✨ support message reaction

Author: RF-Tar-Railt

src/nonebot_plugin_alconna/uniseg/adapters/discord/exporter.py

@@ -195,6 +195,40 @@ async def edit(self, new: Sequence[Segment], mid: Any, bot: Bot, context: Union[
             return await bot.edit_message(channel_id=context.channel_id, message_id=int(mid), **parse_message(new_msg))
         return await bot.edit_message(channel_id=mid.channel_id, message_id=_mid.id, **parse_message(new_msg))
 
+    async def reaction(self, emoji: Emoji, mid: Any, bot: Bot, context: Union[Target, Event], delete: bool = False):
+        assert isinstance(bot, DiscordBot)
+        if isinstance(mid, (str, SnowflakeType)):
+            assert isinstance(context, MessageEvent)
+            if delete:
+                await bot.delete_own_reaction(
+                    channel_id=context.channel_id,
+                    message_id=int(mid),
+                    emoji=emoji.name or emoji.id,
+                    emoji_id=int(emoji.id) if emoji.name else None,
+                )
+            else:
+                await bot.create_reaction(
+                    channel_id=context.channel_id,
+                    message_id=int(mid),
+                    emoji=emoji.name or emoji.id,
+                    emoji_id=int(emoji.id) if emoji.name else None,
+                )
+        _mid: MessageGet = cast(MessageGet, mid)
+        if delete:
+            await bot.delete_own_reaction(
+                channel_id=_mid.channel_id,
+                message_id=_mid.id,
+                emoji=emoji.name or emoji.id,
+                emoji_id=int(emoji.id) if emoji.name else None,
+            )
+        else:
+            await bot.create_reaction(
+                channel_id=_mid.channel_id,
+                message_id=_mid.id,
+                emoji=emoji.name or emoji.id,
+                emoji_id=int(emoji.id) if emoji.name else None,
+            )
+
     def get_reply(self, mid: Any):
         _mid: MessageGet = cast(MessageGet, mid)
         return Reply(str(_mid.id), origin=_mid.id)

src/nonebot_plugin_alconna/uniseg/adapters/dodo/exporter.py

@@ -7,10 +7,11 @@
 from nonebot.adapters.dodo.bot import Bot as DoDoBot
 from nonebot.adapters.dodo.event import MessageEvent
 from nonebot.adapters.dodo.event import Event as DoDoEvent
+from nonebot.adapters.dodo.models import Emoji as DodoEmoji
 from nonebot.adapters.dodo.message import Message, MessageSegment
 
 from nonebot_plugin_alconna.uniseg.constraint import SupportScope
-from nonebot_plugin_alconna.uniseg.segment import At, Text, Image, Reply, Video, Segment
+from nonebot_plugin_alconna.uniseg.segment import At, Text, Emoji, Image, Reply, Video, Segment
 from nonebot_plugin_alconna.uniseg.exporter import Target, SupportAdapter, MessageExporter, SerializeFailed, export
 
 
@@ -117,5 +118,13 @@ async def edit(self, new: Sequence[Segment], mid: Any, bot: Bot, context: Union[
             return await bot.set_channel_message_edit(message_id=mid, message_body=new_msg.to_message_body()[0])
         return None
 
+    async def reaction(self, emoji: Emoji, mid: Any, bot: Bot, context: Union[Target, Event], delete: bool = False):
+        assert isinstance(bot, DoDoBot)
+        emj = DodoEmoji(type=1, id=emoji.id)
+        if delete:
+            await bot.set_channel_message_reaction_remove(message_id=mid, emoji=emj)
+        else:
+            await bot.set_channel_message_reaction_add(message_id=mid, emoji=emj)
+
     def get_reply(self, mid: Any):
         return Reply(mid)

src/nonebot_plugin_alconna/uniseg/adapters/feishu/exporter.py

@@ -10,7 +10,19 @@
 
 from nonebot_plugin_alconna.uniseg.constraint import SupportScope
 from nonebot_plugin_alconna.uniseg.exporter import Target, SupportAdapter, MessageExporter, SerializeFailed, export
-from nonebot_plugin_alconna.uniseg.segment import At, File, Text, AtAll, Audio, Image, Reply, Video, Voice, Segment
+from nonebot_plugin_alconna.uniseg.segment import (
+    At,
+    File,
+    Text,
+    AtAll,
+    Audio,
+    Emoji,
+    Image,
+    Reply,
+    Video,
+    Voice,
+    Segment,
+)
 
 
 class FeishuMessageExporter(MessageExporter[Message]):
@@ -172,10 +184,8 @@ async def send_to(self, target: Union[Target, Event], bot: Bot, message: Message
 
     async def recall(self, mid: Any, bot: Bot, context: Union[Target, Event]):
         assert isinstance(bot, FeishuBot)
-
-        params = {"method": "DELETE"}
         message_id = mid if isinstance(mid, str) else mid["message_id"]
-        return await bot.call_api(f"im/v1/messages/{message_id}", **params)
+        return await bot.delete_msg(message_id)
 
     async def edit(self, new: Sequence[Segment], mid: Any, bot: Bot, context: Union[Target, Event]):
         assert isinstance(bot, FeishuBot)
@@ -184,5 +194,24 @@ async def edit(self, new: Sequence[Segment], mid: Any, bot: Bot, context: Union[
         message_id = mid if isinstance(mid, str) else mid["message_id"]
         return await bot.edit_msg(message_id, content=content, msg_type=msg_type)
 
+    async def reaction(self, emoji: Emoji, mid: Any, bot: Bot, context: Union[Target, Event], delete: bool = False):
+        assert isinstance(bot, FeishuBot)
+        message_id = mid if isinstance(mid, str) else mid["message_id"]
+        if delete:
+            queries = {"reaction_type": emoji.name or emoji.id}
+            resp = await bot.call_api(f"im/v1/messages/{message_id}/reactions", method="GET", params=queries)
+            if items := resp.get("items"):
+                reaction_id = items[0]["reaction_id"]
+                return await bot.call_api(f"im/v1/messages/{message_id}/reactions/{reaction_id}", method="DELETE")
+        else:
+            params = {
+                "method": "POST",
+                "data": {
+                    "emoji": emoji.name,
+                    "reaction_type": {"emoji_type": emoji.name or emoji.id},
+                },
+            }
+            return await bot.call_api(f"im/v1/messages/{message_id}/reactions", **params)
+
     def get_reply(self, mid: Any):
         return Reply(mid["message_id"])

src/nonebot_plugin_alconna/uniseg/adapters/kook/exporter.py

@@ -184,6 +184,28 @@ async def edit(self, new: Sequence[Segment], mid: Any, bot: Bot, context: Union[
             await bot.message_update(**data)
         return
 
+    async def reaction(
+        self,
+        emoji: Emoji,
+        mid: Any,
+        bot: Bot,
+        context: Union[Target, Event],
+        delete: bool = False,
+    ):
+        assert isinstance(bot, KBot)
+        if isinstance(mid, str):
+            if delete:
+                await bot.message_deleteReaction(msg_id=mid, emoji=emoji.name or emoji.id)
+            else:
+                await bot.message_addReaction(msg_id=mid, emoji=emoji.name or emoji.id)
+        _mid: MessageCreateReturn = cast(MessageCreateReturn, mid)
+        if not _mid.msg_id:
+            return
+        if delete:
+            await bot.message_deleteReaction(msg_id=_mid.msg_id, emoji=emoji.name or emoji.id)
+        else:
+            await bot.message_addReaction(msg_id=_mid.msg_id, emoji=emoji.name or emoji.id)
+
     def get_reply(self, mid: Any):
         _mid: MessageCreateReturn = cast(MessageCreateReturn, mid)
         if not _mid.msg_id:

src/nonebot_plugin_alconna/uniseg/adapters/kritor/exporter.py

@@ -315,5 +315,27 @@ async def recall(self, mid: Any, bot: Bot, context: Union[Target, Event]):
             assert isinstance(mid, (SendMessageByResIdResponse, SendMessageResponse))
             await bot.recall_message(message_id=mid.message_id)
 
+    async def reaction(self, emoji: Emoji, mid: Any, bot: Bot, context: Union[Target, Event], delete: bool = False):
+        assert isinstance(bot, KritorBot)
+        if isinstance(context, Event):
+            assert isinstance(context, MessageEvent)
+            contact = context.contact
+        elif context.private:
+            if context.parent_id:
+                contact = Contact(type=SceneType.STRANGER_FROM_GROUP, id=context.id, sub_id=context.parent_id)
+            else:
+                contact = Contact(type=SceneType.FRIEND, id=context.id)
+        elif context.channel:
+            contact = Contact(type=SceneType.GUILD, id=context.parent_id, sub_id=context.id)
+        else:
+            contact = Contact(type=SceneType.GROUP, id=context.id)
+        if isinstance(mid, str):
+            await bot.set_message_comment_emoji(contact=contact, message_id=mid, emoji=int(emoji.id), is_set=not delete)
+        else:
+            assert isinstance(mid, (SendMessageByResIdResponse, SendMessageResponse))
+            await bot.set_message_comment_emoji(
+                contact=contact, message_id=mid.message_id, emoji=int(emoji.id), is_set=not delete
+            )
+
     def get_reply(self, mid: Any):
         return Reply(str(mid["message_id"]))

src/nonebot_plugin_alconna/uniseg/adapters/qq/exporter.py

@@ -459,6 +459,39 @@ async def recall(self, mid: Any, bot: Bot, context: Union[Target, Event]):
                     message_id=mid,
                 )
 
+    async def reaction(self, emoji: Emoji, mid: Any, bot: Bot, context: Union[Target, Event], delete: bool = False):
+        assert isinstance(bot, QQBot)
+        if isinstance(mid, GuildMessage):
+            if delete:
+                await bot.delete_own_message_reaction(
+                    channel_id=mid.channel_id,
+                    message_id=mid.id,
+                    type=1,
+                    id=emoji.id,
+                )
+            else:
+                await bot.put_message_reaction(
+                    channel_id=mid.channel_id,
+                    message_id=mid.id,
+                    type=1,
+                    id=emoji.id,
+                )
+        elif isinstance(mid, str) and isinstance(context, GuildMessageEvent):
+            if delete:
+                await bot.delete_own_message_reaction(
+                    channel_id=context.channel_id,
+                    message_id=mid,
+                    type=1,
+                    id=emoji.id,
+                )
+            else:
+                await bot.put_message_reaction(
+                    channel_id=context.channel_id,
+                    message_id=mid,
+                    type=1,
+                    id=emoji.id,
+                )
+
     def get_reply(self, mid: Any):
         if isinstance(mid, GuildMessage):
             return Reply(mid.id)

src/nonebot_plugin_alconna/uniseg/adapters/satori/exporter.py

@@ -18,6 +18,7 @@
     Text,
     AtAll,
     Audio,
+    Emoji,
     Image,
     Reply,
     Video,
@@ -211,15 +212,42 @@ async def edit(self, new: Sequence[Segment], mid: Any, bot: Bot, context: Union[
             return await bot.update_message(context.channel.id, mid, new_msg)
         _mid: MessageObject = cast(MessageObject, mid)
         if isinstance(context, Target):
+            channel_id = mid.channel.id if mid.channel else context.id
             if context.private:
-                channel = await bot.user_channel_create(user_id=context.id)
-                return await bot.update_message(channel.id, _mid.id, new_msg)
-            return await bot.update_message(context.id, _mid.id, new_msg)
+                channel_id = (await bot.user_channel_create(user_id=context.id)).id
+            return await bot.update_message(channel_id, _mid.id, new_msg)
         if TYPE_CHECKING:
             assert isinstance(context, MessageEvent)
         channel = mid.channel or context.channel
         return await bot.update_message(channel.id, _mid.id, new_msg)
 
+    async def reaction(self, emoji: Emoji, mid: Any, bot: Bot, context: Union[Target, Event], delete: bool = False):
+        assert isinstance(bot, SatoriBot)
+        if isinstance(context, MessageEvent) and isinstance(mid, str):
+            if delete:
+                return await bot.reaction_delete(
+                    channel_id=context.channel.id, message_id=mid, emoji=emoji.name or emoji.id
+                )
+            return await bot.reaction_create(
+                channel_id=context.channel.id, message_id=mid, emoji=emoji.name or emoji.id
+            )
+        _mid: MessageObject = cast(MessageObject, mid)
+        if isinstance(context, Target):
+            channel_id = mid.channel.id if mid.channel else context.id
+            if context.private:
+                channel_id = (await bot.user_channel_create(user_id=context.id)).id
+            if delete:
+                return await bot.reaction_delete(
+                    channel_id=channel_id, message_id=_mid.id, emoji=emoji.name or emoji.id
+                )
+            return await bot.reaction_create(channel_id=channel_id, message_id=_mid.id, emoji=emoji.name or emoji.id)
+        if TYPE_CHECKING:
+            assert isinstance(context, MessageEvent)
+        channel = mid.channel or context.channel
+        if delete:
+            return await bot.reaction_delete(channel_id=channel.id, message_id=_mid.id, emoji=emoji.name or emoji.id)
+        return await bot.reaction_create(channel_id=channel.id, message_id=_mid.id, emoji=emoji.name or emoji.id)
+
     def get_reply(self, mid: Any):
         _mid: MessageObject = cast(MessageObject, mid)
         return Reply(_mid.id)

src/nonebot_plugin_alconna/uniseg/adapters/telegram/exporter.py

@@ -13,7 +13,7 @@
 from nonebot.adapters.telegram.message import Message as TgMessage
 from nonebot.adapters.telegram.model import Message as MessageModel
 from nonebot.adapters.telegram.event import MessageEvent, EventWithChat
-from nonebot.adapters.telegram.model import InlineKeyboardButton, InlineKeyboardMarkup
+from nonebot.adapters.telegram.model import ReactionTypeEmoji, InlineKeyboardButton, InlineKeyboardMarkup
 
 from nonebot_plugin_alconna.uniseg.constraint import SupportScope
 from nonebot_plugin_alconna.uniseg.exporter import Target, SupportAdapter, MessageExporter, SerializeFailed, export
@@ -198,6 +198,22 @@ async def edit(self, new: Sequence[Segment], mid: Any, bot: Bot, context: Union[
             return res
         return None
 
+    async def reaction(self, emoji: Emoji, mid: Any, bot: Bot, context: Union[Target, Event], delete: bool = False):
+        assert isinstance(bot, TgBot)
+        if delete:
+            return  # FIXME:  how to delete reaction in telegram?
+        if isinstance(mid, (str, int)) and isinstance(context, MessageEvent):
+            await bot.set_message_reaction(
+                chat_id=context.chat.id, message_id=int(mid), reaction=[ReactionTypeEmoji(emoji=emoji.name or emoji.id)]
+            )
+        else:
+            _mid: MessageModel = cast(MessageModel, mid)
+            await bot.set_message_reaction(
+                chat_id=_mid.chat.id,
+                message_id=_mid.message_id,
+                reaction=[ReactionTypeEmoji(emoji=emoji.name or emoji.id)],
+            )
+
     def get_reply(self, mid: Any):
         _mid: MessageModel = cast(MessageModel, mid)
         return Reply(str(_mid.message_id))

src/nonebot_plugin_alconna/uniseg/exporter.py

@@ -243,5 +243,8 @@ async def recall(self, mid: Any, bot: Bot, context: Union[Target, Event]):
     async def edit(self, new: Sequence[Segment], mid: Any, bot: Bot, context: Union[Target, Event]):
         raise NotImplementedError
 
+    async def reaction(self, emoji: Emoji, mid: Any, bot: Bot, context: Union[Target, Event], delete: bool = False):
+        raise NotImplementedError
+
     def get_reply(self, mid: Any) -> Reply:
         raise NotImplementedError

src/nonebot_plugin_alconna/uniseg/functions.py

@@ -6,6 +6,7 @@
 from nonebot.adapters import Bot, Event
 from nonebot.internal.matcher import current_bot, current_event
 
+from .segment import Emoji
 from .exporter import SerializeFailed
 from .adapters import alter_get_exporter
 
@@ -31,7 +32,10 @@ async def message_recall(
         _adapter = bot.adapter
         adapter = _adapter.get_name()
     if fn := alter_get_exporter(adapter):
-        return await fn.recall(message_id or fn.get_message_id(event), bot, event)
+        try:
+            return await fn.recall(message_id or fn.get_message_id(event), bot, event)
+        except NotImplementedError:
+            return
     raise SerializeFailed(lang.require("nbp-uniseg", "unsupported").format(adapter=adapter))
 
 
@@ -56,7 +60,40 @@ async def message_edit(
         _adapter = bot.adapter
         adapter = _adapter.get_name()
     if fn := alter_get_exporter(adapter):
-        return await fn.edit(msg, message_id or fn.get_message_id(event), bot, event)
+        try:
+            return await fn.edit(msg, message_id or fn.get_message_id(event), bot, event)
+        except NotImplementedError:
+            return
+    raise SerializeFailed(lang.require("nbp-uniseg", "unsupported").format(adapter=adapter))
+
+
+async def message_reaction(
+    emoji: str | Emoji,
+    message_id: str | None = None,
+    event: Event | None = None,
+    bot: Bot | None = None,
+    adapter: str | None = None,
+    delete: bool = False,
+):
+    if not event:
+        try:
+            event = current_event.get()
+        except LookupError as e:
+            raise SerializeFailed(lang.require("nbp-uniseg", "event_missing")) from e
+    if not bot:
+        try:
+            bot = current_bot.get()
+        except LookupError as e:
+            raise SerializeFailed(lang.require("nbp-uniseg", "bot_missing")) from e
+    if not adapter:
+        _adapter = bot.adapter
+        adapter = _adapter.get_name()
+    emj = Emoji(emoji) if isinstance(emoji, str) else emoji
+    if fn := alter_get_exporter(adapter):
+        try:
+            return await fn.reaction(emj, message_id or fn.get_message_id(event), bot, event, delete=delete)
+        except NotImplementedError:
+            return
     raise SerializeFailed(lang.require("nbp-uniseg", "unsupported").format(adapter=adapter))