init🎉: 首次提交

zhenxun/builtin_plugins/record_request.py

@@ -0,0 +1,182 @@
+import time
+from datetime import datetime
+from typing import Dict
+
+from nonebot import on_message, on_request
+from nonebot.adapters.onebot.v11 import (
+    ActionFailed,
+    Bot,
+    FriendRequestEvent,
+    GroupRequestEvent,
+)
+from nonebot.plugin import PluginMetadata
+from nonebot_plugin_apscheduler import scheduler
+from nonebot_plugin_saa import TargetQQPrivate, Text
+from nonebot_plugin_session import EventSession
+
+from zhenxun.configs.config import NICKNAME, Config
+from zhenxun.configs.utils import PluginExtraData, RegisterConfig
+from zhenxun.models.fg_request import FgRequest
+from zhenxun.models.friend_user import FriendUser
+from zhenxun.models.group_info import GroupInfo
+from zhenxun.services.log import logger
+from zhenxun.utils.enum import PluginType, RequestType
+
+base_config = Config.get("invite_manager")
+
+__plugin_meta__ = PluginMetadata(
+    name="记录请求",
+    description="自定义群欢迎消息",
+    usage="",
+    extra=PluginExtraData(
+        author="HibiKier",
+        version="0.1",
+        plugin_type=PluginType.HIDDEN,
+        configs=[
+            RegisterConfig(
+                module="invite_manager",
+                key="AUTO_ADD_FRIEND",
+                value=False,
+                help="是否自动同意好友添加",
+                type=bool,
+                default_value=False,
+            )
+        ],
+    ).dict(),
+)
+
+
+class Timer:
+    data: Dict[str, float] = {}
+
+    @classmethod
+    def check(cls, uid: int | str):
+        if uid not in cls.data:
+            return True
+        return time.time() - cls.data[uid] > 5 * 60
+
+    @classmethod
+    def clear(cls):
+        now = time.time()
+        cls.data = {k: v for k, v in cls.data.items() if v - now < 5 * 60}
+
+
+friend_req = on_request(priority=5, block=True)
+group_req = on_request(priority=5, block=True)
+_t = on_message(priority=999, block=False, rule=lambda: False)
+
+
+@friend_req.handle()
+async def _(bot: Bot, event: FriendRequestEvent, session: EventSession):
+    if event.user_id and Timer.check(event.user_id):
+        logger.debug(f"收录好友请求...", "好友请求", target=event.user_id)
+        user = await bot.get_stranger_info(user_id=event.user_id)
+        nickname = user["nickname"]
+        # sex = user["sex"]
+        # age = str(user["age"])
+        comment = event.comment
+        superuser = int(list(bot.config.superusers)[0])
+        await Text(
+            f"*****一份好友申请*****\n"
+            f"昵称：{nickname}({event.user_id})\n"
+            f"自动同意：{'√' if base_config.get('AUTO_ADD_FRIEND') else '×'}\n"
+            f"日期：{str(datetime.now()).split('.')[0]}\n"
+            f"备注：{event.comment}"
+        ).send_to(target=TargetQQPrivate(user_id=superuser), bot=bot)
+        if base_config.get("AUTO_ADD_FRIEND"):
+            logger.debug(f"已开启好友请求自动同意，成功通过该请求", "好友请求", target=event.user_id)
+            await bot.set_friend_add_request(flag=event.flag, approve=True)
+            await FriendUser.create(
+                user_id=str(user["user_id"]), user_name=user["nickname"]
+            )
+        else:
+            await FgRequest.create(
+                request_type=RequestType.FRIEND,
+                platform=session.platform,
+                bot_id=bot.self_id,
+                flag=event.flag,
+                user_id=event.user_id,
+                nickname=nickname,
+                comment=comment,
+            )
+    else:
+        logger.debug(f"好友请求五分钟内重复, 已忽略", "好友请求", target=event.user_id)
+
+
+@group_req.handle()
+async def _(bot: Bot, event: GroupRequestEvent, session: EventSession):
+    # 邀请
+    if event.sub_type == "invite":
+        if str(event.user_id) in bot.config.superusers:
+            try:
+                logger.debug(
+                    f"超级用户自动同意加入群聊",
+                    "群聊请求",
+                    session=event.user_id,
+                    target=event.group_id,
+                )
+                await bot.set_group_add_request(
+                    flag=event.flag, sub_type="invite", approve=True
+                )
+                group_info = await bot.get_group_info(group_id=event.group_id)
+                await GroupInfo.update_or_create(
+                    group_id=str(group_info["group_id"]),
+                    defaults={
+                        "group_name": group_info["group_name"],
+                        "max_member_count": group_info["max_member_count"],
+                        "member_count": group_info["member_count"],
+                        "group_flag": 1,
+                    },
+                )
+            except ActionFailed as e:
+                logger.error(
+                    "超级用户自动同意加入群聊发生错误",
+                    "群聊请求",
+                    session=event.user_id,
+                    target=event.group_id,
+                    e=e,
+                )
+        else:
+            if Timer.check(f"{event.user_id}:{event.group_id}"):
+                logger.debug(
+                    f"收录 用户[{event.user_id}] 群聊[{event.group_id}] 群聊请求",
+                    "群聊请求",
+                    target=event.group_id,
+                )
+                user = await bot.get_stranger_info(user_id=event.user_id)
+                nickname = await FriendUser.get_user_name(event.user_id)
+                superuser = int(list(bot.config.superusers)[0])
+                await Text(
+                    f"*****一份入群申请*****\n"
+                    f"申请人：{nickname}({event.user_id})\n"
+                    f"群聊：{event.group_id}\n"
+                    f"邀请日期：{datetime.now().replace(microsecond=0)}"
+                ).send_to(target=TargetQQPrivate(user_id=superuser), bot=bot)
+                await bot.send_private_msg(
+                    user_id=event.user_id,
+                    message=f"想要邀请我偷偷入群嘛~已经提醒{NICKNAME}的管理员大人了\n"
+                    "请确保已经群主或群管理沟通过！\n"
+                    "等待管理员处理吧！",
+                )
+                await FgRequest.create(
+                    request_type=RequestType.FRIEND,
+                    platform=session.platform,
+                    bot_id=bot.self_id,
+                    flag=event.flag,
+                    user_id=event.user_id,
+                    nickname=nickname,
+                )
+            else:
+                logger.debug(
+                    f"群聊请求五分钟内重复, 已忽略",
+                    "群聊请求",
+                    target=f"{event.user_id}:{event.group_id}",
+                )
+
+
+@scheduler.scheduled_job(
+    "interval",
+    minutes=5,
+)
+async def _():
+    Timer.clear()