AstrBot/packages/astrbot_adapter_wechat/wechat_message_event.py

import random, asyncio
from astrbot.core.utils.io import download_image_by_url
from astrbot.api import AstrMessageEvent, MessageChain, logger, AstrBotMessage, PlatformMetadata
from astrbot.api import Plain, Image
from vchat import Core

class WechatPlatformEvent(AstrMessageEvent):
    def __init__(self, message_str: str, message_obj: AstrBotMessage, platform_meta: PlatformMetadata, session_id: str, client: Core):
        super().__init__(message_str, message_obj, platform_meta, session_id)
        self.client = client
        
    @staticmethod
    async def send_with_client(client: Core, message: MessageChain, user_name: str):
        plain = ""
        for comp in message.chain:
            if isinstance(comp, Plain):
                plain += comp.text
            elif isinstance(comp, Image):
                if comp.file and comp.file.startswith("file:///"):
                    file_path = comp.file.replace("file:///", "")
                    with open(file_path, "rb") as f:
                        await client.send_image(user_name, fd=f)
                elif comp.file and comp.file.startswith("http"):
                    image_path = await download_image_by_url(comp.file)
                    with open(image_path, "rb") as f:
                        await client.send_image(user_name, fd=f)
            else:
                logger.error(f"不支持的 vchat(微信适配器) 消息类型: {comp}")
            await asyncio.sleep(random.uniform(0.5, 1.5)) # 🤓
        
        if plain:
            await client.send_msg(plain, user_name)
                
        
    async def send(self, message: MessageChain):
        await WechatPlatformEvent.send_with_client(self.client, message, self.message_obj.raw_message.from_.username)
        await super().send(message)