feat: Implement message edit synchronization with reply-to link (#2)

bot.py

@@ -4,12 +4,14 @@ import aiosqlite
 from telegram import Update, InlineKeyboardButton, InlineKeyboardMarkup
 from telegram.ext import Application, MessageHandler, CommandHandler, filters, ContextTypes
 
+
 # --- 从环境变量读取配置 ---
 BOT_TOKEN = os.getenv('BOT_TOKEN')
 ADMIN_ID = int(os.getenv('ADMIN_ID', '0'))
 DB_FILE = os.getenv('DB_PATH', '/app/data/messages.db')
 GITHUB_URL = os.getenv('GITHUB_URL', 'https://github.com/EchoZenith/TelegramContactBot')
 
+
 # 启用日志
 logging.basicConfig(
     format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', 
@@ -17,80 +19,119 @@ logging.basicConfig(
 )
 logger = logging.getLogger(__name__)
 
+
 # 初始化数据库
 async def init_db():
     async with aiosqlite.connect(DB_FILE) as db:
-        await db.execute('''CREATE TABLE IF NOT EXISTS msg_map 
-                          (admin_msg_id INTEGER PRIMARY KEY, user_id INTEGER)''')
+        # 记录：管理员端消息ID <-> 用户端消息ID <-> 用户ID
+        await db.execute('''CREATE TABLE IF NOT EXISTS msg_pairs 
+                          (admin_msg_id INTEGER PRIMARY KEY, user_msg_id INTEGER, user_id INTEGER)''')
         await db.commit()
 
+
 # 处理 /start 命令
 async def start(update: Update, context: ContextTypes.DEFAULT_TYPE):
     user_id = update.effective_chat.id
+    keyboard = [[InlineKeyboardButton("🌟 查看项目源码", url=GITHUB_URL)]]
+    reply_markup = InlineKeyboardMarkup(keyboard)
+    
     if user_id != ADMIN_ID:
-        # 创建一个带有 URL 的按钮
-        keyboard = [
-            [
-                InlineKeyboardButton("🌟 查看项目源码", url=GITHUB_URL)
-            ]
-        ]
-        reply_markup = InlineKeyboardMarkup(keyboard)
-        
-        await update.message.reply_text(
-            "Hello!\n\nYou can contact us using this bot.",
-            reply_markup=reply_markup
-        )
+        await update.message.reply_text("Hello!\n\nYou can contact us using this bot.", reply_markup=reply_markup)
     else:
         await update.message.reply_text("你好，管理员！有人给机器人发消息时，我会转发给你。你直接【回复】该消息即可回信。")
 
-# 处理所有普通消息（文字、图片、语音、文件等）
-async def handle_message(update: Update, context: ContextTypes.DEFAULT_TYPE):
-    if not update.message:
+# 处理同步修改逻辑
+async def handle_edit(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    edited_msg = update.edited_message
+    if not edited_msg:
         return
-    
+        
+    user_id = edited_msg.chat.id
+    async with aiosqlite.connect(DB_FILE) as db:
+        if user_id != ADMIN_ID:
+            # 1. 查找旧的转发记录
+            async with db.execute("SELECT admin_msg_id FROM msg_pairs WHERE user_msg_id = ? AND user_id = ?", 
+                                 (edited_msg.message_id, user_id)) as cursor:
+                row = await cursor.fetchone()
+                
+                if row:
+                    old_admin_msg_id = row[0]
+                    # 获取新内容（文字或媒体说明）
+                    content = edited_msg.text or edited_msg.caption or ""
+                    new_text = f"【此消息已修改】\n{content}"
+                    
+                    try:
+                        # 2. 判断是纯文本还是带媒体的消息
+                        if edited_msg.text:
+                            # 发送纯文字消息，并回复在旧消息上
+                            new_msg = await context.bot.send_message(
+                                chat_id=ADMIN_ID,
+                                text=new_text,
+                                reply_to_message_id=old_admin_msg_id
+                            )
+                        else:
+                            # 如果是图片/视频等媒体，复制它并修改其 Caption
+                            new_msg = await context.bot.copy_message(
+                                chat_id=ADMIN_ID,
+                                from_chat_id=user_id,
+                                message_id=edited_msg.message_id,
+                                caption=new_text,
+                                reply_to_message_id=old_admin_msg_id
+                            )
+                            
+                        # 3. 更新映射关系，确保管理员回复这条带“【新消息】”提示的消息时，也能回传
+                        await db.execute("INSERT OR REPLACE INTO msg_pairs VALUES (?, ?, ?)", 
+                                         (new_msg.message_id, edited_msg.message_id, user_id))
+                        await db.commit()
+                        
+                    except Exception as e:
+                        logger.error(f"处理用户修改回传失败: {e}")
+        else:
+            # 管理员修改逻辑保持不变（原地编辑用户收到的消息）
+            async with db.execute("SELECT user_id, user_msg_id FROM msg_pairs WHERE admin_msg_id = ?", 
+                                 (edited_msg.message_id,)) as cursor:
+                row = await cursor.fetchone()
+                if row:
+                    try:
+                        new_content = edited_msg.text or edited_msg.caption
+                        await context.bot.edit_message_text(chat_id=row[0], message_id=row[1], text=new_content)
+                    except Exception as e:
+                        logger.warning(f"管理员修改同步略过: {e}")
+
+
+# 处理所有普通消息
+async def handle_message(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    if not update.message: return
     user_id = update.effective_chat.id
     message = update.message
-
-    # 1. 用户发给机器人 -> 转发给管理员
+    
+    # 1. 用户 -> 管理员
     if user_id != ADMIN_ID:
         try:
-            forwarded_msg = await context.bot.forward_message(
-                chat_id=ADMIN_ID,
-                from_chat_id=user_id,
-                message_id=message.message_id
-            )
-            
-            # 记录对应关系到数据库
+            forwarded_msg = await context.bot.forward_message(chat_id=ADMIN_ID, from_chat_id=user_id, message_id=message.message_id)
             async with aiosqlite.connect(DB_FILE) as db:
-                await db.execute("INSERT INTO msg_map VALUES (?, ?)", 
-                                 (forwarded_msg.message_id, user_id))
+                await db.execute("INSERT INTO msg_pairs VALUES (?, ?, ?)", 
+                                 (forwarded_msg.message_id, message.message_id, user_id))
                 await db.commit()
-        except Exception as e:
-            logger.error(f"转发失败: {e}")
+        except Exception as e: logger.error(f"转发失败: {e}")
+        
+    # 2. 管理员回复 -> 用户
+    elif message.reply_to_message:
+        async with aiosqlite.connect(DB_FILE) as db:
+            async with db.execute("SELECT user_id FROM msg_pairs WHERE admin_msg_id = ?", (message.reply_to_message.message_id,)) as cursor:
+                row = await cursor.fetchone()
+        
+        if row:
+            target_user_id = row[0]
+            try:
+                sent_msg = await context.bot.copy_message(chat_id=target_user_id, from_chat_id=ADMIN_ID, message_id=message.message_id)
+                # 记录管理员发出的回复，以便管理员后续修改这条回复时能同步给用户
+                async with aiosqlite.connect(DB_FILE) as db:
+                    await db.execute("INSERT INTO msg_pairs (admin_msg_id, user_msg_id, user_id) VALUES (?, ?, ?)", 
+                                     (message.message_id, sent_msg.message_id, target_user_id))
+                    await db.commit()
+            except Exception as e: await message.reply_text(f"回复失败: {e}")
 
-    # 2. 管理员回复转发的消息 -> 回传给用户
-    else:
-        if message.reply_to_message:
-            reply_id = message.reply_to_message.message_id
-            
-            async with aiosqlite.connect(DB_FILE) as db:
-                async with db.execute("SELECT user_id FROM msg_map WHERE admin_msg_id = ?", (reply_id,)) as cursor:
-                    row = await cursor.fetchone()
-            
-            if row:
-                target_user_id = row[0]
-                try:
-                    # 使用 copy_message 原样复制回复的内容
-                    await context.bot.copy_message(
-                        chat_id=target_user_id,
-                        from_chat_id=ADMIN_ID,
-                        message_id=message.message_id
-                    )
-                except Exception as e:
-                    await message.reply_text(f"回复发送失败，可能用户已停用机器人: {e}")
-            else:
-                await message.reply_text("找不到该消息的来源记录，无法回复。")
-        # 如果管理员没用回复功能，而是直接发消息，不执行任何操作
 
 async def post_init(application: Application):
     await init_db()
@@ -100,13 +141,16 @@ if __name__ == '__main__':
     if not BOT_TOKEN or ADMIN_ID == 0:
         print("错误: 请确保环境变量 BOT_TOKEN 和 ADMIN_ID 已正确设置！")
         exit(1)
-
+        
     # 构建应用
     application = Application.builder().token(BOT_TOKEN).post_init(post_init).build()
     
     # 处理器注册
     application.add_handler(CommandHandler("start", start))
+    # 监听修改消息的更新
+    application.add_handler(MessageHandler(filters.UpdateType.EDITED_MESSAGE & filters.TEXT, handle_edit))
+    # 监听普通消息
     application.add_handler(MessageHandler(filters.ALL & ~filters.COMMAND, handle_message))
     
     print(f"机器人已启动... 管理员ID: {ADMIN_ID}")
-    application.run_polling()
+    application.run_polling()