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,140 @@
|
|
|
1
|
+
import { logger } from "./logging.js";
|
|
2
|
+
import { createTimeoutError } from "./errors.js";
|
|
3
|
+
import { buildRemoteCommand } from "./shell.js";
|
|
4
|
+
function appendBounded(current, data, maxBytes) {
|
|
5
|
+
const currentBytes = Buffer.byteLength(current, "utf8");
|
|
6
|
+
const remaining = maxBytes - currentBytes;
|
|
7
|
+
if (remaining <= 0) {
|
|
8
|
+
return { value: current, truncated: data.length > 0 };
|
|
9
|
+
}
|
|
10
|
+
if (Buffer.byteLength(data, "utf8") <= remaining) {
|
|
11
|
+
return { value: current + data, truncated: false };
|
|
12
|
+
}
|
|
13
|
+
let next = "";
|
|
14
|
+
let used = 0;
|
|
15
|
+
for (const char of data) {
|
|
16
|
+
const charBytes = Buffer.byteLength(char, "utf8");
|
|
17
|
+
if (used + charBytes > remaining) {
|
|
18
|
+
break;
|
|
19
|
+
}
|
|
20
|
+
next += char;
|
|
21
|
+
used += charBytes;
|
|
22
|
+
}
|
|
23
|
+
return { value: current + next, truncated: true };
|
|
24
|
+
}
|
|
25
|
+
export function createStreamingService({ sessionManager, config, policy, }) {
|
|
26
|
+
async function execWithStreaming(options) {
|
|
27
|
+
const { sessionId, command, cwd, env, onChunk } = options;
|
|
28
|
+
logger.debug("Starting streaming execution", { sessionId, command });
|
|
29
|
+
const session = sessionManager.getSession(sessionId);
|
|
30
|
+
if (!session) {
|
|
31
|
+
throw new Error(`Session ${sessionId} not found or expired`);
|
|
32
|
+
}
|
|
33
|
+
const decision = policy.assertAllowed({
|
|
34
|
+
action: "proc.exec",
|
|
35
|
+
command,
|
|
36
|
+
mode: session.info.policyMode,
|
|
37
|
+
});
|
|
38
|
+
if (decision.mode === "explain") {
|
|
39
|
+
return {
|
|
40
|
+
code: 0,
|
|
41
|
+
chunks: [],
|
|
42
|
+
stdout: JSON.stringify({ wouldExecute: true, command, policy: decision }, null, 2),
|
|
43
|
+
stderr: "",
|
|
44
|
+
durationMs: 0,
|
|
45
|
+
truncated: false,
|
|
46
|
+
};
|
|
47
|
+
}
|
|
48
|
+
const osInfo = await sessionManager.getOSInfo(sessionId);
|
|
49
|
+
const startTime = Date.now();
|
|
50
|
+
const chunks = [];
|
|
51
|
+
let fullStdout = "";
|
|
52
|
+
let fullStderr = "";
|
|
53
|
+
let truncated = false;
|
|
54
|
+
return new Promise((resolve, reject) => {
|
|
55
|
+
const shellCommand = buildRemoteCommand(command, osInfo, cwd, env);
|
|
56
|
+
const timeout = setTimeout(() => {
|
|
57
|
+
reject(createTimeoutError(`Streaming command timed out after ${options.timeoutMs ?? config.commandTimeoutMs}ms`, "Increase timeout or use a shorter streaming command."));
|
|
58
|
+
}, options.timeoutMs ?? config.commandTimeoutMs);
|
|
59
|
+
session.ssh
|
|
60
|
+
.execCommand(shellCommand, {
|
|
61
|
+
onStdout: (chunk) => {
|
|
62
|
+
const data = chunk.toString();
|
|
63
|
+
const bounded = appendBounded(fullStdout, data, config.maxCommandOutputBytes);
|
|
64
|
+
fullStdout = bounded.value;
|
|
65
|
+
truncated = truncated || bounded.truncated || chunks.length >= config.maxStreamChunks;
|
|
66
|
+
if (chunks.length < config.maxStreamChunks && !bounded.truncated) {
|
|
67
|
+
const streamChunk = {
|
|
68
|
+
type: "stdout",
|
|
69
|
+
data,
|
|
70
|
+
timestamp: Date.now(),
|
|
71
|
+
};
|
|
72
|
+
chunks.push(streamChunk);
|
|
73
|
+
onChunk?.(streamChunk);
|
|
74
|
+
}
|
|
75
|
+
},
|
|
76
|
+
onStderr: (chunk) => {
|
|
77
|
+
const data = chunk.toString();
|
|
78
|
+
const bounded = appendBounded(fullStderr, data, config.maxCommandOutputBytes);
|
|
79
|
+
fullStderr = bounded.value;
|
|
80
|
+
truncated = truncated || bounded.truncated || chunks.length >= config.maxStreamChunks;
|
|
81
|
+
if (chunks.length < config.maxStreamChunks && !bounded.truncated) {
|
|
82
|
+
const streamChunk = {
|
|
83
|
+
type: "stderr",
|
|
84
|
+
data,
|
|
85
|
+
timestamp: Date.now(),
|
|
86
|
+
};
|
|
87
|
+
chunks.push(streamChunk);
|
|
88
|
+
onChunk?.(streamChunk);
|
|
89
|
+
}
|
|
90
|
+
},
|
|
91
|
+
})
|
|
92
|
+
.then((result) => {
|
|
93
|
+
clearTimeout(timeout);
|
|
94
|
+
if (truncated) {
|
|
95
|
+
const truncatedChunk = {
|
|
96
|
+
type: "truncated",
|
|
97
|
+
data: "Output exceeded configured streaming limits",
|
|
98
|
+
timestamp: Date.now(),
|
|
99
|
+
};
|
|
100
|
+
chunks.push(truncatedChunk);
|
|
101
|
+
onChunk?.(truncatedChunk);
|
|
102
|
+
}
|
|
103
|
+
const exitChunk = {
|
|
104
|
+
type: "exit",
|
|
105
|
+
code: result.code ?? 0,
|
|
106
|
+
timestamp: Date.now(),
|
|
107
|
+
};
|
|
108
|
+
chunks.push(exitChunk);
|
|
109
|
+
onChunk?.(exitChunk);
|
|
110
|
+
const streamResult = {
|
|
111
|
+
code: result.code ?? 0,
|
|
112
|
+
chunks,
|
|
113
|
+
stdout: fullStdout,
|
|
114
|
+
stderr: fullStderr,
|
|
115
|
+
durationMs: Date.now() - startTime,
|
|
116
|
+
truncated,
|
|
117
|
+
};
|
|
118
|
+
logger.debug("Streaming execution completed", {
|
|
119
|
+
sessionId,
|
|
120
|
+
code: streamResult.code,
|
|
121
|
+
chunkCount: chunks.length,
|
|
122
|
+
durationMs: streamResult.durationMs,
|
|
123
|
+
});
|
|
124
|
+
resolve(streamResult);
|
|
125
|
+
})
|
|
126
|
+
.catch((error) => {
|
|
127
|
+
clearTimeout(timeout);
|
|
128
|
+
reject(error);
|
|
129
|
+
});
|
|
130
|
+
});
|
|
131
|
+
}
|
|
132
|
+
return { execWithStreaming };
|
|
133
|
+
}
|
|
134
|
+
export function formatStreamOutput(chunks) {
|
|
135
|
+
return chunks
|
|
136
|
+
.filter((chunk) => chunk.type !== "exit" && chunk.type !== "truncated" && chunk.data)
|
|
137
|
+
.map((chunk) => chunk.data)
|
|
138
|
+
.join("");
|
|
139
|
+
}
|
|
140
|
+
//# sourceMappingURL=streaming.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"streaming.js","sourceRoot":"","sources":["../src/streaming.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AACtC,OAAO,EAAE,kBAAkB,EAAE,MAAM,aAAa,CAAC;AACjD,OAAO,EAAE,kBAAkB,EAAE,MAAM,YAAY,CAAC;AAwChD,SAAS,aAAa,CAAC,OAAe,EAAE,IAAY,EAAE,QAAgB;IACpE,MAAM,YAAY,GAAG,MAAM,CAAC,UAAU,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;IACxD,MAAM,SAAS,GAAG,QAAQ,GAAG,YAAY,CAAC;IAC1C,IAAI,SAAS,IAAI,CAAC,EAAE,CAAC;QACnB,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,SAAS,EAAE,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;IACxD,CAAC;IAED,IAAI,MAAM,CAAC,UAAU,CAAC,IAAI,EAAE,MAAM,CAAC,IAAI,SAAS,EAAE,CAAC;QACjD,OAAO,EAAE,KAAK,EAAE,OAAO,GAAG,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC;IACrD,CAAC;IAED,IAAI,IAAI,GAAG,EAAE,CAAC;IACd,IAAI,IAAI,GAAG,CAAC,CAAC;IACb,KAAK,MAAM,IAAI,IAAI,IAAI,EAAE,CAAC;QACxB,MAAM,SAAS,GAAG,MAAM,CAAC,UAAU,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAClD,IAAI,IAAI,GAAG,SAAS,GAAG,SAAS,EAAE,CAAC;YACjC,MAAM;QACR,CAAC;QACD,IAAI,IAAI,IAAI,CAAC;QACb,IAAI,IAAI,SAAS,CAAC;IACpB,CAAC;IACD,OAAO,EAAE,KAAK,EAAE,OAAO,GAAG,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC;AACpD,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,EACrC,cAAc,EACd,MAAM,EACN,MAAM,GACe;IACrB,KAAK,UAAU,iBAAiB,CAAC,OAAsB;QACrD,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,GAAG,EAAE,GAAG,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC;QAC1D,MAAM,CAAC,KAAK,CAAC,8BAA8B,EAAE,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC,CAAC;QAErE,MAAM,OAAO,GAAG,cAAc,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;QACrD,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CAAC,WAAW,SAAS,uBAAuB,CAAC,CAAC;QAC/D,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,CAAC,aAAa,CAAC;YACpC,MAAM,EAAE,WAAW;YACnB,OAAO;YACP,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,UAAU;SAC9B,CAAC,CAAC;QACH,IAAI,QAAQ,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;YAChC,OAAO;gBACL,IAAI,EAAE,CAAC;gBACP,MAAM,EAAE,EAAE;gBACV,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,YAAY,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;gBAClF,MAAM,EAAE,EAAE;gBACV,UAAU,EAAE,CAAC;gBACb,SAAS,EAAE,KAAK;aACjB,CAAC;QACJ,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QACzD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,MAAM,MAAM,GAAkB,EAAE,CAAC;QACjC,IAAI,UAAU,GAAG,EAAE,CAAC;QACpB,IAAI,UAAU,GAAG,EAAE,CAAC;QACpB,IAAI,SAAS,GAAG,KAAK,CAAC;QAEtB,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,MAAM,YAAY,GAAG,kBAAkB,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;YACnE,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,EAAE;gBAC9B,MAAM,CACJ,kBAAkB,CAChB,qCAAqC,OAAO,CAAC,SAAS,IAAI,MAAM,CAAC,gBAAgB,IAAI,EACrF,sDAAsD,CACvD,CACF,CAAC;YACJ,CAAC,EAAE,OAAO,CAAC,SAAS,IAAI,MAAM,CAAC,gBAAgB,CAAC,CAAC;YAEjD,OAAO,CAAC,GAAG;iBACR,WAAW,CAAC,YAAY,EAAE;gBACzB,QAAQ,EAAE,CAAC,KAAa,EAAE,EAAE;oBAC1B,MAAM,IAAI,GAAG,KAAK,CAAC,QAAQ,EAAE,CAAC;oBAC9B,MAAM,OAAO,GAAG,aAAa,CAAC,UAAU,EAAE,IAAI,EAAE,MAAM,CAAC,qBAAqB,CAAC,CAAC;oBAC9E,UAAU,GAAG,OAAO,CAAC,KAAK,CAAC;oBAC3B,SAAS,GAAG,SAAS,IAAI,OAAO,CAAC,SAAS,IAAI,MAAM,CAAC,MAAM,IAAI,MAAM,CAAC,eAAe,CAAC;oBAEtF,IAAI,MAAM,CAAC,MAAM,GAAG,MAAM,CAAC,eAAe,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC;wBACjE,MAAM,WAAW,GAAgB;4BAC/B,IAAI,EAAE,QAAQ;4BACd,IAAI;4BACJ,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;yBACtB,CAAC;wBACF,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;wBACzB,OAAO,EAAE,CAAC,WAAW,CAAC,CAAC;oBACzB,CAAC;gBACH,CAAC;gBACD,QAAQ,EAAE,CAAC,KAAa,EAAE,EAAE;oBAC1B,MAAM,IAAI,GAAG,KAAK,CAAC,QAAQ,EAAE,CAAC;oBAC9B,MAAM,OAAO,GAAG,aAAa,CAAC,UAAU,EAAE,IAAI,EAAE,MAAM,CAAC,qBAAqB,CAAC,CAAC;oBAC9E,UAAU,GAAG,OAAO,CAAC,KAAK,CAAC;oBAC3B,SAAS,GAAG,SAAS,IAAI,OAAO,CAAC,SAAS,IAAI,MAAM,CAAC,MAAM,IAAI,MAAM,CAAC,eAAe,CAAC;oBAEtF,IAAI,MAAM,CAAC,MAAM,GAAG,MAAM,CAAC,eAAe,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC;wBACjE,MAAM,WAAW,GAAgB;4BAC/B,IAAI,EAAE,QAAQ;4BACd,IAAI;4BACJ,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;yBACtB,CAAC;wBACF,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;wBACzB,OAAO,EAAE,CAAC,WAAW,CAAC,CAAC;oBACzB,CAAC;gBACH,CAAC;aACF,CAAC;iBACD,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE;gBACf,YAAY,CAAC,OAAO,CAAC,CAAC;gBACtB,IAAI,SAAS,EAAE,CAAC;oBACd,MAAM,cAAc,GAAgB;wBAClC,IAAI,EAAE,WAAW;wBACjB,IAAI,EAAE,6CAA6C;wBACnD,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;qBACtB,CAAC;oBACF,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;oBAC5B,OAAO,EAAE,CAAC,cAAc,CAAC,CAAC;gBAC5B,CAAC;gBACD,MAAM,SAAS,GAAgB;oBAC7B,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,MAAM,CAAC,IAAI,IAAI,CAAC;oBACtB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;iBACtB,CAAC;gBACF,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBACvB,OAAO,EAAE,CAAC,SAAS,CAAC,CAAC;gBAErB,MAAM,YAAY,GAAiB;oBACjC,IAAI,EAAE,MAAM,CAAC,IAAI,IAAI,CAAC;oBACtB,MAAM;oBACN,MAAM,EAAE,UAAU;oBAClB,MAAM,EAAE,UAAU;oBAClB,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;oBAClC,SAAS;iBACV,CAAC;gBAEF,MAAM,CAAC,KAAK,CAAC,+BAA+B,EAAE;oBAC5C,SAAS;oBACT,IAAI,EAAE,YAAY,CAAC,IAAI;oBACvB,UAAU,EAAE,MAAM,CAAC,MAAM;oBACzB,UAAU,EAAE,YAAY,CAAC,UAAU;iBACpC,CAAC,CAAC;gBAEH,OAAO,CAAC,YAAY,CAAC,CAAC;YACxB,CAAC,CAAC;iBACD,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBACf,YAAY,CAAC,OAAO,CAAC,CAAC;gBACtB,MAAM,CAAC,KAAK,CAAC,CAAC;YAChB,CAAC,CAAC,CAAC;QACP,CAAC,CAAC,CAAC;IACL,CAAC;IAED,OAAO,EAAE,iBAAiB,EAAE,CAAC;AAC/B,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,MAAqB;IACtD,OAAO,MAAM;SACV,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,MAAM,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW,IAAI,KAAK,CAAC,IAAI,CAAC;SACpF,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC;SAC1B,IAAI,CAAC,EAAE,CAAC,CAAC;AACd,CAAC"}
|
|
@@ -0,0 +1,17 @@
|
|
|
1
|
+
import { type Span, type SpanOptions } from "@opentelemetry/api";
|
|
2
|
+
export interface TelemetryConfig {
|
|
3
|
+
enabled: boolean;
|
|
4
|
+
endpoint?: string | undefined;
|
|
5
|
+
serviceName: string;
|
|
6
|
+
serviceVersion?: string | undefined;
|
|
7
|
+
environment?: string | undefined;
|
|
8
|
+
}
|
|
9
|
+
type SpanWork<T> = (span: Span) => Promise<T> | T;
|
|
10
|
+
export declare function getTelemetryConfig(env?: NodeJS.ProcessEnv, overrides?: Partial<Omit<TelemetryConfig, "enabled">>): TelemetryConfig;
|
|
11
|
+
export declare function normalizeOtlpEndpoint(endpoint: string): string;
|
|
12
|
+
export declare function initTelemetry(overrides?: Partial<Omit<TelemetryConfig, "enabled">>): boolean;
|
|
13
|
+
export declare function shutdownTelemetry(): Promise<void>;
|
|
14
|
+
export declare function withSpan<T>(name: string, work: SpanWork<T>, options?: SpanOptions): Promise<T>;
|
|
15
|
+
export declare function isTelemetryEnabled(): boolean;
|
|
16
|
+
export {};
|
|
17
|
+
//# sourceMappingURL=telemetry.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"telemetry.d.ts","sourceRoot":"","sources":["../src/telemetry.ts"],"names":[],"mappings":"AAAA,OAAO,EAIL,KAAK,IAAI,EACT,KAAK,WAAW,EACjB,MAAM,oBAAoB,CAAC;AAa5B,MAAM,WAAW,eAAe;IAC9B,OAAO,EAAE,OAAO,CAAC;IACjB,QAAQ,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC9B,WAAW,EAAE,MAAM,CAAC;IACpB,cAAc,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IACpC,WAAW,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CAClC;AAED,KAAK,QAAQ,CAAC,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;AAKlD,wBAAgB,kBAAkB,CAChC,GAAG,GAAE,MAAM,CAAC,UAAwB,EACpC,SAAS,GAAE,OAAO,CAAC,IAAI,CAAC,eAAe,EAAE,SAAS,CAAC,CAAM,GACxD,eAAe,CAgBjB;AAED,wBAAgB,qBAAqB,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,CAG9D;AAED,wBAAgB,aAAa,CAAC,SAAS,GAAE,OAAO,CAAC,IAAI,CAAC,eAAe,EAAE,SAAS,CAAC,CAAM,GAAG,OAAO,CAyChG;AAED,wBAAsB,iBAAiB,IAAI,OAAO,CAAC,IAAI,CAAC,CAcvD;AAED,wBAAsB,QAAQ,CAAC,CAAC,EAC9B,IAAI,EAAE,MAAM,EACZ,IAAI,EAAE,QAAQ,CAAC,CAAC,CAAC,EACjB,OAAO,GAAE,WAAgB,GACxB,OAAO,CAAC,CAAC,CAAC,CAmBZ;AAED,wBAAgB,kBAAkB,IAAI,OAAO,CAE5C"}
|
|
@@ -0,0 +1,101 @@
|
|
|
1
|
+
import { trace, SpanStatusCode, } from "@opentelemetry/api";
|
|
2
|
+
import { OTLPTraceExporter } from "@opentelemetry/exporter-trace-otlp-http";
|
|
3
|
+
import { resourceFromAttributes } from "@opentelemetry/resources";
|
|
4
|
+
import { NodeSDK } from "@opentelemetry/sdk-node";
|
|
5
|
+
import { ATTR_DEPLOYMENT_ENVIRONMENT_NAME, ATTR_SERVICE_NAME, ATTR_SERVICE_VERSION, } from "@opentelemetry/semantic-conventions";
|
|
6
|
+
import { logger } from "./logging.js";
|
|
7
|
+
const DEFAULT_SERVICE_NAME = "ssh-mcp-pro";
|
|
8
|
+
let telemetrySdk;
|
|
9
|
+
let telemetryConfig = getTelemetryConfig();
|
|
10
|
+
export function getTelemetryConfig(env = process.env, overrides = {}) {
|
|
11
|
+
const rawEndpoint = overrides.endpoint ?? env.OTEL_EXPORTER_OTLP_ENDPOINT;
|
|
12
|
+
const endpoint = rawEndpoint?.trim() ? normalizeOtlpEndpoint(rawEndpoint) : undefined;
|
|
13
|
+
const serviceName = overrides.serviceName?.trim() ?? env.OTEL_SERVICE_NAME?.trim() ?? DEFAULT_SERVICE_NAME;
|
|
14
|
+
const serviceVersion = overrides.serviceVersion?.trim() ?? env.OTEL_SERVICE_VERSION?.trim() ?? undefined;
|
|
15
|
+
const environment = overrides.environment?.trim() ?? env.NODE_ENV?.trim() ?? undefined;
|
|
16
|
+
return {
|
|
17
|
+
enabled: Boolean(endpoint),
|
|
18
|
+
endpoint,
|
|
19
|
+
serviceName,
|
|
20
|
+
serviceVersion,
|
|
21
|
+
environment,
|
|
22
|
+
};
|
|
23
|
+
}
|
|
24
|
+
export function normalizeOtlpEndpoint(endpoint) {
|
|
25
|
+
const trimmed = endpoint.trim().replace(/\/+$/, "");
|
|
26
|
+
return trimmed.endsWith("/v1/traces") ? trimmed : `${trimmed}/v1/traces`;
|
|
27
|
+
}
|
|
28
|
+
export function initTelemetry(overrides = {}) {
|
|
29
|
+
if (telemetrySdk) {
|
|
30
|
+
return true;
|
|
31
|
+
}
|
|
32
|
+
telemetryConfig = getTelemetryConfig(process.env, overrides);
|
|
33
|
+
if (!telemetryConfig.enabled || !telemetryConfig.endpoint) {
|
|
34
|
+
logger.debug("OpenTelemetry disabled", {
|
|
35
|
+
reason: "OTEL_EXPORTER_OTLP_ENDPOINT is not configured",
|
|
36
|
+
});
|
|
37
|
+
return false;
|
|
38
|
+
}
|
|
39
|
+
const resourceAttributes = {
|
|
40
|
+
[ATTR_SERVICE_NAME]: telemetryConfig.serviceName,
|
|
41
|
+
};
|
|
42
|
+
if (telemetryConfig.serviceVersion) {
|
|
43
|
+
resourceAttributes[ATTR_SERVICE_VERSION] = telemetryConfig.serviceVersion;
|
|
44
|
+
}
|
|
45
|
+
if (telemetryConfig.environment) {
|
|
46
|
+
resourceAttributes[ATTR_DEPLOYMENT_ENVIRONMENT_NAME] = telemetryConfig.environment;
|
|
47
|
+
}
|
|
48
|
+
const exporter = new OTLPTraceExporter({
|
|
49
|
+
url: telemetryConfig.endpoint,
|
|
50
|
+
});
|
|
51
|
+
telemetrySdk = new NodeSDK({
|
|
52
|
+
serviceName: telemetryConfig.serviceName,
|
|
53
|
+
traceExporter: exporter,
|
|
54
|
+
resource: resourceFromAttributes(resourceAttributes),
|
|
55
|
+
});
|
|
56
|
+
telemetrySdk.start();
|
|
57
|
+
logger.info("OpenTelemetry tracing enabled", {
|
|
58
|
+
endpoint: telemetryConfig.endpoint,
|
|
59
|
+
serviceName: telemetryConfig.serviceName,
|
|
60
|
+
});
|
|
61
|
+
return true;
|
|
62
|
+
}
|
|
63
|
+
export async function shutdownTelemetry() {
|
|
64
|
+
if (!telemetrySdk) {
|
|
65
|
+
return;
|
|
66
|
+
}
|
|
67
|
+
const sdk = telemetrySdk;
|
|
68
|
+
telemetrySdk = undefined;
|
|
69
|
+
try {
|
|
70
|
+
await sdk.shutdown();
|
|
71
|
+
logger.info("OpenTelemetry tracing shut down");
|
|
72
|
+
}
|
|
73
|
+
catch (error) {
|
|
74
|
+
logger.warn("Failed to shut down OpenTelemetry cleanly", { error });
|
|
75
|
+
}
|
|
76
|
+
}
|
|
77
|
+
export async function withSpan(name, work, options = {}) {
|
|
78
|
+
const tracer = trace.getTracer(telemetryConfig.serviceName || DEFAULT_SERVICE_NAME);
|
|
79
|
+
return tracer.startActiveSpan(name, options, async (span) => {
|
|
80
|
+
try {
|
|
81
|
+
const result = await work(span);
|
|
82
|
+
span.setStatus({ code: SpanStatusCode.OK });
|
|
83
|
+
return result;
|
|
84
|
+
}
|
|
85
|
+
catch (error) {
|
|
86
|
+
span.recordException(error instanceof Error ? error : new Error(String(error)));
|
|
87
|
+
span.setStatus({
|
|
88
|
+
code: SpanStatusCode.ERROR,
|
|
89
|
+
message: error instanceof Error ? error.message : String(error),
|
|
90
|
+
});
|
|
91
|
+
throw error;
|
|
92
|
+
}
|
|
93
|
+
finally {
|
|
94
|
+
span.end();
|
|
95
|
+
}
|
|
96
|
+
});
|
|
97
|
+
}
|
|
98
|
+
export function isTelemetryEnabled() {
|
|
99
|
+
return Boolean(telemetrySdk);
|
|
100
|
+
}
|
|
101
|
+
//# sourceMappingURL=telemetry.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"telemetry.js","sourceRoot":"","sources":["../src/telemetry.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,KAAK,EACL,cAAc,GAIf,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAAE,iBAAiB,EAAE,MAAM,yCAAyC,CAAC;AAC5E,OAAO,EAAE,sBAAsB,EAAE,MAAM,0BAA0B,CAAC;AAClE,OAAO,EAAE,OAAO,EAAE,MAAM,yBAAyB,CAAC;AAClD,OAAO,EACL,gCAAgC,EAChC,iBAAiB,EACjB,oBAAoB,GACrB,MAAM,qCAAqC,CAAC;AAC7C,OAAO,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AAEtC,MAAM,oBAAoB,GAAG,aAAa,CAAC;AAY3C,IAAI,YAAiC,CAAC;AACtC,IAAI,eAAe,GAAG,kBAAkB,EAAE,CAAC;AAE3C,MAAM,UAAU,kBAAkB,CAChC,MAAyB,OAAO,CAAC,GAAG,EACpC,YAAuD,EAAE;IAEzD,MAAM,WAAW,GAAG,SAAS,CAAC,QAAQ,IAAI,GAAG,CAAC,2BAA2B,CAAC;IAC1E,MAAM,QAAQ,GAAG,WAAW,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,qBAAqB,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IACtF,MAAM,WAAW,GACf,SAAS,CAAC,WAAW,EAAE,IAAI,EAAE,IAAI,GAAG,CAAC,iBAAiB,EAAE,IAAI,EAAE,IAAI,oBAAoB,CAAC;IACzF,MAAM,cAAc,GAClB,SAAS,CAAC,cAAc,EAAE,IAAI,EAAE,IAAI,GAAG,CAAC,oBAAoB,EAAE,IAAI,EAAE,IAAI,SAAS,CAAC;IACpF,MAAM,WAAW,GAAG,SAAS,CAAC,WAAW,EAAE,IAAI,EAAE,IAAI,GAAG,CAAC,QAAQ,EAAE,IAAI,EAAE,IAAI,SAAS,CAAC;IAEvF,OAAO;QACL,OAAO,EAAE,OAAO,CAAC,QAAQ,CAAC;QAC1B,QAAQ;QACR,WAAW;QACX,cAAc;QACd,WAAW;KACZ,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,QAAgB;IACpD,MAAM,OAAO,GAAG,QAAQ,CAAC,IAAI,EAAE,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;IACpD,OAAO,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,OAAO,YAAY,CAAC;AAC3E,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,YAAuD,EAAE;IACrF,IAAI,YAAY,EAAE,CAAC;QACjB,OAAO,IAAI,CAAC;IACd,CAAC;IAED,eAAe,GAAG,kBAAkB,CAAC,OAAO,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;IAC7D,IAAI,CAAC,eAAe,CAAC,OAAO,IAAI,CAAC,eAAe,CAAC,QAAQ,EAAE,CAAC;QAC1D,MAAM,CAAC,KAAK,CAAC,wBAAwB,EAAE;YACrC,MAAM,EAAE,+CAA+C;SACxD,CAAC,CAAC;QACH,OAAO,KAAK,CAAC;IACf,CAAC;IAED,MAAM,kBAAkB,GAAe;QACrC,CAAC,iBAAiB,CAAC,EAAE,eAAe,CAAC,WAAW;KACjD,CAAC;IAEF,IAAI,eAAe,CAAC,cAAc,EAAE,CAAC;QACnC,kBAAkB,CAAC,oBAAoB,CAAC,GAAG,eAAe,CAAC,cAAc,CAAC;IAC5E,CAAC;IACD,IAAI,eAAe,CAAC,WAAW,EAAE,CAAC;QAChC,kBAAkB,CAAC,gCAAgC,CAAC,GAAG,eAAe,CAAC,WAAW,CAAC;IACrF,CAAC;IAED,MAAM,QAAQ,GAAG,IAAI,iBAAiB,CAAC;QACrC,GAAG,EAAE,eAAe,CAAC,QAAQ;KAC9B,CAAC,CAAC;IAEH,YAAY,GAAG,IAAI,OAAO,CAAC;QACzB,WAAW,EAAE,eAAe,CAAC,WAAW;QACxC,aAAa,EAAE,QAAQ;QACvB,QAAQ,EAAE,sBAAsB,CAAC,kBAAkB,CAAC;KACrD,CAAC,CAAC;IACH,YAAY,CAAC,KAAK,EAAE,CAAC;IAErB,MAAM,CAAC,IAAI,CAAC,+BAA+B,EAAE;QAC3C,QAAQ,EAAE,eAAe,CAAC,QAAQ;QAClC,WAAW,EAAE,eAAe,CAAC,WAAW;KACzC,CAAC,CAAC;IAEH,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,iBAAiB;IACrC,IAAI,CAAC,YAAY,EAAE,CAAC;QAClB,OAAO;IACT,CAAC;IAED,MAAM,GAAG,GAAG,YAAY,CAAC;IACzB,YAAY,GAAG,SAAS,CAAC;IAEzB,IAAI,CAAC;QACH,MAAM,GAAG,CAAC,QAAQ,EAAE,CAAC;QACrB,MAAM,CAAC,IAAI,CAAC,iCAAiC,CAAC,CAAC;IACjD,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,CAAC,IAAI,CAAC,2CAA2C,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;IACtE,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,QAAQ,CAC5B,IAAY,EACZ,IAAiB,EACjB,UAAuB,EAAE;IAEzB,MAAM,MAAM,GAAG,KAAK,CAAC,SAAS,CAAC,eAAe,CAAC,WAAW,IAAI,oBAAoB,CAAC,CAAC;IAEpF,OAAO,MAAM,CAAC,eAAe,CAAC,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;QAC1D,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,CAAC;YAChC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,cAAc,CAAC,EAAE,EAAE,CAAC,CAAC;YAC5C,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,eAAe,CAAC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;YAChF,IAAI,CAAC,SAAS,CAAC;gBACb,IAAI,EAAE,cAAc,CAAC,KAAK;gBAC1B,OAAO,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;aAChE,CAAC,CAAC;YACH,MAAM,KAAK,CAAC;QACd,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,GAAG,EAAE,CAAC;QACb,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,kBAAkB;IAChC,OAAO,OAAO,CAAC,YAAY,CAAC,CAAC;AAC/B,CAAC"}
|
|
@@ -0,0 +1,28 @@
|
|
|
1
|
+
import type { Tool } from "@modelcontextprotocol/sdk/types.js";
|
|
2
|
+
import { type ConnectorCredentialRequest } from "../connector-credentials.js";
|
|
3
|
+
import type { ServerConfig } from "../config.js";
|
|
4
|
+
import type { MetricsCollector } from "../metrics.js";
|
|
5
|
+
import type { PolicyEngine } from "../policy.js";
|
|
6
|
+
import type { SessionManager } from "../session.js";
|
|
7
|
+
import type { ToolProvider } from "./types.js";
|
|
8
|
+
export interface ConnectorToolProviderDeps {
|
|
9
|
+
sessionManager: SessionManager;
|
|
10
|
+
metrics: MetricsCollector;
|
|
11
|
+
config: ServerConfig;
|
|
12
|
+
policy: PolicyEngine;
|
|
13
|
+
getConfiguredHosts?: () => Promise<string[]>;
|
|
14
|
+
resolveConnectorCredentials?: (request: ConnectorCredentialRequest, config: ServerConfig) => Promise<Parameters<SessionManager["openSession"]>[0]>;
|
|
15
|
+
}
|
|
16
|
+
export declare class ConnectorToolProvider implements ToolProvider {
|
|
17
|
+
private readonly deps;
|
|
18
|
+
readonly namespace = "connector";
|
|
19
|
+
constructor(deps: ConnectorToolProviderDeps);
|
|
20
|
+
getTools(): Tool[];
|
|
21
|
+
handleTool(toolName: string, args: unknown): Promise<unknown> | undefined;
|
|
22
|
+
private status;
|
|
23
|
+
private listHosts;
|
|
24
|
+
private explainPolicy;
|
|
25
|
+
private planMutation;
|
|
26
|
+
private inspectHost;
|
|
27
|
+
}
|
|
28
|
+
//# sourceMappingURL=connector.provider.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"connector.provider.d.ts","sourceRoot":"","sources":["../../src/tools/connector.provider.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,oCAAoC,CAAC;AAC/D,OAAO,EAEL,KAAK,0BAA0B,EAChC,MAAM,6BAA6B,CAAC;AACrC,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,cAAc,CAAC;AACjD,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,eAAe,CAAC;AACtD,OAAO,KAAK,EAA+B,YAAY,EAAE,MAAM,cAAc,CAAC;AAC9E,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAUpD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AA4B/C,MAAM,WAAW,yBAAyB;IACxC,cAAc,EAAE,cAAc,CAAC;IAC/B,OAAO,EAAE,gBAAgB,CAAC;IAC1B,MAAM,EAAE,YAAY,CAAC;IACrB,MAAM,EAAE,YAAY,CAAC;IACrB,kBAAkB,CAAC,EAAE,MAAM,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;IAC7C,2BAA2B,CAAC,EAAE,CAC5B,OAAO,EAAE,0BAA0B,EACnC,MAAM,EAAE,YAAY,KACjB,OAAO,CAAC,UAAU,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;CAC5D;AAED,qBAAa,qBAAsB,YAAW,YAAY;IAG5C,OAAO,CAAC,QAAQ,CAAC,IAAI;IAFjC,QAAQ,CAAC,SAAS,eAAe;gBAEJ,IAAI,EAAE,yBAAyB;IAE5D,QAAQ,IAAI,IAAI,EAAE;IA+HlB,UAAU,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,GAAG,SAAS;YAiB3D,MAAM;YAwBN,SAAS;YAmBT,aAAa;YA2Bb,YAAY;YAiCZ,WAAW;CA2C1B"}
|