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

package/dist/index.d.ts

@@ -9,7 +9,7 @@
  * - robotName 作为连接索引
  */
 export { WecomClient, initClient } from './client.js';
-export { connectRobot, disconnectRobot, getClient, getConnectionState, getAllConnectionStates, isRobotOccupied, getRobotOccupiedBy, } from './connection-manager.js';
+export { connectRobot, disconnectRobot, getClient, getConnectionState, getAllConnectionStates, } from './connection-manager.js';
 export { startHttpServer, stopHttpServer, HTTP_PORT, HOOK_SCRIPT_PATH, } from './http-server.js';
 export type { ApprovalRequest } from './http-server.js';
 export { registerTools } from './tools/index.js';

package/dist/index.js

@@ -11,7 +11,7 @@
 // Client 模块
 export { WecomClient, initClient } from './client.js';
 // 连接管理模块
-export { connectRobot, disconnectRobot, getClient, getConnectionState, getAllConnectionStates, isRobotOccupied, getRobotOccupiedBy, } from './connection-manager.js';
+export { connectRobot, disconnectRobot, getClient, getConnectionState, getAllConnectionStates, } from './connection-manager.js';
 // HTTP 服务模块
 export { startHttpServer, stopHttpServer, HTTP_PORT, HOOK_SCRIPT_PATH, } from './http-server.js';
 // 工具注册

package/dist/keepalive-monitor.js

@@ -8,6 +8,7 @@
  * - 不再使用 projectDir
  */
 import { getClient, getConnectionState } from './connection-manager.js';
+import { logger } from './logger.js';
 const KEEPALIVE_INTERVAL_MINUTES = 5; // 每 5 分钟
 const CHECK_INTERVAL_MS = 60000; // 每分钟检查一次
 const MAX_KEEPALIVE_COUNT = 2; // 保活提醒最多 2 次
@@ -17,7 +18,7 @@ let monitorTimer = null;
  */
 export function startKeepaliveMonitor() {
     monitorTimer = setInterval(checkAndSendKeepalive, CHECK_INTERVAL_MS);
-    console.log('[keepalive] 保活监控已启动 (每 5 分钟发送审批提醒)');
+    logger.log('[keepalive] 保活监控已启动 (每 5 分钟发送审批提醒)');
 }
 /**
  * 停止保活监控
@@ -27,7 +28,7 @@ export function stopKeepaliveMonitor() {
         clearInterval(monitorTimer);
         monitorTimer = null;
     }
-    console.log('[keepalive] 保活监控已停止');
+    logger.log('[keepalive] 保活监控已停止');
 }
 /**
  * 检查并发送保活消息
@@ -75,12 +76,12 @@ async function sendKeepaliveMessage(approval, minutes, robotName) {
         if (client) {
             const sent = await client.sendText(message);
             if (sent) {
-                console.log(`[keepalive] 已发送审批提醒: ${approval.taskId}, 等待 ${minutes} 分钟`);
+                logger.log(`[keepalive] 已发送审批提醒: ${approval.taskId}, 等待 ${minutes} 分钟`);
             }
         }
     }
     catch (err) {
-        console.error(`[keepalive] 发送审批提醒失败:`, err);
+        logger.error(`[keepalive] 发送审批提醒失败:`, err);
     }
 }
 /**

package/dist/logger.d.ts

@@ -0,0 +1,51 @@
+/**
+ * 统一日志模块
+ *
+ * 通过 --debug 参数控制日志输出：
+ * - debug 模式：所有日志输出到 stderr
+ * - 正常模式：只输出必要信息（如错误、启动消息）
+ *
+ * Debug 标记文件：~/.wecom-aibot-mcp/debug
+ *
+ * 用法与 console 完全一致：
+ * - logger.log() - debug 模式才输出
+ * - logger.error() - 始终输出到 stderr
+ * - logger.warn() - 始终输出到 stderr
+ * - logger.info() - 始终输出到 stdout
+ */
+/**
+ * 检查是否处于 debug 模式
+ */
+export declare function isDebugMode(): boolean;
+/**
+ * 设置 debug 模式
+ */
+export declare function setDebugMode(enabled: boolean): void;
+/**
+ * 日志输出（仅 debug 模式）
+ * 与 console.log 用法完全一致
+ */
+export declare function log(...args: unknown[]): void;
+/**
+ * 错误日志（始终输出到 stderr）
+ * 与 console.error 用法完全一致
+ */
+export declare function error(...args: unknown[]): void;
+/**
+ * 信息日志（始终输出到 stdout）
+ * 与 console.info 用法完全一致
+ */
+export declare function info(...args: unknown[]): void;
+/**
+ * 警告日志（始终输出到 stderr）
+ * 与 console.warn 用法完全一致
+ */
+export declare function warn(...args: unknown[]): void;
+export declare const logger: {
+    log: typeof log;
+    error: typeof error;
+    info: typeof info;
+    warn: typeof warn;
+    isDebug: typeof isDebugMode;
+    setDebug: typeof setDebugMode;
+};

package/dist/logger.js

@@ -0,0 +1,84 @@
+/**
+ * 统一日志模块
+ *
+ * 通过 --debug 参数控制日志输出：
+ * - debug 模式：所有日志输出到 stderr
+ * - 正常模式：只输出必要信息（如错误、启动消息）
+ *
+ * Debug 标记文件：~/.wecom-aibot-mcp/debug
+ *
+ * 用法与 console 完全一致：
+ * - logger.log() - debug 模式才输出
+ * - logger.error() - 始终输出到 stderr
+ * - logger.warn() - 始终输出到 stderr
+ * - logger.info() - 始终输出到 stdout
+ */
+import * as fs from 'fs';
+import * as path from 'path';
+import * as os from 'os';
+const DEBUG_FILE = path.join(os.homedir(), '.wecom-aibot-mcp', 'debug');
+let debugMode = false;
+/**
+ * 检查是否处于 debug 模式
+ */
+export function isDebugMode() {
+    if (debugMode)
+        return true;
+    // 检查标记文件
+    if (fs.existsSync(DEBUG_FILE)) {
+        debugMode = true;
+        return true;
+    }
+    return false;
+}
+/**
+ * 设置 debug 模式
+ */
+export function setDebugMode(enabled) {
+    debugMode = enabled;
+    if (enabled && !fs.existsSync(DEBUG_FILE)) {
+        fs.writeFileSync(DEBUG_FILE, 'true');
+    }
+    else if (!enabled && fs.existsSync(DEBUG_FILE)) {
+        fs.unlinkSync(DEBUG_FILE);
+    }
+}
+/**
+ * 日志输出（仅 debug 模式）
+ * 与 console.log 用法完全一致
+ */
+export function log(...args) {
+    if (isDebugMode()) {
+        console.log(...args);
+    }
+}
+/**
+ * 错误日志（始终输出到 stderr）
+ * 与 console.error 用法完全一致
+ */
+export function error(...args) {
+    console.error(...args);
+}
+/**
+ * 信息日志（始终输出到 stdout）
+ * 与 console.info 用法完全一致
+ */
+export function info(...args) {
+    console.info(...args);
+}
+/**
+ * 警告日志（始终输出到 stderr）
+ * 与 console.warn 用法完全一致
+ */
+export function warn(...args) {
+    console.warn(...args);
+}
+// 导出 logger 对象，用法与 console 完全一致
+export const logger = {
+    log,
+    error,
+    info,
+    warn,
+    isDebug: isDebugMode,
+    setDebug: setDebugMode,
+};

package/dist/message-bus.d.ts

@@ -5,6 +5,12 @@
  * - WecomClient 收到微信消息 → 发布到 bus
  * - MCP Server 订阅消息 → SSE 推送给 CC
  * - 按 CCID 过滤消息（基于引用内容）
+ *
+ * v2.1 使用订阅计数替代 tap 计数：
+ * - subscribe 时计数 +1
+ * - unsubscribe 时计数 -1（自动处理掉线）
+ * - 订阅数 = 1 时直接发给唯一订阅者
+ * - 订阅数 > 1 时检查 ccId
  */
 import { Subject } from 'rxjs';
 export interface WecomMessage {
@@ -17,6 +23,7 @@ export interface WecomMessage {
     timestamp: number;
     quoteContent?: string;
 }
+export declare function getSubscriberCount(robotName: string): number;
 declare const wecomMessage$: Subject<WecomMessage>;
 /**
  * 发布微信消息（由 WecomClient 调用）
@@ -27,7 +34,12 @@ export declare function publishWecomMessage(msg: WecomMessage): void;
  */
 export declare function subscribeWecomMessage(callback: (msg: WecomMessage) => void): import("rxjs").Subscription;
 /**
- * 订阅特定机器人的微信消息
+ * 订阅特定机器人的微信消息（带订阅计数）
  */
 export declare function subscribeWecomMessageByRobot(robotName: string, callback: (msg: WecomMessage) => void): import("rxjs").Subscription;
+/**
+ * 订阅特定机器人且匹配 ccId 的消息
+ * 用于多 CC 共用一个机器人场景的过滤
+ */
+export declare function subscribeWecomMessageByCcId(robotName: string, ccId: string, callback: (msg: WecomMessage) => void): import("rxjs").Subscription;
 export { wecomMessage$ };

package/dist/message-bus.js

@@ -5,9 +5,32 @@
  * - WecomClient 收到微信消息 → 发布到 bus
  * - MCP Server 订阅消息 → SSE 推送给 CC
  * - 按 CCID 过滤消息（基于引用内容）
+ *
+ * v2.1 使用订阅计数替代 tap 计数：
+ * - subscribe 时计数 +1
+ * - unsubscribe 时计数 -1（自动处理掉线）
+ * - 订阅数 = 1 时直接发给唯一订阅者
+ * - 订阅数 > 1 时检查 ccId
  */
 import { Subject } from 'rxjs';
-import { filter } from 'rxjs/operators';
+import { filter, finalize } from 'rxjs/operators';
+// ============================================
+// 订阅计数器（替代 tap 计数）
+// ============================================
+const subscriberCount = new Map(); // robotName → 订阅数
+export function getSubscriberCount(robotName) {
+    return subscriberCount.get(robotName) || 0;
+}
+function incrementSubscriberCount(robotName) {
+    const current = subscriberCount.get(robotName) || 0;
+    subscriberCount.set(robotName, current + 1);
+}
+function decrementSubscriberCount(robotName) {
+    const current = subscriberCount.get(robotName) || 0;
+    if (current > 0) {
+        subscriberCount.set(robotName, current - 1);
+    }
+}
 // ============================================
 // 消息总线（RxJS Subject）
 // ============================================
@@ -28,10 +51,40 @@ export function subscribeWecomMessage(callback) {
     return wecomMessage$.subscribe(callback);
 }
 /**
- * 订阅特定机器人的微信消息
+ * 订阅特定机器人的微信消息（带订阅计数）
  */
 export function subscribeWecomMessageByRobot(robotName, callback) {
-    return wecomMessage$.pipe(filter(msg => msg.robotName === robotName)).subscribe(callback);
+    incrementSubscriberCount(robotName);
+    return wecomMessage$.pipe(filter(msg => msg.robotName === robotName), finalize(() => decrementSubscriberCount(robotName)) // 无论何时结束都减计数
+    ).subscribe(callback);
+}
+/**
+ * 订阅特定机器人且匹配 ccId 的消息
+ * 用于多 CC 共用一个机器人场景的过滤
+ */
+export function subscribeWecomMessageByCcId(robotName, ccId, callback) {
+    incrementSubscriberCount(robotName);
+    return wecomMessage$.pipe(filter(msg => msg.robotName === robotName), filter(msg => isMessageForCcId(msg, ccId)), finalize(() => decrementSubscriberCount(robotName)) // 无论何时结束都减计数
+    ).subscribe(callback);
+}
+/**
+ * 检查消息是否属于指定的 ccId
+ */
+function isMessageForCcId(msg, ccId) {
+    // 有引用内容，检查是否引用该 ccId
+    if (msg.quoteContent) {
+        const match = msg.quoteContent.match(/【([^】]+)】/);
+        if (match && match[1] === ccId) {
+            return true; // 引用了该 ccId，消息属于该 CC
+        }
+        // 引用了其他 ccId，不属于当前 ccId
+        if (match) {
+            return false;
+        }
+    }
+    // 无引用内容，消息不属于任何特定 CC
+    // 在多 CC 场景下，需要用户指定
+    return false;
 }
 // 导出 Observable（供高级用法）
 export { wecomMessage$ };

package/dist/project-config.d.ts

@@ -19,6 +19,7 @@ export interface WechatModeConfig {
     ccId?: string;
     autoApprove?: boolean;
     autoApproveTimeout?: number;
+    heartbeatJobId?: string;
 }
 /**
  * 获取项目配置文件路径
@@ -128,3 +129,18 @@ export declare function removePermissionHook(projectDir: string): {
     path: string;
     existed: boolean;
 };
+/**
+ * 添加 TaskCompleted hook 到项目 settings.json
+ */
+export declare function addTaskCompletedHook(projectDir: string): {
+    success: boolean;
+    path: string;
+};
+/**
+ * 删除 TaskCompleted hook 从项目 settings.json
+ */
+export declare function removeTaskCompletedHook(projectDir: string): {
+    success: boolean;
+    path: string;
+    existed: boolean;
+};

package/dist/project-config.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 CLAUDE_CONFIG_FILE = path.join(os.homedir(), '.claude.json');
 const PROJECT_CONFIG_SUBDIR = '.claude/wecom-aibot';
@@ -37,7 +38,7 @@ export function loadProjectConfig(projectDir) {
         const config = JSON.parse(content);
         // 验证必需字段
         if (!config.botId || !config.secret || !config.defaultUser) {
-            console.error(`[project-config] 项目配置缺少必需字段: ${configPath}`);
+            logger.error(`[project-config] 项目配置缺少必需字段: ${configPath}`);
             return null;
         }
         return {
@@ -48,7 +49,7 @@ export function loadProjectConfig(projectDir) {
         };
     }
     catch (err) {
-        console.error(`[project-config] 解析项目配置失败: ${configPath}`, err);
+        logger.error(`[project-config] 解析项目配置失败: ${configPath}`, err);
         return null;
     }
 }
@@ -66,7 +67,7 @@ export function saveProjectConfig(projectDir, config) {
         fs.mkdirSync(configDir, { recursive: true });
     }
     fs.writeFileSync(configPath, JSON.stringify(config, null, 2));
-    console.log(`[project-config] 已保存项目配置: ${configPath}`);
+    logger.log(`[project-config] 已保存项目配置: ${configPath}`);
 }
 /**
  * 删除项目配置
@@ -77,7 +78,7 @@ export function deleteProjectConfig(projectDir) {
     const configPath = getProjectConfigPath(projectDir);
     if (fs.existsSync(configPath)) {
         fs.unlinkSync(configPath);
-        console.log(`[project-config] 已删除项目配置: ${configPath}`);
+        logger.log(`[project-config] 已删除项目配置: ${configPath}`);
     }
 }
 /**
@@ -129,7 +130,7 @@ export function updateWechatModeConfig(projectDir, updates) {
     // 合并更新
     const newConfig = { ...config, ...updates };
     fs.writeFileSync(configPath, JSON.stringify(newConfig, null, 2));
-    console.log(`[project-config] 已更新微信模式配置: ${configPath}`);
+    logger.log(`[project-config] 已更新微信模式配置: ${configPath}`);
 }
 /**
  * 加载全局配置（从 ~/.claude.json）
@@ -158,7 +159,7 @@ export function loadGlobalConfig() {
         };
     }
     catch (err) {
-        console.error('[project-config] 加载全局配置失败:', err);
+        logger.error('[project-config] 加载全局配置失败:', err);
         return null;
     }
 }
@@ -177,14 +178,14 @@ export function getConfig(projectDir) {
     if (projectDir) {
         const projectConfig = loadProjectConfig(projectDir);
         if (projectConfig) {
-            console.log(`[project-config] 使用项目配置: ${projectDir}`);
+            logger.log(`[project-config] 使用项目配置: ${projectDir}`);
             return projectConfig;
         }
     }
     // 2. 回退全局配置
     const globalConfig = loadGlobalConfig();
     if (globalConfig) {
-        console.log('[project-config] 使用全局配置');
+        logger.log('[project-config] 使用全局配置');
         return globalConfig;
     }
     return null;
@@ -255,6 +256,17 @@ const PERMISSION_HOOK = {
     matcher: '',
     hooks: [{ type: 'command', command: '~/.wecom-aibot-mcp/permission-hook.sh' }],
 };
+/**
+ * TaskCompleted hook 脚本路径
+ */
+const TASK_COMPLETED_HOOK_SCRIPT_PATH = path.join(os.homedir(), '.wecom-aibot-mcp', 'task-completed-hook.sh');
+/**
+ * TaskCompleted hook 配置
+ */
+const TASK_COMPLETED_HOOK = {
+    matcher: '',
+    hooks: [{ type: 'command', command: TASK_COMPLETED_HOOK_SCRIPT_PATH }],
+};
 /**
  * 添加 PermissionRequest hook 到项目 settings.json
  */
@@ -284,11 +296,11 @@ export function addPermissionHook(projectDir) {
     // 写入配置
     try {
         fs.writeFileSync(settingsPath, JSON.stringify(settings, null, 2));
-        console.log(`[project-config] 已添加 PermissionRequest hook: ${settingsPath}`);
+        logger.log(`[project-config] 已添加 PermissionRequest hook: ${settingsPath}`);
         return { success: true, path: settingsPath };
     }
     catch (err) {
-        console.error(`[project-config] 添加 PermissionRequest hook 失败: ${err}`);
+        logger.error(`[project-config] 添加 PermissionRequest hook 失败: ${err}`);
         return { success: false, path: settingsPath };
     }
 }
@@ -312,13 +324,80 @@ export function removePermissionHook(projectDir) {
             }
             // 写入配置
             fs.writeFileSync(settingsPath, JSON.stringify(settings, null, 2));
-            console.log(`[project-config] 已删除 PermissionRequest hook: ${settingsPath}`);
+            logger.log(`[project-config] 已删除 PermissionRequest hook: ${settingsPath}`);
+            return { success: true, path: settingsPath, existed: true };
+        }
+        return { success: true, path: settingsPath, existed: false };
+    }
+    catch (err) {
+        logger.error(`[project-config] 删除 PermissionRequest hook 失败: ${err}`);
+        return { success: false, path: settingsPath, existed: false };
+    }
+}
+/**
+ * 添加 TaskCompleted hook 到项目 settings.json
+ */
+export function addTaskCompletedHook(projectDir) {
+    const settingsPath = getProjectSettingsPath(projectDir);
+    const settingsDir = path.dirname(settingsPath);
+    // 确保目录存在
+    if (!fs.existsSync(settingsDir)) {
+        fs.mkdirSync(settingsDir, { recursive: true });
+    }
+    // 读取现有配置
+    let settings = {};
+    if (fs.existsSync(settingsPath)) {
+        try {
+            const content = fs.readFileSync(settingsPath, 'utf-8');
+            settings = JSON.parse(content);
+        }
+        catch {
+            // ignore
+        }
+    }
+    // 添加 hooks.TaskCompleted
+    if (!settings.hooks) {
+        settings.hooks = {};
+    }
+    settings.hooks.TaskCompleted = [TASK_COMPLETED_HOOK];
+    // 写入配置
+    try {
+        fs.writeFileSync(settingsPath, JSON.stringify(settings, null, 2));
+        logger.log(`[project-config] 已添加 TaskCompleted hook: ${settingsPath}`);
+        return { success: true, path: settingsPath };
+    }
+    catch (err) {
+        logger.error(`[project-config] 添加 TaskCompleted hook 失败: ${err}`);
+        return { success: false, path: settingsPath };
+    }
+}
+/**
+ * 删除 TaskCompleted hook 从项目 settings.json
+ */
+export function removeTaskCompletedHook(projectDir) {
+    const settingsPath = getProjectSettingsPath(projectDir);
+    if (!fs.existsSync(settingsPath)) {
+        return { success: true, path: settingsPath, existed: false };
+    }
+    try {
+        const content = fs.readFileSync(settingsPath, 'utf-8');
+        const settings = JSON.parse(content);
+        // 删除 hooks.TaskCompleted
+        if (settings.hooks && settings.hooks.TaskCompleted) {
+            delete settings.hooks.TaskCompleted;
+            // 如果 hooks 为空，删除整个 hooks 字段
+            if (Object.keys(settings.hooks).length === 0) {
+                delete settings.hooks;
+            }
+            // 写入配置
+            fs.writeFileSync(settingsPath, JSON.stringify(settings, null, 2));
+            logger.log(`[project-config] 已删除 TaskCompleted hook: ${settingsPath}`);
             return { success: true, path: settingsPath, existed: true };
         }
         return { success: true, path: settingsPath, existed: false };
     }
     catch (err) {
-        console.error(`[project-config] 删除 PermissionRequest hook 失败: ${err}`);
+        logger.error(`[project-config] 删除 TaskCompleted hook 失败: ${err}`);
         return { success: false, path: settingsPath, existed: false };
     }
 }