Update main.py

main.py

@@ -1,8 +1,9 @@
 import asyncio
 import os
 import logging
+import json # 【新增部分】引入 json 库解析小程序数据
 from fastapi import FastAPI, Request
-from aiogram import Bot, Dispatcher, types
+from aiogram import Bot, Dispatcher, types, F # 【新增部分】引入 F 用于魔法过滤
 from aiogram.types import Update
 from aiogram.enums import ParseMode
 from aiogram.client.default import DefaultBotProperties
@@ -12,14 +13,14 @@ logging.basicConfig(level=logging.INFO)
 
 # 从环境变量获取配置 (在 HF Space 的 Settings -> Variables and secrets 里设置)
 BOT_TOKEN = os.getenv("BOT_TOKEN")
-WEBHOOK_URL = os.getenv("WEBHOOK_URL")  # 你的 HF Space 网址，例如 https://your-username-space-name.hf.space
+WEBHOOK_URL = os.getenv("WEBHOOK_URL")  # 你的 HF Space 网址
 
 bot = Bot(token=BOT_TOKEN, default=DefaultBotProperties(parse_mode=ParseMode.HTML))
 dp = Dispatcher()
 app = FastAPI()
 
 # ----------------- 模拟数据库 -----------------
-# 警告：HF 重启会清空本地数据！后续请换成 Cloudflare D1 或外部数据库接口
+# 警告：HF 重启会清空本地数据！后续请换成真实的数据库
 CHANNEL_MAP = {
     -1001234567890: -1009876543210 # 来源频道 ID : 目标频道 ID
 }
@@ -27,6 +28,38 @@ CHANNEL_MAP = {
 # ----------------- 媒体组缓冲池 -----------------
 album_buffer = {}
 
+
+# ==========================================
+# 【新增部分】处理从小程序 (TWA) 传回来的数据
+# ==========================================
+@dp.message(F.web_app_data)
+async def handle_web_app_data(message: types.Message):
+    # 获取小程序发来的 JSON 字符串
+    data_str = message.web_app_data.data
+    
+    try:
+        data = json.loads(data_str)
+        if data.get("action") == "set_sync":
+            source = data.get("source")
+            target = data.get("target")
+            
+            logging.info(f"收到新配置: {source} -> {target}")
+            
+            # 动态更新内存中的字典
+            CHANNEL_MAP[source] = target 
+            
+            await message.answer(
+                f"✅ <b>配置保存成功！</b>\n\n将来从 <code>{source}</code> 发布的消息会自动同步到 <code>{target}</code>。", 
+                parse_mode=ParseMode.HTML
+            )
+            
+    except Exception as e:
+        logging.error(f"解析小程序数据失败: {e}")
+        await message.answer("❌ 配置解析失败，请重试。")
+# ==========================================
+
+
+# ----------------- 处理频道消息同步 -----------------
 @dp.channel_post()
 async def handle_channel_post(message: types.Message):
     source_id = message.chat.id
@@ -78,8 +111,8 @@ async def send_album_delayed(mg_id: str, source_id: int, target_id: int):
         except Exception as e:
             logging.error(f"同步媒体组失败: {e}")
 
-# ----------------- FastAPI 路由与 Webhook -----------------
 
+# ----------------- FastAPI 路由与 Webhook -----------------
 @app.on_event("startup")
 async def on_startup():
     # 启动时告诉 Telegram 把消息推送到我们的接口