feat: 适配 at，回复和语音输入

Author: Soulter

tg_message_adapter.py

@@ -4,11 +4,11 @@
 
 from astrbot.api.platform import Platform, AstrBotMessage, MessageMember, PlatformMetadata, MessageType
 from astrbot.api.event import MessageChain
-from astrbot.api.message_components import *
+from astrbot.api.message_components import Plain, Image, Record
 from astrbot.core.platform.astr_message_event import MessageSesion
 from astrbot.api.platform import register_platform_adapter
 
-from telegram import Update
+from telegram import Update, File
 from telegram.ext import ApplicationBuilder, ContextTypes, filters
 from telegram.constants import ChatType
 from telegram.ext import MessageHandler as TelegramMessageHandler
@@ -48,7 +48,10 @@ def meta(self) -> PlatformMetadata:
     @override
     async def run(self):
         self.application = ApplicationBuilder().token(self.config['telegram_token']).build()
-        message_handler = TelegramMessageHandler(filters.TEXT, self.convert_message)
+        message_handler = TelegramMessageHandler(
+            filters=None,
+            callback=self.convert_message
+        )
         self.application.add_handler(message_handler)
         await self.application.initialize()
         await self.application.start()
@@ -62,24 +65,28 @@ async def start(self, update: Update, context: ContextTypes.DEFAULT_TYPE):
 
     async def convert_message(self, update: Update, context: ContextTypes.DEFAULT_TYPE) -> AstrBotMessage:
         message = AstrBotMessage()
-        
         # 获得是群聊还是私聊
         if update.effective_chat.type == ChatType.PRIVATE:
             message.type = MessageType.FRIEND_MESSAGE
         else:
             message.type = MessageType.GROUP_MESSAGE
-        
-        
-        plain_text = update.message.text
+            message.group_id = update.effective_chat.id
         message.message_id = str(update.message.message_id)
-        message.message_str = plain_text
         message.session_id = str(update.effective_chat.id)
-        message.sender = MessageMember(str(update.effective_chat.id), update.effective_chat.effective_name)
-        message.message = [Plain(plain_text),]
-        message.message_str = plain_text
+        message.sender = MessageMember(str(update.effective_user.id), update.effective_user.username)
         message.self_id = str(context.bot.id)
         message.raw_message = update
-
+        message.message_str = ""
+        
+        if update.message.text:
+            plain_text = update.message.text
+            message.message = [Plain(plain_text),]
+            message.message_str = plain_text
+            
+        elif update.message.voice:
+            file = await update.message.voice.get_file()
+            message.message = [Record(file=file.file_path, url=file.file_path),]
+            
 
         await self.handle_msg(message)
 

tg_message_event.py

@@ -1,6 +1,6 @@
 from astrbot.api.event import AstrMessageEvent, MessageChain
-from astrbot.api.platform import AstrBotMessage, PlatformMetadata
-from astrbot.api.message_components import Plain, Image
+from astrbot.api.platform import AstrBotMessage, PlatformMetadata, MessageType
+from astrbot.api.message_components import Plain, Image, Reply, At
 from telegram.ext import ExtBot
 
 class TelegramPlatformEvent(AstrMessageEvent):
@@ -11,18 +11,40 @@ def __init__(self, message_str: str, message_obj: AstrBotMessage, platform_meta:
     @staticmethod
     async def send_with_client(client: ExtBot, message: MessageChain, user_name: str):
         image_path = None
+        
+        has_reply = False
+        reply_message_id = None
+        at_user_id = None
+        for i in message.chain:
+            if isinstance(i, Reply):
+                has_reply = True
+                reply_message_id = i.id
+            if isinstance(i, At):
+                at_user_id = i.name
+        
+        at_flag = False
         for i in message.chain:
+            payload = {
+                "chat_id": user_name,
+            }
+            if has_reply:
+                payload["reply_to_message_id"] = reply_message_id
+            
             if isinstance(i, Plain):
-                await client.send_message(chat_id=user_name, text=i.text)
+                if at_user_id and not at_flag:
+                    i.text = f"@{at_user_id} " + i.text
+                    at_flag = True
+                await client.send_message(text=i.text, **payload)
             elif isinstance(i, Image):
                 if i.path:
                     image_path = i.path
                 else:
                     image_path = i.file
-                await client.send_photo(chat_id=user_name, photo=image_path)
-
-                
+                await client.send_photo(photo=image_path, **payload)
 
     async def send(self, message: MessageChain):
-        await self.send_with_client(self.client, message, self.get_sender_id())
+        if self.get_message_type() == MessageType.GROUP_MESSAGE:
+            await self.send_with_client(self.client, message, self.message_obj.group_id)
+        else:
+            await self.send_with_client(self.client, message, self.get_sender_id())
         await super().send(message)