update v0.0.5.0

plugins/admin_bot_manage/switch_rule.py

@@ -1,78 +1,106 @@
-from nonebot import on_command
+from nonebot import on_command, on_message
 from nonebot.typing import T_State
-from nonebot.adapters.cqhttp import Bot, GroupMessageEvent, MessageEvent
-from .data_source import change_group_switch, set_plugin_status, get_plugin_status
+from nonebot.adapters.cqhttp import Bot, GroupMessageEvent, MessageEvent, GROUP
+from .data_source import (
+    change_group_switch,
+    set_plugin_status,
+    get_plugin_status,
+    group_current_status,
+)
 from services.log import logger
-from configs.config import plugins2info_dict, NICKNAME
+from configs.config import NICKNAME, OC_LEVEL
 from utils.utils import get_message_text, is_number
 from nonebot.permission import SUPERUSER
+from .rule import switch_rule
 
 
-__plugin_name__ = "群功能开关"
+__zx_plugin_name__ = "群功能开关 [Admin]"
 
 __plugin_usage__ = """
-    示例：
-        开启色图
-        关闭色图
-"""
+usage：
+    群内功能与被动技能开关
+    指令：
+        开启/关闭[功能]
+        群被动状态
+        开启全部被动
+        关闭全部被动
+        示例：开启/关闭色图
+""".strip()
+__plugin_superuser_usage__ = """
+usage:
+    功能总开关与指定群禁用
+    指令：
+        功能状态
+        开启/关闭[功能] [group]
+        开启/关闭[功能] ['private'/'group']
+""".strip()
+__plugin_des__ = "群内功能开关"
+__plugin_cmd__ = [
+    "开启/关闭[功能]",
+    "群被动状态",
+    "开启全部被动",
+    "关闭全部被动",
+    "功能状态 [_superuser]",
+    "开启/关闭[功能] [group] [_superuser]",
+    "开启/关闭[功能] ['private'/'group'] [_superuser]",
+]
+__plugin_version__ = 0.1
+__plugin_author__ = "HibiKier"
+__plugin_settings__ = {
+    "admin_level": OC_LEVEL,
+}
+
+switch_rule_matcher = on_message(rule=switch_rule, priority=4, block=True)
+
+plugins_status = on_command("功能状态", permission=SUPERUSER, priority=5, block=True)
+
+group_task_status = on_command("群被动状态", permission=GROUP, priority=5, block=True)
 
 
-cmds = []
-for cmd_list in plugins2info_dict.values():
-    for cmd in cmd_list["cmd"]:
-        cmds.append(f"开启{cmd}")
-        cmds.append(f"关闭{cmd}")
-cmds = set(cmds)
-
-
-switch_rule = on_command(
-    "switch_rule", aliases=cmds, priority=5, block=True
-)
-
-plugins_status = on_command('功能状态', permission=SUPERUSER, priority=5, block=True)
-
-
-@switch_rule.handle()
+@switch_rule_matcher.handle()
 async def _(bot: Bot, event: MessageEvent, state: T_State):
+    _cmd = get_message_text(event.json()).split()[0]
     if isinstance(event, GroupMessageEvent):
-        await switch_rule.send(
-            change_group_switch(
-                state["_prefix"]["raw_command"].strip(), event.group_id
-            )
-        )
-        logger.info(
-            f'USER {event.user_id} GROUP {event.group_id} 使用群功能管理命令 {state["_prefix"]["raw_command"]}'
-        )
+        await switch_rule_matcher.send(await change_group_switch(_cmd, event.group_id))
+        logger.info(f"USER {event.user_id} GROUP {event.group_id} 使用群功能管理命令 {_cmd}")
     else:
         if str(event.user_id) in bot.config.superusers:
-            block_type = get_message_text(event.json())
-            block_type = block_type if block_type else 'a'
-            _cmd = state["_prefix"]["raw_command"].strip()
+            block_type = " ".join(get_message_text(event.json()).split()[1:])
+            block_type = block_type if block_type else "a"
             if is_number(block_type):
-                if not int(block_type) in [g["group_id"] for g in await bot.get_group_list(self_id=int(bot.self_id))]:
-                    await switch_rule.finish(f'{NICKNAME}未加入群聊：{block_type}')
-                change_group_switch(_cmd, int(block_type), True)
-                group_name = (await bot.get_group_info(group_id=int(block_type)))['group_name']
-                await switch_rule.send(f'已禁用群聊 {group_name}({block_type}) 的 {_cmd[2:]} 功能')
-            elif block_type in ['all', 'private', 'group', 'a', 'p', 'g']:
-                block_type = 'all' if block_type == 'a' else block_type
-                block_type = 'private' if block_type == 'p' else block_type
-                block_type = 'group' if block_type == 'g' else block_type
+                if not int(block_type) in [
+                    g["group_id"]
+                    for g in await bot.get_group_list(self_id=int(bot.self_id))
+                ]:
+                    await switch_rule_matcher.finish(f"{NICKNAME}未加入群聊：{block_type}")
+                await change_group_switch(_cmd, int(block_type), True)
+                group_name = (await bot.get_group_info(group_id=int(block_type)))[
+                    "group_name"
+                ]
+                await switch_rule_matcher.send(
+                    f"已禁用群聊 {group_name}({block_type}) 的 {_cmd[2:]} 功能"
+                )
+            elif block_type in ["all", "private", "group", "a", "p", "g"]:
+                block_type = "all" if block_type == "a" else block_type
+                block_type = "private" if block_type == "p" else block_type
+                block_type = "group" if block_type == "g" else block_type
                 set_plugin_status(_cmd, block_type)
-                if block_type == 'all':
-                    await switch_rule.send(f'已{_cmd[:2]}功能：{_cmd[2:]}')
-                elif block_type == 'private':
-                    await switch_rule.send(f'已在私聊中{_cmd[:2]}功能：{_cmd[2:]}')
+                if block_type == "all":
+                    await switch_rule_matcher.send(f"已{_cmd[:2]}功能：{_cmd[2:]}")
+                elif block_type == "private":
+                    await switch_rule_matcher.send(f"已在私聊中{_cmd[:2]}功能：{_cmd[2:]}")
                 else:
-                    await switch_rule.send(f'已在群聊中{_cmd[:2]}功能：{_cmd[2:]}')
+                    await switch_rule_matcher.send(f"已在群聊中{_cmd[:2]}功能：{_cmd[2:]}")
             else:
-                await switch_rule.finish('格式错误：关闭[功能] [group]/[p/g]')
-            logger.info(
-                f'USER {event.user_id} 使用功能管理命令 {state["_prefix"]["raw_command"]} | {block_type}'
-            )
+                await switch_rule_matcher.finish("格式错误：关闭[功能] [group]/[p/g]")
+            logger.info(f"USER {event.user_id} 使用功能管理命令 {_cmd} | {block_type}")
 
 
 @plugins_status.handle()
 async def _(bot: Bot, event: MessageEvent, state: T_State):
     await plugins_status.send(await get_plugin_status())
 
+
+@group_task_status.handle()
+async def _(bot: Bot, event: GroupMessageEvent, state: T_State):
+    await group_task_status.send(await group_current_status(event.group_id))