Merge pull request #1654 from langbot-app/feat/wecom_ai_bot

Feat/wecom ai bot

pkg/platform/sources/wecombot.py

@@ -0,0 +1,167 @@
+from __future__ import annotations
+import typing
+import asyncio
+import traceback
+
+import datetime
+import langbot_plugin.api.definition.abstract.platform.adapter as abstract_platform_adapter
+import langbot_plugin.api.entities.builtin.platform.message as platform_message
+import langbot_plugin.api.entities.builtin.platform.events as platform_events
+import langbot_plugin.api.entities.builtin.platform.entities as platform_entities
+import pydantic
+from ..logger import  EventLogger
+from libs.wecom_ai_bot_api.wecombotevent import WecomBotEvent
+from libs.wecom_ai_bot_api.api import WecomBotClient
+from ...core import app
+
+class WecomBotMessageConverter(abstract_platform_adapter.AbstractMessageConverter):
+    @staticmethod
+    async def yiri2target(message_chain: platform_message.MessageChain):
+        content = ''
+        for msg in message_chain:
+            if type(msg) is platform_message.Plain:
+                content += msg.text
+        return content
+
+    @staticmethod
+    async def target2yiri(event: WecomBotEvent):
+        yiri_msg_list = []
+        if event.type == 'group':
+            yiri_msg_list.append(platform_message.At(target=event.ai_bot_id))
+        yiri_msg_list.append(platform_message.Source(id=event.message_id, time=datetime.datetime.now()))
+        yiri_msg_list.append(platform_message.Plain(text=event.content))
+        if event.picurl != '':
+            yiri_msg_list.append(platform_message.Image(base64=event.picurl))
+        chain = platform_message.MessageChain(yiri_msg_list)
+
+        return chain
+
+class WecomBotEventConverter(abstract_platform_adapter.AbstractEventConverter):
+
+    @staticmethod
+    async def yiri2target(event:platform_events.MessageEvent):
+        return event.source_platform_object
+    
+    @staticmethod
+    async def target2yiri(event:WecomBotEvent):
+        message_chain = await WecomBotMessageConverter.target2yiri(event)
+        if event.type == 'single':
+            return platform_events.FriendMessage(
+                sender=platform_entities.Friend(
+                    id=event.userid,
+                    nickname='',
+                    remark='',
+                ),
+                message_chain=message_chain,
+                time=datetime.datetime.now().timestamp(),
+                source_platform_object=event,
+            )
+        elif event.type == 'group':
+            try:
+                sender = platform_entities.GroupMember(
+                    id=event.userid,
+                    permission='MEMBER',
+                    member_name=event.userid,
+                    group=platform_entities.Group(
+                        id=str(event.chatid),
+                        name='',
+                        permission=platform_entities.Permission.Member,
+                    ),
+                    special_title='',
+                    join_timestamp=0,
+                    last_speak_timestamp=0,
+                    mute_time_remaining=0,
+                )
+                time = datetime.datetime.now().timestamp()
+                return platform_events.GroupMessage(
+                    sender=sender,
+                    message_chain=message_chain,
+                    time=time,
+                    source_platform_object=event,
+                )
+            except Exception:
+                print(traceback.format_exc())
+
+class WecomBotAdapter(abstract_platform_adapter.AbstractMessagePlatformAdapter):
+    bot: WecomBotClient
+    bot_account_id: str
+    message_converter: WecomBotMessageConverter = WecomBotMessageConverter()
+    event_converter: WecomBotEventConverter = WecomBotEventConverter()
+    config: dict
+
+    def __init__(self, config: dict, logger: EventLogger):
+        required_keys = ['Token', 'EncodingAESKey', 'Corpid', 'BotId', 'port']
+        missing_keys = [key for key in required_keys if key not in config]
+        if missing_keys:
+            raise Exception(f'WecomBot 缺少配置项: {missing_keys}')
+
+        # 创建运行时 bot 对象
+        bot = WecomBotClient(
+            Token=config['Token'],
+            EnCodingAESKey=config['EncodingAESKey'],
+            Corpid=config['Corpid'],
+            logger=logger,
+        )
+        bot_account_id = config['BotId']
+
+        super().__init__(
+            config=config,
+            logger=logger,
+            bot=bot,
+            bot_account_id=bot_account_id,
+        )
+
+
+    async def reply_message(self, message_source:platform_events.MessageEvent, message:platform_message.MessageChain,quote_origin: bool = False):
+
+        content = await self.message_converter.yiri2target(message)
+        await self.bot.set_message(message_source.source_platform_object.message_id, content)
+
+    async def send_message(self, target_type, target_id, message):
+        pass
+
+    def register_listener(
+        self,
+        event_type: typing.Type[platform_events.Event],
+        callback: typing.Callable[[platform_events.Event, abstract_platform_adapter.AbstractMessagePlatformAdapter], None],
+    ):
+        async def on_message(event: WecomBotEvent):
+            try:
+                return await callback(await self.event_converter.target2yiri(event), self)
+            except Exception:
+                await self.logger.error(f'Error in wecombot callback: {traceback.format_exc()}')
+                print(traceback.format_exc())
+        try:
+            if event_type == platform_events.FriendMessage:
+                self.bot.on_message('single')(on_message)
+            elif event_type == platform_events.GroupMessage:
+                self.bot.on_message('group')(on_message)
+        except Exception:
+            print(traceback.format_exc())
+            
+
+    async def run_async(self):
+        async def shutdown_trigger_placeholder():
+            while True:
+                await asyncio.sleep(1)
+
+        await self.bot.run_task(
+            host='0.0.0.0',
+            port=self.config['port'],
+            shutdown_trigger=shutdown_trigger_placeholder,
+        )
+
+    async def kill(self) -> bool:
+        return False
+
+    async def unregister_listener(
+        self,
+        event_type: type,
+        callback: typing.Callable[[platform_events.Event, abstract_platform_adapter.AbstractMessagePlatformAdapter], None],
+    ):
+        return super().unregister_listener(event_type, callback)
+    
+    async def is_muted(self, group_id: int) -> bool:
+        pass
+
+    

pkg/platform/sources/wecombot.yaml

@@ -0,0 +1,52 @@
+apiVersion: v1
+kind: MessagePlatformAdapter
+metadata:
+  name: wecombot
+  label:
+    en_US: WeComBot
+    zh_Hans: 企业微信智能机器人
+  description:
+    en_US: WeComBot Adapter
+    zh_Hans: 企业微信智能机器人适配器，请查看文档了解使用方式
+  icon: wecombot.png
+spec:
+  config:
+    - name: port
+      label:
+        en_US: Port
+        zh_Hans: 监听端口
+      type: integer
+      required: true
+      default: 2291
+    - name: Corpid
+      label:
+        en_US: Corpid
+        zh_Hans: 企业ID
+      type: string
+      required: true
+      default: ""
+    - name: Token
+      label:
+        en_US: Token
+        zh_Hans: 令牌
+      type: string
+      required: true
+      default: ""
+    - name: EncodingAESKey
+      label:
+        en_US: EncodingAESKey
+        zh_Hans: 消息加解密密钥
+      type: string
+      required: true
+      default: ""
+    - name: BotId
+      label:
+        en_US: BotId
+        zh_Hans: 机器人ID
+      type: string
+      required: false
+      default: ""
+execution:
+  python:
+    path: ./wecombot.py
+    attr: WecomBotAdapter