AstrBot 4.11.0__py3-none-any.whl → 4.11.2__py3-none-any.whl

astrbot/core/platform/sources/wechatpadpro/wechatpadpro_message_event.py

@@ -1,178 +0,0 @@
-import asyncio
-import base64
-import io
-from collections.abc import AsyncGenerator
-from typing import TYPE_CHECKING
-
-import aiohttp
-from PIL import Image as PILImage  # 使用别名避免冲突
-
-from astrbot import logger
-from astrbot.core.message.components import (
-    Image,
-    Plain,
-    Record,
-    WechatEmoji,
-)  # Import Image
-from astrbot.core.message.message_event_result import MessageChain
-from astrbot.core.platform.astr_message_event import AstrMessageEvent
-from astrbot.core.platform.astrbot_message import AstrBotMessage, MessageType
-from astrbot.core.platform.platform_metadata import PlatformMetadata
-from astrbot.core.utils.tencent_record_helper import audio_to_tencent_silk_base64
-
-if TYPE_CHECKING:
-    from .wechatpadpro_adapter import WeChatPadProAdapter
-
-
-class WeChatPadProMessageEvent(AstrMessageEvent):
-    def __init__(
-        self,
-        message_str: str,
-        message_obj: AstrBotMessage,
-        platform_meta: PlatformMetadata,
-        session_id: str,
-        adapter: "WeChatPadProAdapter",  # 传递适配器实例
-    ):
-        super().__init__(message_str, message_obj, platform_meta, session_id)
-        self.message_obj = message_obj  # Save the full message object
-        self.adapter = adapter  # Save the adapter instance
-
-    async def send(self, message: MessageChain):
-        async with aiohttp.ClientSession() as session:
-            for comp in message.chain:
-                await asyncio.sleep(1)
-                if isinstance(comp, Plain):
-                    await self._send_text(session, comp.text)
-                elif isinstance(comp, Image):
-                    await self._send_image(session, comp)
-                elif isinstance(comp, WechatEmoji):
-                    await self._send_emoji(session, comp)
-                elif isinstance(comp, Record):
-                    await self._send_voice(session, comp)
-        await super().send(message)
-
-    async def send_streaming(
-        self, generator: AsyncGenerator[MessageChain, None], use_fallback: bool = False
-    ):
-        buffer = None
-        async for chain in generator:
-            if not buffer:
-                buffer = chain
-            else:
-                buffer.chain.extend(chain.chain)
-        if not buffer:
-            return None
-        buffer.squash_plain()
-        await self.send(buffer)
-        return await super().send_streaming(generator, use_fallback)
-
-    async def _send_image(self, session: aiohttp.ClientSession, comp: Image):
-        b64 = await comp.convert_to_base64()
-        raw = self._validate_base64(b64)
-        b64c = self._compress_image(raw)
-        payload = {
-            "MsgItem": [
-                {"ImageContent": b64c, "MsgType": 3, "ToUserName": self.session_id},
-            ],
-        }
-        url = f"{self.adapter.base_url}/message/SendImageNewMessage"
-        await self._post(session, url, payload)
-
-    async def _send_text(self, session: aiohttp.ClientSession, text: str):
-        if (
-            self.message_obj.type == MessageType.GROUP_MESSAGE  # 确保是群聊消息
-            and self.adapter.settings.get(
-                "reply_with_mention",
-                False,
-            )  # 检查适配器设置是否启用 reply_with_mention
-            and self.message_obj.sender  # 确保有发送者信息
-            and (
-                self.message_obj.sender.user_id or self.message_obj.sender.nickname
-            )  # 确保发送者有 ID 或昵称
-        ):
-            # 优先使用 nickname，如果没有则使用 user_id
-            mention_text = (
-                self.message_obj.sender.nickname or self.message_obj.sender.user_id
-            )
-            message_text = f"@{mention_text} {text}"
-            # logger.info(f"已添加 @ 信息: {message_text}")
-        else:
-            message_text = text
-        if self.get_group_id() and "#" in self.session_id:
-            session_id = self.session_id.split("#")[0]
-        else:
-            session_id = self.session_id
-        payload = {
-            "MsgItem": [
-                {
-                    "MsgType": 1,
-                    "TextContent": message_text,
-                    "ToUserName": session_id,
-                },
-            ],
-        }
-        url = f"{self.adapter.base_url}/message/SendTextMessage"
-        await self._post(session, url, payload)
-
-    async def _send_emoji(self, session: aiohttp.ClientSession, comp: WechatEmoji):
-        payload = {
-            "EmojiList": [
-                {
-                    "EmojiMd5": comp.md5,
-                    "EmojiSize": comp.md5_len,
-                    "ToUserName": self.session_id,
-                },
-            ],
-        }
-        url = f"{self.adapter.base_url}/message/SendEmojiMessage"
-        await self._post(session, url, payload)
-
-    async def _send_voice(self, session: aiohttp.ClientSession, comp: Record):
-        record_path = await comp.convert_to_file_path()
-        # 默认已经存在 data/temp 中
-        b64, duration = await audio_to_tencent_silk_base64(record_path)
-        payload = {
-            "ToUserName": self.session_id,
-            "VoiceData": b64,
-            "VoiceFormat": 4,
-            "VoiceSecond": duration,
-        }
-        url = f"{self.adapter.base_url}/message/SendVoice"
-        await self._post(session, url, payload)
-
-    @staticmethod
-    def _validate_base64(b64: str) -> bytes:
-        return base64.b64decode(b64, validate=True)
-
-    @staticmethod
-    def _compress_image(data: bytes) -> str:
-        img = PILImage.open(io.BytesIO(data))
-        buf = io.BytesIO()
-        if img.format == "JPEG":
-            img.save(buf, "JPEG", quality=80)
-        else:
-            if img.mode in ("RGBA", "P"):
-                img = img.convert("RGB")
-            img.save(buf, "JPEG", quality=80)
-        # logger.info("图片处理完成！！！")
-        return base64.b64encode(buf.getvalue()).decode()
-
-    async def _post(self, session, url, payload):
-        params = {"key": self.adapter.auth_key}
-        try:
-            async with session.post(url, params=params, json=payload) as resp:
-                data = await resp.json()
-                if resp.status != 200 or data.get("Code") != 200:
-                    logger.error(f"{url} failed: {resp.status} {data}")
-        except Exception as e:
-            logger.error(f"{url} error: {e}")
-
-
-# TODO: 添加对其他消息组件类型的处理 (Record, Video, At等)
-# elif isinstance(component, Record):
-#     pass
-# elif isinstance(component, Video):
-#     pass
-# elif isinstance(component, At):
-#     pass
-# ...

astrbot/core/platform/sources/wechatpadpro/xml_data_parser.py

@@ -1,159 +0,0 @@
-from defusedxml import ElementTree as eT
-
-from astrbot.api import logger
-from astrbot.api.message_components import (
-    BaseMessageComponent,
-    Image,
-    Plain,
-)
-from astrbot.api.message_components import (
-    WechatEmoji as Emoji,
-)
-
-
-class GeweDataParser:
-    def __init__(
-        self,
-        content: str,
-        is_private_chat: bool = False,
-        cached_texts=None,
-        cached_images=None,
-        raw_message: dict | None = None,
-        downloader=None,
-    ):
-        self._xml = None
-        self.content = content
-        self.is_private_chat = is_private_chat
-        self.cached_texts = cached_texts or {}
-        self.cached_images = cached_images or {}
-        self.downloader = downloader
-
-        raw_message = raw_message or {}
-        self.from_user_name = raw_message.get("from_user_name", {}).get("str", "")
-        self.to_user_name = raw_message.get("to_user_name", {}).get("str", "")
-        self.msg_id = raw_message.get("msg_id", "")
-
-    def _format_to_xml(self):
-        if self._xml:
-            return self._xml
-
-        try:
-            msg_str = self.content
-            if not self.is_private_chat:
-                parts = self.content.split(":\n", 1)
-                msg_str = parts[1] if len(parts) == 2 else self.content
-
-            self._xml = eT.fromstring(msg_str)
-            return self._xml
-        except Exception as e:
-            logger.error(f"[XML解析失败] {e}")
-            raise
-
-    async def parse_mutil_49(self) -> list[BaseMessageComponent] | None:
-        """处理 msg_type == 49 的多种 appmsg 类型（目前支持 type==57）"""
-        try:
-            appmsg_type = self._format_to_xml().findtext(".//appmsg/type")
-            if appmsg_type == "57":
-                return await self.parse_reply()
-        except Exception as e:
-            logger.warning(f"[parse_mutil_49] 解析失败: {e}")
-        return None
-
-    async def parse_reply(self) -> list[BaseMessageComponent]:
-        """处理 type == 57 的引用消息：支持文本（1）、图片（3）、嵌套49（49）"""
-        components = []
-
-        try:
-            appmsg = self._format_to_xml().find("appmsg")
-            if appmsg is None:
-                return [Plain("[引用消息解析失败]")]
-
-            refermsg = appmsg.find("refermsg")
-            if refermsg is None:
-                return [Plain("[引用消息解析失败]")]
-
-            quote_type = int(refermsg.findtext("type", "0"))
-            nickname = refermsg.findtext("displayname", "未知发送者")
-            quote_content = refermsg.findtext("content", "")
-            svrid = refermsg.findtext("svrid")
-
-            match quote_type:
-                case 1:  # 文本引用
-                    quoted_text = self.cached_texts.get(str(svrid), quote_content)
-                    components.append(Plain(f"[引用] {nickname}: {quoted_text}"))
-
-                case 3:  # 图片引用
-                    quoted_image_b64 = self.cached_images.get(str(svrid))
-                    if not quoted_image_b64:
-                        try:
-                            quote_xml = eT.fromstring(quote_content)
-                            img = quote_xml.find("img")
-                            cdn_url = (
-                                img.get("cdnbigimgurl") or img.get("cdnmidimgurl")
-                                if img is not None
-                                else None
-                            )
-                            if cdn_url and self.downloader:
-                                image_resp = await self.downloader(
-                                    self.from_user_name,
-                                    self.to_user_name,
-                                    self.msg_id,
-                                )
-                                quoted_image_b64 = (
-                                    image_resp.get("Data", {})
-                                    .get("Data", {})
-                                    .get("Buffer")
-                                )
-                        except Exception as e:
-                            logger.warning(f"[引用图片解析失败] svrid={svrid} err={e}")
-
-                    if quoted_image_b64:
-                        components.extend(
-                            [
-                                Image.fromBase64(quoted_image_b64),
-                                Plain(f"[引用] {nickname}: [引用的图片]"),
-                            ],
-                        )
-                    else:
-                        components.append(
-                            Plain(f"[引用] {nickname}: [引用的图片 - 未能获取]"),
-                        )
-
-                case 49:  # 嵌套引用
-                    try:
-                        nested_root = eT.fromstring(quote_content)
-                        nested_title = nested_root.findtext(".//appmsg/title", "")
-                        components.append(Plain(f"[引用] {nickname}: {nested_title}"))
-                    except Exception as e:
-                        logger.warning(f"[嵌套引用解析失败] err={e}")
-                        components.append(Plain(f"[引用] {nickname}: [嵌套引用消息]"))
-
-                case _:  # 其他未识别类型
-                    logger.info(f"[未知引用类型] quote_type={quote_type}")
-                    components.append(Plain(f"[引用] {nickname}: [不支持的引用类型]"))
-
-            # 主消息标题
-            title = appmsg.findtext("title", "")
-            if title:
-                components.append(Plain(title))
-
-        except Exception as e:
-            logger.error(f"[parse_reply] 总体解析失败: {e}")
-            return [Plain("[引用消息解析失败]")]
-
-        return components
-
-    def parse_emoji(self) -> Emoji | None:
-        """处理 msg_type == 47 的表情消息（emoji）"""
-        try:
-            emoji_element = self._format_to_xml().find(".//emoji")
-            if emoji_element is not None:
-                return Emoji(
-                    md5=emoji_element.get("md5"),
-                    md5_len=emoji_element.get("len"),
-                    cdnurl=emoji_element.get("cdnurl"),
-                )
-        except Exception as e:
-            logger.error(f"[parse_emoji] 解析失败: {e}")
-
-        return None