proxy-api-mcp 1.0.0
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/.env.example +57 -0
- package/README.md +443 -0
- package/api-tools.json.example +226 -0
- package/dist/examples/env-in-json-usage.d.ts +43 -0
- package/dist/examples/env-in-json-usage.d.ts.map +1 -0
- package/dist/examples/env-in-json-usage.js +242 -0
- package/dist/examples/env-in-json-usage.js.map +1 -0
- package/dist/examples/env-usage.d.ts +7 -0
- package/dist/examples/env-usage.d.ts.map +1 -0
- package/dist/examples/env-usage.js +108 -0
- package/dist/examples/env-usage.js.map +1 -0
- package/dist/examples/preprocessor-usage.d.ts +14 -0
- package/dist/examples/preprocessor-usage.d.ts.map +1 -0
- package/dist/examples/preprocessor-usage.js +196 -0
- package/dist/examples/preprocessor-usage.js.map +1 -0
- package/dist/src/config/env.d.ts +97 -0
- package/dist/src/config/env.d.ts.map +1 -0
- package/dist/src/config/env.js +228 -0
- package/dist/src/config/env.js.map +1 -0
- package/dist/src/core/registry.d.ts +59 -0
- package/dist/src/core/registry.d.ts.map +1 -0
- package/dist/src/core/registry.js +126 -0
- package/dist/src/core/registry.js.map +1 -0
- package/dist/src/core/router.d.ts +34 -0
- package/dist/src/core/router.d.ts.map +1 -0
- package/dist/src/core/router.js +108 -0
- package/dist/src/core/router.js.map +1 -0
- package/dist/src/core/server.d.ts +55 -0
- package/dist/src/core/server.d.ts.map +1 -0
- package/dist/src/core/server.js +159 -0
- package/dist/src/core/server.js.map +1 -0
- package/dist/src/core/session.d.ts +70 -0
- package/dist/src/core/session.d.ts.map +1 -0
- package/dist/src/core/session.js +171 -0
- package/dist/src/core/session.js.map +1 -0
- package/dist/src/core/transports/http.d.ts +16 -0
- package/dist/src/core/transports/http.d.ts.map +1 -0
- package/dist/src/core/transports/http.js +169 -0
- package/dist/src/core/transports/http.js.map +1 -0
- package/dist/src/core/transports/stdio.d.ts +9 -0
- package/dist/src/core/transports/stdio.d.ts.map +1 -0
- package/dist/src/core/transports/stdio.js +32 -0
- package/dist/src/core/transports/stdio.js.map +1 -0
- package/dist/src/index.d.ts +19 -0
- package/dist/src/index.d.ts.map +1 -0
- package/dist/src/index.js +22 -0
- package/dist/src/index.js.map +1 -0
- package/dist/src/server.d.ts +6 -0
- package/dist/src/server.d.ts.map +1 -0
- package/dist/src/server.js +121 -0
- package/dist/src/server.js.map +1 -0
- package/dist/src/tools/api/builder.d.ts +34 -0
- package/dist/src/tools/api/builder.d.ts.map +1 -0
- package/dist/src/tools/api/builder.js +196 -0
- package/dist/src/tools/api/builder.js.map +1 -0
- package/dist/src/tools/api/executor.d.ts +57 -0
- package/dist/src/tools/api/executor.d.ts.map +1 -0
- package/dist/src/tools/api/executor.js +317 -0
- package/dist/src/tools/api/executor.js.map +1 -0
- package/dist/src/tools/api/index.d.ts +12 -0
- package/dist/src/tools/api/index.d.ts.map +1 -0
- package/dist/src/tools/api/index.js +12 -0
- package/dist/src/tools/api/index.js.map +1 -0
- package/dist/src/tools/api/json/kingdee/kingdee_caiwukuaiji_chunaguanli.json +879 -0
- package/dist/src/tools/api/json/kingdee/kingdee_caiwukuaiji_yingfukuanguanli.json +1673 -0
- package/dist/src/tools/api/json/kingdee/kingdee_caiwukuaiji_yingshoukuanguanli.json +1081 -0
- package/dist/src/tools/api/json/kingdee/kingdee_caiwukuaiji_zongzhang.json +976 -0
- package/dist/src/tools/api/json/kingdee/kingdee_chengbenguanli_biaozhunchengbenfenxi.json +1 -0
- package/dist/src/tools/api/json/kingdee/kingdee_chengbenguanli_chanpinchengbenhesuan.json +1 -0
- package/dist/src/tools/api/json/kingdee/kingdee_chengbenguanli_cunhuohesuan.json +1858 -0
- package/dist/src/tools/api/json/kingdee/kingdee_gongyinglianguanli_caigouguanli.json +1 -0
- package/dist/src/tools/api/json/kingdee/kingdee_gongyinglianguanli_gongyingshangguanli.json +616 -0
- package/dist/src/tools/api/json/kingdee/kingdee_gongyinglianguanli_kucunguanli.json +858 -0
- package/dist/src/tools/api/json/kingdee/kingdee_gongyinglianguanli_xiaoshouguanli.json +858 -0
- package/dist/src/tools/api/json/kingdee/kingdee_gongyinglianguanli_zuzhijianjiesuan.json +858 -0
- package/dist/src/tools/api/json/kingdee/kingdee_jichuguanli_jichuziliao.json +867 -0
- package/dist/src/tools/api/json/kingdee/kingdee_jichuguanli_xiaoxipingtai.json +374 -0
- package/dist/src/tools/api/json/kingdee/kingdee_shengchanzhizao_chejianguanli.json +923 -0
- package/dist/src/tools/api/json/kingdee/kingdee_shengchanzhizao_gongchengshujuguanli.json +923 -0
- package/dist/src/tools/api/json/kingdee/kingdee_shengchanzhizao_shengchanguanli.json +858 -0
- package/dist/src/tools/api/json/kingdee/kingdee_shuiwuguanli_fapiaoguanli.json +1081 -0
- package/dist/src/tools/api/json/kingdee/kingdee_yuangongfuwu_renrenbaoxiao.json +858 -0
- package/dist/src/tools/api/json/kingdee/kingdee_zichanguanli_gudingzichan.json +1 -0
- package/dist/src/tools/api/json-config-loader.d.ts +28 -0
- package/dist/src/tools/api/json-config-loader.d.ts.map +1 -0
- package/dist/src/tools/api/json-config-loader.js +123 -0
- package/dist/src/tools/api/json-config-loader.js.map +1 -0
- package/dist/src/tools/api/json-config-mapping.d.ts +41 -0
- package/dist/src/tools/api/json-config-mapping.d.ts.map +1 -0
- package/dist/src/tools/api/json-config-mapping.js +138 -0
- package/dist/src/tools/api/json-config-mapping.js.map +1 -0
- package/dist/src/tools/api/loader.d.ts +38 -0
- package/dist/src/tools/api/loader.d.ts.map +1 -0
- package/dist/src/tools/api/loader.js +240 -0
- package/dist/src/tools/api/loader.js.map +1 -0
- package/dist/src/tools/api/preprocessor.d.ts +39 -0
- package/dist/src/tools/api/preprocessor.d.ts.map +1 -0
- package/dist/src/tools/api/preprocessor.js +64 -0
- package/dist/src/tools/api/preprocessor.js.map +1 -0
- package/dist/src/tools/api/preprocessors/example.d.ts +59 -0
- package/dist/src/tools/api/preprocessors/example.d.ts.map +1 -0
- package/dist/src/tools/api/preprocessors/example.js +133 -0
- package/dist/src/tools/api/preprocessors/example.js.map +1 -0
- package/dist/src/tools/api/preprocessors/index.d.ts +10 -0
- package/dist/src/tools/api/preprocessors/index.d.ts.map +1 -0
- package/dist/src/tools/api/preprocessors/index.js +29 -0
- package/dist/src/tools/api/preprocessors/index.js.map +1 -0
- package/dist/src/tools/api/preprocessors/kingdee.d.ts +24 -0
- package/dist/src/tools/api/preprocessors/kingdee.d.ts.map +1 -0
- package/dist/src/tools/api/preprocessors/kingdee.js +149 -0
- package/dist/src/tools/api/preprocessors/kingdee.js.map +1 -0
- package/dist/src/tools/api/types.d.ts +174 -0
- package/dist/src/tools/api/types.d.ts.map +1 -0
- package/dist/src/tools/api/types.js +5 -0
- package/dist/src/tools/api/types.js.map +1 -0
- package/dist/src/tools/examples/echo.d.ts +6 -0
- package/dist/src/tools/examples/echo.d.ts.map +1 -0
- package/dist/src/tools/examples/echo.js +61 -0
- package/dist/src/tools/examples/echo.js.map +1 -0
- package/dist/src/tools/types.d.ts +104 -0
- package/dist/src/tools/types.d.ts.map +1 -0
- package/dist/src/tools/types.js +15 -0
- package/dist/src/tools/types.js.map +1 -0
- package/dist/src/tools/validator.d.ts +27 -0
- package/dist/src/tools/validator.d.ts.map +1 -0
- package/dist/src/tools/validator.js +133 -0
- package/dist/src/tools/validator.js.map +1 -0
- package/dist/src/utils/cache.d.ts +14 -0
- package/dist/src/utils/cache.d.ts.map +1 -0
- package/dist/src/utils/cache.js +42 -0
- package/dist/src/utils/cache.js.map +1 -0
- package/dist/src/utils/env-replacer.d.ts +86 -0
- package/dist/src/utils/env-replacer.d.ts.map +1 -0
- package/dist/src/utils/env-replacer.js +196 -0
- package/dist/src/utils/env-replacer.js.map +1 -0
- package/dist/src/utils/errors.d.ts +102 -0
- package/dist/src/utils/errors.d.ts.map +1 -0
- package/dist/src/utils/errors.js +164 -0
- package/dist/src/utils/errors.js.map +1 -0
- package/dist/src/utils/logger.d.ts +77 -0
- package/dist/src/utils/logger.d.ts.map +1 -0
- package/dist/src/utils/logger.js +155 -0
- package/dist/src/utils/logger.js.map +1 -0
- package/package.json +73 -0
|
@@ -0,0 +1,126 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* 工具注册器
|
|
3
|
+
*/
|
|
4
|
+
import { ToolRegistrationError } from '../utils/errors.js';
|
|
5
|
+
import { SchemaValidator } from '../tools/validator.js';
|
|
6
|
+
import { logger } from '../utils/logger.js';
|
|
7
|
+
/**
|
|
8
|
+
* 工具注册器类
|
|
9
|
+
*/
|
|
10
|
+
export class ToolRegistry {
|
|
11
|
+
tools = new Map();
|
|
12
|
+
/**
|
|
13
|
+
* 注册单个工具
|
|
14
|
+
*/
|
|
15
|
+
register(config) {
|
|
16
|
+
// 验证工具配置
|
|
17
|
+
this.validateToolConfig(config);
|
|
18
|
+
// 检查工具名称唯一性
|
|
19
|
+
if (this.tools.has(config.name)) {
|
|
20
|
+
throw new ToolRegistrationError(`Tool already registered: ${config.name}`, { toolName: config.name });
|
|
21
|
+
}
|
|
22
|
+
// 验证 inputSchema
|
|
23
|
+
if (!SchemaValidator.isValidSchema(config.inputSchema)) {
|
|
24
|
+
throw new ToolRegistrationError(`Invalid input schema for tool: ${config.name}`, { toolName: config.name, schema: config.inputSchema });
|
|
25
|
+
}
|
|
26
|
+
// 验证 outputSchema (如果存在)
|
|
27
|
+
if (config.outputSchema && !SchemaValidator.isValidSchema(config.outputSchema)) {
|
|
28
|
+
throw new ToolRegistrationError(`Invalid output schema for tool: ${config.name}`, { toolName: config.name, schema: config.outputSchema });
|
|
29
|
+
}
|
|
30
|
+
// 注册工具
|
|
31
|
+
this.tools.set(config.name, config);
|
|
32
|
+
logger.info(`Tool registered: ${config.name}`, {
|
|
33
|
+
title: config.title,
|
|
34
|
+
hasOutputSchema: !!config.outputSchema,
|
|
35
|
+
});
|
|
36
|
+
}
|
|
37
|
+
/**
|
|
38
|
+
* 批量注册工具
|
|
39
|
+
*/
|
|
40
|
+
registerBatch(configs) {
|
|
41
|
+
for (const config of configs) {
|
|
42
|
+
this.register(config);
|
|
43
|
+
}
|
|
44
|
+
}
|
|
45
|
+
/**
|
|
46
|
+
* 获取工具配置
|
|
47
|
+
*/
|
|
48
|
+
get(name) {
|
|
49
|
+
return this.tools.get(name);
|
|
50
|
+
}
|
|
51
|
+
/**
|
|
52
|
+
* 检查工具是否存在
|
|
53
|
+
*/
|
|
54
|
+
has(name) {
|
|
55
|
+
return this.tools.has(name);
|
|
56
|
+
}
|
|
57
|
+
/**
|
|
58
|
+
* 获取所有工具名称
|
|
59
|
+
*/
|
|
60
|
+
getToolNames() {
|
|
61
|
+
return Array.from(this.tools.keys());
|
|
62
|
+
}
|
|
63
|
+
/**
|
|
64
|
+
* 获取所有工具配置
|
|
65
|
+
*/
|
|
66
|
+
getAll() {
|
|
67
|
+
return Array.from(this.tools.values());
|
|
68
|
+
}
|
|
69
|
+
/**
|
|
70
|
+
* 注销工具
|
|
71
|
+
*/
|
|
72
|
+
unregister(name) {
|
|
73
|
+
const result = this.tools.delete(name);
|
|
74
|
+
if (result) {
|
|
75
|
+
logger.info(`Tool unregistered: ${name}`);
|
|
76
|
+
}
|
|
77
|
+
return result;
|
|
78
|
+
}
|
|
79
|
+
/**
|
|
80
|
+
* 清空所有工具
|
|
81
|
+
*/
|
|
82
|
+
clear() {
|
|
83
|
+
const count = this.tools.size;
|
|
84
|
+
this.tools.clear();
|
|
85
|
+
logger.info(`All tools cleared (${count} tools)`);
|
|
86
|
+
}
|
|
87
|
+
/**
|
|
88
|
+
* 获取工具数量
|
|
89
|
+
*/
|
|
90
|
+
count() {
|
|
91
|
+
return this.tools.size;
|
|
92
|
+
}
|
|
93
|
+
/**
|
|
94
|
+
* 验证工具配置的完整性
|
|
95
|
+
*/
|
|
96
|
+
validateToolConfig(config) {
|
|
97
|
+
// 验证必填字段
|
|
98
|
+
if (!config.name || typeof config.name !== 'string') {
|
|
99
|
+
throw new ToolRegistrationError('Tool name is required and must be a string');
|
|
100
|
+
}
|
|
101
|
+
if (!config.description || typeof config.description !== 'string') {
|
|
102
|
+
throw new ToolRegistrationError(`Tool description is required: ${config.name}`);
|
|
103
|
+
}
|
|
104
|
+
if (!config.inputSchema || typeof config.inputSchema !== 'object') {
|
|
105
|
+
throw new ToolRegistrationError(`Tool inputSchema is required: ${config.name}`);
|
|
106
|
+
}
|
|
107
|
+
if (!config.handler || typeof config.handler !== 'function') {
|
|
108
|
+
throw new ToolRegistrationError(`Tool handler is required and must be a function: ${config.name}`);
|
|
109
|
+
}
|
|
110
|
+
// 验证名称格式(小写字母、数字和下划线)
|
|
111
|
+
if (!/^[a-z0-9_]+$/.test(config.name)) {
|
|
112
|
+
throw new ToolRegistrationError(`Invalid tool name format: ${config.name}. Use only lowercase letters, numbers, and underscores.`);
|
|
113
|
+
}
|
|
114
|
+
}
|
|
115
|
+
/**
|
|
116
|
+
* 转换为 MCP 工具列表格式
|
|
117
|
+
*/
|
|
118
|
+
toMcpTools() {
|
|
119
|
+
return this.getAll().map(tool => ({
|
|
120
|
+
name: tool.name,
|
|
121
|
+
description: tool.description,
|
|
122
|
+
inputSchema: tool.inputSchema,
|
|
123
|
+
}));
|
|
124
|
+
}
|
|
125
|
+
}
|
|
126
|
+
//# sourceMappingURL=registry.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"registry.js","sourceRoot":"","sources":["../../../src/core/registry.ts"],"names":[],"mappings":"AAAA;;GAEG;AAGH,OAAO,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAC3D,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AACxD,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAE5C;;GAEG;AACH,MAAM,OAAO,YAAY;IACf,KAAK,GAAG,IAAI,GAAG,EAAsB,CAAC;IAE9C;;OAEG;IACH,QAAQ,CAAC,MAAkB;QACzB,SAAS;QACT,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC;QAEhC,YAAY;QACZ,IAAI,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;YAChC,MAAM,IAAI,qBAAqB,CAC7B,4BAA4B,MAAM,CAAC,IAAI,EAAE,EACzC,EAAE,QAAQ,EAAE,MAAM,CAAC,IAAI,EAAE,CAC1B,CAAC;QACJ,CAAC;QAED,iBAAiB;QACjB,IAAI,CAAC,eAAe,CAAC,aAAa,CAAC,MAAM,CAAC,WAAW,CAAC,EAAE,CAAC;YACvD,MAAM,IAAI,qBAAqB,CAC7B,kCAAkC,MAAM,CAAC,IAAI,EAAE,EAC/C,EAAE,QAAQ,EAAE,MAAM,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC,WAAW,EAAE,CACtD,CAAC;QACJ,CAAC;QAED,yBAAyB;QACzB,IAAI,MAAM,CAAC,YAAY,IAAI,CAAC,eAAe,CAAC,aAAa,CAAC,MAAM,CAAC,YAAY,CAAC,EAAE,CAAC;YAC/E,MAAM,IAAI,qBAAqB,CAC7B,mCAAmC,MAAM,CAAC,IAAI,EAAE,EAChD,EAAE,QAAQ,EAAE,MAAM,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC,YAAY,EAAE,CACvD,CAAC;QACJ,CAAC;QAED,OAAO;QACP,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAEpC,MAAM,CAAC,IAAI,CAAC,oBAAoB,MAAM,CAAC,IAAI,EAAE,EAAE;YAC7C,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,eAAe,EAAE,CAAC,CAAC,MAAM,CAAC,YAAY;SACvC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,aAAa,CAAC,OAAqB;QACjC,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;YAC7B,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;QACxB,CAAC;IACH,CAAC;IAED;;OAEG;IACH,GAAG,CAAC,IAAY;QACd,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IAC9B,CAAC;IAED;;OAEG;IACH,GAAG,CAAC,IAAY;QACd,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IAC9B,CAAC;IAED;;OAEG;IACH,YAAY;QACV,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;IACvC,CAAC;IAED;;OAEG;IACH,MAAM;QACJ,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;IACzC,CAAC;IAED;;OAEG;IACH,UAAU,CAAC,IAAY;QACrB,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACvC,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,CAAC,IAAI,CAAC,sBAAsB,IAAI,EAAE,CAAC,CAAC;QAC5C,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACH,KAAK;QACH,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC;QAC9B,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;QACnB,MAAM,CAAC,IAAI,CAAC,sBAAsB,KAAK,SAAS,CAAC,CAAC;IACpD,CAAC;IAED;;OAEG;IACH,KAAK;QACH,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC;IACzB,CAAC;IAED;;OAEG;IACK,kBAAkB,CAAC,MAAkB;QAC3C,SAAS;QACT,IAAI,CAAC,MAAM,CAAC,IAAI,IAAI,OAAO,MAAM,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YACpD,MAAM,IAAI,qBAAqB,CAAC,4CAA4C,CAAC,CAAC;QAChF,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,WAAW,IAAI,OAAO,MAAM,CAAC,WAAW,KAAK,QAAQ,EAAE,CAAC;YAClE,MAAM,IAAI,qBAAqB,CAC7B,iCAAiC,MAAM,CAAC,IAAI,EAAE,CAC/C,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,WAAW,IAAI,OAAO,MAAM,CAAC,WAAW,KAAK,QAAQ,EAAE,CAAC;YAClE,MAAM,IAAI,qBAAqB,CAC7B,iCAAiC,MAAM,CAAC,IAAI,EAAE,CAC/C,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,OAAO,IAAI,OAAO,MAAM,CAAC,OAAO,KAAK,UAAU,EAAE,CAAC;YAC5D,MAAM,IAAI,qBAAqB,CAC7B,oDAAoD,MAAM,CAAC,IAAI,EAAE,CAClE,CAAC;QACJ,CAAC;QAED,sBAAsB;QACtB,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;YACtC,MAAM,IAAI,qBAAqB,CAC7B,6BAA6B,MAAM,CAAC,IAAI,yDAAyD,CAClG,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;OAEG;IACH,UAAU;QAKR,OAAO,IAAI,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAChC,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,WAAW,EAAE,IAAI,CAAC,WAAW;SAC9B,CAAC,CAAC,CAAC;IACN,CAAC;CACF"}
|
|
@@ -0,0 +1,34 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* 工具路由器
|
|
3
|
+
*/
|
|
4
|
+
import { type ToolRegistry } from './registry.js';
|
|
5
|
+
import { type ToolContext, type ToolResult, type Middleware } from '../tools/types.js';
|
|
6
|
+
/**
|
|
7
|
+
* 工具路由器类
|
|
8
|
+
*/
|
|
9
|
+
export declare class ToolRouter {
|
|
10
|
+
private registry;
|
|
11
|
+
private middlewares;
|
|
12
|
+
constructor(registry: ToolRegistry);
|
|
13
|
+
/**
|
|
14
|
+
* 添加中间件
|
|
15
|
+
*/
|
|
16
|
+
use(middleware: Middleware): void;
|
|
17
|
+
/**
|
|
18
|
+
* 调用工具
|
|
19
|
+
*/
|
|
20
|
+
callTool(toolName: string, args: Record<string, unknown>, context: ToolContext): Promise<ToolResult>;
|
|
21
|
+
/**
|
|
22
|
+
* 应用中间件链
|
|
23
|
+
*/
|
|
24
|
+
private applyMiddlewares;
|
|
25
|
+
/**
|
|
26
|
+
* 批量调用工具(并行)
|
|
27
|
+
*/
|
|
28
|
+
callToolsBatch(calls: {
|
|
29
|
+
toolName: string;
|
|
30
|
+
args: Record<string, unknown>;
|
|
31
|
+
context: ToolContext;
|
|
32
|
+
}[]): Promise<ToolResult[]>;
|
|
33
|
+
}
|
|
34
|
+
//# sourceMappingURL=router.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"router.d.ts","sourceRoot":"","sources":["../../../src/core/router.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,KAAK,YAAY,EAAE,MAAM,eAAe,CAAC;AAClD,OAAO,EAAE,KAAK,WAAW,EAAE,KAAK,UAAU,EAAE,KAAK,UAAU,EAAE,MAAM,mBAAmB,CAAC;AASvF;;GAEG;AACH,qBAAa,UAAU;IAGT,OAAO,CAAC,QAAQ;IAF5B,OAAO,CAAC,WAAW,CAAoB;gBAEnB,QAAQ,EAAE,YAAY;IAE1C;;OAEG;IACH,GAAG,CAAC,UAAU,EAAE,UAAU,GAAG,IAAI;IAKjC;;OAEG;IACG,QAAQ,CACZ,QAAQ,EAAE,MAAM,EAChB,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC7B,OAAO,EAAE,WAAW,GACnB,OAAO,CAAC,UAAU,CAAC;IA0EtB;;OAEG;YACW,gBAAgB;IAsB9B;;OAEG;IACG,cAAc,CAClB,KAAK,EAAE;QACL,QAAQ,EAAE,MAAM,CAAC;QACjB,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAC9B,OAAO,EAAE,WAAW,CAAC;KACtB,EAAE,GACF,OAAO,CAAC,UAAU,EAAE,CAAC;CAOzB"}
|
|
@@ -0,0 +1,108 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* 工具路由器
|
|
3
|
+
*/
|
|
4
|
+
import { SchemaValidator } from '../tools/validator.js';
|
|
5
|
+
import { ToolNotFoundError, ToolExecutionError, toMcpError, } from '../utils/errors.js';
|
|
6
|
+
import { logger } from '../utils/logger.js';
|
|
7
|
+
/**
|
|
8
|
+
* 工具路由器类
|
|
9
|
+
*/
|
|
10
|
+
export class ToolRouter {
|
|
11
|
+
registry;
|
|
12
|
+
middlewares = [];
|
|
13
|
+
constructor(registry) {
|
|
14
|
+
this.registry = registry;
|
|
15
|
+
}
|
|
16
|
+
/**
|
|
17
|
+
* 添加中间件
|
|
18
|
+
*/
|
|
19
|
+
use(middleware) {
|
|
20
|
+
this.middlewares.push(middleware);
|
|
21
|
+
logger.debug('Middleware added', { count: this.middlewares.length });
|
|
22
|
+
}
|
|
23
|
+
/**
|
|
24
|
+
* 调用工具
|
|
25
|
+
*/
|
|
26
|
+
async callTool(toolName, args, context) {
|
|
27
|
+
const startTime = Date.now();
|
|
28
|
+
try {
|
|
29
|
+
// 查找工具配置
|
|
30
|
+
const toolConfig = this.registry.get(toolName);
|
|
31
|
+
if (!toolConfig) {
|
|
32
|
+
throw new ToolNotFoundError(toolName);
|
|
33
|
+
}
|
|
34
|
+
logger.debug(`Calling tool: ${toolName}`, { args, context });
|
|
35
|
+
// 验证输入参数
|
|
36
|
+
const validatedArgs = SchemaValidator.validateInput(args, toolConfig.inputSchema);
|
|
37
|
+
// 创建执行函数
|
|
38
|
+
const execute = async () => {
|
|
39
|
+
try {
|
|
40
|
+
// 调用工具处理器
|
|
41
|
+
const result = await toolConfig.handler(validatedArgs, context);
|
|
42
|
+
// 验证输出结果(如果有 outputSchema)
|
|
43
|
+
if (toolConfig.outputSchema && result.structuredContent) {
|
|
44
|
+
SchemaValidator.validateOutput(result.structuredContent, toolConfig.outputSchema);
|
|
45
|
+
}
|
|
46
|
+
return result;
|
|
47
|
+
}
|
|
48
|
+
catch (error) {
|
|
49
|
+
if (error instanceof Error) {
|
|
50
|
+
throw new ToolExecutionError(toolName, error);
|
|
51
|
+
}
|
|
52
|
+
throw error;
|
|
53
|
+
}
|
|
54
|
+
};
|
|
55
|
+
// 应用中间件
|
|
56
|
+
const result = await this.applyMiddlewares(context, execute);
|
|
57
|
+
const duration = Date.now() - startTime;
|
|
58
|
+
logger.info(`Tool executed: ${toolName}`, {
|
|
59
|
+
duration,
|
|
60
|
+
success: !result.isError,
|
|
61
|
+
});
|
|
62
|
+
return result;
|
|
63
|
+
}
|
|
64
|
+
catch (error) {
|
|
65
|
+
const duration = Date.now() - startTime;
|
|
66
|
+
const mcpError = toMcpError(error);
|
|
67
|
+
logger.error(`Tool execution failed: ${toolName}`, {
|
|
68
|
+
duration,
|
|
69
|
+
error: mcpError.message,
|
|
70
|
+
code: mcpError.code,
|
|
71
|
+
});
|
|
72
|
+
return {
|
|
73
|
+
content: [
|
|
74
|
+
{
|
|
75
|
+
type: 'text',
|
|
76
|
+
text: mcpError.message,
|
|
77
|
+
},
|
|
78
|
+
],
|
|
79
|
+
structuredContent: mcpError.toJSON(),
|
|
80
|
+
isError: true,
|
|
81
|
+
};
|
|
82
|
+
}
|
|
83
|
+
}
|
|
84
|
+
/**
|
|
85
|
+
* 应用中间件链
|
|
86
|
+
*/
|
|
87
|
+
async applyMiddlewares(context, finalHandler) {
|
|
88
|
+
if (this.middlewares.length === 0) {
|
|
89
|
+
return finalHandler();
|
|
90
|
+
}
|
|
91
|
+
let index = 0;
|
|
92
|
+
const next = async () => {
|
|
93
|
+
if (index >= this.middlewares.length) {
|
|
94
|
+
return finalHandler();
|
|
95
|
+
}
|
|
96
|
+
const middleware = this.middlewares[index++];
|
|
97
|
+
return middleware(context, next);
|
|
98
|
+
};
|
|
99
|
+
return next();
|
|
100
|
+
}
|
|
101
|
+
/**
|
|
102
|
+
* 批量调用工具(并行)
|
|
103
|
+
*/
|
|
104
|
+
async callToolsBatch(calls) {
|
|
105
|
+
return Promise.all(calls.map(({ toolName, args, context }) => this.callTool(toolName, args, context)));
|
|
106
|
+
}
|
|
107
|
+
}
|
|
108
|
+
//# sourceMappingURL=router.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"router.js","sourceRoot":"","sources":["../../../src/core/router.ts"],"names":[],"mappings":"AAAA;;GAEG;AAIH,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AACxD,OAAO,EACL,iBAAiB,EACjB,kBAAkB,EAClB,UAAU,GACX,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAE5C;;GAEG;AACH,MAAM,OAAO,UAAU;IAGD;IAFZ,WAAW,GAAiB,EAAE,CAAC;IAEvC,YAAoB,QAAsB;QAAtB,aAAQ,GAAR,QAAQ,CAAc;IAAG,CAAC;IAE9C;;OAEG;IACH,GAAG,CAAC,UAAsB;QACxB,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAClC,MAAM,CAAC,KAAK,CAAC,kBAAkB,EAAE,EAAE,KAAK,EAAE,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC;IACvE,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,QAAQ,CACZ,QAAgB,EAChB,IAA6B,EAC7B,OAAoB;QAEpB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE7B,IAAI,CAAC;YACH,SAAS;YACT,MAAM,UAAU,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YAC/C,IAAI,CAAC,UAAU,EAAE,CAAC;gBAChB,MAAM,IAAI,iBAAiB,CAAC,QAAQ,CAAC,CAAC;YACxC,CAAC;YAED,MAAM,CAAC,KAAK,CAAC,iBAAiB,QAAQ,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;YAE7D,SAAS;YACT,MAAM,aAAa,GAAG,eAAe,CAAC,aAAa,CACjD,IAAI,EACJ,UAAU,CAAC,WAAW,CACvB,CAAC;YAEF,SAAS;YACT,MAAM,OAAO,GAAG,KAAK,IAAyB,EAAE;gBAC9C,IAAI,CAAC;oBACH,UAAU;oBACV,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,OAAO,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC;oBAEhE,2BAA2B;oBAC3B,IAAI,UAAU,CAAC,YAAY,IAAI,MAAM,CAAC,iBAAiB,EAAE,CAAC;wBACxD,eAAe,CAAC,cAAc,CAC5B,MAAM,CAAC,iBAAiB,EACxB,UAAU,CAAC,YAAY,CACxB,CAAC;oBACJ,CAAC;oBAED,OAAO,MAAM,CAAC;gBAChB,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;wBAC3B,MAAM,IAAI,kBAAkB,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;oBAChD,CAAC;oBACD,MAAM,KAAK,CAAC;gBACd,CAAC;YACH,CAAC,CAAC;YAEF,QAAQ;YACR,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAE7D,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YACxC,MAAM,CAAC,IAAI,CAAC,kBAAkB,QAAQ,EAAE,EAAE;gBACxC,QAAQ;gBACR,OAAO,EAAE,CAAC,MAAM,CAAC,OAAO;aACzB,CAAC,CAAC;YAEH,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YACxC,MAAM,QAAQ,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC;YAEnC,MAAM,CAAC,KAAK,CAAC,0BAA0B,QAAQ,EAAE,EAAE;gBACjD,QAAQ;gBACR,KAAK,EAAE,QAAQ,CAAC,OAAO;gBACvB,IAAI,EAAE,QAAQ,CAAC,IAAI;aACpB,CAAC,CAAC;YAEH,OAAO;gBACL,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,QAAQ,CAAC,OAAO;qBACvB;iBACF;gBACD,iBAAiB,EAAE,QAAQ,CAAC,MAAM,EAAE;gBACpC,OAAO,EAAE,IAAI;aACd,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,gBAAgB,CAC5B,OAAoB,EACpB,YAAuC;QAEvC,IAAI,IAAI,CAAC,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAClC,OAAO,YAAY,EAAE,CAAC;QACxB,CAAC;QAED,IAAI,KAAK,GAAG,CAAC,CAAC;QAEd,MAAM,IAAI,GAAG,KAAK,IAAyB,EAAE;YAC3C,IAAI,KAAK,IAAI,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC;gBACrC,OAAO,YAAY,EAAE,CAAC;YACxB,CAAC;YAED,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC,CAAC;YAC7C,OAAO,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QACnC,CAAC,CAAC;QAEF,OAAO,IAAI,EAAE,CAAC;IAChB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,cAAc,CAClB,KAIG;QAEH,OAAO,OAAO,CAAC,GAAG,CAChB,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,CACxC,IAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE,IAAI,EAAE,OAAO,CAAC,CACvC,CACF,CAAC;IACJ,CAAC;CACF"}
|
|
@@ -0,0 +1,55 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* MCP Server 封装
|
|
3
|
+
*/
|
|
4
|
+
import { Server } from '@modelcontextprotocol/sdk/server/index.js';
|
|
5
|
+
import { ToolRegistry } from './registry.js';
|
|
6
|
+
import { ToolRouter } from './router.js';
|
|
7
|
+
import { SessionManager } from './session.js';
|
|
8
|
+
import { type ServerConfig, type ToolConfig } from '../tools/types.js';
|
|
9
|
+
/**
|
|
10
|
+
* MCP 服务器封装类
|
|
11
|
+
*/
|
|
12
|
+
export declare class McpServer {
|
|
13
|
+
private server;
|
|
14
|
+
private registry;
|
|
15
|
+
private router;
|
|
16
|
+
private sessionManager;
|
|
17
|
+
constructor(config: ServerConfig);
|
|
18
|
+
/**
|
|
19
|
+
* 注册工具
|
|
20
|
+
*/
|
|
21
|
+
registerTool(config: ToolConfig): void;
|
|
22
|
+
/**
|
|
23
|
+
* 批量注册工具
|
|
24
|
+
*/
|
|
25
|
+
registerTools(configs: ToolConfig[]): void;
|
|
26
|
+
/**
|
|
27
|
+
* 获取工具注册器
|
|
28
|
+
*/
|
|
29
|
+
getRegistry(): ToolRegistry;
|
|
30
|
+
/**
|
|
31
|
+
* 获取工具路由器
|
|
32
|
+
*/
|
|
33
|
+
getRouter(): ToolRouter;
|
|
34
|
+
/**
|
|
35
|
+
* 获取会话管理器
|
|
36
|
+
*/
|
|
37
|
+
getSessionManager(): SessionManager;
|
|
38
|
+
/**
|
|
39
|
+
* 获取底层 MCP Server 实例
|
|
40
|
+
*/
|
|
41
|
+
getServer(): Server;
|
|
42
|
+
/**
|
|
43
|
+
* 设置 MCP Server 处理器
|
|
44
|
+
*/
|
|
45
|
+
private setupHandlers;
|
|
46
|
+
/**
|
|
47
|
+
* 启动会话清理
|
|
48
|
+
*/
|
|
49
|
+
startSessionCleanup(interval?: number): void;
|
|
50
|
+
/**
|
|
51
|
+
* 关闭服务器
|
|
52
|
+
*/
|
|
53
|
+
close(): Promise<void>;
|
|
54
|
+
}
|
|
55
|
+
//# sourceMappingURL=server.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"server.d.ts","sourceRoot":"","sources":["../../../src/core/server.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AAEnE,OAAO,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAC7C,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAC9C,OAAO,EAAE,KAAK,YAAY,EAAE,KAAK,UAAU,EAAoB,MAAM,mBAAmB,CAAC;AAIzF;;GAEG;AACH,qBAAa,SAAS;IACpB,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,QAAQ,CAAe;IAC/B,OAAO,CAAC,MAAM,CAAa;IAC3B,OAAO,CAAC,cAAc,CAAiB;gBAE3B,MAAM,EAAE,YAAY;IAiChC;;OAEG;IACH,YAAY,CAAC,MAAM,EAAE,UAAU,GAAG,IAAI;IAItC;;OAEG;IACH,aAAa,CAAC,OAAO,EAAE,UAAU,EAAE,GAAG,IAAI;IAI1C;;OAEG;IACH,WAAW,IAAI,YAAY;IAI3B;;OAEG;IACH,SAAS,IAAI,UAAU;IAIvB;;OAEG;IACH,iBAAiB,IAAI,cAAc;IAInC;;OAEG;IACH,SAAS,IAAI,MAAM;IAInB;;OAEG;IACH,OAAO,CAAC,aAAa;IAwErB;;OAEG;IACH,mBAAmB,CAAC,QAAQ,CAAC,EAAE,MAAM,GAAG,IAAI;IAI5C;;OAEG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;CAc7B"}
|
|
@@ -0,0 +1,159 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* MCP Server 封装
|
|
3
|
+
*/
|
|
4
|
+
import { Server } from '@modelcontextprotocol/sdk/server/index.js';
|
|
5
|
+
import { CallToolRequestSchema, ListToolsRequestSchema } from '@modelcontextprotocol/sdk/types.js';
|
|
6
|
+
import { ToolRegistry } from './registry.js';
|
|
7
|
+
import { ToolRouter } from './router.js';
|
|
8
|
+
import { SessionManager } from './session.js';
|
|
9
|
+
import { logger } from '../utils/logger.js';
|
|
10
|
+
import { toMcpError } from '../utils/errors.js';
|
|
11
|
+
/**
|
|
12
|
+
* MCP 服务器封装类
|
|
13
|
+
*/
|
|
14
|
+
export class McpServer {
|
|
15
|
+
server;
|
|
16
|
+
registry;
|
|
17
|
+
router;
|
|
18
|
+
sessionManager;
|
|
19
|
+
constructor(config) {
|
|
20
|
+
// 初始化 MCP Server
|
|
21
|
+
this.server = new Server({
|
|
22
|
+
name: config.name,
|
|
23
|
+
version: config.version,
|
|
24
|
+
}, {
|
|
25
|
+
capabilities: {
|
|
26
|
+
tools: {},
|
|
27
|
+
},
|
|
28
|
+
});
|
|
29
|
+
// 初始化组件
|
|
30
|
+
this.registry = new ToolRegistry();
|
|
31
|
+
this.router = new ToolRouter(this.registry);
|
|
32
|
+
this.sessionManager = new SessionManager(config.sessionTimeout);
|
|
33
|
+
// 注册工具
|
|
34
|
+
if (config.tools) {
|
|
35
|
+
this.registerTools(config.tools);
|
|
36
|
+
}
|
|
37
|
+
// 设置 MCP Server 处理器
|
|
38
|
+
this.setupHandlers();
|
|
39
|
+
logger.info(`MCP Server initialized: ${config.name} v${config.version}`, {
|
|
40
|
+
toolCount: this.registry.count(),
|
|
41
|
+
transport: config.transport,
|
|
42
|
+
});
|
|
43
|
+
}
|
|
44
|
+
/**
|
|
45
|
+
* 注册工具
|
|
46
|
+
*/
|
|
47
|
+
registerTool(config) {
|
|
48
|
+
this.registry.register(config);
|
|
49
|
+
}
|
|
50
|
+
/**
|
|
51
|
+
* 批量注册工具
|
|
52
|
+
*/
|
|
53
|
+
registerTools(configs) {
|
|
54
|
+
this.registry.registerBatch(configs);
|
|
55
|
+
}
|
|
56
|
+
/**
|
|
57
|
+
* 获取工具注册器
|
|
58
|
+
*/
|
|
59
|
+
getRegistry() {
|
|
60
|
+
return this.registry;
|
|
61
|
+
}
|
|
62
|
+
/**
|
|
63
|
+
* 获取工具路由器
|
|
64
|
+
*/
|
|
65
|
+
getRouter() {
|
|
66
|
+
return this.router;
|
|
67
|
+
}
|
|
68
|
+
/**
|
|
69
|
+
* 获取会话管理器
|
|
70
|
+
*/
|
|
71
|
+
getSessionManager() {
|
|
72
|
+
return this.sessionManager;
|
|
73
|
+
}
|
|
74
|
+
/**
|
|
75
|
+
* 获取底层 MCP Server 实例
|
|
76
|
+
*/
|
|
77
|
+
getServer() {
|
|
78
|
+
return this.server;
|
|
79
|
+
}
|
|
80
|
+
/**
|
|
81
|
+
* 设置 MCP Server 处理器
|
|
82
|
+
*/
|
|
83
|
+
setupHandlers() {
|
|
84
|
+
// 处理工具列表请求
|
|
85
|
+
this.server.setRequestHandler(ListToolsRequestSchema, () => {
|
|
86
|
+
logger.debug('Handling tools/list request');
|
|
87
|
+
return {
|
|
88
|
+
tools: this.registry.toMcpTools(),
|
|
89
|
+
};
|
|
90
|
+
});
|
|
91
|
+
// 处理工具调用请求
|
|
92
|
+
this.server.setRequestHandler(CallToolRequestSchema, async (request) => {
|
|
93
|
+
const { name, arguments: args } = request.params;
|
|
94
|
+
logger.debug(`Handling tools/call request: ${name}`);
|
|
95
|
+
// 创建工具执行上下文
|
|
96
|
+
const context = {
|
|
97
|
+
timestamp: Date.now(),
|
|
98
|
+
metadata: request.params._meta,
|
|
99
|
+
};
|
|
100
|
+
try {
|
|
101
|
+
// 调用工具
|
|
102
|
+
const result = await this.router.callTool(name, args || {}, context);
|
|
103
|
+
// 如果是错误结果,返回错误响应
|
|
104
|
+
if (result.isError && result.structuredContent) {
|
|
105
|
+
const errorData = result.structuredContent;
|
|
106
|
+
return {
|
|
107
|
+
content: result.content,
|
|
108
|
+
isError: true,
|
|
109
|
+
_meta: {
|
|
110
|
+
error: errorData,
|
|
111
|
+
},
|
|
112
|
+
};
|
|
113
|
+
}
|
|
114
|
+
return {
|
|
115
|
+
content: result.content,
|
|
116
|
+
...(result.structuredContent && {
|
|
117
|
+
_meta: { structuredContent: result.structuredContent },
|
|
118
|
+
}),
|
|
119
|
+
};
|
|
120
|
+
}
|
|
121
|
+
catch (error) {
|
|
122
|
+
const mcpError = toMcpError(error);
|
|
123
|
+
logger.error('Tool call error', mcpError.toJSON());
|
|
124
|
+
return {
|
|
125
|
+
content: [
|
|
126
|
+
{
|
|
127
|
+
type: 'text',
|
|
128
|
+
text: mcpError.message,
|
|
129
|
+
},
|
|
130
|
+
],
|
|
131
|
+
isError: true,
|
|
132
|
+
_meta: {
|
|
133
|
+
error: mcpError.toJSON(),
|
|
134
|
+
},
|
|
135
|
+
};
|
|
136
|
+
}
|
|
137
|
+
});
|
|
138
|
+
}
|
|
139
|
+
/**
|
|
140
|
+
* 启动会话清理
|
|
141
|
+
*/
|
|
142
|
+
startSessionCleanup(interval) {
|
|
143
|
+
this.sessionManager.startCleanup(interval);
|
|
144
|
+
}
|
|
145
|
+
/**
|
|
146
|
+
* 关闭服务器
|
|
147
|
+
*/
|
|
148
|
+
async close() {
|
|
149
|
+
logger.info('Closing MCP Server...');
|
|
150
|
+
// 停止会话清理
|
|
151
|
+
this.sessionManager.destroy();
|
|
152
|
+
// 清空工具注册
|
|
153
|
+
this.registry.clear();
|
|
154
|
+
// 关闭 MCP Server
|
|
155
|
+
await this.server.close();
|
|
156
|
+
logger.info('MCP Server closed');
|
|
157
|
+
}
|
|
158
|
+
}
|
|
159
|
+
//# sourceMappingURL=server.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"server.js","sourceRoot":"","sources":["../../../src/core/server.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EAAE,qBAAqB,EAAE,sBAAsB,EAAE,MAAM,oCAAoC,CAAC;AACnG,OAAO,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAC7C,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAE9C,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAC5C,OAAO,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAEhD;;GAEG;AACH,MAAM,OAAO,SAAS;IACZ,MAAM,CAAS;IACf,QAAQ,CAAe;IACvB,MAAM,CAAa;IACnB,cAAc,CAAiB;IAEvC,YAAY,MAAoB;QAC9B,iBAAiB;QACjB,IAAI,CAAC,MAAM,GAAG,IAAI,MAAM,CACtB;YACE,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,OAAO,EAAE,MAAM,CAAC,OAAO;SACxB,EACD;YACE,YAAY,EAAE;gBACZ,KAAK,EAAE,EAAE;aACV;SACF,CACF,CAAC;QAEF,QAAQ;QACR,IAAI,CAAC,QAAQ,GAAG,IAAI,YAAY,EAAE,CAAC;QACnC,IAAI,CAAC,MAAM,GAAG,IAAI,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC5C,IAAI,CAAC,cAAc,GAAG,IAAI,cAAc,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;QAEhE,OAAO;QACP,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;YACjB,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACnC,CAAC;QAED,oBAAoB;QACpB,IAAI,CAAC,aAAa,EAAE,CAAC;QAErB,MAAM,CAAC,IAAI,CAAC,2BAA2B,MAAM,CAAC,IAAI,KAAK,MAAM,CAAC,OAAO,EAAE,EAAE;YACvE,SAAS,EAAE,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE;YAChC,SAAS,EAAE,MAAM,CAAC,SAAS;SAC5B,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,YAAY,CAAC,MAAkB;QAC7B,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;IACjC,CAAC;IAED;;OAEG;IACH,aAAa,CAAC,OAAqB;QACjC,IAAI,CAAC,QAAQ,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;IACvC,CAAC;IAED;;OAEG;IACH,WAAW;QACT,OAAO,IAAI,CAAC,QAAQ,CAAC;IACvB,CAAC;IAED;;OAEG;IACH,SAAS;QACP,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAED;;OAEG;IACH,iBAAiB;QACf,OAAO,IAAI,CAAC,cAAc,CAAC;IAC7B,CAAC;IAED;;OAEG;IACH,SAAS;QACP,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAED;;OAEG;IACK,aAAa;QACnB,WAAW;QACX,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,sBAAsB,EAAE,GAAG,EAAE;YACzD,MAAM,CAAC,KAAK,CAAC,6BAA6B,CAAC,CAAC;YAC5C,OAAO;gBACL,KAAK,EAAE,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE;aAClC,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,WAAW;QACX,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,qBAAqB,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;YACrE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC;YAEjD,MAAM,CAAC,KAAK,CAAC,gCAAgC,IAAI,EAAE,CAAC,CAAC;YAErD,YAAY;YACZ,MAAM,OAAO,GAAgB;gBAC3B,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;gBACrB,QAAQ,EAAE,OAAO,CAAC,MAAM,CAAC,KAAK;aAC/B,CAAC;YAEF,IAAI,CAAC;gBACH,OAAO;gBACP,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CACvC,IAAI,EACJ,IAAI,IAAI,EAAE,EACV,OAAO,CACR,CAAC;gBAEF,iBAAiB;gBACjB,IAAI,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,iBAAiB,EAAE,CAAC;oBAC/C,MAAM,SAAS,GAAG,MAAM,CAAC,iBAIxB,CAAC;oBAEF,OAAO;wBACL,OAAO,EAAE,MAAM,CAAC,OAAO;wBACvB,OAAO,EAAE,IAAI;wBACb,KAAK,EAAE;4BACL,KAAK,EAAE,SAAS;yBACjB;qBACF,CAAC;gBACJ,CAAC;gBAED,OAAO;oBACL,OAAO,EAAE,MAAM,CAAC,OAAO;oBACvB,GAAG,CAAC,MAAM,CAAC,iBAAiB,IAAI;wBAC9B,KAAK,EAAE,EAAE,iBAAiB,EAAE,MAAM,CAAC,iBAAiB,EAAE;qBACvD,CAAC;iBACH,CAAC;YACJ,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,QAAQ,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC;gBACnC,MAAM,CAAC,KAAK,CAAC,iBAAiB,EAAE,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;gBAEnD,OAAO;oBACL,OAAO,EAAE;wBACP;4BACE,IAAI,EAAE,MAAM;4BACZ,IAAI,EAAE,QAAQ,CAAC,OAAO;yBACvB;qBACF;oBACD,OAAO,EAAE,IAAI;oBACb,KAAK,EAAE;wBACL,KAAK,EAAE,QAAQ,CAAC,MAAM,EAAE;qBACzB;iBACF,CAAC;YACJ,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,mBAAmB,CAAC,QAAiB;QACnC,IAAI,CAAC,cAAc,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;IAC7C,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK;QACT,MAAM,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;QAErC,SAAS;QACT,IAAI,CAAC,cAAc,CAAC,OAAO,EAAE,CAAC;QAE9B,SAAS;QACT,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;QAEtB,gBAAgB;QAChB,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;QAE1B,MAAM,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;IACnC,CAAC;CACF"}
|
|
@@ -0,0 +1,70 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* 会话管理器
|
|
3
|
+
*/
|
|
4
|
+
import { type Session } from '../tools/types.js';
|
|
5
|
+
/**
|
|
6
|
+
* 会话管理器类
|
|
7
|
+
*/
|
|
8
|
+
export declare class SessionManager {
|
|
9
|
+
private sessions;
|
|
10
|
+
private cleanupInterval;
|
|
11
|
+
private sessionTimeout;
|
|
12
|
+
constructor(sessionTimeout?: number);
|
|
13
|
+
/**
|
|
14
|
+
* 启动会话清理定时器
|
|
15
|
+
*/
|
|
16
|
+
startCleanup(interval?: number): void;
|
|
17
|
+
/**
|
|
18
|
+
* 停止会话清理定时器
|
|
19
|
+
*/
|
|
20
|
+
stopCleanup(): void;
|
|
21
|
+
/**
|
|
22
|
+
* 创建新会话
|
|
23
|
+
*/
|
|
24
|
+
create(sessionId: string, context?: Record<string, unknown>): Session;
|
|
25
|
+
/**
|
|
26
|
+
* 获取会话
|
|
27
|
+
*/
|
|
28
|
+
get(sessionId: string): Session;
|
|
29
|
+
/**
|
|
30
|
+
* 更新会话活跃时间
|
|
31
|
+
*/
|
|
32
|
+
touch(sessionId: string): void;
|
|
33
|
+
/**
|
|
34
|
+
* 更新会话上下文
|
|
35
|
+
*/
|
|
36
|
+
updateContext(sessionId: string, context: Record<string, unknown>): void;
|
|
37
|
+
/**
|
|
38
|
+
* 检查会话是否存在
|
|
39
|
+
*/
|
|
40
|
+
has(sessionId: string): boolean;
|
|
41
|
+
/**
|
|
42
|
+
* 检查会话是否过期
|
|
43
|
+
*/
|
|
44
|
+
isExpired(session: Session): boolean;
|
|
45
|
+
/**
|
|
46
|
+
* 关闭会话
|
|
47
|
+
*/
|
|
48
|
+
close(sessionId: string): boolean;
|
|
49
|
+
/**
|
|
50
|
+
* 清理过期会话
|
|
51
|
+
*/
|
|
52
|
+
private cleanupExpiredSessions;
|
|
53
|
+
/**
|
|
54
|
+
* 获取所有活跃会话
|
|
55
|
+
*/
|
|
56
|
+
getActiveSessions(): Session[];
|
|
57
|
+
/**
|
|
58
|
+
* 获取会话数量
|
|
59
|
+
*/
|
|
60
|
+
count(): number;
|
|
61
|
+
/**
|
|
62
|
+
* 清空所有会话
|
|
63
|
+
*/
|
|
64
|
+
clear(): void;
|
|
65
|
+
/**
|
|
66
|
+
* 销毁会话管理器
|
|
67
|
+
*/
|
|
68
|
+
destroy(): void;
|
|
69
|
+
}
|
|
70
|
+
//# sourceMappingURL=session.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"session.d.ts","sourceRoot":"","sources":["../../../src/core/session.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,KAAK,OAAO,EAAiB,MAAM,mBAAmB,CAAC;AAIhE;;GAEG;AACH,qBAAa,cAAc;IACzB,OAAO,CAAC,QAAQ,CAA8B;IAC9C,OAAO,CAAC,eAAe,CAA+B;IACtD,OAAO,CAAC,cAAc,CAAS;gBAEnB,cAAc,SAAU;IAKpC;;OAEG;IACH,YAAY,CAAC,QAAQ,SAAQ,GAAG,IAAI;IAgBpC;;OAEG;IACH,WAAW,IAAI,IAAI;IAQnB;;OAEG;IACH,MAAM,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,GAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAM,GAAG,OAAO;IAmBzE;;OAEG;IACH,GAAG,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO;IAe/B;;OAEG;IACH,KAAK,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI;IAQ9B;;OAEG;IACH,aAAa,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI;IAMxE;;OAEG;IACH,GAAG,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO;IAI/B;;OAEG;IACH,SAAS,CAAC,OAAO,EAAE,OAAO,GAAG,OAAO;IAKpC;;OAEG;IACH,KAAK,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO;IAejC;;OAEG;IACH,OAAO,CAAC,sBAAsB;IAqB9B;;OAEG;IACH,iBAAiB,IAAI,OAAO,EAAE;IAM9B;;OAEG;IACH,KAAK,IAAI,MAAM;IAIf;;OAEG;IACH,KAAK,IAAI,IAAI;IAMb;;OAEG;IACH,OAAO,IAAI,IAAI;CAIhB"}
|