@vrs-soft/wecom-aibot-mcp 2.3.2 → 2.4.0

package/dist/daemon.js

@@ -47,23 +47,7 @@ class ConnectionDaemon {
     }
     loadAllRobots() {
         const robots = [];
-        // 主配置文件
-        const mainConfigPath = path.join(CONFIG_DIR, 'config.json');
-        if (fs.existsSync(mainConfigPath)) {
-            try {
-                const config = JSON.parse(fs.readFileSync(mainConfigPath, 'utf-8'));
-                robots.push({
-                    name: config.nameTag || 'default',
-                    botId: config.botId,
-                    secret: config.secret,
-                    targetUserId: config.targetUserId,
-                });
-            }
-            catch (err) {
-                this.log(`加载主配置失败: ${err}`);
-            }
-        }
-        // 机器人配置文件 (robot-*.json)
+        // 所有机器人配置（统一 robot-*.json 格式）
         const files = fs.readdirSync(CONFIG_DIR).filter(f => f.startsWith('robot-') && f.endsWith('.json'));
         for (const file of files) {
             try {

package/dist/doc-proxy.d.ts

@@ -0,0 +1,21 @@
+/**
+ * 文档 MCP 代理模块
+ *
+ * 将文档工具调用代理转发到机器人专属的企业微信文档 MCP 服务（StreamableHTTP）。
+ * 每次调用建立独立连接，无需维护长连接状态。
+ */
+export interface DocToolResult {
+    content: Array<{
+        type: string;
+        text: string;
+    }>;
+    isError?: boolean;
+}
+/**
+ * 调用机器人文档 MCP 工具
+ *
+ * @param docMcpUrl  机器人专属的文档 MCP URL（含 uaKey）
+ * @param toolName   工具名称，如 create_doc、smartsheet_add_records 等
+ * @param args       工具参数
+ */
+export declare function callDocTool(docMcpUrl: string, toolName: string, args: Record<string, unknown>): Promise<DocToolResult>;

package/dist/doc-proxy.js

@@ -0,0 +1,46 @@
+/**
+ * 文档 MCP 代理模块
+ *
+ * 将文档工具调用代理转发到机器人专属的企业微信文档 MCP 服务（StreamableHTTP）。
+ * 每次调用建立独立连接，无需维护长连接状态。
+ */
+import { Client } from '@modelcontextprotocol/sdk/client/index.js';
+import { StreamableHTTPClientTransport } from '@modelcontextprotocol/sdk/client/streamableHttp.js';
+import { logger } from './logger.js';
+/**
+ * 调用机器人文档 MCP 工具
+ *
+ * @param docMcpUrl  机器人专属的文档 MCP URL（含 uaKey）
+ * @param toolName   工具名称，如 create_doc、smartsheet_add_records 等
+ * @param args       工具参数
+ */
+export async function callDocTool(docMcpUrl, toolName, args) {
+    const client = new Client({ name: 'wecom-aibot-doc-proxy', version: '1.0.0' }, { capabilities: {} });
+    try {
+        const transport = new StreamableHTTPClientTransport(new URL(docMcpUrl));
+        await client.connect(transport);
+        const result = await client.callTool({ name: toolName, arguments: args });
+        return {
+            content: result.content ?? [
+                { type: 'text', text: JSON.stringify(result) },
+            ],
+            isError: result.isError,
+        };
+    }
+    catch (err) {
+        logger.error(`[doc-proxy] 调用 ${toolName} 失败:`, err);
+        const message = err?.message ?? String(err);
+        return {
+            content: [{ type: 'text', text: `文档工具调用失败: ${message}\n\n请检查机器人的文档 MCP URL 是否正确，或该机器人是否已授权文档能力。` }],
+            isError: true,
+        };
+    }
+    finally {
+        try {
+            await client.close();
+        }
+        catch {
+            // 关闭时忽略错误
+        }
+    }
+}

package/dist/http-server.js

@@ -23,7 +23,7 @@ import { isInitializeRequest } from '@modelcontextprotocol/sdk/types.js';
 import { registerTools } from './tools/index.js';
 import { getClient, getConnectionState, getAllConnectionStates, connectAllRobots } from './connection-manager.js';
 import { subscribeWecomMessage, getSubscriberCount } from './message-bus.js';
-import { listAllRobots } from './config-wizard.js';
+import { listAllRobots, VERSION, getAuthToken } from './config-wizard.js';
 import { logger } from './logger.js';
 // ESM 兼容的 __dirname
 const __filename = fileURLToPath(import.meta.url);
@@ -185,7 +185,6 @@ export function getOnlineCcIds() {
 }
 // 使用 Map 存储多个待处理审批（按 taskId 索引）
 const pendingApprovals = new Map();
-const VERSION = '2.0.0';
 const transports = new Map();
 const sseClients = new Map(); // clientId -> SSEClient
 // 初始化 MCP Server（不再全局连接）
@@ -479,6 +478,16 @@ export async function startHttpServer(_server, port = HTTP_PORT) {
                 return;
             }
             const url = req.url || '/';
+            // Auth token 校验（排除 /health）
+            const authToken = getAuthToken();
+            if (authToken && url !== '/health') {
+                const authHeader = req.headers['authorization'];
+                if (authHeader !== `Bearer ${authToken}`) {
+                    res.writeHead(401, { 'Content-Type': 'application/json' });
+                    res.end(JSON.stringify({ error: 'Unauthorized' }));
+                    return;
+                }
+            }
             // MCP endpoint - 每个客户端一个独立的 server 和 transport
             // POST /mcp: 初始化或调用工具
             // GET /mcp: 建立 SSE 流

package/dist/tools/index.js

@@ -20,7 +20,8 @@
  * - 从 Session 自动获取 robotName
  */
 import { z } from 'zod';
-import { listAllRobots, installSkill } from '../config-wizard.js';
+import { listAllRobots, getDocMcpUrl, installSkill, VERSION } from '../config-wizard.js';
+import { callDocTool } from '../doc-proxy.js';
 import { connectRobot, disconnectRobot, getClient, getConnectionState, } from '../connection-manager.js';
 import { registerCcId, unregisterCcId, getRobotByCcId, getProjectDirByCcId, generateCcId, } from '../http-server.js';
 import { subscribeWecomMessageByCcId } from '../message-bus.js';
@@ -249,7 +250,7 @@ npx @vrs-soft/wecom-aibot-mcp
             content: [{
                     type: 'text',
                     text: JSON.stringify({
-                        version: '2.0.0',
+                        version: VERSION,
                         requirements: {
                             // 权限配置需求
                             permissions: {
@@ -312,13 +313,15 @@ npx @vrs-soft/wecom-aibot-mcp
         bot_id: z.string().describe('企业微信 Bot ID'),
         secret: z.string().describe('机器人密钥'),
         default_user: z.string().optional().describe('默认目标用户'),
-    }, async ({ name, bot_id, secret, default_user }) => {
+        doc_mcp_url: z.string().optional().describe('机器人文档 MCP URL（企业微信文档能力）'),
+    }, async ({ name, bot_id, secret, default_user, doc_mcp_url }) => {
         return {
             content: [{
                     type: 'text',
                     text: JSON.stringify({
                         message: '请使用 --add 命令添加机器人配置',
                         command: `npx @vrs-soft/wecom-aibot-mcp --add`,
+                        tip: doc_mcp_url ? '运行向导时在"文档 MCP URL"提示处粘贴您的 URL' : undefined,
                     }, null, 2),
                 }],
         };
@@ -334,6 +337,7 @@ npx @vrs-soft/wecom-aibot-mcp
             name: robot.name,
             botId: robot.botId?.slice(0, 12) + '...', // 只显示前12位
             targetUser: robot.targetUserId,
+            hasDocMcp: !!robot.doc_mcp_url, // 是否配置了文档 MCP 能力
         }));
         return {
             content: [{
@@ -413,11 +417,11 @@ npx @vrs-soft/wecom-aibot-mcp
                     }],
             };
         }
-        // 如果用户传入 cc_id，直接使用；否则自动生成
-        const finalCcId = cc_id || generateCcId(effectiveAgentName);
-        // 检查 wecom-aibot.json 是否已有匹配的 ccId（重连场景）
+        // 检查 wecom-aibot.json 是否已有 ccId（重连场景）
         const existingConfig = loadWechatModeConfig(projectDir);
-        const isReconnect = existingConfig?.ccId === finalCcId;
+        // 优先级：cc_id 参数 > (非 agent_name 指定时) 配置文件已有 ccId > 自动生成
+        const finalCcId = cc_id || (!agent_name && existingConfig?.ccId) || generateCcId(effectiveAgentName);
+        const isReconnect = !!existingConfig?.ccId && existingConfig.ccId === finalCcId;
         // 注册 ccId 到 CC 注册表（重连直接覆盖，首次注册清理超时条目）
         registerCcId(finalCcId, selectedRobot.name, effectiveAgentName, mode, projectDir, isReconnect);
         // 更新项目配置文件中的 wechatMode 为 true
@@ -598,5 +602,205 @@ npx @vrs-soft/wecom-aibot-mcp
                 }],
         };
     });
-    logger.log('[mcp] 已注册 14 个工具');
+    // ============================================
+    // 文档代理工具（企业微信文档 MCP 能力）
+    // 代理转发到机器人专属的 doc_mcp_url
+    // ============================================
+    // 公共辅助：解析 doc MCP URL
+    function resolveDocUrl(robotName) {
+        const { url, error } = getDocMcpUrl(robotName);
+        if (!url) {
+            return {
+                url: null,
+                errorContent: { content: [{ type: 'text', text: error ?? '未配置文档 MCP URL' }] },
+            };
+        }
+        return { url };
+    }
+    server.tool('create_doc', '新建文档或智能表格。doc_type: 3=文档, 10=智能表格。创建成功后返回文档链接。', {
+        doc_type: z.number().int().describe('文档类型：3=文档，10=智能表格'),
+        doc_name: z.string().describe('文档名称，最多 255 个字符'),
+        robot_name: z.string().optional().describe('指定机器人名称（多机器人时必填）'),
+    }, async ({ doc_type, doc_name, robot_name }) => {
+        const { url, errorContent } = resolveDocUrl(robot_name);
+        if (!url)
+            return errorContent;
+        return callDocTool(url, 'create_doc', { doc_type, doc_name });
+    });
+    server.tool('get_doc_content', '获取企业微信文档内容（Markdown 格式）。支持异步轮询：首次调用无需 task_id，若 task_done 为 false 则带 task_id 继续轮询。', {
+        type: z.number().int().describe('内容格式：2=Markdown'),
+        url: z.string().optional().describe('文档访问链接，与 docid 二选一'),
+        docid: z.string().optional().describe('文档 docid，与 url 二选一'),
+        task_id: z.string().optional().describe('任务 ID（轮询时填写）'),
+        robot_name: z.string().optional().describe('指定机器人名称（多机器人时必填）'),
+    }, async ({ type, url: docUrl, docid, task_id, robot_name }) => {
+        const { url, errorContent } = resolveDocUrl(robot_name);
+        if (!url)
+            return errorContent;
+        return callDocTool(url, 'get_doc_content', { type, url: docUrl, docid, task_id });
+    });
+    server.tool('edit_doc_content', '编辑企业微信文档内容，支持 Markdown 格式覆写。', {
+        content: z.string().describe('覆写的文档内容'),
+        content_type: z.number().int().describe('内容类型：1=Markdown'),
+        url: z.string().optional().describe('文档访问链接，与 docid 二选一'),
+        docid: z.string().optional().describe('文档 docid，与 url 二选一'),
+        robot_name: z.string().optional().describe('指定机器人名称（多机器人时必填）'),
+    }, async ({ content, content_type, url: docUrl, docid, robot_name }) => {
+        const { url, errorContent } = resolveDocUrl(robot_name);
+        if (!url)
+            return errorContent;
+        return callDocTool(url, 'edit_doc_content', { content, content_type, url: docUrl, docid });
+    });
+    server.tool('smartsheet_get_sheet', '查询智能表格中的子表信息。', {
+        url: z.string().optional().describe('文档访问链接，与 docid 二选一'),
+        docid: z.string().optional().describe('文档 docid，与 url 二选一'),
+        robot_name: z.string().optional().describe('指定机器人名称（多机器人时必填）'),
+    }, async ({ url: docUrl, docid, robot_name }) => {
+        const { url, errorContent } = resolveDocUrl(robot_name);
+        if (!url)
+            return errorContent;
+        return callDocTool(url, 'smartsheet_get_sheet', { url: docUrl, docid });
+    });
+    server.tool('smartsheet_add_sheet', '在智能表格内添加新子表。', {
+        url: z.string().optional().describe('文档访问链接，与 docid 二选一'),
+        docid: z.string().optional().describe('文档 docid，与 url 二选一'),
+        properties: z.object({ title: z.string().optional() }).optional().describe('子表属性，可设置 title'),
+        robot_name: z.string().optional().describe('指定机器人名称（多机器人时必填）'),
+    }, async ({ url: docUrl, docid, properties, robot_name }) => {
+        const { url, errorContent } = resolveDocUrl(robot_name);
+        if (!url)
+            return errorContent;
+        return callDocTool(url, 'smartsheet_add_sheet', { url: docUrl, docid, properties });
+    });
+    server.tool('smartsheet_update_sheet', '修改智能表格子表的标题。', {
+        properties: z.object({
+            sheet_id: z.string().describe('子表 ID'),
+            title: z.string().describe('新标题'),
+        }).describe('子表属性'),
+        url: z.string().optional().describe('文档访问链接，与 docid 二选一'),
+        docid: z.string().optional().describe('文档 docid，与 url 二选一'),
+        robot_name: z.string().optional().describe('指定机器人名称（多机器人时必填）'),
+    }, async ({ properties, url: docUrl, docid, robot_name }) => {
+        const { url, errorContent } = resolveDocUrl(robot_name);
+        if (!url)
+            return errorContent;
+        return callDocTool(url, 'smartsheet_update_sheet', { properties, url: docUrl, docid });
+    });
+    server.tool('smartsheet_delete_sheet', '删除智能表格中的子表（不可逆）。', {
+        sheet_id: z.string().describe('要删除的子表 ID'),
+        url: z.string().optional().describe('文档访问链接，与 docid 二选一'),
+        docid: z.string().optional().describe('文档 docid，与 url 二选一'),
+        robot_name: z.string().optional().describe('指定机器人名称（多机器人时必填）'),
+    }, async ({ sheet_id, url: docUrl, docid, robot_name }) => {
+        const { url, errorContent } = resolveDocUrl(robot_name);
+        if (!url)
+            return errorContent;
+        return callDocTool(url, 'smartsheet_delete_sheet', { sheet_id, url: docUrl, docid });
+    });
+    server.tool('smartsheet_get_fields', '获取智能表格子表的字段（列）信息。', {
+        sheet_id: z.string().describe('子表 ID'),
+        url: z.string().optional().describe('文档访问链接，与 docid 二选一'),
+        docid: z.string().optional().describe('文档 docid，与 url 二选一'),
+        robot_name: z.string().optional().describe('指定机器人名称（多机器人时必填）'),
+    }, async ({ sheet_id, url: docUrl, docid, robot_name }) => {
+        const { url, errorContent } = resolveDocUrl(robot_name);
+        if (!url)
+            return errorContent;
+        return callDocTool(url, 'smartsheet_get_fields', { sheet_id, url: docUrl, docid });
+    });
+    server.tool('smartsheet_add_fields', '向智能表格子表添加字段（列）。调用前必须先用 smartsheet_get_fields 查看现有字段并用 smartsheet_update_fields 重命名默认字段，否则会多出无用列。', {
+        sheet_id: z.string().describe('子表 ID'),
+        fields: z.array(z.object({
+            field_title: z.string(),
+            field_type: z.string(),
+        })).describe('要添加的字段列表'),
+        url: z.string().optional().describe('文档访问链接，与 docid 二选一'),
+        docid: z.string().optional().describe('文档 docid，与 url 二选一'),
+        robot_name: z.string().optional().describe('指定机器人名称（多机器人时必填）'),
+    }, async ({ sheet_id, fields, url: docUrl, docid, robot_name }) => {
+        const { url, errorContent } = resolveDocUrl(robot_name);
+        if (!url)
+            return errorContent;
+        return callDocTool(url, 'smartsheet_add_fields', { sheet_id, fields, url: docUrl, docid });
+    });
+    server.tool('smartsheet_update_fields', '更新智能表格子表字段的标题（不能更改字段类型）。', {
+        sheet_id: z.string().describe('子表 ID'),
+        fields: z.array(z.object({
+            field_id: z.string(),
+            field_title: z.string(),
+            field_type: z.string(),
+        })).describe('要更新的字段列表'),
+        url: z.string().optional().describe('文档访问链接，与 docid 二选一'),
+        docid: z.string().optional().describe('文档 docid，与 url 二选一'),
+        robot_name: z.string().optional().describe('指定机器人名称（多机器人时必填）'),
+    }, async ({ sheet_id, fields, url: docUrl, docid, robot_name }) => {
+        const { url, errorContent } = resolveDocUrl(robot_name);
+        if (!url)
+            return errorContent;
+        return callDocTool(url, 'smartsheet_update_fields', { sheet_id, fields, url: docUrl, docid });
+    });
+    server.tool('smartsheet_delete_fields', '删除智能表格子表中的字段（列）（不可逆）。', {
+        sheet_id: z.string().describe('子表 ID'),
+        field_ids: z.array(z.string()).describe('要删除的字段 ID 列表'),
+        url: z.string().optional().describe('文档访问链接，与 docid 二选一'),
+        docid: z.string().optional().describe('文档 docid，与 url 二选一'),
+        robot_name: z.string().optional().describe('指定机器人名称（多机器人时必填）'),
+    }, async ({ sheet_id, field_ids, url: docUrl, docid, robot_name }) => {
+        const { url, errorContent } = resolveDocUrl(robot_name);
+        if (!url)
+            return errorContent;
+        return callDocTool(url, 'smartsheet_delete_fields', { sheet_id, field_ids, url: docUrl, docid });
+    });
+    server.tool('smartsheet_get_records', '查询智能表格子表的所有记录。', {
+        sheet_id: z.string().describe('子表 ID'),
+        url: z.string().optional().describe('文档访问链接，与 docid 二选一'),
+        docid: z.string().optional().describe('文档 docid，与 url 二选一'),
+        robot_name: z.string().optional().describe('指定机器人名称（多机器人时必填）'),
+    }, async ({ sheet_id, url: docUrl, docid, robot_name }) => {
+        const { url, errorContent } = resolveDocUrl(robot_name);
+        if (!url)
+            return errorContent;
+        return callDocTool(url, 'smartsheet_get_records', { sheet_id, url: docUrl, docid });
+    });
+    server.tool('smartsheet_add_records', '向智能表格子表添加记录（行）。单次建议不超过 500 行。values 的 key 必须是字段标题（field_title），不能用 field_id。', {
+        sheet_id: z.string().describe('子表 ID'),
+        records: z.array(z.object({ values: z.record(z.string(), z.unknown()) })).describe('记录列表，values key 为字段标题'),
+        url: z.string().optional().describe('文档访问链接，与 docid 二选一'),
+        docid: z.string().optional().describe('文档 docid，与 url 二选一'),
+        robot_name: z.string().optional().describe('指定机器人名称（多机器人时必填）'),
+    }, async ({ sheet_id, records, url: docUrl, docid, robot_name }) => {
+        const { url, errorContent } = resolveDocUrl(robot_name);
+        if (!url)
+            return errorContent;
+        return callDocTool(url, 'smartsheet_add_records', { sheet_id, records, url: docUrl, docid });
+    });
+    server.tool('smartsheet_update_records', '更新智能表格子表中的记录（行）。单次建议不超过 500 行。', {
+        sheet_id: z.string().describe('子表 ID'),
+        records: z.array(z.object({
+            record_id: z.string(),
+            values: z.record(z.string(), z.unknown()),
+        })).describe('要更新的记录列表，含 record_id 和 values'),
+        key_type: z.enum(['CELL_VALUE_KEY_TYPE_FIELD_TITLE', 'CELL_VALUE_KEY_TYPE_FIELD_ID']).describe('values 的 key 类型'),
+        url: z.string().optional().describe('文档访问链接，与 docid 二选一'),
+        docid: z.string().optional().describe('文档 docid，与 url 二选一'),
+        robot_name: z.string().optional().describe('指定机器人名称（多机器人时必填）'),
+    }, async ({ sheet_id, records, key_type, url: docUrl, docid, robot_name }) => {
+        const { url, errorContent } = resolveDocUrl(robot_name);
+        if (!url)
+            return errorContent;
+        return callDocTool(url, 'smartsheet_update_records', { sheet_id, records, key_type, url: docUrl, docid });
+    });
+    server.tool('smartsheet_delete_records', '删除智能表格子表中的记录（行）（不可逆）。单次建议不超过 500 行。', {
+        sheet_id: z.string().describe('子表 ID'),
+        record_ids: z.array(z.string()).describe('要删除的记录 ID 列表'),
+        url: z.string().optional().describe('文档访问链接，与 docid 二选一'),
+        docid: z.string().optional().describe('文档 docid，与 url 二选一'),
+        robot_name: z.string().optional().describe('指定机器人名称（多机器人时必填）'),
+    }, async ({ sheet_id, record_ids, url: docUrl, docid, robot_name }) => {
+        const { url, errorContent } = resolveDocUrl(robot_name);
+        if (!url)
+            return errorContent;
+        return callDocTool(url, 'smartsheet_delete_records', { sheet_id, record_ids, url: docUrl, docid });
+    });
+    logger.log('[mcp] 已注册 29 个工具（含 15 个文档代理工具）');
 }

package/package.json

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