@vrs-soft/wecom-aibot-mcp 1.0.3 → 1.0.5

package/README.md

@@ -60,7 +60,7 @@ PermissionRequest Hook 拦截
 
 ### 审批卡片示例
 
-![审批卡片](docs/approval-card.png)
+<img src="docs/approval-card.png" width="30%" alt="审批卡片示例" />
 
 用户在企业微信中收到审批卡片，点击按钮即可远程决策：
 

package/dist/bin.js

@@ -5,13 +5,13 @@
  * npx 运行入口
  */
 import * as readline from 'readline';
-import { runConfigWizard, loadConfig, deleteConfig, uninstall, ensureHookInstalled } from './config-wizard.js';
+import { runConfigWizard, loadConfig, deleteConfig, uninstall, addMcpConfig, ensureHookInstalled } from './config-wizard.js';
 import { initClient } from './client.js';
 import { registerTools } from './tools/index.js';
 import { startHttpServer } from './http-server.js';
 import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
 import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
-const VERSION = '1.0.0';
+const VERSION = '1.0.5';
 // 等待连接验证（最多等待 10 秒）
 async function waitForConnection(client, timeoutMs = 10000) {
     return new Promise((resolve) => {
@@ -41,7 +41,8 @@ function showHelp() {
 选项:
   --help, -h      显示帮助信息
   --version, -v   显示版本号
-  --config        重新配置（修改 Bot ID / Secret / 目标用户）
+  --config        重新配置默认机器人（修改 Bot ID / Secret / 目标用户）
+  --add           添加新的机器人配置（多机器人场景）
   --status        显示当前配置状态
   --uninstall     卸载并删除所有配置（包括 MCP 配置、hook、skill）
 
@@ -133,6 +134,10 @@ async function main() {
         uninstall();
         process.exit(0);
     }
+    if (args.includes('--add')) {
+        await addMcpConfig();
+        process.exit(0);
+    }
     const reconfig = args.includes('--config');
     const isInteractive = process.stdin.isTTY; // 是否为用户交互模式
     console.log('');
@@ -233,10 +238,20 @@ async function main() {
     const transport = new StdioServerTransport();
     await server.connect(transport);
     // 定期清理过期消息
-    setInterval(() => {
+    const cleanupInterval = setInterval(() => {
         wecomClient.cleanupMessages();
     }, 60000);
     console.log('[mcp] MCP Server 已就绪');
+    // 退出处理：清理资源
+    const gracefulShutdown = () => {
+        console.log('[mcp] 正在关闭...');
+        clearInterval(cleanupInterval);
+        wecomClient.disconnect();
+        process.exit(0);
+    };
+    // 监听进程信号
+    process.on('SIGINT', gracefulShutdown);
+    process.on('SIGTERM', gracefulShutdown);
 }
 main().catch((err) => {
     console.error('[mcp] 启动失败:', err);

package/dist/config-wizard.d.ts

@@ -6,10 +6,10 @@ export interface WecomConfig {
 }
 export declare function loadConfig(): WecomConfig | null;
 export declare function deleteConfig(): void;
-export declare function deleteMcpConfig(): void;
 export declare function deleteHook(): void;
 export declare function deleteSkills(): void;
 export declare function uninstall(): void;
+export declare function addMcpConfig(): Promise<void>;
 export declare function ensureHookInstalled(): void;
 export declare function saveConfig(config: WecomConfig): void;
 /**

package/dist/config-wizard.js

@@ -2,13 +2,13 @@
  * 配置向导模块
  *
  * 首次运行时引导用户配置 Bot ID、Secret 和默认目标用户
+ * 配置直接存储在 ~/.claude.json 的 mcpServers.wecom-aibot.env 中
  */
 import * as readline from 'readline';
 import * as fs from 'fs';
 import * as path from 'path';
 import * as os from 'os';
 const CONFIG_DIR = path.join(os.homedir(), '.wecom-aibot-mcp');
-const CONFIG_FILE = path.join(CONFIG_DIR, 'config.json');
 const CLAUDE_CONFIG_FILE = path.join(os.homedir(), '.claude.json');
 const CLAUDE_SETTINGS_FILE = path.join(os.homedir(), '.claude', 'settings.local.json');
 const HOOK_SCRIPT_PATH = path.join(CONFIG_DIR, 'permission-hook.sh');
@@ -24,18 +24,30 @@ const MCP_TOOL_PERMISSIONS = [
     'mcp__wecom-aibot__enter_headless_mode',
     'mcp__wecom-aibot__exit_headless_mode',
 ];
-// 确保配置目录存在
+// 确保配置目录存在（用于存储端口文件、hook脚本等运行时文件）
 function ensureConfigDir() {
     if (!fs.existsSync(CONFIG_DIR)) {
         fs.mkdirSync(CONFIG_DIR, { recursive: true });
     }
 }
-// 读取已保存的配置
+// 从 ~/.claude.json 读取已保存的配置
 export function loadConfig() {
     try {
-        if (fs.existsSync(CONFIG_FILE)) {
-            const content = fs.readFileSync(CONFIG_FILE, 'utf-8');
-            return JSON.parse(content);
+        // 优先从 ~/.claude.json 读取
+        if (fs.existsSync(CLAUDE_CONFIG_FILE)) {
+            const content = fs.readFileSync(CLAUDE_CONFIG_FILE, 'utf-8');
+            const claudeConfig = JSON.parse(content);
+            const mcpConfig = claudeConfig.mcpServers?.['wecom-aibot'];
+            if (mcpConfig?.env) {
+                const { WECOM_BOT_ID, WECOM_SECRET, WECOM_TARGET_USER } = mcpConfig.env;
+                if (WECOM_BOT_ID && WECOM_SECRET && WECOM_TARGET_USER) {
+                    return {
+                        botId: WECOM_BOT_ID,
+                        secret: WECOM_SECRET,
+                        targetUserId: WECOM_TARGET_USER,
+                    };
+                }
+            }
         }
     }
     catch (err) {
@@ -43,36 +55,21 @@ export function loadConfig() {
     }
     return null;
 }
-// 删除已保存的配置（连接失败时让用户重新输入）
+// 删除配置（从 ~/.claude.json）
 export function deleteConfig() {
-    try {
-        if (fs.existsSync(CONFIG_FILE)) {
-            fs.unlinkSync(CONFIG_FILE);
-            console.log('[config] 已删除配置文件');
-        }
-    }
-    catch (err) {
-        console.error('[config] 删除配置失败:', err);
-    }
-}
-// 删除 MCP Server 配置（从 ~/.claude.json）
-export function deleteMcpConfig() {
     try {
         if (fs.existsSync(CLAUDE_CONFIG_FILE)) {
             const content = fs.readFileSync(CLAUDE_CONFIG_FILE, 'utf-8');
             const claudeConfig = JSON.parse(content);
-            if (claudeConfig.mcpServers && claudeConfig.mcpServers['wecom-aibot']) {
+            if (claudeConfig.mcpServers?.['wecom-aibot']) {
                 delete claudeConfig.mcpServers['wecom-aibot'];
                 fs.writeFileSync(CLAUDE_CONFIG_FILE, JSON.stringify(claudeConfig, null, 2));
-                console.log('[config] 已从 ~/.claude.json 删除 wecom-aibot MCP 配置');
-            }
-            else {
-                console.log('[config] ~/.claude.json 中未找到 wecom-aibot 配置');
+                console.log('[config] 已从 ~/.claude.json 删除 wecom-aibot 配置');
             }
         }
     }
     catch (err) {
-        console.error('[config] 删除 MCP 配置失败:', err);
+        console.error('[config] 删除配置失败:', err);
     }
 }
 // 删除 PermissionRequest hook（从 ~/.claude/settings.local.json）
@@ -117,11 +114,10 @@ export function deleteSkills() {
 // 完全卸载（删除所有相关配置）
 export function uninstall() {
     console.log('\n[config] 开始卸载 wecom-aibot-mcp...\n');
-    deleteConfig();
-    deleteMcpConfig();
+    deleteConfig(); // 删除 ~/.claude.json 中的配置
     deleteHook();
     deleteSkills();
-    // 删除配置目录
+    // 删除运行时文件目录
     if (fs.existsSync(CONFIG_DIR)) {
         try {
             // 删除所有 port-* 和 headless-* 文件
@@ -261,7 +257,7 @@ done
     console.log(`[config] Hook 脚本已写入: ${HOOK_SCRIPT_PATH}`);
 }
 // 写入 MCP Server 配置到 ~/.claude.json
-function writeMcpServerConfig(config) {
+function writeMcpServerConfig(config, instanceName) {
     try {
         // 读取现有配置
         let claudeConfig = {};
@@ -272,13 +268,13 @@ function writeMcpServerConfig(config) {
         // 确保 mcpServers 存在
         if (!claudeConfig.mcpServers)
             claudeConfig.mcpServers = {};
-        // 检查是否已有 wecom-aibot 配置
-        if (claudeConfig.mcpServers['wecom-aibot']) {
-            console.log('[config] ~/.claude.json 中已存在 wecom-aibot 配置，跳过写入');
-            return true;
+        const name = instanceName || 'wecom-aibot';
+        // 检查是否已存在同名配置
+        if (claudeConfig.mcpServers[name] && !instanceName) {
+            console.log(`[config] ~/.claude.json 中已存在 ${name} 配置，将更新`);
         }
         // 写入 MCP Server 配置
-        claudeConfig.mcpServers['wecom-aibot'] = {
+        claudeConfig.mcpServers[name] = {
             command: 'npx',
             args: ['@vrs-soft/wecom-aibot-mcp'],
             env: {
@@ -288,15 +284,16 @@ function writeMcpServerConfig(config) {
             },
         };
         fs.writeFileSync(CLAUDE_CONFIG_FILE, JSON.stringify(claudeConfig, null, 2));
-        console.log(`[config] MCP Server 配置已写入 ~/.claude.json`);
+        console.log(`[config] MCP Server 配置已写入 ~/.claude.json (实例名: ${name})`);
         return true;
     }
     catch (err) {
         console.error('[config] 写入 ~/.claude.json 失败:', err);
         console.log('[config] ⚠️  请手动将以下配置添加到 ~/.claude.json:');
+        const name = instanceName || 'wecom-aibot';
         console.log(JSON.stringify({
             mcpServers: {
-                'wecom-aibot': {
+                [name]: {
                     command: 'npx',
                     args: ['@vrs-soft/wecom-aibot-mcp'],
                     env: {
@@ -310,6 +307,68 @@ function writeMcpServerConfig(config) {
         return false;
     }
 }
+// 添加新的 MCP 配置（用于多 bot 场景）
+export async function addMcpConfig() {
+    const rl = createRL();
+    try {
+        console.log('\n添加新的企业微信机器人配置\n');
+        // 获取实例名称
+        const instanceName = await question(rl, 'MCP 实例名称（如 wecom-aibot-zhangsan）: ');
+        if (!instanceName) {
+            console.log('[config] 实例名称不能为空');
+            return;
+        }
+        // 检查名称是否已存在
+        if (fs.existsSync(CLAUDE_CONFIG_FILE)) {
+            const content = fs.readFileSync(CLAUDE_CONFIG_FILE, 'utf-8');
+            const claudeConfig = JSON.parse(content);
+            if (claudeConfig.mcpServers?.[instanceName]) {
+                console.log(`[config] 实例名称 "${instanceName}" 已存在，请使用其他名称`);
+                return;
+            }
+        }
+        // 获取 Bot ID
+        let botId = await question(rl, 'Bot ID: ');
+        while (!botId) {
+            console.log('Bot ID 不能为空');
+            botId = await question(rl, 'Bot ID: ');
+        }
+        // 获取 Secret
+        let secret = await question(rl, 'Secret: ');
+        while (!secret) {
+            console.log('Secret 不能为空');
+            secret = await question(rl, 'Secret: ');
+        }
+        // 获取默认目标用户
+        console.log('\n请输入默认交互用户的 User ID（企业微信用户账号）\n');
+        let targetUserId = await question(rl, '默认目标用户 ID: ');
+        while (!targetUserId) {
+            console.log('用户 ID 不能为空');
+            targetUserId = await question(rl, '默认目标用户 ID: ');
+        }
+        const config = { botId, secret, targetUserId };
+        // 确认配置
+        console.log('\n─────────────────────────────────────');
+        console.log('配置确认：');
+        console.log(`  实例名称:   ${instanceName}`);
+        console.log(`  Bot ID:     ${botId}`);
+        console.log(`  Secret:     ${secret.slice(0, 8)}...${secret.slice(-4)}`);
+        console.log(`  目标用户:   ${targetUserId}`);
+        console.log('─────────────────────────────────────\n');
+        const confirm = await question(rl, '确认添加配置？(Y/n): ');
+        if (confirm.toLowerCase() === 'n') {
+            console.log('[config] 已取消');
+            return;
+        }
+        // 写入配置
+        writeMcpServerConfig(config, instanceName);
+        console.log(`\n[config] ✅ 已添加新机器人配置: ${instanceName}`);
+        console.log('[config] 请重启 Claude Code 以加载新的 MCP 服务\n');
+    }
+    finally {
+        rl.close();
+    }
+}
 // 安装 skill 文件到 ~/.claude/skills/
 function installSkills() {
     try {
@@ -395,14 +454,12 @@ export function ensureHookInstalled() {
     writeMcpPermissions();
     installSkills();
 }
-// 保存配置（并自动写入 MCP 权限和 Server 配置）
+// 保存配置（直接写入 ~/.claude.json）
 export function saveConfig(config) {
-    ensureConfigDir();
-    fs.writeFileSync(CONFIG_FILE, JSON.stringify(config, null, 2));
-    console.log(`[config] 配置已保存到 ${CONFIG_FILE}`);
-    // 自动写入 MCP Server 配置到 ~/.claude.json
+    ensureConfigDir(); // 确保运行时文件目录存在
+    // 写入 MCP Server 配置到 ~/.claude.json
     writeMcpServerConfig(config);
-    // 自动写入 MCP 工具权限和 Hook 到 ~/.claude/settings.local.json
+    // 写入 MCP 工具权限和 Hook 到 ~/.claude/settings.local.json
     writeMcpPermissions();
 }
 // 创建 readline 接口

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@vrs-soft/wecom-aibot-mcp",
-  "version": "1.0.3",
+  "version": "1.0.5",
   "description": "企业微信智能机器人 MCP 服务 - Claude Code 审批通道",
   "type": "module",
   "main": "dist/index.js",

package/skills/headless-mode/SKILL.md

@@ -7,6 +7,23 @@ description: 当用户说「现在开始通过微信联系」、「我要离开
 
 **触发条件**：「现在开始通过微信联系」、「我要离开电脑前」、「切换到微信模式」
 
+## 多机器人场景
+
+如果用户配置了多个企业微信机器人（如 `wecom-aibot-zhangsan`、`wecom-aibot-lisi`），在进入微信模式前需要确认使用哪个：
+
+```
+【需要确认】检测到您有多个企业微信机器人配置，请选择使用哪个：
+1. wecom-aibot（默认）
+2. wecom-aibot-zhangsan
+3. wecom-aibot-lisi
+
+请回复数字或名称选择。
+```
+
+选择后，后续的 `send_message`、`get_pending_messages` 等操作都使用对应的 MCP 实例。
+
+> 如何添加多个机器人：运行 `npx @vrs-soft/wecom-aibot-mcp --add`
+
 ## 激活后立即执行（必须按顺序）
 
 **Step 1**：进入 headless 模式（写入状态文件，让 hook 知道要发微信审批）