ssh-mcp-pro 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/AGENTS.md +127 -0
- package/ARCHITECTURE.md +145 -0
- package/LICENSE +21 -0
- package/LICENSES/MIT.txt +21 -0
- package/MIGRATION.md +14 -0
- package/README.md +175 -0
- package/REGISTRY_SUBMISSION.md +38 -0
- package/SECURITY.md +40 -0
- package/SECURITY_DECISIONS.md +59 -0
- package/dist/agent-bin.d.ts +3 -0
- package/dist/agent-bin.d.ts.map +1 -0
- package/dist/agent-bin.js +8 -0
- package/dist/agent-bin.js.map +1 -0
- package/dist/audit.d.ts +25 -0
- package/dist/audit.d.ts.map +1 -0
- package/dist/audit.js +50 -0
- package/dist/audit.js.map +1 -0
- package/dist/auth.d.ts +4 -0
- package/dist/auth.d.ts.map +1 -0
- package/dist/auth.js +33 -0
- package/dist/auth.js.map +1 -0
- package/dist/cli.d.ts +16 -0
- package/dist/cli.d.ts.map +1 -0
- package/dist/cli.js +99 -0
- package/dist/cli.js.map +1 -0
- package/dist/config.d.ts +103 -0
- package/dist/config.d.ts.map +1 -0
- package/dist/config.js +490 -0
- package/dist/config.js.map +1 -0
- package/dist/connector-credentials.d.ts +8 -0
- package/dist/connector-credentials.d.ts.map +1 -0
- package/dist/connector-credentials.js +132 -0
- package/dist/connector-credentials.js.map +1 -0
- package/dist/connector-profile.d.ts +17 -0
- package/dist/connector-profile.d.ts.map +1 -0
- package/dist/connector-profile.js +81 -0
- package/dist/connector-profile.js.map +1 -0
- package/dist/container.d.ts +18 -0
- package/dist/container.d.ts.map +1 -0
- package/dist/container.js +52 -0
- package/dist/container.js.map +1 -0
- package/dist/detect.d.ts +7 -0
- package/dist/detect.d.ts.map +1 -0
- package/dist/detect.js +271 -0
- package/dist/detect.js.map +1 -0
- package/dist/ensure.d.ts +17 -0
- package/dist/ensure.d.ts.map +1 -0
- package/dist/ensure.js +531 -0
- package/dist/ensure.js.map +1 -0
- package/dist/errors.d.ts +54 -0
- package/dist/errors.d.ts.map +1 -0
- package/dist/errors.js +84 -0
- package/dist/errors.js.map +1 -0
- package/dist/fs-tools.d.ts +26 -0
- package/dist/fs-tools.d.ts.map +1 -0
- package/dist/fs-tools.js +599 -0
- package/dist/fs-tools.js.map +1 -0
- package/dist/http-rate-limit.d.ts +9 -0
- package/dist/http-rate-limit.d.ts.map +1 -0
- package/dist/http-rate-limit.js +41 -0
- package/dist/http-rate-limit.js.map +1 -0
- package/dist/http-security.d.ts +22 -0
- package/dist/http-security.d.ts.map +1 -0
- package/dist/http-security.js +88 -0
- package/dist/http-security.js.map +1 -0
- package/dist/index.d.ts +10 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +201 -0
- package/dist/index.js.map +1 -0
- package/dist/logging.d.ts +52 -0
- package/dist/logging.d.ts.map +1 -0
- package/dist/logging.js +180 -0
- package/dist/logging.js.map +1 -0
- package/dist/mcp.d.ts +16 -0
- package/dist/mcp.d.ts.map +1 -0
- package/dist/mcp.js +159 -0
- package/dist/mcp.js.map +1 -0
- package/dist/metrics.d.ts +95 -0
- package/dist/metrics.d.ts.map +1 -0
- package/dist/metrics.js +204 -0
- package/dist/metrics.js.map +1 -0
- package/dist/oauth.d.ts +14 -0
- package/dist/oauth.d.ts.map +1 -0
- package/dist/oauth.js +105 -0
- package/dist/oauth.js.map +1 -0
- package/dist/policy.d.ts +64 -0
- package/dist/policy.d.ts.map +1 -0
- package/dist/policy.js +368 -0
- package/dist/policy.js.map +1 -0
- package/dist/process.d.ts +24 -0
- package/dist/process.d.ts.map +1 -0
- package/dist/process.js +212 -0
- package/dist/process.js.map +1 -0
- package/dist/prompts.d.ts +49 -0
- package/dist/prompts.d.ts.map +1 -0
- package/dist/prompts.js +191 -0
- package/dist/prompts.js.map +1 -0
- package/dist/rate-limiter.d.ts +57 -0
- package/dist/rate-limiter.d.ts.map +1 -0
- package/dist/rate-limiter.js +141 -0
- package/dist/rate-limiter.js.map +1 -0
- package/dist/remote/agent-cli.d.ts +2 -0
- package/dist/remote/agent-cli.d.ts.map +1 -0
- package/dist/remote/agent-cli.js +270 -0
- package/dist/remote/agent-cli.js.map +1 -0
- package/dist/remote/agent-executor.d.ts +26 -0
- package/dist/remote/agent-executor.d.ts.map +1 -0
- package/dist/remote/agent-executor.js +400 -0
- package/dist/remote/agent-executor.js.map +1 -0
- package/dist/remote/config.d.ts +3 -0
- package/dist/remote/config.d.ts.map +1 -0
- package/dist/remote/config.js +52 -0
- package/dist/remote/config.js.map +1 -0
- package/dist/remote/control-plane.d.ts +57 -0
- package/dist/remote/control-plane.d.ts.map +1 -0
- package/dist/remote/control-plane.js +1248 -0
- package/dist/remote/control-plane.js.map +1 -0
- package/dist/remote/crypto.d.ts +38 -0
- package/dist/remote/crypto.d.ts.map +1 -0
- package/dist/remote/crypto.js +143 -0
- package/dist/remote/crypto.js.map +1 -0
- package/dist/remote/mcp-tools.d.ts +10 -0
- package/dist/remote/mcp-tools.d.ts.map +1 -0
- package/dist/remote/mcp-tools.js +201 -0
- package/dist/remote/mcp-tools.js.map +1 -0
- package/dist/remote/policy.d.ts +11 -0
- package/dist/remote/policy.d.ts.map +1 -0
- package/dist/remote/policy.js +94 -0
- package/dist/remote/policy.js.map +1 -0
- package/dist/remote/schemas.d.ts +298 -0
- package/dist/remote/schemas.d.ts.map +1 -0
- package/dist/remote/schemas.js +111 -0
- package/dist/remote/schemas.js.map +1 -0
- package/dist/remote/scopes.d.ts +6 -0
- package/dist/remote/scopes.d.ts.map +1 -0
- package/dist/remote/scopes.js +24 -0
- package/dist/remote/scopes.js.map +1 -0
- package/dist/remote/store.d.ts +45 -0
- package/dist/remote/store.d.ts.map +1 -0
- package/dist/remote/store.js +355 -0
- package/dist/remote/store.js.map +1 -0
- package/dist/remote/types.d.ts +183 -0
- package/dist/remote/types.d.ts.map +1 -0
- package/dist/remote/types.js +103 -0
- package/dist/remote/types.js.map +1 -0
- package/dist/remote/util.d.ts +6 -0
- package/dist/remote/util.d.ts.map +1 -0
- package/dist/remote/util.js +45 -0
- package/dist/remote/util.js.map +1 -0
- package/dist/remote/websocket.d.ts +26 -0
- package/dist/remote/websocket.d.ts.map +1 -0
- package/dist/remote/websocket.js +167 -0
- package/dist/remote/websocket.js.map +1 -0
- package/dist/render-http.d.ts +2 -0
- package/dist/render-http.d.ts.map +1 -0
- package/dist/render-http.js +14 -0
- package/dist/render-http.js.map +1 -0
- package/dist/resources.d.ts +19 -0
- package/dist/resources.d.ts.map +1 -0
- package/dist/resources.js +96 -0
- package/dist/resources.js.map +1 -0
- package/dist/retry.d.ts +45 -0
- package/dist/retry.d.ts.map +1 -0
- package/dist/retry.js +120 -0
- package/dist/retry.js.map +1 -0
- package/dist/safety.d.ts +31 -0
- package/dist/safety.d.ts.map +1 -0
- package/dist/safety.js +174 -0
- package/dist/safety.js.map +1 -0
- package/dist/server-http.d.ts +2 -0
- package/dist/server-http.d.ts.map +1 -0
- package/dist/server-http.js +432 -0
- package/dist/server-http.js.map +1 -0
- package/dist/session.d.ts +116 -0
- package/dist/session.d.ts.map +1 -0
- package/dist/session.js +666 -0
- package/dist/session.js.map +1 -0
- package/dist/shell.d.ts +10 -0
- package/dist/shell.d.ts.map +1 -0
- package/dist/shell.js +83 -0
- package/dist/shell.js.map +1 -0
- package/dist/ssh-config.d.ts +94 -0
- package/dist/ssh-config.d.ts.map +1 -0
- package/dist/ssh-config.js +234 -0
- package/dist/ssh-config.js.map +1 -0
- package/dist/streaming.d.ts +36 -0
- package/dist/streaming.d.ts.map +1 -0
- package/dist/streaming.js +140 -0
- package/dist/streaming.js.map +1 -0
- package/dist/telemetry.d.ts +17 -0
- package/dist/telemetry.d.ts.map +1 -0
- package/dist/telemetry.js +101 -0
- package/dist/telemetry.js.map +1 -0
- package/dist/tools/connector.provider.d.ts +28 -0
- package/dist/tools/connector.provider.d.ts.map +1 -0
- package/dist/tools/connector.provider.js +360 -0
- package/dist/tools/connector.provider.js.map +1 -0
- package/dist/tools/ensure.provider.d.ts +18 -0
- package/dist/tools/ensure.provider.d.ts.map +1 -0
- package/dist/tools/ensure.provider.js +173 -0
- package/dist/tools/ensure.provider.js.map +1 -0
- package/dist/tools/fs.provider.d.ts +21 -0
- package/dist/tools/fs.provider.d.ts.map +1 -0
- package/dist/tools/fs.provider.js +259 -0
- package/dist/tools/fs.provider.js.map +1 -0
- package/dist/tools/index.d.ts +4 -0
- package/dist/tools/index.d.ts.map +1 -0
- package/dist/tools/index.js +68 -0
- package/dist/tools/index.js.map +1 -0
- package/dist/tools/metadata.d.ts +11 -0
- package/dist/tools/metadata.d.ts.map +1 -0
- package/dist/tools/metadata.js +10 -0
- package/dist/tools/metadata.js.map +1 -0
- package/dist/tools/output-schemas.d.ts +217 -0
- package/dist/tools/output-schemas.d.ts.map +1 -0
- package/dist/tools/output-schemas.js +300 -0
- package/dist/tools/output-schemas.js.map +1 -0
- package/dist/tools/process.provider.d.ts +22 -0
- package/dist/tools/process.provider.d.ts.map +1 -0
- package/dist/tools/process.provider.js +146 -0
- package/dist/tools/process.provider.js.map +1 -0
- package/dist/tools/registry.d.ts +12 -0
- package/dist/tools/registry.d.ts.map +1 -0
- package/dist/tools/registry.js +163 -0
- package/dist/tools/registry.js.map +1 -0
- package/dist/tools/results.d.ts +4 -0
- package/dist/tools/results.d.ts.map +1 -0
- package/dist/tools/results.js +5 -0
- package/dist/tools/results.js.map +1 -0
- package/dist/tools/session.provider.d.ts +23 -0
- package/dist/tools/session.provider.d.ts.map +1 -0
- package/dist/tools/session.provider.js +299 -0
- package/dist/tools/session.provider.js.map +1 -0
- package/dist/tools/system.provider.d.ts +18 -0
- package/dist/tools/system.provider.d.ts.map +1 -0
- package/dist/tools/system.provider.js +81 -0
- package/dist/tools/system.provider.js.map +1 -0
- package/dist/tools/transfer.provider.d.ts +16 -0
- package/dist/tools/transfer.provider.d.ts.map +1 -0
- package/dist/tools/transfer.provider.js +85 -0
- package/dist/tools/transfer.provider.js.map +1 -0
- package/dist/tools/tunnel.provider.d.ts +18 -0
- package/dist/tools/tunnel.provider.d.ts.map +1 -0
- package/dist/tools/tunnel.provider.js +142 -0
- package/dist/tools/tunnel.provider.js.map +1 -0
- package/dist/tools/types.d.ts +16 -0
- package/dist/tools/types.d.ts.map +1 -0
- package/dist/tools/types.js +2 -0
- package/dist/tools/types.js.map +1 -0
- package/dist/transfer.d.ts +40 -0
- package/dist/transfer.d.ts.map +1 -0
- package/dist/transfer.js +363 -0
- package/dist/transfer.js.map +1 -0
- package/dist/tunnel.d.ts +37 -0
- package/dist/tunnel.d.ts.map +1 -0
- package/dist/tunnel.js +234 -0
- package/dist/tunnel.js.map +1 -0
- package/dist/types.d.ts +341 -0
- package/dist/types.d.ts.map +1 -0
- package/dist/types.js +184 -0
- package/dist/types.js.map +1 -0
- package/docs/docker.md +22 -0
- package/examples/README.md +77 -0
- package/mcp.json +21 -0
- package/package.json +147 -0
- package/registry/ssh-mcp-pro/mcp.json +21 -0
- package/server.json +76 -0
|
@@ -0,0 +1,163 @@
|
|
|
1
|
+
import { filterToolsForProfile, isToolAllowedForProfile, } from "../connector-profile.js";
|
|
2
|
+
import { logger } from "../logging.js";
|
|
3
|
+
const TOOL_ALIASES = {
|
|
4
|
+
"ssh.openSession": "ssh_open_session",
|
|
5
|
+
"ssh.closeSession": "ssh_close_session",
|
|
6
|
+
"proc.exec": "proc_exec",
|
|
7
|
+
"proc.sudo": "proc_sudo",
|
|
8
|
+
"fs.read": "fs_read",
|
|
9
|
+
"fs.write": "fs_write",
|
|
10
|
+
"fs.stat": "fs_stat",
|
|
11
|
+
"fs.list": "fs_list",
|
|
12
|
+
"fs.mkdirp": "fs_mkdirp",
|
|
13
|
+
"fs.rmrf": "fs_rmrf",
|
|
14
|
+
"fs.rename": "fs_rename",
|
|
15
|
+
"ensure.package": "ensure_package",
|
|
16
|
+
"ensure.service": "ensure_service",
|
|
17
|
+
"ensure.linesInFile": "ensure_lines_in_file",
|
|
18
|
+
"patch.apply": "patch_apply",
|
|
19
|
+
"os.detect": "os_detect",
|
|
20
|
+
"ssh.listSessions": "ssh_list_sessions",
|
|
21
|
+
"ssh.ping": "ssh_ping",
|
|
22
|
+
"ssh.listConfiguredHosts": "ssh_list_configured_hosts",
|
|
23
|
+
"ssh.resolveHost": "ssh_resolve_host",
|
|
24
|
+
"connector.status": "connector_status",
|
|
25
|
+
"ssh.hostsList": "ssh_hosts_list",
|
|
26
|
+
"ssh.policyExplain": "ssh_policy_explain",
|
|
27
|
+
"ssh.hostInspect": "ssh_host_inspect",
|
|
28
|
+
"ssh.mutationPlan": "ssh_mutation_plan",
|
|
29
|
+
};
|
|
30
|
+
function optionalString(value) {
|
|
31
|
+
return typeof value === "string" && value.length > 0 ? value : undefined;
|
|
32
|
+
}
|
|
33
|
+
function toToolErrorResponse(payload, fallbackCode = "ETOOL") {
|
|
34
|
+
const response = {
|
|
35
|
+
error: true,
|
|
36
|
+
code: optionalString(payload.code) ?? fallbackCode,
|
|
37
|
+
message: optionalString(payload.message) ?? "Tool call failed.",
|
|
38
|
+
};
|
|
39
|
+
const hint = optionalString(payload.hint);
|
|
40
|
+
if (hint !== undefined) {
|
|
41
|
+
response.hint = hint;
|
|
42
|
+
}
|
|
43
|
+
if (typeof payload.recoverable === "boolean") {
|
|
44
|
+
response.recoverable = payload.recoverable;
|
|
45
|
+
}
|
|
46
|
+
const suggestedAction = optionalString(payload.suggestedAction);
|
|
47
|
+
if (suggestedAction !== undefined) {
|
|
48
|
+
response.suggestedAction = suggestedAction;
|
|
49
|
+
}
|
|
50
|
+
return response;
|
|
51
|
+
}
|
|
52
|
+
function errorResult(payload, fallbackCode) {
|
|
53
|
+
const structuredContent = toToolErrorResponse(payload, fallbackCode);
|
|
54
|
+
return {
|
|
55
|
+
content: [
|
|
56
|
+
{
|
|
57
|
+
type: "text",
|
|
58
|
+
text: JSON.stringify(structuredContent, null, 2),
|
|
59
|
+
},
|
|
60
|
+
],
|
|
61
|
+
structuredContent,
|
|
62
|
+
isError: true,
|
|
63
|
+
};
|
|
64
|
+
}
|
|
65
|
+
function isRecord(value) {
|
|
66
|
+
return value !== null && typeof value === "object" && !Array.isArray(value);
|
|
67
|
+
}
|
|
68
|
+
function isToolCallResult(value) {
|
|
69
|
+
return isRecord(value) && "content" in value && Array.isArray(value.content);
|
|
70
|
+
}
|
|
71
|
+
function normalizeToolCallResult(result) {
|
|
72
|
+
if (!isRecord(result.structuredContent)) {
|
|
73
|
+
return errorResult({
|
|
74
|
+
code: "ESTRUCTUREDCONTENT",
|
|
75
|
+
message: "Tool handlers returning CallToolResult must include structuredContent.",
|
|
76
|
+
});
|
|
77
|
+
}
|
|
78
|
+
if (result.content.length === 0) {
|
|
79
|
+
return errorResult({
|
|
80
|
+
code: "ETOOLCONTENT",
|
|
81
|
+
message: "Tool handlers returning CallToolResult must include at least one content item.",
|
|
82
|
+
});
|
|
83
|
+
}
|
|
84
|
+
return result;
|
|
85
|
+
}
|
|
86
|
+
function normalizeDataResult(data) {
|
|
87
|
+
if (!isRecord(data)) {
|
|
88
|
+
return errorResult({
|
|
89
|
+
code: "ESTRUCTUREDCONTENT",
|
|
90
|
+
message: "Tool handlers must return a structured JSON object or CallToolResult.",
|
|
91
|
+
});
|
|
92
|
+
}
|
|
93
|
+
return {
|
|
94
|
+
content: [{ type: "text", text: JSON.stringify(data, null, 2) }],
|
|
95
|
+
structuredContent: data,
|
|
96
|
+
};
|
|
97
|
+
}
|
|
98
|
+
export class ToolRegistry {
|
|
99
|
+
toolProfile;
|
|
100
|
+
providers = new Map();
|
|
101
|
+
constructor(toolProfile = "full") {
|
|
102
|
+
this.toolProfile = toolProfile;
|
|
103
|
+
}
|
|
104
|
+
register(provider) {
|
|
105
|
+
if (this.providers.has(provider.namespace)) {
|
|
106
|
+
throw new Error(`ToolRegistry: namespace "${provider.namespace}" is already registered`);
|
|
107
|
+
}
|
|
108
|
+
this.providers.set(provider.namespace, provider);
|
|
109
|
+
logger.debug("ToolRegistry: registered provider", {
|
|
110
|
+
namespace: provider.namespace,
|
|
111
|
+
});
|
|
112
|
+
return this;
|
|
113
|
+
}
|
|
114
|
+
getAllTools() {
|
|
115
|
+
const tools = [];
|
|
116
|
+
for (const provider of this.providers.values()) {
|
|
117
|
+
tools.push(...provider.getTools().map((tool) => ({
|
|
118
|
+
...tool,
|
|
119
|
+
title: tool.title ?? tool.annotations?.title,
|
|
120
|
+
})));
|
|
121
|
+
}
|
|
122
|
+
return filterToolsForProfile(tools, this.toolProfile);
|
|
123
|
+
}
|
|
124
|
+
async dispatch(rawToolName, args) {
|
|
125
|
+
const toolName = TOOL_ALIASES[rawToolName] ?? rawToolName;
|
|
126
|
+
if (!isToolAllowedForProfile(toolName, this.toolProfile)) {
|
|
127
|
+
return errorResult({
|
|
128
|
+
code: "ETOOLPROFILE",
|
|
129
|
+
message: `Tool ${toolName} is not exposed by the ${this.toolProfile} connector profile`,
|
|
130
|
+
});
|
|
131
|
+
}
|
|
132
|
+
for (const provider of this.providers.values()) {
|
|
133
|
+
const result = provider.handleTool(toolName, args);
|
|
134
|
+
if (result === undefined) {
|
|
135
|
+
continue;
|
|
136
|
+
}
|
|
137
|
+
try {
|
|
138
|
+
const data = await result;
|
|
139
|
+
if (isToolCallResult(data)) {
|
|
140
|
+
return normalizeToolCallResult(data);
|
|
141
|
+
}
|
|
142
|
+
return normalizeDataResult(data);
|
|
143
|
+
}
|
|
144
|
+
catch (error) {
|
|
145
|
+
logger.error("Tool handler error", {
|
|
146
|
+
toolName,
|
|
147
|
+
error: error instanceof Error ? error.message : String(error),
|
|
148
|
+
});
|
|
149
|
+
if (error &&
|
|
150
|
+
typeof error === "object" &&
|
|
151
|
+
"toJSON" in error &&
|
|
152
|
+
typeof error.toJSON === "function") {
|
|
153
|
+
const structuredError = error.toJSON();
|
|
154
|
+
return errorResult(structuredError);
|
|
155
|
+
}
|
|
156
|
+
const message = error instanceof Error ? error.message : String(error);
|
|
157
|
+
return errorResult({ message });
|
|
158
|
+
}
|
|
159
|
+
}
|
|
160
|
+
return errorResult({ code: "ETOOLNOTFOUND", message: `Unknown tool: ${toolName}` }, "ETOOLNOTFOUND");
|
|
161
|
+
}
|
|
162
|
+
}
|
|
163
|
+
//# sourceMappingURL=registry.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"registry.js","sourceRoot":"","sources":["../../src/tools/registry.ts"],"names":[],"mappings":"AACA,OAAO,EACL,qBAAqB,EACrB,uBAAuB,GAExB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAGvC,MAAM,YAAY,GAA2B;IAC3C,iBAAiB,EAAE,kBAAkB;IACrC,kBAAkB,EAAE,mBAAmB;IACvC,WAAW,EAAE,WAAW;IACxB,WAAW,EAAE,WAAW;IACxB,SAAS,EAAE,SAAS;IACpB,UAAU,EAAE,UAAU;IACtB,SAAS,EAAE,SAAS;IACpB,SAAS,EAAE,SAAS;IACpB,WAAW,EAAE,WAAW;IACxB,SAAS,EAAE,SAAS;IACpB,WAAW,EAAE,WAAW;IACxB,gBAAgB,EAAE,gBAAgB;IAClC,gBAAgB,EAAE,gBAAgB;IAClC,oBAAoB,EAAE,sBAAsB;IAC5C,aAAa,EAAE,aAAa;IAC5B,WAAW,EAAE,WAAW;IACxB,kBAAkB,EAAE,mBAAmB;IACvC,UAAU,EAAE,UAAU;IACtB,yBAAyB,EAAE,2BAA2B;IACtD,iBAAiB,EAAE,kBAAkB;IACrC,kBAAkB,EAAE,kBAAkB;IACtC,eAAe,EAAE,gBAAgB;IACjC,mBAAmB,EAAE,oBAAoB;IACzC,iBAAiB,EAAE,kBAAkB;IACrC,kBAAkB,EAAE,mBAAmB;CACxC,CAAC;AAEF,SAAS,cAAc,CAAC,KAAc;IACpC,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC;AAC3E,CAAC;AAED,SAAS,mBAAmB,CAC1B,OAAgC,EAChC,YAAY,GAAG,OAAO;IAEtB,MAAM,QAAQ,GAAsB;QAClC,KAAK,EAAE,IAAI;QACX,IAAI,EAAE,cAAc,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,YAAY;QAClD,OAAO,EAAE,cAAc,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,mBAAmB;KAChE,CAAC;IACF,MAAM,IAAI,GAAG,cAAc,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IAC1C,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;QACvB,QAAQ,CAAC,IAAI,GAAG,IAAI,CAAC;IACvB,CAAC;IACD,IAAI,OAAO,OAAO,CAAC,WAAW,KAAK,SAAS,EAAE,CAAC;QAC7C,QAAQ,CAAC,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC;IAC7C,CAAC;IACD,MAAM,eAAe,GAAG,cAAc,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;IAChE,IAAI,eAAe,KAAK,SAAS,EAAE,CAAC;QAClC,QAAQ,CAAC,eAAe,GAAG,eAAe,CAAC;IAC7C,CAAC;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,SAAS,WAAW,CAAC,OAAgC,EAAE,YAAqB;IAC1E,MAAM,iBAAiB,GAAG,mBAAmB,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC;IACrE,OAAO;QACL,OAAO,EAAE;YACP;gBACE,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,iBAAiB,EAAE,IAAI,EAAE,CAAC,CAAC;aACjD;SACF;QACD,iBAAiB;QACjB,OAAO,EAAE,IAAI;KACd,CAAC;AACJ,CAAC;AAED,SAAS,QAAQ,CAAC,KAAc;IAC9B,OAAO,KAAK,KAAK,IAAI,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;AAC9E,CAAC;AAED,SAAS,gBAAgB,CAAC,KAAc;IACtC,OAAO,QAAQ,CAAC,KAAK,CAAC,IAAI,SAAS,IAAI,KAAK,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;AAC/E,CAAC;AAED,SAAS,uBAAuB,CAAC,MAAsB;IACrD,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,iBAAiB,CAAC,EAAE,CAAC;QACxC,OAAO,WAAW,CAAC;YACjB,IAAI,EAAE,oBAAoB;YAC1B,OAAO,EAAE,wEAAwE;SAClF,CAAC,CAAC;IACL,CAAC;IAED,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAChC,OAAO,WAAW,CAAC;YACjB,IAAI,EAAE,cAAc;YACpB,OAAO,EAAE,gFAAgF;SAC1F,CAAC,CAAC;IACL,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,mBAAmB,CAAC,IAAa;IACxC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;QACpB,OAAO,WAAW,CAAC;YACjB,IAAI,EAAE,oBAAoB;YAC1B,OAAO,EAAE,uEAAuE;SACjF,CAAC,CAAC;IACL,CAAC;IAED,OAAO;QACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC;QAChE,iBAAiB,EAAE,IAAI;KACxB,CAAC;AACJ,CAAC;AAED,MAAM,OAAO,YAAY;IAGM;IAFZ,SAAS,GAAG,IAAI,GAAG,EAAwB,CAAC;IAE7D,YAA6B,cAA2B,MAAM;QAAjC,gBAAW,GAAX,WAAW,CAAsB;IAAG,CAAC;IAElE,QAAQ,CAAC,QAAsB;QAC7B,IAAI,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;YAC3C,MAAM,IAAI,KAAK,CAAC,4BAA4B,QAAQ,CAAC,SAAS,yBAAyB,CAAC,CAAC;QAC3F,CAAC;QAED,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;QACjD,MAAM,CAAC,KAAK,CAAC,mCAAmC,EAAE;YAChD,SAAS,EAAE,QAAQ,CAAC,SAAS;SAC9B,CAAC,CAAC;QACH,OAAO,IAAI,CAAC;IACd,CAAC;IAED,WAAW;QACT,MAAM,KAAK,GAAW,EAAE,CAAC;QACzB,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,EAAE,CAAC;YAC/C,KAAK,CAAC,IAAI,CACR,GAAG,QAAQ,CAAC,QAAQ,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;gBACpC,GAAG,IAAI;gBACP,KAAK,EAAE,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,WAAW,EAAE,KAAK;aAC7C,CAAC,CAAC,CACJ,CAAC;QACJ,CAAC;QACD,OAAO,qBAAqB,CAAC,KAAK,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;IACxD,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,WAAmB,EAAE,IAAa;QAC/C,MAAM,QAAQ,GAAG,YAAY,CAAC,WAAW,CAAC,IAAI,WAAW,CAAC;QAC1D,IAAI,CAAC,uBAAuB,CAAC,QAAQ,EAAE,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC;YACzD,OAAO,WAAW,CAAC;gBACjB,IAAI,EAAE,cAAc;gBACpB,OAAO,EAAE,QAAQ,QAAQ,0BAA0B,IAAI,CAAC,WAAW,oBAAoB;aACxF,CAAC,CAAC;QACL,CAAC;QAED,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,EAAE,CAAC;YAC/C,MAAM,MAAM,GAAG,QAAQ,CAAC,UAAU,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;YACnD,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;gBACzB,SAAS;YACX,CAAC;YAED,IAAI,CAAC;gBACH,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC;gBAC1B,IAAI,gBAAgB,CAAC,IAAI,CAAC,EAAE,CAAC;oBAC3B,OAAO,uBAAuB,CAAC,IAAI,CAAC,CAAC;gBACvC,CAAC;gBACD,OAAO,mBAAmB,CAAC,IAAI,CAAC,CAAC;YACnC,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,CAAC,KAAK,CAAC,oBAAoB,EAAE;oBACjC,QAAQ;oBACR,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;iBAC9D,CAAC,CAAC;gBAEH,IACE,KAAK;oBACL,OAAO,KAAK,KAAK,QAAQ;oBACzB,QAAQ,IAAI,KAAK;oBACjB,OAAO,KAAK,CAAC,MAAM,KAAK,UAAU,EAClC,CAAC;oBACD,MAAM,eAAe,GAAG,KAAK,CAAC,MAAM,EAA6B,CAAC;oBAClE,OAAO,WAAW,CAAC,eAAe,CAAC,CAAC;gBACtC,CAAC;gBAED,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBACvE,OAAO,WAAW,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC;YAClC,CAAC;QACH,CAAC;QAED,OAAO,WAAW,CAChB,EAAE,IAAI,EAAE,eAAe,EAAE,OAAO,EAAE,iBAAiB,QAAQ,EAAE,EAAE,EAC/D,eAAe,CAChB,CAAC;IACJ,CAAC;CACF"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"results.d.ts","sourceRoot":"","sources":["../../src/tools/results.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAEjD,MAAM,MAAM,iBAAiB,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;AAExD,wBAAgB,UAAU,CACxB,iBAAiB,EAAE,iBAAiB,EACpC,IAAI,SAA6C,GAChD,cAAc,CAGhB"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"results.js","sourceRoot":"","sources":["../../src/tools/results.ts"],"names":[],"mappings":"AAKA,MAAM,UAAU,UAAU,CACxB,iBAAoC,EACpC,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,iBAAiB,EAAE,IAAI,EAAE,CAAC,CAAC;IAEjD,MAAM,OAAO,GAAkB,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;IACxD,OAAO,EAAE,OAAO,EAAE,iBAAiB,EAAE,CAAC;AACxC,CAAC"}
|
|
@@ -0,0 +1,23 @@
|
|
|
1
|
+
import type { Tool } from "@modelcontextprotocol/sdk/types.js";
|
|
2
|
+
import type { MetricsCollector } from "../metrics.js";
|
|
3
|
+
import type { SessionManager } from "../session.js";
|
|
4
|
+
import type { ToolProvider } from "./types.js";
|
|
5
|
+
export interface SessionToolProviderDeps {
|
|
6
|
+
sessionManager: SessionManager;
|
|
7
|
+
metrics: MetricsCollector;
|
|
8
|
+
}
|
|
9
|
+
export declare class SessionToolProvider implements ToolProvider {
|
|
10
|
+
private readonly deps;
|
|
11
|
+
readonly namespace = "session";
|
|
12
|
+
constructor(deps: SessionToolProviderDeps);
|
|
13
|
+
getTools(): Tool[];
|
|
14
|
+
handleTool(toolName: string, args: unknown): Promise<unknown> | undefined;
|
|
15
|
+
private openSession;
|
|
16
|
+
private closeSession;
|
|
17
|
+
private listSessions;
|
|
18
|
+
private ping;
|
|
19
|
+
private listConfiguredHosts;
|
|
20
|
+
private resolveHost;
|
|
21
|
+
private normalizeConnectionParams;
|
|
22
|
+
}
|
|
23
|
+
//# sourceMappingURL=session.provider.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"session.provider.d.ts","sourceRoot":"","sources":["../../src/tools/session.provider.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,oCAAoC,CAAC;AAE/D,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,eAAe,CAAC;AACtD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAYpD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAE/C,MAAM,WAAW,uBAAuB;IACtC,cAAc,EAAE,cAAc,CAAC;IAC/B,OAAO,EAAE,gBAAgB,CAAC;CAC3B;AAED,qBAAa,mBAAoB,YAAW,YAAY;IAG1C,OAAO,CAAC,QAAQ,CAAC,IAAI;IAFjC,QAAQ,CAAC,SAAS,aAAa;gBAEF,IAAI,EAAE,uBAAuB;IAE1D,QAAQ,IAAI,IAAI,EAAE;IA0KlB,UAAU,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,GAAG,SAAS;YAmB3D,WAAW;YAaX,YAAY;YAUZ,YAAY;YAkBZ,IAAI;YA8BJ,mBAAmB;YAMnB,WAAW;IAOzB,OAAO,CAAC,yBAAyB;CAyClC"}
|
|
@@ -0,0 +1,299 @@
|
|
|
1
|
+
import { logger } from "../logging.js";
|
|
2
|
+
import { getConfiguredHosts, resolveSSHHost } from "../ssh-config.js";
|
|
3
|
+
import { ConnectionParamsSchema, HostAliasSchema, SessionIdSchema } from "../types.js";
|
|
4
|
+
import { annotate } from "./metadata.js";
|
|
5
|
+
import { CONFIGURED_HOSTS_OUTPUT_SCHEMA, RESOLVED_HOST_OUTPUT_SCHEMA, SESSION_CLOSE_OUTPUT_SCHEMA, SESSION_LIST_OUTPUT_SCHEMA, SESSION_OPEN_OUTPUT_SCHEMA, SESSION_PING_OUTPUT_SCHEMA, } from "./output-schemas.js";
|
|
6
|
+
export class SessionToolProvider {
|
|
7
|
+
deps;
|
|
8
|
+
namespace = "session";
|
|
9
|
+
constructor(deps) {
|
|
10
|
+
this.deps = deps;
|
|
11
|
+
}
|
|
12
|
+
getTools() {
|
|
13
|
+
return [
|
|
14
|
+
{
|
|
15
|
+
name: "ssh_open_session",
|
|
16
|
+
description: "Opens a new SSH session with authentication",
|
|
17
|
+
annotations: annotate({
|
|
18
|
+
title: "Open SSH Session",
|
|
19
|
+
readOnly: false,
|
|
20
|
+
destructive: false,
|
|
21
|
+
idempotent: false,
|
|
22
|
+
}),
|
|
23
|
+
outputSchema: SESSION_OPEN_OUTPUT_SCHEMA,
|
|
24
|
+
inputSchema: {
|
|
25
|
+
type: "object",
|
|
26
|
+
properties: {
|
|
27
|
+
host: { type: "string", description: "SSH server hostname or IP" },
|
|
28
|
+
username: { type: "string", description: "SSH username" },
|
|
29
|
+
port: { type: "number", description: "SSH port (default: 22)" },
|
|
30
|
+
auth: {
|
|
31
|
+
type: "string",
|
|
32
|
+
enum: ["auto", "password", "key", "agent"],
|
|
33
|
+
description: "Authentication method (default: auto)",
|
|
34
|
+
},
|
|
35
|
+
password: {
|
|
36
|
+
type: "string",
|
|
37
|
+
description: "Password for authentication",
|
|
38
|
+
},
|
|
39
|
+
privateKey: {
|
|
40
|
+
type: "string",
|
|
41
|
+
description: "Inline private key content",
|
|
42
|
+
},
|
|
43
|
+
privateKeyPath: {
|
|
44
|
+
type: "string",
|
|
45
|
+
description: "Path to private key file",
|
|
46
|
+
},
|
|
47
|
+
passphrase: {
|
|
48
|
+
type: "string",
|
|
49
|
+
description: "Passphrase for encrypted private key",
|
|
50
|
+
},
|
|
51
|
+
useAgent: {
|
|
52
|
+
type: "boolean",
|
|
53
|
+
description: "Use SSH agent for authentication",
|
|
54
|
+
},
|
|
55
|
+
readyTimeoutMs: {
|
|
56
|
+
type: "number",
|
|
57
|
+
description: "Connection timeout in milliseconds (default: 20000)",
|
|
58
|
+
},
|
|
59
|
+
ttlMs: {
|
|
60
|
+
type: "number",
|
|
61
|
+
description: "Session TTL in milliseconds (default: 900000)",
|
|
62
|
+
},
|
|
63
|
+
strictHostKeyChecking: {
|
|
64
|
+
type: "boolean",
|
|
65
|
+
description: "Deprecated alias. Prefer hostKeyPolicy.",
|
|
66
|
+
},
|
|
67
|
+
hostKeyPolicy: {
|
|
68
|
+
type: "string",
|
|
69
|
+
enum: ["strict", "accept-new", "insecure"],
|
|
70
|
+
description: "SSH host key verification policy (default: strict)",
|
|
71
|
+
},
|
|
72
|
+
knownHostsPath: {
|
|
73
|
+
type: "string",
|
|
74
|
+
description: "Path to known_hosts file",
|
|
75
|
+
},
|
|
76
|
+
expectedHostKeySha256: {
|
|
77
|
+
type: "string",
|
|
78
|
+
description: "Pinned SHA-256 host key fingerprint, with or without SHA256: prefix",
|
|
79
|
+
},
|
|
80
|
+
policyMode: {
|
|
81
|
+
type: "string",
|
|
82
|
+
enum: ["enforce", "explain"],
|
|
83
|
+
description: "Use explain to return the planned connection without opening SSH",
|
|
84
|
+
},
|
|
85
|
+
},
|
|
86
|
+
required: ["host", "username"],
|
|
87
|
+
},
|
|
88
|
+
},
|
|
89
|
+
{
|
|
90
|
+
name: "ssh_close_session",
|
|
91
|
+
description: "Closes an SSH session",
|
|
92
|
+
annotations: annotate({
|
|
93
|
+
title: "Close SSH Session",
|
|
94
|
+
readOnly: false,
|
|
95
|
+
destructive: false,
|
|
96
|
+
idempotent: true,
|
|
97
|
+
}),
|
|
98
|
+
outputSchema: SESSION_CLOSE_OUTPUT_SCHEMA,
|
|
99
|
+
inputSchema: {
|
|
100
|
+
type: "object",
|
|
101
|
+
properties: {
|
|
102
|
+
sessionId: { type: "string", description: "Session ID to close" },
|
|
103
|
+
},
|
|
104
|
+
required: ["sessionId"],
|
|
105
|
+
},
|
|
106
|
+
},
|
|
107
|
+
{
|
|
108
|
+
name: "ssh_list_sessions",
|
|
109
|
+
description: "Lists all active SSH sessions with their details",
|
|
110
|
+
annotations: annotate({
|
|
111
|
+
title: "List SSH Sessions",
|
|
112
|
+
readOnly: true,
|
|
113
|
+
idempotent: true,
|
|
114
|
+
}),
|
|
115
|
+
outputSchema: SESSION_LIST_OUTPUT_SCHEMA,
|
|
116
|
+
inputSchema: {
|
|
117
|
+
type: "object",
|
|
118
|
+
properties: {},
|
|
119
|
+
required: [],
|
|
120
|
+
},
|
|
121
|
+
},
|
|
122
|
+
{
|
|
123
|
+
name: "ssh_ping",
|
|
124
|
+
description: "Checks if an SSH session is still alive and responsive",
|
|
125
|
+
annotations: annotate({
|
|
126
|
+
title: "Ping SSH Session",
|
|
127
|
+
readOnly: true,
|
|
128
|
+
idempotent: true,
|
|
129
|
+
}),
|
|
130
|
+
outputSchema: SESSION_PING_OUTPUT_SCHEMA,
|
|
131
|
+
inputSchema: {
|
|
132
|
+
type: "object",
|
|
133
|
+
properties: {
|
|
134
|
+
sessionId: {
|
|
135
|
+
type: "string",
|
|
136
|
+
description: "SSH session ID to check",
|
|
137
|
+
},
|
|
138
|
+
},
|
|
139
|
+
required: ["sessionId"],
|
|
140
|
+
},
|
|
141
|
+
},
|
|
142
|
+
{
|
|
143
|
+
name: "ssh_list_configured_hosts",
|
|
144
|
+
description: "Lists all hosts configured in ~/.ssh/config",
|
|
145
|
+
annotations: annotate({
|
|
146
|
+
title: "List Configured Hosts",
|
|
147
|
+
readOnly: true,
|
|
148
|
+
idempotent: true,
|
|
149
|
+
openWorld: false,
|
|
150
|
+
}),
|
|
151
|
+
outputSchema: CONFIGURED_HOSTS_OUTPUT_SCHEMA,
|
|
152
|
+
inputSchema: {
|
|
153
|
+
type: "object",
|
|
154
|
+
properties: {},
|
|
155
|
+
required: [],
|
|
156
|
+
},
|
|
157
|
+
},
|
|
158
|
+
{
|
|
159
|
+
name: "ssh_resolve_host",
|
|
160
|
+
description: "Resolves a host alias from ~/.ssh/config to connection parameters",
|
|
161
|
+
annotations: annotate({
|
|
162
|
+
title: "Resolve SSH Host",
|
|
163
|
+
readOnly: true,
|
|
164
|
+
idempotent: true,
|
|
165
|
+
openWorld: false,
|
|
166
|
+
}),
|
|
167
|
+
outputSchema: RESOLVED_HOST_OUTPUT_SCHEMA,
|
|
168
|
+
inputSchema: {
|
|
169
|
+
type: "object",
|
|
170
|
+
properties: {
|
|
171
|
+
hostAlias: {
|
|
172
|
+
type: "string",
|
|
173
|
+
description: "Host alias from SSH config",
|
|
174
|
+
},
|
|
175
|
+
},
|
|
176
|
+
required: ["hostAlias"],
|
|
177
|
+
},
|
|
178
|
+
},
|
|
179
|
+
];
|
|
180
|
+
}
|
|
181
|
+
handleTool(toolName, args) {
|
|
182
|
+
switch (toolName) {
|
|
183
|
+
case "ssh_open_session":
|
|
184
|
+
return this.openSession(args);
|
|
185
|
+
case "ssh_close_session":
|
|
186
|
+
return this.closeSession(args);
|
|
187
|
+
case "ssh_list_sessions":
|
|
188
|
+
return this.listSessions();
|
|
189
|
+
case "ssh_ping":
|
|
190
|
+
return this.ping(args);
|
|
191
|
+
case "ssh_list_configured_hosts":
|
|
192
|
+
return this.listConfiguredHosts();
|
|
193
|
+
case "ssh_resolve_host":
|
|
194
|
+
return this.resolveHost(args);
|
|
195
|
+
default:
|
|
196
|
+
return undefined;
|
|
197
|
+
}
|
|
198
|
+
}
|
|
199
|
+
async openSession(args) {
|
|
200
|
+
const params = ConnectionParamsSchema.parse(args);
|
|
201
|
+
const result = await this.deps.sessionManager.openSession(this.normalizeConnectionParams(params));
|
|
202
|
+
this.deps.metrics.recordSessionCreated();
|
|
203
|
+
logger.info("SSH session opened", {
|
|
204
|
+
sessionId: result.sessionId,
|
|
205
|
+
host: params.host,
|
|
206
|
+
});
|
|
207
|
+
return result;
|
|
208
|
+
}
|
|
209
|
+
async closeSession(args) {
|
|
210
|
+
const { sessionId } = SessionIdSchema.parse(args);
|
|
211
|
+
const result = await this.deps.sessionManager.closeSession(sessionId);
|
|
212
|
+
if (result) {
|
|
213
|
+
this.deps.metrics.recordSessionClosed();
|
|
214
|
+
}
|
|
215
|
+
logger.info("SSH session closed", { sessionId });
|
|
216
|
+
return { closed: result };
|
|
217
|
+
}
|
|
218
|
+
async listSessions() {
|
|
219
|
+
const sessions = this.deps.sessionManager.getActiveSessions();
|
|
220
|
+
logger.info("Sessions listed", { count: sessions.length });
|
|
221
|
+
return {
|
|
222
|
+
count: sessions.length,
|
|
223
|
+
sessions: sessions.map((session) => ({
|
|
224
|
+
sessionId: session.sessionId,
|
|
225
|
+
host: session.host,
|
|
226
|
+
username: session.username,
|
|
227
|
+
port: session.port,
|
|
228
|
+
createdAt: new Date(session.createdAt).toISOString(),
|
|
229
|
+
expiresAt: new Date(session.expiresAt).toISOString(),
|
|
230
|
+
lastUsed: new Date(session.lastUsed).toISOString(),
|
|
231
|
+
remainingMs: Math.max(0, session.expiresAt - Date.now()),
|
|
232
|
+
})),
|
|
233
|
+
};
|
|
234
|
+
}
|
|
235
|
+
async ping(args) {
|
|
236
|
+
const { sessionId } = SessionIdSchema.parse(args);
|
|
237
|
+
const session = this.deps.sessionManager.getSession(sessionId);
|
|
238
|
+
if (!session) {
|
|
239
|
+
return { alive: false, error: "Session not found or expired" };
|
|
240
|
+
}
|
|
241
|
+
try {
|
|
242
|
+
const startTime = Date.now();
|
|
243
|
+
const pingResult = await session.ssh.execCommand("echo pong");
|
|
244
|
+
const latencyMs = Date.now() - startTime;
|
|
245
|
+
const result = {
|
|
246
|
+
alive: (pingResult.code ?? 0) === 0,
|
|
247
|
+
latencyMs,
|
|
248
|
+
sessionId,
|
|
249
|
+
host: session.info.host,
|
|
250
|
+
remainingMs: Math.max(0, session.info.expiresAt - Date.now()),
|
|
251
|
+
};
|
|
252
|
+
logger.info("Session ping", {
|
|
253
|
+
sessionId,
|
|
254
|
+
alive: result.alive,
|
|
255
|
+
latencyMs,
|
|
256
|
+
});
|
|
257
|
+
return result;
|
|
258
|
+
}
|
|
259
|
+
catch {
|
|
260
|
+
return { alive: false, error: "Connection test failed" };
|
|
261
|
+
}
|
|
262
|
+
}
|
|
263
|
+
async listConfiguredHosts() {
|
|
264
|
+
const hosts = await getConfiguredHosts();
|
|
265
|
+
logger.info("Configured hosts listed", { count: hosts.length });
|
|
266
|
+
return { count: hosts.length, hosts };
|
|
267
|
+
}
|
|
268
|
+
async resolveHost(args) {
|
|
269
|
+
const { hostAlias } = HostAliasSchema.parse(args);
|
|
270
|
+
const resolved = await resolveSSHHost(hostAlias);
|
|
271
|
+
logger.info("Host resolved", { hostAlias, resolved: resolved.host });
|
|
272
|
+
return resolved;
|
|
273
|
+
}
|
|
274
|
+
normalizeConnectionParams(params) {
|
|
275
|
+
return {
|
|
276
|
+
host: params.host,
|
|
277
|
+
username: params.username,
|
|
278
|
+
auth: params.auth,
|
|
279
|
+
readyTimeoutMs: params.readyTimeoutMs,
|
|
280
|
+
ttlMs: params.ttlMs,
|
|
281
|
+
policyMode: params.policyMode,
|
|
282
|
+
...(params.port !== undefined ? { port: params.port } : {}),
|
|
283
|
+
...(params.strictHostKeyChecking !== undefined
|
|
284
|
+
? { strictHostKeyChecking: params.strictHostKeyChecking }
|
|
285
|
+
: {}),
|
|
286
|
+
...(params.hostKeyPolicy ? { hostKeyPolicy: params.hostKeyPolicy } : {}),
|
|
287
|
+
...(params.password ? { password: params.password } : {}),
|
|
288
|
+
...(params.privateKey ? { privateKey: params.privateKey } : {}),
|
|
289
|
+
...(params.privateKeyPath ? { privateKeyPath: params.privateKeyPath } : {}),
|
|
290
|
+
...(params.passphrase ? { passphrase: params.passphrase } : {}),
|
|
291
|
+
...(params.useAgent !== undefined ? { useAgent: params.useAgent } : {}),
|
|
292
|
+
...(params.knownHostsPath ? { knownHostsPath: params.knownHostsPath } : {}),
|
|
293
|
+
...(params.expectedHostKeySha256
|
|
294
|
+
? { expectedHostKeySha256: params.expectedHostKeySha256 }
|
|
295
|
+
: {}),
|
|
296
|
+
};
|
|
297
|
+
}
|
|
298
|
+
}
|
|
299
|
+
//# sourceMappingURL=session.provider.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"session.provider.js","sourceRoot":"","sources":["../../src/tools/session.provider.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAGvC,OAAO,EAAE,kBAAkB,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AACtE,OAAO,EAAE,sBAAsB,EAAE,eAAe,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AACvF,OAAO,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AACzC,OAAO,EACL,8BAA8B,EAC9B,2BAA2B,EAC3B,2BAA2B,EAC3B,0BAA0B,EAC1B,0BAA0B,EAC1B,0BAA0B,GAC3B,MAAM,qBAAqB,CAAC;AAQ7B,MAAM,OAAO,mBAAmB;IAGD;IAFpB,SAAS,GAAG,SAAS,CAAC;IAE/B,YAA6B,IAA6B;QAA7B,SAAI,GAAJ,IAAI,CAAyB;IAAG,CAAC;IAE9D,QAAQ;QACN,OAAO;YACL;gBACE,IAAI,EAAE,kBAAkB;gBACxB,WAAW,EAAE,6CAA6C;gBAC1D,WAAW,EAAE,QAAQ,CAAC;oBACpB,KAAK,EAAE,kBAAkB;oBACzB,QAAQ,EAAE,KAAK;oBACf,WAAW,EAAE,KAAK;oBAClB,UAAU,EAAE,KAAK;iBAClB,CAAC;gBACF,YAAY,EAAE,0BAA0B;gBACxC,WAAW,EAAE;oBACX,IAAI,EAAE,QAAiB;oBACvB,UAAU,EAAE;wBACV,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,2BAA2B,EAAE;wBAClE,QAAQ,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,cAAc,EAAE;wBACzD,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,wBAAwB,EAAE;wBAC/D,IAAI,EAAE;4BACJ,IAAI,EAAE,QAAQ;4BACd,IAAI,EAAE,CAAC,MAAM,EAAE,UAAU,EAAE,KAAK,EAAE,OAAO,CAAC;4BAC1C,WAAW,EAAE,uCAAuC;yBACrD;wBACD,QAAQ,EAAE;4BACR,IAAI,EAAE,QAAQ;4BACd,WAAW,EAAE,6BAA6B;yBAC3C;wBACD,UAAU,EAAE;4BACV,IAAI,EAAE,QAAQ;4BACd,WAAW,EAAE,4BAA4B;yBAC1C;wBACD,cAAc,EAAE;4BACd,IAAI,EAAE,QAAQ;4BACd,WAAW,EAAE,0BAA0B;yBACxC;wBACD,UAAU,EAAE;4BACV,IAAI,EAAE,QAAQ;4BACd,WAAW,EAAE,sCAAsC;yBACpD;wBACD,QAAQ,EAAE;4BACR,IAAI,EAAE,SAAS;4BACf,WAAW,EAAE,kCAAkC;yBAChD;wBACD,cAAc,EAAE;4BACd,IAAI,EAAE,QAAQ;4BACd,WAAW,EAAE,qDAAqD;yBACnE;wBACD,KAAK,EAAE;4BACL,IAAI,EAAE,QAAQ;4BACd,WAAW,EAAE,+CAA+C;yBAC7D;wBACD,qBAAqB,EAAE;4BACrB,IAAI,EAAE,SAAS;4BACf,WAAW,EAAE,yCAAyC;yBACvD;wBACD,aAAa,EAAE;4BACb,IAAI,EAAE,QAAQ;4BACd,IAAI,EAAE,CAAC,QAAQ,EAAE,YAAY,EAAE,UAAU,CAAC;4BAC1C,WAAW,EAAE,oDAAoD;yBAClE;wBACD,cAAc,EAAE;4BACd,IAAI,EAAE,QAAQ;4BACd,WAAW,EAAE,0BAA0B;yBACxC;wBACD,qBAAqB,EAAE;4BACrB,IAAI,EAAE,QAAQ;4BACd,WAAW,EAAE,qEAAqE;yBACnF;wBACD,UAAU,EAAE;4BACV,IAAI,EAAE,QAAQ;4BACd,IAAI,EAAE,CAAC,SAAS,EAAE,SAAS,CAAC;4BAC5B,WAAW,EAAE,kEAAkE;yBAChF;qBACF;oBACD,QAAQ,EAAE,CAAC,MAAM,EAAE,UAAU,CAAC;iBAC/B;aACF;YACD;gBACE,IAAI,EAAE,mBAAmB;gBACzB,WAAW,EAAE,uBAAuB;gBACpC,WAAW,EAAE,QAAQ,CAAC;oBACpB,KAAK,EAAE,mBAAmB;oBAC1B,QAAQ,EAAE,KAAK;oBACf,WAAW,EAAE,KAAK;oBAClB,UAAU,EAAE,IAAI;iBACjB,CAAC;gBACF,YAAY,EAAE,2BAA2B;gBACzC,WAAW,EAAE;oBACX,IAAI,EAAE,QAAiB;oBACvB,UAAU,EAAE;wBACV,SAAS,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,qBAAqB,EAAE;qBAClE;oBACD,QAAQ,EAAE,CAAC,WAAW,CAAC;iBACxB;aACF;YACD;gBACE,IAAI,EAAE,mBAAmB;gBACzB,WAAW,EAAE,kDAAkD;gBAC/D,WAAW,EAAE,QAAQ,CAAC;oBACpB,KAAK,EAAE,mBAAmB;oBAC1B,QAAQ,EAAE,IAAI;oBACd,UAAU,EAAE,IAAI;iBACjB,CAAC;gBACF,YAAY,EAAE,0BAA0B;gBACxC,WAAW,EAAE;oBACX,IAAI,EAAE,QAAiB;oBACvB,UAAU,EAAE,EAAE;oBACd,QAAQ,EAAE,EAAE;iBACb;aACF;YACD;gBACE,IAAI,EAAE,UAAU;gBAChB,WAAW,EAAE,wDAAwD;gBACrE,WAAW,EAAE,QAAQ,CAAC;oBACpB,KAAK,EAAE,kBAAkB;oBACzB,QAAQ,EAAE,IAAI;oBACd,UAAU,EAAE,IAAI;iBACjB,CAAC;gBACF,YAAY,EAAE,0BAA0B;gBACxC,WAAW,EAAE;oBACX,IAAI,EAAE,QAAiB;oBACvB,UAAU,EAAE;wBACV,SAAS,EAAE;4BACT,IAAI,EAAE,QAAQ;4BACd,WAAW,EAAE,yBAAyB;yBACvC;qBACF;oBACD,QAAQ,EAAE,CAAC,WAAW,CAAC;iBACxB;aACF;YACD;gBACE,IAAI,EAAE,2BAA2B;gBACjC,WAAW,EAAE,6CAA6C;gBAC1D,WAAW,EAAE,QAAQ,CAAC;oBACpB,KAAK,EAAE,uBAAuB;oBAC9B,QAAQ,EAAE,IAAI;oBACd,UAAU,EAAE,IAAI;oBAChB,SAAS,EAAE,KAAK;iBACjB,CAAC;gBACF,YAAY,EAAE,8BAA8B;gBAC5C,WAAW,EAAE;oBACX,IAAI,EAAE,QAAiB;oBACvB,UAAU,EAAE,EAAE;oBACd,QAAQ,EAAE,EAAE;iBACb;aACF;YACD;gBACE,IAAI,EAAE,kBAAkB;gBACxB,WAAW,EAAE,mEAAmE;gBAChF,WAAW,EAAE,QAAQ,CAAC;oBACpB,KAAK,EAAE,kBAAkB;oBACzB,QAAQ,EAAE,IAAI;oBACd,UAAU,EAAE,IAAI;oBAChB,SAAS,EAAE,KAAK;iBACjB,CAAC;gBACF,YAAY,EAAE,2BAA2B;gBACzC,WAAW,EAAE;oBACX,IAAI,EAAE,QAAiB;oBACvB,UAAU,EAAE;wBACV,SAAS,EAAE;4BACT,IAAI,EAAE,QAAQ;4BACd,WAAW,EAAE,4BAA4B;yBAC1C;qBACF;oBACD,QAAQ,EAAE,CAAC,WAAW,CAAC;iBACxB;aACF;SACF,CAAC;IACJ,CAAC;IAED,UAAU,CAAC,QAAgB,EAAE,IAAa;QACxC,QAAQ,QAAQ,EAAE,CAAC;YACjB,KAAK,kBAAkB;gBACrB,OAAO,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;YAChC,KAAK,mBAAmB;gBACtB,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;YACjC,KAAK,mBAAmB;gBACtB,OAAO,IAAI,CAAC,YAAY,EAAE,CAAC;YAC7B,KAAK,UAAU;gBACb,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACzB,KAAK,2BAA2B;gBAC9B,OAAO,IAAI,CAAC,mBAAmB,EAAE,CAAC;YACpC,KAAK,kBAAkB;gBACrB,OAAO,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;YAChC;gBACE,OAAO,SAAS,CAAC;QACrB,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,WAAW,CAAC,IAAa;QACrC,MAAM,MAAM,GAAG,sBAAsB,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAClD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,WAAW,CACvD,IAAI,CAAC,yBAAyB,CAAC,MAAM,CAAC,CACvC,CAAC;QACF,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,oBAAoB,EAAE,CAAC;QACzC,MAAM,CAAC,IAAI,CAAC,oBAAoB,EAAE;YAChC,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,IAAI,EAAE,MAAM,CAAC,IAAI;SAClB,CAAC,CAAC;QACH,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,KAAK,CAAC,YAAY,CAAC,IAAa;QACtC,MAAM,EAAE,SAAS,EAAE,GAAG,eAAe,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAClD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;QACtE,IAAI,MAAM,EAAE,CAAC;YACX,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,mBAAmB,EAAE,CAAC;QAC1C,CAAC;QACD,MAAM,CAAC,IAAI,CAAC,oBAAoB,EAAE,EAAE,SAAS,EAAE,CAAC,CAAC;QACjD,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC;IAC5B,CAAC;IAEO,KAAK,CAAC,YAAY;QACxB,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,iBAAiB,EAAE,CAAC;QAC9D,MAAM,CAAC,IAAI,CAAC,iBAAiB,EAAE,EAAE,KAAK,EAAE,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;QAC3D,OAAO;YACL,KAAK,EAAE,QAAQ,CAAC,MAAM;YACtB,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;gBACnC,SAAS,EAAE,OAAO,CAAC,SAAS;gBAC5B,IAAI,EAAE,OAAO,CAAC,IAAI;gBAClB,QAAQ,EAAE,OAAO,CAAC,QAAQ;gBAC1B,IAAI,EAAE,OAAO,CAAC,IAAI;gBAClB,SAAS,EAAE,IAAI,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,WAAW,EAAE;gBACpD,SAAS,EAAE,IAAI,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,WAAW,EAAE;gBACpD,QAAQ,EAAE,IAAI,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE;gBAClD,WAAW,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;aACzD,CAAC,CAAC;SACJ,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,IAAI,CAAC,IAAa;QAC9B,MAAM,EAAE,SAAS,EAAE,GAAG,eAAe,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAClD,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;QAC/D,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,8BAA8B,EAAE,CAAC;QACjE,CAAC;QAED,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC7B,MAAM,UAAU,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC;YAC9D,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YACzC,MAAM,MAAM,GAAG;gBACb,KAAK,EAAE,CAAC,UAAU,CAAC,IAAI,IAAI,CAAC,CAAC,KAAK,CAAC;gBACnC,SAAS;gBACT,SAAS;gBACT,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,IAAI;gBACvB,WAAW,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;aAC9D,CAAC;YAEF,MAAM,CAAC,IAAI,CAAC,cAAc,EAAE;gBAC1B,SAAS;gBACT,KAAK,EAAE,MAAM,CAAC,KAAK;gBACnB,SAAS;aACV,CAAC,CAAC;YACH,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,wBAAwB,EAAE,CAAC;QAC3D,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,mBAAmB;QAC/B,MAAM,KAAK,GAAG,MAAM,kBAAkB,EAAE,CAAC;QACzC,MAAM,CAAC,IAAI,CAAC,yBAAyB,EAAE,EAAE,KAAK,EAAE,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;QAChE,OAAO,EAAE,KAAK,EAAE,KAAK,CAAC,MAAM,EAAE,KAAK,EAAE,CAAC;IACxC,CAAC;IAEO,KAAK,CAAC,WAAW,CAAC,IAAa;QACrC,MAAM,EAAE,SAAS,EAAE,GAAG,eAAe,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAClD,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,SAAS,CAAC,CAAC;QACjD,MAAM,CAAC,IAAI,CAAC,eAAe,EAAE,EAAE,SAAS,EAAE,QAAQ,EAAE,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC;QACrE,OAAO,QAAQ,CAAC;IAClB,CAAC;IAEO,yBAAyB,CAAC,MAiBjC;QACC,OAAO;YACL,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,cAAc,EAAE,MAAM,CAAC,cAAc;YACrC,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,UAAU,EAAE,MAAM,CAAC,UAAU;YAC7B,GAAG,CAAC,MAAM,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC3D,GAAG,CAAC,MAAM,CAAC,qBAAqB,KAAK,SAAS;gBAC5C,CAAC,CAAC,EAAE,qBAAqB,EAAE,MAAM,CAAC,qBAAqB,EAAE;gBACzD,CAAC,CAAC,EAAE,CAAC;YACP,GAAG,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,MAAM,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACxE,GAAG,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACzD,GAAG,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC/D,GAAG,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE,cAAc,EAAE,MAAM,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC3E,GAAG,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC/D,GAAG,CAAC,MAAM,CAAC,QAAQ,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACvE,GAAG,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE,cAAc,EAAE,MAAM,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC3E,GAAG,CAAC,MAAM,CAAC,qBAAqB;gBAC9B,CAAC,CAAC,EAAE,qBAAqB,EAAE,MAAM,CAAC,qBAAqB,EAAE;gBACzD,CAAC,CAAC,EAAE,CAAC;SACR,CAAC;IACJ,CAAC;CACF"}
|
|
@@ -0,0 +1,18 @@
|
|
|
1
|
+
import type { Tool } from "@modelcontextprotocol/sdk/types.js";
|
|
2
|
+
import type { MetricsCollector } from "../metrics.js";
|
|
3
|
+
import type { SessionManager } from "../session.js";
|
|
4
|
+
import type { ToolProvider } from "./types.js";
|
|
5
|
+
export interface SystemToolProviderDeps {
|
|
6
|
+
sessionManager: SessionManager;
|
|
7
|
+
metrics: MetricsCollector;
|
|
8
|
+
}
|
|
9
|
+
export declare class SystemToolProvider implements ToolProvider {
|
|
10
|
+
private readonly deps;
|
|
11
|
+
readonly namespace = "system";
|
|
12
|
+
constructor(deps: SystemToolProviderDeps);
|
|
13
|
+
getTools(): Tool[];
|
|
14
|
+
handleTool(toolName: string, args: unknown): Promise<unknown> | undefined;
|
|
15
|
+
private detect;
|
|
16
|
+
private getMetrics;
|
|
17
|
+
}
|
|
18
|
+
//# sourceMappingURL=system.provider.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"system.provider.d.ts","sourceRoot":"","sources":["../../src/tools/system.provider.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,oCAAoC,CAAC;AAE/D,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,eAAe,CAAC;AACtD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAKpD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAE/C,MAAM,WAAW,sBAAsB;IACrC,cAAc,EAAE,cAAc,CAAC;IAC/B,OAAO,EAAE,gBAAgB,CAAC;CAC3B;AAED,qBAAa,kBAAmB,YAAW,YAAY;IAGzC,OAAO,CAAC,QAAQ,CAAC,IAAI;IAFjC,QAAQ,CAAC,SAAS,YAAY;gBAED,IAAI,EAAE,sBAAsB;IAEzD,QAAQ,IAAI,IAAI,EAAE;IA6ClB,UAAU,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,GAAG,SAAS;YAW3D,MAAM;YAON,UAAU;CASzB"}
|
|
@@ -0,0 +1,81 @@
|
|
|
1
|
+
import { logger } from "../logging.js";
|
|
2
|
+
import { MetricsFormatSchema, SessionIdSchema } from "../types.js";
|
|
3
|
+
import { annotate } from "./metadata.js";
|
|
4
|
+
import { METRICS_OUTPUT_SCHEMA, OS_INFO_OUTPUT_SCHEMA } from "./output-schemas.js";
|
|
5
|
+
import { toolResult } from "./results.js";
|
|
6
|
+
export class SystemToolProvider {
|
|
7
|
+
deps;
|
|
8
|
+
namespace = "system";
|
|
9
|
+
constructor(deps) {
|
|
10
|
+
this.deps = deps;
|
|
11
|
+
}
|
|
12
|
+
getTools() {
|
|
13
|
+
return [
|
|
14
|
+
{
|
|
15
|
+
name: "os_detect",
|
|
16
|
+
description: "Detects operating system and environment information",
|
|
17
|
+
annotations: annotate({
|
|
18
|
+
title: "Detect Remote OS",
|
|
19
|
+
readOnly: true,
|
|
20
|
+
idempotent: true,
|
|
21
|
+
}),
|
|
22
|
+
outputSchema: OS_INFO_OUTPUT_SCHEMA,
|
|
23
|
+
inputSchema: {
|
|
24
|
+
type: "object",
|
|
25
|
+
properties: {
|
|
26
|
+
sessionId: { type: "string", description: "SSH session ID" },
|
|
27
|
+
},
|
|
28
|
+
required: ["sessionId"],
|
|
29
|
+
},
|
|
30
|
+
},
|
|
31
|
+
{
|
|
32
|
+
name: "get_metrics",
|
|
33
|
+
description: "Returns server metrics including session counts, command statistics, and uptime",
|
|
34
|
+
annotations: annotate({
|
|
35
|
+
title: "Get Runtime Metrics",
|
|
36
|
+
readOnly: true,
|
|
37
|
+
idempotent: true,
|
|
38
|
+
openWorld: false,
|
|
39
|
+
}),
|
|
40
|
+
outputSchema: METRICS_OUTPUT_SCHEMA,
|
|
41
|
+
inputSchema: {
|
|
42
|
+
type: "object",
|
|
43
|
+
properties: {
|
|
44
|
+
format: {
|
|
45
|
+
type: "string",
|
|
46
|
+
enum: ["json", "prometheus"],
|
|
47
|
+
description: "Output format (default: json)",
|
|
48
|
+
},
|
|
49
|
+
},
|
|
50
|
+
required: [],
|
|
51
|
+
},
|
|
52
|
+
},
|
|
53
|
+
];
|
|
54
|
+
}
|
|
55
|
+
handleTool(toolName, args) {
|
|
56
|
+
switch (toolName) {
|
|
57
|
+
case "os_detect":
|
|
58
|
+
return this.detect(args);
|
|
59
|
+
case "get_metrics":
|
|
60
|
+
return this.getMetrics(args);
|
|
61
|
+
default:
|
|
62
|
+
return undefined;
|
|
63
|
+
}
|
|
64
|
+
}
|
|
65
|
+
async detect(args) {
|
|
66
|
+
const { sessionId } = SessionIdSchema.parse(args);
|
|
67
|
+
const result = await this.deps.sessionManager.getOSInfo(sessionId);
|
|
68
|
+
logger.info("OS detected", { sessionId });
|
|
69
|
+
return result;
|
|
70
|
+
}
|
|
71
|
+
async getMetrics(args) {
|
|
72
|
+
const { format } = MetricsFormatSchema.parse(args ?? {});
|
|
73
|
+
if (format === "prometheus") {
|
|
74
|
+
const metrics = this.deps.metrics.exportPrometheus();
|
|
75
|
+
return toolResult({ format: "prometheus", metrics }, metrics);
|
|
76
|
+
}
|
|
77
|
+
logger.debug("Metrics retrieved");
|
|
78
|
+
return this.deps.metrics.getMetrics();
|
|
79
|
+
}
|
|
80
|
+
}
|
|
81
|
+
//# sourceMappingURL=system.provider.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"system.provider.js","sourceRoot":"","sources":["../../src/tools/system.provider.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAGvC,OAAO,EAAE,mBAAmB,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AACnE,OAAO,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AACzC,OAAO,EAAE,qBAAqB,EAAE,qBAAqB,EAAE,MAAM,qBAAqB,CAAC;AACnF,OAAO,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAQ1C,MAAM,OAAO,kBAAkB;IAGA;IAFpB,SAAS,GAAG,QAAQ,CAAC;IAE9B,YAA6B,IAA4B;QAA5B,SAAI,GAAJ,IAAI,CAAwB;IAAG,CAAC;IAE7D,QAAQ;QACN,OAAO;YACL;gBACE,IAAI,EAAE,WAAW;gBACjB,WAAW,EAAE,sDAAsD;gBACnE,WAAW,EAAE,QAAQ,CAAC;oBACpB,KAAK,EAAE,kBAAkB;oBACzB,QAAQ,EAAE,IAAI;oBACd,UAAU,EAAE,IAAI;iBACjB,CAAC;gBACF,YAAY,EAAE,qBAAqB;gBACnC,WAAW,EAAE;oBACX,IAAI,EAAE,QAAiB;oBACvB,UAAU,EAAE;wBACV,SAAS,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,gBAAgB,EAAE;qBAC7D;oBACD,QAAQ,EAAE,CAAC,WAAW,CAAC;iBACxB;aACF;YACD;gBACE,IAAI,EAAE,aAAa;gBACnB,WAAW,EACT,iFAAiF;gBACnF,WAAW,EAAE,QAAQ,CAAC;oBACpB,KAAK,EAAE,qBAAqB;oBAC5B,QAAQ,EAAE,IAAI;oBACd,UAAU,EAAE,IAAI;oBAChB,SAAS,EAAE,KAAK;iBACjB,CAAC;gBACF,YAAY,EAAE,qBAAqB;gBACnC,WAAW,EAAE;oBACX,IAAI,EAAE,QAAiB;oBACvB,UAAU,EAAE;wBACV,MAAM,EAAE;4BACN,IAAI,EAAE,QAAQ;4BACd,IAAI,EAAE,CAAC,MAAM,EAAE,YAAY,CAAC;4BAC5B,WAAW,EAAE,+BAA+B;yBAC7C;qBACF;oBACD,QAAQ,EAAE,EAAE;iBACb;aACF;SACF,CAAC;IACJ,CAAC;IAED,UAAU,CAAC,QAAgB,EAAE,IAAa;QACxC,QAAQ,QAAQ,EAAE,CAAC;YACjB,KAAK,WAAW;gBACd,OAAO,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YAC3B,KAAK,aAAa;gBAChB,OAAO,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;YAC/B;gBACE,OAAO,SAAS,CAAC;QACrB,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,MAAM,CAAC,IAAa;QAChC,MAAM,EAAE,SAAS,EAAE,GAAG,eAAe,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAClD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QACnE,MAAM,CAAC,IAAI,CAAC,aAAa,EAAE,EAAE,SAAS,EAAE,CAAC,CAAC;QAC1C,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,KAAK,CAAC,UAAU,CAAC,IAAa;QACpC,MAAM,EAAE,MAAM,EAAE,GAAG,mBAAmB,CAAC,KAAK,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;QACzD,IAAI,MAAM,KAAK,YAAY,EAAE,CAAC;YAC5B,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,gBAAgB,EAAE,CAAC;YACrD,OAAO,UAAU,CAAC,EAAE,MAAM,EAAE,YAAY,EAAE,OAAO,EAAE,EAAE,OAAO,CAAC,CAAC;QAChE,CAAC;QACD,MAAM,CAAC,KAAK,CAAC,mBAAmB,CAAC,CAAC;QAClC,OAAO,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;IACxC,CAAC;CACF"}
|