update 0.0.6.0

admin_manager.py

@@ -1,51 +0,0 @@
-from typing import Optional, Dict
-from .data_class import StaticData
-from utils.utils import FreqLimiter
-from services.log import logger
-from pathlib import Path
-
-
-class AdminManager(StaticData):
-    """
-    管理员命令 管理器
-    """
-
-    def __init__(self):
-        super().__init__(None)
-
-    def add_admin_command(self, plugin: str, level: int):
-        """
-        添加一个管理员命令
-        :param plugin: 模块名
-        :param level: 权限等级
-        """
-        self._data[plugin] = level
-
-    def remove_admin_command(self, plugin: str):
-        """
-        删除一个管理员命令
-        :param plugin: 模块名
-        """
-        if plugin in self._data.keys():
-            del self._data[plugin]
-
-    def check(self, plugin: str, level: int) -> bool:
-        """
-        检查是否满足权限
-        :param plugin: 模块名
-        :param level: 权限等级
-        """
-        if plugin in self._data.keys():
-            return level >= self._data[plugin]
-        return True
-
-    def get_plugin_level(self, plugin: str) -> int:
-        """
-        获取插件等级
-        :param plugin: 模块名
-        """
-        if plugin in self._data.keys():
-            return self._data[plugin]
-        return 0
-
-

basic_plugins/admin_bot_manage/__init__.py

@@ -0,0 +1,30 @@
+from configs.config import Config
+import nonebot
+
+
+Config.add_plugin_config(
+    "admin_bot_manage:custom_welcome_message",
+    "SET_GROUP_WELCOME_MESSAGE_LEVEL",
+    2,
+    name="群管理员操作",
+    help_="设置群欢迎消息权限",
+    default_value=2,
+)
+
+Config.add_plugin_config(
+    "admin_bot_manage:switch_rule",
+    "CHANGE_GROUP_SWITCH_LEVEL [LEVEL]",
+    2,
+    help_="开关群功能权限",
+    default_value=2,
+)
+
+Config.add_plugin_config(
+    "admin_bot_manage",
+    "ADMIN_DEFAULT_AUTH",
+    5,
+    help_="默认群管理员权限",
+    default_value=5
+)
+
+nonebot.load_plugins("basic_plugins/admin_bot_manage")

basic_plugins/admin_bot_manage/admin_config.py

@@ -4,7 +4,7 @@ from nonebot.adapters.cqhttp import Bot, GroupAdminNoticeEvent
 from nonebot.typing import T_State
 from models.level_user import LevelUser
 from models.group_member_info import GroupInfoUser
-from configs.config import ADMIN_DEFAULT_AUTH
+from configs.config import Config
 
 
 __zx_plugin_name__ = "群管理员变动监测 [Hidden]"
@@ -24,8 +24,15 @@ async def _(bot: Bot, event: GroupAdminNoticeEvent, state: T_State):
     except AttributeError:
         nickname = event.user_id
     if event.sub_type == "set":
-        await LevelUser.set_level(event.user_id, event.group_id, ADMIN_DEFAULT_AUTH)
-        logger.info(f"为新晋管理员 {nickname}({event.user_id}) 添加权限等级：{ADMIN_DEFAULT_AUTH}")
+        await LevelUser.set_level(
+            event.user_id,
+            event.group_id,
+            Config.get_config("admin_bot_manage", "ADMIN_DEFAULT_AUTH"),
+        )
+        logger.info(
+            f"为新晋管理员 {nickname}({event.user_id}) "
+            f"添加权限等级：{Config.get_config('admin_bot_manage', 'ADMIN_DEFAULT_AUTH')}"
+        )
     elif event.sub_type == "unset":
         await LevelUser.delete_level(event.user_id, event.group_id)
         logger.info(f"将非管理员 {nickname}({event.user_id}) 取消权限等级")

basic_plugins/admin_bot_manage/custom_welcome_message.py

@@ -4,7 +4,7 @@ from nonebot.typing import T_State
 from nonebot.adapters.cqhttp import Bot, GroupMessageEvent
 from .data_source import custom_group_welcome
 from nonebot.adapters.cqhttp.permission import GROUP
-from configs.config import OC_LEVEL
+from configs.config import Config
 from services.log import logger
 
 
@@ -22,7 +22,7 @@ __plugin_cmd__ = ['自定义群欢迎消息 ?[文本] ?[图片]']
 __plugin_version__ = 0.1
 __plugin_author__ = 'HibiKier'
 __plugin_settings__ = {
-    "admin_level": OC_LEVEL,
+    "admin_level": Config.get_config("admin_bot_manage", "SET_GROUP_WELCOME_MESSAGE_LEVEL"),
 }
 
 custom_welcome = on_command(

basic_plugins/admin_bot_manage/data_source.py

@@ -9,8 +9,8 @@ from models.group_member_info import GroupInfoUser
 from datetime import datetime
 from services.db_context import db
 from models.level_user import LevelUser
-from configs.config import ADMIN_DEFAULT_AUTH
-from utils.manager import group_manager, plugins2settings_manager
+from configs.config import Config
+from utils.manager import group_manager, plugins2settings_manager, plugins_manager
 from utils.image_utils import CreateImg
 import aiofiles
 import aiohttp
@@ -103,9 +103,9 @@ async def change_group_switch(cmd: str, group_id: int, is_super: bool = False):
     group_help_file = Path(DATA_PATH) / "group_help" / f"{group_id}.png"
     status = cmd[:2]
     cmd = cmd[2:]
-    type_ = 'plugin'
+    type_ = "plugin"
     modules = plugins2settings_manager.get_plugin_module(cmd, True)
-    if cmd == '全部被动':
+    if cmd == "全部被动":
         for task in task_data:
             if status == "开启":
                 if not await group_manager.check_group_task_status(group_id, task):
@@ -115,13 +115,13 @@ async def change_group_switch(cmd: str, group_id: int, is_super: bool = False):
                     await group_manager.close_group_task(group_id, task)
         return f"已 {status} 全部被动技能！"
     if cmd in [task_data[x] for x in task_data.keys()]:
-        type_ = 'task'
+        type_ = "task"
         modules = [x for x in task_data.keys() if task_data[x] == cmd]
     for module in modules:
         if is_super:
             module = f"{module}:super"
         if status == "开启":
-            if type_ == 'task':
+            if type_ == "task":
                 if await group_manager.check_group_task_status(group_id, module):
                     return f"被动 {task_data[module]} 正处于开启状态！不要重复开启."
                 await group_manager.open_group_task(group_id, module)
@@ -130,7 +130,7 @@ async def change_group_switch(cmd: str, group_id: int, is_super: bool = False):
                     return f"功能 {cmd} 正处于开启状态！不要重复开启."
                 group_manager.unblock_plugin(module, group_id)
         else:
-            if type_ == 'task':
+            if type_ == "task":
                 if not await group_manager.check_group_task_status(group_id, module):
                     return f"被动 {task_data[module]} 正处于关闭状态！不要重复关闭."
                 await group_manager.close_group_task(group_id, module)
@@ -140,6 +140,14 @@ async def change_group_switch(cmd: str, group_id: int, is_super: bool = False):
                 group_manager.block_plugin(module, group_id)
         if group_help_file.exists():
             group_help_file.unlink()
+    if is_super:
+        for file in os.listdir(Path(DATA_PATH) / 'group_help'):
+            file = Path(DATA_PATH) / 'group_help' / file
+            file.unlink()
+    else:
+        _help_image = Path(DATA_PATH) / 'group_help' / f"{group_id}.png"
+        if _help_image.exists():
+            _help_image.unlink()
     return f"{status} {cmd} 功能！"
 
 
@@ -153,9 +161,12 @@ def set_plugin_status(cmd: str, block_type: str = "all"):
     cmd = cmd[2:]
     module = plugins2settings_manager.get_plugin_module(cmd)
     if status == "开启":
-        group_manager.unblock_plugin(module)
+        plugins_manager.unblock_plugin(module)
     else:
-        group_manager.block_plugin(module, block_type=block_type)
+        plugins_manager.block_plugin(module, block_type=block_type)
+    for file in os.listdir(Path(DATA_PATH) / 'group_help'):
+        file = Path(DATA_PATH) / 'group_help' / file
+        file.unlink()
 
 
 async def get_plugin_status():
@@ -171,22 +182,32 @@ def _get_plugin_status() -> MessageSegment:
     """
     rst = "\t功能\n"
     flag_str = "状态".rjust(4) + "\n"
-    for module in plugins2settings_manager.get_data():
-        flag = group_manager.get_plugin_block_type(module)
+    for module in plugins_manager.get_data():
+        flag = plugins_manager.get_plugin_block_type(module)
         flag = flag.upper() + " CLOSE" if flag else "OPEN"
         try:
-            rst += f'{plugins2settings_manager.get(module)["cmd"][0]}\n'
-        except IndexError:
-            rst += f"{module}\n"
+            plugin_name = plugins_manager.get(module)["plugin_name"]
+            if (
+                "[Hidden]" in plugin_name
+                or "[Admin]" in plugin_name
+                or "[Superuser]" in plugin_name
+            ):
+                continue
+            rst += f"{plugin_name}"
+        except KeyError:
+            rst += f"{module}"
+        if plugins_manager.get(module)["error"]:
+            rst += "[ERROR]"
+        rst += "\n"
         flag_str += f"{flag}\n"
     height = len(rst.split("\n")) * 24
-    a = CreateImg(150, height, font_size=20)
+    a = CreateImg(250, height, font_size=20)
     a.text((10, 10), rst)
     b = CreateImg(200, height, font_size=20)
     b.text((10, 10), flag_str)
-    A = CreateImg(380, height)
+    A = CreateImg(500, height)
     A.paste(a)
-    A.paste(b, (150, 0))
+    A.paste(b, (270, 0))
     return image(b64=A.pic2bs4())
 
 
@@ -217,7 +238,9 @@ async def update_member_info(group_id: int, remind_superuser: bool = False) -> b
                 and not await LevelUser.is_group_flag(user_info["user_id"], group_id)
             ):
                 await LevelUser.set_level(
-                    user_info["user_id"], user_info["group_id"], ADMIN_DEFAULT_AUTH
+                    user_info["user_id"],
+                    user_info["group_id"],
+                    Config.get_config("admin_bot_manage", "ADMIN_DEFAULT_AUTH"),
                 )
             if str(user_info["user_id"]) in bot.config.superusers:
                 await LevelUser.set_level(

basic_plugins/admin_bot_manage/switch_rule.py

@@ -8,7 +8,7 @@ from .data_source import (
     group_current_status,
 )
 from services.log import logger
-from configs.config import NICKNAME, OC_LEVEL
+from configs.config import NICKNAME, Config
 from utils.utils import get_message_text, is_number
 from nonebot.permission import SUPERUSER
 from .rule import switch_rule
@@ -47,7 +47,8 @@ __plugin_cmd__ = [
 __plugin_version__ = 0.1
 __plugin_author__ = "HibiKier"
 __plugin_settings__ = {
-    "admin_level": OC_LEVEL,
+    "admin_level": Config.get_config("admin_bot_manage", "CHANGE_GROUP_SWITCH_LEVEL"),
+    "cmd": ["开启功能", "关闭功能", "开关"]
 }
 
 switch_rule_matcher = on_message(rule=switch_rule, priority=4, block=True)

basic_plugins/admin_bot_manage/timing_task.py

@@ -2,7 +2,7 @@ from utils.utils import scheduler, get_bot
 from .data_source import update_member_info
 from services.log import logger
 from models.group_info import GroupInfo
-from asyncpg.exceptions import ConnectionDoesNotExistError
+from asyncpg.exceptions import ConnectionDoesNotExistError, UndefinedColumnError
 
 
 __zx_plugin_name__ = '管理方面定时任务 [Hidden]'
@@ -47,5 +47,5 @@ async def _():
                 if g not in all_group:
                     await update_member_info(g, False)
                     logger.info(f"快速更新群信息以及权限：{g}")
-    except (IndexError, ConnectionDoesNotExistError):
+    except (IndexError, ConnectionDoesNotExistError, UndefinedColumnError):
         pass

basic_plugins/admin_help/data_source.py

@@ -18,13 +18,16 @@ admin_help_image = Path(IMAGE_PATH) / 'admin_help_img.png'
 
 
 @driver.on_bot_connect
-async def create_help_image(bot: Bot = None):
-    """
-    创建管理员帮助图片
-    """
+async def init_task(bot: Bot = None):
     if not group_manager.get_task_data():
         await group_manager.init_group_task()
         logger.info(f'已成功加载 {len(group_manager.get_task_data())} 个被动技能.')
+
+
+async def create_help_image():
+    """
+    创建管理员帮助图片
+    """
     await asyncio.get_event_loop().run_in_executor(
         None, _create_help_image
     )

basic_plugins/apscheduler/__init__.py

@@ -8,7 +8,7 @@ from nonebot.adapters.cqhttp.exception import ActionFailed
 from configs.config import NICKNAME
 from utils.manager import group_manager
 
-__zx_plugin_name__ = "群定时任务相关 [Hidden]"
+__zx_plugin_name__ = "定时任务相关 [Hidden]"
 __plugin_version__ = 0.1
 __plugin_author__ = "HibiKier"
 __plugin_task__ = {'zwa': '早晚安'}
@@ -106,7 +106,7 @@ async def _():
         logger.error(f"自动更新群组信息错误 e:{e}")
 
 
-#  一次性任务
+    #  一次性任务
 # 固定时间触发，仅触发一次：
 #
 # from datetime import datetime

basic_plugins/auto_invite/__init__.py

@@ -1,24 +1,38 @@
-from nonebot import on_request
-from nonebot.adapters.cqhttp import Bot, FriendRequestEvent, GroupRequestEvent
+from nonebot import on_request, on_message
+from nonebot.adapters.cqhttp import (
+    Bot,
+    FriendRequestEvent,
+    GroupRequestEvent,
+    MessageEvent,
+)
 from models.friend_user import FriendUser
 from datetime import datetime
-from configs.config import AUTO_ADD_FRIEND, NICKNAME
+from configs.config import NICKNAME, Config
 from nonebot.adapters.cqhttp.exception import ActionFailed
+from utils.manager import requests_manager
+from models.group_info import GroupInfo
 from utils.utils import scheduler
+import asyncio
 import time
+import re
 
 __zx_plugin_name__ = "好友群聊处理请求 [Hidden]"
 __plugin_version__ = 0.1
-__plugin_author__ = 'HibiKier'
-
-friend_req = on_request(priority=5)
+__plugin_author__ = "HibiKier"
+__plugin_configs__ = {
+    "AUTO_ADD_FRIEND": {"value": False, "help": "是否自动同意好友添加", "default_value": False}
+}
 
+friend_req = on_request(priority=5, block=True)
+group_req = on_request(priority=5, block=True)
+x = on_message(priority=5, block=False)
 
 exists_data = {"private": {}, "group": {}}
 
 
 @friend_req.handle()
 async def _(bot: Bot, event: FriendRequestEvent, state: dict):
+    print(event)
     global exists_data
     if exists_data["private"].get(event.user_id):
         if time.time() - exists_data["private"][event.user_id] < 60 * 5:
@@ -26,20 +40,30 @@ async def _(bot: Bot, event: FriendRequestEvent, state: dict):
     exists_data["private"][event.user_id] = time.time()
     user = await bot.get_stranger_info(user_id=event.user_id)
     nickname = user["nickname"]
+    sex = user["sex"]
+    age = str(user["age"])
+    comment = event.comment
     await bot.send_private_msg(
         user_id=int(list(bot.config.superusers)[0]),
         message=f"*****一份好友申请*****\n"
         f"昵称：{nickname}({event.user_id})\n"
-        f"自动同意：{'√' if AUTO_ADD_FRIEND else '×'}\n"
+        f"自动同意：{'√' if Config.get_config('auto_invite', 'AUTO_ADD_FRIEND') else '×'}\n"
         f"日期：{str(datetime.now()).split('.')[0]}\n"
         f"备注：{event.comment}",
     )
-    if AUTO_ADD_FRIEND:
+    if Config.get_config("auto_invite", "AUTO_ADD_FRIEND"):
         await bot.set_friend_add_request(flag=event.flag, approve=True)
         await FriendUser.add_friend_info(user["user_id"], user["nickname"])
-
-
-group_req = on_request(priority=5, block=True)
+    else:
+        requests_manager.add_request(
+            event.user_id,
+            "private",
+            event.flag,
+            nickname=nickname,
+            sex=sex,
+            age=age,
+            comment=comment,
+        )
 
 
 @group_req.handle()
@@ -51,9 +75,23 @@ async def _(bot: Bot, event: GroupRequestEvent, state: dict):
                 await bot.set_group_add_request(
                     flag=event.flag, sub_type="invite", approve=True
                 )
+                if await GroupInfo.get_group_info(event.group_id):
+                    await GroupInfo.set_group_flag(event.group_id, 1)
+                else:
+                    group_info = await bot.get_group_info(group_id=event.group_id)
+                    await GroupInfo.add_group_info(
+                        group_info["group_id"],
+                        group_info["group_name"],
+                        group_info["max_member_count"],
+                        group_info["member_count"],
+                        1,
+                    )
             except ActionFailed:
                 pass
         else:
+            user = await bot.get_stranger_info(user_id=event.user_id)
+            sex = user["sex"]
+            age = str(user["age"])
             if exists_data["group"].get(f"{event.user_id}:{event.group_id}"):
                 if (
                     time.time()
@@ -76,6 +114,31 @@ async def _(bot: Bot, event: GroupRequestEvent, state: dict):
                 "请确保已经群主或群管理沟通过！\n"
                 "等待管理员处理吧！",
             )
+            requests_manager.add_request(
+                event.user_id,
+                "group",
+                event.flag,
+                invite_group=event.group_id,
+                nickname=nickname,
+                sex=sex,
+                age=age,
+            )
+
+
+@x.handle()
+async def _(bot: Bot, event: MessageEvent, state: dict):
+    await asyncio.sleep(0.1)
+    r = re.search(r'groupcode="(.*?)"', str(event.get_message()))
+    if r:
+        group_id = int(r.group(1))
+    else:
+        return
+    r = re.search(r'groupname="(.*?)"', str(event.get_message()))
+    if r:
+        group_name = r.group(1)
+    else:
+        group_name = "None"
+    requests_manager.set_group_name(group_name, group_id)
 
 
 @scheduler.scheduled_job(

basic_plugins/ban/__init__.py

@@ -5,7 +5,7 @@ from nonebot.typing import T_State
 from nonebot.adapters.cqhttp import Bot
 from nonebot.adapters.cqhttp import GroupMessageEvent, PrivateMessageEvent
 from utils.utils import get_message_at, get_message_text, is_number
-from configs.config import NICKNAME, BAN_LEVEL
+from configs.config import NICKNAME, Config
 from nonebot.permission import SUPERUSER
 from services.log import logger
 
@@ -34,7 +34,15 @@ __plugin_cmd__ = ['.ban [at] ?[小时] ?[分钟]', '.unban [at]', 'b了 [at] [_s
 __plugin_version__ = 0.1
 __plugin_author__ = 'HibiKier'
 __plugin_settings__ = {
-    "admin_level": BAN_LEVEL,
+    "admin_level": Config.get_config("ban", "BAN_LEVEL"),
+    "cmd": ['.ban', '.unban', 'ban', 'unban']
+}
+__plugin_configs__ = {
+    "BAN_LEVEL [LEVEL]": {
+        "value": 5,
+        "help": "ban/unban所需要的管理员权限等级",
+        "default_value": 5
+    }
 }
 
 

basic_plugins/group_handle/__init__.py

@@ -10,8 +10,11 @@ from nonebot.adapters.cqhttp import (
     GroupDecreaseNoticeEvent,
 )
 from nonebot.adapters.cqhttp.exception import ActionFailed
-from utils.manager import group_manager, plugins2settings_manager
+from utils.manager import group_manager, plugins2settings_manager, requests_manager
+from configs.config import NICKNAME
 from models.group_info import GroupInfo
+from utils.utils import FreqLimiter
+from configs.config import Config
 from pathlib import Path
 import random
 import os
@@ -25,10 +28,17 @@ except ModuleNotFoundError:
 __zx_plugin_name__ = "群事件处理 [Hidden]"
 __plugin_version__ = 0.1
 __plugin_author__ = "HibiKier"
-__plugin_task__ = {
-    'group_welcome': '进群欢迎',
-    'refund_group_remind': '退群提醒'
-}
+__plugin_task__ = {"group_welcome": "进群欢迎", "refund_group_remind": "退群提醒"}
+Config.add_plugin_config(
+    "auto_invite", "message", f"请不要未经同意就拉{NICKNAME}入群！告辞！", help_="强制拉群后进群回复的内容.."
+)
+Config.add_plugin_config(
+    "auto_invite", "flag", True, help_="被强制拉群后是否直接退出", default_value=True
+)
+Config.add_plugin_config(
+    "auto_invite", "welcome_msg_cd", 5, help_="群欢迎消息cd", default_value=5
+)
+_flmt = FreqLimiter(Config.get_config("auto_invite", "welcome_msg_cd"))
 
 
 # 群员增加处理
@@ -42,7 +52,30 @@ add_group = on_request(priority=1, block=False)
 @group_increase_handle.handle()
 async def _(bot: Bot, event: GroupIncreaseNoticeEvent, state: dict):
     if event.user_id == int(bot.self_id):
-        if event.group_id not in group_manager["group_manager"].keys():
+        group = await GroupInfo.get_group_info(event.group_id)
+        # 群聊不存在或被强制拉群，退出该群
+        if (not group or group.group_flag == 0) and Config.get_config(
+            "auto_invite", "flag"
+        ):
+            try:
+                msg = Config.get_config("auto_invite", "message")
+                if msg:
+                    await bot.send_group_msg(group_id=event.group_id, message=msg)
+                await bot.set_group_leave(group_id=event.group_id)
+                await bot.send_private_msg(
+                    user_id=int(list(bot.config.superusers)[0]),
+                    message=f"触发强制入群保护，已成功退出群聊 {event.group_id}..",
+                )
+                logger.info(f"强制拉群或未有群信息，退出群聊 {group} 成功")
+                requests_manager.remove_request("group", event.group_id)
+            except Exception as e:
+                logger.info(f"强制拉群或未有群信息，退出群聊 {group} 失败 e:{e}")
+                await bot.send_private_msg(
+                    user_id=int(list(bot.config.superusers)[0]),
+                    message=f"触发强制入群保护，退出群聊 {event.group_id} 失败..",
+                )
+        # 默认群功能开关
+        elif event.group_id not in group_manager["group_manager"].keys():
             data = plugins2settings_manager.get_data()
             for plugin in data.keys():
                 if not data[plugin]["default_status"]:
@@ -63,7 +96,10 @@ async def _(bot: Bot, event: GroupIncreaseNoticeEvent, state: dict):
             logger.info(f"用户{user_info['user_id']} 所属{user_info['group_id']} 更新失败")
 
         # 群欢迎消息
-        if await group_manager.check_group_task_status(event.group_id, 'group_welcome'):
+        if await group_manager.check_group_task_status(
+            event.group_id, "group_welcome"
+        ) and _flmt.check(event.group_id):
+            _flmt.start_cd(event.group_id)
             msg = ""
             img = ""
             at_flag = False
@@ -115,6 +151,9 @@ async def _(bot: Bot, event: GroupDecreaseNoticeEvent, state: dict):
             f"日期：{str(datetime.now()).split('.')[0]}",
         )
         return
+    if event.user_id == int(bot.self_id):
+        group_manager.delete_group(event.group_id)
+        return
     try:
         user_name = (
             await GroupInfoUser.get_member_info(event.user_id, event.group_id)
@@ -125,7 +164,9 @@ async def _(bot: Bot, event: GroupDecreaseNoticeEvent, state: dict):
         logger.info(f"用户{user_name}, qq={event.user_id} 所属{event.group_id} 删除成功")
     else:
         logger.info(f"用户{user_name}, qq={event.user_id} 所属{event.group_id} 删除失败")
-    if await group_manager.check_group_task_status(event.group_id, 'refund_group_remind'):
+    if await group_manager.check_group_task_status(
+        event.group_id, "refund_group_remind"
+    ):
         rst = ""
         if event.sub_type == "leave":
             rst = f"{user_name}离开了我们..."
@@ -133,7 +174,9 @@ async def _(bot: Bot, event: GroupDecreaseNoticeEvent, state: dict):
             operator = await bot.get_group_member_info(
                 user_id=event.operator_id, group_id=event.group_id
             )
-            operator_name = operator["card"] if operator["card"] else operator["nickname"]
+            operator_name = (
+                operator["card"] if operator["card"] else operator["nickname"]
+            )
             rst = f"{user_name} 被 {operator_name} 送走了."
         try:
             await group_decrease_handle.send(f"{rst}")

basic_plugins/help/__init__.py

@@ -2,25 +2,30 @@ from nonebot import on_command
 from nonebot.adapters.cqhttp import (
     Bot,
     MessageEvent,
+    GroupMessageEvent
 )
 from nonebot.typing import T_State
 from nonebot.rule import to_me
-from configs.path_config import IMAGE_PATH
+from configs.path_config import IMAGE_PATH, DATA_PATH
 from utils.message_builder import image
 from .data_source import create_help_img, get_plugin_help
 from utils.utils import get_message_text
 from pathlib import Path
+import os
 
 
 __zx_plugin_name__ = "帮助"
 
-
+group_help_path = Path(DATA_PATH) / "group_help"
 help_image = Path(IMAGE_PATH) / "help.png"
 simple_help_image = Path(IMAGE_PATH) / "simple_help.png"
 if help_image.exists():
     help_image.unlink()
 if simple_help_image.exists():
     simple_help_image.unlink()
+for x in os.listdir(group_help_path):
+    group_help_image = group_help_path / x
+    group_help_image.unlink()
 
 _help = on_command("详细功能", rule=to_me(), aliases={"详细帮助"}, priority=1, block=True)
 simple_help = on_command("功能", rule=to_me(), aliases={"help", "帮助"}, priority=1, block=True)
@@ -33,7 +38,7 @@ async def _(bot: Bot, event: MessageEvent, state: T_State):
             help_image.unlink()
         if simple_help_image.exists():
             simple_help_image.unlink()
-        await create_help_img(help_image, simple_help_image)
+        await create_help_img(None, help_image, simple_help_image)
     await _help.finish(image("help.png"))
 
 
@@ -45,17 +50,23 @@ async def _(bot: Bot, event: MessageEvent, state: T_State):
         if '-super' in msg:
             if str(event.user_id) in bot.config.superusers:
                 is_super = True
-            msg = msg.replace('-super', '')
+            msg = msg.replace('-super', '').strip()
         msg = get_plugin_help(msg, is_super)
         if msg:
             await _help.send(image(b64=msg))
         else:
             await _help.send("没有此功能的帮助信息...")
     else:
-        if not simple_help_image.exists():
-            if help_image.exists():
-                help_image.unlink()
-            if simple_help_image.exists():
-                simple_help_image.unlink()
-            await create_help_img(help_image, simple_help_image)
-        await _help.finish(image("simple_help.png"))
+        if isinstance(event, GroupMessageEvent):
+            _image_path = group_help_path / f"{event.group_id}.png"
+            if not _image_path.exists():
+                await create_help_img(event.group_id, help_image, _image_path)
+            await simple_help.send(image(_image_path))
+        else:
+            if not simple_help_image.exists():
+                if help_image.exists():
+                    help_image.unlink()
+                if simple_help_image.exists():
+                    simple_help_image.unlink()
+                await create_help_img(None, help_image, simple_help_image)
+            await _help.finish(image("simple_help.png"))

basic_plugins/help/data_source.py

@@ -1,6 +1,11 @@
 from utils.image_utils import CreateImg
 from configs.path_config import IMAGE_PATH
-from utils.manager import plugins2settings_manager
+from utils.manager import (
+    plugins2settings_manager,
+    admin_manager,
+    plugins_manager,
+    group_manager,
+)
 from typing import Optional
 from services.log import logger
 from pathlib import Path
@@ -16,20 +21,26 @@ random_bk_path = Path(IMAGE_PATH) / "background" / "help" / "simple_help"
 background = Path(IMAGE_PATH) / "background" / "0.png"
 
 
-async def create_help_img(help_image: Path, simple_help_image: Path):
+async def create_help_img(
+    group_id: Optional[int], help_image: Path, simple_help_image: Path
+):
     """
     生成帮助图片
+    :param group_id: 群号
     :param help_image: 图片路径
     :param simple_help_image: 简易帮助图片路径
     """
     return await asyncio.get_event_loop().run_in_executor(
-        None, _create_help_img, help_image, simple_help_image
+        None, _create_help_img, group_id, help_image, simple_help_image
     )
 
 
-def _create_help_img(help_image: Path, simple_help_image: Path):
+def _create_help_img(
+    group_id: Optional[int], help_image: Path, simple_help_image: Path
+):
     """
     生成帮助图片
+    :param group_id: 群号
     :param help_image: 图片路径
     :param simple_help_image: 简易帮助图片路径
     """
@@ -41,6 +52,7 @@ def _create_help_img(help_image: Path, simple_help_image: Path):
     _tmp = []
     tmp_img = CreateImg(0, 0, plain_text="1", font_size=24)
     font_height = tmp_img.h
+    # 插件分类
     for matcher in _matchers:
         plugin_name = None
         _plugin = nonebot.plugin.get_plugin(matcher.module)
@@ -99,6 +111,7 @@ def _create_help_img(help_image: Path, simple_help_image: Path):
                     logger.warning(f"{type(e)}: {e}")
             else:
                 matchers_data[plugin_type][plugin_name] = {
+                    "module": matcher.module,
                     "des": plugin_des,
                     "cmd": plugin_cmd,
                     "text_type": text_type,
@@ -124,13 +137,28 @@ def _create_help_img(help_image: Path, simple_help_image: Path):
     types = list(matchers_data.keys())
     types.sort()
     ix = 0
+    # 详细帮助
     for type_ in types:
         keys = list(matchers_data[type_].keys())
         keys.sort()
         help_str = f"{type_ if type_ != 'normal' else '功能'}:\n\n"
         simple_help_str = f"{type_ if type_ != 'normal' else '功能'}:\n\n"
         for i, k in enumerate(keys):
-            simple_help_str += f"{i+1}.{k}\n"
+            # 禁用flag
+            flag = True
+            if plugins_manager.get_plugin_status(
+                matchers_data[type_][k]["module"], "all"
+            ):
+                flag = False
+            if group_id:
+                flag = flag and plugins_manager.get_plugin_status(
+                    matchers_data[type_][k]["module"], "group"
+                )
+            simple_help_str += (
+                f"{i+1}.{k}<|_|~|>"
+                f"{group_manager.get_plugin_status(matchers_data[type_][k]['module'], group_id) if group_id else '_'}|"
+                f"{flag}\n"
+            )
             if matchers_data[type_][k]["text_type"] == 1:
                 _x = tmp_img.getsize(
                     f"{i+1}".rjust(5)
@@ -157,19 +185,48 @@ def _create_help_img(help_image: Path, simple_help_image: Path):
             width + 150, height, font_size=24, color="white" if not ix % 2 else "black"
         )
         A.text((10, 10), help_str, (255, 255, 255) if ix % 2 else (0, 0, 0))
+        # 生成各个分类的插件简易帮助图片
         simple_width = 0
-        for x in [tmp_img.getsize(x)[0] for x in simple_help_str.split("\n")]:
+        for x in [
+            tmp_img.getsize(x.split("<|_|~|>")[0])[0]
+            for x in simple_help_str.split("\n")
+        ]:
             simple_width = simple_width if simple_width > x else x
-        bk = CreateImg(
-            simple_width + 20, simple_height, font_size=24, color="#6495ED"
-        )
+        bk = CreateImg(simple_width + 20, simple_height, font_size=24, color="#6495ED")
         B = CreateImg(
             simple_width + 20,
             simple_height,
             font_size=24,
             color="white" if not ix % 2 else "black",
         )
-        B.text((10, 10), simple_help_str, (255, 255, 255) if ix % 2 else (0, 0, 0))
+        # 切分，判断插件开关状态
+        _s_height = 10
+        for _s in simple_help_str.split("\n"):
+            text_color = (255, 255, 255) if ix % 2 else (0, 0, 0)
+            _line_flag = False
+            if "<|_|~|>" in _s:
+                _x = _s.split("<|_|~|>")
+                _flag_sp = _x[-1].split("|")
+                if group_id:
+                    if _flag_sp[0].lower() != "true":
+                        text_color = (252, 75, 13)
+                if _flag_sp[1].lower() == "true":
+                    _line_flag = True
+                _s = _x[0]
+            B.text((10, _s_height), _s, text_color)
+            if _line_flag:
+                B.line(
+                    (
+                        7,
+                        _s_height + int(B.getsize(_s)[1] / 2) + 2,
+                        B.getsize(_s)[0] + 11,
+                        _s_height + int(B.getsize(_s)[1] / 2) + 2,
+                    ),
+                    (236, 66, 7),
+                    3,
+                )
+            _s_height += B.getsize("1")[1] + 5
+        # B.text((10, 10), simple_help_str, (255, 255, 255) if ix % 2 else (0, 0, 0))
         bk.paste(B, center_type="center")
         bk.transparent(2)
         ix += 1
@@ -178,13 +235,18 @@ def _create_help_img(help_image: Path, simple_help_image: Path):
     height = 0
     for img in help_img_list:
         height += img.h
-    A = CreateImg(width + 150, height + 50, font_size=24)
-    A.text((10, 10), '*  注: ‘*’ 代表可有多个相同参数 ‘?’ 代表可省略该参数  *\n\n" "功能名: 功能简介 -> 指令\n\n')
-    current_height = 50
-    for img in help_img_list:
-        A.paste(img, (0, current_height))
-        current_height += img.h
-    A.save(help_image)
+    if not group_id:
+        A = CreateImg(width + 150, height + 50, font_size=24)
+        A.text(
+            (10, 10), '*  注: ‘*’ 代表可有多个相同参数 ‘?’ 代表可省略该参数  *\n\n" "功能名: 功能简介 -> 指令\n\n'
+        )
+        current_height = 50
+        for img in help_img_list:
+            A.paste(img, (0, current_height))
+            current_height += img.h
+        A.save(help_image)
+    # 详细帮助生成完毕
+    # 简易帮助图片合成
     height = 0
     width = 0
     for img in simple_help_img_list:
@@ -209,19 +271,30 @@ def _create_help_img(help_image: Path, simple_help_image: Path):
     _, B = get_max_width_or_paste(simple_help_img_list, B, True)
     w = 10
     h = 10
-    for msg in ['目前支持的功能列表:', '可以通过 ‘帮助[功能名称]’ 来获取对应功能的使用方法', '或者使用 ‘详细帮助’ 来获取所有功能方法']:
+    for msg in ["目前支持的功能列表:", "可以通过 ‘帮助[功能名称]’ 来获取对应功能的使用方法", "或者使用 ‘详细帮助’ 来获取所有功能方法"]:
         text = CreateImg(
             0,
             0,
             plain_text=msg,
             font_size=24,
-            color=(255, 255, 255, 0),
-            font='yuanshen.ttf'
+            font="yuanshen.ttf",
         )
         B.paste(text, (w, h), True)
         h += 50
-        if msg == '目前支持的功能列表:':
+        if msg == "目前支持的功能列表:":
             w += 50
+    B.paste(
+        CreateImg(
+            0,
+            0,
+            plain_text="注: 红字代表功能被群管理员禁用，红线代表功能正在维护",
+            font_size=24,
+            font="yuanshen.ttf",
+            font_color=(231, 74, 57)
+        ),
+        (300, 10),
+        True,
+    )
     B.save(simple_help_image)
 
 
@@ -262,6 +335,8 @@ def get_plugin_help(msg: str, is_super: bool = False) -> Optional[str]:
     :param is_super: 是否为超级用户
     """
     module = plugins2settings_manager.get_plugin_module(msg)
+    if not module:
+        module = admin_manager.get_plugin_module(msg)
     if module:
         try:
             plugin = nonebot.plugin.get_plugin(module)
@@ -272,18 +347,21 @@ def get_plugin_help(msg: str, is_super: bool = False) -> Optional[str]:
                     )
                 else:
                     result = plugin.module.__getattribute__("__plugin_usage__")
-                width = 0
-                for x in result.split("\n"):
-                    _width = len(x) * 24
-                    width = width if width > _width else _width
-                height = len(result.split("\n")) * 45
-                A = CreateImg(width, height, font_size=24)
-                bk = CreateImg(
-                    width, height, background=Path(IMAGE_PATH) / "background" / "1.png"
-                )
-                A.paste(bk, alpha=True)
-                A.text((int(width * 0.048), int(height * 0.21)), result)
-                return A.pic2bs4()
+                if result:
+                    width = 0
+                    for x in result.split("\n"):
+                        _width = len(x) * 24
+                        width = width if width > _width else _width
+                    height = len(result.split("\n")) * 45
+                    A = CreateImg(width, height, font_size=24)
+                    bk = CreateImg(
+                        width,
+                        height,
+                        background=Path(IMAGE_PATH) / "background" / "1.png",
+                    )
+                    A.paste(bk, alpha=True)
+                    A.text((int(width * 0.048), int(height * 0.21)), result)
+                    return A.pic2bs4()
         except AttributeError:
             pass
     return None

basic_plugins/hooks/__init__.py

@@ -0,0 +1,38 @@
+from configs.config import Config
+import nonebot
+
+
+Config.add_plugin_config(
+    "hook",
+    "CHECK_NOTICE_INFO_CD",
+    300,
+    name="基础hook配置",
+    help_="群检测，个人权限检测等各种检测提示信息cd",
+    default_value=300
+)
+
+Config.add_plugin_config(
+    "hook",
+    "MALICIOUS_BAN_TIME",
+    30,
+    help_="恶意命令触发检测触发后ban的时长（分钟）",
+    default_value=30
+)
+
+Config.add_plugin_config(
+    "hook",
+    "MALICIOUS_CHECK_TIME",
+    5,
+    help_="恶意命令触发检测规定时间内（秒）",
+    default_value=5
+)
+
+Config.add_plugin_config(
+    "hook",
+    "MALICIOUS_BAN_COUNT",
+    6,
+    help_="恶意命令触发检测最大触发次数",
+    default_value=6
+)
+
+nonebot.load_plugins("basic_plugins/hooks")

basic_plugins/hooks/auth_hook.py

@@ -0,0 +1,210 @@
+from nonebot.matcher import Matcher
+from nonebot.message import run_preprocessor, IgnoredException
+from nonebot.adapters.cqhttp.exception import ActionFailed
+from utils.manager import (
+    plugins2settings_manager,
+    admin_manager,
+    group_manager,
+    plugins_manager,
+)
+from .utils import set_block_limit_false, status_message_manager
+from nonebot.typing import T_State
+from nonebot.adapters.cqhttp import (
+    Bot,
+    MessageEvent,
+    GroupMessageEvent,
+    PokeNotifyEvent,
+)
+from configs.config import Config
+from models.ban_user import BanUser
+from utils.utils import FreqLimiter
+from utils.message_builder import at
+from models.level_user import LevelUser
+
+
+_flmt = FreqLimiter(Config.get_config("hook", "CHECK_NOTICE_INFO_CD"))
+_flmt_g = FreqLimiter(Config.get_config("hook", "CHECK_NOTICE_INFO_CD"))
+_flmt_s = FreqLimiter(Config.get_config("hook", "CHECK_NOTICE_INFO_CD"))
+_flmt_c = FreqLimiter(Config.get_config("hook", "CHECK_NOTICE_INFO_CD"))
+
+
+ignore_rst_module = ["ai", "poke", "dialogue"]
+
+
+# 权限检测
+@run_preprocessor
+async def _(matcher: Matcher, bot: Bot, event: MessageEvent, state: T_State):
+    module = matcher.module
+    plugins2info_dict = plugins2settings_manager.get_data()
+    if (
+        (not isinstance(event, MessageEvent) and module != "poke")
+        or await BanUser.is_ban(event.user_id)
+        and str(event.user_id) not in bot.config.superusers
+    ) or (
+        str(event.user_id) in bot.config.superusers
+        and plugins2info_dict.get(module)
+        and not plugins2info_dict[module]["limit_superuser"]
+    ):
+        return
+    # 群黑名单检测
+    if isinstance(event, GroupMessageEvent):
+        if group_manager.get_group_level(event.group_id) < 0:
+            raise IgnoredException("群黑名单")
+    if module in admin_manager.keys() and matcher.priority not in [1, 9]:
+        if isinstance(event, GroupMessageEvent):
+            # 个人权限
+            if (
+                not await LevelUser.check_level(
+                    event.user_id,
+                    event.group_id,
+                    admin_manager.get_plugin_level(module),
+                )
+                and admin_manager.get_plugin_level(module) > 0
+            ):
+                try:
+                    if _flmt.check(event.user_id):
+                        _flmt.start_cd(event.user_id)
+                        await bot.send_group_msg(
+                            group_id=event.group_id,
+                            message=f"{at(event.user_id)}你的权限不足喔，该功能需要的权限等级："
+                            f"{admin_manager.get_plugin_level(module)}",
+                        )
+                except ActionFailed:
+                    pass
+                set_block_limit_false(event, module)
+                if event.is_tome():
+                    status_message_manager.add(event.group_id)
+                raise IgnoredException("权限不足")
+        else:
+            if not await LevelUser.check_level(
+                event.user_id, 0, admin_manager.get_plugin_level(module)
+            ):
+                try:
+                    await bot.send_private_msg(
+                        user_id=event.user_id,
+                        message=f"你的权限不足喔，该功能需要的权限等级：{admin_manager.get_plugin_level(module)}",
+                    )
+                except ActionFailed:
+                    pass
+                set_block_limit_false(event, module)
+                if event.is_tome():
+                    status_message_manager.add(event.user_id)
+                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 status_message_manager.get(event.group_id) is None:
+                status_message_manager.delete(event.group_id)
+            if plugins2info_dict[module]["level"] > group_manager.get_group_level(
+                event.group_id
+            ):
+                try:
+                    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
+                if event.is_tome():
+                    status_message_manager.add(event.group_id)
+                set_block_limit_false(event, module)
+                raise IgnoredException("群权限不足")
+            # 插件状态
+            if not group_manager.get_plugin_status(module, event.group_id):
+                try:
+                    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
+                if event.is_tome():
+                    status_message_manager.add(event.group_id)
+                set_block_limit_false(event, module)
+                raise IgnoredException("未开启此功能...")
+            # 管理员禁用
+            if not group_manager.get_plugin_status(f"{module}:super", event.group_id):
+                try:
+                    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
+                if event.is_tome():
+                    status_message_manager.add(event.group_id)
+                set_block_limit_false(event, module)
+                raise IgnoredException("管理员禁用了此群该功能...")
+            # 群聊禁用
+            if not plugins_manager.get_plugin_status(module, block_type="group"):
+                try:
+                    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
+                if event.is_tome():
+                    status_message_manager.add(event.group_id)
+                set_block_limit_false(event, module)
+                raise IgnoredException("该插件在群聊中已被禁用...")
+        else:
+            # 私聊禁用
+            if not plugins_manager.get_plugin_status(module, block_type="private"):
+                try:
+                    if _flmt_c.check(event.user_id):
+                        _flmt_c.start_cd(event.user_id)
+                        await bot.send_private_msg(
+                            user_id=event.user_id, message="该功能在私聊中已被禁用..."
+                        )
+                except ActionFailed:
+                    pass
+                if event.is_tome():
+                    status_message_manager.add(event.user_id)
+                set_block_limit_false(event, module)
+                raise IgnoredException("该插件在私聊中已被禁用...")
+        # 维护
+        if not plugins_manager.get_plugin_status(module, block_type="all"):
+            if isinstance(
+                event, GroupMessageEvent
+            ) and group_manager.check_group_is_white(event.group_id):
+                return
+            try:
+                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="此功能正在维护..."
+                    )
+            except ActionFailed:
+                pass
+            if event.is_tome():
+                id_ = (
+                    event.group_id
+                    if isinstance(event, GroupMessageEvent)
+                    else event.user_id
+                )
+                status_message_manager.add(id_)
+            set_block_limit_false(event, module)
+            raise IgnoredException("此功能正在维护...")

basic_plugins/hooks/ban_hook.py

@@ -0,0 +1,83 @@
+from nonebot.matcher import Matcher
+from nonebot.message import run_preprocessor, IgnoredException
+from nonebot.adapters.cqhttp.exception import ActionFailed
+from nonebot.typing import T_State
+from nonebot.adapters.cqhttp import (
+    Bot,
+    MessageEvent,
+    GroupMessageEvent,
+)
+from configs.config import Config
+from models.ban_user import BanUser
+from utils.utils import is_number, static_flmt
+from utils.message_builder import at
+
+
+Config.add_plugin_config(
+    "hook",
+    "BAN_RESULT",
+    "才不会给你发消息.",
+    help_="对被ban用户发送的消息",
+)
+
+
+# 检查是否被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]:
+        if (
+            await BanUser.is_ban(event.user_id)
+            and str(event.user_id) not in bot.config.superusers
+        ):
+            time = await BanUser.check_ban_time(event.user_id)
+            if is_number(time):
+                time = abs(int(time))
+                if time < 60:
+                    time = str(time) + " 秒"
+                else:
+                    time = str(int(time / 60)) + " 分钟"
+            else:
+                time = str(time) + " 分钟"
+            if isinstance(event, GroupMessageEvent):
+                if not static_flmt.check(event.user_id):
+                    raise IgnoredException("用户处于黑名单中")
+                static_flmt.start_cd(event.user_id)
+                if matcher.priority != 9:
+                    try:
+                        ban_result = Config.get_config("hook", "BAN_RESULT")
+                        if ban_result:
+                            await bot.send_group_msg(
+                                group_id=event.group_id,
+                                message=at(event.user_id)
+                                + ban_result
+                                + f" 在..在 {time} 后才会理你喔",
+                            )
+                    except ActionFailed:
+                        pass
+            else:
+                if not static_flmt.check(event.user_id):
+                    raise IgnoredException("用户处于黑名单中")
+                static_flmt.start_cd(event.user_id)
+                if matcher.priority != 9:
+                    try:
+                        ban_result = Config.get_config("hook", "BAN_RESULT")
+                        if ban_result:
+                            await bot.send_private_msg(
+                                user_id=event.user_id,
+                                message=at(event.user_id)
+                                + ban_result
+                                + f" 在..在 {time}后才会理你喔",
+                            )
+                    except ActionFailed:
+                        pass
+            raise IgnoredException("用户处于黑名单中")

basic_plugins/hooks/chkdsk_hook.py

@@ -0,0 +1,54 @@
+from nonebot.matcher import Matcher
+from nonebot.message import run_preprocessor, IgnoredException
+from nonebot.adapters.cqhttp.exception import ActionFailed
+from nonebot.typing import T_State
+from nonebot.adapters.cqhttp import (
+    Bot,
+    MessageEvent,
+    GroupMessageEvent,
+)
+from configs.config import Config
+from models.ban_user import BanUser
+from utils.utils import BanCheckLimiter
+from utils.message_builder import at
+from services.log import logger
+
+
+_blmt = BanCheckLimiter(
+    Config.get_config("hook", "MALICIOUS_CHECK_TIME"),
+    Config.get_config("hook", "MALICIOUS_BAN_COUNT"),
+)
+
+
+# 恶意触发命令检测
+@run_preprocessor
+async def _(matcher: Matcher, bot: Bot, event: GroupMessageEvent, state: T_State):
+    if not isinstance(event, MessageEvent):
+        return
+    if matcher.type == "message" and matcher.priority not in [1, 9]:
+        if state["_prefix"]["raw_command"]:
+            if _blmt.check(f'{event.user_id}{state["_prefix"]["raw_command"]}'):
+                if await BanUser.ban(
+                    event.user_id,
+                    9,
+                    Config.get_config("hook", "MALICIOUS_BAN_TIME") * 60,
+                ):
+                    logger.info(f"USER {event.user_id} 触发了恶意触发检测")
+                if isinstance(event, GroupMessageEvent):
+                    try:
+                        await bot.send_group_msg(
+                            group_id=event.group_id,
+                            message=at(event.user_id) + "检测到恶意触发命令，您将被封禁 30 分钟",
+                        )
+                    except ActionFailed:
+                        pass
+                else:
+                    try:
+                        await bot.send_private_msg(
+                            user_id=event.user_id,
+                            message=at(event.user_id) + "检测到恶意触发命令，您将被封禁 30 分钟",
+                        )
+                    except ActionFailed:
+                        pass
+                raise IgnoredException("检测到恶意触发命令")
+        _blmt.add(f'{event.user_id}{state["_prefix"]["raw_command"]}')

basic_plugins/hooks/limit_hook.py

@@ -0,0 +1,146 @@
+from nonebot.matcher import Matcher
+from nonebot.message import run_preprocessor, run_postprocessor, IgnoredException
+from nonebot.adapters.cqhttp.exception import ActionFailed
+from models.friend_user import FriendUser
+from models.group_member_info import GroupInfoUser
+from utils.message_builder import at
+from .utils import status_message_manager, set_block_limit_false
+from utils.manager import (
+    plugins2cd_manager,
+    plugins2block_manager,
+    plugins2count_manager,
+)
+from typing import Optional
+from nonebot.typing import T_State
+from nonebot.adapters.cqhttp import (
+    Bot,
+    Event,
+    MessageEvent,
+    PrivateMessageEvent,
+    GroupMessageEvent,
+    Message,
+)
+
+
+# 命令cd | 命令阻塞 | 命令次数
+@run_preprocessor
+async def _(matcher: Matcher, bot: Bot, event: MessageEvent, state: T_State):
+    if not isinstance(event, MessageEvent) and matcher.module != "poke":
+        return
+    module = matcher.module
+    if (
+        isinstance(event, GroupMessageEvent)
+        and status_message_manager.get(event.group_id) is None
+    ):
+        status_message_manager.delete(event.group_id)
+    # Count
+    if (
+        plugins2count_manager.check_plugin_count_status(module)
+        and event.user_id not in bot.config.superusers
+    ):
+        plugin_count_data = plugins2count_manager.get_plugin_count_data(module)
+        limit_type = plugin_count_data["limit_type"]
+        rst = plugin_count_data["rst"]
+        count_type_ = event.user_id
+        if limit_type == "group" and isinstance(event, GroupMessageEvent):
+            count_type_ = event.group_id
+        if not plugins2count_manager.check(module, count_type_):
+            if rst:
+                rst = await init_rst(rst, event)
+                await send_msg(rst, bot, event)
+            raise IgnoredException(f"{module} count次数限制...")
+        else:
+            plugins2count_manager.increase(module, count_type_)
+    # Cd
+    if plugins2cd_manager.check_plugin_cd_status(module):
+        plugin_cd_data = plugins2cd_manager.get_plugin_cd_data(module)
+        check_type = plugin_cd_data["check_type"]
+        limit_type = plugin_cd_data["limit_type"]
+        rst = plugin_cd_data["rst"]
+        if (
+            (isinstance(event, PrivateMessageEvent) and check_type == "private")
+            or (isinstance(event, GroupMessageEvent) and check_type == "group")
+            or plugins2cd_manager.get_plugin_data(module).get("check_type") == "all"
+        ):
+            cd_type_ = event.user_id
+            if limit_type == "group" and isinstance(event, GroupMessageEvent):
+                cd_type_ = event.group_id
+            if not plugins2cd_manager.check(module, cd_type_):
+                if rst:
+                    rst = await init_rst(rst, event)
+                    await send_msg(rst, bot, event)
+                raise IgnoredException(f"{module} 正在cd中...")
+            else:
+                plugins2cd_manager.start_cd(module, cd_type_)
+    # Block
+    if plugins2block_manager.check_plugin_block_status(module):
+        plugin_block_data = plugins2block_manager.get_plugin_block_data(module)
+        check_type = plugin_block_data["check_type"]
+        limit_type = plugin_block_data["limit_type"]
+        rst = plugin_block_data["rst"]
+        if (
+            (isinstance(event, PrivateMessageEvent) and check_type == "private")
+            or (isinstance(event, GroupMessageEvent) and check_type == "group")
+            or check_type == "all"
+        ):
+            block_type_ = event.user_id
+            if limit_type == "group" and isinstance(event, GroupMessageEvent):
+                block_type_ = event.group_id
+            if plugins2block_manager.check(block_type_, module):
+                if rst:
+                    rst = await init_rst(rst, event)
+                    await send_msg(rst, bot, event)
+                raise IgnoredException(f"{event.user_id}正在调用{module}....")
+            else:
+                plugins2block_manager.set_true(block_type_, module)
+
+
+async def send_msg(rst: str, bot: Bot, event: MessageEvent):
+    """
+    发送信息
+    :param rst: pass
+    :param bot: pass
+    :param event: pass
+    """
+    rst = await init_rst(rst, event)
+    try:
+        if isinstance(event, GroupMessageEvent):
+            status_message_manager.add(event.group_id)
+            await bot.send_group_msg(group_id=event.group_id, message=Message(rst))
+        else:
+            status_message_manager.add(event.user_id)
+            await bot.send_private_msg(user_id=event.user_id, message=Message(rst))
+    except ActionFailed:
+        pass
+
+
+# 解除命令block阻塞
+@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
+    set_block_limit_false(event, module)
+
+
+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

basic_plugins/hooks/other_hook.py

@@ -0,0 +1,41 @@
+from nonebot.matcher import Matcher
+from nonebot.message import run_preprocessor, IgnoredException
+from nonebot.typing import T_State
+from .utils import status_message_manager
+from nonebot.adapters.cqhttp import (
+    Bot,
+    MessageEvent,
+    PrivateMessageEvent,
+    GroupMessageEvent,
+)
+
+
+# 为什么AI会自己和自己聊天
+@run_preprocessor
+async def _(matcher: Matcher, bot: Bot, event: PrivateMessageEvent, state: T_State):
+    if not isinstance(event, MessageEvent):
+        return
+    if event.user_id == int(bot.self_id):
+        raise IgnoredException("为什么AI会自己和自己聊天")
+
+
+# 有命令就别说话了
+@run_preprocessor
+async def _(matcher: Matcher, bot: Bot, event: MessageEvent, state: T_State):
+    if not isinstance(event, MessageEvent):
+        return
+    if matcher.type == "message" and matcher.module == "ai":
+        if (
+            isinstance(event, GroupMessageEvent)
+            and not status_message_manager.check(event.group_id)
+        ):
+            status_message_manager.delete(event.group_id)
+            raise IgnoredException("有命令就别说话了")
+        elif (
+            isinstance(event, PrivateMessageEvent)
+            and not status_message_manager.check(event.user_id)
+        ):
+            print(status_message_manager)
+            status_message_manager.delete(event.user_id)
+            raise IgnoredException("有命令就别说话了")
+

basic_plugins/hooks/utils.py

@@ -0,0 +1,48 @@
+from nonebot.adapters.cqhttp import GroupMessageEvent, PrivateMessageEvent
+from utils.manager import plugins2block_manager, StaticData
+import time
+
+
+class StatusMessageManager(StaticData):
+
+    def __init__(self):
+        super().__init__(None)
+
+    def add(self, id_: int):
+        self._data[id_] = time.time()
+
+    def delete(self, id_: int):
+        if self._data.get(id_):
+            del self._data[id_]
+
+    def check(self, id_: int, t: int = 30) -> bool:
+        if self._data.get(id_):
+            if time.time() - self._data[id_] > t:
+                del self._data[id_]
+                return True
+            return False
+        return True
+
+
+status_message_manager = StatusMessageManager()
+
+
+def set_block_limit_false(event, module):
+    """
+    设置用户block为false
+    :param event: event
+    :param module: 插件模块
+    """
+    if plugins2block_manager.check_plugin_block_status(module):
+        plugin_block_data = plugins2block_manager.get_plugin_block_data(module)
+        check_type = plugin_block_data["check_type"]
+        limit_type = plugin_block_data["limit_type"]
+        if not (
+            (isinstance(event, GroupMessageEvent) and check_type == "private")
+            or (isinstance(event, PrivateMessageEvent) and check_type == "group")
+        ):
+            block_type_ = event.user_id
+            if limit_type == "group" and isinstance(event, GroupMessageEvent):
+                block_type_ = event.group_id
+            plugins2block_manager.set_false(block_type_, module)
+

basic_plugins/hooks/withdraw_message_hook.py

@@ -0,0 +1,28 @@
+from nonebot.matcher import Matcher
+from nonebot.message import run_postprocessor
+from typing import Optional
+from nonebot.typing import T_State
+from nonebot.adapters.cqhttp import Bot, Event
+from utils.manager import withdraw_message_manager
+import asyncio
+
+
+# 消息撤回
+@run_postprocessor
+async def _(
+    matcher: Matcher,
+    exception: Optional[Exception],
+    bot: Bot,
+    event: Event,
+    state: T_State,
+):
+    tasks = []
+    for id_, time in withdraw_message_manager.data:
+        tasks.append(asyncio.ensure_future(_withdraw_message(bot, id_, time)))
+        withdraw_message_manager.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))

basic_plugins/init_plugin_config/__init__.py

@@ -0,0 +1,51 @@
+from .init_group_manager import init_group_manager, group_manager
+from .init_plugins_config import init_plugins_config
+from .init_plugins_data import init_plugins_data, plugins_manager
+from .init_none_plugin_count_manager import init_none_plugin_count_manager
+from .init_plugins_resources import init_plugins_resources
+from .init_plugins_settings import init_plugins_settings
+from .init_plugins_limit import (
+    init_plugins_block_limit,
+    init_plugins_count_limit,
+    init_plugins_cd_limit,
+)
+from .check_plugin_status import check_plugin_status
+from nonebot.adapters.cqhttp import Bot
+from configs.path_config import DATA_PATH
+from services.log import logger
+from nonebot import Driver
+import nonebot
+
+
+__zx_plugin_name__ = "初始化插件数据 [Hidden]"
+__plugin_version__ = 0.1
+__plugin_author__ = "HibiKier"
+
+
+driver: Driver = nonebot.get_driver()
+
+
+@driver.on_startup
+def _():
+    """
+    初始化数据
+    """
+    init_plugins_settings(DATA_PATH)
+    init_plugins_cd_limit(DATA_PATH)
+    init_plugins_block_limit(DATA_PATH)
+    init_plugins_count_limit(DATA_PATH)
+    init_plugins_data(DATA_PATH)
+    init_plugins_config(DATA_PATH)
+    init_plugins_resources()
+    init_none_plugin_count_manager()
+    x = group_manager.get_super_old_data()
+    if x:
+        for key in x.keys():
+            plugins_manager.block_plugin(key, block_type=x[key])
+    logger.info("初始化数据完成...")
+
+
+@driver.on_bot_connect
+async def _(bot: Bot):
+    await init_group_manager()
+    await check_plugin_status(bot)

basic_plugins/init_plugin_config/check_plugin_status.py

@@ -0,0 +1,18 @@
+from utils.manager import plugins_manager
+from nonebot.adapters.cqhttp import Bot
+
+
+async def check_plugin_status(bot: Bot):
+    """
+    遍历查看插件加载情况
+    """
+    rst = ""
+    for plugin in plugins_manager.keys():
+        data = plugins_manager.get(plugin)
+        if data.get("error") or data.get("error") is None:
+            rst += f'{plugin}:{data["plugin_name"]}\n'
+    if rst:
+        rst = "以下插件加载失败..\n" + rst
+        await bot.send_private_msg(
+            user_id=int(list(bot.config.superusers)[0]), message=rst[:-1]
+        )

basic_plugins/init_plugin_config/init_group_manager.py

@@ -0,0 +1,73 @@
+from pathlib import Path
+from utils.manager import group_manager
+from services.db_context import db
+from asyncpg.exceptions import DuplicateColumnError
+from services.log import logger
+
+try:
+    import ujson as json
+except ModuleNotFoundError:
+    import json
+try:
+    from models.group_remind import GroupRemind
+except ModuleNotFoundError:
+    pass
+
+
+async def init_group_manager():
+    """
+    旧数据格式替换为新格式
+    初始化数据
+    """
+    old_group_level_file = Path() / "data" / "manager" / "group_level.json"
+    old_plugin_list_file = Path() / "data" / "manager" / "plugin_list.json"
+    if old_group_level_file.exists():
+        data = json.load(open(old_group_level_file, "r", encoding="utf8"))
+        for key in data.keys():
+            group = key
+            level = data[key]
+            group_manager.set_group_level(group, level)
+        old_group_level_file.unlink()
+        group_manager.save()
+
+    if old_plugin_list_file.exists():
+        data = json.load(open(old_plugin_list_file, "r", encoding="utf8"))
+        for plugin in data.keys():
+            for group in data[plugin].keys():
+                if group == "default" and not data[plugin]["default"]:
+                    group_manager.block_plugin(plugin)
+                elif not data[plugin][group]:
+                    group_manager.block_plugin(plugin, group)
+        old_plugin_list_file.unlink()
+    old_data_table = Path() / "models" / "group_remind.py"
+    try:
+        if old_data_table.exists():
+            b = {
+                "hy": "group_welcome",
+                "kxcz": "open_case_reset_remind",
+                "zwa": "zwa",
+                "blpar": "bilibili_parse",
+                "epic": "epic_free_game",
+                "pa": "pa",
+                "almanac": "genshin_alc",
+            }
+            for group in group_manager.get_data()["group_manager"]:
+                for remind in b:
+                    try:
+                        status = await GroupRemind.get_status(int(group), remind)
+                        if status is not None:
+                            if status:
+                                await group_manager.open_group_task(group, b[remind])
+                                logger.info(f"读取旧数据-->{group} 开启 {b[remind]}")
+                            else:
+                                await group_manager.close_group_task(group, b[remind])
+                                logger.info(f"读取旧数据-->{group} 关闭 {b[remind]}")
+                    except Exception as e:
+                        pass
+            query = db.text("DROP TABLE group_reminds;")
+            await db.first(query)
+            old_data_table.unlink()
+            logger.info("旧数据读取完毕，删除了舍弃表 group_reminds...")
+    except (ModuleNotFoundError, DuplicateColumnError):
+        pass
+    group_manager.save()

basic_plugins/init_plugin_config/init_none_plugin_count_manager.py

@@ -0,0 +1,45 @@
+from utils.manager import (
+    none_plugin_count_manager,
+    plugins2count_manager,
+    plugins2cd_manager,
+    plugins2settings_manager,
+    plugins2block_manager,
+    plugins_manager,
+    resources_manager
+)
+from services.log import logger
+from utils.utils import get_matchers
+
+try:
+    import ujson as json
+except ModuleNotFoundError:
+    import json
+
+
+def init_none_plugin_count_manager():
+    """
+    清除已删除插件数据
+    """
+    modules = [x.module for x in get_matchers()]
+    for module in none_plugin_count_manager.keys():
+        if module not in modules:
+            none_plugin_count_manager.add_count(module)
+        else:
+            none_plugin_count_manager.reset(module)
+        if none_plugin_count_manager.check(module):
+            try:
+                plugin_name = plugins_manager.get(module)["plugin_name"]
+            except (AttributeError, KeyError):
+                plugin_name = ""
+            try:
+                plugins2settings_manager.delete(module)
+                plugins2count_manager.delete(module)
+                plugins2cd_manager.delete(module)
+                plugins2block_manager.delete(module)
+                plugins_manager.delete(module)
+                resources_manager.remove_resource(module)
+                logger.info(f"{module}:{plugin_name} 插件疑似已删除，清除对应插件数据...")
+            except Exception as e:
+                logger.error(
+                    f"{module}:{plugin_name} 插件疑似已删除，清除对应插件数据失败...{type(e)}：{e}"
+                )

basic_plugins/init_plugin_config/init_plugins_config.py

@@ -0,0 +1,147 @@
+from pathlib import Path
+from ruamel.yaml import round_trip_load, round_trip_dump, YAML
+from utils.manager import admin_manager, plugins_manager
+from configs.config import Config
+from services.log import logger
+from utils.utils import get_matchers
+from ruamel import yaml
+import nonebot
+
+
+_yaml = YAML(typ="safe")
+
+
+def init_plugins_config(data_path):
+    """
+    初始化插件数据配置
+    """
+    plugins2config_file = Path(data_path) / "configs" / "plugins2config.yaml"
+    plugins2config_file.parent.mkdir(parents=True, exist_ok=True)
+    _data = {}
+    if plugins2config_file.exists():
+        _data = _yaml.load(open(plugins2config_file, "r", encoding="utf8"))
+    _matchers = get_matchers()
+    for matcher in _matchers:
+        _plugin = nonebot.plugin.get_plugin(matcher.module)
+        try:
+            _module = _plugin.module
+        except AttributeError:
+            continue
+        try:
+            plugin_version = _module.__getattribute__("__plugin_version__")
+        except AttributeError:
+            plugin_version = None
+        try:
+            plugin_configs = _module.__getattribute__("__plugin_configs__")
+        except AttributeError:
+            continue
+        # 插件配置版本更新或为Version为None或不在存储配置内
+        if (
+            plugin_version is None
+            or (
+                _data.get(matcher.module)
+                and _data[matcher.module].keys() != plugin_configs.keys()
+            )
+            or plugin_version > plugins_manager.get(matcher.module)["version"]
+            or matcher.module not in _data.keys()
+        ):
+            for key in plugin_configs:
+                if isinstance(plugin_configs[key], dict):
+                    Config.add_plugin_config(
+                        matcher.module,
+                        key,
+                        plugin_configs[key].get("value"),
+                        help_=plugin_configs[key].get("help"),
+                        default_value=plugin_configs[key].get("default_value"),
+                        _override=True,
+                    )
+                else:
+                    Config.add_plugin_config(matcher.module, key, plugin_configs[key])
+        else:
+            plugin_configs = _data[matcher.module]
+            for key in plugin_configs:
+                Config.add_plugin_config(
+                    matcher.module,
+                    key,
+                    plugin_configs[key]["value"],
+                    help_=plugin_configs[key]["help"],
+                    default_value=plugin_configs[key]["default_value"],
+                    _override=True,
+                )
+    if not Config.is_empty():
+        Config.save()
+        _data = round_trip_load(open(plugins2config_file, encoding="utf8"))
+        for plugin in _data.keys():
+            try:
+                plugin_name = plugins_manager.get(plugin)["plugin_name"]
+            except (AttributeError, TypeError):
+                plugin_name = plugin
+            _data[plugin].yaml_set_start_comment(plugin_name, indent=2)
+        # 初始化未设置的管理员权限等级
+        for k, v in Config.get_admin_level_data():
+            admin_manager.set_admin_level(k, v)
+        # 存完插件基本设置
+        with open(plugins2config_file, "w", encoding="utf8") as wf:
+            round_trip_dump(
+                _data, wf, indent=2, Dumper=yaml.RoundTripDumper, allow_unicode=True
+            )
+        # 再开始读取用户配置
+        user_config_file = Path() / "configs" / "config.yaml"
+        _data = {}
+        _tmp_data = {}
+        if user_config_file.exists():
+            with open(user_config_file, "r", encoding="utf8") as f:
+                _data = _yaml.load(f)
+        for plugin in Config.keys():
+            _tmp_data[plugin] = {}
+            for k in Config[plugin].keys():
+                if _data.get(plugin) and k in _data[plugin].keys():
+                    Config.set_config(plugin, k, _data[plugin][k])
+                _tmp_data[plugin][k] = Config.get_config(plugin, k)
+        Config.save()
+        temp_file = Path() / "configs" / "temp_config.yaml"
+        try:
+            with open(temp_file, "w", encoding="utf8") as wf:
+                yaml.dump(
+                    _tmp_data, wf, Dumper=yaml.RoundTripDumper, allow_unicode=True
+                )
+            with open(temp_file, "r", encoding="utf8") as rf:
+                _data = round_trip_load(rf)
+            # 添加注释
+            for plugin in _data.keys():
+                rst = ""
+                plugin_name = None
+                try:
+                    plugin_data = Config.get(plugin)
+                    for x in list(Config.get(plugin).keys()):
+                        try:
+                            _x = plugin_data[x].get("name")
+                            if _x:
+                                plugin_name = _x
+                        except AttributeError:
+                            pass
+                except (KeyError, AttributeError):
+                    plugin_name = None
+                if not plugin_name:
+                    try:
+                        plugin_name = plugins_manager.get(plugin)["plugin_name"]
+                    except (AttributeError, TypeError):
+                        plugin_name = plugin
+                plugin_name = (
+                    plugin_name.replace("[Hidden]", "")
+                    .replace("[Superuser]", "")
+                    .replace("[Admin]", "")
+                    .strip()
+                )
+                rst += plugin_name + "\n"
+                for k in _data[plugin].keys():
+                    rst += f'{k}: {Config[plugin][k]["help"]}' + "\n"
+                _data[plugin].yaml_set_start_comment(rst[:-1], indent=2)
+            with open(Path() / "configs" / "config.yaml", "w", encoding="utf8") as wf:
+                round_trip_dump(
+                    _data, wf, Dumper=yaml.RoundTripDumper, allow_unicode=True
+                )
+        except Exception as e:
+            logger.error(f"生成简易配置注释错误 {type(e)}：{e}")
+        if temp_file.exists():
+            temp_file.unlink()

basic_plugins/init_plugin_config/init_plugins_data.py

@@ -0,0 +1,82 @@
+from pathlib import Path
+from ruamel.yaml import YAML
+from utils.manager import plugins_manager
+from utils.utils import get_matchers
+import nonebot
+
+try:
+    import ujson as json
+except ModuleNotFoundError:
+    import json
+
+
+_yaml = YAML(typ="safe")
+
+
+def init_plugins_data(data_path):
+    """
+    初始化插件数据信息
+    """
+    plugin2data_file = Path(data_path) / "manager" / "plugin_manager.json"
+    plugin2data_file.parent.mkdir(parents=True, exist_ok=True)
+    _data = {}
+    if plugin2data_file.exists():
+        _data = json.load(open(plugin2data_file, "r", encoding="utf8"))
+    _matchers = get_matchers()
+    for matcher in _matchers:
+        _plugin = nonebot.plugin.get_plugin(matcher.module)
+        try:
+            _module = _plugin.module
+        except AttributeError:
+            if matcher.module not in _data.keys():
+                plugins_manager.add_plugin_data(
+                    matcher.module, matcher.module, error=True
+                )
+            else:
+                plugins_manager.set_module_data(matcher.module, "error", True)
+                plugin_data = plugins_manager.get(matcher.module)
+                if plugin_data:
+                    plugins_manager.set_module_data(
+                        matcher.module, "version", plugin_data.get("version")
+                    )
+        else:
+            try:
+                plugin_version = _module.__getattribute__("__plugin_version__")
+            except AttributeError:
+                plugin_version = None
+            try:
+                plugin_name = _module.__getattribute__("__zx_plugin_name__")
+            except AttributeError:
+                plugin_name = matcher.module
+            try:
+                plugin_author = _module.__getattribute__("__plugin_author__")
+            except AttributeError:
+                plugin_author = None
+            if matcher.module in plugins_manager.keys():
+                plugins_manager.set_module_data(matcher.module, "error", False)
+            if matcher.module not in plugins_manager.keys():
+                plugins_manager.add_plugin_data(
+                    matcher.module,
+                    plugin_name=plugin_name,
+                    author=plugin_author,
+                    version=plugin_version,
+                )
+            elif plugins_manager[matcher.module]["version"] is None or (
+                plugin_version is not None
+                and plugin_version > plugins_manager[matcher.module]["version"]
+            ):
+                plugins_manager.set_module_data(
+                    matcher.module, "plugin_name", plugin_name
+                )
+                plugins_manager.set_module_data(matcher.module, "author", plugin_author)
+                plugins_manager.set_module_data(
+                    matcher.module, "version", plugin_version
+                )
+            if matcher.module in _data.keys():
+                plugins_manager.set_module_data(
+                    matcher.module, "error", _data[matcher.module]["error"]
+                )
+                plugins_manager.set_module_data(
+                    matcher.module, "plugin_name", _data[matcher.module]["plugin_name"]
+                )
+    plugins_manager.save()

basic_plugins/init_plugin_config/init_plugins_limit.py

@@ -0,0 +1,157 @@
+from pathlib import Path
+from ruamel.yaml import round_trip_load, round_trip_dump, YAML
+from utils.manager import (
+    plugins2cd_manager,
+    plugins2block_manager,
+    plugins2count_manager,
+)
+from utils.utils import get_matchers
+from ruamel import yaml
+import nonebot
+
+
+_yaml = YAML(typ="safe")
+
+
+def init_plugins_cd_limit(data_path):
+    """
+    加载 cd 限制
+    """
+    plugins2cd_file = Path(data_path) / "configs" / "plugins2cd.yaml"
+    plugins2cd_file.parent.mkdir(exist_ok=True, parents=True)
+    _data = {}
+    _matchers = get_matchers()
+    for matcher in _matchers:
+        if not plugins2cd_manager.get_plugin_cd_data(matcher.module):
+            _plugin = nonebot.plugin.get_plugin(matcher.module)
+            try:
+                _module = _plugin.module
+                plugin_cd_limit = _module.__getattribute__("__plugin_cd_limit__")
+                plugins2cd_manager.add_cd_limit(
+                    matcher.module, data_dict=plugin_cd_limit
+                )
+            except AttributeError:
+                pass
+    if not plugins2cd_manager.keys():
+        plugins2cd_manager.add_cd_limit(
+            "这是一个示例"
+        )
+    _tmp_data = {"PluginCdLimit": plugins2cd_manager.get_data()}
+    with open(plugins2cd_file, "w", encoding="utf8") as wf:
+        yaml.dump(_tmp_data, wf, Dumper=yaml.RoundTripDumper, allow_unicode=True)
+    _data = round_trip_load(open(plugins2cd_file, encoding="utf8"))
+    _data["PluginCdLimit"].yaml_set_start_comment(
+        """# 需要cd的功能
+# 自定义的功能需要cd也可以在此配置
+# key：模块名称
+# cd：cd 时长（秒）
+# status：此限制的开关状态
+# check_type：'private'/'group'/'all'，限制私聊/群聊/全部
+# limit_type：监听对象，以user_id或group_id作为键来限制，'user'：用户id，'group'：群id
+#                                     示例：'user'：用户N秒内触发1次，'group'：群N秒内触发1次
+# rst：回复的话，可以添加[at]，[uname]，[nickname]来对应艾特，用户群名称，昵称系统昵称
+# rst 为 "" 或 None 时则不回复
+# rst示例："[uname]你冲的太快了，[nickname]先生，请稍后再冲[at]"
+# rst回复："老色批你冲的太快了，欧尼酱先生，请稍后再冲@老色批"
+#      用户昵称↑     昵称系统的昵称↑          艾特用户↑""",
+        indent=2,
+    )
+    with open(plugins2cd_file, "w", encoding="utf8") as wf:
+        round_trip_dump(_data, wf, Dumper=yaml.RoundTripDumper, allow_unicode=True)
+    plugins2cd_manager.reload_cd_limit()
+
+
+def init_plugins_block_limit(data_path):
+    """
+    加载阻塞限制
+    """
+    plugins2block_file = Path(data_path) / "configs" / "plugins2block.yaml"
+    plugins2block_file.parent.mkdir(exist_ok=True, parents=True)
+    _data = {}
+    _matchers = get_matchers()
+    for matcher in _matchers:
+        if not plugins2block_manager.get_plugin_block_data(matcher.module):
+            _plugin = nonebot.plugin.get_plugin(matcher.module)
+            try:
+                _module = _plugin.module
+                plugin_block_limit = _module.__getattribute__("__plugin_block_limit__")
+                plugins2block_manager.add_block_limit(
+                    matcher.module, data_dict=plugin_block_limit
+                )
+            except AttributeError:
+                pass
+    if not plugins2block_manager.keys():
+        plugins2block_manager.add_block_limit(
+            "这是一个示例"
+        )
+    _tmp_data = {"PluginBlockLimit": plugins2block_manager.get_data()}
+    with open(plugins2block_file, "w", encoding="utf8") as wf:
+        yaml.dump(_tmp_data, wf, Dumper=yaml.RoundTripDumper, allow_unicode=True)
+    _data = round_trip_load(open(plugins2block_file, encoding="utf8"))
+    _data["PluginBlockLimit"].yaml_set_start_comment(
+        """# 用户调用阻塞
+# 即 当用户调用此功能还未结束时
+# 用发送消息阻止用户重复调用此命令直到该命令结束
+# key：模块名称
+# status：此限制的开关状态
+# check_type：'private'/'group'/'all'，限制私聊/群聊/全部
+# limit_type：监听对象，以user_id或group_id作为键来限制，'user'：用户id，'group'：群id
+#                                     示例：'user'：阻塞用户，'group'：阻塞群聊
+# rst：回复的话，可以添加[at]，[uname]，[nickname]来对应艾特，用户群名称，昵称系统昵称
+# rst 为 "" 或 None 时则不回复
+# rst示例："[uname]你冲的太快了，[nickname]先生，请稍后再冲[at]"
+# rst回复："老色批你冲的太快了，欧尼酱先生，请稍后再冲@老色批"
+#      用户昵称↑     昵称系统的昵称↑          艾特用户↑""",
+        indent=2,
+    )
+    with open(plugins2block_file, "w", encoding="utf8") as wf:
+        round_trip_dump(_data, wf, Dumper=yaml.RoundTripDumper, allow_unicode=True)
+    plugins2block_manager.reload_block_limit()
+
+
+def init_plugins_count_limit(data_path):
+    """
+    加载次数限制
+    """
+    plugins2count_file = Path(data_path) / "configs" / "plugins2count.yaml"
+    plugins2count_file.parent.mkdir(exist_ok=True, parents=True)
+    _data = {}
+    _matchers = get_matchers()
+    for matcher in _matchers:
+        if not plugins2count_manager.get_plugin_count_data(matcher.module):
+            _plugin = nonebot.plugin.get_plugin(matcher.module)
+            try:
+                _module = _plugin.module
+                plugin_count_limit = _module.__getattribute__("__plugin_count_limit__")
+                plugins2count_manager.add_count_limit(
+                    matcher.module, data_dict=plugin_count_limit
+                )
+            except AttributeError:
+                pass
+    if not plugins2count_manager.keys():
+        plugins2count_manager.add_count_limit(
+            "这是一个示例"
+        )
+    _tmp_data = {"PluginCountLimit": plugins2count_manager.get_data()}
+    with open(plugins2count_file, "w", encoding="utf8") as wf:
+        yaml.dump(_tmp_data, wf, Dumper=yaml.RoundTripDumper, allow_unicode=True)
+    _data = round_trip_load(open(plugins2count_file, encoding="utf8"))
+    _data["PluginCountLimit"].yaml_set_start_comment(
+        """# 命令每日次数限制
+# 即 用户/群聊 每日可调用命令的次数 [数据内存存储，重启将会重置]
+# 每日调用直到 00:00 刷新
+# key：模块名称
+# max_count: 每日调用上限
+# status：此限制的开关状态
+# limit_type：监听对象，以user_id或group_id作为键来限制，'user'：用户id，'group'：群id
+#                                     示例：'user'：用户上限，'group'：群聊上限
+# rst：回复的话，可以添加[at]，[uname]，[nickname]来对应艾特，用户群名称，昵称系统昵称
+# rst 为 "" 或 None 时则不回复
+# rst示例："[uname]你冲的太快了，[nickname]先生，请稍后再冲[at]"
+# rst回复："老色批你冲的太快了，欧尼酱先生，请稍后再冲@老色批"
+#      用户昵称↑     昵称系统的昵称↑          艾特用户↑""",
+        indent=2,
+    )
+    with open(plugins2count_file, "w", encoding="utf8") as wf:
+        round_trip_dump(_data, wf, Dumper=yaml.RoundTripDumper, allow_unicode=True)
+    plugins2count_manager.reload_count_limit()

basic_plugins/init_plugin_config/init_plugins_resources.py

@@ -0,0 +1,43 @@
+from utils.manager import resources_manager
+from utils.utils import get_matchers
+from services.log import logger
+from pathlib import Path
+import nonebot
+
+
+def init_plugins_resources():
+    """
+    资源文件路径的移动
+    """
+    _tmp = []
+    for matcher in get_matchers():
+        if matcher.module not in _tmp:
+            _tmp.append(matcher.module)
+            _plugin = nonebot.plugin.get_plugin(matcher.module)
+            try:
+                _module = _plugin.module
+            except AttributeError:
+                logger.warning(f"插件 {matcher.module} 加载失败...，资源控制未加载...")
+            else:
+                try:
+                    resources = _module.__getattribute__("__plugin_resources__")
+                except AttributeError:
+                    pass
+                else:
+                    path = Path(_module.__getattribute__("__file__")).parent
+                    for resource in resources.keys():
+                        resources_manager.add_resource(matcher.module, (path / resource).absolute(), resources[resource])
+    resources_manager.save()
+    resources_manager.start_move()
+
+
+
+
+
+
+
+
+
+
+
+

basic_plugins/init_plugin_config/init_plugins_settings.py

@@ -0,0 +1,130 @@
+from pathlib import Path
+from ruamel.yaml import round_trip_load, round_trip_dump, YAML
+from utils.manager import plugins2settings_manager, admin_manager
+from services.log import logger
+from utils.utils import get_matchers
+from ruamel import yaml
+import nonebot
+
+
+_yaml = YAML(typ="safe")
+
+
+def init_plugins_settings(data_path: str):
+    """
+    初始化插件设置，从插件中获取 __zx_plugin_name__，__plugin_cmd__，__plugin_settings__
+    """
+    plugins2settings_file = Path(data_path) / "configs" / "plugins2settings.yaml"
+    plugins2settings_file.parent.mkdir(exist_ok=True, parents=True)
+    _matchers = get_matchers()
+    _data = {}
+    if plugins2settings_file.exists():
+        with open(plugins2settings_file, "r", encoding="utf8") as f:
+            _data = _yaml.load(f)
+            _data = _data["PluginSettings"] if _data else {}
+    _tmp_module = {}
+    _tmp = []
+    for matcher in _matchers:
+        if matcher.module in _data.keys():
+            plugins2settings_manager.add_plugin_settings(
+                matcher.module,
+                plugin_type=_data[matcher.module]["plugin_type"],
+                data_dict=_data[matcher.module],
+            )
+            if _data[matcher.module]["cmd"]:
+                _tmp_module[matcher.module] = _data[matcher.module]["cmd"][0]
+        else:
+            _plugin = nonebot.plugin.get_plugin(matcher.module)
+            try:
+                _module = _plugin.module
+            except AttributeError:
+                logger.warning(f"插件 {matcher.module} 加载失败...，插件控制未加载.")
+            else:
+                try:
+                    plugin_name = _module.__getattribute__("__zx_plugin_name__")
+                    if "[admin]" in plugin_name.lower():
+                        try:
+                            admin_settings = _module.__getattribute__(
+                                "__plugin_settings__"
+                            )
+                            level = admin_settings["admin_level"]
+                            cmd = admin_settings.get("cmd")
+                        except (AttributeError, KeyError):
+                            level = 5
+                            cmd = None
+                        if not level:
+                            level = 5
+                        admin_manager.add_admin_plugin_settings(
+                            matcher.module, cmd, level
+                        )
+                    if (
+                        "[hidden]" in plugin_name.lower()
+                        or "[admin]" in plugin_name.lower()
+                        or "[superuser]" in plugin_name.lower()
+                        or matcher.module in plugins2settings_manager.keys()
+                    ):
+                        continue
+                except AttributeError:
+                    if matcher.module not in _tmp:
+                        logger.warning(
+                            f"获取插件 {matcher.module} __zx_plugin_name__ 失败...，插件控制未加载."
+                        )
+                else:
+                    try:
+                        _tmp_module[matcher.module] = plugin_name
+                        plugin_settings = _module.__getattribute__(
+                            "__plugin_settings__"
+                        )
+                        if (
+                            plugin_settings["cmd"]
+                            and plugin_name not in plugin_settings["cmd"]
+                        ):
+                            plugin_settings["cmd"].append(plugin_name)
+                        if plugins2settings_manager.get(
+                            matcher.module
+                        ) and plugins2settings_manager[matcher.module].get(
+                            "plugin_type"
+                        ):
+                            plugin_type = tuple(
+                                plugins2settings_manager.get_plugin_data(
+                                    matcher.module
+                                )["plugin_type"]
+                            )
+                        else:
+                            try:
+                                plugin_type = _module.__getattribute__(
+                                    "__plugin_type__"
+                                )
+                            except AttributeError:
+                                plugin_type = ("normal",)
+                        if plugin_settings and matcher.module:
+                            plugins2settings_manager.add_plugin_settings(
+                                matcher.module,
+                                plugin_type=plugin_type,
+                                data_dict=plugin_settings,
+                            )
+                    except AttributeError:
+                        pass
+        _tmp.append(matcher.module)
+    _tmp_data = {"PluginSettings": plugins2settings_manager.get_data()}
+    with open(plugins2settings_file, "w", encoding="utf8") as wf:
+        yaml.dump(_tmp_data, wf, Dumper=yaml.RoundTripDumper, allow_unicode=True)
+    _data = round_trip_load(open(plugins2settings_file, encoding="utf8"))
+    _data["PluginSettings"].yaml_set_start_comment(
+        """# 模块与对应命令和对应群权限
+# 用于生成帮助图片 和 开关功能
+# key：模块名称
+# level：需要的群等级
+# default_status：加入群时功能的默认开关状态
+# limit_superuser: 功能状态是否限制超级用户
+# cmd: 关闭[cmd] 都会触发命令 关闭对应功能，cmd列表第一个词为统计的功能名称
+# plugin_type: 帮助类别 示例：('原神相关',) 或 ('原神相关', 1)，1代表帮助命令列向排列，否则为横向排列""",
+        indent=2,
+    )
+    for plugin in _data["PluginSettings"].keys():
+        _data["PluginSettings"][plugin].yaml_set_start_comment(
+            f"{plugin}：{_tmp_module[plugin]}", indent=2
+        )
+    with open(plugins2settings_file, "w", encoding="utf8") as wf:
+        round_trip_dump(_data, wf, Dumper=yaml.RoundTripDumper, allow_unicode=True)
+    logger.info(f"已成功加载 {len(plugins2settings_manager.get_data())} 个非限制插件.")

basic_plugins/scripts.py

@@ -0,0 +1,111 @@
+from nonebot import Driver
+from services.db_context import db
+from asyncpg.exceptions import DuplicateColumnError
+from models.group_info import GroupInfo
+from nonebot.adapters.cqhttp import Bot
+from utils.user_agent import get_user_agent
+from services.log import logger
+from configs.path_config import TEXT_PATH
+from asyncio.exceptions import TimeoutError
+from pathlib import Path
+import aiohttp
+import nonebot
+
+try:
+    import ujson as json
+except ModuleNotFoundError:
+    import json
+
+
+driver: Driver = nonebot.get_driver()
+
+
+@driver.on_startup
+async def update_city():
+    """
+    部分插件需要中国省份城市
+    这里直接更新，避免插件内代码重复
+    :return:
+    """
+    china_city = Path(TEXT_PATH) / "china_city.json"
+    data = {}
+    try:
+        async with aiohttp.ClientSession(headers=get_user_agent()) as session:
+            async with session.get(
+                "http://www.weather.com.cn/data/city3jdata/china.html", timeout=5
+            ) as res:
+                provinces_data = json.loads(await res.text(encoding="utf8"))
+            for province in provinces_data.keys():
+                data[provinces_data[province]] = []
+                async with session.get(
+                    f"http://www.weather.com.cn/data/city3jdata/provshi/{province}.html",
+                    timeout=5,
+                ) as res:
+                    city_data = json.loads(await res.text(encoding="utf8"))
+                    for city in city_data.keys():
+                        data[provinces_data[province]].append(city_data[city])
+        with open(china_city, "w", encoding="utf8") as f:
+            json.dump(data, f, indent=4, ensure_ascii=False)
+        logger.info("自动更新城市列表完成.....")
+    except TimeoutError:
+        logger.info("自动更新城市列表超时.....")
+
+
+@driver.on_startup
+async def _():
+    """
+    数据库表结构变换
+    """
+    sql_str = [
+        "ALTER TABLE group_info ADD group_flag Integer NOT NULL DEFAULT 0;"     # group_info表添加一个group_flag
+    ]
+    for sql in sql_str:
+        try:
+            query = db.text(sql)
+            await db.first(query)
+        except DuplicateColumnError:
+            pass
+
+
+@driver.on_bot_connect
+async def _(bot: Bot):
+    """
+    版本某些需要的变换
+    """
+    # 清空不存在的群聊信息，并将已所有已存在的群聊group_flag设置为1（认证所有已存在的群）
+    if not await GroupInfo.get_group_info(114514):
+        # 标识符，该功能只需执行一次
+        await GroupInfo.add_group_info(
+            114514,
+            "114514",
+            114514,
+            114514,
+            1
+        )
+        group_list = await bot.get_group_list()
+        group_list = [g["group_id"] for g in group_list]
+        _gl = [x.group_id for x in await GroupInfo.get_all_group()]
+        if 114514 in _gl:
+            _gl.remove(114514)
+        for group_id in _gl:
+            if group_id in group_list:
+                if await GroupInfo.get_group_info(group_id):
+                    await GroupInfo.set_group_flag(group_id, 1)
+                else:
+                    group_info = await bot.get_group_info(group_id=group_id)
+                    await GroupInfo.add_group_info(
+                        group_info["group_id"],
+                        group_info["group_name"],
+                        group_info["max_member_count"],
+                        group_info["member_count"],
+                        1
+                    )
+                logger.info(f"已将群聊 {group_id} 添加认证...")
+            else:
+                await GroupInfo.delete_group_info(group_id)
+                logger.info(f"移除不存在的群聊信息：{group_id}")
+
+
+
+
+

basic_plugins/super_cmd/__init__.py

@@ -0,0 +1,10 @@
+import nonebot
+
+
+nonebot.load_plugins('basic_plugins/super_cmd')
+
+
+
+
+
+

basic_plugins/super_cmd/bot_friend_group.py

@@ -0,0 +1,152 @@
+from nonebot import on_command
+from nonebot.permission import SUPERUSER
+from nonebot.typing import T_State
+from nonebot.adapters.cqhttp import Bot, MessageEvent, Message
+from nonebot.rule import to_me
+from utils.utils import get_message_text, is_number
+from utils.manager import requests_manager
+from utils.message_builder import image
+from models.group_info import GroupInfo
+
+
+__zx_plugin_name__ = "显示所有好友群组 [Superuser]"
+__plugin_usage__ = """
+usage：
+    显示所有好友群组
+    指令：
+        查看所有好友/查看所有群组
+        同意好友请求 [id]
+        拒绝好友请求 [id]
+        同意群聊请求 [id]
+        拒绝群聊请求 [id]
+        查看所有请求
+        清空所有请求
+""".strip()
+__plugin_des__ = "显示所有好友群组"
+__plugin_cmd__ = [
+    "查看所有好友/查看所有群组",
+    "同意好友请求 [id]",
+    "拒绝好友请求 [id]",
+    "同意群聊请求 [id]",
+    "拒绝群聊请求 [id]",
+    "查看所有请求",
+    "清空所有请求",
+]
+__plugin_version__ = 0.1
+__plugin_author__ = "HibiKier"
+
+
+cls_group = on_command(
+    "查看所有群组", rule=to_me(), permission=SUPERUSER, priority=1, block=True
+)
+cls_friend = on_command(
+    "查看所有好友", rule=to_me(), permission=SUPERUSER, priority=1, block=True
+)
+
+friend_handle = on_command(
+    "同意好友请求", aliases={"拒绝好友请求"}, permission=SUPERUSER, priority=1, block=True
+)
+
+group_handle = on_command(
+    "同意群聊请求", aliases={"拒绝群聊请求"}, permission=SUPERUSER, priority=1, block=True
+)
+
+clear_request = on_command("清空所有请求", permission=SUPERUSER, priority=1, block=True)
+
+cls_request = on_command("查看所有请求", permission=SUPERUSER, priority=1, block=True)
+
+
+@cls_group.handle()
+async def _(bot: Bot, event: MessageEvent, state: T_State):
+    gl = await bot.get_group_list(self_id=int(bot.self_id))
+    msg = ["{group_id} {group_name}".format_map(g) for g in gl]
+    msg = "\n".join(msg)
+    msg = f"bot:{bot.self_id}\n| 群号 | 群名 | 共{len(gl)}个群\n" + msg
+    await bot.send_private_msg(
+        self_id=int(bot.self_id),
+        user_id=int(list(bot.config.superusers)[0]),
+        message=msg,
+    )
+
+
+@cls_friend.handle()
+async def _(bot: Bot, event: MessageEvent, state: T_State):
+    gl = await bot.get_friend_list(self_id=int(bot.self_id))
+    msg = ["{user_id} {nickname}".format_map(g) for g in gl]
+    msg = "\n".join(msg)
+    msg = f"| QQ号 | 昵称 | 共{len(gl)}个好友\n" + msg
+    await bot.send_private_msg(
+        self_id=int(bot.self_id),
+        user_id=int(list(bot.config.superusers)[0]),
+        message=msg,
+    )
+
+
+@friend_handle.handle()
+async def _(bot: Bot, event: MessageEvent, state: T_State):
+    id_ = get_message_text(event.json())
+    if is_number(id_):
+        id_ = int(id_)
+        if state["_prefix"]["raw_command"][:2] == "同意":
+            if await requests_manager.approve(bot, id_, "private"):
+                await friend_handle.send("同意好友请求成功..")
+            else:
+                await friend_handle.send("同意好友请求失败，可能是未找到此id的请求..")
+        else:
+            if await requests_manager.refused(bot, id_, "private"):
+                await friend_handle.send("拒绝好友请求成功..")
+            else:
+                await friend_handle.send("拒绝好友请求失败，可能是未找到此id的请求..")
+    else:
+        await friend_handle.send("id必须为纯数字！")
+
+
+@group_handle.handle()
+async def _(bot: Bot, event: MessageEvent, state: T_State):
+    id_ = get_message_text(event.json())
+    if is_number(id_):
+        id_ = int(id_)
+        if state["_prefix"]["raw_command"][:2] == "同意":
+            rid = await requests_manager.approve(bot, id_, "group")
+            if rid:
+                await friend_handle.send("同意群聊请求成功..")
+                if await GroupInfo.get_group_info(rid):
+                    await GroupInfo.set_group_flag(rid, 1)
+                else:
+                    group_info = await bot.get_group_info(group_id=rid)
+                    await GroupInfo.add_group_info(
+                        rid,
+                        group_info["group_name"],
+                        group_info["max_member_count"],
+                        group_info["member_count"],
+                        1
+                    )
+            else:
+                await friend_handle.send("同意群聊请求失败，可能是未找到此id的请求..")
+        else:
+            if await requests_manager.refused(bot, id_, "group"):
+                await friend_handle.send("拒绝群聊请求成功..")
+            else:
+                await friend_handle.send("拒绝群聊请求失败，可能是未找到此id的请求..")
+    else:
+        await friend_handle.send("id必须为纯数字！")
+
+
+@cls_request.handle()
+async def _(bot: Bot, event: MessageEvent, state: T_State):
+    _str = ""
+    for type_ in ["private", "group"]:
+        msg = await requests_manager.show(type_)
+        if msg:
+            _str += image(b64=msg)
+        else:
+            _str += "没有任何好友请求.." if type_ == "private" else "没有任何群聊请求.."
+        if type_ == "private":
+            _str += '\n--------------------\n'
+    await cls_request.send(Message(_str))
+
+
+@clear_request.handle()
+async def _(bot: Bot, event: MessageEvent, state: T_State):
+    requests_manager.clear()
+    await cls_request.send("已清空所有好友/群聊请求..")

basic_plugins/super_cmd/clear_data.py

@@ -0,0 +1,67 @@
+from nonebot import on_command
+from nonebot.permission import SUPERUSER
+from nonebot.typing import T_State
+from nonebot.adapters.cqhttp import Bot, MessageEvent
+from configs.path_config import TEMP_PATH
+from nonebot.rule import to_me
+from utils.utils import scheduler
+from services.log import logger
+from utils.manager import resources_manager
+import asyncio
+import time
+import os
+
+__zx_plugin_name__ = "清理临时数据 [Superuser]"
+__plugin_usage__ = """
+usage：
+    清理临时数据
+    指令：
+        清理临时数据
+""".strip()
+__plugin_des__ = "清理临时数据"
+__plugin_cmd__ = [
+    "清理临时数据",
+]
+__plugin_version__ = 0.1
+__plugin_author__ = "HibiKier"
+
+
+clear_data = on_command(
+    "清理临时数据", rule=to_me(), permission=SUPERUSER, priority=1, block=True
+)
+
+
+resources_manager.add_temp_dir(TEMP_PATH)
+
+
+@clear_data.handle()
+async def _(bot: Bot, event: MessageEvent, state: T_State):
+    await clear_data.send("开始清理临时数据....")
+    size = await asyncio.get_event_loop().run_in_executor(None, _clear_data)
+    await clear_data.send("共清理了 {:.2f}MB 的数据...".format(size / 1024 / 1024))
+
+
+def _clear_data() -> float:
+    size = 0
+    for dir_ in resources_manager.get_temp_data_dir():
+        if dir_.exists():
+            for file in os.listdir(dir_):
+                file = dir_ / file
+                try:
+                    if time.time() - os.path.getatime(file) > 300:
+                        file_size = os.path.getsize(file)
+                        file.unlink()
+                        size += file_size
+                except Exception as e:
+                    logger.error(f"清理临时数据错误...{type(e)}：{e}")
+    return float(size)
+
+
+@scheduler.scheduled_job(
+    "cron",
+    hour=1,
+    minute=1,
+)
+async def _():
+    size = await asyncio.get_event_loop().run_in_executor(None, _clear_data)
+    logger.info("自动清理临时数据完成，" + "共清理了 {:.2f}MB 的数据...".format(size / 1024 / 1024))

basic_plugins/super_cmd/manager_group.py

@@ -1,34 +1,45 @@
 from nonebot import on_command, on_regex
-from utils.utils import get_message_text, is_number
-from nonebot.rule import to_me
-from services.log import logger
-from nonebot.adapters.cqhttp import Bot, GroupMessageEvent, MessageEvent, GROUP
-from nonebot.typing import T_State
 from nonebot.permission import SUPERUSER
-from configs.config import NICKNAME
+from nonebot.typing import T_State
+from nonebot.adapters.cqhttp import Bot, MessageEvent, GROUP, GroupMessageEvent
+from nonebot.rule import to_me
+from utils.utils import get_message_text, is_number
 from utils.manager import group_manager, plugins2settings_manager
+from models.group_info import GroupInfo
+from services.log import logger
+from configs.config import NICKNAME
+from nonebot.adapters.cqhttp.exception import ActionFailed
 
-try:
-    import ujson as json
-except ModuleNotFoundError:
-    import json
 
-__zx_plugin_name__ = "群权限操作 [Superuser]"
+__zx_plugin_name__ = "管理群操作 [Superuser]"
 __plugin_usage__ = """
 usage：
-    对群权限 | 群白名单 的操作
+    群权限 | 群白名单 | 退出群 操作
     指令：
-        修改群权限 [group] [等级]
-        添加群白名单 *[group]
-        删除群白名单 *[group]
+        退群 [group_id]
+        修改群权限 [group_id] [等级]
+        添加群白名单 *[group_id]
+        删除群白名单 *[group_id]
+        添加群认证 *[group_id]
+        删除群认证 *[group_id]
         查看群白名单
 """.strip()
-__plugin_des__ = "对群权限 | 群白名单 的操作"
-__plugin_cmd__ = ["修改群权限 [group] [等级]", "添加群白名单 *[group]", "删除群白名单 *[group]", "查看群白名单"]
+__plugin_des__ = "管理群操作"
+__plugin_cmd__ = [
+    "退群 [group_id]",
+    "修改群权限 [group_id] [等级]",
+    "添加群白名单 *[group_id]",
+    "删除群白名单 *[group_id]",
+    "添加群认证 *[group_id]",
+    "删除群认证 *[group_id]",
+    "查看群白名单",
+]
 __plugin_version__ = 0.1
 __plugin_author__ = "HibiKier"
 
 
+del_group = on_command("退群", rule=to_me(), permission=SUPERUSER, priority=1, block=True)
+
 add_group_level = on_command("修改群权限", priority=1, permission=SUPERUSER, block=True)
 my_group_level = on_command(
     "查看群权限", aliases={"群权限"}, priority=5, permission=GROUP, block=True
@@ -40,7 +51,6 @@ what_up_group_level = on_regex(
     permission=GROUP,
     block=True,
 )
-
 manager_group_whitelist = on_command(
     "添加群白名单", aliases={"删除群白名单"}, priority=1, permission=SUPERUSER, block=True
 )
@@ -49,6 +59,29 @@ show_group_whitelist = on_command(
     "查看群白名单", priority=1, permission=SUPERUSER, block=True
 )
 
+group_auth = on_command(
+    "添加群认证", aliases={"删除群认证"}, priority=1, permission=SUPERUSER, block=True
+)
+
+
+@del_group.handle()
+async def _(bot: Bot, event: MessageEvent, state: T_State):
+    group_id = get_message_text(event.json())
+    if group_id:
+        if is_number(group_id):
+            try:
+                await bot.set_group_leave(group_id=int(group_id))
+                logger.info(f"退出群聊 {group_id} 成功")
+                await del_group.send(f"退出群聊 {group_id} 成功", at_sender=True)
+                group_manager.delete_group(int(group_id))
+                await GroupInfo.delete_group_info(int(group_id))
+            except Exception as e:
+                logger.info(f"退出群聊 {group_id} 失败 e:{e}")
+        else:
+            await del_group.finish(f"请输入正确的群号", at_sender=True)
+    else:
+        await del_group.finish(f"请输入群号", at_sender=True)
+
 
 @add_group_level.handle()
 async def _(bot: Bot, event: MessageEvent, state: T_State):
@@ -130,3 +163,38 @@ async def _(bot: Bot, event: MessageEvent, state: T_State):
         await show_group_whitelist.send("目前的群白名单：\n" + "\n".join(x))
     else:
         await show_group_whitelist.send("没有任何群在群白名单...")
+
+
+@group_auth.handle()
+async def _(bot: Bot, event: MessageEvent, state: T_State):
+    msg = get_message_text(event.json()).split()
+    for group_id in msg:
+        if not is_number(group_id):
+            await group_auth.send(f"{group_id}非纯数字，已跳过该项..")
+        group_id = int(group_id)
+        if state["_prefix"]["raw_command"][:2] == "添加":
+            if await GroupInfo.get_group_info(group_id):
+                await GroupInfo.set_group_flag(group_id, 1)
+            else:
+                try:
+                    group_info = await bot.get_group_info(group_id=group_id)
+                except ActionFailed:
+                    group_info = {
+                        "group_id": group_id,
+                        "group_name": "_",
+                        "max_member_count": -1,
+                        "member_count": -1,
+                    }
+                await GroupInfo.add_group_info(
+                    group_info["group_id"],
+                    group_info["group_name"],
+                    group_info["max_member_count"],
+                    group_info["member_count"],
+                    1,
+                )
+        else:
+            if await GroupInfo.get_group_info(group_id):
+                await GroupInfo.set_group_flag(group_id, 0)
+        await group_auth.send(
+            f'已为 {group_id} {state["_prefix"]["raw_command"][:2]}群认证..'
+        )

basic_plugins/super_cmd/reload_setting.py

@@ -0,0 +1,40 @@
+from nonebot import on_command
+from nonebot.permission import SUPERUSER
+from nonebot.rule import to_me
+from utils.manager import (
+    plugins2cd_manager,
+    plugins2settings_manager,
+    plugins2block_manager,
+    group_manager,
+)
+from nonebot.typing import T_State
+from nonebot.adapters.cqhttp import Bot, MessageEvent
+
+
+__zx_plugin_name__ = "重载插件配置 [Superuser]"
+__plugin_usage__ = """
+usage：
+    重载插件配置
+    指令：
+        重载插件配置
+""".strip()
+__plugin_des__ = "重载插件配置"
+__plugin_cmd__ = [
+    "重载插件配置",
+]
+__plugin_version__ = 0.1
+__plugin_author__ = "HibiKier"
+
+
+reload_plugins_manager = on_command(
+    "重载插件限制", rule=to_me(), permission=SUPERUSER, priority=1, block=True
+)
+
+
+@reload_plugins_manager.handle()
+async def _(bot: Bot, event: MessageEvent, state: T_State):
+    plugins2settings_manager.reload()
+    plugins2cd_manager.reload()
+    plugins2block_manager.reload()
+    group_manager.reload()
+    await reload_plugins_manager.send("重载完成...")

basic_plugins/super_cmd/set_admin_permissions.py

@@ -0,0 +1,89 @@
+from nonebot import on_command
+from nonebot.permission import SUPERUSER
+from models.level_user import LevelUser
+from nonebot.typing import T_State
+from nonebot.adapters.cqhttp import Bot, GroupMessageEvent, Message
+from utils.utils import get_message_at, get_message_text, is_number
+from services.log import logger
+from utils.message_builder import at
+
+__zx_plugin_name__ = "用户权限管理 [Superuser]"
+__plugin_usage__ = """
+usage：
+    增删改用户的权限
+    指令：
+        添加权限 [at] [权限]
+        添加权限 [qq] [group_id] [权限]
+        删除权限 [at]
+""".strip()
+__plugin_des__ = "增删改用户的权限"
+__plugin_cmd__ = [
+    "添加权限 [at] [权限]",
+    "添加权限 [qq] [group_id] [权限]",
+    "删除权限 [at]",
+]
+__plugin_version__ = 0.1
+__plugin_author__ = "HibiKier"
+
+
+super_cmd = on_command(
+    "添加管理",
+    aliases={"删除管理", "添加权限", "删除权限"},
+    priority=1,
+    permission=SUPERUSER,
+    block=True,
+)
+
+
+@super_cmd.handle()
+async def _(bot: Bot, event: GroupMessageEvent, state: T_State):
+    group_id = -1
+    level = 0
+    try:
+        args = get_message_text(event.json()).strip().split()
+        qq = get_message_at(event.json())
+        flag = -1
+        if not qq:
+            if len(args) > 2:
+                if is_number(args[0]) and is_number(args[1]) and is_number(args[2]):
+                    qq = int(args[0])
+                    group_id = int(args[1])
+                    level = int(args[2])
+                    flag = 1
+                else:
+                    await super_cmd.finish("所有参数必须是数字！", at_sender=True)
+            else:
+                await super_cmd.finish(
+                    "权限参数不完全\n\t格式：添加/删除权限 [at] [level]"
+                    "\n\t格式：添加/删除权限 [qq] [group] [level]",
+                    at_sender=True,
+                )
+        else:
+            qq = qq[0]
+            group_id = event.group_id
+            flag = 2
+        if state["_prefix"]["raw_command"][:2] == "添加":
+            if is_number(args[0]):
+                level = int(args[0])
+            else:
+                await super_cmd.finish("权限等级必须是数字！", at_sender=True)
+            if await LevelUser.set_level(qq, group_id, level, 1):
+                result = "添加管理成功, 权限: " + str(level)
+            else:
+                result = "管理已存在, 更新权限: " + str(level)
+        else:
+            if await LevelUser.delete_level(qq, event.group_id):
+                result = "删除管理成功!"
+            else:
+                result = "该账号无管理权限!"
+        if flag == 2:
+            await super_cmd.send(result)
+        elif flag == 1:
+            await bot.send_group_msg(
+                group_id=group_id,
+                message=Message(f"{at(qq)}管理员修改了你的权限" f"\n--------\n你当前的权限等级：{level}"),
+            )
+            await super_cmd.send("修改成功")
+    except Exception as e:
+        await super_cmd.send("执行指令失败!")
+        logger.error(f"执行指令失败 e：{e}")

basic_plugins/super_cmd/super_task_switch.py

@@ -0,0 +1,56 @@
+from nonebot import on_command
+from nonebot.permission import SUPERUSER
+from nonebot.typing import T_State
+from nonebot.adapters.cqhttp import Bot, MessageEvent
+from nonebot.rule import to_me
+from utils.utils import get_message_text, is_number
+from services.log import logger
+from utils.manager import group_manager
+
+
+__zx_plugin_name__ = "超级用户被动开关 [Superuser]"
+__plugin_usage__ = """
+usage：
+    超级用户被动开关
+    指令：
+        开启/关闭广播通知
+""".strip()
+__plugin_des__ = "超级用户被动开关"
+__plugin_cmd__ = [
+    "开启/关闭广播通知",
+]
+__plugin_version__ = 0.1
+__plugin_author__ = "HibiKier"
+
+
+oc_gb = on_command(
+    "开启广播通知",
+    aliases={"关闭广播通知"},
+    rule=to_me(),
+    permission=SUPERUSER,
+    priority=1,
+    block=True,
+)
+
+
+@oc_gb.handle()
+async def _(bot: Bot, event: MessageEvent, state: T_State):
+    group = get_message_text(event.json())
+    if group:
+        if is_number(group):
+            group = int(group)
+            for g in await bot.get_group_list():
+                if g["group_id"] == group:
+                    break
+            else:
+                await oc_gb.finish("没有加入这个群...", at_sender=True)
+            if state["_prefix"]["raw_command"] == "开启广播通知":
+                logger.info(f"USER {event.user_id} 开启了 GROUP {group} 的广播")
+                await oc_gb.finish(await group_manager.open_group_task(group, "broadcast",), at_sender=True)
+            else:
+                logger.info(f"USER {event.user_id} 关闭了 GROUP {group} 的广播")
+                await oc_gb.finish(await group_manager.close_group_task(group, "broadcast"), at_sender=True)
+        else:
+            await oc_gb.finish("请输入正确的群号", at_sender=True)
+    else:
+        await oc_gb.finish("请输入要关闭广播的群号", at_sender=True)

basic_plugins/super_cmd/update_friend_group_info.py

@@ -0,0 +1,71 @@
+from nonebot import on_command
+from nonebot.permission import SUPERUSER
+from nonebot.typing import T_State
+from nonebot.adapters.cqhttp import Bot, MessageEvent
+from nonebot.rule import to_me
+from utils.utils import get_bot
+from services.log import logger
+from models.group_info import GroupInfo
+from models.friend_user import FriendUser
+
+
+__zx_plugin_name__ = "更新群/好友信息 [Superuser]"
+__plugin_usage__ = """
+usage：
+    更新群/好友信息
+    指令：
+        更新群信息
+        更新好友信息
+""".strip()
+__plugin_des__ = "更新群/好友信息"
+__plugin_cmd__ = [
+    "更新群信息",
+    "更新好友信息",
+]
+__plugin_version__ = 0.1
+__plugin_author__ = "HibiKier"
+
+update_group_info = on_command(
+    "更新群信息", rule=to_me(), permission=SUPERUSER, priority=1, block=True
+)
+update_friend_info = on_command(
+    "更新好友信息", rule=to_me(), permission=SUPERUSER, priority=1, block=True
+)
+
+
+@update_group_info.handle()
+async def _(bot: Bot, event: MessageEvent, state: T_State):
+    bot = get_bot()
+    gl = await bot.get_group_list()
+    gl = [g["group_id"] for g in gl]
+    num = 0
+    rst = ""
+    for g in gl:
+        group_info = await bot.get_group_info(group_id=g)
+        if await GroupInfo.add_group_info(
+            group_info["group_id"],
+            group_info["group_name"],
+            group_info["max_member_count"],
+            group_info["member_count"],
+        ):
+            num += 1
+            logger.info(f"自动更新群组 {g} 信息成功")
+        else:
+            logger.info(f"自动更新群组 {g} 信息失败")
+            rst += f"{g} 更新失败\n"
+    await update_group_info.send(f"成功更新了 {num} 个群的信息\n{rst[:-1]}")
+
+
+@update_friend_info.handle()
+async def _(bot: Bot, event: MessageEvent, state: T_State):
+    num = 0
+    rst = ""
+    fl = await get_bot().get_friend_list()
+    for f in fl:
+        if await FriendUser.add_friend_info(f["user_id"], f["nickname"]):
+            logger.info(f'自动更新好友 {f["user_id"]} 信息成功')
+            num += 1
+        else:
+            logger.warning(f'自动更新好友 {f["user_id"]} 信息失败')
+            rst += f'{f["user_id"]} 更新失败\n'
+    await update_friend_info.send(f"成功更新了 {num} 个好友的信息\n{rst[:-1]}")

basic_plugins/super_help/data_source.py

@@ -33,8 +33,8 @@ def _create_help_image():
     width = 0
     help_str = "超级用户帮助\n\n*  注: ‘*’ 代表可有多个相同参数 ‘?’ 代表可省略该参数  *\n\n"
     tmp_img = CreateImg(0, 0, plain_text='1', font_size=24)
-    plugin_name = ""
     for matcher in _matchers:
+        plugin_name = ""
         try:
             _plugin = nonebot.plugin.get_plugin(matcher.module)
             _module = _plugin.module

bot.py

@@ -9,6 +9,9 @@ driver.register_adapter("cqhttp", CQHTTPBot)
 config = driver.config
 driver.on_startup(init)
 driver.on_shutdown(disconnect)
+# 优先加载定时任务插件
+nonebot.load_plugin("nonebot_plugin_apscheduler")
+nonebot.load_plugins("basic_plugins")
 nonebot.load_plugins("plugins")
 
 

configs/config.py

@@ -1,21 +1,11 @@
-from typing import List, Optional, Tuple
-from services.service_config import TL_M_KEY, SYSTEM_M_PROXY, ALAPI_M_TOKEN
-try:
-    import ujson as json
-except ModuleNotFoundError:
-    import json
+from typing import Optional
+from pathlib import Path
+from .utils import ConfigsManager
 
 
 # 回复消息名称
 NICKNAME: str = "小真寻"
 
-# API KEY（必要）
-RSSHUBAPP: str = "https://rsshub.app"  # rsshub
-ALAPI_TOKEN: str = ""  # ALAPI  https://admin.alapi.cn/user/login
-HIBIAPI: str = "https://api.obfs.dev"
-# 图灵
-TL_KEY: List[str] = []
-
 # 数据库（必要）
 # 如果填写了bind就不需要再填写后面的字段了#）
 # 示例："bind": "postgresql://user:password@127.0.0.1:5432/database"
@@ -27,113 +17,8 @@ address: str = ""  # 数据库地址
 port: str = ""  # 数据库端口
 database: str = ""  # 数据库名称
 
-# 代理
-SYSTEM_PROXY: Optional[str] = None  # 全局代理
-BUFF_PROXY: Optional[str] = None  # Buff代理
-
-# 公开图库列表
-IMAGE_DIR_LIST: List[str] = ["美图", "萝莉", "壁纸"]
-
-# 对被ban用户发送的消息
-BAN_RESULT: str = "才不会给你发消息."
-
-# PIX图库下载的画质 可能的值：original：原图，master：缩略图（加快发送速度）
-PIX_IMAGE_SIZE: str = "master"
-
-
-# 插件配置
-MAXINFO_REIMU: int = 7  # 上车(reimu)功能查找目的地的最大数
-COUNT_PER_DAY_REIMU: int = 5  # 每日上车(reimu)次数限制
-MAXINFO_BT: int = 10  # bt功能单次查找最大数
-MAXINFO_PRIVATE_ANIME: int = 20  # 私聊搜索动漫返回的最大数量
-MAXINFO_GROUP_ANIME: int = 5  # 群搜索动漫返回的最大数量
-MAX_FIND_IMG_COUNT: int = 3  # 识图最大返回数
-# 参1：延迟撤回色图时间(秒)，0 为关闭 | 参2：监控聊天类型，0(私聊) 1(群聊) 2(群聊+私聊)
-WITHDRAW_SETU_TIME: Tuple[int, int] = (0, 1)
-# 参1：延迟撤回PIX图片时间(秒)，0 为关闭 | 参2：监控聊天类型，0(私聊) 1(群聊) 2(群聊+私聊)
-WITHDRAW_PIX_TIME: Tuple[int, int] = (0, 1)
-
-# PIX图库 与 额外图库OmegaPixivIllusts 混合搜索的比例 参1：PIX图库 参2：OmegaPixivIllusts扩展图库（没有此图库请设置为0）
-PIX_OMEGA_PIXIV_RATIO: Tuple[int, int] = (10, 0)
-
-# 各种卡池的开关
-PRTS_FLAG = True  # 明日方舟
-GENSHIN_FLAG = True  # 原神
-PRETTY_FLAG = True  # 赛马娘
-GUARDIAN_FLAG = True  # 坎公骑冠剑
-PCR_FLAG = True  # 公主连结
-AZUR_FLAG = True  # 碧蓝航线
-FGO_FLAG = True  # 命运-冠位指定（FGO）
-ONMYOJI_FLAG = True  # 阴阳师
-
-PCR_TAI = True  # pcr是否开启台服卡池
-SEMAPHORE = 5  # 限制碧蓝航线和FGO并发数
-
-ADMIN_DEFAULT_AUTH: int = 5  # 默认群管理员权限
-
-MAX_SIGN_GOLD: int = 200  # 签到好感度加成额外获得的最大金币数
-MAX_RUSSIAN_BET_GOLD: int = 1000  # 俄罗斯轮盘最大赌注金额
-
-INITIAL_SETU_PROBABILITY: float = 0.7  # 色图概率
-FUDU_PROBABILITY: float = 0.7  # 复读概率
-
-INITIAL_OPEN_CASE_COUNT: int = 20  # 初始开箱次数
-MUTE_DEFAULT_COUNT: int = 10  # 刷屏禁言默认检测次数
-MUTE_DEFAULT_TIME: int = 7  # 刷屏检测默认规定时间
-MUTE_DEFAULT_DURATION: int = 10  # 刷屏检测默禁言时长（分钟）
-
-CHECK_NOTICE_INFO_CD = 300  # 群检测，个人权限检测等各种检测提示信息cd
-
-# 注：即在 MALICIOUS_CHECK_TIME 时间内触发相同命令 MALICIOUS_BAN_COUNT 将被ban MALICIOUS_BAN_TIME 分钟
-MALICIOUS_BAN_TIME: int = 30  # 恶意命令触发检测触发后ban的时长（分钟）
-MALICIOUS_BAN_COUNT: int = 6  # 恶意命令触发检测最大触发次数
-MALICIOUS_CHECK_TIME: int = 5  # 恶意命令触发检测规定时间内（秒）
-
-# LEVEL
-DELETE_IMG_LEVEL: int = 7  # 删除图片权限
-MOVE_IMG_LEVEL: int = 7  # 移动图片权限
-UPLOAD_IMG_LEVEL: int = 6  # 上传图片权限
-BAN_LEVEL: int = 5  # BAN权限
-OC_LEVEL: int = 2  # 开关群功能权限
-MUTE_LEVEL: int = 5  # 更改禁言设置权限
-MEMBER_ACTIVITY_LEVEL = 5  # 群员活跃检测设置权限
-GROUP_BILIBILI_SUB_LEVEL = 5  # 群内bilibili订阅需要的权限
-
-DEFAULT_GROUP_LEVEL = 5  # 默认群等级
-
-# 是否开启HIBIAPI搜图功能（该功能会搜索群友提交的xp）
-HIBIAPI_FLAG: bool = True
-# HIBIAPI搜图图片的最低收藏
-HIBIAPI_BOOKMARKS: int = 5000
-
-# 需要为哪些群更新最新版gocq吗？（上传最新版gocq）
-# 示例：[434995955, 239483248]
-UPDATE_GOCQ_GROUP: List[int] = []
-
-# 是否存储色图
-DOWNLOAD_SETU: bool = True
-# 仅仅使用本地色图
-ONLY_USE_LOCAL_SETU: bool = False
-# 是否自动同意好友添加
-AUTO_ADD_FRIEND: bool = True
-# 当含有ALAPI_TOKEN时是否检测文本合规，开启检测会减慢回复速度
-ALAPI_AI_CHECK: bool = True
-# 导入商店自带的三个商品
-IMPORT_DEFAULT_SHOP_GOODS: bool = True
-# 真寻是否自动更新
-AUTO_UPDATE_ZHENXUN: bool = True
-
-
-if TL_M_KEY:
-    TL_KEY = TL_M_KEY
-if SYSTEM_M_PROXY:
-    SYSTEM_PROXY = SYSTEM_M_PROXY
-if ALAPI_M_TOKEN:
-    ALAPI_TOKEN = ALAPI_M_TOKEN
-
-
-HIBIAPI = HIBIAPI[:-1] if HIBIAPI[-1] == "/" else HIBIAPI
-RSSHUBAPP = RSSHUBAPP[:-1] if RSSHUBAPP[-1] == "/" else RSSHUBAPP
-
+# 全局代理，例如 "http://127.0.0.1:7890"
+SYSTEM_PROXY: Optional[str] = None
 
 
+Config = ConfigsManager(Path() / "data" / "configs" / "plugins2config.yaml")

configs/utils/__init__.py

@@ -0,0 +1,159 @@
+from typing import Optional, Any, Union
+from pathlib import Path
+from ruamel.yaml import YAML
+from ruamel import yaml
+
+
+class ConfigsManager:
+    """
+    插件配置 与 资源 管理器
+    """
+
+    def __init__(self, file: Path):
+        self._data: dict = {}
+        self._admin_level_data = []
+        if file:
+            file.parent.mkdir(exist_ok=True, parents=True)
+            self.file = file
+            if file.exists():
+                _yaml = YAML()
+                with open(file, "r", encoding="utf8") as f:
+                    self._data = _yaml.load(f)
+
+    def add_plugin_config(
+        self,
+        module: str,
+        key: str,
+        value: Optional[Any],
+        *,
+        name: Optional[str] = None,
+        help_: Optional[str] = None,
+        default_value: Optional[str] = None,
+        _override: bool = False
+    ):
+        """
+        为插件添加一个配置，不会被覆盖，只有第一个生效
+        :param module: 模块
+        :param key: 键
+        :param value: 值
+        :param name: 插件名称
+        :param help_: 配置注解
+        :param default_value: 默认值
+        :param _override: 覆盖前值
+        """
+        if (
+            not (module in self._data.keys() and self._data[module].get(key))
+            or _override
+        ):
+            _module = None
+            if ":" in module:
+                module = module.split(":")
+                _module = module[-1]
+                module = module[0]
+            if "[LEVEL]" in key and _module:
+                key = key.replace("[LEVEL]", "").strip()
+                self._admin_level_data.append((_module, value))
+            if self._data.get(module) is None:
+                self._data[module] = {}
+            key = key.upper()
+            self._data[module][key] = {
+                "value": value,
+                "name": name.strip() if isinstance(name, str) else name,
+                "help": help_.strip() if isinstance(help_, str) else help_,
+                "default_value": default_value,
+            }
+
+    def remove_plugin_config(self, module: str):
+        """
+        为插件删除一个配置
+        :param module: 模块名
+        """
+        if module in self._data.keys():
+            del self._data[module]
+
+    def set_config(self, module: str, key: str, value: str):
+        """
+        设置配置值
+        :param module: 模块名
+        :param key: 配置名称
+        :param value: 值
+        """
+        if module in self._data.keys():
+            if self._data[module].get(key) is not None:
+                self._data[module][key]["value"] = value
+
+    def set_help(self, module: str, key: str, help_: str):
+        """
+        设置配置注释
+        :param module: 模块名
+        :param key: 配置名称
+        :param help_: 注释文本
+        """
+        if module in self._data.keys():
+            if self._data[module].get(key) is not None:
+                self._data[module][key]["help"] = help_
+
+    def set_default_value(self, module: str, key: str, value: str):
+        """
+        设置配置默认值
+        :param module: 模块名
+        :param key: 配置名称
+        :param value: 值
+        """
+        if module in self._data.keys():
+            if self._data[module].get(key) is not None:
+                self._data[module][key]["default_value"] = value
+
+    def get_config(self, module: str, key: str) -> Optional[Any]:
+        """
+        获取指定配置值
+        :param module: 模块名
+        :param key: 配置名称
+        """
+        key = key.upper()
+        if module in self._data.keys():
+            if self._data[module].get(key) is not None:
+                if self._data[module][key]["value"] is None:
+                    return self._data[module][key]["default_value"]
+                return self._data[module][key]["value"]
+        return None
+
+    def get(self, key: str):
+        """
+        获取插件配置数据
+        :param key: 名称
+        """
+        if key in self._data.keys():
+            return self._data[key]
+
+    def save(self, path: Union[str, Path] = None):
+        """
+        保存数据
+        :param path: 路径
+        """
+        path = path if path else self.file
+        with open(path, "w", encoding="utf8") as f:
+            yaml.dump(
+                self._data, f, indent=2, Dumper=yaml.RoundTripDumper, allow_unicode=True
+            )
+
+    def get_admin_level_data(self):
+        """
+        获取管理插件等级
+        """
+        return self._admin_level_data
+
+    def is_empty(self) -> bool:
+        return not bool(self._data)
+
+    def keys(self):
+        return self._data.keys()
+
+    def __str__(self):
+        return str(self._data)
+
+    def __setitem__(self, key, value):
+        self._data[key] = value
+
+    def __getitem__(self, key):
+        return self._data[key]

models/count_user.py

@@ -1,63 +0,0 @@
-from services.db_context import db
-
-
-class UserCount(db.Model):
-    __tablename__ = "count_users"
-
-    user_qq = db.Column(db.BigInteger(), nullable=False, primary_key=True)
-    reimu_count = db.Column(db.Integer(), nullable=False, default=0)
-    setu_r18_count = db.Column(db.Integer(), nullable=False, default=0)
-
-    _idx1 = db.Index("sign_reimu_users_idx1", "user_qq", unique=True)
-
-    @classmethod
-    async def add_count(cls, user_qq: int, name: str, count: int = 1):
-        """
-        说明：
-            用户添加次数
-        参数：
-            :param user_qq: qq号
-            :param name: 目标名称
-            :param count: 增加次数
-        """
-        query = cls.query.where((cls.user_qq == user_qq))
-        query = query.with_for_update()
-        user = await query.gino.first()
-        user = user if user else await cls.create(user_qq=user_qq)
-        if name == "reimu":
-            await user.update(reimu_count=cls.reimu_count + count).apply()
-        if name == "setu_r18":
-            await user.update(setu_r18_count=cls.setu_r18_count + count).apply()
-
-    @classmethod
-    async def check_count(cls, user_qq: int, name: str, max_count: int) -> bool:
-        """
-        说明：
-            检测次数是否到达最大值
-        参数：
-            :param user_qq: qq号
-            :param name: 目标名称
-            :param max_count: 最大值
-        """
-        query = cls.query.where((cls.user_qq == user_qq))
-        user = await query.gino.first()
-        user = user if user else await cls.create(user_qq=user_qq)
-        if name == "reimu":
-            if user.reimu_count == max_count:
-                return True
-            else:
-                return False
-        if name == "setu_r18":
-            if user.setu_r18_count == max_count:
-                return True
-            else:
-                return False
-
-    @classmethod
-    async def reset_count(cls):
-        """
-        说明：
-            重置每日次数
-        """
-        for user in await cls.query.gino.all():
-            await user.update(reimu_count=0, setu_r18_count=0).apply()

models/group_info.py

@@ -1,5 +1,6 @@
 from services.db_context import db
-from typing import List
+from services.log import logger
+from typing import List, Optional
 
 
 class GroupInfo(db.Model):
@@ -9,6 +10,7 @@ class GroupInfo(db.Model):
     group_name = db.Column(db.Unicode(), nullable=False, default="")
     max_member_count = db.Column(db.Integer(), nullable=False, default=0)
     member_count = db.Column(db.Integer(), nullable=False, default=0)
+    group_flag = db.Column(db.Integer(), nullable=False, default=0)
 
     _idx1 = db.Index("group_info_idx1", "group_id", unique=True)
 
@@ -25,7 +27,12 @@ class GroupInfo(db.Model):
 
     @classmethod
     async def add_group_info(
-        cls, group_id: int, group_name: str, max_member_count: int, member_count: int
+        cls,
+        group_id: int,
+        group_name: str,
+        max_member_count: int,
+        member_count: int,
+        group_flag: Optional[int] = None,
     ) -> bool:
         """
         说明：
@@ -35,6 +42,7 @@ class GroupInfo(db.Model):
             :param group_name: 群名称
             :param max_member_count: 群员最大数量
             :param member_count: 群员数量
+            :param group_flag: 群认证，0为未认证，1为认证
         """
         try:
             group = (
@@ -43,36 +51,35 @@ class GroupInfo(db.Model):
                 .gino.first()
             )
             if group:
-                await cls.update(
-                    group_id=group_id,
+                await group.update(
                     group_name=group_name,
                     max_member_count=max_member_count,
                     member_count=member_count,
                 ).apply()
+                if group_flag is not None:
+                    await group.update(group_flag=group_flag).apply()
             else:
                 await cls.create(
                     group_id=group_id,
                     group_name=group_name,
                     max_member_count=max_member_count,
                     member_count=member_count,
+                    group_flag=group_flag,
                 )
             return True
-        except Exception:
+        except Exception as e:
+            logger.info(f"GroupInfo 调用 add_group_info 发生错误 {type(e)}：{e}")
             return False
 
     @classmethod
-    async def delete_group_info(cls, group_id: int) -> bool:
+    async def delete_group_info(cls, group_id: int):
         """
         说明：
             删除群信息
         参数：
             :param group_id: 群号
         """
-        try:
-            await cls.delete.where(cls.group_id == group_id).gino.status()
-            return True
-        except Exception:
-            return False
+        await cls.delete.where(cls.group_id == group_id).gino.status()
 
     @classmethod
     async def get_all_group(cls) -> List["GroupInfo"]:
@@ -82,3 +89,23 @@ class GroupInfo(db.Model):
         """
         query = await cls.query.gino.all()
         return query
+
+    @classmethod
+    async def set_group_flag(cls, group_id: int, group_flag: int) -> bool:
+        """
+        设置群认证
+        :param group_id: 群号
+        :param group_flag: 群认证，0为未认证，1为认证
+        """
+        group = (
+            await cls.query.where(cls.group_id == group_id)
+            .with_for_update()
+            .gino.first()
+        )
+        if group:
+            if group.group_flag != group_flag:
+                await group.update(
+                    group_flag=group_flag,
+                ).apply()
+            return True
+        return False

models/group_remind.py

@@ -1,105 +0,0 @@
-from services.db_context import db
-
-
-class GroupRemind(db.Model):
-    __tablename__ = "group_reminds"
-
-    id = db.Column(db.Integer(), primary_key=True)
-    group_id = db.Column(db.BigInteger(), nullable=False)
-    hy = db.Column(db.Boolean(), default=False)  # 进群欢迎
-    kxcz = db.Column(db.Boolean(), default=False)  # 开箱重置
-    zwa = db.Column(db.Boolean(), default=False)  # 早晚安
-    gb = db.Column(db.Boolean(), default=True)  # 广播
-    blpar = db.Column(db.Boolean(), default=True)  # bilibili转发解析
-    pa = db.Column(db.Boolean(), default=True)  # 爬
-    epic = db.Column(db.Boolean(), default=False)  # epic
-    almanac = db.Column(db.Boolean(), default=False)  # 原神黄历
-
-    _idx1 = db.Index("info_group_reminds_idx1", "group_id", unique=True)
-
-    @classmethod
-    async def get_status(cls, group_id: int, name: str) -> bool:
-        """
-        说明：
-            获取群通知状态
-        参数：
-            :param group_id: 群号
-            :param name: 目标名称
-        """
-        group = await cls.query.where((cls.group_id == group_id)).gino.first()
-        if not group:
-            group = await cls.create(
-                group_id=group_id,
-            )
-        if name == "hy":
-            return group.hy
-        if name == "kxcz":
-            return group.kxcz
-        if name == "zwa":
-            return group.zwa
-        if name == "gb":
-            return group.gb
-        if name == "blpar":
-            return group.blpar
-        if name == "epic":
-            return group.epic
-        if name == "pa":
-            return group.pa
-        if name == "almanac":
-            return group.almanac
-
-    @classmethod
-    async def set_status(cls, group_id: int, name: str, status: bool) -> bool:
-        """
-        说明：
-            设置群通知状态
-        参数：
-            :param group_id: 群号
-            :param name: 目标名称
-            :param status: 通知状态
-        """
-        try:
-            group = (
-                await cls.query.where((cls.group_id == group_id))
-                .with_for_update()
-                .gino.first()
-            )
-            if not group:
-                group = await cls.create(
-                    group_id=group_id,
-                )
-            if name == "hy":
-                await group.update(
-                    hy=status,
-                ).apply()
-            if name == "kxcz":
-                await group.update(
-                    kxcz=status,
-                ).apply()
-            if name == "zwa":
-                await group.update(
-                    zwa=status,
-                ).apply()
-            if name == "gb":
-                await group.update(
-                    gb=status,
-                ).apply()
-            if name == "blpar":
-                await group.update(
-                    blpar=status,
-                ).apply()
-            if name == "epic":
-                await group.update(
-                    epic=status,
-                ).apply()
-            if name == "pa":
-                await group.update(
-                    pa=status,
-                ).apply()
-            if name == "almanac":
-                await group.update(
-                    almanac=status,
-                ).apply()
-            return True
-        except Exception as e:
-            return False