update code

plugins/hook.py

@@ -1,13 +1,18 @@
 from nonebot.matcher import Matcher
-from nonebot.message import run_preprocessor, IgnoredException
+from nonebot.message import run_preprocessor, run_postprocessor, IgnoredException
 from nonebot.adapters.cqhttp.exception import ActionFailed
+from models.group_member_info import GroupInfoUser
+from models.friend_user import FriendUser
+from typing import Optional
 from nonebot.typing import T_State
 from nonebot.adapters.cqhttp import (
     Bot,
+    Event,
     MessageEvent,
     PrivateMessageEvent,
     GroupMessageEvent,
-    PokeNotifyEvent
+    PokeNotifyEvent,
+    Message,
 )
 from configs.config import (
     BAN_RESULT,
@@ -17,24 +22,43 @@ from configs.config import (
     MALICIOUS_BAN_COUNT,
     CHECK_NOTICE_INFO_CD,
     plugins2info_dict,
+    plugins2cd_dict,
+    plugins2exists_dict,
 )
 from models.ban_user import BanUser
-from utils.utils import is_number, static_flmt, BanCheckLimiter
+from utils.utils import (
+    is_number,
+    static_flmt,
+    BanCheckLimiter,
+    FreqLimiter,
+    UserExistLimiter,
+)
+from utils.static_data import withdraw_message_id_manager
 from utils.message_builder import at
 from services.log import logger
 from models.level_user import LevelUser
 from utils.static_data import group_manager
-from utils.utils import FreqLimiter
+import asyncio
 
 try:
     import ujson as json
 except ModuleNotFoundError:
     import json
 
+withdraw_message_id_manager["message_id"] = []
+
 
 # 检查是否被ban
 @run_preprocessor
 async def _(matcher: Matcher, bot: Bot, event: MessageEvent, state: T_State):
+    try:
+        if (
+            await BanUser.is_super_ban(event.user_id)
+            and str(event.user_id) not in bot.config.superusers
+        ):
+            raise IgnoredException("用户处于超级黑名单中")
+    except AttributeError:
+        pass
     if not isinstance(event, MessageEvent):
         return
     if matcher.type == "message" and matcher.priority not in [1, 9]:
@@ -48,9 +72,9 @@ async def _(matcher: Matcher, bot: Bot, event: MessageEvent, state: T_State):
                 if time < 60:
                     time = str(time) + " 秒"
                 else:
-                    time = str(int(time / 60)) + " 分"
+                    time = str(int(time / 60)) + " 分钟"
             else:
-                time = str(time) + " 分"
+                time = str(time) + " 分钟"
             if isinstance(event, GroupMessageEvent):
                 if not static_flmt.check(event.user_id):
                     raise IgnoredException("用户处于黑名单中")
@@ -61,7 +85,7 @@ async def _(matcher: Matcher, bot: Bot, event: MessageEvent, state: T_State):
                             group_id=event.group_id,
                             message=at(event.user_id)
                             + BAN_RESULT
-                            + f" 在..在 {time}后才会理你喔",
+                            + f" 在..在 {time} 后才会理你喔",
                         )
                     except ActionFailed:
                         pass
@@ -127,11 +151,16 @@ _flmt = FreqLimiter(CHECK_NOTICE_INFO_CD)
 _flmt_g = FreqLimiter(CHECK_NOTICE_INFO_CD)
 _flmt_s = FreqLimiter(CHECK_NOTICE_INFO_CD)
 _flmt_c = FreqLimiter(CHECK_NOTICE_INFO_CD)
+_exists_msg = {}
+
+
+ignore_rst_module = ["ai", "poke"]
 
 
 # 权限检测
 @run_preprocessor
 async def _(matcher: Matcher, bot: Bot, event: MessageEvent, state: T_State):
+    global _exists_msg
     if (
         (not isinstance(event, MessageEvent) and matcher.module != "poke")
         or await BanUser.is_ban(event.user_id)
@@ -170,54 +199,70 @@ async def _(matcher: Matcher, bot: Bot, event: MessageEvent, state: T_State):
                 raise IgnoredException("权限不足")
     if module in plugins2info_dict.keys() and matcher.priority not in [1, 9]:
         # 戳一戳单独判断
-        if isinstance(event, GroupMessageEvent) or (isinstance(event, PokeNotifyEvent) and event.group_id):
+        if isinstance(event, GroupMessageEvent) or (
+            isinstance(event, PokeNotifyEvent) and event.group_id
+        ):
+            if _exists_msg.get(event.group_id) is None:
+                _exists_msg[event.group_id] = False
             # 群权限
             if plugins2info_dict[module]["level"] > group_manager.get_group_level(
                 str(event.group_id)
             ):
                 try:
-                    if _flmt_g.check(event.user_id):
+                    if _flmt_g.check(event.user_id) and module not in ignore_rst_module:
                         _flmt_g.start_cd(event.user_id)
                         await bot.send_group_msg(
                             group_id=event.group_id, message="群权限不足..."
                         )
                 except ActionFailed:
                     pass
+                _exists_msg[event.group_id] = True
                 raise IgnoredException("群权限不足")
             # 插件状态
             if not group_manager.get_plugin_status(module, str(event.group_id)):
                 try:
-                    if module != "poke" and _flmt_s.check(event.group_id):
+                    if module not in ignore_rst_module and _flmt_s.check(
+                        event.group_id
+                    ):
                         _flmt_s.start_cd(event.group_id)
                         await bot.send_group_msg(
                             group_id=event.group_id, message="该群未开启此功能.."
                         )
                 except ActionFailed:
                     pass
+                _exists_msg[event.group_id] = True
                 raise IgnoredException("未开启此功能...")
             # 管理员禁用
             if not group_manager.get_plugin_status(
                 f"{module}:super", str(event.group_id)
             ):
                 try:
-                    if _flmt_s.check(event.group_id):
+                    if (
+                        _flmt_s.check(event.group_id)
+                        and module not in ignore_rst_module
+                    ):
                         _flmt_s.start_cd(event.group_id)
                         await bot.send_group_msg(
                             group_id=event.group_id, message="管理员禁用了此群该功能..."
                         )
                 except ActionFailed:
                     pass
+                _exists_msg[event.group_id] = True
                 raise IgnoredException("管理员禁用了此群该功能...")
             # 群聊禁用
             if not group_manager.get_plugin_status(module, block_type="group"):
                 try:
-                    if _flmt_c.check(event.group_id):
+                    if (
+                        _flmt_c.check(event.group_id)
+                        and module not in ignore_rst_module
+                    ):
                         _flmt_c.start_cd(event.group_id)
                         await bot.send_group_msg(
                             group_id=event.group_id, message="该功能在群聊中已被禁用..."
                         )
                 except ActionFailed:
                     pass
+                _exists_msg[event.group_id] = True
                 raise IgnoredException("该插件在群聊中已被禁用...")
         else:
             # 私聊禁用
@@ -238,21 +283,187 @@ async def _(matcher: Matcher, bot: Bot, event: MessageEvent, state: T_State):
             ) and group_manager.check_group_is_white(event.group_id):
                 return
             try:
-                if _flmt_c.check(event.user_id):
-                    _flmt_c.start_cd(event.user_id)
-                    if isinstance(event, GroupMessageEvent):
+                if isinstance(event, GroupMessageEvent):
+                    if (
+                        _flmt_c.check(event.group_id)
+                        and module not in ignore_rst_module
+                    ):
+                        _flmt_c.start_cd(event.group_id)
                         await bot.send_group_msg(
                             group_id=event.group_id, message="此功能正在维护..."
                         )
-                    else:
-                        await bot.send_private_msg(
-                            user_id=event.user_id, message="此功能正在维护..."
-                        )
+                else:
+                    await bot.send_private_msg(
+                        user_id=event.user_id, message="此功能正在维护..."
+                    )
             except ActionFailed:
                 pass
+            if isinstance(event, GroupMessageEvent):
+                _exists_msg[event.group_id] = True
             raise IgnoredException("此功能正在维护...")
 
 
+check_flmt = {}
+for plugin in plugins2cd_dict.keys():
+    if plugins2cd_dict[plugin]["status"]:
+        check_flmt[plugin] = FreqLimiter(plugins2cd_dict[plugin]["cd"])
+
+
+check_elmt = {}
+for plugin in plugins2exists_dict.keys():
+    if plugins2exists_dict[plugin]["status"]:
+        check_elmt[plugin] = UserExistLimiter()
+
+
+# 命令cd 和 命令阻塞
+@run_preprocessor
+async def _(matcher: Matcher, bot: Bot, event: MessageEvent, state: T_State):
+    global _exists_msg
+    if not isinstance(event, MessageEvent) and matcher.module != "poke":
+        return
+    module = matcher.module
+    if isinstance(event, GroupMessageEvent) and _exists_msg.get(event.group_id) is None:
+        _exists_msg[event.group_id] = False
+    if module in plugins2cd_dict.keys() and module in check_flmt.keys():
+        if (
+            (
+                isinstance(event, PrivateMessageEvent)
+                and plugins2cd_dict[module]["check_type"] == "private"
+            )
+            or (
+                isinstance(event, GroupMessageEvent)
+                and plugins2cd_dict[module]["check_type"] == "group"
+            )
+            or plugins2cd_dict[module]["check_type"] == "all"
+        ) and plugins2cd_dict[module]["cd"] > 0:
+            cd_type_ = event.user_id
+            if plugins2cd_dict[module]["limit_type"] == "group" and isinstance(
+                event, GroupMessageEvent
+            ):
+                cd_type_ = event.group_id
+            if not check_flmt[module].check(cd_type_):
+                rst = plugins2cd_dict[module]["rst"]
+                if rst:
+                    rst = await init_rst(rst, event)
+                    await send_msg(rst, bot, event)
+                raise IgnoredException(f"{module} 正在cd中...")
+            else:
+                check_flmt[module].start_cd(cd_type_)
+    if module in plugins2exists_dict.keys() and module in check_elmt.keys():
+        if (
+            (
+                isinstance(event, PrivateMessageEvent)
+                and plugins2exists_dict[module]["check_type"] == "private"
+            )
+            or (
+                isinstance(event, GroupMessageEvent)
+                and plugins2exists_dict[module]["check_type"] == "group"
+            )
+            or plugins2exists_dict[module]["check_type"] == "all"
+        ):
+            exists_type_ = event.user_id
+            if plugins2exists_dict[module]["limit_type"] == "group" and isinstance(
+                event, GroupMessageEvent
+            ):
+                exists_type_ = event.group_id
+            if check_elmt[module].check(exists_type_):
+                rst = plugins2exists_dict[module]["rst"]
+                if rst:
+                    rst = await init_rst(rst, event)
+                    await send_msg(rst, bot, event)
+                raise IgnoredException(f"{event.user_id}正在调用{module}....")
+            else:
+                check_elmt[module].set_true(exists_type_)
+
+
+async def send_msg(rst: str, bot: Bot, event: MessageEvent):
+    """
+    发送信息
+    :param rst: pass
+    :param bot: pass
+    :param event: pass
+    """
+    global _exists_msg
+    rst = await init_rst(rst, event)
+    try:
+        if isinstance(event, GroupMessageEvent):
+            _exists_msg[event.group_id] = True
+            await bot.send_group_msg(group_id=event.group_id, message=Message(rst))
+        else:
+            _exists_msg[event.user_id] = True
+            await bot.send_private_msg(user_id=event.user_id, message=Message(rst))
+    except ActionFailed:
+        pass
+
+
+@run_postprocessor
+async def _(
+    matcher: Matcher,
+    exception: Optional[Exception],
+    bot: Bot,
+    event: Event,
+    state: T_State,
+):
+    if not isinstance(event, MessageEvent) and matcher.module != "poke":
+        return
+    module = matcher.module
+    if module in plugins2exists_dict.keys() and module in check_elmt.keys():
+        if not (
+            (
+                isinstance(event, GroupMessageEvent)
+                and plugins2exists_dict[module]["check_type"] == "private"
+            )
+            or (
+                isinstance(event, PrivateMessageEvent)
+                and plugins2exists_dict[module]["check_type"] == "group"
+            )
+        ):
+            exists_type_ = event.user_id
+            if plugins2exists_dict[module]["limit_type"] == "group" and isinstance(
+                event, GroupMessageEvent
+            ):
+                exists_type_ = event.group_id
+            check_elmt[module].set_false(exists_type_)
+
+
+async def init_rst(rst: str, event: MessageEvent):
+    if "[uname]" in rst:
+        uname = event.sender.card if event.sender.card else event.sender.nickname
+        rst = rst.replace("[uname]", uname)
+    if "[nickname]" in rst:
+        if isinstance(event, GroupMessageEvent):
+            nickname = await GroupInfoUser.get_group_member_nickname(
+                event.user_id, event.group_id
+            )
+        else:
+            nickname = await FriendUser.get_friend_nickname(event.user_id)
+        rst = rst.replace("[nickname]", nickname)
+    if "[at]" in rst and isinstance(event, GroupMessageEvent):
+        rst = rst.replace("[at]", str(at(event.user_id)))
+    return rst
+
+
+# 消息撤回
+@run_postprocessor
+async def _(
+    matcher: Matcher,
+    exception: Optional[Exception],
+    bot: Bot,
+    event: Event,
+    state: T_State,
+):
+    tasks = []
+    for id_, time in withdraw_message_id_manager["message_id"]:
+        tasks.append(asyncio.ensure_future(_withdraw_message(bot, id_, time)))
+        withdraw_message_id_manager["message_id"].remove((id_, time))
+    await asyncio.gather(*tasks)
+
+
+async def _withdraw_message(bot: Bot, id_: int, time: int):
+    await asyncio.sleep(time)
+    await bot.delete_msg(message_id=id_, self_id=int(bot.self_id))
+
+
 # 为什么AI会自己和自己聊天
 @run_preprocessor
 async def _(matcher: Matcher, bot: Bot, event: PrivateMessageEvent, state: T_State):
@@ -265,8 +476,19 @@ async def _(matcher: Matcher, bot: Bot, event: PrivateMessageEvent, state: T_Sta
 # 有命令就别说话了
 @run_preprocessor
 async def _(matcher: Matcher, bot: Bot, event: MessageEvent, state: T_State):
+    global _exists_msg
     if not isinstance(event, MessageEvent):
         return
     if matcher.type == "message":
-        if state["_prefix"]["raw_command"] and matcher.module == "ai":
+        if matcher.module == "ai":
+            if (
+                isinstance(event, GroupMessageEvent)
+                and _exists_msg.get(event.group_id) is True
+            ):
+                _exists_msg[event.group_id] = False
+            elif (
+                isinstance(event, PrivateMessageEvent)
+                and _exists_msg.get(event.user_id) is True
+            ):
+                _exists_msg[event.user_id] = False
             raise IgnoredException("有命令就别说话了")