mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-14 21:52:56 +08:00
276 lines
10 KiB
Python
276 lines
10 KiB
Python
from io import BytesIO
|
|
|
|
from arclet.alconna import Args, Option
|
|
from arclet.alconna.typing import CommandMeta
|
|
from nonebot.adapters import Bot
|
|
from nonebot.permission import SUPERUSER
|
|
from nonebot.plugin import PluginMetadata
|
|
from nonebot.rule import to_me
|
|
from nonebot_plugin_alconna import (
|
|
Alconna,
|
|
AlconnaQuery,
|
|
Arparma,
|
|
Query,
|
|
on_alconna,
|
|
store_true,
|
|
)
|
|
from nonebot_plugin_session import EventSession
|
|
|
|
from zhenxun.configs.path_config import IMAGE_PATH
|
|
from zhenxun.configs.utils import PluginExtraData
|
|
from zhenxun.models.fg_request import FgRequest
|
|
from zhenxun.services.log import logger
|
|
from zhenxun.utils.enum import PluginType, RequestHandleType, RequestType
|
|
from zhenxun.utils.exception import NotFoundError
|
|
from zhenxun.utils.image_utils import BuildImage
|
|
from zhenxun.utils.message import MessageUtils
|
|
from zhenxun.utils.utils import get_user_avatar
|
|
|
|
usage = """
|
|
查看请求
|
|
清空请求
|
|
请求处理 -fa [id] / 同意好友请求 [id] # 同意好友请求
|
|
请求处理 -fr [id] / 拒绝好友请求 [id] # 拒绝好友请求
|
|
请求处理 -fi [id] / 忽略好友请求 [id] # 忽略好友请求
|
|
请求处理 -ga [id] / 同意群组请求 [id] # 同意群聊请求
|
|
请求处理 -gr [id] / 拒绝群组请求 [id] # 拒绝群聊请求
|
|
请求处理 -gi [id] / 忽略群组请求 [id] # 忽略群聊请求
|
|
""".strip()
|
|
|
|
|
|
__plugin_meta__ = PluginMetadata(
|
|
name="请求处理",
|
|
description="好友与邀请群组请求处理",
|
|
usage=usage,
|
|
extra=PluginExtraData(
|
|
author="HibiKier",
|
|
version="0.1",
|
|
plugin_type=PluginType.SUPERUSER,
|
|
).to_dict(),
|
|
)
|
|
|
|
|
|
_req_matcher = on_alconna(
|
|
Alconna(
|
|
"请求处理",
|
|
Args["handle", ["-fa", "-fr", "-fi", "-ga", "-gr", "-gi"]]["id", int],
|
|
meta=CommandMeta(
|
|
description="好友/群组请求处理",
|
|
usage=usage,
|
|
example="同意好友请求 20",
|
|
compact=True,
|
|
),
|
|
),
|
|
permission=SUPERUSER,
|
|
priority=1,
|
|
rule=to_me(),
|
|
block=True,
|
|
)
|
|
|
|
_read_matcher = on_alconna(
|
|
Alconna(
|
|
"查看请求",
|
|
Option("-f|--friend", action=store_true, help_text="查看好友请求"),
|
|
Option("-g|--group", action=store_true, help_text="查看群组请求"),
|
|
meta=CommandMeta(
|
|
description="查看所有请求或好友群组请求",
|
|
usage="查看请求\n查看请求 -f\n查看请求-g",
|
|
example="查看请求 -f",
|
|
compact=True,
|
|
),
|
|
),
|
|
permission=SUPERUSER,
|
|
priority=1,
|
|
rule=to_me(),
|
|
block=True,
|
|
)
|
|
|
|
_clear_matcher = on_alconna(
|
|
Alconna(
|
|
"清空请求",
|
|
Option("-f|--friend", action=store_true, help_text="清空好友请求"),
|
|
Option("-g|--group", action=store_true, help_text="清空群组请求"),
|
|
meta=CommandMeta(
|
|
description="清空请求",
|
|
usage="清空请求\n清空请求 -f\n清空请求-g",
|
|
example="清空请求 -f",
|
|
compact=True,
|
|
),
|
|
),
|
|
permission=SUPERUSER,
|
|
priority=1,
|
|
rule=to_me(),
|
|
block=True,
|
|
)
|
|
|
|
reg_arg_list = [
|
|
(r"同意好友请求", ["-fa", "{%0}"]),
|
|
(r"拒绝好友请求", ["-fr", "{%0}"]),
|
|
(r"忽略好友请求", ["-fi", "{%0}"]),
|
|
(r"同意群组请求", ["-ga", "{%0}"]),
|
|
(r"拒绝群组请求", ["-gr", "{%0}"]),
|
|
(r"忽略群组请求", ["-gi", "{%0}"]),
|
|
]
|
|
|
|
for r in reg_arg_list:
|
|
_req_matcher.shortcut(
|
|
r[0],
|
|
command="请求处理",
|
|
arguments=r[1],
|
|
prefix=True,
|
|
)
|
|
|
|
|
|
@_req_matcher.handle()
|
|
async def _(
|
|
bot: Bot,
|
|
session: EventSession,
|
|
handle: str,
|
|
id: int,
|
|
arparma: Arparma,
|
|
):
|
|
type_dict = {
|
|
"a": RequestHandleType.APPROVE,
|
|
"r": RequestHandleType.REFUSED,
|
|
"i": RequestHandleType.IGNORE,
|
|
}
|
|
handle_type = type_dict[handle[-1]]
|
|
try:
|
|
if handle_type == RequestHandleType.APPROVE:
|
|
await FgRequest.approve(bot, id)
|
|
if handle_type == RequestHandleType.REFUSED:
|
|
await FgRequest.refused(bot, id)
|
|
if handle_type == RequestHandleType.IGNORE:
|
|
await FgRequest.ignore(id)
|
|
except NotFoundError:
|
|
await MessageUtils.build_message("未发现此id的请求...").finish(reply_to=True)
|
|
except Exception:
|
|
await MessageUtils.build_message("其他错误, 可能flag已失效...").finish(
|
|
reply_to=True
|
|
)
|
|
logger.info("处理请求", arparma.header_result, session=session)
|
|
await MessageUtils.build_message("成功处理请求!").finish(reply_to=True)
|
|
|
|
|
|
@_read_matcher.handle()
|
|
async def _(
|
|
session: EventSession,
|
|
arparma: Arparma,
|
|
is_friend: Query[bool] = AlconnaQuery("friend.value", False),
|
|
is_group: Query[bool] = AlconnaQuery("group.value", False),
|
|
):
|
|
if all_request := await FgRequest.filter(handle_type__isnull=True).all():
|
|
req_list = list(all_request)
|
|
req_list.reverse()
|
|
friend_req: list[FgRequest] = []
|
|
group_req: list[FgRequest] = []
|
|
for req in req_list:
|
|
if req.request_type == RequestType.FRIEND:
|
|
friend_req.append(req)
|
|
else:
|
|
group_req.append(req)
|
|
if is_friend.result:
|
|
group_req = []
|
|
elif is_group.result:
|
|
friend_req = []
|
|
req_image_list: list[BuildImage] = []
|
|
for i, req_list in enumerate([friend_req, group_req]):
|
|
img_list = []
|
|
for req in req_list:
|
|
content = await get_user_avatar(req.user_id)
|
|
ava_img = BuildImage(
|
|
80, 80, background=BytesIO(content) if content else None
|
|
)
|
|
await ava_img.circle()
|
|
handle_img = BuildImage(
|
|
130, 32, font_size=15, color="#EEEFF4", font="HYWenHei-85W.ttf"
|
|
)
|
|
await handle_img.text((0, 0), "同意/拒绝/忽略", center_type="center")
|
|
await handle_img.circle_corner(10)
|
|
background = BuildImage(500, 100, font_size=22, color=(255, 255, 255))
|
|
await background.paste(ava_img, (55, 0), center_type="height")
|
|
if session.platform and session.platform != "unknown":
|
|
platform_icon = BuildImage(
|
|
30,
|
|
30,
|
|
background=IMAGE_PATH / "_icon" / f"{session.platform}.png",
|
|
)
|
|
await background.paste(platform_icon, (46, 10))
|
|
await background.text((150, 12), req.nickname)
|
|
if i == 0:
|
|
comment_img = await BuildImage.build_text_image(
|
|
f"对方留言：{req.comment}", size=15, font_color=(140, 140, 143)
|
|
)
|
|
else:
|
|
comment_img = await BuildImage.build_text_image(
|
|
f"群组：{req.group_id}", size=15, font_color=(140, 140, 143)
|
|
)
|
|
await background.paste(comment_img, (150, 65))
|
|
tag = await BuildImage.build_text_image(
|
|
f"{req.platform}",
|
|
size=13,
|
|
color=(0, 167, 250),
|
|
font="HYWenHei-85W.ttf",
|
|
font_color=(255, 255, 255),
|
|
padding=(1, 6, 1, 6),
|
|
)
|
|
await tag.circle_corner(5)
|
|
await background.paste(tag, (150, 42))
|
|
await background.paste(handle_img, (360, 35))
|
|
_id_img = BuildImage(
|
|
32, 32, font_size=15, color="#EEEFF4", font="HYWenHei-85W.ttf"
|
|
)
|
|
await _id_img.text((0, 0), f"{req.id}", center_type="center")
|
|
await _id_img.circle_corner(10)
|
|
await background.paste(_id_img, (10, 0), center_type="height")
|
|
img_list.append(background)
|
|
if img_list:
|
|
A = await BuildImage.auto_paste(img_list, 1)
|
|
result_image = BuildImage(
|
|
A.width, A.height + 30, color=(255, 255, 255), font_size=20
|
|
)
|
|
await result_image.paste(A, (0, 30))
|
|
_type_text = "好友请求" if i == 0 else "群组请求"
|
|
await result_image.text((15, 13), _type_text, fill=(140, 140, 143))
|
|
req_image_list.append(result_image)
|
|
if not req_image_list:
|
|
await MessageUtils.build_message("没有任何请求喔...").finish(reply_to=True)
|
|
if len(req_image_list) == 1:
|
|
await MessageUtils.build_message(req_image_list[0]).finish()
|
|
width = sum(img.width for img in req_image_list)
|
|
height = max(img.height for img in req_image_list)
|
|
background = BuildImage(width, height)
|
|
await background.paste(req_image_list[0])
|
|
await req_image_list[1].line((0, 10, 1, req_image_list[1].height - 10), width=1)
|
|
await background.paste(req_image_list[1], (req_image_list[1].width, 0))
|
|
logger.info("查看请求", arparma.header_result, session=session)
|
|
await MessageUtils.build_message(background).finish()
|
|
await MessageUtils.build_message("没有任何请求喔...").finish(reply_to=True)
|
|
|
|
|
|
@_clear_matcher.handle()
|
|
async def _(
|
|
session: EventSession,
|
|
arparma: Arparma,
|
|
is_friend: Query[bool] = AlconnaQuery("friend.value", False),
|
|
is_group: Query[bool] = AlconnaQuery("group.value", False),
|
|
):
|
|
_type = ""
|
|
if is_friend.result:
|
|
_type = "好友"
|
|
await FgRequest.filter(
|
|
handle_type__isnull=True, request_type=RequestType.FRIEND
|
|
).update(handle_type=RequestHandleType.IGNORE)
|
|
elif is_group.result:
|
|
_type = "群组"
|
|
await FgRequest.filter(
|
|
handle_type__isnull=True, request_type=RequestType.GROUP
|
|
).update(handle_type=RequestHandleType.IGNORE)
|
|
else:
|
|
_type = "所有"
|
|
await FgRequest.filter(handle_type__isnull=True).update(
|
|
handle_type=RequestHandleType.IGNORE
|
|
)
|
|
logger.info(f"清空{_type}请求", arparma.header_result, session=session)
|
|
await MessageUtils.build_message(f"已清空{_type}请求!").finish()
|