feat: support adapter qq (#20)

* feat: support adapter qq

* style: auto fix by pre-commit hooks

* remove openid uuid check

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

README.md

@@ -86,6 +86,7 @@ async def handle(user_info: UserInfo = BotUserInfo()):  # 获取Bot用户信息
 - [x] Discord
 - [x] DoDo
 - [x] Satori
+- [x] QQ
 
 ### 鸣谢
 

nonebot_plugin_userinfo/__init__.py

@@ -25,5 +25,6 @@
         "~discord",
         "~dodo",
         "~satori",
+        "~qq",
     },
 )

nonebot_plugin_userinfo/adapters/__init__.py

@@ -5,6 +5,7 @@
 from . import kaiheila as kaiheila
 from . import onebot_v11 as onebot_v11
 from . import onebot_v12 as onebot_v12
+from . import qq as qq
 from . import red as red
 from . import satori as satori
 from . import telegram as telegram

nonebot_plugin_userinfo/adapters/qq.py

@@ -0,0 +1,63 @@
+from typing import Optional
+
+from nonebot.exception import ActionFailed
+from nonebot.log import logger
+
+from ..getter import UserInfoGetter, register_user_info_getter
+from ..image_source import ImageUrl, QQAvatarOpenId
+from ..user_info import UserInfo
+
+try:
+    from nonebot.adapters.qq import (
+        Bot,
+        C2CMessageCreateEvent,
+        Event,
+        GroupAtMessageCreateEvent,
+        GuildMessageEvent,
+    )
+
+    @register_user_info_getter(Bot, Event)
+    class Getter(UserInfoGetter[Bot, Event]):
+        async def _get_info(self, user_id: str) -> Optional[UserInfo]:
+            user = None
+
+            if self.bot.self_id == user_id:
+                try:
+                    user = await self.bot.me()
+                except ActionFailed as e:
+                    logger.warning(f"Error calling me: {e}")
+
+            if not user and isinstance(self.event, GuildMessageEvent):
+                if self.event.author.id == user_id:
+                    user = self.event.author
+                else:
+                    guild_id = self.event.guild_id
+                    try:
+                        member = await self.bot.get_member(
+                            guild_id=guild_id, user_id=user_id
+                        )
+                        if member:
+                            user = member.user
+                    except ActionFailed as e:
+                        logger.warning(f"Error calling get_member: {e}")
+
+            if user:
+                return UserInfo(
+                    user_id=user.id,
+                    user_name=user.username or "",
+                    user_avatar=ImageUrl(url=user.avatar) if user.avatar else None,
+                )
+
+            if isinstance(
+                self.event, (C2CMessageCreateEvent, GroupAtMessageCreateEvent)
+            ):
+                return UserInfo(
+                    user_id=user_id,
+                    user_name="",
+                    user_avatar=QQAvatarOpenId(
+                        appid=self.bot.bot_info.id, user_openid=user_id
+                    ),
+                )
+
+except ImportError:
+    pass

nonebot_plugin_userinfo/image_source.py

@@ -64,6 +64,15 @@ async def get_image(self) -> bytes:
         return data
 
 
+class QQAvatarOpenId(ImageSource):
+    appid: str
+    user_openid: str
+
+    async def get_image(self) -> bytes:
+        url = f"https://q.qlogo.cn/qqapp/{self.appid}/{self.user_openid}/100"
+        return await download_url(url)
+
+
 class TelegramFile(ImageSource):
     token: str
     file_path: str

pyproject.toml

@@ -33,9 +33,10 @@ nonebot-adapter-kaiheila = "^0.3.1"
 nonebot-adapter-telegram = { git = "https://github.com/nonebot/adapter-telegram.git" }
 nonebot-adapter-feishu = "^2.4.0"
 nonebot-adapter-red = "^0.9.0"
-nonebot-adapter-discord = { git = "https://github.com/nonebot/adapter-discord.git" }
+nonebot-adapter-discord = "^0.1.4"
 nonebot-adapter-dodo = "^0.2.0"
 nonebot-adapter-satori = "^0.9.3"
+nonebot-adapter-qq = "^1.4.1"
 
 [tool.pytest.ini_options]
 asyncio_mode = "auto"

tests/plugins/echo.py

@@ -22,5 +22,7 @@ async def _(bot: Bot, event: Event, arg: Message = CommandArg()):
 
 
 @bot_user_info_cmd.handle()
-async def _(user_info: Optional[UserInfo] = BotUserInfo(use_cache=False)):  # 获取Bot用户信息
+async def _(
+    user_info: Optional[UserInfo] = BotUserInfo(use_cache=False),
+):  # 获取Bot用户信息
     await bot_user_info_cmd.send("", user_info=user_info)

tests/test_qq.py

@@ -0,0 +1,107 @@
+from datetime import datetime
+
+from nonebot.adapters.qq import (
+    Bot,
+    EventType,
+    GroupAtMessageCreateEvent,
+    MessageCreateEvent,
+)
+from nonebot.adapters.qq.config import BotInfo
+from nonebot.adapters.qq.models import GroupMemberAuthor, Member, User
+from nonebug import App
+
+
+def _fake_message_create_event(msg: str) -> MessageCreateEvent:
+    return MessageCreateEvent(
+        id="id",
+        __type__=EventType.MESSAGE_CREATE,
+        channel_id="6677",
+        guild_id="5566",
+        author=User(id="3344", username="MyUser", avatar="http://xxx.jpg"),
+        content=msg,
+    )
+
+
+def _fake_group_at_message_create_event(msg: str) -> GroupAtMessageCreateEvent:
+    return GroupAtMessageCreateEvent(
+        id="id",
+        content=msg,
+        timestamp="1234567890",
+        __type__=EventType.GROUP_AT_MESSAGE_CREATE,
+        author=GroupMemberAuthor(
+            id="id",
+            member_openid="3F21411B784D403E811E68BF3E2944D8",
+        ),
+        group_openid="19303D3617EF432999F24342F99AEC65",
+    )
+
+
+async def test_message_event(app: App):
+    from nonebot_plugin_userinfo import UserInfo
+    from nonebot_plugin_userinfo.image_source import ImageUrl, QQAvatarOpenId
+
+    async with app.test_matcher() as ctx:
+        bot = ctx.create_bot(
+            base=Bot, self_id="2233", bot_info=BotInfo(id="2233", token="", secret="")
+        )
+
+        user_info = UserInfo(
+            user_id="3344",
+            user_name="MyUser",
+            user_displayname=None,
+            user_remark=None,
+            user_avatar=ImageUrl(url="http://xxx.jpg"),
+            user_gender="unknown",
+        )
+        event = _fake_message_create_event("/user_info")
+        ctx.receive_event(bot, event)
+        ctx.should_call_send(event, "", True, user_info=user_info)
+
+        user_info = UserInfo(
+            user_id="1234",
+            user_name="member",
+            user_displayname=None,
+            user_remark=None,
+            user_avatar=ImageUrl(url="http://xxx.jpg"),
+            user_gender="unknown",
+        )
+        event = _fake_message_create_event("/user_info 1234")
+        ctx.receive_event(bot, event)
+        ctx.should_call_api(
+            "get_member",
+            {"guild_id": "5566", "user_id": "1234"},
+            Member(
+                user=User(id="1234", username="member", avatar="http://xxx.jpg"),
+                joined_at=datetime.now(),
+            ),
+        )
+        ctx.should_call_send(event, "", True, user_info=user_info)
+
+        user_info = UserInfo(
+            user_id="2233",
+            user_name="Bot",
+            user_displayname=None,
+            user_remark=None,
+            user_avatar=ImageUrl(url="http://xxx.jpg"),
+            user_gender="unknown",
+        )
+        event = _fake_message_create_event("/bot_user_info")
+        ctx.receive_event(bot, event)
+        ctx.should_call_api(
+            "me", {}, User(id="2233", username="Bot", avatar="http://xxx.jpg")
+        )
+        ctx.should_call_send(event, "", True, user_info=user_info)
+
+        user_info = UserInfo(
+            user_id="3F21411B784D403E811E68BF3E2944D8",
+            user_name="",
+            user_displayname=None,
+            user_remark=None,
+            user_avatar=QQAvatarOpenId(
+                appid="2233", user_openid="3F21411B784D403E811E68BF3E2944D8"
+            ),
+            user_gender="unknown",
+        )
+        event = _fake_group_at_message_create_event("/user_info")
+        ctx.receive_event(bot, event)
+        ctx.should_call_send(event, "", True, user_info=user_info)