@wecom/wecom-openclaw-plugin 2026.3.31-beta.0 → 2026.4.1-beta.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +4 -251
- package/dist/cjs/index.js +45 -0
- package/dist/cjs/index.js.map +1 -0
- package/dist/cjs/src/channel.js +339 -0
- package/dist/cjs/src/channel.js.map +1 -0
- package/dist/cjs/src/const.js +137 -0
- package/dist/cjs/src/const.js.map +1 -0
- package/dist/cjs/src/dm-policy.js +89 -0
- package/dist/cjs/src/dm-policy.js.map +1 -0
- package/dist/cjs/src/group-policy.js +133 -0
- package/dist/cjs/src/group-policy.js.map +1 -0
- package/dist/cjs/src/mcp/interceptors/biz-error.js +70 -0
- package/dist/cjs/src/mcp/interceptors/biz-error.js.map +1 -0
- package/dist/cjs/src/mcp/interceptors/index.js +73 -0
- package/dist/cjs/src/mcp/interceptors/index.js.map +1 -0
- package/dist/cjs/src/mcp/interceptors/msg-media.js +159 -0
- package/dist/cjs/src/mcp/interceptors/msg-media.js.map +1 -0
- package/dist/cjs/src/mcp/interceptors/smartpage-create.js +157 -0
- package/dist/cjs/src/mcp/interceptors/smartpage-create.js.map +1 -0
- package/dist/cjs/src/mcp/interceptors/smartpage-export.js +93 -0
- package/dist/cjs/src/mcp/interceptors/smartpage-export.js.map +1 -0
- package/dist/cjs/src/mcp/schema.js +104 -0
- package/dist/cjs/src/mcp/schema.js.map +1 -0
- package/dist/cjs/src/mcp/tool.js +208 -0
- package/dist/cjs/src/mcp/tool.js.map +1 -0
- package/dist/cjs/src/mcp/transport.js +417 -0
- package/dist/cjs/src/mcp/transport.js.map +1 -0
- package/dist/cjs/src/media-handler.js +129 -0
- package/dist/cjs/src/media-handler.js.map +1 -0
- package/dist/cjs/src/media-uploader.js +284 -0
- package/dist/cjs/src/media-uploader.js.map +1 -0
- package/dist/cjs/src/message-parser.js +134 -0
- package/dist/cjs/src/message-parser.js.map +1 -0
- package/dist/cjs/src/message-sender.js +89 -0
- package/dist/cjs/src/message-sender.js.map +1 -0
- package/dist/cjs/src/monitor.js +991 -0
- package/dist/cjs/src/monitor.js.map +1 -0
- package/dist/cjs/src/onboarding.js +182 -0
- package/dist/cjs/src/onboarding.js.map +1 -0
- package/dist/cjs/src/openclaw-compat.js +400 -0
- package/dist/cjs/src/openclaw-compat.js.map +1 -0
- package/dist/cjs/src/reqid-store.js +88 -0
- package/dist/cjs/src/reqid-store.js.map +1 -0
- package/dist/cjs/src/runtime.js +16 -0
- package/dist/cjs/src/runtime.js.map +1 -0
- package/dist/cjs/src/state-dir-resolve.js +35 -0
- package/dist/cjs/src/state-dir-resolve.js.map +1 -0
- package/dist/cjs/src/state-manager.js +158 -0
- package/dist/cjs/src/state-manager.js.map +1 -0
- package/dist/cjs/src/template-card-parser.js +641 -0
- package/dist/cjs/src/template-card-parser.js.map +1 -0
- package/dist/cjs/src/timeout.js +42 -0
- package/dist/cjs/src/timeout.js.map +1 -0
- package/dist/cjs/src/utils.js +60 -0
- package/dist/cjs/src/utils.js.map +1 -0
- package/dist/cjs/src/version.js +26 -0
- package/dist/cjs/src/version.js.map +1 -0
- package/dist/esm/index.js +41 -0
- package/dist/esm/index.js.map +1 -0
- package/dist/esm/src/channel.js +337 -0
- package/dist/esm/src/channel.js.map +1 -0
- package/dist/{src → esm/src}/const.js +30 -77
- package/dist/esm/src/const.js.map +1 -0
- package/dist/{src → esm/src}/dm-policy.js +9 -6
- package/dist/esm/src/dm-policy.js.map +1 -0
- package/dist/{src → esm/src}/group-policy.js +7 -3
- package/dist/esm/src/group-policy.js.map +1 -0
- package/dist/{src → esm/src}/mcp/interceptors/biz-error.js +6 -2
- package/dist/esm/src/mcp/interceptors/biz-error.js.map +1 -0
- package/dist/{src → esm/src}/mcp/interceptors/index.js +10 -6
- package/dist/esm/src/mcp/interceptors/index.js.map +1 -0
- package/dist/{src → esm/src}/mcp/interceptors/msg-media.js +10 -6
- package/dist/esm/src/mcp/interceptors/msg-media.js.map +1 -0
- package/dist/{src → esm/src}/mcp/interceptors/smartpage-create.js +6 -2
- package/dist/esm/src/mcp/interceptors/smartpage-create.js.map +1 -0
- package/dist/{src → esm/src}/mcp/interceptors/smartpage-export.js +6 -2
- package/dist/esm/src/mcp/interceptors/smartpage-export.js.map +1 -0
- package/dist/{src → esm/src}/mcp/schema.js +4 -1
- package/dist/esm/src/mcp/schema.js.map +1 -0
- package/dist/{src → esm/src}/mcp/tool.js +8 -4
- package/dist/esm/src/mcp/tool.js.map +1 -0
- package/dist/{src → esm/src}/mcp/transport.js +16 -15
- package/dist/esm/src/mcp/transport.js.map +1 -0
- package/dist/{src → esm/src}/media-handler.js +10 -6
- package/dist/esm/src/media-handler.js.map +1 -0
- package/dist/{src → esm/src}/media-uploader.js +10 -64
- package/dist/esm/src/media-uploader.js.map +1 -0
- package/dist/{src → esm/src}/message-parser.js +4 -1
- package/dist/esm/src/message-parser.js.map +1 -0
- package/dist/{src → esm/src}/message-sender.js +11 -46
- package/dist/esm/src/message-sender.js.map +1 -0
- package/dist/{src → esm/src}/monitor.js +282 -151
- package/dist/esm/src/monitor.js.map +1 -0
- package/dist/{src → esm/src}/onboarding.js +26 -22
- package/dist/esm/src/onboarding.js.map +1 -0
- package/dist/{src → esm/src}/openclaw-compat.js +27 -143
- package/dist/esm/src/openclaw-compat.js.map +1 -0
- package/dist/{src → esm/src}/reqid-store.js +7 -4
- package/dist/esm/src/reqid-store.js.map +1 -0
- package/dist/{src → esm/src}/runtime.js +5 -2
- package/dist/esm/src/runtime.js.map +1 -0
- package/dist/esm/src/state-dir-resolve.js +13 -0
- package/dist/esm/src/state-dir-resolve.js.map +1 -0
- package/dist/{src → esm/src}/state-manager.js +16 -84
- package/dist/esm/src/state-manager.js.map +1 -0
- package/dist/{src → esm/src}/template-card-parser.js +8 -17
- package/dist/esm/src/template-card-parser.js.map +1 -0
- package/dist/{src → esm/src}/timeout.js +5 -2
- package/dist/esm/src/timeout.js.map +1 -0
- package/dist/{src → esm/src}/utils.js +10 -24
- package/dist/esm/src/utils.js.map +1 -0
- package/dist/esm/src/version.js +23 -0
- package/dist/esm/src/version.js.map +1 -0
- package/dist/esm/types/index.d.ts +9 -0
- package/dist/{src → esm/types/src}/const.d.ts +0 -46
- package/dist/{src → esm/types/src}/interface.d.ts +2 -0
- package/dist/{src → esm/types/src}/message-sender.d.ts +0 -18
- package/dist/{src → esm/types/src}/openclaw-compat.d.ts +0 -49
- package/dist/esm/types/src/state-dir-resolve.d.ts +2 -0
- package/dist/{src → esm/types/src}/utils.d.ts +0 -39
- package/dist/esm/types/src/version.d.ts +2 -0
- package/dist/index.d.ts +4 -2
- package/package.json +7 -15
- package/dist/index.js +0 -113
- package/dist/src/accounts.d.ts +0 -57
- package/dist/src/accounts.js +0 -222
- package/dist/src/agent/api-client.d.ts +0 -94
- package/dist/src/agent/api-client.js +0 -292
- package/dist/src/agent/handler.d.ts +0 -63
- package/dist/src/agent/handler.js +0 -532
- package/dist/src/agent/index.d.ts +0 -5
- package/dist/src/agent/index.js +0 -5
- package/dist/src/agent/webhook.d.ts +0 -23
- package/dist/src/agent/webhook.js +0 -226
- package/dist/src/agent/xml.d.ts +0 -21
- package/dist/src/agent/xml.js +0 -41
- package/dist/src/channel.js +0 -564
- package/dist/src/chat-queue.d.ts +0 -31
- package/dist/src/chat-queue.js +0 -45
- package/dist/src/http.d.ts +0 -26
- package/dist/src/http.js +0 -93
- package/dist/src/interface.js +0 -4
- package/dist/src/mcp/index.js +0 -6
- package/dist/src/mcp/interceptors/types.js +0 -7
- package/dist/src/shared/command-auth.d.ts +0 -23
- package/dist/src/shared/command-auth.js +0 -73
- package/dist/src/shared/xml-parser.d.ts +0 -46
- package/dist/src/shared/xml-parser.js +0 -210
- package/dist/src/target.d.ts +0 -35
- package/dist/src/target.js +0 -67
- package/dist/src/template-card-manager.d.ts +0 -55
- package/dist/src/template-card-manager.js +0 -205
- package/dist/src/types/account.d.ts +0 -29
- package/dist/src/types/account.js +0 -4
- package/dist/src/types/config.d.ts +0 -80
- package/dist/src/types/config.js +0 -7
- package/dist/src/types/constants.d.ts +0 -42
- package/dist/src/types/constants.js +0 -42
- package/dist/src/types/index.d.ts +0 -7
- package/dist/src/types/index.js +0 -5
- package/dist/src/types/message.d.ts +0 -238
- package/dist/src/types/message.js +0 -5
- package/dist/src/version.d.ts +0 -2
- package/dist/src/version.js +0 -24
- package/dist/src/webhook/command-auth.d.ts +0 -47
- package/dist/src/webhook/command-auth.js +0 -96
- package/dist/src/webhook/gateway.d.ts +0 -36
- package/dist/src/webhook/gateway.js +0 -227
- package/dist/src/webhook/handler.d.ts +0 -18
- package/dist/src/webhook/handler.js +0 -420
- package/dist/src/webhook/helpers.d.ts +0 -156
- package/dist/src/webhook/helpers.js +0 -833
- package/dist/src/webhook/http.d.ts +0 -26
- package/dist/src/webhook/http.js +0 -94
- package/dist/src/webhook/index.d.ts +0 -11
- package/dist/src/webhook/index.js +0 -14
- package/dist/src/webhook/media.d.ts +0 -29
- package/dist/src/webhook/media.js +0 -88
- package/dist/src/webhook/monitor.d.ts +0 -59
- package/dist/src/webhook/monitor.js +0 -1147
- package/dist/src/webhook/state.d.ts +0 -220
- package/dist/src/webhook/state.js +0 -491
- package/dist/src/webhook/target.d.ts +0 -41
- package/dist/src/webhook/target.js +0 -132
- package/dist/src/webhook/types.d.ts +0 -347
- package/dist/src/webhook/types.js +0 -33
- /package/dist/{src → esm/types/src}/channel.d.ts +0 -0
- /package/dist/{src → esm/types/src}/dm-policy.d.ts +0 -0
- /package/dist/{src → esm/types/src}/group-policy.d.ts +0 -0
- /package/dist/{src → esm/types/src}/mcp/index.d.ts +0 -0
- /package/dist/{src → esm/types/src}/mcp/interceptors/biz-error.d.ts +0 -0
- /package/dist/{src → esm/types/src}/mcp/interceptors/index.d.ts +0 -0
- /package/dist/{src → esm/types/src}/mcp/interceptors/msg-media.d.ts +0 -0
- /package/dist/{src → esm/types/src}/mcp/interceptors/smartpage-create.d.ts +0 -0
- /package/dist/{src → esm/types/src}/mcp/interceptors/smartpage-export.d.ts +0 -0
- /package/dist/{src → esm/types/src}/mcp/interceptors/types.d.ts +0 -0
- /package/dist/{src → esm/types/src}/mcp/schema.d.ts +0 -0
- /package/dist/{src → esm/types/src}/mcp/tool.d.ts +0 -0
- /package/dist/{src → esm/types/src}/mcp/transport.d.ts +0 -0
- /package/dist/{src → esm/types/src}/media-handler.d.ts +0 -0
- /package/dist/{src → esm/types/src}/media-uploader.d.ts +0 -0
- /package/dist/{src → esm/types/src}/message-parser.d.ts +0 -0
- /package/dist/{src → esm/types/src}/monitor.d.ts +0 -0
- /package/dist/{src → esm/types/src}/onboarding.d.ts +0 -0
- /package/dist/{src → esm/types/src}/reqid-store.d.ts +0 -0
- /package/dist/{src → esm/types/src}/runtime.d.ts +0 -0
- /package/dist/{src → esm/types/src}/state-manager.d.ts +0 -0
- /package/dist/{src → esm/types/src}/template-card-parser.d.ts +0 -0
- /package/dist/{src → esm/types/src}/timeout.d.ts +0 -0
|
@@ -0,0 +1,208 @@
|
|
|
1
|
+
'use strict';
|
|
2
|
+
|
|
3
|
+
var transport = require('./transport.js');
|
|
4
|
+
var schema = require('./schema.js');
|
|
5
|
+
var index = require('./interceptors/index.js');
|
|
6
|
+
|
|
7
|
+
/**
|
|
8
|
+
* wecom_mcp — 模拟 MCP 调用的 Agent Tool
|
|
9
|
+
*
|
|
10
|
+
* 通过 MCP Streamable HTTP 传输协议调用企业微信 MCP Server,
|
|
11
|
+
* 提供 list(列出所有工具)和 call(调用工具)两个操作。
|
|
12
|
+
*
|
|
13
|
+
* 在 skills 中的使用方式:
|
|
14
|
+
* wecom_mcp list <category>
|
|
15
|
+
* wecom_mcp call <category> <method> '<jsonArgs>'
|
|
16
|
+
*
|
|
17
|
+
* 示例:
|
|
18
|
+
* wecom_mcp list contact
|
|
19
|
+
* wecom_mcp call contact getContact '{}'
|
|
20
|
+
*/
|
|
21
|
+
// ============================================================================
|
|
22
|
+
// 响应构造辅助
|
|
23
|
+
// ============================================================================
|
|
24
|
+
/** 构造统一的文本响应结构 */
|
|
25
|
+
const textResult = (data) => ({
|
|
26
|
+
content: [{ type: "text", text: JSON.stringify(data, null, 2) }],
|
|
27
|
+
});
|
|
28
|
+
/** 构造错误响应 */
|
|
29
|
+
const errorResult = (err) => {
|
|
30
|
+
// 适配企业微信 API 返回的 { errcode, errmsg } 结构
|
|
31
|
+
if (err && typeof err === "object" && "errcode" in err) {
|
|
32
|
+
const { errcode, errmsg } = err;
|
|
33
|
+
return textResult({ error: errmsg ?? `错误码: ${errcode}`, errcode });
|
|
34
|
+
}
|
|
35
|
+
const message = err instanceof Error ? err.message : String(err);
|
|
36
|
+
return textResult({ error: message });
|
|
37
|
+
};
|
|
38
|
+
// ============================================================================
|
|
39
|
+
// list 操作:列出某品类的所有 MCP 工具
|
|
40
|
+
// ============================================================================
|
|
41
|
+
const handleList = async (category) => {
|
|
42
|
+
const result = await transport.sendJsonRpc(category, "tools/list");
|
|
43
|
+
const tools = result?.tools ?? [];
|
|
44
|
+
if (tools.length === 0) {
|
|
45
|
+
return { message: `品类 "${category}" 下暂无可用工具`, tools: [] };
|
|
46
|
+
}
|
|
47
|
+
return {
|
|
48
|
+
category,
|
|
49
|
+
count: tools.length,
|
|
50
|
+
tools: tools.map((t) => ({
|
|
51
|
+
name: t.name,
|
|
52
|
+
description: t.description ?? "",
|
|
53
|
+
// 清洗 inputSchema,内联 $ref/$defs 引用并移除 Gemini 不支持的关键词,
|
|
54
|
+
// 避免 Gemini 模型解析 function response 时报 400 错误
|
|
55
|
+
inputSchema: t.inputSchema ? schema.cleanSchemaForGemini(t.inputSchema) : undefined,
|
|
56
|
+
})),
|
|
57
|
+
};
|
|
58
|
+
};
|
|
59
|
+
// ============================================================================
|
|
60
|
+
// call 操作:调用某品类的某个 MCP 工具
|
|
61
|
+
// ============================================================================
|
|
62
|
+
const handleCall = async (category, method, args) => {
|
|
63
|
+
const ctx = { category, method, args };
|
|
64
|
+
const callStart = performance.now();
|
|
65
|
+
console.log(`[mcp] handleCall ${category}/${method} 入参: ${JSON.stringify(args)}`);
|
|
66
|
+
// 1. 收集拦截器的 beforeCall 配置(如超时时间、替换 args)
|
|
67
|
+
const { options, args: resolvedArgs } = await index.resolveBeforeCall(ctx);
|
|
68
|
+
const finalArgs = resolvedArgs ?? args;
|
|
69
|
+
if (resolvedArgs) {
|
|
70
|
+
console.log(`[mcp] handleCall ${category}/${method} 拦截器替换 args: ${JSON.stringify(resolvedArgs).slice(0, 500)}` +
|
|
71
|
+
(JSON.stringify(resolvedArgs).length > 500 ? "...(truncated)" : ""));
|
|
72
|
+
}
|
|
73
|
+
if (options) {
|
|
74
|
+
console.log(`[mcp] handleCall ${category}/${method} 拦截器选项: ${JSON.stringify(options)}`);
|
|
75
|
+
}
|
|
76
|
+
// 2. 执行 MCP 调用
|
|
77
|
+
const result = await transport.sendJsonRpc(category, "tools/call", {
|
|
78
|
+
name: method,
|
|
79
|
+
arguments: finalArgs,
|
|
80
|
+
}, options);
|
|
81
|
+
const rpcDone = performance.now();
|
|
82
|
+
const rpcMs = (rpcDone - callStart).toFixed(1);
|
|
83
|
+
const resultStr = JSON.stringify(result);
|
|
84
|
+
console.log(`[mcp] handleCall ${category}/${method} MCP 响应 (${rpcMs}ms): ${resultStr.slice(0, 800)}` +
|
|
85
|
+
(resultStr.length > 800 ? "...(truncated)" : ""));
|
|
86
|
+
// 3. 管道式执行 afterCall 拦截器(业务错误码检查、响应变换等)
|
|
87
|
+
const finalResult = await index.runAfterCall(ctx, result);
|
|
88
|
+
const totalMs = (performance.now() - callStart).toFixed(1);
|
|
89
|
+
const interceptMs = (performance.now() - rpcDone).toFixed(1);
|
|
90
|
+
// 有拦截器处理时打印详细耗时,否则只打印 RPC 耗时
|
|
91
|
+
if (finalResult !== result) {
|
|
92
|
+
const finalStr = JSON.stringify(finalResult);
|
|
93
|
+
console.log(`[mcp] handleCall ${category}/${method} afterCall 变换后 (${interceptMs}ms): ${finalStr.slice(0, 500)}` +
|
|
94
|
+
(finalStr.length > 500 ? "...(truncated)" : ""));
|
|
95
|
+
console.log(`[mcp] handleCall ${category}/${method} 总耗时: ${totalMs}ms` +
|
|
96
|
+
` (MCP请求: ${rpcMs}ms, 拦截处理: ${interceptMs}ms)`);
|
|
97
|
+
}
|
|
98
|
+
else {
|
|
99
|
+
console.log(`[mcp] handleCall ${category}/${method} 耗时: ${rpcMs}ms`);
|
|
100
|
+
}
|
|
101
|
+
return finalResult;
|
|
102
|
+
};
|
|
103
|
+
// ============================================================================
|
|
104
|
+
// 参数解析
|
|
105
|
+
// ============================================================================
|
|
106
|
+
/**
|
|
107
|
+
* 解析 args 参数:支持 JSON 字符串或直接的对象
|
|
108
|
+
*/
|
|
109
|
+
const parseArgs = (args) => {
|
|
110
|
+
if (!args)
|
|
111
|
+
return {};
|
|
112
|
+
if (typeof args === "object")
|
|
113
|
+
return args;
|
|
114
|
+
try {
|
|
115
|
+
return JSON.parse(args);
|
|
116
|
+
}
|
|
117
|
+
catch (err) {
|
|
118
|
+
const detail = err instanceof SyntaxError ? err.message : String(err);
|
|
119
|
+
throw new Error(`args 参数不是合法的 JSON: ${args} (${detail})`);
|
|
120
|
+
}
|
|
121
|
+
};
|
|
122
|
+
// ============================================================================
|
|
123
|
+
// 工具定义 & 导出
|
|
124
|
+
// ============================================================================
|
|
125
|
+
/**
|
|
126
|
+
* 创建 wecom_mcp Agent Tool 定义
|
|
127
|
+
*/
|
|
128
|
+
function createWeComMcpTool() {
|
|
129
|
+
return {
|
|
130
|
+
name: "wecom_mcp",
|
|
131
|
+
label: "企业微信 MCP 工具",
|
|
132
|
+
description: [
|
|
133
|
+
"通过 HTTP 直接调用企业微信 MCP Server。",
|
|
134
|
+
"支持两种操作:",
|
|
135
|
+
" - list: 列出指定品类的所有 MCP 工具",
|
|
136
|
+
" - call: 调用指定品类的某个 MCP 工具",
|
|
137
|
+
"",
|
|
138
|
+
"使用方式:",
|
|
139
|
+
" wecom_mcp list <category>",
|
|
140
|
+
" wecom_mcp call <category> <method> '<jsonArgs>'",
|
|
141
|
+
"",
|
|
142
|
+
"示例:",
|
|
143
|
+
" 列出 contact 品类所有工具:wecom_mcp list contact",
|
|
144
|
+
" 调用 contact 的 getContact:wecom_mcp call contact getContact '{}'",
|
|
145
|
+
].join("\n"),
|
|
146
|
+
parameters: {
|
|
147
|
+
type: "object",
|
|
148
|
+
properties: {
|
|
149
|
+
action: {
|
|
150
|
+
type: "string",
|
|
151
|
+
enum: ["list", "call"],
|
|
152
|
+
description: "操作类型:list(列出工具)或 call(调用工具)",
|
|
153
|
+
},
|
|
154
|
+
category: {
|
|
155
|
+
type: "string",
|
|
156
|
+
description: "MCP 品类名称,如 doc、contact 等,对应 mcpConfig 中的 key",
|
|
157
|
+
},
|
|
158
|
+
method: {
|
|
159
|
+
type: "string",
|
|
160
|
+
description: "要调用的 MCP 方法名(action=call 时必填)",
|
|
161
|
+
},
|
|
162
|
+
args: {
|
|
163
|
+
type: ["string", "object"],
|
|
164
|
+
description: "调用 MCP 方法的参数,可以是 JSON 字符串或对象(action=call 时使用,默认 {})",
|
|
165
|
+
},
|
|
166
|
+
},
|
|
167
|
+
required: ["action", "category"],
|
|
168
|
+
},
|
|
169
|
+
async execute(_toolCallId, params) {
|
|
170
|
+
const p = params;
|
|
171
|
+
console.log(`[mcp] execute: action=${p.action}, category=${p.category}` +
|
|
172
|
+
(p.method ? `, method=${p.method}` : "") +
|
|
173
|
+
(p.args ? `, args=${typeof p.args === "string" ? p.args : JSON.stringify(p.args)}` : ""));
|
|
174
|
+
try {
|
|
175
|
+
let result;
|
|
176
|
+
switch (p.action) {
|
|
177
|
+
case "list":
|
|
178
|
+
result = textResult(await handleList(p.category));
|
|
179
|
+
break;
|
|
180
|
+
case "call": {
|
|
181
|
+
if (!p.method) {
|
|
182
|
+
result = textResult({ error: "action 为 call 时必须提供 method 参数" });
|
|
183
|
+
break;
|
|
184
|
+
}
|
|
185
|
+
const args = parseArgs(p.args);
|
|
186
|
+
result = textResult(await handleCall(p.category, p.method, args));
|
|
187
|
+
break;
|
|
188
|
+
}
|
|
189
|
+
default:
|
|
190
|
+
result = textResult({ error: `未知操作类型: ${String(p.action)},支持 list 和 call` });
|
|
191
|
+
}
|
|
192
|
+
console.log(`[mcp] execute: action=${p.action}, category=${p.category}` +
|
|
193
|
+
(p.method ? `, method=${p.method}` : "") +
|
|
194
|
+
` → 响应长度=${result.content[0].text.length} chars`);
|
|
195
|
+
return result;
|
|
196
|
+
}
|
|
197
|
+
catch (err) {
|
|
198
|
+
console.error(`[mcp] execute: action=${p.action}, category=${p.category}` +
|
|
199
|
+
(p.method ? `, method=${p.method}` : "") +
|
|
200
|
+
` → 异常: ${err instanceof Error ? err.message : String(err)}`);
|
|
201
|
+
return errorResult(err);
|
|
202
|
+
}
|
|
203
|
+
},
|
|
204
|
+
};
|
|
205
|
+
}
|
|
206
|
+
|
|
207
|
+
exports.createWeComMcpTool = createWeComMcpTool;
|
|
208
|
+
//# sourceMappingURL=tool.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tool.js","sources":["../../../../../../src/mcp/tool.ts"],"sourcesContent":["/**\n * wecom_mcp — 模拟 MCP 调用的 Agent Tool\n *\n * 通过 MCP Streamable HTTP 传输协议调用企业微信 MCP Server,\n * 提供 list(列出所有工具)和 call(调用工具)两个操作。\n *\n * 在 skills 中的使用方式:\n * wecom_mcp list <category>\n * wecom_mcp call <category> <method> '<jsonArgs>'\n *\n * 示例:\n * wecom_mcp list contact\n * wecom_mcp call contact getContact '{}'\n */\n\nimport { sendJsonRpc, type McpToolInfo } from \"./transport.js\";\nimport { cleanSchemaForGemini } from \"./schema.js\";\nimport { resolveBeforeCall, runAfterCall } from \"./interceptors/index.js\";\n\n// ============================================================================\n// 类型定义\n// ============================================================================\n\n/** wecom_mcp 的入参 */\ninterface WeComToolsParams {\n /** 操作类型:list | call */\n action: \"list\" | \"call\";\n /** MCP 品类,对应 mcpConfig 中的 key,如 doc、contact */\n category: string;\n /** 调用的 MCP 方法名(action=call 时必填) */\n method?: string;\n /** 调用 MCP 方法的 JSON 参数(action=call 时使用) */\n args?: string | Record<string, unknown>;\n}\n\n// ============================================================================\n// 响应构造辅助\n// ============================================================================\n\n/** 构造统一的文本响应结构 */\nconst textResult = (data: unknown) => ({\n content: [{ type: \"text\" as const, text: JSON.stringify(data, null, 2) }],\n});\n\n/** 构造错误响应 */\nconst errorResult = (err: unknown) => {\n // 适配企业微信 API 返回的 { errcode, errmsg } 结构\n if (err && typeof err === \"object\" && \"errcode\" in err) {\n const { errcode, errmsg } = err as { errcode: number; errmsg?: string };\n return textResult({ error: errmsg ?? `错误码: ${errcode}`, errcode });\n }\n\n const message = err instanceof Error ? err.message : String(err);\n return textResult({ error: message });\n};\n\n// ============================================================================\n// list 操作:列出某品类的所有 MCP 工具\n// ============================================================================\n\nconst handleList = async (category: string): Promise<unknown> => {\n const result = await sendJsonRpc(category, \"tools/list\") as { tools?: McpToolInfo[] } | undefined;\n\n const tools = result?.tools ?? [];\n if (tools.length === 0) {\n return { message: `品类 \"${category}\" 下暂无可用工具`, tools: [] };\n }\n\n return {\n category,\n count: tools.length,\n tools: tools.map((t) => ({\n name: t.name,\n description: t.description ?? \"\",\n // 清洗 inputSchema,内联 $ref/$defs 引用并移除 Gemini 不支持的关键词,\n // 避免 Gemini 模型解析 function response 时报 400 错误\n inputSchema: t.inputSchema ? cleanSchemaForGemini(t.inputSchema) : undefined,\n })),\n };\n};\n\n// ============================================================================\n// call 操作:调用某品类的某个 MCP 工具\n// ============================================================================\n\nconst handleCall = async (\n category: string,\n method: string,\n args: Record<string, unknown>,\n): Promise<unknown> => {\n const ctx = { category, method, args };\n const callStart = performance.now();\n\n console.log(`[mcp] handleCall ${category}/${method} 入参: ${JSON.stringify(args)}`);\n\n // 1. 收集拦截器的 beforeCall 配置(如超时时间、替换 args)\n const { options, args: resolvedArgs } = await resolveBeforeCall(ctx);\n const finalArgs = resolvedArgs ?? args;\n\n if (resolvedArgs) {\n console.log(\n `[mcp] handleCall ${category}/${method} 拦截器替换 args: ${JSON.stringify(resolvedArgs).slice(0, 500)}` +\n (JSON.stringify(resolvedArgs).length > 500 ? \"...(truncated)\" : \"\"),\n );\n }\n if (options) {\n console.log(`[mcp] handleCall ${category}/${method} 拦截器选项: ${JSON.stringify(options)}`);\n }\n\n // 2. 执行 MCP 调用\n const result = await sendJsonRpc(category, \"tools/call\", {\n name: method,\n arguments: finalArgs,\n }, options);\n\n const rpcDone = performance.now();\n const rpcMs = (rpcDone - callStart).toFixed(1);\n\n const resultStr = JSON.stringify(result);\n console.log(\n `[mcp] handleCall ${category}/${method} MCP 响应 (${rpcMs}ms): ${resultStr.slice(0, 800)}` +\n (resultStr.length > 800 ? \"...(truncated)\" : \"\"),\n );\n\n // 3. 管道式执行 afterCall 拦截器(业务错误码检查、响应变换等)\n const finalResult = await runAfterCall(ctx, result);\n\n const totalMs = (performance.now() - callStart).toFixed(1);\n const interceptMs = (performance.now() - rpcDone).toFixed(1);\n\n // 有拦截器处理时打印详细耗时,否则只打印 RPC 耗时\n if (finalResult !== result) {\n const finalStr = JSON.stringify(finalResult);\n console.log(\n `[mcp] handleCall ${category}/${method} afterCall 变换后 (${interceptMs}ms): ${finalStr.slice(0, 500)}` +\n (finalStr.length > 500 ? \"...(truncated)\" : \"\"),\n );\n console.log(\n `[mcp] handleCall ${category}/${method} 总耗时: ${totalMs}ms` +\n ` (MCP请求: ${rpcMs}ms, 拦截处理: ${interceptMs}ms)`,\n );\n } else {\n console.log(`[mcp] handleCall ${category}/${method} 耗时: ${rpcMs}ms`);\n }\n\n return finalResult;\n};\n\n// ============================================================================\n// 参数解析\n// ============================================================================\n\n/**\n * 解析 args 参数:支持 JSON 字符串或直接的对象\n */\nconst parseArgs = (args: string | Record<string, unknown> | undefined): Record<string, unknown> => {\n if (!args) return {};\n if (typeof args === \"object\") return args;\n try {\n return JSON.parse(args) as Record<string, unknown>;\n } catch (err) {\n const detail = err instanceof SyntaxError ? err.message : String(err);\n throw new Error(`args 参数不是合法的 JSON: ${args} (${detail})`);\n }\n};\n\n// ============================================================================\n// 工具定义 & 导出\n// ============================================================================\n\n/**\n * 创建 wecom_mcp Agent Tool 定义\n */\nexport function createWeComMcpTool() {\n return {\n name: \"wecom_mcp\",\n label: \"企业微信 MCP 工具\",\n description: [\n \"通过 HTTP 直接调用企业微信 MCP Server。\",\n \"支持两种操作:\",\n \" - list: 列出指定品类的所有 MCP 工具\",\n \" - call: 调用指定品类的某个 MCP 工具\",\n \"\",\n \"使用方式:\",\n \" wecom_mcp list <category>\",\n \" wecom_mcp call <category> <method> '<jsonArgs>'\",\n \"\",\n \"示例:\",\n \" 列出 contact 品类所有工具:wecom_mcp list contact\",\n \" 调用 contact 的 getContact:wecom_mcp call contact getContact '{}'\",\n ].join(\"\\n\"),\n parameters: {\n type: \"object\" as const,\n properties: {\n action: {\n type: \"string\",\n enum: [\"list\", \"call\"],\n description: \"操作类型:list(列出工具)或 call(调用工具)\",\n },\n category: {\n type: \"string\",\n description: \"MCP 品类名称,如 doc、contact 等,对应 mcpConfig 中的 key\",\n },\n method: {\n type: \"string\",\n description: \"要调用的 MCP 方法名(action=call 时必填)\",\n },\n args: {\n type: [\"string\", \"object\"],\n description: \"调用 MCP 方法的参数,可以是 JSON 字符串或对象(action=call 时使用,默认 {})\",\n },\n },\n required: [\"action\", \"category\"],\n },\n async execute(_toolCallId: string, params: unknown) {\n const p = params as WeComToolsParams;\n console.log(\n `[mcp] execute: action=${p.action}, category=${p.category}` +\n (p.method ? `, method=${p.method}` : \"\") +\n (p.args ? `, args=${typeof p.args === \"string\" ? p.args : JSON.stringify(p.args)}` : \"\"),\n );\n try {\n let result: ReturnType<typeof textResult>;\n switch (p.action) {\n case \"list\":\n result = textResult(await handleList(p.category));\n break;\n case \"call\": {\n if (!p.method) {\n result = textResult({ error: \"action 为 call 时必须提供 method 参数\" });\n break;\n }\n const args = parseArgs(p.args);\n result = textResult(await handleCall(p.category, p.method, args));\n break;\n }\n default:\n result = textResult({ error: `未知操作类型: ${String(p.action)},支持 list 和 call` });\n }\n console.log(\n `[mcp] execute: action=${p.action}, category=${p.category}` +\n (p.method ? `, method=${p.method}` : \"\") +\n ` → 响应长度=${result.content[0].text.length} chars`,\n );\n return result;\n } catch (err) {\n console.error(\n `[mcp] execute: action=${p.action}, category=${p.category}` +\n (p.method ? `, method=${p.method}` : \"\") +\n ` → 异常: ${err instanceof Error ? err.message : String(err)}`,\n );\n return errorResult(err);\n }\n },\n };\n}\n"],"names":["sendJsonRpc","cleanSchemaForGemini","resolveBeforeCall","runAfterCall"],"mappings":";;;;;;AAAA;;;;;;;;;;;;;AAaG;AAsBH;AACA;AACA;AAEA;AACA,MAAM,UAAU,GAAG,CAAC,IAAa,MAAM;IACrC,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC;AAC1E,CAAA,CAAC;AAEF;AACA,MAAM,WAAW,GAAG,CAAC,GAAY,KAAI;;IAEnC,IAAI,GAAG,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,SAAS,IAAI,GAAG,EAAE;AACtD,QAAA,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,GAA2C;AACvE,QAAA,OAAO,UAAU,CAAC,EAAE,KAAK,EAAE,MAAM,IAAI,CAAA,KAAA,EAAQ,OAAO,CAAA,CAAE,EAAE,OAAO,EAAE,CAAC;IACpE;AAEA,IAAA,MAAM,OAAO,GAAG,GAAG,YAAY,KAAK,GAAG,GAAG,CAAC,OAAO,GAAG,MAAM,CAAC,GAAG,CAAC;IAChE,OAAO,UAAU,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC;AACvC,CAAC;AAED;AACA;AACA;AAEA,MAAM,UAAU,GAAG,OAAO,QAAgB,KAAsB;IAC9D,MAAM,MAAM,GAAG,MAAMA,qBAAW,CAAC,QAAQ,EAAE,YAAY,CAA0C;AAEjG,IAAA,MAAM,KAAK,GAAG,MAAM,EAAE,KAAK,IAAI,EAAE;AACjC,IAAA,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE;QACtB,OAAO,EAAE,OAAO,EAAE,CAAA,IAAA,EAAO,QAAQ,CAAA,SAAA,CAAW,EAAE,KAAK,EAAE,EAAE,EAAE;IAC3D;IAEA,OAAO;QACL,QAAQ;QACR,KAAK,EAAE,KAAK,CAAC,MAAM;QACnB,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM;YACvB,IAAI,EAAE,CAAC,CAAC,IAAI;AACZ,YAAA,WAAW,EAAE,CAAC,CAAC,WAAW,IAAI,EAAE;;;AAGhC,YAAA,WAAW,EAAE,CAAC,CAAC,WAAW,GAAGC,2BAAoB,CAAC,CAAC,CAAC,WAAW,CAAC,GAAG,SAAS;AAC7E,SAAA,CAAC,CAAC;KACJ;AACH,CAAC;AAED;AACA;AACA;AAEA,MAAM,UAAU,GAAG,OACjB,QAAgB,EAChB,MAAc,EACd,IAA6B,KACT;IACpB,MAAM,GAAG,GAAG,EAAE,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE;AACtC,IAAA,MAAM,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE;AAEnC,IAAA,OAAO,CAAC,GAAG,CAAC,CAAA,iBAAA,EAAoB,QAAQ,IAAI,MAAM,CAAA,KAAA,EAAQ,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAA,CAAE,CAAC;;AAGjF,IAAA,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,YAAY,EAAE,GAAG,MAAMC,uBAAiB,CAAC,GAAG,CAAC;AACpE,IAAA,MAAM,SAAS,GAAG,YAAY,IAAI,IAAI;IAEtC,IAAI,YAAY,EAAE;QAChB,OAAO,CAAC,GAAG,CACT,CAAA,iBAAA,EAAoB,QAAQ,CAAA,CAAA,EAAI,MAAM,gBAAgB,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAA,CAAE;aACjG,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC,MAAM,GAAG,GAAG,GAAG,gBAAgB,GAAG,EAAE,CAAC,CACpE;IACH;IACA,IAAI,OAAO,EAAE;AACX,QAAA,OAAO,CAAC,GAAG,CAAC,CAAA,iBAAA,EAAoB,QAAQ,IAAI,MAAM,CAAA,QAAA,EAAW,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAA,CAAE,CAAC;IACzF;;IAGA,MAAM,MAAM,GAAG,MAAMF,qBAAW,CAAC,QAAQ,EAAE,YAAY,EAAE;AACvD,QAAA,IAAI,EAAE,MAAM;AACZ,QAAA,SAAS,EAAE,SAAS;KACrB,EAAE,OAAO,CAAC;AAEX,IAAA,MAAM,OAAO,GAAG,WAAW,CAAC,GAAG,EAAE;AACjC,IAAA,MAAM,KAAK,GAAG,CAAC,OAAO,GAAG,SAAS,EAAE,OAAO,CAAC,CAAC,CAAC;IAE9C,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;AACxC,IAAA,OAAO,CAAC,GAAG,CACT,oBAAoB,QAAQ,CAAA,CAAA,EAAI,MAAM,CAAA,SAAA,EAAY,KAAK,CAAA,KAAA,EAAQ,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAA,CAAE;AACxF,SAAC,SAAS,CAAC,MAAM,GAAG,GAAG,GAAG,gBAAgB,GAAG,EAAE,CAAC,CACjD;;IAGD,MAAM,WAAW,GAAG,MAAMG,kBAAY,CAAC,GAAG,EAAE,MAAM,CAAC;AAEnD,IAAA,MAAM,OAAO,GAAG,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,SAAS,EAAE,OAAO,CAAC,CAAC,CAAC;AAC1D,IAAA,MAAM,WAAW,GAAG,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;;AAG5D,IAAA,IAAI,WAAW,KAAK,MAAM,EAAE;QAC1B,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC;AAC5C,QAAA,OAAO,CAAC,GAAG,CACT,oBAAoB,QAAQ,CAAA,CAAA,EAAI,MAAM,CAAA,gBAAA,EAAmB,WAAW,CAAA,KAAA,EAAQ,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAA,CAAE;AACpG,aAAC,QAAQ,CAAC,MAAM,GAAG,GAAG,GAAG,gBAAgB,GAAG,EAAE,CAAC,CAChD;QACD,OAAO,CAAC,GAAG,CACT,CAAA,iBAAA,EAAoB,QAAQ,CAAA,CAAA,EAAI,MAAM,CAAA,MAAA,EAAS,OAAO,CAAA,EAAA,CAAI;AAC1D,YAAA,CAAA,SAAA,EAAY,KAAK,CAAA,UAAA,EAAa,WAAW,CAAA,GAAA,CAAK,CAC/C;IACH;SAAO;QACL,OAAO,CAAC,GAAG,CAAC,CAAA,iBAAA,EAAoB,QAAQ,CAAA,CAAA,EAAI,MAAM,CAAA,KAAA,EAAQ,KAAK,CAAA,EAAA,CAAI,CAAC;IACtE;AAEA,IAAA,OAAO,WAAW;AACpB,CAAC;AAED;AACA;AACA;AAEA;;AAEG;AACH,MAAM,SAAS,GAAG,CAAC,IAAkD,KAA6B;AAChG,IAAA,IAAI,CAAC,IAAI;AAAE,QAAA,OAAO,EAAE;IACpB,IAAI,OAAO,IAAI,KAAK,QAAQ;AAAE,QAAA,OAAO,IAAI;AACzC,IAAA,IAAI;AACF,QAAA,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAA4B;IACpD;IAAE,OAAO,GAAG,EAAE;AACZ,QAAA,MAAM,MAAM,GAAG,GAAG,YAAY,WAAW,GAAG,GAAG,CAAC,OAAO,GAAG,MAAM,CAAC,GAAG,CAAC;QACrE,MAAM,IAAI,KAAK,CAAC,CAAA,mBAAA,EAAsB,IAAI,CAAA,EAAA,EAAK,MAAM,CAAA,CAAA,CAAG,CAAC;IAC3D;AACF,CAAC;AAED;AACA;AACA;AAEA;;AAEG;SACa,kBAAkB,GAAA;IAChC,OAAO;AACL,QAAA,IAAI,EAAE,WAAW;AACjB,QAAA,KAAK,EAAE,aAAa;AACpB,QAAA,WAAW,EAAE;YACX,8BAA8B;YAC9B,SAAS;YACT,4BAA4B;YAC5B,4BAA4B;YAC5B,EAAE;YACF,OAAO;YACP,6BAA6B;YAC7B,mDAAmD;YACnD,EAAE;YACF,KAAK;YACL,4CAA4C;YAC5C,kEAAkE;SACnE,CAAC,IAAI,CAAC,IAAI,CAAC;AACZ,QAAA,UAAU,EAAE;AACV,YAAA,IAAI,EAAE,QAAiB;AACvB,YAAA,UAAU,EAAE;AACV,gBAAA,MAAM,EAAE;AACN,oBAAA,IAAI,EAAE,QAAQ;AACd,oBAAA,IAAI,EAAE,CAAC,MAAM,EAAE,MAAM,CAAC;AACtB,oBAAA,WAAW,EAAE,6BAA6B;AAC3C,iBAAA;AACD,gBAAA,QAAQ,EAAE;AACR,oBAAA,IAAI,EAAE,QAAQ;AACd,oBAAA,WAAW,EAAE,8CAA8C;AAC5D,iBAAA;AACD,gBAAA,MAAM,EAAE;AACN,oBAAA,IAAI,EAAE,QAAQ;AACd,oBAAA,WAAW,EAAE,+BAA+B;AAC7C,iBAAA;AACD,gBAAA,IAAI,EAAE;AACJ,oBAAA,IAAI,EAAE,CAAC,QAAQ,EAAE,QAAQ,CAAC;AAC1B,oBAAA,WAAW,EAAE,qDAAqD;AACnE,iBAAA;AACF,aAAA;AACD,YAAA,QAAQ,EAAE,CAAC,QAAQ,EAAE,UAAU,CAAC;AACjC,SAAA;AACD,QAAA,MAAM,OAAO,CAAC,WAAmB,EAAE,MAAe,EAAA;YAChD,MAAM,CAAC,GAAG,MAA0B;YACpC,OAAO,CAAC,GAAG,CACT,CAAA,sBAAA,EAAyB,CAAC,CAAC,MAAM,CAAA,WAAA,EAAc,CAAC,CAAC,QAAQ,CAAA,CAAE;AAC3D,iBAAC,CAAC,CAAC,MAAM,GAAG,CAAA,SAAA,EAAY,CAAC,CAAC,MAAM,CAAA,CAAE,GAAG,EAAE,CAAC;AACxC,iBAAC,CAAC,CAAC,IAAI,GAAG,CAAA,OAAA,EAAU,OAAO,CAAC,CAAC,IAAI,KAAK,QAAQ,GAAG,CAAC,CAAC,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC,CACzF;AACD,YAAA,IAAI;AACF,gBAAA,IAAI,MAAqC;AACzC,gBAAA,QAAQ,CAAC,CAAC,MAAM;AACd,oBAAA,KAAK,MAAM;wBACT,MAAM,GAAG,UAAU,CAAC,MAAM,UAAU,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;wBACjD;oBACF,KAAK,MAAM,EAAE;AACX,wBAAA,IAAI,CAAC,CAAC,CAAC,MAAM,EAAE;4BACb,MAAM,GAAG,UAAU,CAAC,EAAE,KAAK,EAAE,+BAA+B,EAAE,CAAC;4BAC/D;wBACF;wBACA,MAAM,IAAI,GAAG,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC;AAC9B,wBAAA,MAAM,GAAG,UAAU,CAAC,MAAM,UAAU,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;wBACjE;oBACF;AACA,oBAAA;AACE,wBAAA,MAAM,GAAG,UAAU,CAAC,EAAE,KAAK,EAAE,CAAA,QAAA,EAAW,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAA,eAAA,CAAiB,EAAE,CAAC;;gBAEhF,OAAO,CAAC,GAAG,CACT,CAAA,sBAAA,EAAyB,CAAC,CAAC,MAAM,CAAA,WAAA,EAAc,CAAC,CAAC,QAAQ,CAAA,CAAE;AAC3D,qBAAC,CAAC,CAAC,MAAM,GAAG,CAAA,SAAA,EAAY,CAAC,CAAC,MAAM,CAAA,CAAE,GAAG,EAAE,CAAC;AACxC,oBAAA,CAAA,QAAA,EAAW,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAA,MAAA,CAAQ,CACjD;AACD,gBAAA,OAAO,MAAM;YACf;YAAE,OAAO,GAAG,EAAE;gBACZ,OAAO,CAAC,KAAK,CACX,CAAA,sBAAA,EAAyB,CAAC,CAAC,MAAM,CAAA,WAAA,EAAc,CAAC,CAAC,QAAQ,CAAA,CAAE;AAC3D,qBAAC,CAAC,CAAC,MAAM,GAAG,CAAA,SAAA,EAAY,CAAC,CAAC,MAAM,CAAA,CAAE,GAAG,EAAE,CAAC;AACxC,oBAAA,CAAA,OAAA,EAAU,GAAG,YAAY,KAAK,GAAG,GAAG,CAAC,OAAO,GAAG,MAAM,CAAC,GAAG,CAAC,CAAA,CAAE,CAC7D;AACD,gBAAA,OAAO,WAAW,CAAC,GAAG,CAAC;YACzB;QACF,CAAC;KACF;AACH;;;;"}
|
|
@@ -0,0 +1,417 @@
|
|
|
1
|
+
'use strict';
|
|
2
|
+
|
|
3
|
+
var aibotNodeSdk = require('@wecom/aibot-node-sdk');
|
|
4
|
+
var openclawCompat = require('../openclaw-compat.js');
|
|
5
|
+
var stateManager = require('../state-manager.js');
|
|
6
|
+
var _const = require('../const.js');
|
|
7
|
+
var timeout = require('../timeout.js');
|
|
8
|
+
var version = require('../version.js');
|
|
9
|
+
|
|
10
|
+
/**
|
|
11
|
+
* MCP Streamable HTTP 传输层模块
|
|
12
|
+
*
|
|
13
|
+
* 负责:
|
|
14
|
+
* - MCP JSON-RPC over HTTP 通信(发送请求、解析响应)
|
|
15
|
+
* - Streamable HTTP session 生命周期管理(initialize 握手 → Mcp-Session-Id 维护 → 失效重建)
|
|
16
|
+
* - 自动检测无状态 Server:如果 initialize 响应未返回 Mcp-Session-Id,
|
|
17
|
+
* 则标记为无状态模式,后续请求跳过握手和 session 管理
|
|
18
|
+
* - SSE 流式响应解析
|
|
19
|
+
* - MCP 配置运行时缓存(通过 WSClient 拉取 URL 并缓存在内存中)
|
|
20
|
+
*/
|
|
21
|
+
// ============================================================================
|
|
22
|
+
// 内部状态
|
|
23
|
+
// ============================================================================
|
|
24
|
+
/** HTTP 请求超时时间(毫秒) */
|
|
25
|
+
const HTTP_REQUEST_TIMEOUT_MS = 30000;
|
|
26
|
+
/** 媒体下载请求超时时间(毫秒),base64 编码的媒体文件最大可达 ~27MB */
|
|
27
|
+
const MEDIA_DOWNLOAD_TIMEOUT_MS = 120000;
|
|
28
|
+
/** 日志前缀 */
|
|
29
|
+
const LOG_TAG = "[mcp]";
|
|
30
|
+
/**
|
|
31
|
+
* MCP JSON-RPC 错误
|
|
32
|
+
*
|
|
33
|
+
* 携带服务端返回的 JSON-RPC error.code,
|
|
34
|
+
* 用于上层按错误码进行差异化处理(如特定错误码触发缓存清理)。
|
|
35
|
+
*/
|
|
36
|
+
class McpRpcError extends Error {
|
|
37
|
+
constructor(code, message, data) {
|
|
38
|
+
super(message);
|
|
39
|
+
this.code = code;
|
|
40
|
+
this.data = data;
|
|
41
|
+
this.name = "McpRpcError";
|
|
42
|
+
}
|
|
43
|
+
}
|
|
44
|
+
/**
|
|
45
|
+
* MCP HTTP 错误
|
|
46
|
+
*
|
|
47
|
+
* 携带 HTTP 状态码,用于精确判断 session 失效(404)等场景,
|
|
48
|
+
* 避免通过字符串匹配 "404" 导致的误判。
|
|
49
|
+
*/
|
|
50
|
+
class McpHttpError extends Error {
|
|
51
|
+
constructor(statusCode, message) {
|
|
52
|
+
super(message);
|
|
53
|
+
this.statusCode = statusCode;
|
|
54
|
+
this.name = "McpHttpError";
|
|
55
|
+
}
|
|
56
|
+
}
|
|
57
|
+
/**
|
|
58
|
+
* 需要清理缓存的 JSON-RPC 错误码集合
|
|
59
|
+
*
|
|
60
|
+
* 当 MCP Server 返回以下错误码时,说明服务端状态已发生变化(如配置变更、
|
|
61
|
+
* 服务重启等),需要清理对应 category 的全部缓存,确保下次请求重新
|
|
62
|
+
* 拉取配置并重建会话。
|
|
63
|
+
*
|
|
64
|
+
* - -32001: 服务不可用(Server Unavailable)
|
|
65
|
+
* - -32002: 配置已变更(Config Changed)
|
|
66
|
+
* - -32003: 认证失败(Auth Failed)
|
|
67
|
+
*/
|
|
68
|
+
const CACHE_CLEAR_ERROR_CODES = new Set([-32001, -32002, -32003]);
|
|
69
|
+
/** MCP 配置缓存:category → response.body(完整配置) */
|
|
70
|
+
const mcpConfigCache = new Map();
|
|
71
|
+
/** Streamable HTTP 会话缓存:category → session */
|
|
72
|
+
const mcpSessionCache = new Map();
|
|
73
|
+
/** 已确认为无状态的 MCP Server 品类集合(跳过后续握手) */
|
|
74
|
+
const statelessCategories = new Set();
|
|
75
|
+
/** 正在进行中的 initialize 请求(防止并发重复初始化),key 为 category */
|
|
76
|
+
const inflightInitRequests = new Map();
|
|
77
|
+
// ============================================================================
|
|
78
|
+
// MCP 配置拉取与缓存
|
|
79
|
+
// ============================================================================
|
|
80
|
+
/**
|
|
81
|
+
* 通过 WSClient 拉取指定 category 的 MCP 完整配置
|
|
82
|
+
*
|
|
83
|
+
* @param category - MCP 品类名称,如 doc、contact
|
|
84
|
+
* @returns 完整的 response.body 配置对象(至少包含 url 字段)
|
|
85
|
+
*/
|
|
86
|
+
async function fetchMcpConfig(category) {
|
|
87
|
+
const wsClient = stateManager.getWeComWebSocket(openclawCompat.DEFAULT_ACCOUNT_ID);
|
|
88
|
+
if (!wsClient) {
|
|
89
|
+
throw new Error("WSClient 未连接,无法拉取 MCP 配置");
|
|
90
|
+
}
|
|
91
|
+
const reqId = aibotNodeSdk.generateReqId("mcp_config");
|
|
92
|
+
const response = await timeout.withTimeout(wsClient.reply({ headers: { req_id: reqId } }, { biz_type: category, plugin_version: version.PLUGIN_VERSION }, _const.MCP_GET_CONFIG_CMD), _const.MCP_CONFIG_FETCH_TIMEOUT_MS, `MCP config fetch for "${category}" timed out after ${_const.MCP_CONFIG_FETCH_TIMEOUT_MS}ms`);
|
|
93
|
+
if (response.errcode !== undefined && response.errcode !== 0) {
|
|
94
|
+
const errMsg = `MCP 配置请求失败: errcode=${response.errcode}, errmsg=${response.errmsg ?? "unknown"}`;
|
|
95
|
+
console.error(`${LOG_TAG} ${errMsg}`);
|
|
96
|
+
throw new Error(errMsg);
|
|
97
|
+
}
|
|
98
|
+
const body = response.body;
|
|
99
|
+
if (!body?.url) {
|
|
100
|
+
throw new Error(`MCP 配置响应缺少 url 字段 (category="${category}")`);
|
|
101
|
+
}
|
|
102
|
+
console.log(`${LOG_TAG} 配置拉取成功 (category="${category}")`);
|
|
103
|
+
return body;
|
|
104
|
+
}
|
|
105
|
+
/**
|
|
106
|
+
* 获取指定品类的 MCP Server URL
|
|
107
|
+
*
|
|
108
|
+
* 优先从内存缓存中读取,未命中时通过 WSClient 拉取并缓存。
|
|
109
|
+
*
|
|
110
|
+
* @param category - MCP 品类名称
|
|
111
|
+
* @returns MCP Server URL
|
|
112
|
+
*/
|
|
113
|
+
async function getMcpUrl(category) {
|
|
114
|
+
// 查内存缓存
|
|
115
|
+
const cached = mcpConfigCache.get(category);
|
|
116
|
+
if (cached)
|
|
117
|
+
return cached.url;
|
|
118
|
+
// 缓存未命中,通过 WSClient 拉取
|
|
119
|
+
const body = await fetchMcpConfig(category);
|
|
120
|
+
// 写入缓存
|
|
121
|
+
mcpConfigCache.set(category, body);
|
|
122
|
+
console.log(`${LOG_TAG} getMcpUrl ${category}: ${body.url}`);
|
|
123
|
+
return body.url;
|
|
124
|
+
}
|
|
125
|
+
// ============================================================================
|
|
126
|
+
// HTTP 底层通信
|
|
127
|
+
// ============================================================================
|
|
128
|
+
/**
|
|
129
|
+
* 发送原始 HTTP 请求到 MCP Server(底层方法)
|
|
130
|
+
*
|
|
131
|
+
* 自动携带 Mcp-Session-Id 请求头(如果有),
|
|
132
|
+
* 并从响应头中更新 sessionId。
|
|
133
|
+
*/
|
|
134
|
+
async function sendRawJsonRpc(url, session, body, timeoutMs = HTTP_REQUEST_TIMEOUT_MS) {
|
|
135
|
+
const controller = new AbortController();
|
|
136
|
+
const timeoutId = setTimeout(() => controller.abort(), timeoutMs);
|
|
137
|
+
const headers = {
|
|
138
|
+
"Content-Type": "application/json",
|
|
139
|
+
Accept: "application/json, text/event-stream",
|
|
140
|
+
};
|
|
141
|
+
// Streamable HTTP:携带会话 ID
|
|
142
|
+
if (session.sessionId) {
|
|
143
|
+
headers["Mcp-Session-Id"] = session.sessionId;
|
|
144
|
+
}
|
|
145
|
+
let response;
|
|
146
|
+
try {
|
|
147
|
+
response = await fetch(url, {
|
|
148
|
+
method: "POST",
|
|
149
|
+
headers,
|
|
150
|
+
body: JSON.stringify(body),
|
|
151
|
+
signal: controller.signal,
|
|
152
|
+
});
|
|
153
|
+
}
|
|
154
|
+
catch (err) {
|
|
155
|
+
if (err instanceof DOMException && err.name === "AbortError") {
|
|
156
|
+
throw new Error(`MCP 请求超时 (${timeoutMs}ms)`);
|
|
157
|
+
}
|
|
158
|
+
throw new Error(`MCP 网络请求失败: ${err instanceof Error ? err.message : String(err)}`);
|
|
159
|
+
}
|
|
160
|
+
finally {
|
|
161
|
+
clearTimeout(timeoutId);
|
|
162
|
+
}
|
|
163
|
+
// 从响应头提取新的 sessionId(不直接修改入参,由调用方决定如何更新)
|
|
164
|
+
const newSessionId = response.headers.get("mcp-session-id");
|
|
165
|
+
if (!response.ok) {
|
|
166
|
+
throw new McpHttpError(response.status, `MCP HTTP 请求失败: ${response.status} ${response.statusText}`);
|
|
167
|
+
}
|
|
168
|
+
// Streamable HTTP:notification 响应可能无响应体(204 或 content-length: 0)
|
|
169
|
+
const contentLength = response.headers.get("content-length");
|
|
170
|
+
if (response.status === 204 || contentLength === "0") {
|
|
171
|
+
return { response, rpcResult: undefined, newSessionId };
|
|
172
|
+
}
|
|
173
|
+
const contentType = response.headers.get("content-type") ?? "";
|
|
174
|
+
// 处理 SSE 流式响应
|
|
175
|
+
if (contentType.includes("text/event-stream")) {
|
|
176
|
+
return { response, rpcResult: await parseSseResponse(response), newSessionId };
|
|
177
|
+
}
|
|
178
|
+
// 普通 JSON 响应 — 先读取文本,防止空内容导致 JSON.parse 报错
|
|
179
|
+
const text = await response.text();
|
|
180
|
+
if (!text.trim()) {
|
|
181
|
+
return { response, rpcResult: undefined, newSessionId };
|
|
182
|
+
}
|
|
183
|
+
const rpc = JSON.parse(text);
|
|
184
|
+
if (rpc.error) {
|
|
185
|
+
throw new McpRpcError(rpc.error.code, `MCP 调用错误 [${rpc.error.code}]: ${rpc.error.message}`, rpc.error.data);
|
|
186
|
+
}
|
|
187
|
+
return { response, rpcResult: rpc.result, newSessionId };
|
|
188
|
+
}
|
|
189
|
+
// ============================================================================
|
|
190
|
+
// Session 管理
|
|
191
|
+
// ============================================================================
|
|
192
|
+
/**
|
|
193
|
+
* 对指定 URL 执行 Streamable HTTP 的 initialize 握手
|
|
194
|
+
*
|
|
195
|
+
* 发送 initialize → 接收 serverInfo → 发送 initialized 通知。
|
|
196
|
+
* 如果服务端未返回 Mcp-Session-Id,则标记为无状态模式,后续请求跳过 session 管理。
|
|
197
|
+
*/
|
|
198
|
+
async function initializeSession(url, category) {
|
|
199
|
+
const session = { sessionId: null, initialized: false, stateless: false };
|
|
200
|
+
console.log(`${LOG_TAG} 开始 initialize 握手 (category="${category}")`);
|
|
201
|
+
// 1. 发送 initialize 请求
|
|
202
|
+
const initBody = {
|
|
203
|
+
jsonrpc: "2.0",
|
|
204
|
+
id: aibotNodeSdk.generateReqId("mcp_init"),
|
|
205
|
+
method: "initialize",
|
|
206
|
+
params: {
|
|
207
|
+
protocolVersion: "2025-03-26",
|
|
208
|
+
capabilities: {},
|
|
209
|
+
clientInfo: { name: "wecom_mcp", version: "1.0.0" },
|
|
210
|
+
},
|
|
211
|
+
};
|
|
212
|
+
const { newSessionId: initSessionId } = await sendRawJsonRpc(url, session, initBody);
|
|
213
|
+
// 用返回的 newSessionId 更新 session(不再依赖副作用修改)
|
|
214
|
+
if (initSessionId) {
|
|
215
|
+
session.sessionId = initSessionId;
|
|
216
|
+
}
|
|
217
|
+
// 检查服务端是否返回了 Mcp-Session-Id
|
|
218
|
+
// 如果没有返回,说明该 Server 是无状态实现,无需维护 session
|
|
219
|
+
if (!session.sessionId) {
|
|
220
|
+
session.stateless = true;
|
|
221
|
+
session.initialized = true;
|
|
222
|
+
statelessCategories.add(category);
|
|
223
|
+
mcpSessionCache.set(category, session);
|
|
224
|
+
console.log(`${LOG_TAG} 无状态 Server 确认 (category="${category}")`);
|
|
225
|
+
return session;
|
|
226
|
+
}
|
|
227
|
+
// 2. 发送 initialized 通知(JSON-RPC notification 不带 id 字段)
|
|
228
|
+
const notifyBody = {
|
|
229
|
+
jsonrpc: "2.0",
|
|
230
|
+
method: "notifications/initialized",
|
|
231
|
+
};
|
|
232
|
+
// initialized 通知不需要等待响应,但 Streamable HTTP 要求通过 POST 发送
|
|
233
|
+
const { newSessionId: notifySessionId } = await sendRawJsonRpc(url, session, notifyBody);
|
|
234
|
+
// 如果 initialized 通知的响应也携带了 sessionId,以最新的为准
|
|
235
|
+
if (notifySessionId) {
|
|
236
|
+
session.sessionId = notifySessionId;
|
|
237
|
+
}
|
|
238
|
+
session.initialized = true;
|
|
239
|
+
mcpSessionCache.set(category, session);
|
|
240
|
+
console.log(`${LOG_TAG} 有状态 Session 建立成功 (category="${category}", sessionId="${session.sessionId}")`);
|
|
241
|
+
return session;
|
|
242
|
+
}
|
|
243
|
+
/**
|
|
244
|
+
* 获取或创建指定 URL 的 MCP 会话
|
|
245
|
+
*
|
|
246
|
+
* - 已确认无状态的 category:直接返回空 session,跳过握手
|
|
247
|
+
* - 已有可用有状态会话:直接返回缓存
|
|
248
|
+
* - 其他情况:执行 initialize 握手,并发请求会被合并
|
|
249
|
+
*/
|
|
250
|
+
async function getOrCreateSession(url, category) {
|
|
251
|
+
// 已确认为无状态的 Server,直接返回空 session 跳过握手
|
|
252
|
+
if (statelessCategories.has(category)) {
|
|
253
|
+
const cached = mcpSessionCache.get(category);
|
|
254
|
+
if (cached)
|
|
255
|
+
return cached;
|
|
256
|
+
// 首次发现被清除(理论上不会走到这里),重新走握手探测
|
|
257
|
+
}
|
|
258
|
+
const cached = mcpSessionCache.get(category);
|
|
259
|
+
if (cached?.initialized)
|
|
260
|
+
return cached;
|
|
261
|
+
// 防止并发重复初始化
|
|
262
|
+
const inflight = inflightInitRequests.get(category);
|
|
263
|
+
if (inflight)
|
|
264
|
+
return inflight;
|
|
265
|
+
const promise = initializeSession(url, category).finally(() => {
|
|
266
|
+
inflightInitRequests.delete(category);
|
|
267
|
+
});
|
|
268
|
+
inflightInitRequests.set(category, promise);
|
|
269
|
+
return promise;
|
|
270
|
+
}
|
|
271
|
+
// ============================================================================
|
|
272
|
+
// SSE 解析
|
|
273
|
+
// ============================================================================
|
|
274
|
+
/**
|
|
275
|
+
* 解析 SSE 流式响应,提取最终的 JSON-RPC result
|
|
276
|
+
*
|
|
277
|
+
* 按照 SSE 规范,同一事件中的多个 `data:` 行会用换行符拼接。
|
|
278
|
+
* 空行分隔不同事件,取最后一个完整事件的数据。
|
|
279
|
+
*/
|
|
280
|
+
async function parseSseResponse(response) {
|
|
281
|
+
const text = await response.text();
|
|
282
|
+
const lines = text.split("\n");
|
|
283
|
+
// 按 SSE 规范解析:空行分隔事件,同一事件内的 data 行用换行拼接
|
|
284
|
+
let currentDataParts = [];
|
|
285
|
+
let lastEventData = "";
|
|
286
|
+
for (const line of lines) {
|
|
287
|
+
if (line.startsWith("data: ")) {
|
|
288
|
+
currentDataParts.push(line.slice(6));
|
|
289
|
+
}
|
|
290
|
+
else if (line.startsWith("data:")) {
|
|
291
|
+
// data: 后无空格时,值为空字符串
|
|
292
|
+
currentDataParts.push(line.slice(5));
|
|
293
|
+
}
|
|
294
|
+
else if (line.trim() === "" && currentDataParts.length > 0) {
|
|
295
|
+
// 空行表示事件结束,拼接所有 data 行
|
|
296
|
+
lastEventData = currentDataParts.join("\n").trim();
|
|
297
|
+
currentDataParts = [];
|
|
298
|
+
}
|
|
299
|
+
}
|
|
300
|
+
// 处理最后一个未以空行结尾的事件
|
|
301
|
+
if (currentDataParts.length > 0) {
|
|
302
|
+
lastEventData = currentDataParts.join("\n").trim();
|
|
303
|
+
}
|
|
304
|
+
if (!lastEventData) {
|
|
305
|
+
throw new Error("SSE 响应中未包含有效数据");
|
|
306
|
+
}
|
|
307
|
+
try {
|
|
308
|
+
const rpc = JSON.parse(lastEventData);
|
|
309
|
+
if (rpc.error) {
|
|
310
|
+
throw new McpRpcError(rpc.error.code, `MCP 调用错误 [${rpc.error.code}]: ${rpc.error.message}`, rpc.error.data);
|
|
311
|
+
}
|
|
312
|
+
return rpc.result;
|
|
313
|
+
}
|
|
314
|
+
catch (err) {
|
|
315
|
+
if (err instanceof SyntaxError) {
|
|
316
|
+
throw new Error(`SSE 响应解析失败: ${lastEventData.slice(0, 200)}`);
|
|
317
|
+
}
|
|
318
|
+
throw err;
|
|
319
|
+
}
|
|
320
|
+
}
|
|
321
|
+
// ============================================================================
|
|
322
|
+
// 公共 API
|
|
323
|
+
// ============================================================================
|
|
324
|
+
/**
|
|
325
|
+
* 清理指定品类的所有 MCP 缓存(配置、会话、无状态标记)
|
|
326
|
+
*
|
|
327
|
+
* 当 MCP Server 返回特定错误码时调用,确保下次请求重新拉取配置并重建会话。
|
|
328
|
+
*
|
|
329
|
+
* @param category - MCP 品类名称
|
|
330
|
+
*/
|
|
331
|
+
function clearCategoryCache(category) {
|
|
332
|
+
console.log(`${LOG_TAG} 清理缓存 (category="${category}")`);
|
|
333
|
+
mcpConfigCache.delete(category);
|
|
334
|
+
mcpSessionCache.delete(category);
|
|
335
|
+
statelessCategories.delete(category);
|
|
336
|
+
inflightInitRequests.delete(category);
|
|
337
|
+
}
|
|
338
|
+
/**
|
|
339
|
+
* 发送 JSON-RPC 请求到 MCP Server(Streamable HTTP 协议)
|
|
340
|
+
*
|
|
341
|
+
* 自动管理 session 生命周期:
|
|
342
|
+
* - 无状态 Server:跳过 session 管理,直接发送请求
|
|
343
|
+
* - 有状态 Server:首次调用先执行 initialize 握手,session 失效(404)时自动重建并重试
|
|
344
|
+
*
|
|
345
|
+
* @param category - MCP 品类名称
|
|
346
|
+
* @param method - JSON-RPC 方法名
|
|
347
|
+
* @param params - JSON-RPC 参数
|
|
348
|
+
* @param options - 可选配置(如自定义超时)
|
|
349
|
+
* @returns JSON-RPC result
|
|
350
|
+
*/
|
|
351
|
+
async function sendJsonRpc(category, method, params, options) {
|
|
352
|
+
const url = await getMcpUrl(category);
|
|
353
|
+
const timeoutMs = options?.timeoutMs;
|
|
354
|
+
const body = {
|
|
355
|
+
jsonrpc: "2.0",
|
|
356
|
+
id: aibotNodeSdk.generateReqId("mcp_rpc"),
|
|
357
|
+
method,
|
|
358
|
+
...(params !== undefined ? { params } : {}),
|
|
359
|
+
};
|
|
360
|
+
let session = await getOrCreateSession(url, category);
|
|
361
|
+
try {
|
|
362
|
+
const { rpcResult, newSessionId } = await sendRawJsonRpc(url, session, body, timeoutMs);
|
|
363
|
+
// 用最新的 sessionId 更新 session
|
|
364
|
+
if (newSessionId) {
|
|
365
|
+
session.sessionId = newSessionId;
|
|
366
|
+
}
|
|
367
|
+
return rpcResult;
|
|
368
|
+
}
|
|
369
|
+
catch (err) {
|
|
370
|
+
// 特定 JSON-RPC 错误码触发缓存清理(统一在传输层处理,上层无需关心)
|
|
371
|
+
if (err instanceof McpRpcError && CACHE_CLEAR_ERROR_CODES.has(err.code)) {
|
|
372
|
+
clearCategoryCache(category);
|
|
373
|
+
}
|
|
374
|
+
// 无状态 Server 不存在 session 失效问题,直接抛出错误
|
|
375
|
+
if (session.stateless)
|
|
376
|
+
throw err;
|
|
377
|
+
// 有状态 Server:session 失效时服务端返回 404,需要重新初始化并重试一次
|
|
378
|
+
// 使用 McpHttpError.statusCode 精确匹配,避免字符串匹配 "404" 导致误判
|
|
379
|
+
if (err instanceof McpHttpError && err.statusCode === 404) {
|
|
380
|
+
console.log(`${LOG_TAG} Session 失效 (category="${category}"),开始重建...`);
|
|
381
|
+
mcpSessionCache.delete(category);
|
|
382
|
+
// 使用 rebuildSession 合并并发的 session 重建请求,避免竞态条件
|
|
383
|
+
session = await rebuildSession(url, category);
|
|
384
|
+
const { rpcResult, newSessionId } = await sendRawJsonRpc(url, session, body, timeoutMs);
|
|
385
|
+
if (newSessionId) {
|
|
386
|
+
session.sessionId = newSessionId;
|
|
387
|
+
}
|
|
388
|
+
return rpcResult;
|
|
389
|
+
}
|
|
390
|
+
// 其他错误记录日志后抛出
|
|
391
|
+
console.error(`${LOG_TAG} RPC 请求失败 (category="${category}", method="${method}"): ${err instanceof Error ? err.message : String(err)}`);
|
|
392
|
+
throw err;
|
|
393
|
+
}
|
|
394
|
+
}
|
|
395
|
+
/**
|
|
396
|
+
* 合并并发的 session 重建请求
|
|
397
|
+
*
|
|
398
|
+
* 与 getOrCreateSession 类似,使用 inflightInitRequests 防止
|
|
399
|
+
* 多个并发请求同时遇到 404 时重复执行 initialize 握手。
|
|
400
|
+
*/
|
|
401
|
+
async function rebuildSession(url, category) {
|
|
402
|
+
const inflight = inflightInitRequests.get(category);
|
|
403
|
+
if (inflight)
|
|
404
|
+
return inflight;
|
|
405
|
+
const promise = initializeSession(url, category).finally(() => {
|
|
406
|
+
inflightInitRequests.delete(category);
|
|
407
|
+
});
|
|
408
|
+
inflightInitRequests.set(category, promise);
|
|
409
|
+
return promise;
|
|
410
|
+
}
|
|
411
|
+
|
|
412
|
+
exports.MEDIA_DOWNLOAD_TIMEOUT_MS = MEDIA_DOWNLOAD_TIMEOUT_MS;
|
|
413
|
+
exports.McpHttpError = McpHttpError;
|
|
414
|
+
exports.McpRpcError = McpRpcError;
|
|
415
|
+
exports.clearCategoryCache = clearCategoryCache;
|
|
416
|
+
exports.sendJsonRpc = sendJsonRpc;
|
|
417
|
+
//# sourceMappingURL=transport.js.map
|