mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
107 lines
4.1 KiB
Python
107 lines
4.1 KiB
Python
from nonebot import on_command, on_message
|
||
from nonebot.typing import T_State
|
||
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 NICKNAME, OC_LEVEL
|
||
from utils.utils import get_message_text, is_number
|
||
from nonebot.permission import SUPERUSER
|
||
from .rule import switch_rule
|
||
|
||
|
||
__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)
|
||
|
||
|
||
@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_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 = " ".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_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_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_matcher.send(f"已在群聊中{_cmd[:2]}功能：{_cmd[2:]}")
|
||
else:
|
||
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))
|