zhenxun_bot/basic_plugins/super_cmd/manager_group.py

from typing import Tuple

from nonebot import on_command, on_regex
from nonebot.adapters.onebot.v11 import (
    GROUP,
    Bot,
    GroupMessageEvent,
    Message,
    MessageEvent,
)
from nonebot.params import Command, CommandArg
from nonebot.permission import SUPERUSER
from nonebot.rule import to_me

from configs.config import NICKNAME
from models.group_info import GroupInfo
from services.log import logger
from utils.depends import OneCommand
from utils.image_utils import text2image
from utils.manager import group_manager, plugins2settings_manager
from utils.message_builder import image
from utils.utils import is_number

__zx_plugin_name__ = "管理群操作 [Superuser]"
__plugin_usage__ = """
usage：
    群权限 | 群白名单 | 退出群 操作
    退群，添加/删除群白名单，添加/删除群认证，当在群聊中这五个命令且没有指定群号时，默认指定当前群聊
    指令:
        退群 ?[group_id]
        修改群权限 [group_id] [等级]
        修改群权限 [等级]: 该命令仅在群聊时生效，默认修改当前群聊
        添加群白名单 ?*[group_id]
        删除群白名单 ?*[group_id]
        添加群认证 ?*[group_id]
        删除群认证 ?*[group_id]
        查看群白名单
""".strip()
__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
)
what_up_group_level = on_regex(
    "(:?提高|提升|升高|增加|加上).*?群权限",
    rule=to_me(),
    priority=5,
    permission=GROUP,
    block=True,
)
manager_group_whitelist = on_command(
    "添加群白名单", aliases={"删除群白名单"}, priority=1, permission=SUPERUSER, block=True
)

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, arg: Message = CommandArg()):
    group_id = arg.extract_plain_text().strip()
    if not group_id and isinstance(event, GroupMessageEvent):
        group_id = event.group_id
    if group_id:
        if is_number(group_id):
            group_list = [x["group_id"] for x in await bot.get_group_list()]
            group_id = int(group_id)
            if group_id not in group_list:
                logger.debug("群聊不存在", "退群", event.user_id, target=group_id)
                await del_group.finish(f"{NICKNAME}未在该群聊中...")
            try:
                await bot.set_group_leave(group_id=group_id)
                logger.info(f"{NICKNAME}退出群聊成功", "退群", event.user_id, target=group_id)
                await del_group.send(f"退出群聊 {group_id} 成功", at_sender=True)
                group_manager.delete_group(group_id)
                await GroupInfo.filter(group_id=group_id).delete()
            except Exception as e:
                logger.error(f"退出群聊失败", "退群", event.user_id, target=group_id, e=e)
                await del_group.send(f"退出群聊 {group_id} 失败", at_sender=True)
        else:
            await del_group.send(f"请输入正确的群号", at_sender=True)
    else:
        await del_group.send(f"请输入群号", at_sender=True)


@add_group_level.handle()
async def _(bot: Bot, event: MessageEvent, arg: Message = CommandArg()):
    msg = arg.extract_plain_text().strip()
    msg = msg.split()
    group_id = 0
    level = 0
    if isinstance(event, GroupMessageEvent) and len(msg) == 1:
        msg = [event.group_id, msg[0]]
    if not msg:
        await add_group_level.finish("缺失参数...")
    if len(msg) < 2:
        await add_group_level.finish("缺失参数...")
    if is_number(msg[0]) and is_number(msg[1]):
        group_id = int(msg[0])
        level = int(msg[1])
    else:
        await add_group_level.finish("参数错误...群号和等级必须是数字..")
    old_level = group_manager.get_group_level(group_id)
    group_manager.set_group_level(group_id, level)
    await add_group_level.send("修改成功...", at_sender=True)
    if level > -1:
        await bot.send_group_msg(
            group_id=group_id, message=f"管理员修改了此群权限：{old_level} -> {level}"
        )
    logger.info(f"修改群权限：{level}", "修改群权限", event.user_id, target=group_id)


@my_group_level.handle()
async def _(event: GroupMessageEvent):
    level = group_manager.get_group_level(event.group_id)
    tmp = ""
    data = plugins2settings_manager.get_data()
    for module in data:
        if data[module].level > level:
            plugin_name = data[module].cmd[0]
            if plugin_name == "pixiv":
                plugin_name = "搜图 p站排行"
            tmp += f"{plugin_name}\n"
    if not tmp:
        await my_group_level.finish(f"当前群权限：{level}")
    await my_group_level.finish(
        f"当前群权限：{level}\n目前无法使用的功能:\n"
        + image(await text2image(tmp, padding=10, color="#f9f6f2"))
    )


@what_up_group_level.handle()
async def _():
    await what_up_group_level.finish(
        f"[此功能用于防止内鬼，如果引起不便那真是抱歉了]\n" f"目前提高群权限的方法：\n" f"\t1.超级管理员修改权限"
    )


@manager_group_whitelist.handle()
async def _(
    bot: Bot, event: MessageEvent, cmd: str = OneCommand(), arg: Message = CommandArg()
):
    msg = arg.extract_plain_text().strip().split()
    if not msg and isinstance(event, GroupMessageEvent):
        msg = [event.group_id]
    if not msg:
        await manager_group_whitelist.finish("请输入群号")
    all_group = [g["group_id"] for g in await bot.get_group_list()]
    error_group = []
    group_list = []
    for group_id in msg:
        if is_number(group_id) and int(group_id) in all_group:
            group_list.append(int(group_id))
        else:
            logger.debug(f"群号不合法或不存在", cmd, target=group_id)
            error_group.append(group_id)
    if group_list:
        for group_id in group_list:
            if cmd in ["添加群白名单"]:
                group_manager.add_group_white_list(group_id)
            else:
                group_manager.delete_group_white_list(group_id)
        group_list = [str(x) for x in group_list]
        await manager_group_whitelist.send("已成功将 " + "\n".join(group_list) + " " + cmd)
        group_manager.save()
    if error_group:
        await manager_group_whitelist.send("以下群聊不合法或不存在:\n" + "\n".join(error_group))


@show_group_whitelist.handle()
async def _():
    group = [str(g) for g in group_manager.get_group_white_list()]
    if not group:
        await show_group_whitelist.finish("没有任何群在群白名单...")
    await show_group_whitelist.send("目前的群白名单:\n" + "\n".join(group))


@group_auth.handle()
async def _(
    bot: Bot, event: MessageEvent, cmd: str = OneCommand(), arg: Message = CommandArg()
):
    msg = arg.extract_plain_text().strip().split()
    if isinstance(event, GroupMessageEvent) and not msg:
        msg = [event.group_id]
    if not msg:
        await manager_group_whitelist.finish("请输入群号")
    error_group = []
    all_group = [g["group_id"] for g in await bot.get_group_list()]
    for group_id in msg:
        group_id = int(group_id)
        if is_number(group_id) and group_id in all_group:
            if cmd[:2] == "添加":
                try:
                    group_info = await bot.get_group_info(group_id=group_id)
                    await GroupInfo.update_or_create(
                        group_id=group_info["group_id"],
                        defaults={
                            "group_flag": 1,
                            "group_name": group_info["group_name"],
                            "max_member_count": group_info["max_member_count"],
                            "member_count": group_info["member_count"],
                        },
                    )
                except Exception as e:
                    await group_auth.send(f"添加群认证 {group_id} 发生错误！")
                    logger.error(f"添加群认证发生错误", cmd, target=group_id, e=e)
                else:
                    await group_auth.send(f"已为 {group_id} {cmd[:2]}群认证..")
                    logger.info(f"添加群认证成功", cmd, target=group_id)
            else:
                if group := await GroupInfo.filter(group_id=group_id).first():
                    await group.update_or_create(
                        group_id=group_id, defaults={"group_flag": 0}
                    )
                    await group_auth.send(f"已删除 {group_id} 群认证..")
                    logger.info(f"删除群认证成功", cmd, target=group_id)
                else:
                    await group_auth.send(f"未查找到群聊: {group_id}")
                    logger.info(f"未找到群聊", cmd, target=group_id)
        else:
            logger.debug(f"群号不合法或不存在", cmd, target=group_id)
            error_group.append(str(group_id))
    if error_group:
        await manager_group_whitelist.send("以下群聊不合法或不存在:\n" + "\n".join(error_group))