myagent-ai 1.0.0

package/chatbot/manager.py

@@ -0,0 +1,164 @@
+"""
+chatbot/manager.py - 聊天平台管理器
+=====================================
+统一管理所有聊天平台的生命周期和消息路由。
+纯 Python 实现，不依赖 Node.js 或额外网关。
+"""
+from __future__ import annotations
+
+import asyncio
+from typing import Any, Callable, Dict, List, Optional
+
+from core.logger import get_logger
+from chatbot.base import BaseChatBot, ChatMessage, ChatResponse
+from config import ChatPlatformConfig
+
+logger = get_logger("myagent.chatbot")
+
+
+class ChatBotManager:
+    """
+    聊天平台管理器。
+
+    功能:
+      - 统一管理多个聊天平台
+      - 消息路由到主 Agent
+      - 多用户/多会话隔离
+      - 平台独立的生命周期管理
+      - 后台异步运行
+
+    使用示例:
+        manager = ChatBotManager()
+        manager.setup_platforms(config.chat_platforms, message_handler=handle_message)
+        await manager.start_all()
+    """
+
+    def __init__(self):
+        self._bots: Dict[str, BaseChatBot] = {}
+        self._session_map: Dict[str, str] = {}  # session_id -> last_message
+
+    def setup_platforms(
+        self,
+        platform_configs: List[ChatPlatformConfig],
+        message_handler: Callable,
+    ):
+        """
+        初始化所有聊天平台。
+
+        Args:
+            platform_configs: 平台配置列表
+            message_handler: 统一消息处理回调
+        """
+        for cfg in platform_configs:
+            if not cfg.enabled:
+                continue
+            try:
+                bot = self._create_bot(cfg, message_handler)
+                if bot:
+                    self._bots[cfg.platform] = bot
+                    logger.info(f"聊天平台已配置: {cfg.platform}")
+            except Exception as e:
+                logger.error(f"平台 {cfg.platform} 初始化失败: {e}")
+
+    def _create_bot(
+        self,
+        config: ChatPlatformConfig,
+        message_handler: Callable,
+    ) -> Optional[BaseChatBot]:
+        """根据配置创建对应的聊天机器人实例"""
+        platform = config.platform
+
+        if platform == "telegram":
+            from chatbot.telegram_bot import TelegramBot
+            return TelegramBot(
+                token=config.token,
+                allowed_users=config.allowed_users,
+                message_handler=message_handler,
+                **config.extra,
+            )
+
+        elif platform == "discord":
+            from chatbot.discord_bot import DiscordBot
+            return DiscordBot(
+                token=config.token,
+                allowed_users=config.allowed_users,
+                message_handler=message_handler,
+                **config.extra,
+            )
+
+        elif platform == "feishu":
+            from chatbot.feishu_bot import FeishuBot
+            return FeishuBot(
+                token=config.token,
+                app_id=config.app_id,
+                app_secret=config.app_secret,
+                allowed_users=config.allowed_users,
+                message_handler=message_handler,
+                **config.extra,
+            )
+
+        elif platform == "qq":
+            from chatbot.qq_bot import QQBot
+            return QQBot(
+                token=config.token,
+                allowed_users=config.allowed_users,
+                message_handler=message_handler,
+                **config.extra,
+            )
+
+        elif platform == "wechat":
+            from chatbot.wechat_bot import WeChatBot
+            return WeChatBot(
+                token=config.token,
+                allowed_users=config.allowed_users,
+                message_handler=message_handler,
+                **config.extra,
+            )
+
+        else:
+            logger.warning(f"不支持的平台: {platform}")
+            return None
+
+    async def start_all(self):
+        """启动所有聊天平台"""
+        tasks = []
+        for name, bot in self._bots.items():
+            logger.info(f"启动聊天平台: {name}")
+            task = asyncio.create_task(self._run_bot(name, bot))
+            tasks.append(task)
+        await asyncio.gather(*tasks, return_exceptions=True)
+
+    async def _run_bot(self, name: str, bot: BaseChatBot):
+        """安全运行单个聊天平台"""
+        try:
+            await bot.start()
+        except Exception as e:
+            logger.error(f"聊天平台 {name} 运行异常: {e}")
+
+    async def stop_all(self):
+        """停止所有聊天平台"""
+        for name, bot in self._bots.items():
+            try:
+                await bot.stop()
+            except Exception as e:
+                logger.error(f"停止 {name} 失败: {e}")
+        logger.info("所有聊天平台已停止")
+
+    async def send_to_all(self, text: str):
+        """向所有平台广播消息"""
+        for bot in self._bots.values():
+            try:
+                await bot.send_message(ChatResponse(text=text))
+            except Exception as e:
+                logger.error(f"广播失败: {e}")
+
+    def get_active_platforms(self) -> List[str]:
+        """获取活跃平台列表"""
+        return list(self._bots.keys())
+
+    def get_stats(self) -> Dict[str, Any]:
+        """获取状态"""
+        return {
+            "active_platforms": list(self._bots.keys()),
+            "platform_count": len(self._bots),
+        }

package/chatbot/qq_bot.py

@@ -0,0 +1,189 @@
+"""
+chatbot/qq_bot.py - QQ 机器人
+===============================
+QQ 机器人适配器。
+
+注意: QQ 官方机器人 API 有严格限制，本实现支持:
+  1. QQ 官方机器人 (go-cqhttp / Lagrange / LLOneBot 等 OneBot 协议)
+  2. 纯 HTTP/Websocket 连接，无需 Node.js
+
+配置方式:
+  - 通过 OneBot v11 HTTP API 连接
+  - 需要先运行 OneBot 实现 (如 Lagrange.Core)
+"""
+from __future__ import annotations
+
+import asyncio
+import json
+from typing import Optional
+
+from chatbot.base import BaseChatBot, ChatMessage, ChatResponse
+
+try:
+    import aiohttp
+    HAS_AIOHTTP = True
+except ImportError:
+    HAS_AIOHTTP = False
+
+
+class QQBot(BaseChatBot):
+    """
+    QQ 机器人适配器 (OneBot v11 协议)。
+
+    配置要求:
+      - token: OneBot HTTP API 地址 (如 http://127.0.0.1:3000)
+      - extra.ws_url: OneBot WebSocket 地址 (如 ws://127.0.0.1:3001)
+      - extra.self_id: 机器人 QQ 号
+
+    需要预先运行 OneBot 实现 (推荐 Lagrange.Core):
+      https://github.com/LagrangeDev/Lagrange.Core
+    """
+
+    platform_name = "qq"
+
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)
+        # token 字段存储 OneBot HTTP API 地址
+        self.api_url = self.token.rstrip("/") if self.token else ""
+        self.ws_url = self.config.get("ws_url", "")
+        self.self_id = self.config.get("self_id", "")
+        self._session = None
+
+    async def start(self):
+        """启动 QQ 机器人"""
+        if not HAS_AIOHTTP:
+            self.logger.error("请安装 aiohttp: pip install aiohttp")
+            return
+
+        if not self.api_url:
+            self.logger.error("QQ OneBot API 地址未配置")
+            return
+
+        self._session = aiohttp.ClientSession()
+        self._running = True
+
+        # 测试连接
+        try:
+            async with self._session.get(f"{self.api_url}/get_login_info",
+                                          timeout=aiohttp.ClientTimeout(total=5)) as resp:
+                info = await resp.json()
+                if info.get("status") == "ok":
+                    self.self_id = str(info["data"].get("user_id", ""))
+                    self.logger.info(f"QQ 机器人已连接: {self.self_id}")
+                else:
+                    self.logger.warning(f"QQ 连接测试失败: {info}")
+        except Exception as e:
+            self.logger.warning(f"无法连接 OneBot API ({self.api_url}): {e}")
+
+        # 启动 WebSocket 监听
+        if self.ws_url:
+            asyncio.create_task(self._ws_listen())
+        else:
+            self.logger.info("QQ 机器人使用 HTTP API 模式(仅发送)")
+            while self._running:
+                await asyncio.sleep(1)
+
+    async def stop(self):
+        """停止 QQ 机器人"""
+        self._running = False
+        if self._session:
+            await self._session.close()
+        self.logger.info("QQ 机器人已停止")
+
+    async def send_message(self, response: ChatResponse) -> bool:
+        """发送消息到 QQ"""
+        if not self._session or not self.api_url:
+            return False
+
+        try:
+            if response.is_group:
+                # 群消息
+                url = f"{self.api_url}/send_group_msg"
+                payload = {
+                    "group_id": int(response.chat_id),
+                    "message": response.text[:4000],
+                }
+            else:
+                # 私聊
+                url = f"{self.api_url}/send_private_msg"
+                payload = {
+                    "user_id": int(response.user_id or response.chat_id),
+                    "message": response.text[:4000],
+                }
+
+            if response.reply_to:
+                payload["message"] = f"[CQ:reply,id={response.reply_to}]{payload['message']}"
+
+            async with self._session.post(url, json=payload) as resp:
+                result = await resp.json()
+                if result.get("status") == "ok":
+                    return True
+                self.logger.error(f"QQ 发送失败: {result}")
+                return False
+        except Exception as e:
+            self.logger.error(f"QQ 发送消息异常: {e}")
+            return False
+
+    async def _ws_listen(self):
+        """WebSocket 监听消息"""
+        if not self.ws_url:
+            return
+
+        while self._running:
+            try:
+                async with self._session.ws_connect(self.ws_url) as ws:
+                    self.logger.info(f"QQ WebSocket 已连接: {self.ws_url}")
+
+                    async for msg in ws:
+                        if msg.type == aiohttp.WSMsgType.TEXT:
+                            data = json.loads(msg.data)
+
+                            # OneBot v11 事件格式
+                            if data.get("post_type") == "message":
+                                await self._handle_event(data)
+                        elif msg.type in (
+                            aiohttp.WSMsgType.ERROR,
+                            aiohttp.WSMsgType.CLOSED,
+                        ):
+                            break
+
+            except Exception as e:
+                self.logger.error(f"QQ WebSocket 异常: {e}")
+                if self._running:
+                    await asyncio.sleep(5)
+
+    async def _handle_event(self, data: dict):
+        """处理 OneBot 事件"""
+        user_id = str(data.get("user_id", ""))
+        raw_msg = data.get("raw_message", data.get("message", ""))
+
+        # 去除 CQ 码
+        import re
+        clean_msg = re.sub(r'\[CQ:[^\]]+\]', '', raw_msg).strip()
+
+        if not clean_msg:
+            return
+
+        # 只处理 @机器人 或私信
+        is_group = data.get("message_type") == "group"
+        if is_group:
+            group_id = str(data.get("group_id", ""))
+            # 检查是否被 @
+            at_pattern = f"[CQ:at,qq={self.self_id}]"
+            if at_pattern not in data.get("message", ""):
+                return  # 群聊中未被 @，忽略
+            chat_id = group_id
+        else:
+            chat_id = user_id
+
+        message = ChatMessage(
+            platform=self.platform_name,
+            chat_id=chat_id,
+            user_id=user_id,
+            username=data.get("sender", {}).get("nickname", ""),
+            text=clean_msg,
+            is_group=is_group,
+            reply_to=str(data.get("message_id", "")),
+            raw_data=data,
+        )
+        await self._handle_message(message)

package/chatbot/telegram_bot.py

@@ -0,0 +1,167 @@
+"""
+chatbot/telegram_bot.py - Telegram 机器人
+===========================================
+使用 python-telegram-bot 库接入 Telegram。
+纯 Python 实现，异步运行。
+"""
+from __future__ import annotations
+
+import asyncio
+from typing import Optional, List
+
+from chatbot.base import BaseChatBot, ChatMessage, ChatResponse
+
+try:
+    from telegram import Update
+    from telegram.ext import (
+        Application, CommandHandler, MessageHandler,
+        filters, ContextTypes,
+    )
+    HAS_TELEGRAM = True
+except ImportError:
+    HAS_TELEGRAM = False
+
+
+class TelegramBot(BaseChatBot):
+    """
+    Telegram 机器人适配器。
+
+    配置要求:
+      - token: Bot Token (从 @BotFather 获取)
+    """
+
+    platform_name = "telegram"
+
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)
+        self._app: Optional[Application] = None
+
+    async def start(self):
+        """启动 Telegram 机器人"""
+        if not HAS_TELEGRAM:
+            self.logger.error("请安装 python-telegram-bot: pip install python-telegram-bot")
+            return
+
+        if not self.token:
+            self.logger.error("Telegram Bot Token 未配置")
+            return
+
+        # 创建 Application
+        self._app = Application.builder().token(self.token).build()
+
+        # 注册处理器
+        self._app.add_handler(CommandHandler("start", self._cmd_start))
+        self._app.add_handler(CommandHandler("help", self._cmd_help))
+        self._app.add_handler(CommandHandler("status", self._cmd_status))
+        self._app.add_handler(CommandHandler("clear", self._cmd_clear))
+        self._app.add_handler(
+            MessageHandler(filters.TEXT & ~filters.COMMAND, self._on_message)
+        )
+
+        self._running = True
+
+        # 启动轮询
+        await self._app.initialize()
+        await self._app.start()
+        await self._app.updater.start_polling(
+            allowed_updates=Update.ALL_TYPES,
+            drop_pending_updates=True,
+        )
+
+        self.logger.info("Telegram 机器人已启动")
+
+        # 保持运行
+        while self._running:
+            await asyncio.sleep(1)
+
+    async def stop(self):
+        """停止 Telegram 机器人"""
+        self._running = False
+        if self._app:
+            await self._app.updater.stop()
+            await self._app.stop()
+            await self._app.shutdown()
+        self.logger.info("Telegram 机器人已停止")
+
+    async def send_message(self, response: ChatResponse) -> bool:
+        """发送消息到 Telegram"""
+        if not self._app or not response.chat_id:
+            return False
+
+        try:
+            kwargs = {
+                "chat_id": int(response.chat_id) if response.chat_id.isdigit() else response.chat_id,
+                "text": response.text[:4096],  # Telegram 消息限制
+            }
+            if response.parse_mode:
+                kwargs["parse_mode"] = response.parse_mode
+            if response.reply_to:
+                kwargs["reply_to_message_id"] = int(response.reply_to)
+
+            await self._app.bot.send_message(**kwargs)
+            return True
+        except Exception as e:
+            self.logger.error(f"发送消息失败: {e}")
+            return False
+
+    # ==========================================================================
+    # 命令处理器
+    # ==========================================================================
+
+    async def _cmd_start(self, update: Update, context: ContextTypes.DEFAULT_TYPE):
+        """处理 /start 命令"""
+        await update.message.reply_text(
+            "👋 你好！我是 MyAgent，你的本地AI助手。\n\n"
+            "直接发送消息即可与我对话。\n"
+            "我可以帮你执行代码、操作文件、搜索信息等。\n\n"
+            "命令列表:\n"
+            "/help - 查看帮助\n"
+            "/status - 查看状态\n"
+            "/clear - 清除对话历史"
+        )
+
+    async def _cmd_help(self, update: Update, context: ContextTypes.DEFAULT_TYPE):
+        """处理 /help 命令"""
+        await update.message.reply_text(
+            "🤖 MyAgent 帮助\n\n"
+            "我能做什么:\n"
+            "• 执行 Python/Shell/PowerShell 代码\n"
+            "• 读写搜索文件\n"
+            "• 搜索互联网\n"
+            "• 查询系统信息\n"
+            "• 浏览器自动化\n"
+            "• 记住你的偏好和经验\n\n"
+            "使用方式:\n"
+            "直接用自然语言描述你的需求即可。"
+        )
+
+    async def _cmd_status(self, update: Update, context: ContextTypes.DEFAULT_TYPE):
+        """处理 /status 命令"""
+        await update.message.reply_text("✅ MyAgent 正在运行中...")
+
+    async def _cmd_clear(self, update: Update, context: ContextTypes.DEFAULT_TYPE):
+        """处理 /clear 命令"""
+        await self._handle_message(ChatMessage(
+            platform=self.platform_name,
+            chat_id=str(update.effective_chat.id),
+            user_id=str(update.effective_user.id),
+            username=update.effective_user.username or update.effective_user.first_name or "",
+            text="__cmd_clear__",
+            is_group=update.effective_chat.type != "private",
+            raw_data={"command": "clear"},
+        ))
+        await update.message.reply_text("🗑️ 对话历史已清除。")
+
+    async def _on_message(self, update: Update, context: ContextTypes.DEFAULT_TYPE):
+        """处理普通消息"""
+        message = ChatMessage(
+            platform=self.platform_name,
+            chat_id=str(update.effective_chat.id),
+            user_id=str(update.effective_user.id),
+            username=update.effective_user.username or update.effective_user.first_name or "",
+            text=update.message.text or "",
+            is_group=update.effective_chat.type != "private",
+            reply_to=str(update.message.reply_to_message.message_id) if update.message.reply_to_message else "",
+            raw_data={"update": str(update)},
+        )
+        await self._handle_message(message)