LangBot/src/langbot/pkg/platform/sources/officialaccount.py

from __future__ import annotations
import typing
import asyncio
import traceback
import pydantic
import datetime
import langbot_plugin.api.definition.abstract.platform.adapter as abstract_platform_adapter
from langbot.libs.official_account_api.oaevent import OAEvent
from langbot.libs.official_account_api.api import OAClient
from langbot.libs.official_account_api.api import OAClientForLongerResponse
import langbot_plugin.api.entities.builtin.platform.entities as platform_entities
import langbot_plugin.api.entities.builtin.platform.message as platform_message
import langbot_plugin.api.entities.builtin.platform.events as platform_events
from ..logger import EventLogger


class OAMessageConverter(abstract_platform_adapter.AbstractMessageConverter):
    @staticmethod
    async def yiri2target(message_chain: platform_message.MessageChain):
        for msg in message_chain:
            if type(msg) is platform_message.Plain:
                return msg.text

    @staticmethod
    async def target2yiri(message: str, message_id=-1):
        yiri_msg_list = []
        yiri_msg_list.append(platform_message.Source(id=message_id, time=datetime.datetime.now()))

        yiri_msg_list.append(platform_message.Plain(text=message))
        chain = platform_message.MessageChain(yiri_msg_list)

        return chain


class OAEventConverter(abstract_platform_adapter.AbstractEventConverter):
    @staticmethod
    async def target2yiri(event: OAEvent):
        if event.type == 'text':
            yiri_chain = await OAMessageConverter.target2yiri(event.message, event.message_id)

            friend = platform_entities.Friend(
                id=event.user_id,
                nickname=str(event.user_id),
                remark='',
            )

            return platform_events.FriendMessage(
                sender=friend,
                message_chain=yiri_chain,
                time=event.timestamp,
                source_platform_object=event,
            )
        else:
            return None


class OfficialAccountAdapter(abstract_platform_adapter.AbstractMessagePlatformAdapter):
    message_converter: OAMessageConverter = OAMessageConverter()
    event_converter: OAEventConverter = OAEventConverter()
    bot: typing.Union[OAClient, OAClientForLongerResponse] = pydantic.Field(exclude=True)
    bot_uuid: str = None

    def __init__(self, config: dict, logger: EventLogger):
        # 校验必填项
        required_keys = ['token', 'EncodingAESKey', 'AppSecret', 'AppID', 'Mode']
        missing_keys = [k for k in required_keys if k not in config]
        if missing_keys:
            raise Exception(f'OfficialAccount 缺少配置项: {missing_keys}')

        # 创建运行时 bot 对象，始终使用统一 webhook 模式
        if config['Mode'] == 'drop':
            bot = OAClient(
                token=config['token'],
                EncodingAESKey=config['EncodingAESKey'],
                Appsecret=config['AppSecret'],
                AppID=config['AppID'],
                logger=logger,
                unified_mode=True,
            )
        elif config['Mode'] == 'passive':
            bot = OAClientForLongerResponse(
                token=config['token'],
                EncodingAESKey=config['EncodingAESKey'],
                Appsecret=config['AppSecret'],
                AppID=config['AppID'],
                LoadingMessage=config.get('LoadingMessage', ''),
                logger=logger,
                unified_mode=True,
            )
        else:
            raise KeyError('请设置微信公众号通信模式')

        bot_account_id = config.get('AppID', '')

        super().__init__(
            bot=bot,
            bot_account_id=bot_account_id,
            config=config,
            logger=logger,
        )

    async def reply_message(
        self,
        message_source: platform_events.FriendMessage,
        message: platform_message.MessageChain,
        quote_origin: bool = False,
    ):
        content = await OAMessageConverter.yiri2target(message)
        if isinstance(self.bot, OAClient):
            await self.bot.set_message(message_source.message_chain.message_id, content)
        elif isinstance(self.bot, OAClientForLongerResponse):
            from_user = message_source.sender.id
            await self.bot.set_message(from_user, message_source.message_chain.message_id, content)

    async def send_message(self, target_type: str, target_id: str, message: platform_message.MessageChain):
        pass

    def register_listener(
        self,
        event_type: type,
        callback: typing.Callable[
            [platform_events.Event, abstract_platform_adapter.AbstractMessagePlatformAdapter], None
        ],
    ):
        async def on_message(event: OAEvent):
            self.bot_account_id = event.receiver_id
            try:
                return await callback(await self.event_converter.target2yiri(event), self)
            except Exception:
                await self.logger.error(f'Error in officialaccount callback: {traceback.format_exc()}')

        if event_type == platform_events.FriendMessage:
            self.bot.on_message('text')(on_message)
        elif event_type == platform_events.GroupMessage:
            pass

    def set_bot_uuid(self, bot_uuid: str):
        """设置 bot UUID（用于生成 webhook URL）"""
        self.bot_uuid = bot_uuid

    async def handle_unified_webhook(self, bot_uuid: str, path: str, request):
        """处理统一 webhook 请求。

        Args:
            bot_uuid: Bot 的 UUID
            path: 子路径（如果有的话）
            request: Quart Request 对象

        Returns:
            响应数据
        """
        return await self.bot.handle_unified_webhook(request)

    async def run_async(self):
        # 统一 webhook 模式下，不启动独立的 Quart 应用
        # 保持运行但不启动独立端口

        async def keep_alive():
            while True:
                await asyncio.sleep(1)

        await keep_alive()

    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: str,
    ) -> bool:
        pass