@vrs-soft/wecom-aibot-mcp 1.0.7 → 1.0.9

package/dist/bin.d.ts

@@ -7,5 +7,10 @@
  * 支持 HTTP Transport 模式：
  * - 固定端口 18963
  * - 支持多 Claude Code 同时连接
+ *
+ * 连接管理：
+ * - 启动时不建立 WebSocket 连接
+ * - enter_headless_mode 时按需建立连接
+ * - exit_headless_mode 时断开连接
  */
 export {};

package/dist/bin.js

@@ -7,30 +7,26 @@
  * 支持 HTTP Transport 模式：
  * - 固定端口 18963
  * - 支持多 Claude Code 同时连接
+ *
+ * 连接管理：
+ * - 启动时不建立 WebSocket 连接
+ * - enter_headless_mode 时按需建立连接
+ * - exit_headless_mode 时断开连接
  */
-import { runConfigWizard, loadConfig, saveConfig, deleteConfig, deleteMcpConfigInteractive, uninstall, addMcpConfig, detectUserIdFromMessage, ensureHookInstalled, } from './config-wizard.js';
+import { spawn } from 'child_process';
+import * as fs from 'fs';
+import * as path from 'path';
+import * as os from 'os';
+import { runConfigWizard, loadConfig, saveConfig, deleteConfig, deleteMcpConfigInteractive, uninstall, addMcpConfig, detectUserIdFromMessage, ensureHookInstalled, listAllRobots, } from './config-wizard.js';
 import { initClient } from './client.js';
 import { registerTools } from './tools/index.js';
-import { startHttpServer, HTTP_PORT } from './http-server.js';
+import { startHttpServer, stopHttpServer, HTTP_PORT } from './http-server.js';
 import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
-import { clearAllProjectHooks } from './headless-state.js';
-const VERSION = '1.0.6';
-// 等待连接验证（最多等待 10 秒）
-async function waitForConnection(client, timeoutMs = 10000) {
-    return new Promise((resolve) => {
-        const startTime = Date.now();
-        const checkInterval = setInterval(() => {
-            if (client.isConnected()) {
-                clearInterval(checkInterval);
-                resolve(true);
-            }
-            else if (Date.now() - startTime > timeoutMs) {
-                clearInterval(checkInterval);
-                resolve(false);
-            }
-        }, 500);
-    });
-}
+import { clearAllProjectHooks, getAllHeadlessStates } from './headless-state.js';
+import { loadStats, cleanupOldLogs } from './connection-log.js';
+import { startKeepaliveMonitor, stopKeepaliveMonitor } from './keepalive-monitor.js';
+const VERSION = '1.0.9';
+const PID_FILE = path.join(os.homedir(), '.wecom-aibot-mcp', 'server.pid');
 function showHelp() {
     console.log(`
 企业微信智能机器人 MCP 服务 v${VERSION}
@@ -44,12 +40,27 @@ function showHelp() {
 选项:
   --help, -h      显示帮助信息
   --version, -v   显示版本号
+  --start         启动 MCP Server（后台服务模式）
+  --stop          停止 MCP Server
+  --status        显示服务状态和机器人配置
   --config        重新配置默认机器人（修改 Bot ID / Secret / 目标用户）
   --add           添加新的机器人配置（多机器人场景）
+  --list          列出所有已配置的机器人及其占用状态
   --delete [名称] 删除指定的机器人配置（无参数则显示列表选择）
-  --status        显示当前配置状态
   --uninstall     卸载并删除所有配置（包括 MCP 配置、hook、skill）
 
+使用流程:
+  1. 首次安装: npx @vrs-soft/wecom-aibot-mcp
+     （进入配置向导，完成后自动后台启动服务）
+
+  2. 已有配置: npx @vrs-soft/wecom-aibot-mcp
+     （显示状态，提示使用 --start 启动）
+
+  3. 启动服务: npx @vrs-soft/wecom-aibot-mcp --start
+     （后台启动 MCP HTTP Server）
+
+  4. 停止服务: npx @vrs-soft/wecom-aibot-mcp --stop
+
 MCP 配置（HTTP Transport）:
 
   编辑 ~/.claude.json：
@@ -58,7 +69,7 @@ MCP 配置（HTTP Transport）:
     "mcpServers": {
       "wecom-aibot": {
         "type": "http",
-        "url": "http://127.0.0.1:${HTTP_PORT}/mcp"
+        "url": "http://127.0.0.1:18963/mcp"
       }
     }
   }
@@ -80,16 +91,179 @@ function showVersion() {
     console.log(`wecom-aibot-mcp v${VERSION}`);
 }
 function showStatus() {
-    const config = loadConfig();
-    if (config) {
-        console.log('当前配置:');
-        console.log(`  Bot ID:     ${config.botId}`);
-        console.log(`  Secret:     ${config.secret.slice(0, 8)}...${config.secret.slice(-4)}`);
-        console.log(`  目标用户:   ${config.targetUserId}`);
+    const allRobots = listAllRobots();
+    const headlessStates = getAllHeadlessStates();
+    // 检查服务是否运行
+    const serverRunning = isServerRunning();
+    console.log(`\n服务状态: ${serverRunning ? '✅ 运行中' : '❌ 未启动'}\n`);
+    if (allRobots.length === 0) {
+        console.log('尚未配置机器人，请运行 npx @vrs-soft/wecom-aibot-mcp 启动配置向导');
+        return;
     }
-    else {
-        console.log('尚未配置，请运行 npx @vrs-soft/wecom-aibot-mcp 启动配置向导');
+    // 构建机器人占用信息
+    const robotUsage = new Map();
+    for (const { state } of headlessStates) {
+        if (state.robotName) {
+            robotUsage.set(state.robotName, {
+                agentName: state.agentName || '未知',
+                projectDir: state.projectDir,
+            });
+        }
+    }
+    console.log(`已配置 ${allRobots.length} 个机器人:\n`);
+    for (const robot of allRobots) {
+        const defaultTag = robot.isDefault ? ' (默认)' : '';
+        const usage = robotUsage.get(robot.name);
+        const statusTag = usage ? ` [使用中]` : '';
+        console.log(`  ${robot.name}${defaultTag}${statusTag}`);
+        console.log(`    Bot ID:     ${robot.botId}`);
+        console.log(`    目标用户:   ${robot.targetUserId}`);
+        if (usage) {
+            console.log(`    使用者:     ${usage.agentName} (${usage.projectDir})`);
+        }
+        console.log('');
+    }
+}
+// 检查服务是否运行
+function isServerRunning() {
+    if (!fs.existsSync(PID_FILE)) {
+        return false;
+    }
+    try {
+        const pid = parseInt(fs.readFileSync(PID_FILE, 'utf-8').trim());
+        // 检查进程是否存在
+        process.kill(pid, 0);
+        return true;
+    }
+    catch {
+        // 进程不存在，清理 PID 文件
+        fs.unlinkSync(PID_FILE);
+        return false;
+    }
+}
+// 停止服务
+function stopServer() {
+    if (!fs.existsSync(PID_FILE)) {
+        console.log('[mcp] 服务未运行');
+        return false;
+    }
+    try {
+        const pid = parseInt(fs.readFileSync(PID_FILE, 'utf-8').trim());
+        process.kill(pid, 'SIGTERM');
+        // 等待进程退出
+        let attempts = 0;
+        while (attempts < 10) {
+            try {
+                process.kill(pid, 0);
+                // 进程还存在，等待
+                setTimeout(() => { }, 500);
+                attempts++;
+            }
+            catch {
+                // 进程已退出
+                break;
+            }
+        }
+        fs.unlinkSync(PID_FILE);
+        console.log('[mcp] 服务已停止');
+        return true;
+    }
+    catch (err) {
+        console.error('[mcp] 停止服务失败:', err);
+        fs.unlinkSync(PID_FILE);
+        return false;
+    }
+}
+// 等待连接验证（用于配置向导验证凭证）
+async function waitForConnection(client, timeoutMs = 10000) {
+    return new Promise((resolve) => {
+        const startTime = Date.now();
+        const checkInterval = setInterval(() => {
+            if (client.isConnected()) {
+                clearInterval(checkInterval);
+                resolve(true);
+            }
+            else if (Date.now() - startTime > timeoutMs) {
+                clearInterval(checkInterval);
+                resolve(false);
+            }
+        }, 500);
+    });
+}
+// 启动 MCP Server（前台运行，供 --start 使用）
+async function startMcpServerForeground() {
+    const savedConfig = loadConfig();
+    if (!savedConfig || !savedConfig.botId || !savedConfig.secret || !savedConfig.targetUserId) {
+        console.error('[mcp] 未找到配置，请先运行: npx @vrs-soft/wecom-aibot-mcp');
+        process.exit(1);
+    }
+    // 写入 PID 文件
+    fs.writeFileSync(PID_FILE, String(process.pid));
+    // 确保 hook 已安装
+    ensureHookInstalled();
+    // 清理残留的 headless 状态
+    clearAllProjectHooks();
+    // 加载统计并清理旧日志
+    loadStats();
+    cleanupOldLogs(1 / 24);
+    // 创建 MCP Server
+    const server = new McpServer({
+        name: 'wecom-aibot-mcp',
+        version: VERSION,
+    });
+    registerTools(server);
+    // 启动 HTTP 服务
+    console.log('');
+    console.log('  ╔════════════════════════════════════════════════════════╗');
+    console.log(`  ║   企业微信智能机器人 MCP 服务 v${VERSION}                   ║`);
+    console.log('  ║   Claude Code 审批通道                                 ║');
+    console.log('  ╚════════════════════════════════════════════════════════╝');
+    console.log('');
+    console.log(`[mcp] 启动 MCP HTTP Server (端口: ${HTTP_PORT})...`);
+    await startHttpServer(server);
+    startKeepaliveMonitor();
+    console.log(`[mcp] MCP Server 已就绪`);
+    console.log(`[mcp] HTTP endpoint: http://127.0.0.1:${HTTP_PORT}/mcp`);
+    console.log(`[mcp] 健康检查: http://127.0.0.1:${HTTP_PORT}/health`);
+    console.log(`[mcp] 微信模式：enter_headless_mode 时建立连接`);
+    console.log(`[mcp] PID: ${process.pid}`);
+    // 退出处理
+    const gracefulShutdown = () => {
+        console.log('[mcp] 正在关闭...');
+        stopKeepaliveMonitor();
+        stopHttpServer();
+        if (fs.existsSync(PID_FILE)) {
+            fs.unlinkSync(PID_FILE);
+        }
+        process.exit(0);
+    };
+    process.on('SIGINT', gracefulShutdown);
+    process.on('SIGTERM', gracefulShutdown);
+}
+// 后台启动 MCP Server（使用 spawn）
+function startMcpServerBackground() {
+    // 检查配置是否存在
+    const savedConfig = loadConfig();
+    if (!savedConfig || !savedConfig.botId || !savedConfig.secret || !savedConfig.targetUserId) {
+        console.error('[mcp] 未找到配置，请先运行: npx @vrs-soft/wecom-aibot-mcp');
+        process.exit(1);
+    }
+    // 检查是否已运行
+    if (isServerRunning()) {
+        console.log('[mcp] 服务已在运行中');
+        return;
     }
+    const nodePath = process.execPath;
+    const scriptPath = process.argv[1];
+    const child = spawn(nodePath, [scriptPath, '--start', '--foreground'], {
+        detached: true,
+        stdio: 'ignore',
+    });
+    child.unref();
+    console.log('[mcp] MCP Server 已在后台启动');
+    console.log(`[mcp] HTTP endpoint: http://127.0.0.1:18963/mcp`);
+    console.log('[mcp] 健康检查: curl http://127.0.0.1:18963/health');
+    console.log('[mcp] 停止服务: npx @vrs-soft/wecom-aibot-mcp --stop');
 }
 async function main() {
     const args = process.argv.slice(2);
@@ -102,11 +276,21 @@ async function main() {
         showVersion();
         process.exit(0);
     }
-    if (args.includes('--status')) {
+    if (args.includes('--status') || args.includes('--list')) {
         showStatus();
         process.exit(0);
     }
+    // --stop 命令：停止服务
+    if (args.includes('--stop')) {
+        stopServer();
+        process.exit(0);
+    }
+    // --uninstall 命令：先停止服务再卸载
     if (args.includes('--uninstall')) {
+        if (isServerRunning()) {
+            console.log('[mcp] 正在停止服务...');
+            stopServer();
+        }
         uninstall();
         process.exit(0);
     }
@@ -121,6 +305,16 @@ async function main() {
         await deleteMcpConfigInteractive(instanceName);
         process.exit(0);
     }
+    // --start --foreground：前台启动（内部调用）
+    if (args.includes('--start') && args.includes('--foreground')) {
+        await startMcpServerForeground();
+        return; // 保持运行，不 exit
+    }
+    // --start：后台启动
+    if (args.includes('--start')) {
+        startMcpServerBackground();
+        process.exit(0);
+    }
     const reconfig = args.includes('--config');
     const isInteractive = process.stdin.isTTY; // 是否为用户交互模式
     console.log('');
@@ -129,6 +323,9 @@ async function main() {
     console.log('  ║   Claude Code 审批通道                                 ║');
     console.log('  ╚════════════════════════════════════════════════════════╝');
     console.log('');
+    // 加载统计并清理旧日志（保留 1 小时）
+    loadStats();
+    cleanupOldLogs(1 / 24);
     // 获取或初始化配置
     let config;
     let ranWizard = false; // 是否运行了配置向导
@@ -165,35 +362,33 @@ async function main() {
     ensureHookInstalled();
     // 清理残留的 headless 状态和 Hook 配置
     clearAllProjectHooks();
-    // 初始化 WebSocket 客户端
-    console.log(`[mcp] 初始化企业微信客户端...`);
-    const wecomClient = initClient(config.botId, config.secret, config.targetUserId || 'placeholder');
-    // 等待连接验证
-    console.log('[mcp] 等待连接验证...');
-    const connected = await waitForConnection(wecomClient, 10000);
-    if (!connected) {
-        console.log('[mcp] 连接失败，可能是配置错误或机器人未授权');
-        console.log('[mcp] 请检查上面的错误提示，修复后重新配置');
-        // 删除无效配置，让用户重新输入
-        deleteConfig();
-        if (isInteractive) {
+    // 配置向导模式：验证连接并识别用户 ID
+    if (isInteractive && (ranWizard || reconfig)) {
+        console.log('[mcp] 验证机器人连接...');
+        // 临时建立连接验证凭证
+        const tempClient = initClient(config.botId, config.secret, config.targetUserId || 'placeholder');
+        const connected = await waitForConnection(tempClient, 10000);
+        if (!connected) {
+            console.log('[mcp] 连接失败，可能是配置错误或机器人未授权');
+            console.log('[mcp] 请检查上面的错误提示，修复后重新配置');
+            // 删除无效配置，让用户重新输入
+            deleteConfig();
             console.log('\n请检查：');
             console.log('  1. Bot ID 和 Secret 是否正确');
             console.log('  2. 新建机器人需等待约 2 分钟同步');
             console.log('  3. 是否已完成授权（机器人详情 → 可使用权限 → 授权）');
             console.log('\n修复后重新运行: npx @vrs-soft/wecom-aibot-mcp --config');
+            tempClient.disconnect();
+            process.exit(1);
         }
-        process.exit(1);
-    }
-    // 连接成功
-    if (isInteractive && (ranWizard || reconfig)) {
-        // 需要通过消息自动识别用户 ID
+        // 连接成功
         console.log('\n[mcp] ✅ 机器人连接成功！');
         // 提示用户发送消息来识别用户 ID
-        const userId = await detectUserIdFromMessage(wecomClient, 60);
+        const userId = await detectUserIdFromMessage(tempClient, 180);
         if (!userId) {
             console.log('\n[mcp] 未能在规定时间内识别用户 ID');
             console.log('[mcp] 请重新运行配置：npx @vrs-soft/wecom-aibot-mcp --config');
+            tempClient.disconnect();
             process.exit(1);
         }
         // 更新配置中的用户 ID
@@ -202,36 +397,18 @@ async function main() {
         saveConfig(config, instanceName);
         console.log('\n[mcp] ✅ 配置完成！');
         console.log(`[mcp] 用户 ID: ${userId}`);
+        // 配置完成后断开连接
+        tempClient.disconnect();
+        // 首次安装后自动后台启动服务
+        console.log('\n[mcp] 正在后台启动 MCP Server...');
+        startMcpServerBackground();
         console.log('[mcp] 请重启 Claude Code 以加载 MCP 服务\n');
         process.exit(0);
     }
-    // 创建 MCP Server
-    const server = new McpServer({
-        name: 'wecom-aibot-mcp',
-        version: VERSION,
-    });
-    // 注册工具
-    registerTools(server, wecomClient);
-    // 启动 HTTP 服务（包含 MCP endpoint）
-    console.log(`[mcp] 启动 MCP HTTP Server (端口: ${HTTP_PORT})...`);
-    await startHttpServer(wecomClient, server);
-    // 定期清理过期消息
-    const cleanupInterval = setInterval(() => {
-        wecomClient.cleanupMessages();
-    }, 60000);
-    console.log(`[mcp] MCP Server 已就绪`);
-    console.log(`[mcp] HTTP endpoint: http://127.0.0.1:${HTTP_PORT}/mcp`);
-    console.log(`[mcp] 健康检查: http://127.0.0.1:${HTTP_PORT}/health`);
-    // 退出处理：清理资源
-    const gracefulShutdown = () => {
-        console.log('[mcp] 正在关闭...');
-        clearInterval(cleanupInterval);
-        wecomClient.disconnect();
-        process.exit(0);
-    };
-    // 监听进程信号
-    process.on('SIGINT', gracefulShutdown);
-    process.on('SIGTERM', gracefulShutdown);
+    // 已有配置，显示状态并提示启动命令
+    showStatus();
+    console.log('\n[mcp] 使用 --start 启动服务，--stop 停止服务');
+    console.log('[mcp] 命令: npx @vrs-soft/wecom-aibot-mcp --start\n');
 }
 main().catch((err) => {
     console.error('[mcp] 启动失败:', err);

package/dist/client.d.ts

@@ -4,6 +4,10 @@ interface ApprovalRecord {
     result?: 'allow-once' | 'allow-always' | 'deny';
     timestamp: number;
     toolName?: string;
+    toolInput?: Record<string, unknown>;
+    projectDir?: string;
+    lastKeepaliveMinute?: number;
+    keepaliveCount?: number;
 }
 interface MessageRecord {
     msgid: string;
@@ -41,6 +45,8 @@ declare class WecomClient {
     sendApprovalRequest(title: string, description: string, requestId: string, targetUser?: string): Promise<string>;
     getApprovalResult(taskId: string): 'pending' | 'allow-once' | 'allow-always' | 'deny';
     getPendingApprovals(): string[];
+    getPendingApprovalsRecords(): ApprovalRecord[];
+    getApprovalRecord(taskId: string): ApprovalRecord | undefined;
     getLatestMessage(afterTimestamp: number): MessageRecord | undefined;
     getPendingMessages(clear?: boolean): MessageRecord[];
     cleanupMessages(maxAgeMs?: number): void;

package/dist/client.js

@@ -5,6 +5,7 @@
  * 管理消息队列和审批状态。
  */
 import AiBot from '@wecom/aibot-node-sdk';
+import { logConnected, logAuthenticated, logDisconnected, logReconnecting, logError, } from './connection-log.js';
 class WecomClient {
     wsClient;
     approvals = new Map();
@@ -22,7 +23,7 @@ class WecomClient {
         this.wsClient = new AiBot.WSClient({
             botId,
             secret,
-            heartbeatInterval: 30000,
+            heartbeatInterval: 15000, // 15 秒心跳，更快检测断线
             maxReconnectAttempts: -1, // 无限重连
         });
         this.setupEventHandlers();
@@ -37,14 +38,14 @@ class WecomClient {
     }
     setupEventHandlers() {
         this.wsClient.on('connected', () => {
-            console.log('[wecom] WebSocket 连接已建立');
+            logConnected();
         });
         this.wsClient.on('authenticated', () => {
             const wasReconnecting = this.wasReconnecting;
             this.connected = true;
             this.wasReconnecting = false;
             this.reconnectAttempt = 0;
-            console.log('[wecom] 认证成功，长连接已就绪');
+            logAuthenticated();
             // 重连成功后发送通知
             if (wasReconnecting) {
                 this.sendText('【系统】连接已恢复').catch(err => {
@@ -58,7 +59,7 @@ class WecomClient {
             this.connected = false;
             this.wasReconnecting = true;
             this.lastDisconnectTime = Date.now();
-            console.log(`[wecom] 连接断开: ${reason}`);
+            logDisconnected(reason);
             // 发送断线通知
             this.sendText('【系统】连接中断，正在重连...').catch(err => {
                 console.error('[wecom] 发送断线通知失败:', err);
@@ -66,10 +67,10 @@ class WecomClient {
         });
         this.wsClient.on('reconnecting', (attempt) => {
             this.reconnectAttempt = attempt;
-            console.log(`[wecom] 正在重连 (第 ${attempt} 次)`);
+            logReconnecting(attempt);
         });
         this.wsClient.on('error', (err) => {
-            console.error(`[wecom] 错误: ${err.message}`);
+            logError(err.message);
             // 检测授权相关错误（40058: invalid Request Parameter）
             if (err.message.includes('40058') || err.message.includes('invalid Request Parameter')) {
                 console.log('');
@@ -90,6 +91,7 @@ class WecomClient {
         });
         // 监听模板卡片事件（审批结果）
         this.wsClient.on('event.template_card_event', (frame) => {
+            console.log('[wecom] 收到 template_card_event 事件');
             this.handleApprovalResponse(frame);
         });
         // 监听进入会话事件
@@ -304,6 +306,15 @@ class WecomClient {
             .filter(([_, a]) => !a.resolved)
             .map(([taskId, _]) => taskId);
     }
+    // 获取所有待处理审批记录（供保活监控使用）
+    getPendingApprovalsRecords() {
+        return Array.from(this.approvals.values())
+            .filter(a => !a.resolved);
+    }
+    // 获取单个审批记录
+    getApprovalRecord(taskId) {
+        return this.approvals.get(taskId);
+    }
     // 获取最新消息（用于等待回复）
     getLatestMessage(afterTimestamp) {
         return this.messages.find(m => m.timestamp > afterTimestamp && m.from_userid === this.targetUserId);

package/dist/config-wizard.d.ts

@@ -16,8 +16,18 @@ export declare function deleteMcpConfig(instanceName: string): boolean;
 export declare function deleteMcpConfigInteractive(instanceName?: string): Promise<void>;
 export declare function uninstall(): void;
 export declare function addMcpConfig(): Promise<void>;
+export declare function listAllRobots(): Array<{
+    name: string;
+    botId: string;
+    targetUserId: string;
+    isDefault: boolean;
+}>;
 export declare function ensureHookInstalled(): void;
 export declare function saveConfig(config: WecomConfig, instanceName?: string): void;
+/**
+ * 安装 headless-mode skill 到项目目录
+ */
+export declare function installSkill(projectDir: string): void;
 /**
  * 运行配置向导
  */