nonebot-plugin-genshinuid 4.0.1__tar.gz

nonebot_plugin_genshinuid-4.0.1/GenshinUID/__init__.py

@@ -0,0 +1,442 @@
+import os
+import re
+import asyncio
+from pathlib import Path
+from base64 import b64encode
+from typing import Any, List, Union, Optional
+
+import aiofiles
+from nonebot.log import logger
+from nonebot.adapters import Bot
+from nonebot.matcher import Matcher
+from nonebot.permission import SUPERUSER
+from nonebot.internal.adapter import Event
+from nonebot import on_notice, get_driver, on_message, on_fullmatch
+
+from .client import GsClient
+from .auto_install import start, install
+from .models import Message, MessageReceive
+
+get_message = on_message(priority=999)
+get_notice = on_notice(priority=999)
+install_core = on_fullmatch('gs一键安装', permission=SUPERUSER, block=True)
+start_core = on_fullmatch('启动core', permission=SUPERUSER, block=True)
+connect_core = on_fullmatch(
+    ('连接core', '链接core'), permission=SUPERUSER, block=True
+)
+driver = get_driver()
+gsclient: Optional[GsClient] = None
+
+command_start = driver.config.command_start
+
+
+@get_notice.handle()
+async def get_notice_message(bot: Bot, ev: Event):
+    if gsclient is None or not gsclient.is_alive:
+        return await connect()
+
+    raw_data = ev.dict()
+    logger.debug(raw_data)
+
+    try:
+        user_id = str(ev.get_user_id())
+    except ValueError:
+        user_id = '未知'
+
+    group_id = None
+    sp_user_type = None
+    sp_bot_id = None
+    self_id = str(bot.self_id)
+    msg_id = ''
+    pm = 6
+
+    if await SUPERUSER(bot, ev):
+        pm = 1
+
+    if 'group_id' in raw_data:
+        group_id = str(raw_data['group_id'])
+
+    if 'user_id' in raw_data:
+        user_id = str(raw_data['user_id'])
+
+    if sp_bot_id:
+        bot_id = sp_bot_id
+    else:
+        bot_id = ev.__class__.__module__.split('.')[2]
+
+    user_type = 'group' if group_id else 'direct'
+
+    if 'notice_type' in raw_data and raw_data['notice_type'] in [
+        'group_upload',
+        'offline_file',
+    ]:
+        val = raw_data['file']['url']
+        name = raw_data['file']['name']
+        message = [Message('file', f'{name}|{val}')]
+        # onebot_v11
+    else:
+        return
+
+    msg = MessageReceive(
+        bot_id=bot_id,
+        bot_self_id=self_id,
+        user_type=sp_user_type if sp_user_type else user_type,
+        group_id=group_id,
+        user_id=user_id,
+        content=message,
+        msg_id=msg_id,
+        user_pm=pm,
+    )
+    logger.info(f'【发送】[gsuid-core]: {msg.bot_id}')
+    await gsclient._input(msg)
+
+
+@get_message.handle()
+async def get_all_message(bot: Bot, ev: Event):
+    if gsclient is None or not gsclient.is_alive:
+        return await connect()
+
+    # 通用字段获取
+    group_id = None
+    user_id = ev.get_user_id()
+    messages = ev.get_message()
+    logger.debug(ev)
+
+    self_id = str(bot.self_id)
+    message: List[Message] = []
+    sp_bot_id: Optional[str] = None
+
+    pm = 6
+    msg_id = ''
+
+    # qqguild
+    if bot.adapter.get_name() == 'QQ Guild':
+        from nonebot.adapters.qqguild.event import (
+            MessageEvent,
+            DirectMessageCreateEvent,
+        )
+
+        # 私聊
+        if isinstance(ev, DirectMessageCreateEvent):
+            user_type = 'direct'
+            group_id = str(ev.guild_id)
+            msg_id = ev.id
+        # 群聊
+        elif isinstance(ev, MessageEvent):
+            user_type = 'group'
+            group_id = str(ev.channel_id)
+            if ev.member and ev.member.roles:
+                if 4 in ev.member.roles:
+                    pm = 2
+                elif 2 in ev.member.roles:
+                    pm = 3
+                elif 5 in ev.member.roles:
+                    pm = 5
+            msg_id = ev.id
+        else:
+            logger.debug('[gsuid] 不支持该 QQ Guild 事件...')
+            return
+    # telegram
+    elif bot.adapter.get_name() == 'Telegram':
+        from nonebot.adapters.telegram.event import (
+            GroupMessageEvent,
+            PrivateMessageEvent,
+        )
+
+        if isinstance(ev, GroupMessageEvent) or isinstance(
+            ev, PrivateMessageEvent
+        ):
+            if ev.from_.is_bot:
+                return
+
+            user_id = str(ev.from_.id)
+            msg_id = str(ev.message_id)
+            if isinstance(ev, GroupMessageEvent):
+                user_type = 'group'
+                group_id = str(ev.chat.id)
+            else:
+                user_type = 'direct'
+        else:
+            logger.debug('[gsuid] 不支持该 Telegram 事件...')
+            return
+    # kaiheila
+    elif bot.adapter.get_name() == 'Kaiheila':
+        from nonebot.adapters.kaiheila.event import (
+            ChannelMessageEvent,
+            PrivateMessageEvent,
+        )
+
+        if isinstance(ev, ChannelMessageEvent) or isinstance(
+            ev, PrivateMessageEvent
+        ):
+            if ev.event.author.bot:
+                return
+
+            user_id = ev.author_id
+            msg_id = ev.msg_id
+            if isinstance(ev, ChannelMessageEvent):
+                user_type = 'group'
+                group_id = ev.target_id
+            else:
+                user_type = 'direct'
+        else:
+            logger.debug('[gsuid] 不支持该 kaiheila 事件...')
+            return
+    # onebot
+    elif bot.adapter.get_name() == 'OneBot V11':
+        from nonebot.adapters.onebot.v11.event import (
+            GroupMessageEvent,
+            PrivateMessageEvent,
+        )
+
+        if isinstance(ev, GroupMessageEvent) or isinstance(
+            ev, PrivateMessageEvent
+        ):
+            messages = ev.original_message
+            msg_id = str(ev.message_id)
+            if ev.sender.role == 'owner':
+                pm = 2
+            elif ev.sender.role == 'admin':
+                pm = 3
+
+            if isinstance(ev, GroupMessageEvent):
+                user_type = 'group'
+                group_id = str(ev.group_id)
+            else:
+                user_type = 'direct'
+        else:
+            logger.debug('[gsuid] 不支持该 onebotv11 事件...')
+            return
+    elif bot.adapter.get_name() == 'Feishu':
+        from nonebot.adapters.feishu.event import (
+            GroupEventMessage,
+            PrivateEventMessage,
+        )
+
+        if isinstance(ev, GroupEventMessage) or isinstance(
+            ev, PrivateEventMessage
+        ):
+            for feishu_msg in messages:
+                if 'image_key' in feishu_msg.data:
+                    feishu_msg.data['url'] = feishu_msg.data['image_key']
+            user_id = ev.get_user_id()
+            msg_id = ev.message_id
+            if isinstance(ev, GroupEventMessage):
+                user_type = 'group'
+                group_id = ev.chat_id
+            else:
+                user_type = 'direct'
+        else:
+            logger.debug('[gsuid] 不支持该 Feishu 事件...')
+            return
+    # ntchat
+    elif bot.adapter.get_name() == 'ntchat':
+        from nonebot.adapters.ntchat.event import (
+            FileMessageEvent,
+            TextMessageEvent,
+        )
+
+        if isinstance(ev, TextMessageEvent):
+            user_id = ev.from_wxid
+            msg_id = ev.msgid
+            if 'chatroom' in ev.to_wxid:
+                user_type = 'group'
+                group_id = ev.to_wxid
+            else:
+                user_type = 'direct'
+            if 'image' in ev.data:
+                message.append(Message('image', ev.data['image']))
+            if 'from_wxid' in ev.data:
+                if 'raw_msg' in ev.data and 'xml' in ev.data['raw_msg']:
+                    match = re.search(
+                        r'<svrid>(\d+)</svrid>', ev.data['raw_msg']
+                    )
+                    if match:
+                        message.append(Message('reply', match.group(1)))
+            if ev.at_user_list:
+                at_list = [Message('at', i) for i in ev.at_user_list]
+                at_list.pop(0)
+                message.extend(at_list)
+        elif isinstance(ev, FileMessageEvent):
+            if 'chatroom' in ev.to_wxid:
+                group_id = ev.to_wxid
+                user_type = 'group'
+            else:
+                user_type = 'direct'
+            val = ev.file
+            name = ev.file_name
+            await asyncio.sleep(2)
+            if (
+                os.path.exists(val)
+                and os.path.getsize(val) <= 5 * 1024 * 1024
+                and str(name).endswith('.json')
+            ):
+                message.append(await convert_file(val, name))
+        else:
+            logger.debug('[gsuid] 不支持该 ntchat 事件...')
+            return
+    # OneBot V12 (仅在 ComWechatClient 测试)
+    elif bot.adapter.get_name() == 'OneBot V12':
+        from nonebot.adapters.onebot.v12.event import (
+            GroupMessageEvent,
+            PrivateMessageEvent,
+        )
+
+        # v12msgid = raw_data['id']  # V12的消息id
+        # self = raw_data['self']  # 返回 platform='xxx' user_id='wxid_xxxxx'
+        # platform = self.platform  # 机器人平台
+        # V12还支持频道等其他平台，速速Pr！
+
+        if isinstance(ev, GroupMessageEvent) or isinstance(
+            ev, PrivateMessageEvent
+        ):
+            messages = ev.original_message
+            msg_id = ev.message_id
+            sp_bot_id = 'onebot_v12'
+
+            if '[文件]' in ev.alt_message:
+                file_id = messages[0].data.get('file_id')
+                logger.info('[OB12文件ID]', file_id)
+                if file_id and file_id in messages[0].data.values():
+                    data = await get_file(bot, file_id)
+                    logger.info('[OB12文件]', data)
+                    name = data['name']
+                    path = data['path']
+                    message.append(await convert_file(path, name))
+
+            if isinstance(ev, GroupMessageEvent):
+                user_type = 'group'
+                group_id = ev.group_id
+            else:
+                user_type = 'direct'
+        else:
+            logger.debug('[gsuid] 不支持该 onebotv12 事件...')
+            return
+    else:
+        logger.debug(f'[gsuid] 不支持该 {bot.adapter.get_name()} 事件...')
+        return
+
+    if sp_bot_id:
+        bot_id = sp_bot_id
+    else:
+        bot_id = messages.__class__.__module__.split('.')[2]
+
+    # 确认超管权限
+    if await SUPERUSER(bot, ev):
+        pm = 1
+
+    # 如果有at提及，增加AT
+    if ev.is_tome():
+        message.append(Message('at', self_id))
+
+    # 处理消息
+    for index, _msg in enumerate(messages):
+        message = convert_message(_msg, message, index)
+
+    if not message:
+        return
+
+    msg = MessageReceive(
+        bot_id=bot_id,
+        bot_self_id=self_id,
+        user_type=user_type,
+        group_id=group_id,
+        user_id=user_id,
+        content=message,
+        msg_id=msg_id if msg_id else '',
+        user_pm=pm,
+    )
+    logger.info(f'【发送】[gsuid-core]: {msg.bot_id}')
+    await gsclient._input(msg)
+
+
+@install_core.handle()
+async def send_install_msg(matcher: Matcher):
+    await matcher.send('即将开始安装...会持续一段时间, 且期间无法使用Bot!')
+    await matcher.send(await install())
+
+
+@connect_core.handle()
+async def send_connect_msg(matcher: Matcher):
+    await connect()
+    await matcher.send('链接成功！')
+
+
+@start_core.handle()
+async def send_start_msg(matcher: Matcher):
+    await matcher.send(await start())
+
+
+@driver.on_bot_connect
+async def start_client():
+    if gsclient is None:
+        await connect()
+
+
+async def connect():
+    global gsclient
+    try:
+        gsclient = await GsClient().async_connect()
+        await gsclient.start()
+    except ConnectionRefusedError:
+        logger.error('Core服务器连接失败...请稍后使用[启动core]命令启动...')
+
+
+def convert_message(_msg: Any, message: List[Message], index: int):
+    if _msg.type == 'text':
+        data: str = (
+            _msg.data['text'] if 'text' in _msg.data else _msg.data['content']
+        )
+        if index == 0:
+            if data.startswith(tuple(command_start)):
+                for word in command_start:
+                    data = data.replace(word, '', 1)
+        message.append(Message('text', data))
+    elif _msg.type == 'image':
+        file_id = _msg.data.get('file_id')
+        if file_id in _msg.data.values():
+            message.append(Message('image', _msg.data['file_id']))
+            logger.debug('[OB12图片]', _msg.data['file_id'])
+        else:
+            message.append(Message('image', _msg.data['url']))
+    elif _msg.type == 'at':
+        message.append(Message('at', _msg.data['qq']))
+    elif _msg.type == 'reply':
+        message_id = _msg.data.get('message_id')
+        if message_id in _msg.data.values():
+            message.append(Message('reply', _msg.data['message_id']))
+        else:
+            message.append(Message('reply', _msg.data['id']))
+    elif _msg.type == 'mention':
+        if 'user_id' in _msg.data:
+            message.append(Message('at', _msg.data['user_id']))
+    return message
+
+
+# 读取文件为base64
+async def convert_file(
+    content: Union[Path, str, bytes], file_name: str
+) -> Message:
+    if isinstance(content, Path):
+        print(content)
+        async with aiofiles.open(str(content), 'rb') as fp:
+            file = await fp.read()
+    elif isinstance(content, bytes):
+        file = content
+    else:
+        async with aiofiles.open(content, 'rb') as fp:
+            file = await fp.read()
+    return Message(
+        type='file',
+        data=f'{file_name}|{b64encode(file).decode()}',
+    )
+
+
+# 获取文件
+async def get_file(bot: Bot, file_id: str):
+    data = await bot.call_api(
+        api='get_file',
+        file_id=f'{file_id}',
+        type='path',
+    )
+    return data

nonebot_plugin_genshinuid-4.0.1/GenshinUID/auto_install.py

@@ -0,0 +1,42 @@
+import asyncio
+import subprocess
+
+from git.repo import Repo
+from nonebot.log import logger
+from pip._internal import main as pip_install
+
+from .path import CORE_PATH, GSUID_PATH
+
+GS_GIT = 'https://ghproxy.com/https://github.com/KimigaiiWuyi/GenshinUID.git'
+CORE_GIT = 'https://ghproxy.com/https://github.com/Genshin-bots/gsuid_core.git'
+
+
+async def _install():
+    if not CORE_PATH.exists():
+        Repo.clone_from(CORE_GIT, CORE_PATH, single_branch=True, depth=1)
+    if not GSUID_PATH.exists():
+        Repo.clone_from(
+            GS_GIT, GSUID_PATH, b='v4', single_branch=True, depth=1
+        )
+    try:
+        import poetry.__version__ as poetry_version
+
+        if float(poetry_version.__version__[:3]) < 1.4:
+            pip_install(['install', 'poetry'])
+    except ImportError:
+        pip_install(['install', 'poetry'])
+    subprocess.call(['poetry', 'install'], cwd=f'{CORE_PATH}')
+
+
+async def install():
+    done = await asyncio.gather(_install())
+    logger.info(done)
+    return '安装成功...'
+
+
+async def start():
+    subprocess.Popen(
+        ['poetry', 'run', 'python', 'gsuid_core/core.py'],
+        cwd=f'{CORE_PATH}',
+    )
+    return '启动成功完成...'