@onesclawkolor/onesclaw 0.1.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/README.md +65 -0
- package/dist/agent/conversation.d.ts +34 -0
- package/dist/agent/conversation.js +61 -0
- package/dist/agent/conversation.js.map +1 -0
- package/dist/agent/loop.d.ts +40 -0
- package/dist/agent/loop.js +213 -0
- package/dist/agent/loop.js.map +1 -0
- package/dist/agent/session.d.ts +73 -0
- package/dist/agent/session.js +225 -0
- package/dist/agent/session.js.map +1 -0
- package/dist/audit/file-sink.d.ts +6 -0
- package/dist/audit/file-sink.js +21 -0
- package/dist/audit/file-sink.js.map +1 -0
- package/dist/audit/ids.d.ts +3 -0
- package/dist/audit/ids.js +15 -0
- package/dist/audit/ids.js.map +1 -0
- package/dist/audit/types.d.ts +13 -0
- package/dist/audit/types.js +2 -0
- package/dist/audit/types.js.map +1 -0
- package/dist/cli/commands.d.ts +31 -0
- package/dist/cli/commands.js +433 -0
- package/dist/cli/commands.js.map +1 -0
- package/dist/cli/logger.d.ts +57 -0
- package/dist/cli/logger.js +79 -0
- package/dist/cli/logger.js.map +1 -0
- package/dist/cli/output.d.ts +59 -0
- package/dist/cli/output.js +306 -0
- package/dist/cli/output.js.map +1 -0
- package/dist/cli/prompt.d.ts +29 -0
- package/dist/cli/prompt.js +197 -0
- package/dist/cli/prompt.js.map +1 -0
- package/dist/cli/repl.d.ts +4 -0
- package/dist/cli/repl.js +486 -0
- package/dist/cli/repl.js.map +1 -0
- package/dist/commands/init.d.ts +4 -0
- package/dist/commands/init.js +84 -0
- package/dist/commands/init.js.map +1 -0
- package/dist/commands/model.d.ts +3 -0
- package/dist/commands/model.js +178 -0
- package/dist/commands/model.js.map +1 -0
- package/dist/commands/plugin-init.d.ts +23 -0
- package/dist/commands/plugin-init.js +303 -0
- package/dist/commands/plugin-init.js.map +1 -0
- package/dist/commands/provider-auth.d.ts +14 -0
- package/dist/commands/provider-auth.js +153 -0
- package/dist/commands/provider-auth.js.map +1 -0
- package/dist/config/manager.d.ts +71 -0
- package/dist/config/manager.js +201 -0
- package/dist/config/manager.js.map +1 -0
- package/dist/config/types.d.ts +43 -0
- package/dist/config/types.js +2 -0
- package/dist/config/types.js.map +1 -0
- package/dist/errors.d.ts +37 -0
- package/dist/errors.js +55 -0
- package/dist/errors.js.map +1 -0
- package/dist/index.d.ts +2 -0
- package/dist/index.js +207 -0
- package/dist/index.js.map +1 -0
- package/dist/llm/anthropic.d.ts +83 -0
- package/dist/llm/anthropic.js +190 -0
- package/dist/llm/anthropic.js.map +1 -0
- package/dist/llm/factory.d.ts +105 -0
- package/dist/llm/factory.js +164 -0
- package/dist/llm/factory.js.map +1 -0
- package/dist/llm/openai-codex.d.ts +102 -0
- package/dist/llm/openai-codex.js +445 -0
- package/dist/llm/openai-codex.js.map +1 -0
- package/dist/llm/openai.d.ts +71 -0
- package/dist/llm/openai.js +289 -0
- package/dist/llm/openai.js.map +1 -0
- package/dist/llm/provider.d.ts +1 -0
- package/dist/llm/provider.js +2 -0
- package/dist/llm/provider.js.map +1 -0
- package/dist/llm/types.d.ts +72 -0
- package/dist/llm/types.js +2 -0
- package/dist/llm/types.js.map +1 -0
- package/dist/mcp/bridge.d.ts +17 -0
- package/dist/mcp/bridge.js +74 -0
- package/dist/mcp/bridge.js.map +1 -0
- package/dist/mcp/config.d.ts +10 -0
- package/dist/mcp/config.js +27 -0
- package/dist/mcp/config.js.map +1 -0
- package/dist/mcp/manager.d.ts +67 -0
- package/dist/mcp/manager.js +207 -0
- package/dist/mcp/manager.js.map +1 -0
- package/dist/memory/scope.d.ts +4 -0
- package/dist/memory/scope.js +14 -0
- package/dist/memory/scope.js.map +1 -0
- package/dist/memory/store.d.ts +28 -0
- package/dist/memory/store.js +126 -0
- package/dist/memory/store.js.map +1 -0
- package/dist/prompts/system.d.ts +4 -0
- package/dist/prompts/system.js +197 -0
- package/dist/prompts/system.js.map +1 -0
- package/dist/skills/loader.d.ts +25 -0
- package/dist/skills/loader.js +98 -0
- package/dist/skills/loader.js.map +1 -0
- package/dist/skills/registry.d.ts +57 -0
- package/dist/skills/registry.js +132 -0
- package/dist/skills/registry.js.map +1 -0
- package/dist/skills/sync.d.ts +33 -0
- package/dist/skills/sync.js +180 -0
- package/dist/skills/sync.js.map +1 -0
- package/dist/skills/types.d.ts +1 -0
- package/dist/skills/types.js +2 -0
- package/dist/skills/types.js.map +1 -0
- package/dist/tools/ask.d.ts +30 -0
- package/dist/tools/ask.js +60 -0
- package/dist/tools/ask.js.map +1 -0
- package/dist/tools/confirmation.d.ts +12 -0
- package/dist/tools/confirmation.js +75 -0
- package/dist/tools/confirmation.js.map +1 -0
- package/dist/tools/plugin-init.d.ts +29 -0
- package/dist/tools/plugin-init.js +47 -0
- package/dist/tools/plugin-init.js.map +1 -0
- package/dist/tools/registry.d.ts +56 -0
- package/dist/tools/registry.js +123 -0
- package/dist/tools/registry.js.map +1 -0
- package/dist/tools/safety.d.ts +2 -0
- package/dist/tools/safety.js +30 -0
- package/dist/tools/safety.js.map +1 -0
- package/dist/tools/server.d.ts +91 -0
- package/dist/tools/server.js +149 -0
- package/dist/tools/server.js.map +1 -0
- package/dist/tools/shell.d.ts +11 -0
- package/dist/tools/shell.js +174 -0
- package/dist/tools/shell.js.map +1 -0
- package/dist/tools/skills-sync.d.ts +19 -0
- package/dist/tools/skills-sync.js +29 -0
- package/dist/tools/skills-sync.js.map +1 -0
- package/dist/tools/ssh.d.ts +62 -0
- package/dist/tools/ssh.js +343 -0
- package/dist/tools/ssh.js.map +1 -0
- package/dist/tools/types.d.ts +10 -0
- package/dist/tools/types.js +2 -0
- package/dist/tools/types.js.map +1 -0
- package/dist/utils/retry.d.ts +7 -0
- package/dist/utils/retry.js +34 -0
- package/dist/utils/retry.js.map +1 -0
- package/install.sh +171 -0
- package/package.json +49 -0
|
@@ -0,0 +1,75 @@
|
|
|
1
|
+
const PROOF_TTL_MS = 120_000;
|
|
2
|
+
const AFFIRMATIVE_PATTERNS = [
|
|
3
|
+
/\b(yes|y|ok|okay|sure|confirm|confirmed|approve|approved|proceed|do it)\b/i,
|
|
4
|
+
/(确认|同意|可以|继续|执行|好的|是的|确认执行)/,
|
|
5
|
+
];
|
|
6
|
+
const NEGATIVE_PATTERNS = [
|
|
7
|
+
/\b(no|n|cancel|stop|deny|reject|abort|skip)\b/i,
|
|
8
|
+
/(取消|不要|停止|拒绝|不同意|算了|否)/,
|
|
9
|
+
];
|
|
10
|
+
let activeProof = null;
|
|
11
|
+
function classifyAnswer(answer) {
|
|
12
|
+
const normalized = answer.trim();
|
|
13
|
+
if (!normalized) {
|
|
14
|
+
return 'empty';
|
|
15
|
+
}
|
|
16
|
+
if (AFFIRMATIVE_PATTERNS.some((p) => p.test(normalized))) {
|
|
17
|
+
return 'affirmative';
|
|
18
|
+
}
|
|
19
|
+
if (NEGATIVE_PATTERNS.some((p) => p.test(normalized))) {
|
|
20
|
+
return 'negative';
|
|
21
|
+
}
|
|
22
|
+
return 'ambiguous';
|
|
23
|
+
}
|
|
24
|
+
export function issueFromAskUser(question, answer, now = Date.now()) {
|
|
25
|
+
const answerClass = classifyAnswer(answer);
|
|
26
|
+
if (answerClass === 'affirmative') {
|
|
27
|
+
activeProof = {
|
|
28
|
+
issuedAt: now,
|
|
29
|
+
consumed: false,
|
|
30
|
+
question: question.trim(),
|
|
31
|
+
answer: answer.trim(),
|
|
32
|
+
};
|
|
33
|
+
return { issued: true, status: 'issued' };
|
|
34
|
+
}
|
|
35
|
+
// 负向/模糊回答都会清空旧 proof,避免误复用。
|
|
36
|
+
activeProof = null;
|
|
37
|
+
if (answerClass === 'negative') {
|
|
38
|
+
return { issued: false, status: 'rejected' };
|
|
39
|
+
}
|
|
40
|
+
if (answerClass === 'empty') {
|
|
41
|
+
return { issued: false, status: 'empty' };
|
|
42
|
+
}
|
|
43
|
+
return { issued: false, status: 'ambiguous' };
|
|
44
|
+
}
|
|
45
|
+
export function consumeForHighRisk(_command, now = Date.now()) {
|
|
46
|
+
if (!activeProof) {
|
|
47
|
+
return { ok: false, status: 'missing' };
|
|
48
|
+
}
|
|
49
|
+
if (activeProof.consumed) {
|
|
50
|
+
return { ok: false, status: 'already_used' };
|
|
51
|
+
}
|
|
52
|
+
if (now - activeProof.issuedAt > PROOF_TTL_MS) {
|
|
53
|
+
activeProof = null;
|
|
54
|
+
return { ok: false, status: 'expired' };
|
|
55
|
+
}
|
|
56
|
+
activeProof.consumed = true;
|
|
57
|
+
return { ok: true, status: 'valid' };
|
|
58
|
+
}
|
|
59
|
+
export function requireConfirmationProofError(command, status) {
|
|
60
|
+
const reason = status === 'expired'
|
|
61
|
+
? '确认已过期(超过 120 秒)'
|
|
62
|
+
: status === 'already_used'
|
|
63
|
+
? '确认已被消费,不可复用'
|
|
64
|
+
: '未检测到有效确认';
|
|
65
|
+
return [
|
|
66
|
+
'检测到高危命令,但缺少有效确认凭证,已被安全策略拦截。',
|
|
67
|
+
`原因: ${reason}`,
|
|
68
|
+
'请先调用 ask_user 工具获得明确同意,然后立即重试并传入 confirmed=true。',
|
|
69
|
+
`命令: ${command}`,
|
|
70
|
+
].join('\n');
|
|
71
|
+
}
|
|
72
|
+
export function resetForTest() {
|
|
73
|
+
activeProof = null;
|
|
74
|
+
}
|
|
75
|
+
//# sourceMappingURL=confirmation.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"confirmation.js","sourceRoot":"","sources":["../../src/tools/confirmation.ts"],"names":[],"mappings":"AAAA,MAAM,YAAY,GAAG,OAAO,CAAC;AAa7B,MAAM,oBAAoB,GAAa;IACrC,4EAA4E;IAC5E,6BAA6B;CAC9B,CAAC;AAEF,MAAM,iBAAiB,GAAa;IAClC,gDAAgD;IAChD,wBAAwB;CACzB,CAAC;AAEF,IAAI,WAAW,GAA6B,IAAI,CAAC;AAEjD,SAAS,cAAc,CAAC,MAAc;IACpC,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC;IACjC,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,IAAI,oBAAoB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,EAAE,CAAC;QACzD,OAAO,aAAa,CAAC;IACvB,CAAC;IAED,IAAI,iBAAiB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,EAAE,CAAC;QACtD,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,OAAO,WAAW,CAAC;AACrB,CAAC;AAED,MAAM,UAAU,gBAAgB,CAC9B,QAAgB,EAChB,MAAc,EACd,MAAc,IAAI,CAAC,GAAG,EAAE;IAExB,MAAM,WAAW,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC;IAE3C,IAAI,WAAW,KAAK,aAAa,EAAE,CAAC;QAClC,WAAW,GAAG;YACZ,QAAQ,EAAE,GAAG;YACb,QAAQ,EAAE,KAAK;YACf,QAAQ,EAAE,QAAQ,CAAC,IAAI,EAAE;YACzB,MAAM,EAAE,MAAM,CAAC,IAAI,EAAE;SACtB,CAAC;QACF,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC;IAC5C,CAAC;IAED,4BAA4B;IAC5B,WAAW,GAAG,IAAI,CAAC;IAEnB,IAAI,WAAW,KAAK,UAAU,EAAE,CAAC;QAC/B,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC;IAC/C,CAAC;IACD,IAAI,WAAW,KAAK,OAAO,EAAE,CAAC;QAC5B,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC;IAC5C,CAAC;IACD,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC;AAChD,CAAC;AAED,MAAM,UAAU,kBAAkB,CAChC,QAAgB,EAChB,MAAc,IAAI,CAAC,GAAG,EAAE;IAExB,IAAI,CAAC,WAAW,EAAE,CAAC;QACjB,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC;IAC1C,CAAC;IAED,IAAI,WAAW,CAAC,QAAQ,EAAE,CAAC;QACzB,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,cAAc,EAAE,CAAC;IAC/C,CAAC;IAED,IAAI,GAAG,GAAG,WAAW,CAAC,QAAQ,GAAG,YAAY,EAAE,CAAC;QAC9C,WAAW,GAAG,IAAI,CAAC;QACnB,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC;IAC1C,CAAC;IAED,WAAW,CAAC,QAAQ,GAAG,IAAI,CAAC;IAC5B,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC;AACvC,CAAC;AAED,MAAM,UAAU,6BAA6B,CAAC,OAAe,EAAE,MAA0B;IACvF,MAAM,MAAM,GACV,MAAM,KAAK,SAAS;QAClB,CAAC,CAAC,iBAAiB;QACnB,CAAC,CAAC,MAAM,KAAK,cAAc;YACzB,CAAC,CAAC,aAAa;YACf,CAAC,CAAC,UAAU,CAAC;IAEnB,OAAO;QACL,6BAA6B;QAC7B,OAAO,MAAM,EAAE;QACf,kDAAkD;QAClD,OAAO,OAAO,EAAE;KACjB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACf,CAAC;AAED,MAAM,UAAU,YAAY;IAC1B,WAAW,GAAG,IAAI,CAAC;AACrB,CAAC"}
|
|
@@ -0,0 +1,29 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* 创建 plugin_init 工具
|
|
3
|
+
* 使 LLM 能够在理解用户意图后直接初始化 ONES 插件项目
|
|
4
|
+
*/
|
|
5
|
+
export declare function createPluginInitTool(): {
|
|
6
|
+
name: string;
|
|
7
|
+
description: string;
|
|
8
|
+
inputSchema: {
|
|
9
|
+
type: string;
|
|
10
|
+
properties: {
|
|
11
|
+
pluginName: {
|
|
12
|
+
type: string;
|
|
13
|
+
description: string;
|
|
14
|
+
};
|
|
15
|
+
pluginDir: {
|
|
16
|
+
type: string;
|
|
17
|
+
description: string;
|
|
18
|
+
};
|
|
19
|
+
};
|
|
20
|
+
required: string[];
|
|
21
|
+
};
|
|
22
|
+
execute: (args: any) => Promise<{
|
|
23
|
+
content: string;
|
|
24
|
+
isError: boolean;
|
|
25
|
+
} | {
|
|
26
|
+
content: string;
|
|
27
|
+
isError?: undefined;
|
|
28
|
+
}>;
|
|
29
|
+
};
|
|
@@ -0,0 +1,47 @@
|
|
|
1
|
+
// @ts-nocheck
|
|
2
|
+
import { pluginInitCore } from '../commands/plugin-init.js';
|
|
3
|
+
/**
|
|
4
|
+
* 创建 plugin_init 工具
|
|
5
|
+
* 使 LLM 能够在理解用户意图后直接初始化 ONES 插件项目
|
|
6
|
+
*/
|
|
7
|
+
export function createPluginInitTool() {
|
|
8
|
+
return {
|
|
9
|
+
name: 'plugin_init',
|
|
10
|
+
description: '初始化一个 ONES 插件项目。包含环境预检(Node.js 18、npm、ONES CLI、npx)、项目创建和配置。调用前请先通过对话确认插件名称。',
|
|
11
|
+
inputSchema: {
|
|
12
|
+
type: 'object',
|
|
13
|
+
properties: {
|
|
14
|
+
pluginName: {
|
|
15
|
+
type: 'string',
|
|
16
|
+
description: '插件名称,仅允许字母、数字、连字符(-)和下划线(_)',
|
|
17
|
+
},
|
|
18
|
+
pluginDir: {
|
|
19
|
+
type: 'string',
|
|
20
|
+
description: '项目目录路径,留空则在当前目录创建',
|
|
21
|
+
},
|
|
22
|
+
},
|
|
23
|
+
required: ['pluginName'],
|
|
24
|
+
},
|
|
25
|
+
execute: async (args) => {
|
|
26
|
+
const pluginName = args.pluginName;
|
|
27
|
+
const pluginDir = args.pluginDir;
|
|
28
|
+
const result = await pluginInitCore({ pluginName, pluginDir });
|
|
29
|
+
if (!result.success) {
|
|
30
|
+
return { content: result.message, isError: true };
|
|
31
|
+
}
|
|
32
|
+
const lines = [result.message];
|
|
33
|
+
lines.push(`项目路径: ${result.projectPath}`);
|
|
34
|
+
if (result.nextSteps) {
|
|
35
|
+
lines.push('');
|
|
36
|
+
lines.push('后续调试步骤:');
|
|
37
|
+
result.nextSteps.forEach((step, i) => {
|
|
38
|
+
lines.push(`${i + 1}. ${step}`);
|
|
39
|
+
});
|
|
40
|
+
}
|
|
41
|
+
lines.push('');
|
|
42
|
+
lines.push('开发文档: https://developer.ones.cn/zh-CN/docs/guide/getting-started/development');
|
|
43
|
+
return { content: lines.join('\n') };
|
|
44
|
+
},
|
|
45
|
+
};
|
|
46
|
+
}
|
|
47
|
+
//# sourceMappingURL=plugin-init.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"plugin-init.js","sourceRoot":"","sources":["../../src/tools/plugin-init.ts"],"names":[],"mappings":"AAAA,cAAc;AACd,OAAO,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AAC5D;;;GAGG;AACH,MAAM,UAAU,oBAAoB;IAChC,OAAO;QACH,IAAI,EAAE,aAAa;QACnB,WAAW,EAAE,8EAA8E;QAC3F,WAAW,EAAE;YACT,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACR,UAAU,EAAE;oBACR,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,6BAA6B;iBAC7C;gBACD,SAAS,EAAE;oBACP,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,mBAAmB;iBACnC;aACJ;YACD,QAAQ,EAAE,CAAC,YAAY,CAAC;SAC3B;QACD,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YACpB,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC;YACnC,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC;YACjC,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC,CAAC;YAC/D,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBAClB,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;YACtD,CAAC;YACD,MAAM,KAAK,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YAC/B,KAAK,CAAC,IAAI,CAAC,SAAS,MAAM,CAAC,WAAW,EAAE,CAAC,CAAC;YAC1C,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;gBACnB,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBACf,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBACtB,MAAM,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE;oBACjC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;gBACpC,CAAC,CAAC,CAAC;YACP,CAAC;YACD,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACf,KAAK,CAAC,IAAI,CAAC,8EAA8E,CAAC,CAAC;YAC3F,OAAO,EAAE,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QACzC,CAAC;KACJ,CAAC;AACN,CAAC"}
|
|
@@ -0,0 +1,56 @@
|
|
|
1
|
+
import { SSHConnectionPool } from './ssh.js';
|
|
2
|
+
import { SkillRegistry } from '../skills/registry.js';
|
|
3
|
+
/**
|
|
4
|
+
* 统一工具注册中心
|
|
5
|
+
* 管理所有可用工具(MCP 工具 + 未来的原生工具)
|
|
6
|
+
*/
|
|
7
|
+
export declare class ToolRegistry {
|
|
8
|
+
tools: Map<any, any>;
|
|
9
|
+
sshPool: SSHConnectionPool;
|
|
10
|
+
skillRegistry: SkillRegistry;
|
|
11
|
+
/**
|
|
12
|
+
* 注册单个工具
|
|
13
|
+
*/
|
|
14
|
+
register(tool: any): void;
|
|
15
|
+
/**
|
|
16
|
+
* 注册 Native Tools(本地工具)
|
|
17
|
+
*/
|
|
18
|
+
registerNativeTools(): void;
|
|
19
|
+
/**
|
|
20
|
+
* 加载 Skill 索引
|
|
21
|
+
*/
|
|
22
|
+
loadSkills(indexPath: any): Promise<void>;
|
|
23
|
+
/**
|
|
24
|
+
* 获取已加载的技能列表(供 System Prompt 构建使用)
|
|
25
|
+
*/
|
|
26
|
+
getSkills(): any[];
|
|
27
|
+
/**
|
|
28
|
+
* 清理资源(关闭 SSH 连接等)
|
|
29
|
+
*/
|
|
30
|
+
cleanup(): Promise<void>;
|
|
31
|
+
/**
|
|
32
|
+
* 批量注册 MCP 工具
|
|
33
|
+
* 从 MCPManager 获取所有工具,转换为统一 Tool 接口
|
|
34
|
+
*/
|
|
35
|
+
registerMCPTools(manager: any): void;
|
|
36
|
+
/**
|
|
37
|
+
* 获取所有工具的 Anthropic 格式定义(传给 LLM)
|
|
38
|
+
*/
|
|
39
|
+
getToolDefinitions(): {
|
|
40
|
+
name: any;
|
|
41
|
+
description: any;
|
|
42
|
+
input_schema: any;
|
|
43
|
+
}[];
|
|
44
|
+
/**
|
|
45
|
+
* 执行工具
|
|
46
|
+
*/
|
|
47
|
+
executeTool(name: any, args: any): Promise<any>;
|
|
48
|
+
/**
|
|
49
|
+
* 查询工具是否存在
|
|
50
|
+
*/
|
|
51
|
+
hasTool(name: any): boolean;
|
|
52
|
+
/**
|
|
53
|
+
* 已注册工具数量
|
|
54
|
+
*/
|
|
55
|
+
get size(): number;
|
|
56
|
+
}
|
|
@@ -0,0 +1,123 @@
|
|
|
1
|
+
// @ts-nocheck
|
|
2
|
+
import { UserInterruptError } from '../errors.js';
|
|
3
|
+
import { getToolDefinitions, callMcpTool } from '../mcp/bridge.js';
|
|
4
|
+
import { createShellTool } from './shell.js';
|
|
5
|
+
import { SSHConnectionPool, createSSHTool } from './ssh.js';
|
|
6
|
+
import { createAskUserTool } from './ask.js';
|
|
7
|
+
import { SkillRegistry, createListSkillsTool } from '../skills/registry.js';
|
|
8
|
+
import { createLoadSkillTool } from '../skills/loader.js';
|
|
9
|
+
import { createPluginInitTool } from './plugin-init.js';
|
|
10
|
+
import { createSkillsSyncTool } from './skills-sync.js';
|
|
11
|
+
import { createServerListTool, createServerAddTool, createServerRemoveTool } from './server.js';
|
|
12
|
+
/**
|
|
13
|
+
* 统一工具注册中心
|
|
14
|
+
* 管理所有可用工具(MCP 工具 + 未来的原生工具)
|
|
15
|
+
*/
|
|
16
|
+
export class ToolRegistry {
|
|
17
|
+
tools = new Map();
|
|
18
|
+
sshPool = new SSHConnectionPool();
|
|
19
|
+
skillRegistry = new SkillRegistry();
|
|
20
|
+
/**
|
|
21
|
+
* 注册单个工具
|
|
22
|
+
*/
|
|
23
|
+
register(tool) {
|
|
24
|
+
this.tools.set(tool.name, tool);
|
|
25
|
+
}
|
|
26
|
+
/**
|
|
27
|
+
* 注册 Native Tools(本地工具)
|
|
28
|
+
*/
|
|
29
|
+
registerNativeTools() {
|
|
30
|
+
this.register(createShellTool());
|
|
31
|
+
this.register(createSSHTool(this.sshPool));
|
|
32
|
+
this.register(createAskUserTool());
|
|
33
|
+
this.register(createListSkillsTool(this.skillRegistry));
|
|
34
|
+
this.register(createLoadSkillTool(this.skillRegistry));
|
|
35
|
+
this.register(createPluginInitTool());
|
|
36
|
+
this.register(createSkillsSyncTool());
|
|
37
|
+
this.register(createServerListTool());
|
|
38
|
+
this.register(createServerAddTool());
|
|
39
|
+
this.register(createServerRemoveTool());
|
|
40
|
+
}
|
|
41
|
+
/**
|
|
42
|
+
* 加载 Skill 索引
|
|
43
|
+
*/
|
|
44
|
+
async loadSkills(indexPath) {
|
|
45
|
+
await this.skillRegistry.load(indexPath);
|
|
46
|
+
}
|
|
47
|
+
/**
|
|
48
|
+
* 获取已加载的技能列表(供 System Prompt 构建使用)
|
|
49
|
+
*/
|
|
50
|
+
getSkills() {
|
|
51
|
+
return this.skillRegistry.getAll();
|
|
52
|
+
}
|
|
53
|
+
/**
|
|
54
|
+
* 清理资源(关闭 SSH 连接等)
|
|
55
|
+
*/
|
|
56
|
+
async cleanup() {
|
|
57
|
+
await this.sshPool.closeAll();
|
|
58
|
+
}
|
|
59
|
+
/**
|
|
60
|
+
* 批量注册 MCP 工具
|
|
61
|
+
* 从 MCPManager 获取所有工具,转换为统一 Tool 接口
|
|
62
|
+
*/
|
|
63
|
+
registerMCPTools(manager) {
|
|
64
|
+
const definitions = getToolDefinitions(manager);
|
|
65
|
+
for (const def of definitions) {
|
|
66
|
+
const tool = {
|
|
67
|
+
name: def.name,
|
|
68
|
+
description: def.description,
|
|
69
|
+
inputSchema: def.input_schema,
|
|
70
|
+
execute: async (args) => {
|
|
71
|
+
const result = await callMcpTool(manager, def.name, args);
|
|
72
|
+
return { content: result.content, isError: result.isError };
|
|
73
|
+
},
|
|
74
|
+
};
|
|
75
|
+
this.tools.set(tool.name, tool);
|
|
76
|
+
}
|
|
77
|
+
}
|
|
78
|
+
/**
|
|
79
|
+
* 获取所有工具的 Anthropic 格式定义(传给 LLM)
|
|
80
|
+
*/
|
|
81
|
+
getToolDefinitions() {
|
|
82
|
+
return Array.from(this.tools.values()).map((tool) => ({
|
|
83
|
+
name: tool.name,
|
|
84
|
+
description: tool.description,
|
|
85
|
+
input_schema: tool.inputSchema,
|
|
86
|
+
}));
|
|
87
|
+
}
|
|
88
|
+
/**
|
|
89
|
+
* 执行工具
|
|
90
|
+
*/
|
|
91
|
+
async executeTool(name, args) {
|
|
92
|
+
const tool = this.tools.get(name);
|
|
93
|
+
if (!tool) {
|
|
94
|
+
return { content: `未知工具: ${name}`, isError: true };
|
|
95
|
+
}
|
|
96
|
+
try {
|
|
97
|
+
return await tool.execute(args);
|
|
98
|
+
}
|
|
99
|
+
catch (error) {
|
|
100
|
+
// 用户中断:放行,让错误传播到 agent loop 以终止当前任务
|
|
101
|
+
if (error instanceof UserInterruptError) {
|
|
102
|
+
throw error;
|
|
103
|
+
}
|
|
104
|
+
return {
|
|
105
|
+
content: `工具执行失败 [${name}]: ${error instanceof Error ? error.message : String(error)}`,
|
|
106
|
+
isError: true,
|
|
107
|
+
};
|
|
108
|
+
}
|
|
109
|
+
}
|
|
110
|
+
/**
|
|
111
|
+
* 查询工具是否存在
|
|
112
|
+
*/
|
|
113
|
+
hasTool(name) {
|
|
114
|
+
return this.tools.has(name);
|
|
115
|
+
}
|
|
116
|
+
/**
|
|
117
|
+
* 已注册工具数量
|
|
118
|
+
*/
|
|
119
|
+
get size() {
|
|
120
|
+
return this.tools.size;
|
|
121
|
+
}
|
|
122
|
+
}
|
|
123
|
+
//# sourceMappingURL=registry.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"registry.js","sourceRoot":"","sources":["../../src/tools/registry.ts"],"names":[],"mappings":"AAAA,cAAc;AACd,OAAO,EAAE,kBAAkB,EAAE,MAAM,cAAc,CAAC;AAClD,OAAO,EAAE,kBAAkB,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AACnE,OAAO,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AAC7C,OAAO,EAAE,iBAAiB,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AAC5D,OAAO,EAAE,iBAAiB,EAAE,MAAM,UAAU,CAAC;AAC7C,OAAO,EAAE,aAAa,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAC;AAC5E,OAAO,EAAE,mBAAmB,EAAE,MAAM,qBAAqB,CAAC;AAC1D,OAAO,EAAE,oBAAoB,EAAE,MAAM,kBAAkB,CAAC;AACxD,OAAO,EAAE,oBAAoB,EAAE,MAAM,kBAAkB,CAAC;AACxD,OAAO,EAAE,oBAAoB,EAAE,mBAAmB,EAAE,sBAAsB,EAAE,MAAM,aAAa,CAAC;AAChG;;;GAGG;AACH,MAAM,OAAO,YAAY;IACrB,KAAK,GAAG,IAAI,GAAG,EAAE,CAAC;IAClB,OAAO,GAAG,IAAI,iBAAiB,EAAE,CAAC;IAClC,aAAa,GAAG,IAAI,aAAa,EAAE,CAAC;IACpC;;OAEG;IACH,QAAQ,CAAC,IAAI;QACT,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;IACpC,CAAC;IACD;;OAEG;IACH,mBAAmB;QACf,IAAI,CAAC,QAAQ,CAAC,eAAe,EAAE,CAAC,CAAC;QACjC,IAAI,CAAC,QAAQ,CAAC,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC;QAC3C,IAAI,CAAC,QAAQ,CAAC,iBAAiB,EAAE,CAAC,CAAC;QACnC,IAAI,CAAC,QAAQ,CAAC,oBAAoB,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC;QACxD,IAAI,CAAC,QAAQ,CAAC,mBAAmB,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC;QACvD,IAAI,CAAC,QAAQ,CAAC,oBAAoB,EAAE,CAAC,CAAC;QACtC,IAAI,CAAC,QAAQ,CAAC,oBAAoB,EAAE,CAAC,CAAC;QACtC,IAAI,CAAC,QAAQ,CAAC,oBAAoB,EAAE,CAAC,CAAC;QACtC,IAAI,CAAC,QAAQ,CAAC,mBAAmB,EAAE,CAAC,CAAC;QACrC,IAAI,CAAC,QAAQ,CAAC,sBAAsB,EAAE,CAAC,CAAC;IAC5C,CAAC;IACD;;OAEG;IACH,KAAK,CAAC,UAAU,CAAC,SAAS;QACtB,MAAM,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAC7C,CAAC;IACD;;OAEG;IACH,SAAS;QACL,OAAO,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,CAAC;IACvC,CAAC;IACD;;OAEG;IACH,KAAK,CAAC,OAAO;QACT,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC;IAClC,CAAC;IACD;;;OAGG;IACH,gBAAgB,CAAC,OAAO;QACpB,MAAM,WAAW,GAAG,kBAAkB,CAAC,OAAO,CAAC,CAAC;QAChD,KAAK,MAAM,GAAG,IAAI,WAAW,EAAE,CAAC;YAC5B,MAAM,IAAI,GAAG;gBACT,IAAI,EAAE,GAAG,CAAC,IAAI;gBACd,WAAW,EAAE,GAAG,CAAC,WAAW;gBAC5B,WAAW,EAAE,GAAG,CAAC,YAAY;gBAC7B,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;oBACpB,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,OAAO,EAAE,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;oBAC1D,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC;gBAChE,CAAC;aACJ,CAAC;YACF,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;QACpC,CAAC;IACL,CAAC;IACD;;OAEG;IACH,kBAAkB;QACd,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YAClD,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,YAAY,EAAE,IAAI,CAAC,WAAW;SACjC,CAAC,CAAC,CAAC;IACR,CAAC;IACD;;OAEG;IACH,KAAK,CAAC,WAAW,CAAC,IAAI,EAAE,IAAI;QACxB,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAClC,IAAI,CAAC,IAAI,EAAE,CAAC;YACR,OAAO,EAAE,OAAO,EAAE,SAAS,IAAI,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;QACvD,CAAC;QACD,IAAI,CAAC;YACD,OAAO,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QACpC,CAAC;QACD,OAAO,KAAK,EAAE,CAAC;YACX,oCAAoC;YACpC,IAAI,KAAK,YAAY,kBAAkB,EAAE,CAAC;gBACtC,MAAM,KAAK,CAAC;YAChB,CAAC;YACD,OAAO;gBACH,OAAO,EAAE,WAAW,IAAI,MAAM,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE;gBACtF,OAAO,EAAE,IAAI;aAChB,CAAC;QACN,CAAC;IACL,CAAC;IACD;;OAEG;IACH,OAAO,CAAC,IAAI;QACR,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IAChC,CAAC;IACD;;OAEG;IACH,IAAI,IAAI;QACJ,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC;IAC3B,CAAC;CACJ"}
|
|
@@ -0,0 +1,30 @@
|
|
|
1
|
+
const HIGH_RISK_PATTERNS = [
|
|
2
|
+
/\brm\s+-rf\b/i,
|
|
3
|
+
/\bmkfs(\.\w+)?\b/i,
|
|
4
|
+
/\bdd\s+if=/i,
|
|
5
|
+
/\bshutdown\b/i,
|
|
6
|
+
/\breboot\b/i,
|
|
7
|
+
/\bpoweroff\b/i,
|
|
8
|
+
/\bk3s-killall\.sh\b/i,
|
|
9
|
+
/\bsystemctl\s+(stop|disable|mask|restart)\b/i,
|
|
10
|
+
/\bservice\s+\S+\s+stop\b/i,
|
|
11
|
+
/\b(drop|truncate)\s+table\b/i,
|
|
12
|
+
/\bdelete\s+from\b/i,
|
|
13
|
+
/\bupdate\s+\S+\s+set\b/i,
|
|
14
|
+
/\buninstall\b/i,
|
|
15
|
+
/\bkill\s+-9\b/i,
|
|
16
|
+
];
|
|
17
|
+
export function isHighRiskCommand(command) {
|
|
18
|
+
if (!command || !command.trim()) {
|
|
19
|
+
return false;
|
|
20
|
+
}
|
|
21
|
+
return HIGH_RISK_PATTERNS.some((pattern) => pattern.test(command));
|
|
22
|
+
}
|
|
23
|
+
export function requireConfirmationError(command) {
|
|
24
|
+
return [
|
|
25
|
+
'检测到高危命令,已被安全策略拦截。',
|
|
26
|
+
'请先调用 ask_user 工具明确征得用户确认,再重试并传入 confirmed=true。',
|
|
27
|
+
`命令: ${command}`,
|
|
28
|
+
].join('\n');
|
|
29
|
+
}
|
|
30
|
+
//# sourceMappingURL=safety.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"safety.js","sourceRoot":"","sources":["../../src/tools/safety.ts"],"names":[],"mappings":"AAAA,MAAM,kBAAkB,GAAa;IACnC,eAAe;IACf,mBAAmB;IACnB,aAAa;IACb,eAAe;IACf,aAAa;IACb,eAAe;IACf,sBAAsB;IACtB,8CAA8C;IAC9C,2BAA2B;IAC3B,8BAA8B;IAC9B,oBAAoB;IACpB,yBAAyB;IACzB,gBAAgB;IAChB,gBAAgB;CACjB,CAAC;AAEF,MAAM,UAAU,iBAAiB,CAAC,OAAe;IAC/C,IAAI,CAAC,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC;QAChC,OAAO,KAAK,CAAC;IACf,CAAC;IAED,OAAO,kBAAkB,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC;AACrE,CAAC;AAED,MAAM,UAAU,wBAAwB,CAAC,OAAe;IACtD,OAAO;QACL,mBAAmB;QACnB,iDAAiD;QACjD,OAAO,OAAO,EAAE;KACjB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACf,CAAC"}
|
|
@@ -0,0 +1,91 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* 创建 server_list 工具
|
|
3
|
+
*/
|
|
4
|
+
export declare function createServerListTool(): {
|
|
5
|
+
name: string;
|
|
6
|
+
description: string;
|
|
7
|
+
inputSchema: {
|
|
8
|
+
type: string;
|
|
9
|
+
properties: {};
|
|
10
|
+
};
|
|
11
|
+
execute: () => Promise<{
|
|
12
|
+
content: string;
|
|
13
|
+
}>;
|
|
14
|
+
};
|
|
15
|
+
/**
|
|
16
|
+
* 创建 server_add 工具
|
|
17
|
+
*/
|
|
18
|
+
export declare function createServerAddTool(): {
|
|
19
|
+
name: string;
|
|
20
|
+
description: string;
|
|
21
|
+
inputSchema: {
|
|
22
|
+
type: string;
|
|
23
|
+
properties: {
|
|
24
|
+
name: {
|
|
25
|
+
type: string;
|
|
26
|
+
description: string;
|
|
27
|
+
};
|
|
28
|
+
host: {
|
|
29
|
+
type: string;
|
|
30
|
+
description: string;
|
|
31
|
+
};
|
|
32
|
+
user: {
|
|
33
|
+
type: string;
|
|
34
|
+
description: string;
|
|
35
|
+
};
|
|
36
|
+
port: {
|
|
37
|
+
type: string;
|
|
38
|
+
description: string;
|
|
39
|
+
};
|
|
40
|
+
authType: {
|
|
41
|
+
type: string;
|
|
42
|
+
enum: string[];
|
|
43
|
+
description: string;
|
|
44
|
+
};
|
|
45
|
+
identityFile: {
|
|
46
|
+
type: string;
|
|
47
|
+
description: string;
|
|
48
|
+
};
|
|
49
|
+
password: {
|
|
50
|
+
type: string;
|
|
51
|
+
description: string;
|
|
52
|
+
};
|
|
53
|
+
description: {
|
|
54
|
+
type: string;
|
|
55
|
+
description: string;
|
|
56
|
+
};
|
|
57
|
+
};
|
|
58
|
+
required: string[];
|
|
59
|
+
};
|
|
60
|
+
execute: (args: any) => Promise<{
|
|
61
|
+
content: string;
|
|
62
|
+
isError: boolean;
|
|
63
|
+
} | {
|
|
64
|
+
content: string;
|
|
65
|
+
isError?: undefined;
|
|
66
|
+
}>;
|
|
67
|
+
};
|
|
68
|
+
/**
|
|
69
|
+
* 创建 server_remove 工具
|
|
70
|
+
*/
|
|
71
|
+
export declare function createServerRemoveTool(): {
|
|
72
|
+
name: string;
|
|
73
|
+
description: string;
|
|
74
|
+
inputSchema: {
|
|
75
|
+
type: string;
|
|
76
|
+
properties: {
|
|
77
|
+
name: {
|
|
78
|
+
type: string;
|
|
79
|
+
description: string;
|
|
80
|
+
};
|
|
81
|
+
};
|
|
82
|
+
required: string[];
|
|
83
|
+
};
|
|
84
|
+
execute: (args: any) => Promise<{
|
|
85
|
+
content: string;
|
|
86
|
+
isError: boolean;
|
|
87
|
+
} | {
|
|
88
|
+
content: string;
|
|
89
|
+
isError?: undefined;
|
|
90
|
+
}>;
|
|
91
|
+
};
|
|
@@ -0,0 +1,149 @@
|
|
|
1
|
+
// @ts-nocheck
|
|
2
|
+
import { loadServers, saveServers } from '../config/manager.js';
|
|
3
|
+
/**
|
|
4
|
+
* 创建 server_list 工具
|
|
5
|
+
*/
|
|
6
|
+
export function createServerListTool() {
|
|
7
|
+
return {
|
|
8
|
+
name: 'server_list',
|
|
9
|
+
description: '列出所有已配置的 SSH 服务器。',
|
|
10
|
+
inputSchema: {
|
|
11
|
+
type: 'object',
|
|
12
|
+
properties: {},
|
|
13
|
+
},
|
|
14
|
+
execute: async () => {
|
|
15
|
+
const { servers } = await loadServers();
|
|
16
|
+
if (servers.length === 0) {
|
|
17
|
+
return { content: '没有已配置的服务器。可以通过 server_add 工具添加。' };
|
|
18
|
+
}
|
|
19
|
+
const lines = [`共 ${servers.length} 台服务器:`, ''];
|
|
20
|
+
for (const s of servers) {
|
|
21
|
+
const auth = s.auth_type === 'password' ? 'password' : 'ssh-key';
|
|
22
|
+
lines.push(`- ${s.name}: ${s.user}@${s.host}:${s.port} (${auth})${s.description ? ` — ${s.description}` : ''}`);
|
|
23
|
+
}
|
|
24
|
+
return { content: lines.join('\n') };
|
|
25
|
+
},
|
|
26
|
+
};
|
|
27
|
+
}
|
|
28
|
+
/**
|
|
29
|
+
* 创建 server_add 工具
|
|
30
|
+
*/
|
|
31
|
+
export function createServerAddTool() {
|
|
32
|
+
return {
|
|
33
|
+
name: 'server_add',
|
|
34
|
+
description: '添加一台 SSH 服务器配置。调用前请通过对话确认服务器名称、主机地址等信息。如需密码认证,应通过 ask_user 工具收集密码。',
|
|
35
|
+
inputSchema: {
|
|
36
|
+
type: 'object',
|
|
37
|
+
properties: {
|
|
38
|
+
name: {
|
|
39
|
+
type: 'string',
|
|
40
|
+
description: '服务器名称标识(唯一)',
|
|
41
|
+
},
|
|
42
|
+
host: {
|
|
43
|
+
type: 'string',
|
|
44
|
+
description: '主机地址(IP 或域名)',
|
|
45
|
+
},
|
|
46
|
+
user: {
|
|
47
|
+
type: 'string',
|
|
48
|
+
description: 'SSH 用户名,默认 root',
|
|
49
|
+
},
|
|
50
|
+
port: {
|
|
51
|
+
type: 'number',
|
|
52
|
+
description: 'SSH 端口,默认 22',
|
|
53
|
+
},
|
|
54
|
+
authType: {
|
|
55
|
+
type: 'string',
|
|
56
|
+
enum: ['ssh-key', 'password'],
|
|
57
|
+
description: '认证方式,默认 ssh-key',
|
|
58
|
+
},
|
|
59
|
+
identityFile: {
|
|
60
|
+
type: 'string',
|
|
61
|
+
description: 'SSH 密钥文件路径(authType=ssh-key 时使用),默认 ~/.ssh/id_rsa',
|
|
62
|
+
},
|
|
63
|
+
password: {
|
|
64
|
+
type: 'string',
|
|
65
|
+
description: 'SSH 密码(authType=password 时使用)',
|
|
66
|
+
},
|
|
67
|
+
description: {
|
|
68
|
+
type: 'string',
|
|
69
|
+
description: '服务器描述(可选)',
|
|
70
|
+
},
|
|
71
|
+
},
|
|
72
|
+
required: ['name', 'host'],
|
|
73
|
+
},
|
|
74
|
+
execute: async (args) => {
|
|
75
|
+
const name = args.name;
|
|
76
|
+
const host = args.host;
|
|
77
|
+
const user = args.user || 'root';
|
|
78
|
+
const port = typeof args.port === 'number' ? args.port : 22;
|
|
79
|
+
const authType = args.authType === 'password' ? 'password' : 'ssh-key';
|
|
80
|
+
const identityFile = args.identityFile || (authType === 'ssh-key' ? '~/.ssh/id_rsa' : undefined);
|
|
81
|
+
const password = args.password;
|
|
82
|
+
const description = args.description || '';
|
|
83
|
+
// 校验
|
|
84
|
+
if (!name) {
|
|
85
|
+
return { content: '服务器名称不能为空', isError: true };
|
|
86
|
+
}
|
|
87
|
+
if (!host) {
|
|
88
|
+
return { content: '主机地址不能为空', isError: true };
|
|
89
|
+
}
|
|
90
|
+
if (isNaN(port) || port < 1 || port > 65535) {
|
|
91
|
+
return { content: '端口号无效(1-65535)', isError: true };
|
|
92
|
+
}
|
|
93
|
+
if (authType === 'password' && !password) {
|
|
94
|
+
return { content: '密码认证方式需要提供密码,请通过 ask_user 工具向用户收集', isError: true };
|
|
95
|
+
}
|
|
96
|
+
const config = await loadServers();
|
|
97
|
+
// 重名检测
|
|
98
|
+
if (config.servers.some((s) => s.name === name)) {
|
|
99
|
+
return { content: `服务器 "${name}" 已存在,请使用其他名称`, isError: true };
|
|
100
|
+
}
|
|
101
|
+
config.servers.push({
|
|
102
|
+
name,
|
|
103
|
+
host,
|
|
104
|
+
user,
|
|
105
|
+
port,
|
|
106
|
+
auth_type: authType,
|
|
107
|
+
identity_file: identityFile,
|
|
108
|
+
password,
|
|
109
|
+
description,
|
|
110
|
+
});
|
|
111
|
+
await saveServers(config);
|
|
112
|
+
return { content: `服务器 "${name}" 添加成功 (${user}@${host}:${port}, ${authType})` };
|
|
113
|
+
},
|
|
114
|
+
};
|
|
115
|
+
}
|
|
116
|
+
/**
|
|
117
|
+
* 创建 server_remove 工具
|
|
118
|
+
*/
|
|
119
|
+
export function createServerRemoveTool() {
|
|
120
|
+
return {
|
|
121
|
+
name: 'server_remove',
|
|
122
|
+
description: '删除一台已配置的 SSH 服务器。这是不可逆操作,调用前应通过 ask_user 工具确认用户意图。',
|
|
123
|
+
inputSchema: {
|
|
124
|
+
type: 'object',
|
|
125
|
+
properties: {
|
|
126
|
+
name: {
|
|
127
|
+
type: 'string',
|
|
128
|
+
description: '要删除的服务器名称',
|
|
129
|
+
},
|
|
130
|
+
},
|
|
131
|
+
required: ['name'],
|
|
132
|
+
},
|
|
133
|
+
execute: async (args) => {
|
|
134
|
+
const name = args.name;
|
|
135
|
+
if (!name) {
|
|
136
|
+
return { content: '服务器名称不能为空', isError: true };
|
|
137
|
+
}
|
|
138
|
+
const config = await loadServers();
|
|
139
|
+
const index = config.servers.findIndex((s) => s.name === name);
|
|
140
|
+
if (index === -1) {
|
|
141
|
+
return { content: `未找到服务器: ${name}`, isError: true };
|
|
142
|
+
}
|
|
143
|
+
config.servers.splice(index, 1);
|
|
144
|
+
await saveServers(config);
|
|
145
|
+
return { content: `服务器 "${name}" 已删除` };
|
|
146
|
+
},
|
|
147
|
+
};
|
|
148
|
+
}
|
|
149
|
+
//# sourceMappingURL=server.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"server.js","sourceRoot":"","sources":["../../src/tools/server.ts"],"names":[],"mappings":"AAAA,cAAc;AACd,OAAO,EAAE,WAAW,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AAChE;;GAEG;AACH,MAAM,UAAU,oBAAoB;IAChC,OAAO;QACH,IAAI,EAAE,aAAa;QACnB,WAAW,EAAE,mBAAmB;QAChC,WAAW,EAAE;YACT,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE,EAAE;SACjB;QACD,OAAO,EAAE,KAAK,IAAI,EAAE;YAChB,MAAM,EAAE,OAAO,EAAE,GAAG,MAAM,WAAW,EAAE,CAAC;YACxC,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACvB,OAAO,EAAE,OAAO,EAAE,iCAAiC,EAAE,CAAC;YAC1D,CAAC;YACD,MAAM,KAAK,GAAG,CAAC,KAAK,OAAO,CAAC,MAAM,QAAQ,EAAE,EAAE,CAAC,CAAC;YAChD,KAAK,MAAM,CAAC,IAAI,OAAO,EAAE,CAAC;gBACtB,MAAM,IAAI,GAAG,CAAC,CAAC,SAAS,KAAK,UAAU,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC;gBACjE,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,IAAI,KAAK,IAAI,IAAI,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACpH,CAAC;YACD,OAAO,EAAE,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QACzC,CAAC;KACJ,CAAC;AACN,CAAC;AACD;;GAEG;AACH,MAAM,UAAU,mBAAmB;IAC/B,OAAO;QACH,IAAI,EAAE,YAAY;QAClB,WAAW,EAAE,oEAAoE;QACjF,WAAW,EAAE;YACT,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACR,IAAI,EAAE;oBACF,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,aAAa;iBAC7B;gBACD,IAAI,EAAE;oBACF,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,cAAc;iBAC9B;gBACD,IAAI,EAAE;oBACF,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,iBAAiB;iBACjC;gBACD,IAAI,EAAE;oBACF,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,cAAc;iBAC9B;gBACD,QAAQ,EAAE;oBACN,IAAI,EAAE,QAAQ;oBACd,IAAI,EAAE,CAAC,SAAS,EAAE,UAAU,CAAC;oBAC7B,WAAW,EAAE,iBAAiB;iBACjC;gBACD,YAAY,EAAE;oBACV,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,mDAAmD;iBACnE;gBACD,QAAQ,EAAE;oBACN,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,+BAA+B;iBAC/C;gBACD,WAAW,EAAE;oBACT,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,WAAW;iBAC3B;aACJ;YACD,QAAQ,EAAE,CAAC,MAAM,EAAE,MAAM,CAAC;SAC7B;QACD,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YACpB,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC;YACvB,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC;YACvB,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,IAAI,MAAM,CAAC;YACjC,MAAM,IAAI,GAAG,OAAO,IAAI,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC;YAC5D,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,KAAK,UAAU,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC;YACvE,MAAM,YAAY,GAAG,IAAI,CAAC,YAAY,IAAI,CAAC,QAAQ,KAAK,SAAS,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;YACjG,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC;YAC/B,MAAM,WAAW,GAAG,IAAI,CAAC,WAAW,IAAI,EAAE,CAAC;YAC3C,KAAK;YACL,IAAI,CAAC,IAAI,EAAE,CAAC;gBACR,OAAO,EAAE,OAAO,EAAE,WAAW,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;YACnD,CAAC;YACD,IAAI,CAAC,IAAI,EAAE,CAAC;gBACR,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;YAClD,CAAC;YACD,IAAI,KAAK,CAAC,IAAI,CAAC,IAAI,IAAI,GAAG,CAAC,IAAI,IAAI,GAAG,KAAK,EAAE,CAAC;gBAC1C,OAAO,EAAE,OAAO,EAAE,gBAAgB,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;YACxD,CAAC;YACD,IAAI,QAAQ,KAAK,UAAU,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACvC,OAAO,EAAE,OAAO,EAAE,mCAAmC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;YAC3E,CAAC;YACD,MAAM,MAAM,GAAG,MAAM,WAAW,EAAE,CAAC;YACnC,OAAO;YACP,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,CAAC,EAAE,CAAC;gBAC9C,OAAO,EAAE,OAAO,EAAE,QAAQ,IAAI,eAAe,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;YACnE,CAAC;YACD,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC;gBAChB,IAAI;gBACJ,IAAI;gBACJ,IAAI;gBACJ,IAAI;gBACJ,SAAS,EAAE,QAAQ;gBACnB,aAAa,EAAE,YAAY;gBAC3B,QAAQ;gBACR,WAAW;aACd,CAAC,CAAC;YACH,MAAM,WAAW,CAAC,MAAM,CAAC,CAAC;YAC1B,OAAO,EAAE,OAAO,EAAE,QAAQ,IAAI,WAAW,IAAI,IAAI,IAAI,IAAI,IAAI,KAAK,QAAQ,GAAG,EAAE,CAAC;QACpF,CAAC;KACJ,CAAC;AACN,CAAC;AACD;;GAEG;AACH,MAAM,UAAU,sBAAsB;IAClC,OAAO;QACH,IAAI,EAAE,eAAe;QACrB,WAAW,EAAE,oDAAoD;QACjE,WAAW,EAAE;YACT,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACR,IAAI,EAAE;oBACF,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,WAAW;iBAC3B;aACJ;YACD,QAAQ,EAAE,CAAC,MAAM,CAAC;SACrB;QACD,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YACpB,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC;YACvB,IAAI,CAAC,IAAI,EAAE,CAAC;gBACR,OAAO,EAAE,OAAO,EAAE,WAAW,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;YACnD,CAAC;YACD,MAAM,MAAM,GAAG,MAAM,WAAW,EAAE,CAAC;YACnC,MAAM,KAAK,GAAG,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC;YAC/D,IAAI,KAAK,KAAK,CAAC,CAAC,EAAE,CAAC;gBACf,OAAO,EAAE,OAAO,EAAE,WAAW,IAAI,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;YACzD,CAAC;YACD,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;YAChC,MAAM,WAAW,CAAC,MAAM,CAAC,CAAC;YAC1B,OAAO,EAAE,OAAO,EAAE,QAAQ,IAAI,OAAO,EAAE,CAAC;QAC5C,CAAC;KACJ,CAAC;AACN,CAAC"}
|
|
@@ -0,0 +1,11 @@
|
|
|
1
|
+
import type { Tool } from './types.js';
|
|
2
|
+
/**
|
|
3
|
+
* 截断过长的输出
|
|
4
|
+
* 超过 MAX_OUTPUT_LENGTH 时保留前后各 KEEP_LENGTH 字符
|
|
5
|
+
*/
|
|
6
|
+
export declare function truncateOutput(output: string): string;
|
|
7
|
+
export declare function shouldBlockCommand(command: string, confirmed?: boolean): boolean;
|
|
8
|
+
/**
|
|
9
|
+
* 创建 Shell 命令执行工具
|
|
10
|
+
*/
|
|
11
|
+
export declare function createShellTool(): Tool;
|