feihong-code 0.2.2 → 0.5.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/.github/FUNDING.yml +4 -0
- package/.github/ISSUE_TEMPLATE/bug_report.md +37 -37
- package/.github/ISSUE_TEMPLATE/config.yml +14 -14
- package/.github/ISSUE_TEMPLATE/feature_request.md +28 -28
- package/.github/PULL_REQUEST_TEMPLATE.md +46 -46
- package/.github/SECURITY.md +67 -67
- package/.github/dependabot.yml +16 -0
- package/.github/workflows/ci.yml +1 -1
- package/AGENT-GUIDE.md +382 -237
- package/CHANGELOG.md +405 -73
- package/CODE_OF_CONDUCT.md +56 -56
- package/CONTRIBUTING.md +68 -68
- package/LICENSE +22 -22
- package/README.md +31 -25
- package/README.self-evolve.md +274 -0
- package/dist/agent/code-review.js +2 -0
- package/dist/agent/code-review.js.map +1 -1
- package/dist/agent/code-writer.js +70 -17
- package/dist/agent/code-writer.js.map +1 -1
- package/dist/agent/context-compactor.js +13 -4
- package/dist/agent/context-compactor.js.map +1 -1
- package/dist/agent/experience.js +262 -62
- package/dist/agent/experience.js.map +1 -1
- package/dist/agent/orchestrator.js +148 -68
- package/dist/agent/orchestrator.js.map +1 -1
- package/dist/agent/quality-gate.js +10 -4
- package/dist/agent/quality-gate.js.map +1 -1
- package/dist/agent/repo-context.js +181 -0
- package/dist/agent/repo-context.js.map +1 -0
- package/dist/agent/repo-reader.js +92 -99
- package/dist/agent/repo-reader.js.map +1 -1
- package/dist/agent/self-heal.js +38 -48
- package/dist/agent/self-heal.js.map +1 -1
- package/dist/agent/self-improver.js +65 -61
- package/dist/agent/self-improver.js.map +1 -1
- package/dist/agent/subagent-summary.js +31 -0
- package/dist/agent/subagent-summary.js.map +1 -0
- package/dist/agent/subagent.js +52 -2
- package/dist/agent/subagent.js.map +1 -1
- package/dist/agent/symbol-index.js +160 -0
- package/dist/agent/symbol-index.js.map +1 -0
- package/dist/agent/team.js +193 -0
- package/dist/agent/team.js.map +1 -0
- package/dist/cli/commands.js +124 -143
- package/dist/cli/commands.js.map +1 -1
- package/dist/cli/index.js +113 -106
- package/dist/cli/index.js.map +1 -1
- package/dist/cli/repl.js +82 -7
- package/dist/cli/repl.js.map +1 -1
- package/dist/cli/run.js +601 -95
- package/dist/cli/run.js.map +1 -1
- package/dist/cli/tui.js +145 -0
- package/dist/cli/tui.js.map +1 -0
- package/dist/cli/version.js +1 -1
- package/dist/enterprise/audit.js +75 -12
- package/dist/enterprise/audit.js.map +1 -1
- package/dist/enterprise/index.js +14 -11
- package/dist/enterprise/index.js.map +1 -1
- package/dist/enterprise/policy.js +22 -10
- package/dist/enterprise/policy.js.map +1 -1
- package/dist/harness/executor.js +127 -0
- package/dist/harness/executor.js.map +1 -0
- package/dist/harness/harness.js +87 -0
- package/dist/harness/harness.js.map +1 -0
- package/dist/harness/index.js +31 -0
- package/dist/harness/index.js.map +1 -0
- package/dist/harness/loader.js +138 -0
- package/dist/harness/loader.js.map +1 -0
- package/dist/harness/reporter.js +34 -0
- package/dist/harness/reporter.js.map +1 -0
- package/dist/harness/types.js +10 -0
- package/dist/harness/types.js.map +1 -0
- package/dist/harness/verifier.js +48 -0
- package/dist/harness/verifier.js.map +1 -0
- package/dist/hello.js +14 -0
- package/dist/hello.js.map +1 -0
- package/dist/memory/auto-summarize.js +208 -0
- package/dist/memory/auto-summarize.js.map +1 -0
- package/dist/memory/index.js +228 -0
- package/dist/memory/index.js.map +1 -0
- package/dist/models/model-router.js +23 -10
- package/dist/models/model-router.js.map +1 -1
- package/dist/plugins/plugin-loader.js +179 -0
- package/dist/plugins/plugin-loader.js.map +1 -0
- package/dist/runtime/event-log.js.map +1 -1
- package/dist/runtime/hooks.js +80 -0
- package/dist/runtime/hooks.js.map +1 -0
- package/dist/self-evolve/hook.js +60 -0
- package/dist/self-evolve/hook.js.map +1 -0
- package/dist/shared/config.js +35 -3
- package/dist/shared/config.js.map +1 -1
- package/dist/shared/i18n.js +535 -0
- package/dist/shared/i18n.js.map +1 -0
- package/dist/skills/grill.js +2 -1
- package/dist/skills/grill.js.map +1 -1
- package/dist/skills/self-heal.js +73 -0
- package/dist/skills/self-heal.js.map +1 -0
- package/dist/skills/skill-loader.js +131 -0
- package/dist/skills/skill-loader.js.map +1 -0
- package/dist/skills/skill-market.js +195 -0
- package/dist/skills/skill-market.js.map +1 -0
- package/dist/tools/analysis/code-analyzer.js +45 -18
- package/dist/tools/analysis/code-analyzer.js.map +1 -1
- package/dist/tools/index.js +5 -0
- package/dist/tools/index.js.map +1 -1
- package/dist/tools/mcp/index.js +84 -0
- package/dist/tools/mcp/index.js.map +1 -0
- package/dist/tools/mcp/mcp-client.js +172 -0
- package/dist/tools/mcp/mcp-client.js.map +1 -0
- package/dist/tools/sandbox.js +126 -0
- package/dist/tools/sandbox.js.map +1 -0
- package/dist/tools/shell/exec.js +25 -0
- package/dist/tools/shell/exec.js.map +1 -1
- package/dist/tools/shell/run-shell.tool.js +4 -1
- package/dist/tools/shell/run-shell.tool.js.map +1 -1
- package/dist/tools/skills/load-skill.tool.js +36 -0
- package/dist/tools/skills/load-skill.tool.js.map +1 -0
- package/dist/tools/tool.interface.js.map +1 -1
- package/dist/tools/tool.registry.js +37 -1
- package/dist/tools/tool.registry.js.map +1 -1
- package/dist/tools/web/web.tool.js +139 -0
- package/dist/tools/web/web.tool.js.map +1 -0
- package/dist/web/auth.js +42 -3
- package/dist/web/auth.js.map +1 -1
- package/dist/web/channels.js +171 -0
- package/dist/web/channels.js.map +1 -0
- package/dist/web/public/index.html +3253 -39
- package/dist/web/public/index.html.tmp +3117 -0
- package/dist/web/public/index_new.html +3196 -0
- package/dist/web/server.js +790 -11
- package/dist/web/server.js.map +1 -1
- package/dist/web/task-queue.js +352 -0
- package/dist/web/task-queue.js.map +1 -0
- package/dist/web/web-config.js +143 -0
- package/dist/web/web-config.js.map +1 -0
- package/docs/Deployment_Guide_EN.md +288 -0
- package/docs/Technical_Manual_EN.md +216 -0
- package/docs/User_Manual_EN.md +314 -0
- package/docs/self-evolve-implementation.md +165 -0
- package/docs/self-evolve.md +166 -0
- package/docs//344/272/247/345/223/201/345/274/200/345/217/221/346/226/207/346/241/243.md +614 -614
- package/docs//344/274/201/344/270/232/351/203/250/347/275/262/344/270/216/345/220/210/350/247/204.md +267 -267
- package/docs//344/275/277/347/224/250/350/257/264/346/230/216/344/271/246.md +318 -358
- package/docs//345/270/270/350/247/201/351/227/256/351/242/230/344/270/216/346/225/205/351/232/234/346/216/222/346/237/245.md +130 -130
- package/docs//346/212/200/346/234/257/350/257/264/346/230/216/344/271/246.md +216 -303
- package/docs//346/236/266/346/236/204/344/270/216API.md +238 -238
- package/docs//347/224/250/346/210/267/346/211/213/345/206/214.md +196 -196
- package/docs//351/203/250/347/275/262/346/214/207/345/215/227.md +165 -165
- package/docs//351/203/250/347/275/262/350/257/264/346/230/216/344/271/246.md +288 -0
- package/docs//351/205/215/347/275/256/345/217/202/350/200/203.md +127 -127
- package/package.json +114 -109
- package/tool-schema.json +127 -127
|
@@ -0,0 +1,84 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.parseMcpServers = parseMcpServers;
|
|
4
|
+
exports.attachMcpTools = attachMcpTools;
|
|
5
|
+
exports.closeMcpClients = closeMcpClients;
|
|
6
|
+
/**
|
|
7
|
+
* 飞虹 Code (Muse Code 参照复刻)
|
|
8
|
+
* 晋江市飞虹智科技企业管理有限公司 · 飞扬企源研发中心 · 负责人:吴赐虹
|
|
9
|
+
*
|
|
10
|
+
* P0-3 MCP 工具适配层:把 MCP 服务器的工具包装成本地 Tool 注册进 ToolRegistry。
|
|
11
|
+
* - 工具名以 `<serverName>_<toolName>` 前缀注册,避免与内置工具冲突
|
|
12
|
+
* - 调用时透传 JSON-RPC tools/call,结果归一为 ToolResult
|
|
13
|
+
* - 沙箱/守卫在 ToolRegistry.execute 层统一生效(MCP 工具同样受约束)
|
|
14
|
+
*/
|
|
15
|
+
const zod_1 = require("zod");
|
|
16
|
+
const mcp_client_1 = require("./mcp-client");
|
|
17
|
+
const logger_1 = require("../../shared/logger");
|
|
18
|
+
/** MCP 服务器配置解析:支持 FH_MCP_SERVERS(JSON 数组)与配置文件 mcpServers 字段 */
|
|
19
|
+
function parseMcpServers(raw) {
|
|
20
|
+
if (!raw?.trim())
|
|
21
|
+
return [];
|
|
22
|
+
try {
|
|
23
|
+
const parsed = JSON.parse(raw);
|
|
24
|
+
if (!Array.isArray(parsed))
|
|
25
|
+
return [];
|
|
26
|
+
return parsed
|
|
27
|
+
.filter((s) => !!s && typeof s === 'object' && typeof s.name === 'string' && typeof s.command === 'string')
|
|
28
|
+
.map((s) => ({ args: [], ...s }));
|
|
29
|
+
}
|
|
30
|
+
catch {
|
|
31
|
+
logger_1.logger.warn('FH_MCP_SERVERS 不是合法 JSON,已忽略');
|
|
32
|
+
return [];
|
|
33
|
+
}
|
|
34
|
+
}
|
|
35
|
+
/** 包装一个 MCP 工具为本地 Tool(schema 采用宽松 object,运行时再转发给服务器校验) */
|
|
36
|
+
function wrapMcpTool(serverName, tool, client) {
|
|
37
|
+
const fullName = `${serverName}_${tool.name}`;
|
|
38
|
+
return {
|
|
39
|
+
name: fullName,
|
|
40
|
+
description: `[MCP ${serverName}] ${tool.description ?? tool.name}`,
|
|
41
|
+
jsonSchema: (tool.inputSchema ?? { type: 'object', properties: {} }),
|
|
42
|
+
schema: zod_1.z.record(zod_1.z.string(), zod_1.z.unknown()),
|
|
43
|
+
async execute(args, _ctx) {
|
|
44
|
+
try {
|
|
45
|
+
const res = await client.callTool(tool.name, args);
|
|
46
|
+
return res.ok
|
|
47
|
+
? { ok: true, output: res.output || '(MCP 无文本输出)' }
|
|
48
|
+
: { ok: false, output: '', error: res.error ?? 'MCP 工具调用失败' };
|
|
49
|
+
}
|
|
50
|
+
catch (e) {
|
|
51
|
+
const msg = e instanceof Error ? e.message : String(e);
|
|
52
|
+
logger_1.logger.warn('MCP tool call failed', { tool: fullName, error: msg });
|
|
53
|
+
return { ok: false, output: '', error: msg };
|
|
54
|
+
}
|
|
55
|
+
},
|
|
56
|
+
};
|
|
57
|
+
}
|
|
58
|
+
/** 连接一组 MCP 服务器并把它们的工具注册进 registry;返回已连接客户端(供关闭) */
|
|
59
|
+
async function attachMcpTools(registry, servers) {
|
|
60
|
+
const clients = [];
|
|
61
|
+
for (const cfg of servers) {
|
|
62
|
+
try {
|
|
63
|
+
const client = await (0, mcp_client_1.connectMcp)(cfg);
|
|
64
|
+
const tools = await client.listTools();
|
|
65
|
+
for (const t of tools) {
|
|
66
|
+
registry.register(wrapMcpTool(cfg.name, t, client));
|
|
67
|
+
}
|
|
68
|
+
clients.push(client);
|
|
69
|
+
logger_1.logger.info('MCP server attached', { name: cfg.name, tools: tools.length });
|
|
70
|
+
}
|
|
71
|
+
catch (e) {
|
|
72
|
+
logger_1.logger.warn('MCP server attach failed(跳过,不影响主流程)', {
|
|
73
|
+
name: cfg.name,
|
|
74
|
+
error: e instanceof Error ? e.message : String(e),
|
|
75
|
+
});
|
|
76
|
+
}
|
|
77
|
+
}
|
|
78
|
+
return clients;
|
|
79
|
+
}
|
|
80
|
+
/** 关闭全部 MCP 客户端 */
|
|
81
|
+
async function closeMcpClients(clients) {
|
|
82
|
+
await Promise.all(clients.map((c) => c.close().catch(() => undefined)));
|
|
83
|
+
}
|
|
84
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/tools/mcp/index.ts"],"names":[],"mappings":";;AAeA,0CAYC;AA0BD,wCAmBC;AAGD,0CAEC;AA7ED;;;;;;;;GAQG;AACH,6BAAwB;AAExB,6CAA2E;AAC3E,gDAA6C;AAE7C,gEAAgE;AAChE,SAAgB,eAAe,CAAC,GAAmB;IACjD,IAAI,CAAC,GAAG,EAAE,IAAI,EAAE;QAAE,OAAO,EAAE,CAAC;IAC5B,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAC/B,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC;YAAE,OAAO,EAAE,CAAC;QACtC,OAAO,MAAM;aACV,MAAM,CAAC,CAAC,CAAC,EAAwB,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,OAAO,CAAC,KAAK,QAAQ,IAAI,OAAO,CAAC,CAAC,IAAI,KAAK,QAAQ,IAAI,OAAO,CAAC,CAAC,OAAO,KAAK,QAAQ,CAAC;aAChI,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;IACtC,CAAC;IAAC,MAAM,CAAC;QACP,eAAM,CAAC,IAAI,CAAC,8BAA8B,CAAC,CAAC;QAC5C,OAAO,EAAE,CAAC;IACZ,CAAC;AACH,CAAC;AAED,2DAA2D;AAC3D,SAAS,WAAW,CAAC,UAAkB,EAAE,IAAmF,EAAE,MAAiB;IAC7I,MAAM,QAAQ,GAAG,GAAG,UAAU,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;IAC9C,OAAO;QACL,IAAI,EAAE,QAAQ;QACd,WAAW,EAAE,QAAQ,UAAU,KAAK,IAAI,CAAC,WAAW,IAAI,IAAI,CAAC,IAAI,EAAE;QACnE,UAAU,EAAE,CAAC,IAAI,CAAC,WAAW,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,EAAE,CAA4B;QAC/F,MAAM,EAAE,OAAC,CAAC,MAAM,CAAC,OAAC,CAAC,MAAM,EAAE,EAAE,OAAC,CAAC,OAAO,EAAE,CAAC;QACzC,KAAK,CAAC,OAAO,CAAC,IAA6B,EAAE,IAAiB;YAC5D,IAAI,CAAC;gBACH,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;gBACnD,OAAO,GAAG,CAAC,EAAE;oBACX,CAAC,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,IAAI,aAAa,EAAE;oBACnD,CAAC,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,GAAG,CAAC,KAAK,IAAI,YAAY,EAAE,CAAC;YAClE,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,MAAM,GAAG,GAAG,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;gBACvD,eAAM,CAAC,IAAI,CAAC,sBAAsB,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;gBACpE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC;YAC/C,CAAC;QACH,CAAC;KACF,CAAC;AACJ,CAAC;AAED,oDAAoD;AAC7C,KAAK,UAAU,cAAc,CAAC,QAAqC,EAAE,OAA0B;IACpG,MAAM,OAAO,GAAgB,EAAE,CAAC;IAChC,KAAK,MAAM,GAAG,IAAI,OAAO,EAAE,CAAC;QAC1B,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAA,uBAAU,EAAC,GAAG,CAAC,CAAC;YACrC,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC,SAAS,EAAE,CAAC;YACvC,KAAK,MAAM,CAAC,IAAI,KAAK,EAAE,CAAC;gBACtB,QAAQ,CAAC,QAAQ,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC;YACtD,CAAC;YACD,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACrB,eAAM,CAAC,IAAI,CAAC,qBAAqB,EAAE,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,EAAE,KAAK,EAAE,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;QAC9E,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,eAAM,CAAC,IAAI,CAAC,qCAAqC,EAAE;gBACjD,IAAI,EAAE,GAAG,CAAC,IAAI;gBACd,KAAK,EAAE,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;aAClD,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,mBAAmB;AACZ,KAAK,UAAU,eAAe,CAAC,OAAoB;IACxD,MAAM,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;AAC1E,CAAC"}
|
|
@@ -0,0 +1,172 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.McpClient = void 0;
|
|
4
|
+
exports.connectMcp = connectMcp;
|
|
5
|
+
/**
|
|
6
|
+
* 飞虹 Code (Muse Code 参照复刻)
|
|
7
|
+
* 晋江市飞虹智科技企业管理有限公司 · 飞扬企源研发中心 · 负责人:吴赐虹
|
|
8
|
+
*
|
|
9
|
+
* P0-3 轻量 MCP(Model Context Protocol)stdio 客户端。
|
|
10
|
+
*
|
|
11
|
+
* 设计原则:
|
|
12
|
+
* - 零第三方依赖:纯 Node 内置模块(child_process / readline),离线可用。
|
|
13
|
+
* - 传输:MCP 官方 stdio transport 采用"每行一个 JSON-RPC 2.0 消息"(NDJSON)。
|
|
14
|
+
* - 生命周期:spawn 子进程 → initialize 握手 → notifications/initialized → tools/list → tools/call。
|
|
15
|
+
* - 约定:服务器一切非协议输出写 stderr(协议规范要求),stdout 仅走协议;本客户端将 stderr 归入日志。
|
|
16
|
+
*/
|
|
17
|
+
const child_process_1 = require("child_process");
|
|
18
|
+
const readline_1 = require("readline");
|
|
19
|
+
const logger_1 = require("../../shared/logger");
|
|
20
|
+
/** JSON-RPC 请求/通知写入 stdin(NDJSON 一行一条) */
|
|
21
|
+
function writeMessage(child, msg) {
|
|
22
|
+
child.stdin?.write(JSON.stringify(msg) + '\n');
|
|
23
|
+
}
|
|
24
|
+
/**
|
|
25
|
+
* MCP stdio 客户端实例。
|
|
26
|
+
* 用法:await client.connect() → await client.listTools() → await client.callTool(name, args) → await client.close()。
|
|
27
|
+
*/
|
|
28
|
+
class McpClient {
|
|
29
|
+
cfg;
|
|
30
|
+
child = null;
|
|
31
|
+
rl = null;
|
|
32
|
+
pending = new Map();
|
|
33
|
+
idSeq = 0;
|
|
34
|
+
connected = false;
|
|
35
|
+
stderrBuf = '';
|
|
36
|
+
constructor(cfg) {
|
|
37
|
+
this.cfg = cfg;
|
|
38
|
+
}
|
|
39
|
+
/** 启动子进程并完成 initialize 握手 */
|
|
40
|
+
async connect() {
|
|
41
|
+
if (this.connected)
|
|
42
|
+
return;
|
|
43
|
+
const { command, args = [], cwd, env, name } = this.cfg;
|
|
44
|
+
this.child = (0, child_process_1.spawn)(command, args, {
|
|
45
|
+
cwd,
|
|
46
|
+
env: env ? { ...process.env, ...env } : process.env,
|
|
47
|
+
stdio: ['pipe', 'pipe', 'pipe'],
|
|
48
|
+
windowsHide: true,
|
|
49
|
+
});
|
|
50
|
+
this.rl = (0, readline_1.createInterface)({ input: this.child.stdout ?? undefined });
|
|
51
|
+
this.rl.on('line', (line) => this.handleLine(line));
|
|
52
|
+
this.child.stderr?.on('data', (d) => {
|
|
53
|
+
this.stderrBuf = (this.stderrBuf + d.toString()).slice(-4000);
|
|
54
|
+
});
|
|
55
|
+
this.child.on('error', (e) => this.rejectAll(`MCP 进程启动失败: ${e.message}`));
|
|
56
|
+
this.child.on('exit', (code) => {
|
|
57
|
+
this.connected = false;
|
|
58
|
+
this.rejectAll(`MCP 进程退出(code=${code})`);
|
|
59
|
+
});
|
|
60
|
+
// initialize 握手(协议版本用官方最新稳定版)
|
|
61
|
+
const initResp = await this.request('initialize', {
|
|
62
|
+
protocolVersion: '2024-11-05',
|
|
63
|
+
capabilities: {},
|
|
64
|
+
clientInfo: { name: 'feihong-code', version: '0.5.0' },
|
|
65
|
+
}, this.cfg.initTimeoutMs ?? 15000);
|
|
66
|
+
if (!initResp.result) {
|
|
67
|
+
throw new Error(`MCP ${name} initialize 失败: ${initResp.error?.message ?? '无响应'}`);
|
|
68
|
+
}
|
|
69
|
+
// 通知服务器客户端已就绪
|
|
70
|
+
writeMessage(this.child, { jsonrpc: '2.0', method: 'notifications/initialized' });
|
|
71
|
+
this.connected = true;
|
|
72
|
+
}
|
|
73
|
+
/** 列出服务器工具 */
|
|
74
|
+
async listTools() {
|
|
75
|
+
const resp = await this.request('tools/list', {});
|
|
76
|
+
if (!resp.result) {
|
|
77
|
+
throw new Error(`tools/list 失败: ${resp.error?.message ?? '无响应'}`);
|
|
78
|
+
}
|
|
79
|
+
const tools = resp.result.tools ?? [];
|
|
80
|
+
return tools;
|
|
81
|
+
}
|
|
82
|
+
/** 调用服务器工具(参数自动序列化,文本结果拼接返回) */
|
|
83
|
+
async callTool(toolName, args) {
|
|
84
|
+
const resp = await this.request('tools/call', { name: toolName, arguments: args }, this.cfg.callTimeoutMs ?? 120000);
|
|
85
|
+
if (!resp.result) {
|
|
86
|
+
return { ok: false, output: '', error: `tools/call 失败: ${resp.error?.message ?? '无响应'}` };
|
|
87
|
+
}
|
|
88
|
+
const result = resp.result;
|
|
89
|
+
const text = (result.content ?? [])
|
|
90
|
+
.map((c) => c.text ?? '')
|
|
91
|
+
.join('\n');
|
|
92
|
+
return {
|
|
93
|
+
ok: !result.isError,
|
|
94
|
+
output: text,
|
|
95
|
+
error: result.isError ? text.slice(0, 500) : undefined,
|
|
96
|
+
};
|
|
97
|
+
}
|
|
98
|
+
/** 关闭连接(结束子进程) */
|
|
99
|
+
async close() {
|
|
100
|
+
this.connected = false;
|
|
101
|
+
this.rl?.close();
|
|
102
|
+
this.rl = null;
|
|
103
|
+
if (this.child && this.child.exitCode === null) {
|
|
104
|
+
this.child.kill();
|
|
105
|
+
}
|
|
106
|
+
this.child = null;
|
|
107
|
+
}
|
|
108
|
+
/** stderr 摘要(诊断用) */
|
|
109
|
+
get stderrTail() {
|
|
110
|
+
return this.stderrBuf;
|
|
111
|
+
}
|
|
112
|
+
handleLine(line) {
|
|
113
|
+
if (!line.trim())
|
|
114
|
+
return;
|
|
115
|
+
let msg;
|
|
116
|
+
try {
|
|
117
|
+
msg = JSON.parse(line);
|
|
118
|
+
}
|
|
119
|
+
catch {
|
|
120
|
+
logger_1.logger.warn('MCP 收到非 JSON 行,已忽略', { server: this.cfg.name, line: line.slice(0, 200) });
|
|
121
|
+
return;
|
|
122
|
+
}
|
|
123
|
+
if (msg.id !== undefined) {
|
|
124
|
+
const entry = this.pending.get(msg.id);
|
|
125
|
+
if (entry) {
|
|
126
|
+
this.pending.delete(msg.id);
|
|
127
|
+
entry.resolve(msg);
|
|
128
|
+
}
|
|
129
|
+
}
|
|
130
|
+
// 服务器主动通知(如 logs/listened)直接忽略
|
|
131
|
+
}
|
|
132
|
+
request(method, params, timeoutMs = 120000) {
|
|
133
|
+
if (!this.child || this.child.exitCode !== null) {
|
|
134
|
+
return Promise.reject(new Error('MCP 进程未运行'));
|
|
135
|
+
}
|
|
136
|
+
const id = ++this.idSeq;
|
|
137
|
+
return new Promise((resolve, reject) => {
|
|
138
|
+
const timer = setTimeout(() => {
|
|
139
|
+
this.pending.delete(id);
|
|
140
|
+
reject(new Error(`MCP 请求超时(${method} > ${timeoutMs}ms)`));
|
|
141
|
+
}, timeoutMs);
|
|
142
|
+
this.pending.set(id, {
|
|
143
|
+
resolve: (msg) => {
|
|
144
|
+
clearTimeout(timer);
|
|
145
|
+
resolve(msg);
|
|
146
|
+
},
|
|
147
|
+
reject,
|
|
148
|
+
});
|
|
149
|
+
writeMessage(this.child, { jsonrpc: '2.0', id, method, params });
|
|
150
|
+
});
|
|
151
|
+
}
|
|
152
|
+
rejectAll(reason) {
|
|
153
|
+
for (const [, entry] of this.pending) {
|
|
154
|
+
entry.reject(new Error(reason));
|
|
155
|
+
}
|
|
156
|
+
this.pending.clear();
|
|
157
|
+
}
|
|
158
|
+
}
|
|
159
|
+
exports.McpClient = McpClient;
|
|
160
|
+
/** 便捷工厂:创建客户端并完成握手 */
|
|
161
|
+
async function connectMcp(cfg) {
|
|
162
|
+
const client = new McpClient(cfg);
|
|
163
|
+
try {
|
|
164
|
+
await client.connect();
|
|
165
|
+
return client;
|
|
166
|
+
}
|
|
167
|
+
catch (e) {
|
|
168
|
+
await client.close().catch(() => undefined);
|
|
169
|
+
throw e;
|
|
170
|
+
}
|
|
171
|
+
}
|
|
172
|
+
//# sourceMappingURL=mcp-client.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"mcp-client.js","sourceRoot":"","sources":["../../../src/tools/mcp/mcp-client.ts"],"names":[],"mappings":";;;AA2MA,gCASC;AApND;;;;;;;;;;;GAWG;AACH,iDAAyD;AACzD,uCAA2D;AAC3D,gDAA6C;AAsC7C,0CAA0C;AAC1C,SAAS,YAAY,CAAC,KAAmB,EAAE,GAAY;IACrD,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,CAAC;AACjD,CAAC;AAED;;;GAGG;AACH,MAAa,SAAS;IAWS;IAVrB,KAAK,GAAwB,IAAI,CAAC;IAClC,EAAE,GAAqB,IAAI,CAAC;IACnB,OAAO,GAAG,IAAI,GAAG,EAG/B,CAAC;IACI,KAAK,GAAG,CAAC,CAAC;IACV,SAAS,GAAG,KAAK,CAAC;IAClB,SAAS,GAAG,EAAE,CAAC;IAEvB,YAA6B,GAAoB;QAApB,QAAG,GAAH,GAAG,CAAiB;IAAG,CAAC;IAErD,6BAA6B;IAC7B,KAAK,CAAC,OAAO;QACX,IAAI,IAAI,CAAC,SAAS;YAAE,OAAO;QAC3B,MAAM,EAAE,OAAO,EAAE,IAAI,GAAG,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,IAAI,CAAC,GAAG,CAAC;QACxD,IAAI,CAAC,KAAK,GAAG,IAAA,qBAAK,EAAC,OAAO,EAAE,IAAI,EAAE;YAChC,GAAG;YACH,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,OAAO,CAAC,GAAG,EAAE,GAAG,GAAG,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG;YACnD,KAAK,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC;YAC/B,WAAW,EAAE,IAAI;SAClB,CAAC,CAAC;QAEH,IAAI,CAAC,EAAE,GAAG,IAAA,0BAAe,EAAC,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,MAAM,IAAK,SAAmB,EAAE,CAAC,CAAC;QAChF,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC;QAEpD,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,CAAS,EAAE,EAAE;YAC1C,IAAI,CAAC,SAAS,GAAG,CAAC,IAAI,CAAC,SAAS,GAAG,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;QAC1E,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE;YAC7B,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;YACvB,IAAI,CAAC,SAAS,CAAC,iBAAiB,IAAI,GAAG,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;QAEH,8BAA8B;QAC9B,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE;YAChD,eAAe,EAAE,YAAY;YAC7B,YAAY,EAAE,EAAE;YAChB,UAAU,EAAE,EAAE,IAAI,EAAE,cAAc,EAAE,OAAO,EAAE,OAAO,EAAE;SACvD,EAAE,IAAI,CAAC,GAAG,CAAC,aAAa,IAAI,KAAK,CAAC,CAAC;QACpC,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC;YACrB,MAAM,IAAI,KAAK,CAAC,OAAO,IAAI,mBAAmB,QAAQ,CAAC,KAAK,EAAE,OAAO,IAAI,KAAK,EAAE,CAAC,CAAC;QACpF,CAAC;QACD,cAAc;QACd,YAAY,CAAC,IAAI,CAAC,KAAK,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,2BAA2B,EAAE,CAAC,CAAC;QAClF,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;IACxB,CAAC;IAED,cAAc;IACd,KAAK,CAAC,SAAS;QACb,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC;QAClD,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,kBAAkB,IAAI,CAAC,KAAK,EAAE,OAAO,IAAI,KAAK,EAAE,CAAC,CAAC;QACpE,CAAC;QACD,MAAM,KAAK,GAAI,IAAI,CAAC,MAAmC,CAAC,KAAK,IAAI,EAAE,CAAC;QACpE,OAAO,KAAK,CAAC;IACf,CAAC;IAED,gCAAgC;IAChC,KAAK,CAAC,QAAQ,CAAC,QAAgB,EAAE,IAA6B;QAC5D,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,SAAS,EAAE,IAAI,EAAE,EAAE,IAAI,CAAC,GAAG,CAAC,aAAa,IAAI,MAAM,CAAC,CAAC;QACrH,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,kBAAkB,IAAI,CAAC,KAAK,EAAE,OAAO,IAAI,KAAK,EAAE,EAAE,CAAC;QAC5F,CAAC;QACD,MAAM,MAAM,GAAG,IAAI,CAAC,MAAkF,CAAC;QACvG,MAAM,IAAI,GAAG,CAAC,MAAM,CAAC,OAAO,IAAI,EAAE,CAAC;aAChC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,IAAI,EAAE,CAAC;aACxB,IAAI,CAAC,IAAI,CAAC,CAAC;QACd,OAAO;YACL,EAAE,EAAE,CAAC,MAAM,CAAC,OAAO;YACnB,MAAM,EAAE,IAAI;YACZ,KAAK,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS;SACvD,CAAC;IACJ,CAAC;IAED,kBAAkB;IAClB,KAAK,CAAC,KAAK;QACT,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;QACvB,IAAI,CAAC,EAAE,EAAE,KAAK,EAAE,CAAC;QACjB,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC;QACf,IAAI,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,QAAQ,KAAK,IAAI,EAAE,CAAC;YAC/C,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;QACpB,CAAC;QACD,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;IACpB,CAAC;IAED,qBAAqB;IACrB,IAAI,UAAU;QACZ,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAEO,UAAU,CAAC,IAAY;QAC7B,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE;YAAE,OAAO;QACzB,IAAI,GAAoB,CAAC;QACzB,IAAI,CAAC;YACH,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAoB,CAAC;QAC5C,CAAC;QAAC,MAAM,CAAC;YACP,eAAM,CAAC,IAAI,CAAC,oBAAoB,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC;YACvF,OAAO;QACT,CAAC;QACD,IAAI,GAAG,CAAC,EAAE,KAAK,SAAS,EAAE,CAAC;YACzB,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YACvC,IAAI,KAAK,EAAE,CAAC;gBACV,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBAC5B,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;YACrB,CAAC;QACH,CAAC;QACD,+BAA+B;IACjC,CAAC;IAEO,OAAO,CAAC,MAAc,EAAE,MAA+B,EAAE,SAAS,GAAG,MAAM;QACjF,IAAI,CAAC,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,QAAQ,KAAK,IAAI,EAAE,CAAC;YAChD,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC;QAChD,CAAC;QACD,MAAM,EAAE,GAAG,EAAE,IAAI,CAAC,KAAK,CAAC;QACxB,OAAO,IAAI,OAAO,CAAkB,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACtD,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE;gBAC5B,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;gBACxB,MAAM,CAAC,IAAI,KAAK,CAAC,YAAY,MAAM,MAAM,SAAS,KAAK,CAAC,CAAC,CAAC;YAC5D,CAAC,EAAE,SAAS,CAAC,CAAC;YACd,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,EAAE;gBACnB,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;oBACf,YAAY,CAAC,KAAK,CAAC,CAAC;oBACpB,OAAO,CAAC,GAAG,CAAC,CAAC;gBACf,CAAC;gBACD,MAAM;aACP,CAAC,CAAC;YACH,YAAY,CAAC,IAAI,CAAC,KAAM,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;QACpE,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,SAAS,CAAC,MAAc;QAC9B,KAAK,MAAM,CAAC,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACrC,KAAK,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;QAClC,CAAC;QACD,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;IACvB,CAAC;CACF;AA3ID,8BA2IC;AAED,sBAAsB;AACf,KAAK,UAAU,UAAU,CAAC,GAAoB;IACnD,MAAM,MAAM,GAAG,IAAI,SAAS,CAAC,GAAG,CAAC,CAAC;IAClC,IAAI,CAAC;QACH,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;QACvB,OAAO,MAAM,CAAC;IAChB,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QAC5C,MAAM,CAAC,CAAC;IACV,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1,126 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.normalizeSandboxMode = normalizeSandboxMode;
|
|
4
|
+
exports.extractNetworkHosts = extractNetworkHosts;
|
|
5
|
+
exports.checkNetworkUrl = checkNetworkUrl;
|
|
6
|
+
exports.checkSandbox = checkSandbox;
|
|
7
|
+
exports.describeSandboxMode = describeSandboxMode;
|
|
8
|
+
/** 规范化沙箱模式:非法值回退默认 workspace-write */
|
|
9
|
+
function normalizeSandboxMode(raw) {
|
|
10
|
+
const v = raw?.trim().toLowerCase();
|
|
11
|
+
if (v === 'read-only' || v === 'readonly')
|
|
12
|
+
return 'read-only';
|
|
13
|
+
if (v === 'danger-full-access' || v === 'full-access' || v === 'full')
|
|
14
|
+
return 'danger-full-access';
|
|
15
|
+
if (v === 'container' || v === 'docker')
|
|
16
|
+
return 'container';
|
|
17
|
+
return 'workspace-write';
|
|
18
|
+
}
|
|
19
|
+
/** 写类工具:read-only 模式下禁止 */
|
|
20
|
+
const WRITE_TOOLS = new Set(['write_file', 'edit_file']);
|
|
21
|
+
/** 提取命令中出现的 http(s) 目标主机名(简单解析,不做 DNS 解析) */
|
|
22
|
+
function extractNetworkHosts(cmd) {
|
|
23
|
+
const hosts = new Set();
|
|
24
|
+
const re = /https?:\/\/([^/\s'"]+)/gi;
|
|
25
|
+
let m;
|
|
26
|
+
while ((m = re.exec(cmd)) !== null) {
|
|
27
|
+
let host = m[1].toLowerCase();
|
|
28
|
+
// 剥离端口与路径尾(host:port 或 host/path 已由正则排除 /,仅需剥端口)
|
|
29
|
+
const colon = host.indexOf(':');
|
|
30
|
+
if (colon > 0)
|
|
31
|
+
host = host.slice(0, colon);
|
|
32
|
+
if (host)
|
|
33
|
+
hosts.add(host);
|
|
34
|
+
}
|
|
35
|
+
return [...hosts];
|
|
36
|
+
}
|
|
37
|
+
function hostMatches(rule, host) {
|
|
38
|
+
const r = rule.trim().toLowerCase();
|
|
39
|
+
if (!r)
|
|
40
|
+
return false;
|
|
41
|
+
// 精确匹配,或规则为子域通配(如 .github.com 匹配 api.github.com)
|
|
42
|
+
if (r.startsWith('.'))
|
|
43
|
+
return host === r.slice(1) || host.endsWith(r);
|
|
44
|
+
return host === r;
|
|
45
|
+
}
|
|
46
|
+
/** 网络域名级检查(web_fetch/web_search 等 URL 入参工具复用):返回 null 表示放行 */
|
|
47
|
+
function checkNetworkUrl(rules, url) {
|
|
48
|
+
let target = '';
|
|
49
|
+
try {
|
|
50
|
+
const u = new URL(url);
|
|
51
|
+
target = u.hostname.toLowerCase();
|
|
52
|
+
}
|
|
53
|
+
catch {
|
|
54
|
+
// 非法 URL:交给工具自身报错,沙箱不额外拦截
|
|
55
|
+
return null;
|
|
56
|
+
}
|
|
57
|
+
if (rules.networkDeny.some((r) => hostMatches(r, target))) {
|
|
58
|
+
return `目标域名 ${target} 命中网络黑名单`;
|
|
59
|
+
}
|
|
60
|
+
if (rules.networkAllow.length > 0 && !rules.networkAllow.some((r) => hostMatches(r, target))) {
|
|
61
|
+
return `目标域名 ${target} 不在网络白名单`;
|
|
62
|
+
}
|
|
63
|
+
return null;
|
|
64
|
+
}
|
|
65
|
+
/** 沙箱判定(纯函数):先 deny 域名 → 再 read-only 写限制 → 网络 allow 约束 */
|
|
66
|
+
function checkSandbox(mode, tool, args, rules) {
|
|
67
|
+
// 域名黑名单:无论何种模式都拦截(安全基线,与 danger-full-access 也生效)
|
|
68
|
+
const cmd = String(args.command ?? args.cmd ?? '');
|
|
69
|
+
if (tool === 'run_shell' && cmd) {
|
|
70
|
+
const hosts = extractNetworkHosts(cmd);
|
|
71
|
+
for (const host of hosts) {
|
|
72
|
+
if (rules.networkDeny.some((r) => hostMatches(r, host))) {
|
|
73
|
+
return { blocked: true, reason: `目标域名 ${host} 命中网络黑名单` };
|
|
74
|
+
}
|
|
75
|
+
}
|
|
76
|
+
}
|
|
77
|
+
if (mode === 'danger-full-access')
|
|
78
|
+
return { blocked: false };
|
|
79
|
+
// read-only:禁写文件 + 禁 shell
|
|
80
|
+
if (mode === 'read-only') {
|
|
81
|
+
if (WRITE_TOOLS.has(tool)) {
|
|
82
|
+
return { blocked: true, reason: 'read-only 模式禁止写文件(仅允许 read/list/grep)' };
|
|
83
|
+
}
|
|
84
|
+
if (tool === 'run_shell') {
|
|
85
|
+
return { blocked: true, reason: 'read-only 模式禁止执行 shell 命令' };
|
|
86
|
+
}
|
|
87
|
+
return { blocked: false };
|
|
88
|
+
}
|
|
89
|
+
// container(P5-4):文件工具放行(容器内无文件工具),shell 交给容器执行层;
|
|
90
|
+
// 网络域名规则仍生效(容器执行同样受 allow/deny 约束)
|
|
91
|
+
if (mode === 'container') {
|
|
92
|
+
if (tool === 'run_shell' && cmd && rules.networkAllow.length > 0) {
|
|
93
|
+
const hosts = extractNetworkHosts(cmd);
|
|
94
|
+
for (const host of hosts) {
|
|
95
|
+
if (!rules.networkAllow.some((r) => hostMatches(r, host))) {
|
|
96
|
+
return { blocked: true, reason: `目标域名 ${host} 不在网络白名单` };
|
|
97
|
+
}
|
|
98
|
+
}
|
|
99
|
+
}
|
|
100
|
+
return { blocked: false };
|
|
101
|
+
}
|
|
102
|
+
// workspace-write:网络 allow 白名单约束(配置了 allow 列表时,未命中即拦截)
|
|
103
|
+
if (tool === 'run_shell' && cmd && rules.networkAllow.length > 0) {
|
|
104
|
+
const hosts = extractNetworkHosts(cmd);
|
|
105
|
+
for (const host of hosts) {
|
|
106
|
+
if (!rules.networkAllow.some((r) => hostMatches(r, host))) {
|
|
107
|
+
return { blocked: true, reason: `目标域名 ${host} 不在网络白名单` };
|
|
108
|
+
}
|
|
109
|
+
}
|
|
110
|
+
}
|
|
111
|
+
return { blocked: false };
|
|
112
|
+
}
|
|
113
|
+
/** 沙箱模式 → 人类可读描述(doctor / help 用) */
|
|
114
|
+
function describeSandboxMode(mode) {
|
|
115
|
+
switch (mode) {
|
|
116
|
+
case 'read-only':
|
|
117
|
+
return '只读勘察(禁写禁执行)';
|
|
118
|
+
case 'danger-full-access':
|
|
119
|
+
return '全权限(绕过写限制与审批,危险命令仍拦截)';
|
|
120
|
+
case 'container':
|
|
121
|
+
return '容器隔离(shell 在 Docker 容器内执行,镜像 FH_SANDBOX_IMAGE)';
|
|
122
|
+
default:
|
|
123
|
+
return '工作区可写(shell 受白名单+审批约束)';
|
|
124
|
+
}
|
|
125
|
+
}
|
|
126
|
+
//# sourceMappingURL=sandbox.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"sandbox.js","sourceRoot":"","sources":["../../src/tools/sandbox.ts"],"names":[],"mappings":";;AA+BA,oDAMC;AAMD,kDAYC;AAWD,0CAgBC;AAGD,oCAuDC;AAGD,kDAWC;AA5HD,sCAAsC;AACtC,SAAgB,oBAAoB,CAAC,GAAmB;IACtD,MAAM,CAAC,GAAG,GAAG,EAAE,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IACpC,IAAI,CAAC,KAAK,WAAW,IAAI,CAAC,KAAK,UAAU;QAAE,OAAO,WAAW,CAAC;IAC9D,IAAI,CAAC,KAAK,oBAAoB,IAAI,CAAC,KAAK,aAAa,IAAI,CAAC,KAAK,MAAM;QAAE,OAAO,oBAAoB,CAAC;IACnG,IAAI,CAAC,KAAK,WAAW,IAAI,CAAC,KAAK,QAAQ;QAAE,OAAO,WAAW,CAAC;IAC5D,OAAO,iBAAiB,CAAC;AAC3B,CAAC;AAED,2BAA2B;AAC3B,MAAM,WAAW,GAAG,IAAI,GAAG,CAAC,CAAC,YAAY,EAAE,WAAW,CAAC,CAAC,CAAC;AAEzD,6CAA6C;AAC7C,SAAgB,mBAAmB,CAAC,GAAW;IAC7C,MAAM,KAAK,GAAG,IAAI,GAAG,EAAU,CAAC;IAChC,MAAM,EAAE,GAAG,0BAA0B,CAAC;IACtC,IAAI,CAAyB,CAAC;IAC9B,OAAO,CAAC,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;QACnC,IAAI,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;QAC9B,iDAAiD;QACjD,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAChC,IAAI,KAAK,GAAG,CAAC;YAAE,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;QAC3C,IAAI,IAAI;YAAE,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IAC5B,CAAC;IACD,OAAO,CAAC,GAAG,KAAK,CAAC,CAAC;AACpB,CAAC;AAED,SAAS,WAAW,CAAC,IAAY,EAAE,IAAY;IAC7C,MAAM,CAAC,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IACpC,IAAI,CAAC,CAAC;QAAE,OAAO,KAAK,CAAC;IACrB,iDAAiD;IACjD,IAAI,CAAC,CAAC,UAAU,CAAC,GAAG,CAAC;QAAE,OAAO,IAAI,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;IACtE,OAAO,IAAI,KAAK,CAAC,CAAC;AACpB,CAAC;AAED,8DAA8D;AAC9D,SAAgB,eAAe,CAAC,KAAmB,EAAE,GAAW;IAC9D,IAAI,MAAM,GAAG,EAAE,CAAC;IAChB,IAAI,CAAC;QACH,MAAM,CAAC,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC,CAAC;QACvB,MAAM,GAAG,CAAC,CAAC,QAAQ,CAAC,WAAW,EAAE,CAAC;IACpC,CAAC;IAAC,MAAM,CAAC;QACP,0BAA0B;QAC1B,OAAO,IAAI,CAAC;IACd,CAAC;IACD,IAAI,KAAK,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,WAAW,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,EAAE,CAAC;QAC1D,OAAO,QAAQ,MAAM,UAAU,CAAC;IAClC,CAAC;IACD,IAAI,KAAK,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,WAAW,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,EAAE,CAAC;QAC7F,OAAO,QAAQ,MAAM,UAAU,CAAC;IAClC,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,0DAA0D;AAC1D,SAAgB,YAAY,CAC1B,IAAiB,EACjB,IAAY,EACZ,IAA6B,EAC7B,KAAmB;IAEnB,iDAAiD;IACjD,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,GAAG,IAAI,EAAE,CAAC,CAAC;IACnD,IAAI,IAAI,KAAK,WAAW,IAAI,GAAG,EAAE,CAAC;QAChC,MAAM,KAAK,GAAG,mBAAmB,CAAC,GAAG,CAAC,CAAC;QACvC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,KAAK,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,WAAW,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,EAAE,CAAC;gBACxD,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,QAAQ,IAAI,UAAU,EAAE,CAAC;YAC3D,CAAC;QACH,CAAC;IACH,CAAC;IAED,IAAI,IAAI,KAAK,oBAAoB;QAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;IAE7D,2BAA2B;IAC3B,IAAI,IAAI,KAAK,WAAW,EAAE,CAAC;QACzB,IAAI,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;YAC1B,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,uCAAuC,EAAE,CAAC;QAC5E,CAAC;QACD,IAAI,IAAI,KAAK,WAAW,EAAE,CAAC;YACzB,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,2BAA2B,EAAE,CAAC;QAChE,CAAC;QACD,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;IAC5B,CAAC;IAED,kDAAkD;IAClD,mCAAmC;IACnC,IAAI,IAAI,KAAK,WAAW,EAAE,CAAC;QACzB,IAAI,IAAI,KAAK,WAAW,IAAI,GAAG,IAAI,KAAK,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACjE,MAAM,KAAK,GAAG,mBAAmB,CAAC,GAAG,CAAC,CAAC;YACvC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,WAAW,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,EAAE,CAAC;oBAC1D,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,QAAQ,IAAI,UAAU,EAAE,CAAC;gBAC3D,CAAC;YACH,CAAC;QACH,CAAC;QACD,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;IAC5B,CAAC;IAED,uDAAuD;IACvD,IAAI,IAAI,KAAK,WAAW,IAAI,GAAG,IAAI,KAAK,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACjE,MAAM,KAAK,GAAG,mBAAmB,CAAC,GAAG,CAAC,CAAC;QACvC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,WAAW,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,EAAE,CAAC;gBAC1D,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,QAAQ,IAAI,UAAU,EAAE,CAAC;YAC3D,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;AAC5B,CAAC;AAED,qCAAqC;AACrC,SAAgB,mBAAmB,CAAC,IAAiB;IACnD,QAAQ,IAAI,EAAE,CAAC;QACb,KAAK,WAAW;YACd,OAAO,aAAa,CAAC;QACvB,KAAK,oBAAoB;YACvB,OAAO,uBAAuB,CAAC;QACjC,KAAK,WAAW;YACd,OAAO,gDAAgD,CAAC;QAC1D;YACE,OAAO,wBAAwB,CAAC;IACpC,CAAC;AACH,CAAC"}
|
package/dist/tools/shell/exec.js
CHANGED
|
@@ -1,5 +1,7 @@
|
|
|
1
1
|
"use strict";
|
|
2
2
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.containerImage = containerImage;
|
|
4
|
+
exports.runCommandInContainer = runCommandInContainer;
|
|
3
5
|
exports.runCommand = runCommand;
|
|
4
6
|
exports.sanitizeManagedCommand = sanitizeManagedCommand;
|
|
5
7
|
exports.commandHead = commandHead;
|
|
@@ -8,8 +10,31 @@ exports.commandHead = commandHead;
|
|
|
8
10
|
* 晋江市飞虹智科技企业管理有限公司 · 飞扬企源研发中心 · 负责人:吴赐虹
|
|
9
11
|
*
|
|
10
12
|
* 命令执行辅助(跨平台,spawn shell)
|
|
13
|
+
* P5-4:容器执行(docker run 挂载工作区,镜像 FH_SANDBOX_IMAGE)
|
|
11
14
|
*/
|
|
12
15
|
const child_process_1 = require("child_process");
|
|
16
|
+
/** 容器镜像(FH_SANDBOX_IMAGE 可覆盖;默认 node:22-alpine 轻量且覆盖常见工具) */
|
|
17
|
+
function containerImage() {
|
|
18
|
+
return process.env.FH_SANDBOX_IMAGE || 'node:22-alpine';
|
|
19
|
+
}
|
|
20
|
+
/** 容器模式下应把命令包装成 docker run 执行;镜像不存在时由 docker 自动拉取 */
|
|
21
|
+
function runCommandInContainer(cmd, cwd, timeoutMs = 120000) {
|
|
22
|
+
const image = containerImage();
|
|
23
|
+
// 挂载工作区到 /workspace,容器内 cwd=/workspace;--rm 用完即删
|
|
24
|
+
const dockerArgs = [
|
|
25
|
+
'run',
|
|
26
|
+
'--rm',
|
|
27
|
+
'-v',
|
|
28
|
+
`${cwd}:/workspace`,
|
|
29
|
+
'-w',
|
|
30
|
+
'/workspace',
|
|
31
|
+
image,
|
|
32
|
+
'sh',
|
|
33
|
+
'-c',
|
|
34
|
+
cmd,
|
|
35
|
+
];
|
|
36
|
+
return runCommand('docker ' + dockerArgs.map((a) => `"${a.replace(/"/g, '\\"')}"`).join(' '), cwd, timeoutMs);
|
|
37
|
+
}
|
|
13
38
|
function runCommand(cmd, cwd, timeoutMs = 120000) {
|
|
14
39
|
return new Promise((resolve) => {
|
|
15
40
|
const child = (0, child_process_1.spawn)(cmd, { cwd, shell: true, timeout: timeoutMs });
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"exec.js","sourceRoot":"","sources":["../../../src/tools/shell/exec.ts"],"names":[],"mappings":";;
|
|
1
|
+
{"version":3,"file":"exec.js","sourceRoot":"","sources":["../../../src/tools/shell/exec.ts"],"names":[],"mappings":";;AAgBA,wCAEC;AAGD,sDAgBC;AAED,gCAcC;AAUD,wDAMC;AAGD,kCAGC;AA3ED;;;;;;GAMG;AACH,iDAAsC;AAQtC,6DAA6D;AAC7D,SAAgB,cAAc;IAC5B,OAAO,OAAO,CAAC,GAAG,CAAC,gBAAgB,IAAI,gBAAgB,CAAC;AAC1D,CAAC;AAED,qDAAqD;AACrD,SAAgB,qBAAqB,CAAC,GAAW,EAAE,GAAW,EAAE,SAAS,GAAG,MAAM;IAChF,MAAM,KAAK,GAAG,cAAc,EAAE,CAAC;IAC/B,iDAAiD;IACjD,MAAM,UAAU,GAAG;QACjB,KAAK;QACL,MAAM;QACN,IAAI;QACJ,GAAG,GAAG,aAAa;QACnB,IAAI;QACJ,YAAY;QACZ,KAAK;QACL,IAAI;QACJ,IAAI;QACJ,GAAG;KACJ,CAAC;IACF,OAAO,UAAU,CAAC,SAAS,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,IAAI,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,SAAS,CAAC,CAAC;AAChH,CAAC;AAED,SAAgB,UAAU,CAAC,GAAW,EAAE,GAAW,EAAE,SAAS,GAAG,MAAM;IACrE,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QAC7B,MAAM,KAAK,GAAG,IAAA,qBAAK,EAAC,GAAG,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,IAAI,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC;QACnE,IAAI,MAAM,GAAG,EAAE,CAAC;QAChB,IAAI,MAAM,GAAG,EAAE,CAAC;QAChB,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,MAAM,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC;QAC1D,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,MAAM,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC;QAC1D,kEAAkE;QAClE,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC,EAAE,EAAE;YACtB,MAAM,IAAI,CAAC,CAAC,OAAO,CAAC;YACpB,OAAO,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;QACH,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,IAAI,IAAI,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;IAC5E,CAAC,CAAC,CAAC;AACL,CAAC;AAED;;;GAGG;AACH,MAAM,oBAAoB,GAAG,qDAAqD,CAAC;AACnF,MAAM,oBAAoB,GAAG,mEAAmE,CAAC;AACjG,MAAM,cAAc,GAAG,4BAA4B,CAAC;AAEpD,SAAgB,sBAAsB,CAAC,GAAY,EAAE,QAAQ,GAAG,UAAU;IACxE,MAAM,GAAG,GAAG,CAAC,GAAG,IAAI,QAAQ,CAAC,CAAC,IAAI,EAAE,CAAC;IACrC,IAAI,oBAAoB,CAAC,IAAI,CAAC,GAAG,CAAC;QAAE,OAAO,IAAI,CAAC;IAChD,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,GAAG,CAAC;QAAE,OAAO,IAAI,CAAC;IAC3C,IAAI,oBAAoB,CAAC,IAAI,CAAC,GAAG,CAAC;QAAE,OAAO,IAAI,CAAC;IAChD,OAAO,GAAG,CAAC;AACb,CAAC;AAED,qBAAqB;AACrB,SAAgB,WAAW,CAAC,GAAW;IACrC,MAAM,CAAC,GAAG,GAAG,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;IACrC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;AACvB,CAAC"}
|
|
@@ -38,7 +38,10 @@ exports.runShellTool = {
|
|
|
38
38
|
if (!approved)
|
|
39
39
|
return { ok: false, output: '', error: '已拒绝执行(需审批)' };
|
|
40
40
|
}
|
|
41
|
-
|
|
41
|
+
// P5-4:container 沙箱模式下命令在 Docker 容器内执行(挂载工作区)
|
|
42
|
+
const res = ctx.security.sandboxMode === 'container'
|
|
43
|
+
? await (0, exec_1.runCommandInContainer)(command, ctx.cwd)
|
|
44
|
+
: await (0, exec_1.runCommand)(command, ctx.cwd);
|
|
42
45
|
return {
|
|
43
46
|
ok: res.code === 0,
|
|
44
47
|
output: `${res.stdout}${res.stderr}`.slice(0, 4000),
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"run-shell.tool.js","sourceRoot":"","sources":["../../../src/tools/shell/run-shell.tool.ts"],"names":[],"mappings":";;;AAAA;;;;;GAKG;AACH,6BAAwB;AAExB,
|
|
1
|
+
{"version":3,"file":"run-shell.tool.js","sourceRoot":"","sources":["../../../src/tools/shell/run-shell.tool.ts"],"names":[],"mappings":";;;AAAA;;;;;GAKG;AACH,6BAAwB;AAExB,iCAAwE;AAExE,6BAA6B;AAC7B,MAAM,kBAAkB,GAAG,qDAAqD,CAAC;AAEpE,QAAA,YAAY,GAAS;IAChC,IAAI,EAAE,WAAW;IACjB,WAAW,EAAE,8BAA8B;IAC3C,UAAU,EAAE;QACV,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,UAAU,EAAE,EAAE;QACpE,QAAQ,EAAE,CAAC,SAAS,CAAC;KACtB;IACD,MAAM,EAAE,OAAC,CAAC,MAAM,CAAC,EAAE,OAAO,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;IAChD,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,GAAgB;QAClC,MAAM,EAAE,OAAO,EAAE,GAAG,IAA2B,CAAC;QAChD,MAAM,IAAI,GAAG,IAAA,kBAAW,EAAC,OAAO,CAAC,CAAC;QAClC,uBAAuB;QACvB,IAAI,GAAG,CAAC,QAAQ,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC3C,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,cAAc,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;gBAChD,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,YAAY,IAAI,EAAE,EAAE,CAAC;YAC9D,CAAC;QACH,CAAC;QACD,qCAAqC;QACrC,IAAI,kBAAkB,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC;YACrC,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,wBAAwB,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,EAAE,CAAC;QAC3F,CAAC;QACD,IAAI,GAAG,CAAC,QAAQ,CAAC,eAAe,EAAE,CAAC;YACjC,MAAM,QAAQ,GAAG,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,OAAO,CAAC,cAAc,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;YAClF,IAAI,CAAC,QAAQ;gBAAE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC;QACvE,CAAC;QACD,8CAA8C;QAC9C,MAAM,GAAG,GACP,GAAG,CAAC,QAAQ,CAAC,WAAW,KAAK,WAAW;YACtC,CAAC,CAAC,MAAM,IAAA,4BAAqB,EAAC,OAAO,EAAE,GAAG,CAAC,GAAG,CAAC;YAC/C,CAAC,CAAC,MAAM,IAAA,iBAAU,EAAC,OAAO,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC;QACzC,OAAO;YACL,EAAE,EAAE,GAAG,CAAC,IAAI,KAAK,CAAC;YAClB,MAAM,EAAE,GAAG,GAAG,CAAC,MAAM,GAAG,GAAG,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC;YACnD,KAAK,EAAE,GAAG,CAAC,IAAI,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,aAAa,GAAG,CAAC,IAAI,EAAE;SAC5D,CAAC;IACJ,CAAC;CACF,CAAC"}
|
|
@@ -0,0 +1,36 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.loadSkillTool = void 0;
|
|
4
|
+
/**
|
|
5
|
+
* 飞虹 Code (Muse Code 参照复刻)
|
|
6
|
+
* 晋江市飞虹智科技企业管理有限公司 · 飞扬企源研发中心 · 负责人:吴赐虹
|
|
7
|
+
*
|
|
8
|
+
* P1-2 工具:load_skill —— 按名加载 SKILL.md 完整指令(渐进式披露 Tier-2)。
|
|
9
|
+
* 模型从 system prompt 的技能索引中看到 name+description,
|
|
10
|
+
* 需要时调用本工具取回完整指令,避免把全部技能正文常驻上下文。
|
|
11
|
+
*/
|
|
12
|
+
const zod_1 = require("zod");
|
|
13
|
+
const skill_loader_1 = require("../../skills/skill-loader");
|
|
14
|
+
/** 打包技能目录(仓库内 skills/,随 dist 分发) */
|
|
15
|
+
const PACKAGED_SKILL_DIRS = [require('path').join(__dirname, '..', '..', '..', 'skills')];
|
|
16
|
+
exports.loadSkillTool = {
|
|
17
|
+
name: 'load_skill',
|
|
18
|
+
description: '加载指定技能的完整指令(技能名见 system prompt 中的技能索引)',
|
|
19
|
+
jsonSchema: {
|
|
20
|
+
type: 'object',
|
|
21
|
+
properties: { name: { type: 'string', description: '技能名,如 plan / grill / goal' } },
|
|
22
|
+
required: ['name'],
|
|
23
|
+
},
|
|
24
|
+
schema: zod_1.z.object({ name: zod_1.z.string().min(1) }),
|
|
25
|
+
async execute(args, ctx) {
|
|
26
|
+
const { name } = args;
|
|
27
|
+
const skills = (0, skill_loader_1.discoverSkills)(ctx.cwd, PACKAGED_SKILL_DIRS);
|
|
28
|
+
const body = (0, skill_loader_1.loadSkillBody)(skills, name);
|
|
29
|
+
if (body === null) {
|
|
30
|
+
const available = skills.map((s) => s.name).join(', ') || '(无)';
|
|
31
|
+
return { ok: false, output: '', error: `未找到技能 ${name};可用技能: ${available}` };
|
|
32
|
+
}
|
|
33
|
+
return { ok: true, output: body.slice(0, 12000) };
|
|
34
|
+
},
|
|
35
|
+
};
|
|
36
|
+
//# sourceMappingURL=load-skill.tool.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"load-skill.tool.js","sourceRoot":"","sources":["../../../src/tools/skills/load-skill.tool.ts"],"names":[],"mappings":";;;AAAA;;;;;;;GAOG;AACH,6BAAwB;AAExB,4DAAsF;AAEtF,oCAAoC;AACpC,MAAM,mBAAmB,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,CAAC,CAAC,CAAC;AAE7E,QAAA,aAAa,GAAS;IACjC,IAAI,EAAE,YAAY;IAClB,WAAW,EAAE,wCAAwC;IACrD,UAAU,EAAE;QACV,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,2BAA2B,EAAE,EAAE;QAClF,QAAQ,EAAE,CAAC,MAAM,CAAC;KACnB;IACD,MAAM,EAAE,OAAC,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;IAC7C,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,GAAgB;QAClC,MAAM,EAAE,IAAI,EAAE,GAAG,IAAwB,CAAC;QAC1C,MAAM,MAAM,GAAY,IAAA,6BAAc,EAAC,GAAG,CAAC,GAAG,EAAE,mBAAmB,CAAC,CAAC;QACrE,MAAM,IAAI,GAAG,IAAA,4BAAa,EAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QACzC,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;YAClB,MAAM,SAAS,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,KAAK,CAAC;YAChE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,SAAS,IAAI,UAAU,SAAS,EAAE,EAAE,CAAC;QAC9E,CAAC;QACD,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,EAAE,CAAC;IACpD,CAAC;CACF,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"tool.interface.js","sourceRoot":"","sources":["../../src/tools/tool.interface.ts"],"names":[],"mappings":";;
|
|
1
|
+
{"version":3,"file":"tool.interface.js","sourceRoot":"","sources":["../../src/tools/tool.interface.ts"],"names":[],"mappings":";;AAkEA,oCAEC;AAHD,qBAAqB;AACrB,SAAgB,YAAY,CAAC,IAAU;IACrC,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,CAAC;AACzF,CAAC"}
|
|
@@ -9,6 +9,8 @@ exports.ToolRegistry = void 0;
|
|
|
9
9
|
*/
|
|
10
10
|
const logger_1 = require("../shared/logger");
|
|
11
11
|
const tool_interface_1 = require("./tool.interface");
|
|
12
|
+
const sandbox_1 = require("./sandbox");
|
|
13
|
+
const hooks_1 = require("../runtime/hooks");
|
|
12
14
|
class ToolRegistry {
|
|
13
15
|
tools = new Map();
|
|
14
16
|
register(tool) {
|
|
@@ -41,6 +43,30 @@ class ToolRegistry {
|
|
|
41
43
|
return { ok: false, output: '', error: `参数校验失败: ${parsed.error.message}` };
|
|
42
44
|
}
|
|
43
45
|
const args = parsed.data;
|
|
46
|
+
// P0-2:沙箱技术边界(先于 RBAC 守卫执行 —— 沙箱是"能否做"的硬边界,
|
|
47
|
+
// 审批/策略都无权放行被沙箱拦截的动作)
|
|
48
|
+
if (ctx.security.sandboxMode) {
|
|
49
|
+
const rules = ctx.security.networkRules ?? { networkAllow: [], networkDeny: [] };
|
|
50
|
+
const decision = (0, sandbox_1.checkSandbox)(ctx.security.sandboxMode, name, args, rules);
|
|
51
|
+
if (decision.blocked) {
|
|
52
|
+
logger_1.logger.warn('sandbox blocked tool call', { tool: name, reason: decision.reason });
|
|
53
|
+
return { ok: false, output: '', error: `[沙箱拦截] ${decision.reason}` };
|
|
54
|
+
}
|
|
55
|
+
}
|
|
56
|
+
// P2-1:PreToolUse hook(确定性拦截,命令退出码非 0 → 阻止执行)
|
|
57
|
+
const hookCtx = {
|
|
58
|
+
cwd: ctx.cwd,
|
|
59
|
+
runId: ctx.runId,
|
|
60
|
+
tool: name,
|
|
61
|
+
path: typeof args.path === 'string' ? args.path : typeof args.file === 'string' ? args.file : undefined,
|
|
62
|
+
};
|
|
63
|
+
if (ctx.security.hooks?.length) {
|
|
64
|
+
const pre = await (0, hooks_1.runHooks)(ctx.security.hooks, 'PreToolUse', hookCtx);
|
|
65
|
+
if (pre.blocked) {
|
|
66
|
+
logger_1.logger.warn('hook denied tool call', { tool: name, reason: pre.reason });
|
|
67
|
+
return { ok: false, output: '', error: `[hook 拦截] ${pre.reason}` };
|
|
68
|
+
}
|
|
69
|
+
}
|
|
44
70
|
// M4:企业守卫前置(策略 → 审批 → 审计)。未注入 guard 时行为不变。
|
|
45
71
|
if (ctx.guard) {
|
|
46
72
|
let verdict;
|
|
@@ -61,14 +87,24 @@ class ToolRegistry {
|
|
|
61
87
|
// 此处关闭工具内的二次审批与白名单硬拦截,避免重复弹审批 / 决策打架。
|
|
62
88
|
ctx = { ...ctx, security: { shellAllowlist: [], requireApproval: false } };
|
|
63
89
|
}
|
|
90
|
+
let result;
|
|
64
91
|
try {
|
|
65
|
-
|
|
92
|
+
result = await tool.execute(args, ctx);
|
|
66
93
|
}
|
|
67
94
|
catch (e) {
|
|
68
95
|
const msg = e instanceof Error ? e.message : String(e);
|
|
69
96
|
logger_1.logger.error('tool execution failed', { tool: name, error: msg });
|
|
70
97
|
return { ok: false, output: '', error: msg };
|
|
71
98
|
}
|
|
99
|
+
// P2-1:PostToolUse / PostEdit hook(记录结果;编辑成功时对文件触发)
|
|
100
|
+
if (ctx.security.hooks?.length) {
|
|
101
|
+
const postCtx = { ...hookCtx, ok: result.ok };
|
|
102
|
+
await (0, hooks_1.runHooks)(ctx.security.hooks, 'PostToolUse', postCtx);
|
|
103
|
+
if (result.ok && hookCtx.path) {
|
|
104
|
+
await (0, hooks_1.runHooks)(ctx.security.hooks, 'PostEdit', postCtx);
|
|
105
|
+
}
|
|
106
|
+
}
|
|
107
|
+
return result;
|
|
72
108
|
}
|
|
73
109
|
}
|
|
74
110
|
exports.ToolRegistry = ToolRegistry;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"tool.registry.js","sourceRoot":"","sources":["../../src/tools/tool.registry.ts"],"names":[],"mappings":";;;AAAA;;;;;GAKG;AACH,6CAA0C;AAE1C,qDAAgD;
|
|
1
|
+
{"version":3,"file":"tool.registry.js","sourceRoot":"","sources":["../../src/tools/tool.registry.ts"],"names":[],"mappings":";;;AAAA;;;;;GAKG;AACH,6CAA0C;AAE1C,qDAAgD;AAChD,uCAA8E;AAC9E,4CAA4C;AAG5C,MAAa,YAAY;IACN,KAAK,GAAG,IAAI,GAAG,EAAgB,CAAC;IAEjD,QAAQ,CAAC,IAAU;QACjB,IAAI,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;YAC9B,eAAM,CAAC,IAAI,CAAC,kBAAkB,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;QACvD,CAAC;QACD,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;IAClC,CAAC;IAED,GAAG,CAAC,IAAY;QACd,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IAC9B,CAAC;IAED,GAAG,CAAC,IAAY;QACd,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IAC9B,CAAC;IAED,IAAI;QACF,OAAO,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;IAClC,CAAC;IAED,kBAAkB;IAClB,WAAW;QACT,OAAO,IAAI,CAAC,IAAI,EAAE,CAAC,GAAG,CAAC,6BAAY,CAAC,CAAC;IACvC,CAAC;IAED,yCAAyC;IACzC,KAAK,CAAC,OAAO,CAAC,IAAY,EAAE,OAAgB,EAAE,GAAgB;QAC5D,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAClC,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,SAAS,IAAI,EAAE,EAAE,CAAC;QAC3D,CAAC;QACD,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QAC9C,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACpB,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,WAAW,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,EAAE,CAAC;QAC7E,CAAC;QACD,MAAM,IAAI,GAAG,MAAM,CAAC,IAA+B,CAAC;QAEpD,4CAA4C;QAC5C,sBAAsB;QACtB,IAAI,GAAG,CAAC,QAAQ,CAAC,WAAW,EAAE,CAAC;YAC7B,MAAM,KAAK,GAAiB,GAAG,CAAC,QAAQ,CAAC,YAAY,IAAI,EAAE,YAAY,EAAE,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC;YAC/F,MAAM,QAAQ,GAAG,IAAA,sBAAY,EAAC,GAAG,CAAC,QAAQ,CAAC,WAA0B,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC;YAC1F,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;gBACrB,eAAM,CAAC,IAAI,CAAC,2BAA2B,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;gBAClF,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,UAAU,QAAQ,CAAC,MAAM,EAAE,EAAE,CAAC;YACvE,CAAC;QACH,CAAC;QAED,8CAA8C;QAC9C,MAAM,OAAO,GAAG;YACd,GAAG,EAAE,GAAG,CAAC,GAAG;YACZ,KAAK,EAAE,GAAG,CAAC,KAAK;YAChB,IAAI,EAAE,IAAI;YACV,IAAI,EAAE,OAAO,IAAI,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,IAAI,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS;SACxG,CAAC;QACF,IAAI,GAAG,CAAC,QAAQ,CAAC,KAAK,EAAE,MAAM,EAAE,CAAC;YAC/B,MAAM,GAAG,GAAG,MAAM,IAAA,gBAAQ,EAAC,GAAG,CAAC,QAAQ,CAAC,KAAK,EAAE,YAAY,EAAE,OAAO,CAAC,CAAC;YACtE,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;gBAChB,eAAM,CAAC,IAAI,CAAC,uBAAuB,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC;gBACzE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,aAAa,GAAG,CAAC,MAAM,EAAE,EAAE,CAAC;YACrE,CAAC;QACH,CAAC;QAED,2CAA2C;QAC3C,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;YACd,IAAI,OAAO,CAAC;YACZ,IAAI,CAAC;gBACH,OAAO,GAAG,MAAM,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YAC9C,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,6BAA6B;gBAC7B,MAAM,GAAG,GAAG,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;gBACvD,eAAM,CAAC,KAAK,CAAC,gBAAgB,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;gBAC3D,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,iBAAiB,GAAG,EAAE,EAAE,CAAC;YAClE,CAAC;YACD,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;gBACrB,eAAM,CAAC,IAAI,CAAC,WAAW,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;gBACjE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC;YAC1D,CAAC;YACD,gCAAgC;YAChC,sCAAsC;YACtC,GAAG,GAAG,EAAE,GAAG,GAAG,EAAE,QAAQ,EAAE,EAAE,cAAc,EAAE,EAAE,EAAE,eAAe,EAAE,KAAK,EAAE,EAAE,CAAC;QAC7E,CAAC;QAED,IAAI,MAAkB,CAAC;QACvB,IAAI,CAAC;YACH,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;QACzC,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,MAAM,GAAG,GAAG,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;YACvD,eAAM,CAAC,KAAK,CAAC,uBAAuB,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;YAClE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC;QAC/C,CAAC;QAED,oDAAoD;QACpD,IAAI,GAAG,CAAC,QAAQ,CAAC,KAAK,EAAE,MAAM,EAAE,CAAC;YAC/B,MAAM,OAAO,GAAG,EAAE,GAAG,OAAO,EAAE,EAAE,EAAE,MAAM,CAAC,EAAE,EAAE,CAAC;YAC9C,MAAM,IAAA,gBAAQ,EAAC,GAAG,CAAC,QAAQ,CAAC,KAAK,EAAE,aAAa,EAAE,OAAO,CAAC,CAAC;YAC3D,IAAI,MAAM,CAAC,EAAE,IAAI,OAAO,CAAC,IAAI,EAAE,CAAC;gBAC9B,MAAM,IAAA,gBAAQ,EAAC,GAAG,CAAC,QAAQ,CAAC,KAAK,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;YAC1D,CAAC;QACH,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;CACF;AAzGD,oCAyGC"}
|