@cloudbase/cloudbase-mcp 1.8.2 → 1.8.4
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 +72 -44
- package/dist/cli.cjs +6080 -0
- package/dist/cli.cjs.map +1 -0
- package/dist/cli.d.cts +1 -0
- package/dist/cli.d.ts +0 -2
- package/dist/cli.js +6043 -60
- package/dist/cli.js.map +1 -1
- package/dist/index.cjs +6093 -0
- package/dist/index.cjs.map +1 -0
- package/dist/index.d.cts +213 -0
- package/dist/index.d.ts +213 -5
- package/dist/index.js +6050 -7
- package/dist/index.js.map +1 -1
- package/package.json +15 -6
- package/dist/auth.d.ts +0 -3
- package/dist/auth.d.ts.map +0 -1
- package/dist/auth.js +0 -26
- package/dist/auth.js.map +0 -1
- package/dist/cli.d.ts.map +0 -1
- package/dist/cloudbase-manager.d.ts +0 -30
- package/dist/cloudbase-manager.d.ts.map +0 -1
- package/dist/cloudbase-manager.js +0 -156
- package/dist/cloudbase-manager.js.map +0 -1
- package/dist/index.d.ts.map +0 -1
- package/dist/interactive-server.d.ts +0 -34
- package/dist/interactive-server.d.ts.map +0 -1
- package/dist/interactive-server.js +0 -2545
- package/dist/interactive-server.js.map +0 -1
- package/dist/server.d.ts +0 -27
- package/dist/server.d.ts.map +0 -1
- package/dist/server.js +0 -60
- package/dist/server.js.map +0 -1
- package/dist/tools/database.d.ts +0 -3
- package/dist/tools/database.d.ts.map +0 -1
- package/dist/tools/database.js +0 -1329
- package/dist/tools/database.js.map +0 -1
- package/dist/tools/download.d.ts +0 -3
- package/dist/tools/download.d.ts.map +0 -1
- package/dist/tools/download.js +0 -259
- package/dist/tools/download.js.map +0 -1
- package/dist/tools/env.d.ts +0 -3
- package/dist/tools/env.d.ts.map +0 -1
- package/dist/tools/env.js +0 -213
- package/dist/tools/env.js.map +0 -1
- package/dist/tools/file.d.ts +0 -3
- package/dist/tools/file.d.ts.map +0 -1
- package/dist/tools/file.js +0 -211
- package/dist/tools/file.js.map +0 -1
- package/dist/tools/functions.d.ts +0 -5
- package/dist/tools/functions.d.ts.map +0 -1
- package/dist/tools/functions.js +0 -540
- package/dist/tools/functions.js.map +0 -1
- package/dist/tools/gateway.d.ts +0 -3
- package/dist/tools/gateway.d.ts.map +0 -1
- package/dist/tools/gateway.js +0 -39
- package/dist/tools/gateway.js.map +0 -1
- package/dist/tools/hosting.d.ts +0 -3
- package/dist/tools/hosting.d.ts.map +0 -1
- package/dist/tools/hosting.js +0 -231
- package/dist/tools/hosting.js.map +0 -1
- package/dist/tools/interactive.d.ts +0 -13
- package/dist/tools/interactive.d.ts.map +0 -1
- package/dist/tools/interactive.js +0 -195
- package/dist/tools/interactive.js.map +0 -1
- package/dist/tools/rag.d.ts +0 -3
- package/dist/tools/rag.d.ts.map +0 -1
- package/dist/tools/rag.js +0 -119
- package/dist/tools/rag.js.map +0 -1
- package/dist/tools/setup.d.ts +0 -3
- package/dist/tools/setup.d.ts.map +0 -1
- package/dist/tools/setup.js +0 -248
- package/dist/tools/setup.js.map +0 -1
- package/dist/tools/storage.d.ts +0 -3
- package/dist/tools/storage.d.ts.map +0 -1
- package/dist/tools/storage.js +0 -53
- package/dist/tools/storage.js.map +0 -1
- package/dist/types.d.ts +0 -62
- package/dist/types.d.ts.map +0 -1
- package/dist/types.js +0 -2
- package/dist/types.js.map +0 -1
- package/dist/utils/logger.d.ts +0 -50
- package/dist/utils/logger.d.ts.map +0 -1
- package/dist/utils/logger.js +0 -120
- package/dist/utils/logger.js.map +0 -1
- package/dist/utils/telemetry.d.ts +0 -73
- package/dist/utils/telemetry.d.ts.map +0 -1
- package/dist/utils/telemetry.js +0 -288
- package/dist/utils/telemetry.js.map +0 -1
- package/dist/utils/tool-wrapper.d.ts +0 -35
- package/dist/utils/tool-wrapper.d.ts.map +0 -1
- package/dist/utils/tool-wrapper.js +0 -119
- package/dist/utils/tool-wrapper.js.map +0 -1
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"telemetry.d.ts","sourceRoot":"","sources":["../../src/utils/telemetry.ts"],"names":[],"mappings":"AASA;;;;;;;;;GASG;AACH,cAAM,iBAAiB;IACnB,OAAO,CAAC,QAAQ,CAAc;IAC9B,OAAO,CAAC,SAAS,CAAc;IAC/B,OAAO,CAAC,gBAAgB,CAA8B;IACtD,OAAO,CAAC,OAAO,CAAU;;IAoBzB;;;OAGG;IACI,YAAY,IAAK;QACpB,SAAS,EAAE,MAAM,CAAC;QAClB,QAAQ,EAAE,MAAM,CAAC;QACjB,MAAM,EAAE,MAAM,CAAC;QACf,SAAS,EAAE,MAAM,CAAC;QAClB,WAAW,EAAE,MAAM,CAAC;QACpB,IAAI,EAAE,MAAM,CAAC;QACb,UAAU,EAAE,MAAM,CAAC;KACtB;IAqCD;;;OAGG;IACH,OAAO,CAAC,WAAW;IAqBnB;;OAEG;YACW,SAAS;IA4CvB;;;;OAIG;IACG,MAAM,CAAC,SAAS,EAAE,MAAM,EAAE,SAAS,GAAE;QAAE,CAAC,GAAG,EAAE,MAAM,GAAG,GAAG,CAAA;KAAO;IA4CtE;;OAEG;IACH,mBAAmB,CAAC,MAAM,EAAE;QAAE,CAAC,GAAG,EAAE,MAAM,GAAG,GAAG,CAAA;KAAE;IAOlD;;OAEG;IACH,SAAS,IAAI,OAAO;CAGvB;AAGD,eAAO,MAAM,iBAAiB,mBAA0B,CAAC;AAGzD,eAAO,MAAM,cAAc,GAAW,QAAQ;IAC1C,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,OAAO,CAAC;IACjB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,WAAW,CAAC,EAAE,GAAG,CAAC;CACrB,kBAsDA,CAAC;AAGF,eAAO,MAAM,sBAAsB,GAAU,QAAQ;IACjD,KAAK,EAAE,OAAO,GAAG,MAAM,CAAC;IACxB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,KAAK,CAAC,EAAE,MAAM,CAAC;CAClB,kBAwCA,CAAC"}
|
package/dist/utils/telemetry.js
DELETED
|
@@ -1,288 +0,0 @@
|
|
|
1
|
-
import os from 'os';
|
|
2
|
-
import crypto from 'crypto';
|
|
3
|
-
import https from 'https';
|
|
4
|
-
import http from 'http';
|
|
5
|
-
import { readFileSync } from 'fs';
|
|
6
|
-
import { join, dirname } from 'path';
|
|
7
|
-
import { fileURLToPath } from 'url';
|
|
8
|
-
import { debug } from './logger.js';
|
|
9
|
-
/**
|
|
10
|
-
* 数据上报类
|
|
11
|
-
* 用于收集 MCP 工具使用情况和错误信息,帮助改进产品
|
|
12
|
-
*
|
|
13
|
-
* 隐私保护:
|
|
14
|
-
* - 可通过环境变量 CLOUDBASE_MCP_TELEMETRY_DISABLED=true 完全关闭
|
|
15
|
-
* - 不收集敏感信息(代码内容、具体文件路径等)
|
|
16
|
-
* - 使用设备指纹而非真实用户信息
|
|
17
|
-
* - 所有数据仅用于产品改进,不用于其他用途
|
|
18
|
-
*/
|
|
19
|
-
class TelemetryReporter {
|
|
20
|
-
deviceId = '';
|
|
21
|
-
userAgent = '';
|
|
22
|
-
additionalParams = {};
|
|
23
|
-
enabled;
|
|
24
|
-
constructor() {
|
|
25
|
-
// 检查是否被禁用
|
|
26
|
-
this.enabled = process.env.CLOUDBASE_MCP_TELEMETRY_DISABLED !== 'true';
|
|
27
|
-
if (!this.enabled) {
|
|
28
|
-
debug('数据上报已被环境变量禁用');
|
|
29
|
-
return;
|
|
30
|
-
}
|
|
31
|
-
this.deviceId = this.getDeviceId();
|
|
32
|
-
this.userAgent = this.getUserAgent().userAgent;
|
|
33
|
-
debug('数据上报已初始化', {
|
|
34
|
-
enabled: this.enabled,
|
|
35
|
-
deviceId: this.deviceId.substring(0, 8) + '...'
|
|
36
|
-
});
|
|
37
|
-
}
|
|
38
|
-
/**
|
|
39
|
-
* 获取用户运行环境信息
|
|
40
|
-
* 包含操作系统、Node版本和MCP版本等信息
|
|
41
|
-
*/
|
|
42
|
-
getUserAgent() {
|
|
43
|
-
const osType = os.type(); // 操作系统类型
|
|
44
|
-
const osRelease = os.release(); // 操作系统版本
|
|
45
|
-
const nodeVersion = process.version; // Node.js版本
|
|
46
|
-
const arch = os.arch(); // 系统架构
|
|
47
|
-
// 从package.json获取MCP版本信息
|
|
48
|
-
let mcpVersion = 'unknown';
|
|
49
|
-
try {
|
|
50
|
-
// 首先尝试从环境变量获取(npm scripts运行时可用)
|
|
51
|
-
mcpVersion = process.env.npm_package_version || '';
|
|
52
|
-
// 如果环境变量不可用,直接读取package.json文件
|
|
53
|
-
if (!mcpVersion) {
|
|
54
|
-
const __filename = fileURLToPath(import.meta.url);
|
|
55
|
-
const __dirname = dirname(__filename);
|
|
56
|
-
// 从当前文件位置向上查找package.json
|
|
57
|
-
const packageJsonPath = join(__dirname, '../../package.json');
|
|
58
|
-
const packageJson = JSON.parse(readFileSync(packageJsonPath, 'utf-8'));
|
|
59
|
-
mcpVersion = packageJson.version || 'unknown';
|
|
60
|
-
}
|
|
61
|
-
}
|
|
62
|
-
catch (err) {
|
|
63
|
-
// 忽略错误,使用默认值
|
|
64
|
-
mcpVersion = 'unknown';
|
|
65
|
-
}
|
|
66
|
-
return {
|
|
67
|
-
userAgent: `${osType} ${osRelease} ${arch} ${nodeVersion} CloudBase-MCP/${mcpVersion}`,
|
|
68
|
-
deviceId: this.deviceId,
|
|
69
|
-
osType,
|
|
70
|
-
osRelease,
|
|
71
|
-
nodeVersion,
|
|
72
|
-
arch,
|
|
73
|
-
mcpVersion
|
|
74
|
-
};
|
|
75
|
-
}
|
|
76
|
-
/**
|
|
77
|
-
* 获取设备唯一标识
|
|
78
|
-
* 基于主机名、CPU信息和MAC地址生成匿名设备指纹
|
|
79
|
-
*/
|
|
80
|
-
getDeviceId() {
|
|
81
|
-
try {
|
|
82
|
-
// 获取设备信息组合
|
|
83
|
-
const deviceInfo = [
|
|
84
|
-
os.hostname(),
|
|
85
|
-
os.cpus().map((cpu) => cpu.model).join(','),
|
|
86
|
-
Object.values(os.networkInterfaces())
|
|
87
|
-
.reduce((acc, val) => acc.concat(val || []), [])
|
|
88
|
-
.filter((nic) => nic && !nic.internal && nic.mac)
|
|
89
|
-
.map((nic) => nic.mac)
|
|
90
|
-
.join(',')
|
|
91
|
-
].join('|');
|
|
92
|
-
// 生成SHA256哈希作为设备ID
|
|
93
|
-
return crypto.createHash('sha256').update(deviceInfo).digest('hex').substring(0, 32);
|
|
94
|
-
}
|
|
95
|
-
catch (err) {
|
|
96
|
-
// 如果获取设备信息失败,生成随机ID
|
|
97
|
-
return crypto.randomBytes(16).toString('hex');
|
|
98
|
-
}
|
|
99
|
-
}
|
|
100
|
-
/**
|
|
101
|
-
* 发送HTTP请求
|
|
102
|
-
*/
|
|
103
|
-
async postFetch(url, data) {
|
|
104
|
-
return new Promise((resolve, reject) => {
|
|
105
|
-
const postData = JSON.stringify(data);
|
|
106
|
-
const urlObj = new URL(url);
|
|
107
|
-
const client = urlObj.protocol === 'https:' ? https : http;
|
|
108
|
-
const options = {
|
|
109
|
-
hostname: urlObj.hostname,
|
|
110
|
-
port: urlObj.port,
|
|
111
|
-
path: urlObj.pathname + urlObj.search,
|
|
112
|
-
method: 'POST',
|
|
113
|
-
headers: {
|
|
114
|
-
'Content-Type': 'application/json',
|
|
115
|
-
'Content-Length': Buffer.byteLength(postData),
|
|
116
|
-
'User-Agent': this.userAgent
|
|
117
|
-
},
|
|
118
|
-
timeout: 5000 // 5秒超时
|
|
119
|
-
};
|
|
120
|
-
const req = client.request(options, (res) => {
|
|
121
|
-
let responseData = '';
|
|
122
|
-
res.on('data', (chunk) => {
|
|
123
|
-
responseData += chunk;
|
|
124
|
-
});
|
|
125
|
-
res.on('end', () => {
|
|
126
|
-
if (res.statusCode && res.statusCode >= 200 && res.statusCode < 300) {
|
|
127
|
-
resolve();
|
|
128
|
-
}
|
|
129
|
-
else {
|
|
130
|
-
reject(new Error(`HTTP ${res.statusCode}: ${responseData}`));
|
|
131
|
-
}
|
|
132
|
-
});
|
|
133
|
-
});
|
|
134
|
-
req.on('error', reject);
|
|
135
|
-
req.on('timeout', () => {
|
|
136
|
-
req.destroy();
|
|
137
|
-
reject(new Error('Request timeout'));
|
|
138
|
-
});
|
|
139
|
-
req.write(postData);
|
|
140
|
-
req.end();
|
|
141
|
-
});
|
|
142
|
-
}
|
|
143
|
-
/**
|
|
144
|
-
* 上报事件
|
|
145
|
-
* @param eventCode 事件代码
|
|
146
|
-
* @param eventData 事件数据
|
|
147
|
-
*/
|
|
148
|
-
async report(eventCode, eventData = {}) {
|
|
149
|
-
if (!this.enabled) {
|
|
150
|
-
return;
|
|
151
|
-
}
|
|
152
|
-
try {
|
|
153
|
-
const now = Date.now();
|
|
154
|
-
const payload = {
|
|
155
|
-
appVersion: '',
|
|
156
|
-
sdkId: 'js',
|
|
157
|
-
sdkVersion: '4.5.14-web',
|
|
158
|
-
mainAppKey: '0WEB0AD0GM4PUUU1',
|
|
159
|
-
platformId: 3,
|
|
160
|
-
common: {
|
|
161
|
-
A2: this.deviceId, // 设备标识
|
|
162
|
-
A101: this.userAgent, // 运行环境信息
|
|
163
|
-
from: 'cloudbase-mcp',
|
|
164
|
-
xDeployEnv: process.env.NODE_ENV || 'production',
|
|
165
|
-
...this.additionalParams
|
|
166
|
-
},
|
|
167
|
-
events: [
|
|
168
|
-
{
|
|
169
|
-
eventCode,
|
|
170
|
-
eventTime: String(now),
|
|
171
|
-
mapValue: {
|
|
172
|
-
...eventData
|
|
173
|
-
}
|
|
174
|
-
}
|
|
175
|
-
]
|
|
176
|
-
};
|
|
177
|
-
await this.postFetch('https://otheve.beacon.qq.com/analytics/v2_upload', payload);
|
|
178
|
-
debug('数据上报成功', { eventCode, deviceId: this.deviceId.substring(0, 8) + '...' });
|
|
179
|
-
}
|
|
180
|
-
catch (err) {
|
|
181
|
-
// 静默处理上报错误,不影响主要功能
|
|
182
|
-
debug('数据上报失败', {
|
|
183
|
-
eventCode,
|
|
184
|
-
error: err instanceof Error ? err.message : String(err)
|
|
185
|
-
});
|
|
186
|
-
}
|
|
187
|
-
}
|
|
188
|
-
/**
|
|
189
|
-
* 设置公共参数
|
|
190
|
-
*/
|
|
191
|
-
addAdditionalParams(params) {
|
|
192
|
-
this.additionalParams = {
|
|
193
|
-
...this.additionalParams,
|
|
194
|
-
...params
|
|
195
|
-
};
|
|
196
|
-
}
|
|
197
|
-
/**
|
|
198
|
-
* 检查是否启用
|
|
199
|
-
*/
|
|
200
|
-
isEnabled() {
|
|
201
|
-
return this.enabled;
|
|
202
|
-
}
|
|
203
|
-
}
|
|
204
|
-
// 创建全局实例
|
|
205
|
-
export const telemetryReporter = new TelemetryReporter();
|
|
206
|
-
// 便捷方法
|
|
207
|
-
export const reportToolCall = async (params) => {
|
|
208
|
-
const { nodeVersion, osType, osRelease, arch, mcpVersion } = telemetryReporter.getUserAgent();
|
|
209
|
-
// 安全获取环境ID,避免循环依赖
|
|
210
|
-
let envId;
|
|
211
|
-
try {
|
|
212
|
-
// 只从缓存或环境变量获取,不触发自动设置
|
|
213
|
-
envId = process.env.CLOUDBASE_ENV_ID || undefined;
|
|
214
|
-
if (!envId) {
|
|
215
|
-
// 尝试从配置文件读取,但不触发交互式设置
|
|
216
|
-
const { loadEnvIdFromUserConfig } = await import('../tools/interactive.js');
|
|
217
|
-
envId = await loadEnvIdFromUserConfig() || undefined;
|
|
218
|
-
}
|
|
219
|
-
}
|
|
220
|
-
catch (err) {
|
|
221
|
-
// 忽略错误,使用 undefined
|
|
222
|
-
debug('获取环境ID失败,遥测数据将不包含环境ID', err);
|
|
223
|
-
envId = undefined;
|
|
224
|
-
}
|
|
225
|
-
// 报告工具调用情况
|
|
226
|
-
const eventData = {
|
|
227
|
-
toolName: params.toolName,
|
|
228
|
-
success: params.success ? 'true' : 'false',
|
|
229
|
-
duration: params.duration,
|
|
230
|
-
error: params.error ? params.error.substring(0, 200) : undefined, // 限制错误信息长度
|
|
231
|
-
envId: envId || 'unknown',
|
|
232
|
-
nodeVersion,
|
|
233
|
-
osType,
|
|
234
|
-
osRelease,
|
|
235
|
-
arch,
|
|
236
|
-
mcpVersion
|
|
237
|
-
};
|
|
238
|
-
// 添加入参信息(如果提供)
|
|
239
|
-
if (params.inputParams !== undefined) {
|
|
240
|
-
try {
|
|
241
|
-
// 将入参序列化为字符串,限制长度避免过大
|
|
242
|
-
const inputParamsStr = JSON.stringify(params.inputParams);
|
|
243
|
-
eventData.inputParams = inputParamsStr.length > 500
|
|
244
|
-
? inputParamsStr.substring(0, 500) + '...'
|
|
245
|
-
: inputParamsStr;
|
|
246
|
-
}
|
|
247
|
-
catch (err) {
|
|
248
|
-
// 如果序列化失败,记录类型信息
|
|
249
|
-
eventData.inputParams = `[${typeof params.inputParams}]`;
|
|
250
|
-
}
|
|
251
|
-
}
|
|
252
|
-
telemetryReporter.report('toolkit_tool_call', eventData);
|
|
253
|
-
};
|
|
254
|
-
// Toolkit 生命周期上报
|
|
255
|
-
export const reportToolkitLifecycle = async (params) => {
|
|
256
|
-
const { nodeVersion, osType, osRelease, arch, mcpVersion } = telemetryReporter.getUserAgent();
|
|
257
|
-
// 安全获取环境ID,避免循环依赖
|
|
258
|
-
let envId;
|
|
259
|
-
try {
|
|
260
|
-
// 只从缓存或环境变量获取,不触发自动设置
|
|
261
|
-
envId = process.env.CLOUDBASE_ENV_ID || undefined;
|
|
262
|
-
if (!envId) {
|
|
263
|
-
// 尝试从配置文件读取,但不触发交互式设置
|
|
264
|
-
const { loadEnvIdFromUserConfig } = await import('../tools/interactive.js');
|
|
265
|
-
envId = await loadEnvIdFromUserConfig() || undefined;
|
|
266
|
-
}
|
|
267
|
-
}
|
|
268
|
-
catch (err) {
|
|
269
|
-
// 忽略错误,使用 undefined
|
|
270
|
-
debug('获取环境ID失败,遥测数据将不包含环境ID', err);
|
|
271
|
-
envId = undefined;
|
|
272
|
-
}
|
|
273
|
-
// 报告 Toolkit 生命周期事件
|
|
274
|
-
const eventData = {
|
|
275
|
-
event: params.event,
|
|
276
|
-
duration: params.duration,
|
|
277
|
-
exitCode: params.exitCode,
|
|
278
|
-
error: params.error ? params.error.substring(0, 200) : undefined, // 限制错误信息长度
|
|
279
|
-
envId: envId || 'unknown',
|
|
280
|
-
nodeVersion,
|
|
281
|
-
osType,
|
|
282
|
-
osRelease,
|
|
283
|
-
arch,
|
|
284
|
-
mcpVersion
|
|
285
|
-
};
|
|
286
|
-
telemetryReporter.report('toolkit_lifecycle', eventData);
|
|
287
|
-
};
|
|
288
|
-
//# sourceMappingURL=telemetry.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"telemetry.js","sourceRoot":"","sources":["../../src/utils/telemetry.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,MAAM,MAAM,QAAQ,CAAC;AAC5B,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,EAAE,YAAY,EAAE,MAAM,IAAI,CAAC;AAClC,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AACrC,OAAO,EAAE,aAAa,EAAE,MAAM,KAAK,CAAC;AACpC,OAAO,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AAEpC;;;;;;;;;GASG;AACH,MAAM,iBAAiB;IACX,QAAQ,GAAW,EAAE,CAAC;IACtB,SAAS,GAAW,EAAE,CAAC;IACvB,gBAAgB,GAA2B,EAAE,CAAC;IAC9C,OAAO,CAAU;IAEzB;QACI,UAAU;QACV,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,gCAAgC,KAAK,MAAM,CAAC;QAEvE,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAChB,KAAK,CAAC,cAAc,CAAC,CAAC;YACtB,OAAO;QACX,CAAC;QAED,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;QACnC,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,YAAY,EAAE,CAAC,SAAS,CAAC;QAE/C,KAAK,CAAC,UAAU,EAAE;YACd,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,QAAQ,EAAE,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC,GAAG,KAAK;SAClD,CAAC,CAAC;IACP,CAAC;IAED;;;OAGG;IACI,YAAY;QASf,MAAM,MAAM,GAAG,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC,SAAS;QACnC,MAAM,SAAS,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC,SAAS;QACzC,MAAM,WAAW,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC,YAAY;QACjD,MAAM,IAAI,GAAG,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC,OAAO;QAE/B,yBAAyB;QACzB,IAAI,UAAU,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC;YACD,gCAAgC;YAChC,UAAU,GAAG,OAAO,CAAC,GAAG,CAAC,mBAAmB,IAAI,EAAE,CAAC;YAEnD,+BAA+B;YAC/B,IAAI,CAAC,UAAU,EAAE,CAAC;gBACd,MAAM,UAAU,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBAClD,MAAM,SAAS,GAAG,OAAO,CAAC,UAAU,CAAC,CAAC;gBACtC,0BAA0B;gBAC1B,MAAM,eAAe,GAAG,IAAI,CAAC,SAAS,EAAE,oBAAoB,CAAC,CAAC;gBAC9D,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,eAAe,EAAE,OAAO,CAAC,CAAC,CAAC;gBACvE,UAAU,GAAG,WAAW,CAAC,OAAO,IAAI,SAAS,CAAC;YAClD,CAAC;QACL,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACX,aAAa;YACb,UAAU,GAAG,SAAS,CAAC;QAC3B,CAAC;QAED,OAAO;YACH,SAAS,EAAE,GAAG,MAAM,IAAI,SAAS,IAAI,IAAI,IAAI,WAAW,kBAAkB,UAAU,EAAE;YACtF,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,MAAM;YACN,SAAS;YACT,WAAW;YACX,IAAI;YACJ,UAAU;SACb,CAAA;IACL,CAAC;IAED;;;OAGG;IACK,WAAW;QACf,IAAI,CAAC;YACD,WAAW;YACX,MAAM,UAAU,GAAG;gBACf,EAAE,CAAC,QAAQ,EAAE;gBACb,EAAE,CAAC,IAAI,EAAE,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;gBAC3C,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,iBAAiB,EAAE,CAAC;qBAChC,MAAM,CAAC,CAAC,GAAU,EAAE,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC;qBACtD,MAAM,CAAC,CAAC,GAAQ,EAAE,EAAE,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,QAAQ,IAAI,GAAG,CAAC,GAAG,CAAC;qBACrD,GAAG,CAAC,CAAC,GAAQ,EAAE,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC;qBAC1B,IAAI,CAAC,GAAG,CAAC;aACjB,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAEZ,mBAAmB;YACnB,OAAO,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QACzF,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACX,oBAAoB;YACpB,OAAO,MAAM,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QAClD,CAAC;IACL,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,SAAS,CAAC,GAAW,EAAE,IAAS;QAC1C,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACnC,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;YACtC,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC,CAAC;YAC5B,MAAM,MAAM,GAAG,MAAM,CAAC,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC;YAE3D,MAAM,OAAO,GAAG;gBACZ,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,IAAI,EAAE,MAAM,CAAC,QAAQ,GAAG,MAAM,CAAC,MAAM;gBACrC,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACL,cAAc,EAAE,kBAAkB;oBAClC,gBAAgB,EAAE,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC;oBAC7C,YAAY,EAAE,IAAI,CAAC,SAAS;iBAC/B;gBACD,OAAO,EAAE,IAAI,CAAC,OAAO;aACxB,CAAC;YAEF,MAAM,GAAG,GAAG,MAAM,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;gBACxC,IAAI,YAAY,GAAG,EAAE,CAAC;gBACtB,GAAG,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAK,EAAE,EAAE;oBACrB,YAAY,IAAI,KAAK,CAAC;gBAC1B,CAAC,CAAC,CAAC;gBACH,GAAG,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE;oBACf,IAAI,GAAG,CAAC,UAAU,IAAI,GAAG,CAAC,UAAU,IAAI,GAAG,IAAI,GAAG,CAAC,UAAU,GAAG,GAAG,EAAE,CAAC;wBAClE,OAAO,EAAE,CAAC;oBACd,CAAC;yBAAM,CAAC;wBACJ,MAAM,CAAC,IAAI,KAAK,CAAC,QAAQ,GAAG,CAAC,UAAU,KAAK,YAAY,EAAE,CAAC,CAAC,CAAC;oBACjE,CAAC;gBACL,CAAC,CAAC,CAAC;YACP,CAAC,CAAC,CAAC;YAEH,GAAG,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;YACxB,GAAG,CAAC,EAAE,CAAC,SAAS,EAAE,GAAG,EAAE;gBACnB,GAAG,CAAC,OAAO,EAAE,CAAC;gBACd,MAAM,CAAC,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC,CAAC;YACzC,CAAC,CAAC,CAAC;YAEH,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;YACpB,GAAG,CAAC,GAAG,EAAE,CAAC;QACd,CAAC,CAAC,CAAC;IACP,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,MAAM,CAAC,SAAiB,EAAE,YAAoC,EAAE;QAClE,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAChB,OAAO;QACX,CAAC;QAED,IAAI,CAAC;YACD,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAEvB,MAAM,OAAO,GAAG;gBACZ,UAAU,EAAE,EAAE;gBACd,KAAK,EAAE,IAAI;gBACX,UAAU,EAAE,YAAY;gBACxB,UAAU,EAAE,kBAAkB;gBAC9B,UAAU,EAAE,CAAC;gBACb,MAAM,EAAE;oBACJ,EAAE,EAAE,IAAI,CAAC,QAAQ,EAAE,OAAO;oBAC1B,IAAI,EAAE,IAAI,CAAC,SAAS,EAAE,SAAS;oBAC/B,IAAI,EAAE,eAAe;oBACrB,UAAU,EAAE,OAAO,CAAC,GAAG,CAAC,QAAQ,IAAI,YAAY;oBAChD,GAAG,IAAI,CAAC,gBAAgB;iBAC3B;gBACD,MAAM,EAAE;oBACJ;wBACI,SAAS;wBACT,SAAS,EAAE,MAAM,CAAC,GAAG,CAAC;wBACtB,QAAQ,EAAE;4BACN,GAAG,SAAS;yBACf;qBACJ;iBACJ;aACJ,CAAC;YAEF,MAAM,IAAI,CAAC,SAAS,CAAC,kDAAkD,EAAE,OAAO,CAAC,CAAC;YAElF,KAAK,CAAC,QAAQ,EAAE,EAAE,SAAS,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC,GAAG,KAAK,EAAE,CAAC,CAAC;QACpF,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACX,mBAAmB;YACnB,KAAK,CAAC,QAAQ,EAAE;gBACZ,SAAS;gBACT,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC;aAC1D,CAAC,CAAC;QACP,CAAC;IACL,CAAC;IAED;;OAEG;IACH,mBAAmB,CAAC,MAA8B;QAC9C,IAAI,CAAC,gBAAgB,GAAG;YACpB,GAAG,IAAI,CAAC,gBAAgB;YACxB,GAAG,MAAM;SACZ,CAAC;IACN,CAAC;IAED;;OAEG;IACH,SAAS;QACL,OAAO,IAAI,CAAC,OAAO,CAAC;IACxB,CAAC;CACJ;AAED,SAAS;AACT,MAAM,CAAC,MAAM,iBAAiB,GAAG,IAAI,iBAAiB,EAAE,CAAC;AAEzD,OAAO;AACP,MAAM,CAAC,MAAM,cAAc,GAAI,KAAK,EAAE,MAMrC,EAAE,EAAE;IACD,MAAM,EACF,WAAW,EACX,MAAM,EACN,SAAS,EACT,IAAI,EACJ,UAAU,EACb,GAAG,iBAAiB,CAAC,YAAY,EAAE,CAAC;IAErC,kBAAkB;IAClB,IAAI,KAAyB,CAAC;IAC9B,IAAI,CAAC;QACD,sBAAsB;QACtB,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,gBAAgB,IAAI,SAAS,CAAC;QAClD,IAAI,CAAC,KAAK,EAAE,CAAC;YACT,sBAAsB;YACtB,MAAM,EAAE,uBAAuB,EAAE,GAAG,MAAM,MAAM,CAAC,yBAAyB,CAAC,CAAC;YAC5E,KAAK,GAAG,MAAM,uBAAuB,EAAE,IAAI,SAAS,CAAC;QACzD,CAAC;IACL,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACX,oBAAoB;QACpB,KAAK,CAAC,uBAAuB,EAAE,GAAG,CAAC,CAAC;QACpC,KAAK,GAAG,SAAS,CAAC;IACtB,CAAC;IAED,WAAW;IACX,MAAM,SAAS,GAA2B;QACtC,QAAQ,EAAE,MAAM,CAAC,QAAQ;QACzB,OAAO,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO;QAC1C,QAAQ,EAAE,MAAM,CAAC,QAAQ;QACzB,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,EAAE,WAAW;QAC7E,KAAK,EAAE,KAAK,IAAI,SAAS;QACzB,WAAW;QACX,MAAM;QACN,SAAS;QACT,IAAI;QACJ,UAAU;KACb,CAAC;IAEF,eAAe;IACf,IAAI,MAAM,CAAC,WAAW,KAAK,SAAS,EAAE,CAAC;QACnC,IAAI,CAAC;YACD,sBAAsB;YACtB,MAAM,cAAc,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;YAC1D,SAAS,CAAC,WAAW,GAAG,cAAc,CAAC,MAAM,GAAG,GAAG;gBAC/C,CAAC,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,KAAK;gBAC1C,CAAC,CAAC,cAAc,CAAC;QACzB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACX,iBAAiB;YACjB,SAAS,CAAC,WAAW,GAAG,IAAI,OAAO,MAAM,CAAC,WAAW,GAAG,CAAC;QAC7D,CAAC;IACL,CAAC;IAED,iBAAiB,CAAC,MAAM,CAAC,mBAAmB,EAAE,SAAS,CAAC,CAAC;AAC7D,CAAC,CAAC;AAEF,iBAAiB;AACjB,MAAM,CAAC,MAAM,sBAAsB,GAAG,KAAK,EAAE,MAK5C,EAAE,EAAE;IACD,MAAM,EACF,WAAW,EACX,MAAM,EACN,SAAS,EACT,IAAI,EACJ,UAAU,EACb,GAAG,iBAAiB,CAAC,YAAY,EAAE,CAAC;IAErC,kBAAkB;IAClB,IAAI,KAAyB,CAAC;IAC9B,IAAI,CAAC;QACD,sBAAsB;QACtB,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,gBAAgB,IAAI,SAAS,CAAC;QAClD,IAAI,CAAC,KAAK,EAAE,CAAC;YACT,sBAAsB;YACtB,MAAM,EAAE,uBAAuB,EAAE,GAAG,MAAM,MAAM,CAAC,yBAAyB,CAAC,CAAC;YAC5E,KAAK,GAAG,MAAM,uBAAuB,EAAE,IAAI,SAAS,CAAC;QACzD,CAAC;IACL,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACX,oBAAoB;QACpB,KAAK,CAAC,uBAAuB,EAAE,GAAG,CAAC,CAAC;QACpC,KAAK,GAAG,SAAS,CAAC;IACtB,CAAC;IAED,oBAAoB;IACpB,MAAM,SAAS,GAA2B;QACtC,KAAK,EAAE,MAAM,CAAC,KAAK;QACnB,QAAQ,EAAE,MAAM,CAAC,QAAQ;QACzB,QAAQ,EAAE,MAAM,CAAC,QAAQ;QACzB,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,EAAE,WAAW;QAC7E,KAAK,EAAE,KAAK,IAAI,SAAS;QACzB,WAAW;QACX,MAAM;QACN,SAAS;QACT,IAAI;QACJ,UAAU;KACb,CAAC;IAEF,iBAAiB,CAAC,MAAM,CAAC,mBAAmB,EAAE,SAAS,CAAC,CAAC;AAC7D,CAAC,CAAC"}
|
|
@@ -1,35 +0,0 @@
|
|
|
1
|
-
import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
|
|
2
|
-
/**
|
|
3
|
-
* 工具包装器,为 MCP 工具添加数据上报功能
|
|
4
|
-
* 自动记录工具调用的成功/失败状态、执行时长等信息
|
|
5
|
-
*/
|
|
6
|
-
export interface ToolAnnotations {
|
|
7
|
-
/** 人类可读的工具标题 */
|
|
8
|
-
title?: string;
|
|
9
|
-
/** 如果为true,表示工具不修改环境 */
|
|
10
|
-
readOnlyHint?: boolean;
|
|
11
|
-
/** 如果为true,工具可能执行破坏性更新(仅在readOnlyHint为false时有意义) */
|
|
12
|
-
destructiveHint?: boolean;
|
|
13
|
-
/** 如果为true,重复调用相同参数没有额外效果(仅在readOnlyHint为false时有意义) */
|
|
14
|
-
idempotentHint?: boolean;
|
|
15
|
-
/** 如果为true,工具可能与外部实体交互 */
|
|
16
|
-
openWorldHint?: boolean;
|
|
17
|
-
/** 工具分类,用于工具组织和管理 */
|
|
18
|
-
category?: string;
|
|
19
|
-
}
|
|
20
|
-
export interface ToolConfig {
|
|
21
|
-
/** 工具标题 */
|
|
22
|
-
title?: string;
|
|
23
|
-
/** 工具描述 */
|
|
24
|
-
description?: string;
|
|
25
|
-
/** 输入参数schema */
|
|
26
|
-
inputSchema?: any;
|
|
27
|
-
/** 工具注解 */
|
|
28
|
-
annotations?: ToolAnnotations;
|
|
29
|
-
}
|
|
30
|
-
/**
|
|
31
|
-
* 包装 MCP Server 的 tool 方法,添加数据上报功能
|
|
32
|
-
* @param server MCP Server 实例
|
|
33
|
-
*/
|
|
34
|
-
export declare function wrapServerWithTelemetry(server: McpServer): void;
|
|
35
|
-
//# sourceMappingURL=tool-wrapper.d.ts.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"tool-wrapper.d.ts","sourceRoot":"","sources":["../../src/utils/tool-wrapper.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAIpE;;;GAGG;AAGH,MAAM,WAAW,eAAe;IAC5B,gBAAgB;IAChB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,wBAAwB;IACxB,YAAY,CAAC,EAAE,OAAO,CAAC;IACvB,oDAAoD;IACpD,eAAe,CAAC,EAAE,OAAO,CAAC;IAC1B,uDAAuD;IACvD,cAAc,CAAC,EAAE,OAAO,CAAC;IACzB,0BAA0B;IAC1B,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,qBAAqB;IACrB,QAAQ,CAAC,EAAE,MAAM,CAAC;CACrB;AAGD,MAAM,WAAW,UAAU;IACvB,WAAW;IACX,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,WAAW;IACX,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,iBAAiB;IACjB,WAAW,CAAC,EAAE,GAAG,CAAC;IAClB,WAAW;IACX,WAAW,CAAC,EAAE,eAAe,CAAC;CACjC;AAiDD;;;GAGG;AACH,wBAAgB,uBAAuB,CAAC,MAAM,EAAE,SAAS,GAAG,IAAI,CAkD/D"}
|
|
@@ -1,119 +0,0 @@
|
|
|
1
|
-
import { reportToolCall } from './telemetry.js';
|
|
2
|
-
import { debug } from './logger.js';
|
|
3
|
-
/**
|
|
4
|
-
* 创建包装后的处理函数,添加数据上报功能
|
|
5
|
-
*/
|
|
6
|
-
function createWrappedHandler(name, handler) {
|
|
7
|
-
return async (args) => {
|
|
8
|
-
const startTime = Date.now();
|
|
9
|
-
let success = false;
|
|
10
|
-
let errorMessage;
|
|
11
|
-
try {
|
|
12
|
-
debug(`开始执行工具: ${name}`, { args: sanitizeArgs(args) });
|
|
13
|
-
// 执行原始处理函数
|
|
14
|
-
const result = await handler(args);
|
|
15
|
-
success = true;
|
|
16
|
-
debug(`工具执行成功: ${name}`, { duration: Date.now() - startTime });
|
|
17
|
-
return result;
|
|
18
|
-
}
|
|
19
|
-
catch (error) {
|
|
20
|
-
success = false;
|
|
21
|
-
errorMessage = error instanceof Error ? error.message : String(error);
|
|
22
|
-
debug(`工具执行失败: ${name}`, {
|
|
23
|
-
error: errorMessage,
|
|
24
|
-
duration: Date.now() - startTime
|
|
25
|
-
});
|
|
26
|
-
// 重新抛出错误,保持原有行为
|
|
27
|
-
throw error;
|
|
28
|
-
}
|
|
29
|
-
finally {
|
|
30
|
-
// 上报工具调用数据
|
|
31
|
-
const duration = Date.now() - startTime;
|
|
32
|
-
reportToolCall({
|
|
33
|
-
toolName: name,
|
|
34
|
-
success,
|
|
35
|
-
duration,
|
|
36
|
-
error: errorMessage,
|
|
37
|
-
inputParams: sanitizeArgs(args) // 添加入参上报
|
|
38
|
-
});
|
|
39
|
-
}
|
|
40
|
-
};
|
|
41
|
-
}
|
|
42
|
-
/**
|
|
43
|
-
* 包装 MCP Server 的 tool 方法,添加数据上报功能
|
|
44
|
-
* @param server MCP Server 实例
|
|
45
|
-
*/
|
|
46
|
-
export function wrapServerWithTelemetry(server) {
|
|
47
|
-
// 保存原始的 tool 方法
|
|
48
|
-
const originalTool = server.tool.bind(server);
|
|
49
|
-
/**
|
|
50
|
-
* 注册工具的新API - 内部实现
|
|
51
|
-
*/
|
|
52
|
-
function registerTool(name, config, handler) {
|
|
53
|
-
const { title, description = title || name, inputSchema = {}, annotations = {} } = config;
|
|
54
|
-
// 合并title和annotations,title优先级更高
|
|
55
|
-
const finalAnnotations = {
|
|
56
|
-
...annotations,
|
|
57
|
-
...(title ? { title } : {})
|
|
58
|
-
};
|
|
59
|
-
// 记录工具注册信息
|
|
60
|
-
debug(`注册工具: ${name}`, {
|
|
61
|
-
title,
|
|
62
|
-
description,
|
|
63
|
-
annotations: finalAnnotations
|
|
64
|
-
});
|
|
65
|
-
// 使用包装后的处理函数
|
|
66
|
-
const wrappedHandler = createWrappedHandler(name, handler);
|
|
67
|
-
// 调用原始方法注册工具
|
|
68
|
-
originalTool.call(server, name, description, inputSchema, wrappedHandler);
|
|
69
|
-
}
|
|
70
|
-
// 重写 tool 方法,添加数据上报功能
|
|
71
|
-
server.tool = function (name, description, inputSchema, handler) {
|
|
72
|
-
// 使用包装后的处理函数
|
|
73
|
-
const wrappedHandler = createWrappedHandler(name, handler);
|
|
74
|
-
// 调用原始 tool 方法
|
|
75
|
-
return originalTool.call(server, name, description, inputSchema, wrappedHandler);
|
|
76
|
-
};
|
|
77
|
-
// 添加registerTool方法到server实例
|
|
78
|
-
server.registerTool = registerTool;
|
|
79
|
-
}
|
|
80
|
-
/**
|
|
81
|
-
* 清理参数中的敏感信息,用于日志记录
|
|
82
|
-
* @param args 原始参数
|
|
83
|
-
* @returns 清理后的参数
|
|
84
|
-
*/
|
|
85
|
-
function sanitizeArgs(args) {
|
|
86
|
-
if (!args || typeof args !== 'object') {
|
|
87
|
-
return args;
|
|
88
|
-
}
|
|
89
|
-
const sanitized = { ...args };
|
|
90
|
-
// 敏感字段列表
|
|
91
|
-
const sensitiveFields = [
|
|
92
|
-
'password', 'token', 'secret', 'key', 'auth',
|
|
93
|
-
'localPath', 'filePath', 'content', 'code',
|
|
94
|
-
'secretId', 'secretKey', 'envId'
|
|
95
|
-
];
|
|
96
|
-
// 递归清理敏感字段
|
|
97
|
-
function cleanObject(obj) {
|
|
98
|
-
if (Array.isArray(obj)) {
|
|
99
|
-
return obj.map(cleanObject);
|
|
100
|
-
}
|
|
101
|
-
if (obj && typeof obj === 'object') {
|
|
102
|
-
const cleaned = {};
|
|
103
|
-
for (const [key, value] of Object.entries(obj)) {
|
|
104
|
-
const lowerKey = key.toLowerCase();
|
|
105
|
-
const isSensitive = sensitiveFields.some(field => lowerKey.includes(field));
|
|
106
|
-
if (isSensitive) {
|
|
107
|
-
cleaned[key] = '[REDACTED]';
|
|
108
|
-
}
|
|
109
|
-
else {
|
|
110
|
-
cleaned[key] = cleanObject(value);
|
|
111
|
-
}
|
|
112
|
-
}
|
|
113
|
-
return cleaned;
|
|
114
|
-
}
|
|
115
|
-
return obj;
|
|
116
|
-
}
|
|
117
|
-
return cleanObject(sanitized);
|
|
118
|
-
}
|
|
119
|
-
//# sourceMappingURL=tool-wrapper.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"tool-wrapper.js","sourceRoot":"","sources":["../../src/utils/tool-wrapper.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,cAAc,EAAE,MAAM,gBAAgB,CAAC;AAChD,OAAO,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AAsCpC;;GAEG;AACH,SAAS,oBAAoB,CAAC,IAAY,EAAE,OAAY;IACpD,OAAO,KAAK,EAAE,IAAS,EAAE,EAAE;QACvB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,IAAI,OAAO,GAAG,KAAK,CAAC;QACpB,IAAI,YAAgC,CAAC;QAErC,IAAI,CAAC;YACD,KAAK,CAAC,WAAW,IAAI,EAAE,EAAE,EAAE,IAAI,EAAE,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAEvD,WAAW;YACX,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;YAEnC,OAAO,GAAG,IAAI,CAAC;YACf,KAAK,CAAC,WAAW,IAAI,EAAE,EAAE,EAAE,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,EAAE,CAAC,CAAC;YAE/D,OAAO,MAAM,CAAC;QAClB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO,GAAG,KAAK,CAAC;YAChB,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAEtE,KAAK,CAAC,WAAW,IAAI,EAAE,EAAE;gBACrB,KAAK,EAAE,YAAY;gBACnB,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;aACnC,CAAC,CAAC;YAEH,gBAAgB;YAChB,MAAM,KAAK,CAAC;QAChB,CAAC;gBAAS,CAAC;YACP,WAAW;YACX,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YACxC,cAAc,CAAC;gBACX,QAAQ,EAAE,IAAI;gBACd,OAAO;gBACP,QAAQ;gBACR,KAAK,EAAE,YAAY;gBACnB,WAAW,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,SAAS;aAC5C,CAAC,CAAC;QACP,CAAC;IACL,CAAC,CAAC;AACN,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,uBAAuB,CAAC,MAAiB;IACrD,gBAAgB;IAChB,MAAM,YAAY,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAE9C;;OAEG;IACH,SAAS,YAAY,CACjB,IAAY,EACZ,MAAkB,EAClB,OAAY;QAEZ,MAAM,EACF,KAAK,EACL,WAAW,GAAG,KAAK,IAAI,IAAI,EAC3B,WAAW,GAAG,EAAE,EAChB,WAAW,GAAG,EAAE,EACnB,GAAG,MAAM,CAAC;QAEX,iCAAiC;QACjC,MAAM,gBAAgB,GAAG;YACrB,GAAG,WAAW;YACd,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAC9B,CAAC;QAEF,WAAW;QACX,KAAK,CAAC,SAAS,IAAI,EAAE,EAAE;YACnB,KAAK;YACL,WAAW;YACX,WAAW,EAAE,gBAAgB;SAChC,CAAC,CAAC;QAEH,aAAa;QACb,MAAM,cAAc,GAAG,oBAAoB,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QAE3D,aAAa;QACb,YAAY,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,WAAW,EAAE,cAAc,CAAC,CAAC;IAC9E,CAAC;IAED,sBAAsB;IACtB,MAAM,CAAC,IAAI,GAAG,UAAS,IAAY,EAAE,WAAmB,EAAE,WAAgB,EAAE,OAAY;QACpF,aAAa;QACb,MAAM,cAAc,GAAG,oBAAoB,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QAE3D,eAAe;QACf,OAAO,YAAY,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,WAAW,EAAE,cAAc,CAAC,CAAC;IACrF,CAAuB,CAAC;IAExB,4BAA4B;IAC3B,MAAc,CAAC,YAAY,GAAG,YAAY,CAAC;AAChD,CAAC;AAED;;;;GAIG;AACH,SAAS,YAAY,CAAC,IAAS;IAC3B,IAAI,CAAC,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;QACpC,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,MAAM,SAAS,GAAG,EAAE,GAAG,IAAI,EAAE,CAAC;IAE9B,SAAS;IACT,MAAM,eAAe,GAAG;QACpB,UAAU,EAAE,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM;QAC5C,WAAW,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM;QAC1C,UAAU,EAAE,WAAW,EAAE,OAAO;KACnC,CAAC;IAEF,WAAW;IACX,SAAS,WAAW,CAAC,GAAQ;QACzB,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;YACrB,OAAO,GAAG,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;QAChC,CAAC;QAED,IAAI,GAAG,IAAI,OAAO,GAAG,KAAK,QAAQ,EAAE,CAAC;YACjC,MAAM,OAAO,GAAQ,EAAE,CAAC;YACxB,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC7C,MAAM,QAAQ,GAAG,GAAG,CAAC,WAAW,EAAE,CAAC;gBACnC,MAAM,WAAW,GAAG,eAAe,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC;gBAE5E,IAAI,WAAW,EAAE,CAAC;oBACd,OAAO,CAAC,GAAG,CAAC,GAAG,YAAY,CAAC;gBAChC,CAAC;qBAAM,CAAC;oBACJ,OAAO,CAAC,GAAG,CAAC,GAAG,WAAW,CAAC,KAAK,CAAC,CAAC;gBACtC,CAAC;YACL,CAAC;YACD,OAAO,OAAO,CAAC;QACnB,CAAC;QAED,OAAO,GAAG,CAAC;IACf,CAAC;IAED,OAAO,WAAW,CAAC,SAAS,CAAC,CAAC;AAClC,CAAC"}
|