@vrs-soft/wecom-aibot-mcp 1.5.0 → 2.3.0

package/dist/connection-log.js

@@ -5,6 +5,7 @@
  */
 import * as fs from 'fs';
 import * as path from 'path';
+import { logger } from './logger.js';
 const CONFIG_DIR = path.join(process.env.HOME || '', '.wecom-aibot-mcp');
 const LOG_FILE = path.join(CONFIG_DIR, 'connection.log');
 // 全局状态
@@ -30,20 +31,20 @@ function writeLog(record) {
     const time = record.timestamp;
     switch (record.event) {
         case 'connected':
-            console.log(`[${time}] [conn] WebSocket 连接已建立`);
+            logger.log(`[${time}] [conn] WebSocket 连接已建立`);
             break;
         case 'authenticated':
-            console.log(`[${time}] [conn] 认证成功`);
+            logger.log(`[${time}] [conn] 认证成功`);
             break;
         case 'disconnected':
             const duration = record.connectionDuration ? ` (持续 ${record.connectionDuration}s)` : '';
-            console.log(`[${time}] [conn] 连接断开: ${record.reason || '未知'}${duration}`);
+            logger.log(`[${time}] [conn] 连接断开: ${record.reason || '未知'}${duration}`);
             break;
         case 'reconnecting':
-            console.log(`[${time}] [conn] 正在重连 (第 ${record.attempt} 次)`);
+            logger.log(`[${time}] [conn] 正在重连 (第 ${record.attempt} 次)`);
             break;
         case 'error':
-            console.error(`[${time}] [conn] 错误: ${record.errorMessage}`);
+            logger.error(`[${time}] [conn] 错误: ${record.errorMessage}`);
             break;
     }
 }
@@ -199,7 +200,7 @@ export function cleanupOldLogs(daysToKeep = 30) {
         }
     });
     fs.writeFileSync(LOG_FILE, recentLines.join('\n') + '\n', 'utf-8');
-    console.log(`[conn] 已清理 ${lines.length - recentLines.length} 条旧日志`);
+    logger.log(`[conn] 已清理 ${lines.length - recentLines.length} 条旧日志`);
 }
 // 导出日志文件路径
 export function getLogFilePath() {

package/dist/connection-manager.d.ts

@@ -13,14 +13,6 @@
  * - 集成消息总线（用户消息通过 SSE 推送）
  */
 import { WecomClient } from './client.js';
-/**
- * 检查机器人是否被占用（当前进程内）
- */
-export declare function isRobotOccupied(robotName: string): boolean;
-/**
- * 获取占用机器人的智能体名称
- */
-export declare function getRobotOccupiedBy(robotName: string): string | undefined;
 /**
  * 连接到指定机器人
  * 注意：MCP Server 启动时已自动连接所有机器人

package/dist/connection-manager.js

@@ -17,6 +17,7 @@ import * as path from 'path';
 import * as os from 'os';
 import { WecomClient } from './client.js';
 import { listAllRobots } from './config-wizard.js';
+import { logger } from './logger.js';
 // 连接池：robotName → ConnectionState
 const connectionPool = new Map();
 const CONFIG_DIR = path.join(os.homedir(), '.wecom-aibot-mcp');
@@ -68,19 +69,6 @@ function waitForConnection(client, timeoutMs) {
         }, 500);
     });
 }
-/**
- * 检查机器人是否被占用（当前进程内）
- */
-export function isRobotOccupied(robotName) {
-    return connectionPool.has(robotName);
-}
-/**
- * 获取占用机器人的智能体名称
- */
-export function getRobotOccupiedBy(robotName) {
-    const state = connectionPool.get(robotName);
-    return state?.agentName;
-}
 /**
  * 连接到指定机器人
  * 注意：MCP Server 启动时已自动连接所有机器人
@@ -125,7 +113,7 @@ export async function connectRobot(robotName, agentName) {
         agentName,
     };
     connectionPool.set(robot.name, state);
-    console.log(`[connection] 已连接机器人: ${robot.name}`);
+    logger.log(`[connection] 已连接机器人: ${robot.name}`);
     return {
         success: true,
         client,
@@ -139,7 +127,7 @@ export function disconnectRobot(robotName) {
     if (state) {
         state.client.disconnect();
         connectionPool.delete(robotName);
-        console.log(`[connection] 已断开机器人: ${robotName}`);
+        logger.log(`[connection] 已断开机器人: ${robotName}`);
     }
 }
 /**
@@ -157,16 +145,16 @@ export async function getClient(robotName) {
     // 断开了，尝试重连
     const robot = await findRobotConfig(state.robotName);
     if (robot) {
-        console.log(`[connection] 重连机器人: ${robot.name}`);
+        logger.log(`[connection] 重连机器人: ${robot.name}`);
         state.client = new WecomClient(robot.botId, robot.secret, robot.targetUserId, robot.name);
         state.client.connect();
         const connected = await waitForConnection(state.client, 5000);
         if (connected) {
-            console.log(`[connection] 重连成功: ${robot.name}`);
+            logger.log(`[connection] 重连成功: ${robot.name}`);
             return state.client;
         }
         else {
-            console.log(`[connection] 重连失败: ${robot.name}`);
+            logger.log(`[connection] 重连失败: ${robot.name}`);
             return null;
         }
     }
@@ -224,21 +212,21 @@ export function updateAgentName(robotName, agentName) {
 export async function connectAllRobots() {
     const robots = listAllRobots();
     if (robots.length === 0) {
-        console.log('[connection] 未配置任何机器人');
+        logger.log('[connection] 未配置任何机器人');
         return;
     }
     // 多机器人场景：不自动连接，等待用户选择
     if (robots.length > 1) {
-        console.log(`[connection] 检测到 ${robots.length} 个机器人，等待用户选择`);
+        logger.log(`[connection] 检测到 ${robots.length} 个机器人，等待用户选择`);
         return;
     }
     // 单机器人场景：自动连接
-    console.log(`[connection] 自动连接机器人: ${robots[0].name}`);
+    logger.log(`[connection] 自动连接机器人: ${robots[0].name}`);
     const result = await connectRobot(robots[0].name);
     if (result.success) {
-        console.log(`[connection] ✅ ${robots[0].name} 已连接`);
+        logger.log(`[connection] ✅ ${robots[0].name} 已连接`);
     }
     else {
-        console.log(`[connection] ❌ ${robots[0].name} 连接失败: ${result.error}`);
+        logger.log(`[connection] ❌ ${robots[0].name} 连接失败: ${result.error}`);
     }
 }

package/dist/daemon.js

@@ -12,6 +12,7 @@ import * as fs from 'fs';
 import * as path from 'path';
 import * as os from 'os';
 import { WecomClient } from './client.js';
+import { logger } from './logger.js';
 const CONFIG_DIR = path.join(os.homedir(), '.wecom-aibot-mcp');
 const DAEMON_PORT = 18964;
 const PID_FILE = path.join(CONFIG_DIR, 'daemon.pid');
@@ -311,7 +312,7 @@ class ConnectionDaemon {
     log(message) {
         const timestamp = new Date().toISOString();
         const line = `[${timestamp}] ${message}`;
-        console.log(line);
+        logger.log(line);
         fs.appendFileSync(LOG_FILE, line + '\n');
     }
     async shutdown() {
@@ -341,15 +342,15 @@ if (command === '--stop') {
         const pid = parseInt(fs.readFileSync(PID_FILE, 'utf-8'));
         try {
             process.kill(pid, 'SIGTERM');
-            console.log(`守护进程已停止 (PID: ${pid})`);
+            logger.log(`守护进程已停止 (PID: ${pid})`);
         }
         catch {
-            console.log('守护进程未运行');
+            logger.log('守护进程未运行');
         }
         fs.unlinkSync(PID_FILE);
     }
     else {
-        console.log('守护进程未运行');
+        logger.log('守护进程未运行');
     }
 }
 else if (command === '--status') {
@@ -358,10 +359,10 @@ else if (command === '--status') {
         let data = '';
         res.on('data', chunk => data += chunk);
         res.on('end', () => {
-            console.log(JSON.stringify(JSON.parse(data), null, 2));
+            logger.log(JSON.stringify(JSON.parse(data), null, 2));
         });
     }).on('error', () => {
-        console.log('守护进程未运行');
+        logger.log('守护进程未运行');
     });
 }
 else {

package/dist/headless-state.d.ts

@@ -62,18 +62,6 @@ export declare function getAllHeadlessStates(): Array<{
     projectDir: string;
     state: HeadlessState;
 }>;
-/**
- * 检查机器人是否被占用
- *
- * 扫描所有 headless 项目，检查是否有使用该机器人
- */
-export declare function checkRobotOccupied(robotName: string, excludeProjectDir?: string): {
-    occupied: boolean;
-    by?: {
-        projectDir: string;
-        agentName: string;
-    };
-};
 /**
  * 清理所有项目 Hook 配置（服务重启时）
  */

package/dist/headless-state.js

@@ -10,6 +10,7 @@
 import * as fs from 'fs';
 import * as path from 'path';
 import * as os from 'os';
+import { logger } from './logger.js';
 // 配置目录
 const CONFIG_DIR = path.join(os.homedir(), '.wecom-aibot-mcp');
 // 全局索引文件
@@ -77,7 +78,7 @@ export function enterHeadlessMode(projectDir, agentName, robotName) {
         fs.mkdirSync(stateDir, { recursive: true });
     }
     fs.writeFileSync(stateFilePath, JSON.stringify(state, null, 2));
-    console.log(`[headless] 已进入微信模式: ${stateFilePath}`);
+    logger.log(`[headless] 已进入微信模式: ${stateFilePath}`);
     // 2. 添加到全局索引
     const index = readHeadlessIndex();
     if (!index.includes(projectDir)) {
@@ -100,14 +101,14 @@ export function exitHeadlessMode(projectDir) {
     const dir = projectDir || process.cwd();
     const state = loadHeadlessState(dir);
     if (!state) {
-        console.log('[headless] 未在微信模式');
+        logger.log('[headless] 未在微信模式');
         return null;
     }
     // 1. 删除项目状态文件
     const stateFilePath = getProjectHeadlessFile(state.projectDir);
     if (fs.existsSync(stateFilePath)) {
         fs.unlinkSync(stateFilePath);
-        console.log(`[headless] 已删除状态文件: ${stateFilePath}`);
+        logger.log(`[headless] 已删除状态文件: ${stateFilePath}`);
     }
     // 2. 从全局索引移除
     const index = readHeadlessIndex();
@@ -130,7 +131,7 @@ export function setAutoApprove(enabled, projectDir) {
     // 写入状态文件
     const stateFilePath = getProjectHeadlessFile(state.projectDir);
     fs.writeFileSync(stateFilePath, JSON.stringify(state, null, 2));
-    console.log(`[headless] 已${enabled ? '启用' : '禁用'}智能代批`);
+    logger.log(`[headless] 已${enabled ? '启用' : '禁用'}智能代批`);
     return state;
 }
 /**
@@ -147,7 +148,7 @@ export function loadHeadlessState(projectDir) {
         return JSON.parse(content);
     }
     catch (err) {
-        console.error(`[headless] 解析状态文件失败: ${stateFilePath}`, err);
+        logger.error(`[headless] 解析状态文件失败: ${stateFilePath}`, err);
         return null;
     }
 }
@@ -176,7 +177,7 @@ function configureProjectHook(projectDir) {
             settings = JSON.parse(content);
         }
         catch (err) {
-            console.error(`[headless] 读取 settings.json 失败: ${settingsPath}`, err);
+            logger.error(`[headless] 读取 settings.json 失败: ${settingsPath}`, err);
         }
     }
     // 设置 PermissionRequest hook
@@ -197,7 +198,7 @@ function configureProjectHook(projectDir) {
     ];
     // 写入配置
     fs.writeFileSync(settingsPath, JSON.stringify(settings, null, 2));
-    console.log(`[headless] 已配置项目 Hook: ${settingsPath}`);
+    logger.log(`[headless] 已配置项目 Hook: ${settingsPath}`);
 }
 /**
  * 清除项目级 Hook 配置
@@ -217,11 +218,11 @@ function clearProjectHook(projectDir) {
                 delete settings.hooks;
             }
             fs.writeFileSync(settingsPath, JSON.stringify(settings, null, 2));
-            console.log(`[headless] 已清除项目 Hook: ${settingsPath}`);
+            logger.log(`[headless] 已清除项目 Hook: ${settingsPath}`);
         }
     }
     catch (err) {
-        console.error(`[headless] 清除项目 Hook 失败: ${settingsPath}`, err);
+        logger.error(`[headless] 清除项目 Hook 失败: ${settingsPath}`, err);
     }
 }
 /**
@@ -242,7 +243,7 @@ export function cleanupOrphanFiles() {
         else {
             // 状态文件不存在，清理 Hook 配置
             clearProjectHook(projectDir);
-            console.log(`[headless] 清理孤儿项目: ${projectDir}`);
+            logger.log(`[headless] 清理孤儿项目: ${projectDir}`);
         }
     }
     // 更新索引
@@ -267,31 +268,6 @@ export function getAllHeadlessStates() {
     }
     return results;
 }
-/**
- * 检查机器人是否被占用
- *
- * 扫描所有 headless 项目，检查是否有使用该机器人
- */
-export function checkRobotOccupied(robotName, excludeProjectDir) {
-    const allStates = getAllHeadlessStates();
-    for (const { projectDir, state } of allStates) {
-        // 排除当前项目
-        if (excludeProjectDir && projectDir === excludeProjectDir) {
-            continue;
-        }
-        // 检查是否使用同一机器人
-        if (state.robotName === robotName) {
-            return {
-                occupied: true,
-                by: {
-                    projectDir,
-                    agentName: state.agentName || '未知',
-                },
-            };
-        }
-    }
-    return { occupied: false };
-}
 /**
  * 清理所有项目 Hook 配置（服务重启时）
  */

package/dist/http-server.d.ts

@@ -15,7 +15,7 @@
 import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
 export declare const HTTP_PORT = 18963;
 export declare const HOOK_SCRIPT_PATH: string;
-export declare function generateCcId(): string;
+export declare function generateCcId(agentName?: string): string;
 export declare function pushMessageToSession(robotName: string, message: {
     msgid: string;
     content: string;
@@ -24,6 +24,14 @@ export declare function pushMessageToSession(robotName: string, message: {
     chattype: 'single' | 'group';
     timestamp: number;
 }): Promise<void>;
+export declare function pushMessageToSSEClient(robotName: string, message: {
+    msgid: string;
+    content: string;
+    from_userid: string;
+    chatid: string;
+    chattype: 'single' | 'group';
+    timestamp: number;
+}, targetCcId?: string): Promise<void>;
 export interface ApprovalRequest {
     tool_name: string;
     tool_input: Record<string, unknown>;
@@ -34,10 +42,21 @@ export interface ApprovalRequest {
 interface CCRegistryEntry {
     robotName: string;
     agentName?: string;
+    mode?: 'channel' | 'http';
+    projectDir?: string;
+    lastOnline: number;
 }
-export declare function registerCcId(ccId: string, robotName: string, agentName?: string): void;
+export declare function registerCcId(ccId: string, robotName: string, agentName?: string, mode?: 'channel' | 'http', projectDir?: string, isReconnect?: boolean): {
+    success: boolean;
+    ccId: string;
+};
 export declare function unregisterCcId(ccId: string): void;
+export declare function clearCcIdRegistry(): {
+    cleared: number;
+    entries: string[];
+};
 export declare function getRobotByCcId(ccId: string): string | null;
+export declare function getProjectDirByCcId(ccId: string): string | null;
 export declare function getCCRegistryEntry(ccId: string): CCRegistryEntry | null;
 export declare function getCCCount(): number;
 export declare function getCCCountByRobot(robotName: string): number;