diff --git a/bot.py b/bot.py index b223612..40fe80b 100644 --- a/bot.py +++ b/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() \ No newline at end of file