import random from typing import Any, List from nonebot import on_regex from nonebot.adapters import Bot from nonebot.matcher import Matcher from nonebot.params import Depends, RegexGroup from nonebot.plugin import PluginMetadata from nonebot.rule import to_me from nonebot_plugin_alconna import Alconna, Option, UniMsg, on_alconna, store_true from nonebot_plugin_saa import Text from nonebot_plugin_session import EventSession from nonebot_plugin_userinfo import EventUserInfo, UserInfo from zhenxun.configs.config import NICKNAME, Config from zhenxun.configs.utils import PluginExtraData, RegisterConfig from zhenxun.models.ban_console import BanConsole from zhenxun.models.friend_user import FriendUser from zhenxun.models.group_member_info import GroupInfoUser from zhenxun.services.log import logger from zhenxun.utils.enum import PluginType __plugin_meta__ = PluginMetadata( name="昵称系统", description="区区昵称，才不想叫呢！", usage=f""" 个人昵称，将替换{NICKNAME}称呼你的名称，群聊 与 私聊 昵称相互独立，全局昵称设置将更改您目前所有群聊中及私聊的昵称 指令： 以后叫我 [昵称]: 设置当前群聊/私聊的昵称 全局昵称设置 [昵称]: 设置当前所有群聊和私聊的昵称 {NICKNAME}我是谁 """.strip(), extra=PluginExtraData( author="HibiKier", version="0.1", plugin_type=PluginType.NORMAL, menu_type="其他", configs=[ RegisterConfig( key="BLACK_WORD", value=["爸", "爹", "爷", "父"], help="昵称所屏蔽的关键词，已设置的昵称会被替换为 *，未设置的昵称会在设置时提示", default_value=None, type=List[str], ) ], ).dict(), ) _nickname_matcher = on_regex( "(?:以后)?(?:叫我|请叫我|称呼我)(.*)", rule=to_me(), priority=5, block=True, ) _global_nickname_matcher = on_regex( "设置全局昵称(.*)", rule=to_me(), priority=5, block=True ) _matcher = on_alconna( Alconna( "nickname", Option("--name", action=store_true, help_text="用户昵称"), Option("--cancel", action=store_true, help_text="取消昵称"), ), rule=to_me(), priority=5, block=True, ) CALL_NAME = [ "好啦好啦，我知道啦，{}，以后就这么叫你吧", f"嗯嗯，{NICKNAME}" + "记住你的昵称了哦，{}", "好突然，突然要叫你昵称什么的...{}..", f"{NICKNAME}" + "会好好记住{}的，放心吧", "好..好.，那窝以后就叫你{}了.", ] REMIND = [ "我肯定记得你啊，你是{}啊", "我不会忘记你的，你也不要忘记我！{}", f"哼哼，{NICKNAME}" + "记忆力可是很好的，{}", "嗯？你是失忆了嘛...{}..", f"不要小看{NICKNAME}" + "的记忆力啊！笨蛋{}！QAQ", "哎？{}..怎么了吗..突然这样问..", ] CANCEL = [ f"呜..{NICKNAME}" + "睡一觉就会忘记的..和梦一样..{}", "窝知道了..{}..", f"是{NICKNAME}" + "哪里做的不好嘛..好吧..晚安{}", "呃，{}，下次我绝对绝对绝对不会再忘记你！", "可..可恶！{}！太可恶了！呜", ] def CheckNickname(): """ 检查名称是否合法 """ async def dependency( bot: Bot, matcher: Matcher, session: EventSession, message: UniMsg, reg_group: tuple[Any, ...] = RegexGroup(), ): black_word = Config.get_config("nickname", "BLACK_WORD") (name,) = reg_group logger.debug(f"昵称检查: {name}", "昵称设置", session=session) if not name: await Text("叫你空白？叫你虚空？叫你无名??").finish(at_sender=True) if session.id1 in bot.config.superusers: logger.debug( f"超级用户设置昵称, 跳过合法检测: {name}", "昵称设置", session=session ) return if len(name) > 20: await Text("昵称可不能超过20个字!").finish(at_sender=True) if name in bot.config.nickname: await Text("笨蛋！休想占用我的名字! #").finish(at_sender=True) if black_word: for x in name: if x in black_word: logger.debug("昵称设置禁止字符: [{x}]", "昵称设置", session=session) await Text(f"字符 [{x}] 为禁止字符!").finish(at_sender=True) for word in black_word: if word in name: logger.debug( "昵称设置禁止字符: [{word}]", "昵称设置", session=session ) await Text(f"字符 [{x}] 为禁止字符!").finish(at_sender=True) return Depends(dependency) @_nickname_matcher.handle(parameterless=[CheckNickname()]) async def _( session: EventSession, user_info: UserInfo = EventUserInfo(), reg_group: tuple[Any, ...] = RegexGroup(), ): if session.id1: (name,) = reg_group if len(name) < 5: if random.random() < 0.3: name = "~".join(name) if gid := session.id3 or session.id2: await GroupInfoUser.set_user_nickname( session.id1, gid, name, user_info.user_displayname or user_info.user_remark or user_info.user_name, session.platform, ) logger.info(f"设置群昵称成功: {name}", "昵称设置", session=session) await Text(random.choice(CALL_NAME).format(name)).finish(reply=True) else: await FriendUser.set_user_nickname( session.id1, name, user_info.user_displayname or user_info.user_remark or user_info.user_name, session.platform, ) logger.info(f"设置私聊昵称成功: {name}", "昵称设置", session=session) await Text(random.choice(CALL_NAME).format(name)).finish(reply=True) await Text("用户id为空...").send() @_global_nickname_matcher.handle(parameterless=[CheckNickname()]) async def _( session: EventSession, user_info: UserInfo = EventUserInfo(), reg_group: tuple[Any, ...] = RegexGroup(), ): if session.id1: (name,) = reg_group await FriendUser.set_user_nickname( session.id1, name, user_info.user_displayname or user_info.user_remark or user_info.user_name, session.platform, ) await GroupInfoUser.filter(user_id=session.id1).update(nickname=name) logger.info(f"设置全局昵称成功: {name}", "设置全局昵称", session=session) await Text(random.choice(CALL_NAME).format(name)).finish(reply=True) await Text("用户id为空...").send() @_matcher.assign("name") async def _(session: EventSession, user_info: UserInfo = EventUserInfo()): if session.id1: if gid := session.id3 or session.id2: nickname = await GroupInfoUser.get_user_nickname(session.id1, gid) card = user_info.user_displayname or user_info.user_name else: nickname = await FriendUser.get_user_nickname(session.id1) card = user_info.user_name if nickname: await Text(random.choice(REMIND).format(nickname)).finish(reply=True) else: await Text( random.choice( [ "没..没有昵称嘛，{}", "啊，你是{}啊，我想叫你的昵称！", "是{}啊，有什么事吗？", "你是{}？", ] ).format(card) ).finish(reply=True) await Text("用户id为空...").send() @_matcher.assign("cancel") async def _(bot: Bot, session: EventSession, user_info: UserInfo = EventUserInfo()): if session.id1: gid = session.id3 or session.id2 if gid: nickname = await GroupInfoUser.get_user_nickname(session.id1, gid) else: nickname = await FriendUser.get_user_nickname(session.id1) if nickname: await Text(random.choice(CANCEL).format(nickname)).send(reply=True) if gid: await GroupInfoUser.set_user_nickname(session.id1, gid, "") else: await FriendUser.set_user_nickname(session.id1, "") await BanConsole.ban(session.id1, gid, 9, 60, bot.self_id) return else: await Text("你在做梦吗？你没有昵称啊").finish(reply=True) await Text("用户id为空...").send()