@duckmind/dm-windows-x64 0.63.4 → 0.63.9
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/dm.exe +0 -0
- package/extensions/.dm-extensions.json +236 -13
- package/extensions/dm-context/package.json +1 -1
- package/extensions/dm-context/skills/context-management/SKILL.md +173 -223
- package/extensions/dm-context/skills/context-management/references/development-and-troubleshooting.md +75 -0
- package/extensions/dm-context/skills/context-management/references/interleaved-async-work.md +143 -0
- package/extensions/dm-context/skills/context-management/references/planning-and-execution.md +81 -0
- package/extensions/dm-context/skills/context-management/references/repeated-items-and-batch-work.md +69 -0
- package/extensions/dm-context/skills/context-management/references/retry-branch-and-pivot.md +80 -0
- package/extensions/dm-context/skills/context-management/references/search-research-and-reading.md +103 -0
- package/extensions/dm-context/skills/context-management/references/task-switching-and-cleanup.md +73 -0
- package/extensions/dm-context/src/context.js +3 -2
- package/extensions/dm-context/src/index.js +168 -84
- package/extensions/dm-mcp-adapter/OAUTH.md +3 -0
- package/extensions/dm-mcp-adapter/THIRD_PARTY_NOTICES.md +4 -0
- package/extensions/dm-mcp-adapter/abort.js +35 -0
- package/extensions/dm-mcp-adapter/agent-dir.js +45 -0
- package/extensions/dm-mcp-adapter/agent-plugin-loader.js +349 -0
- package/extensions/dm-mcp-adapter/app-bridge.bundle.js +5843 -0
- package/extensions/dm-mcp-adapter/cli.js +354 -0
- package/extensions/dm-mcp-adapter/commands.js +619 -0
- package/extensions/dm-mcp-adapter/config.js +1060 -0
- package/extensions/dm-mcp-adapter/consent-manager.js +60 -0
- package/extensions/dm-mcp-adapter/direct-tools.js +544 -0
- package/extensions/dm-mcp-adapter/dist/agent-dir.js +45 -0
- package/extensions/dm-mcp-adapter/dist/agent-plugin-loader.js +349 -0
- package/extensions/dm-mcp-adapter/dist/config.js +1060 -0
- package/extensions/dm-mcp-adapter/dist/mcp-bearer-store.js +311 -0
- package/extensions/dm-mcp-adapter/dist/metadata-cache.js +285 -0
- package/extensions/dm-mcp-adapter/dist/package-mcp-loader.js +144 -0
- package/extensions/dm-mcp-adapter/dist/resource-tools.js +7 -0
- package/extensions/dm-mcp-adapter/dist/types.js +247 -0
- package/extensions/dm-mcp-adapter/dist/ui-app-bridge-helpers.js +36 -0
- package/extensions/dm-mcp-adapter/dist/ui-stream-types.js +58 -0
- package/extensions/dm-mcp-adapter/dist/ui-tool-visibility.js +26 -0
- package/extensions/dm-mcp-adapter/dist/utils.js +307 -0
- package/extensions/dm-mcp-adapter/elicitation-handler.js +306 -0
- package/extensions/dm-mcp-adapter/error-signal.js +16 -0
- package/extensions/dm-mcp-adapter/errors.js +225 -0
- package/extensions/dm-mcp-adapter/failure-backoff.js +19 -0
- package/extensions/dm-mcp-adapter/glimpse-ui.js +61 -0
- package/extensions/dm-mcp-adapter/host-html-template.js +464 -0
- package/extensions/dm-mcp-adapter/index.js +1079 -0
- package/extensions/dm-mcp-adapter/init.js +624 -0
- package/extensions/dm-mcp-adapter/json-schema-validator.js +53 -0
- package/extensions/dm-mcp-adapter/lifecycle.js +385 -0
- package/extensions/dm-mcp-adapter/logger.js +115 -0
- package/extensions/dm-mcp-adapter/mcp-auth-flow.js +824 -0
- package/extensions/dm-mcp-adapter/mcp-auth.js +732 -0
- package/extensions/dm-mcp-adapter/mcp-bearer-store.js +311 -0
- package/extensions/dm-mcp-adapter/mcp-callback-server.js +388 -0
- package/extensions/dm-mcp-adapter/mcp-code.js +291 -0
- package/extensions/dm-mcp-adapter/mcp-keyring-helper.cjs +88 -0
- package/extensions/dm-mcp-adapter/mcp-oauth-provider.js +490 -0
- package/extensions/dm-mcp-adapter/mcp-output-guard.js +386 -0
- package/extensions/dm-mcp-adapter/mcp-panel.js +951 -0
- package/extensions/dm-mcp-adapter/mcp-probe.js +151 -0
- package/extensions/dm-mcp-adapter/mcp-references.js +263 -0
- package/extensions/dm-mcp-adapter/mcp-script-worker.mjs +126 -0
- package/extensions/dm-mcp-adapter/mcp-setup-panel.js +578 -0
- package/extensions/dm-mcp-adapter/mcp-status.js +80 -0
- package/extensions/dm-mcp-adapter/mcp-trace.js +215 -0
- package/extensions/dm-mcp-adapter/metadata-cache.js +292 -0
- package/extensions/dm-mcp-adapter/namespace-tools.js +155 -0
- package/extensions/dm-mcp-adapter/npx-resolver.js +465 -0
- package/extensions/dm-mcp-adapter/oauth-handler.js +17 -0
- package/extensions/dm-mcp-adapter/oauth.js +17 -0
- package/extensions/dm-mcp-adapter/onboarding-state.js +62 -0
- package/extensions/dm-mcp-adapter/package-mcp-loader.js +144 -0
- package/extensions/dm-mcp-adapter/package.json +68 -0
- package/extensions/dm-mcp-adapter/panel-keys.js +26 -0
- package/extensions/dm-mcp-adapter/prompts.js +258 -0
- package/extensions/dm-mcp-adapter/proxy-modes.js +1336 -0
- package/extensions/dm-mcp-adapter/request-headers-command.js +293 -0
- package/extensions/dm-mcp-adapter/resource-tools.js +7 -0
- package/extensions/dm-mcp-adapter/runtime-owner.js +79 -0
- package/extensions/dm-mcp-adapter/sampling-handler.js +222 -0
- package/extensions/dm-mcp-adapter/sandbox-proxy-template.js +190 -0
- package/extensions/dm-mcp-adapter/search-ranking.js +198 -0
- package/extensions/dm-mcp-adapter/server-manager.js +1258 -0
- package/extensions/dm-mcp-adapter/session-recovery.js +82 -0
- package/extensions/dm-mcp-adapter/skills/mcp-scripting/SKILL.md +39 -0
- package/extensions/dm-mcp-adapter/state.js +0 -0
- package/extensions/dm-mcp-adapter/test-runner.mjs +18 -0
- package/extensions/dm-mcp-adapter/tool-approval.js +151 -0
- package/extensions/dm-mcp-adapter/tool-metadata.js +247 -0
- package/extensions/dm-mcp-adapter/tool-registrar.js +212 -0
- package/extensions/dm-mcp-adapter/tool-result-renderer.js +355 -0
- package/extensions/dm-mcp-adapter/ts-shape.js +157 -0
- package/extensions/dm-mcp-adapter/types.js +264 -0
- package/extensions/dm-mcp-adapter/ui-app-bridge-helpers.js +36 -0
- package/extensions/dm-mcp-adapter/ui-resource-handler.js +209 -0
- package/extensions/dm-mcp-adapter/ui-server.js +815 -0
- package/extensions/dm-mcp-adapter/ui-session.js +455 -0
- package/extensions/dm-mcp-adapter/ui-stream-types.js +58 -0
- package/extensions/dm-mcp-adapter/ui-tool-visibility.js +26 -0
- package/extensions/dm-mcp-adapter/unix-socket-transport.js +83 -0
- package/extensions/dm-mcp-adapter/utils.js +307 -0
- package/extensions/dm-skills-manager/THIRD_PARTY_NOTICES.md +27 -0
- package/extensions/dm-skills-manager/extensions/skills-manager/components.js +265 -0
- package/extensions/dm-skills-manager/extensions/skills-manager/constants.js +31 -0
- package/extensions/dm-skills-manager/extensions/skills-manager/creation-fallback.js +20 -0
- package/extensions/dm-skills-manager/extensions/skills-manager/creation.js +145 -0
- package/extensions/dm-skills-manager/extensions/skills-manager/dialog.js +738 -0
- package/extensions/dm-skills-manager/extensions/skills-manager/dm-ai-compat.js +15 -0
- package/extensions/dm-skills-manager/extensions/skills-manager/format.js +125 -0
- package/extensions/dm-skills-manager/extensions/skills-manager/glyphs.js +142 -0
- package/extensions/dm-skills-manager/extensions/skills-manager/layout.js +93 -0
- package/extensions/dm-skills-manager/extensions/skills-manager/paths.js +76 -0
- package/extensions/dm-skills-manager/extensions/skills-manager/registry.js +95 -0
- package/extensions/dm-skills-manager/extensions/skills-manager/settings.js +93 -0
- package/extensions/dm-skills-manager/extensions/skills-manager/startup.js +32 -0
- package/extensions/dm-skills-manager/extensions/skills-manager/toggle.js +54 -0
- package/extensions/dm-skills-manager/extensions/skills-manager/types.js +14 -0
- package/extensions/dm-skills-manager/extensions/skills-manager/ui.js +121 -0
- package/extensions/dm-skills-manager/extensions/skills-manager.js +111 -0
- package/extensions/dm-skills-manager/package.json +121 -0
- package/package.json +13 -1
- package/extensions/dm-localllm-provider/detect.test.js +0 -1051
- package/extensions/dm-localllm-provider/index.test.js +0 -149
- package/extensions/dm-localllm-provider/keychain.test.js +0 -61
|
@@ -0,0 +1,311 @@
|
|
|
1
|
+
import { createHash } from "node:crypto";
|
|
2
|
+
import { createRequire } from "node:module";
|
|
3
|
+
import { dirname, join } from "node:path";
|
|
4
|
+
const require = createRequire(import.meta.url);
|
|
5
|
+
const BEARER_SECRET_SERVICE = "dm-mcp-adapter.bearer";
|
|
6
|
+
const TEST_AUTH_STORE_ENV = "DM_MCP_ADAPTER_TEST_AUTH_STORE";
|
|
7
|
+
const BEARER_SECRET_CHUNK_SIZE = 1000;
|
|
8
|
+
const BEARER_SECRET_VALUE_LIMIT = 1280;
|
|
9
|
+
let KeyringEntryClass;
|
|
10
|
+
const memoryBearerEntries = new Map;
|
|
11
|
+
let testBearerSecretStoreReadCount = 0;
|
|
12
|
+
|
|
13
|
+
export class BearerCredentialStoreError extends Error {
|
|
14
|
+
code = "BEARER_CREDENTIAL_STORE_UNAVAILABLE";
|
|
15
|
+
operation;
|
|
16
|
+
constructor(message, operation, cause) {
|
|
17
|
+
super(message, { cause });
|
|
18
|
+
this.name = "BearerCredentialStoreError";
|
|
19
|
+
this.operation = operation;
|
|
20
|
+
}
|
|
21
|
+
}
|
|
22
|
+
const memoryBearerSecretStore = {
|
|
23
|
+
read(account) {
|
|
24
|
+
testBearerSecretStoreReadCount++;
|
|
25
|
+
return memoryBearerEntries.get(account);
|
|
26
|
+
},
|
|
27
|
+
write(account, payload) {
|
|
28
|
+
memoryBearerEntries.set(account, payload);
|
|
29
|
+
},
|
|
30
|
+
remove(account) {
|
|
31
|
+
memoryBearerEntries.delete(account);
|
|
32
|
+
}
|
|
33
|
+
};
|
|
34
|
+
const sizeLimitedBearerSecretStore = {
|
|
35
|
+
read(account) {
|
|
36
|
+
testBearerSecretStoreReadCount++;
|
|
37
|
+
return memoryBearerEntries.get(account);
|
|
38
|
+
},
|
|
39
|
+
write(account, payload) {
|
|
40
|
+
if (payload.length > BEARER_SECRET_VALUE_LIMIT) {
|
|
41
|
+
throw new Error(`Value of 'password encoded as UTF-16' is longer than the platform limit of ${BEARER_SECRET_VALUE_LIMIT * 2} chars`);
|
|
42
|
+
}
|
|
43
|
+
memoryBearerEntries.set(account, payload);
|
|
44
|
+
},
|
|
45
|
+
remove(account) {
|
|
46
|
+
memoryBearerEntries.delete(account);
|
|
47
|
+
}
|
|
48
|
+
};
|
|
49
|
+
const unavailableBearerSecretStore = {
|
|
50
|
+
read() {
|
|
51
|
+
testBearerSecretStoreReadCount++;
|
|
52
|
+
throw new Error("simulated secure credential store unavailable");
|
|
53
|
+
},
|
|
54
|
+
write() {
|
|
55
|
+
throw new Error("simulated secure credential store unavailable");
|
|
56
|
+
},
|
|
57
|
+
remove() {
|
|
58
|
+
throw new Error("simulated secure credential store unavailable");
|
|
59
|
+
}
|
|
60
|
+
};
|
|
61
|
+
const keyringBearerSecretStore = {
|
|
62
|
+
read(account) {
|
|
63
|
+
return getKeyringEntry(account).getPassword() ?? undefined;
|
|
64
|
+
},
|
|
65
|
+
write(account, payload) {
|
|
66
|
+
getKeyringEntry(account).setPassword(payload);
|
|
67
|
+
},
|
|
68
|
+
remove(account) {
|
|
69
|
+
getKeyringEntry(account).deleteCredential();
|
|
70
|
+
}
|
|
71
|
+
};
|
|
72
|
+
function getBearerSecretStore() {
|
|
73
|
+
if (process.env[TEST_AUTH_STORE_ENV] === "memory")
|
|
74
|
+
return memoryBearerSecretStore;
|
|
75
|
+
if (process.env[TEST_AUTH_STORE_ENV] === "sizelimited")
|
|
76
|
+
return sizeLimitedBearerSecretStore;
|
|
77
|
+
if (process.env[TEST_AUTH_STORE_ENV] === "unavailable")
|
|
78
|
+
return unavailableBearerSecretStore;
|
|
79
|
+
return keyringBearerSecretStore;
|
|
80
|
+
}
|
|
81
|
+
function getKeyringEntry(account) {
|
|
82
|
+
try {
|
|
83
|
+
KeyringEntryClass ??= loadKeyringEntryClass();
|
|
84
|
+
return new KeyringEntryClass(BEARER_SECRET_SERVICE, account);
|
|
85
|
+
} catch (error) {
|
|
86
|
+
throw new Error("Bearer token secure credential storage is unavailable. Configure the OS credential store and retry.", { cause: error });
|
|
87
|
+
}
|
|
88
|
+
}
|
|
89
|
+
function loadKeyringEntryClass(keyringRequire = require, platform = process.platform, arch = process.arch) {
|
|
90
|
+
try {
|
|
91
|
+
return keyringRequire("@napi-rs/keyring").Entry;
|
|
92
|
+
} catch (loaderError) {
|
|
93
|
+
try {
|
|
94
|
+
return loadKeyringNativeBindingFallback(keyringRequire, platform, arch).Entry;
|
|
95
|
+
} catch (fallbackError) {
|
|
96
|
+
const message = fallbackError instanceof Error ? fallbackError.message : String(fallbackError);
|
|
97
|
+
throw new Error(`Failed to load @napi-rs/keyring; absolute-path native binding fallback also failed: ${message}`, { cause: loaderError });
|
|
98
|
+
}
|
|
99
|
+
}
|
|
100
|
+
}
|
|
101
|
+
function loadKeyringNativeBindingFallback(keyringRequire, platform, arch) {
|
|
102
|
+
const targets = getKeyringNativeBindingSuffixes(platform, arch).map((suffix) => ({
|
|
103
|
+
packageName: `@napi-rs/keyring-${suffix}`,
|
|
104
|
+
bindingFile: `keyring.${suffix}.node`
|
|
105
|
+
}));
|
|
106
|
+
let lastError;
|
|
107
|
+
for (const target of targets) {
|
|
108
|
+
try {
|
|
109
|
+
const packageJsonPath = keyringRequire.resolve(`${target.packageName}/package.json`);
|
|
110
|
+
return keyringRequire(join(dirname(packageJsonPath), target.bindingFile));
|
|
111
|
+
} catch (error) {
|
|
112
|
+
lastError = error;
|
|
113
|
+
}
|
|
114
|
+
}
|
|
115
|
+
throw lastError instanceof Error ? lastError : new Error(String(lastError));
|
|
116
|
+
}
|
|
117
|
+
function getKeyringNativeBindingSuffixes(platform, arch) {
|
|
118
|
+
if (platform === "darwin") {
|
|
119
|
+
if (arch === "arm64")
|
|
120
|
+
return ["darwin-arm64"];
|
|
121
|
+
if (arch === "x64")
|
|
122
|
+
return ["darwin-x64"];
|
|
123
|
+
}
|
|
124
|
+
if (platform === "win32") {
|
|
125
|
+
if (arch === "arm64")
|
|
126
|
+
return ["win32-arm64-msvc"];
|
|
127
|
+
if (arch === "x64")
|
|
128
|
+
return ["win32-x64-msvc"];
|
|
129
|
+
if (arch === "ia32")
|
|
130
|
+
return ["win32-ia32-msvc"];
|
|
131
|
+
}
|
|
132
|
+
if (platform === "linux") {
|
|
133
|
+
if (arch === "arm64")
|
|
134
|
+
return ["linux-arm64-gnu", "linux-arm64-musl"];
|
|
135
|
+
if (arch === "arm")
|
|
136
|
+
return ["linux-arm-gnueabihf"];
|
|
137
|
+
if (arch === "riscv64")
|
|
138
|
+
return ["linux-riscv64-gnu"];
|
|
139
|
+
if (arch === "x64")
|
|
140
|
+
return ["linux-x64-gnu", "linux-x64-musl"];
|
|
141
|
+
}
|
|
142
|
+
if (platform === "freebsd" && arch === "x64")
|
|
143
|
+
return ["freebsd-x64"];
|
|
144
|
+
return [];
|
|
145
|
+
}
|
|
146
|
+
function getBearerAccount(serverName) {
|
|
147
|
+
if (typeof serverName !== "string") {
|
|
148
|
+
throw new Error(`Invalid MCP server name: ${JSON.stringify(serverName)}`);
|
|
149
|
+
}
|
|
150
|
+
return `sha256-${createHash("sha256").update(serverName, "utf8").digest("hex")}`;
|
|
151
|
+
}
|
|
152
|
+
function parseBearerPayload(serverName, payload) {
|
|
153
|
+
const parsed = parseStoredBearerJson(serverName, payload);
|
|
154
|
+
if (typeof parsed !== "object" || parsed === null || Array.isArray(parsed)) {
|
|
155
|
+
throw new Error(`Stored bearer token record for ${serverName} has invalid shape`);
|
|
156
|
+
}
|
|
157
|
+
const record = parsed;
|
|
158
|
+
if (typeof record.token !== "string" || typeof record.serverUrl !== "string") {
|
|
159
|
+
throw new Error(`Stored bearer token record for ${serverName} has invalid shape`);
|
|
160
|
+
}
|
|
161
|
+
return { token: record.token, serverUrl: record.serverUrl };
|
|
162
|
+
}
|
|
163
|
+
function parseStoredBearerJson(serverName, payload) {
|
|
164
|
+
try {
|
|
165
|
+
return JSON.parse(payload);
|
|
166
|
+
} catch {
|
|
167
|
+
throw new Error(`Failed to parse stored bearer token record for ${serverName}`);
|
|
168
|
+
}
|
|
169
|
+
}
|
|
170
|
+
function isBearerChunkManifest(value) {
|
|
171
|
+
if (typeof value !== "object" || value === null)
|
|
172
|
+
return false;
|
|
173
|
+
const manifest = value;
|
|
174
|
+
return manifest.__piMcpAdapterBearerChunked === 1 && typeof manifest.chunkCount === "number" && Number.isInteger(manifest.chunkCount) && manifest.chunkCount > 0 && typeof manifest.chunkDigest === "string" && /^[a-f0-9]{16}$/.test(manifest.chunkDigest);
|
|
175
|
+
}
|
|
176
|
+
function readManifest(serverName, payload) {
|
|
177
|
+
const parsed = parseStoredBearerJson(serverName, payload);
|
|
178
|
+
return isBearerChunkManifest(parsed) ? parsed : undefined;
|
|
179
|
+
}
|
|
180
|
+
function chunkAccount(account, manifest, index) {
|
|
181
|
+
return `${account}.chunk.${manifest.chunkDigest}.${index}`;
|
|
182
|
+
}
|
|
183
|
+
function chunkAccounts(account, manifest) {
|
|
184
|
+
return Array.from({ length: manifest.chunkCount }, (_, index) => chunkAccount(account, manifest, index));
|
|
185
|
+
}
|
|
186
|
+
function createManifest(payload) {
|
|
187
|
+
return {
|
|
188
|
+
__piMcpAdapterBearerChunked: 1,
|
|
189
|
+
chunkCount: Math.ceil(payload.length / BEARER_SECRET_CHUNK_SIZE),
|
|
190
|
+
chunkDigest: createHash("sha256").update(payload, "utf8").digest("hex").slice(0, 16)
|
|
191
|
+
};
|
|
192
|
+
}
|
|
193
|
+
function readExistingManifest(store, serverName, account) {
|
|
194
|
+
try {
|
|
195
|
+
const payload = store.read(account);
|
|
196
|
+
return payload === undefined ? undefined : readManifest(serverName, payload);
|
|
197
|
+
} catch {
|
|
198
|
+
return;
|
|
199
|
+
}
|
|
200
|
+
}
|
|
201
|
+
function removeChunks(store, account, manifest) {
|
|
202
|
+
if (!manifest)
|
|
203
|
+
return;
|
|
204
|
+
for (const accountName of chunkAccounts(account, manifest))
|
|
205
|
+
store.remove(accountName);
|
|
206
|
+
}
|
|
207
|
+
function tryRemoveChunks(store, account, manifest) {
|
|
208
|
+
try {
|
|
209
|
+
removeChunks(store, account, manifest);
|
|
210
|
+
} catch {}
|
|
211
|
+
}
|
|
212
|
+
function readBearerRecordFromStore(store, serverName) {
|
|
213
|
+
const account = getBearerAccount(serverName);
|
|
214
|
+
let payload;
|
|
215
|
+
try {
|
|
216
|
+
payload = store.read(account);
|
|
217
|
+
} catch (error) {
|
|
218
|
+
throw new BearerCredentialStoreError(`Failed to read bearer token for ${serverName} from the OS secure credential store`, "read", error);
|
|
219
|
+
}
|
|
220
|
+
if (payload === undefined)
|
|
221
|
+
return;
|
|
222
|
+
try {
|
|
223
|
+
const manifest = readManifest(serverName, payload);
|
|
224
|
+
const recordPayload = manifest ? chunkAccounts(account, manifest).map((chunkName) => {
|
|
225
|
+
const chunk = store.read(chunkName);
|
|
226
|
+
if (chunk === undefined)
|
|
227
|
+
throw new Error(`Missing bearer token chunk for ${serverName}`);
|
|
228
|
+
return chunk;
|
|
229
|
+
}).join("") : payload;
|
|
230
|
+
return parseBearerPayload(serverName, recordPayload);
|
|
231
|
+
} catch (error) {
|
|
232
|
+
throw new BearerCredentialStoreError(`Failed to read bearer token for ${serverName} from the OS secure credential store`, "read", error);
|
|
233
|
+
}
|
|
234
|
+
}
|
|
235
|
+
function writeBearerRecordToStore(store, serverName, record) {
|
|
236
|
+
const account = getBearerAccount(serverName);
|
|
237
|
+
const payload = JSON.stringify(record);
|
|
238
|
+
const previousManifest = readExistingManifest(store, serverName, account);
|
|
239
|
+
const manifest = payload.length > BEARER_SECRET_CHUNK_SIZE ? createManifest(payload) : undefined;
|
|
240
|
+
try {
|
|
241
|
+
if (manifest) {
|
|
242
|
+
for (let index = 0;index < manifest.chunkCount; index++) {
|
|
243
|
+
store.write(chunkAccount(account, manifest, index), payload.slice(index * BEARER_SECRET_CHUNK_SIZE, (index + 1) * BEARER_SECRET_CHUNK_SIZE));
|
|
244
|
+
}
|
|
245
|
+
store.write(account, JSON.stringify(manifest));
|
|
246
|
+
} else {
|
|
247
|
+
store.write(account, payload);
|
|
248
|
+
}
|
|
249
|
+
if (previousManifest?.chunkDigest !== manifest?.chunkDigest) {
|
|
250
|
+
tryRemoveChunks(store, account, previousManifest);
|
|
251
|
+
}
|
|
252
|
+
} catch (error) {
|
|
253
|
+
if (previousManifest?.chunkDigest !== manifest?.chunkDigest) {
|
|
254
|
+
tryRemoveChunks(store, account, manifest);
|
|
255
|
+
}
|
|
256
|
+
throw new BearerCredentialStoreError(`Failed to write bearer token for ${serverName} to the OS secure credential store`, "write", error);
|
|
257
|
+
}
|
|
258
|
+
}
|
|
259
|
+
function removeBearerRecordFromStore(store, serverName) {
|
|
260
|
+
const account = getBearerAccount(serverName);
|
|
261
|
+
try {
|
|
262
|
+
const payload = store.read(account);
|
|
263
|
+
const manifest = payload === undefined ? undefined : readManifest(serverName, payload);
|
|
264
|
+
removeChunks(store, account, manifest);
|
|
265
|
+
store.remove(account);
|
|
266
|
+
} catch (error) {
|
|
267
|
+
throw new BearerCredentialStoreError(`Failed to remove bearer token for ${serverName} from the OS secure credential store`, "remove", error);
|
|
268
|
+
}
|
|
269
|
+
}
|
|
270
|
+
export function getBearerTokenForUrl(serverName, serverUrl) {
|
|
271
|
+
const record = readBearerRecordFromStore(getBearerSecretStore(), serverName);
|
|
272
|
+
if (!record)
|
|
273
|
+
return;
|
|
274
|
+
return record.serverUrl === serverUrl ? record.token : undefined;
|
|
275
|
+
}
|
|
276
|
+
export function saveBearerTokenForUrl(serverName, token, serverUrl) {
|
|
277
|
+
writeBearerRecordToStore(getBearerSecretStore(), serverName, { token, serverUrl });
|
|
278
|
+
}
|
|
279
|
+
export function removeBearerToken(serverName) {
|
|
280
|
+
removeBearerRecordFromStore(getBearerSecretStore(), serverName);
|
|
281
|
+
}
|
|
282
|
+
export function inspectBearerTokenForUrl(serverName, serverUrl) {
|
|
283
|
+
try {
|
|
284
|
+
const record = readBearerRecordFromStore(getBearerSecretStore(), serverName);
|
|
285
|
+
if (!record)
|
|
286
|
+
return { status: "missing" };
|
|
287
|
+
if (record.serverUrl !== serverUrl)
|
|
288
|
+
return { status: "url-mismatch" };
|
|
289
|
+
return { status: "present" };
|
|
290
|
+
} catch (error) {
|
|
291
|
+
if (!(error instanceof BearerCredentialStoreError))
|
|
292
|
+
throw error;
|
|
293
|
+
return { status: "unavailable", message: "Bearer token secure credential store unavailable. Configure or unlock the OS credential store and retry." };
|
|
294
|
+
}
|
|
295
|
+
}
|
|
296
|
+
export function resetTestBearerTokenStore() {
|
|
297
|
+
memoryBearerEntries.clear();
|
|
298
|
+
testBearerSecretStoreReadCount = 0;
|
|
299
|
+
}
|
|
300
|
+
export function getTestBearerTokenStoreEntries() {
|
|
301
|
+
return [...memoryBearerEntries.entries()];
|
|
302
|
+
}
|
|
303
|
+
export function removeTestBearerTokenStoreEntry(account) {
|
|
304
|
+
memoryBearerEntries.delete(account);
|
|
305
|
+
}
|
|
306
|
+
export function setTestBearerTokenStoreEntry(account, payload) {
|
|
307
|
+
memoryBearerEntries.set(account, payload);
|
|
308
|
+
}
|
|
309
|
+
export function getTestBearerTokenStoreReadCount() {
|
|
310
|
+
return testBearerSecretStoreReadCount;
|
|
311
|
+
}
|
|
@@ -0,0 +1,285 @@
|
|
|
1
|
+
import { existsSync, readFileSync, writeFileSync, renameSync, mkdirSync } from "node:fs";
|
|
2
|
+
import { dirname } from "node:path";
|
|
3
|
+
import { getAgentPath } from "./agent-dir.js";
|
|
4
|
+
import { createHash } from "node:crypto";
|
|
5
|
+
import { getToolUiResourceUri } from "./ui-app-bridge-helpers.js";
|
|
6
|
+
import { createToolSelectorCandidateIndex, formatPromptCommandName, formatToolName, getToolNameCandidates, isServerDisabled, isToolAllowed, resolveToolPrefix } from "./types.js";
|
|
7
|
+
import { resourceNameToToolName } from "./resource-tools.js";
|
|
8
|
+
import { extractToolUiStreamMode, interpolateEnvRecord, interpolateEnvVars, resolveBearerToken, resolveConfigPath, resolveServerUrl } from "./utils.js";
|
|
9
|
+
import { extractUiToolVisibility, isUiToolVisibleToModel } from "./ui-tool-visibility.js";
|
|
10
|
+
const CACHE_VERSION = 1;
|
|
11
|
+
const CACHE_MAX_AGE_MS = 7 * 24 * 60 * 60 * 1000;
|
|
12
|
+
export function getMetadataCachePath() {
|
|
13
|
+
return getAgentPath("mcp-cache.json");
|
|
14
|
+
}
|
|
15
|
+
export function loadMetadataCache() {
|
|
16
|
+
const cachePath = getMetadataCachePath();
|
|
17
|
+
if (!existsSync(cachePath))
|
|
18
|
+
return null;
|
|
19
|
+
try {
|
|
20
|
+
const raw = JSON.parse(readFileSync(cachePath, "utf-8"));
|
|
21
|
+
if (!raw || typeof raw !== "object")
|
|
22
|
+
return null;
|
|
23
|
+
if (raw.version !== CACHE_VERSION)
|
|
24
|
+
return null;
|
|
25
|
+
if (!raw.servers || typeof raw.servers !== "object")
|
|
26
|
+
return null;
|
|
27
|
+
return raw;
|
|
28
|
+
} catch {
|
|
29
|
+
return null;
|
|
30
|
+
}
|
|
31
|
+
}
|
|
32
|
+
export function saveMetadataCache(cache) {
|
|
33
|
+
const cachePath = getMetadataCachePath();
|
|
34
|
+
const dir = dirname(cachePath);
|
|
35
|
+
mkdirSync(dir, { recursive: true });
|
|
36
|
+
let merged = { version: CACHE_VERSION, servers: {} };
|
|
37
|
+
try {
|
|
38
|
+
if (existsSync(cachePath)) {
|
|
39
|
+
const existing = JSON.parse(readFileSync(cachePath, "utf-8"));
|
|
40
|
+
if (existing && existing.version === CACHE_VERSION && existing.servers) {
|
|
41
|
+
merged.servers = { ...existing.servers };
|
|
42
|
+
}
|
|
43
|
+
}
|
|
44
|
+
} catch {}
|
|
45
|
+
merged.version = CACHE_VERSION;
|
|
46
|
+
merged.servers = { ...merged.servers, ...cache.servers };
|
|
47
|
+
const tmpPath = `${cachePath}.${process.pid}.tmp`;
|
|
48
|
+
writeFileSync(tmpPath, JSON.stringify(merged), "utf-8");
|
|
49
|
+
renameSync(tmpPath, cachePath);
|
|
50
|
+
}
|
|
51
|
+
export function computeServerHash(definition, environment = process.env) {
|
|
52
|
+
const identity = {
|
|
53
|
+
command: definition.command,
|
|
54
|
+
args: definition.args,
|
|
55
|
+
socket: resolveConfigPath(definition.socket, environment),
|
|
56
|
+
env: interpolateEnvRecord(definition.env, environment),
|
|
57
|
+
cwd: resolveConfigPath(definition.cwd, environment),
|
|
58
|
+
url: resolveServerUrl(definition, environment),
|
|
59
|
+
headers: interpolateEnvRecord(definition.headers, environment),
|
|
60
|
+
requestHeadersCommand: definition.requestHeadersCommand ? {
|
|
61
|
+
command: interpolateEnvVars(definition.requestHeadersCommand.command, environment),
|
|
62
|
+
args: definition.requestHeadersCommand.args?.map((argument) => interpolateEnvVars(argument, environment)),
|
|
63
|
+
env: interpolateEnvRecord(definition.requestHeadersCommand.env, environment),
|
|
64
|
+
timeoutMs: definition.requestHeadersCommand.timeoutMs
|
|
65
|
+
} : undefined,
|
|
66
|
+
auth: definition.auth,
|
|
67
|
+
protocolVersion: definition.protocolVersion,
|
|
68
|
+
bearerToken: resolveBearerToken(definition, environment),
|
|
69
|
+
bearerTokenEnv: definition.bearerTokenEnv,
|
|
70
|
+
exposeResources: definition.exposeResources,
|
|
71
|
+
includeTools: definition.includeTools,
|
|
72
|
+
excludeTools: definition.excludeTools
|
|
73
|
+
};
|
|
74
|
+
const normalized = stableStringify(identity);
|
|
75
|
+
return createHash("sha256").update(normalized).digest("hex");
|
|
76
|
+
}
|
|
77
|
+
export function isServerCacheValid(entry, definition, maxAgeMs = CACHE_MAX_AGE_MS, environment = process.env) {
|
|
78
|
+
let configHash;
|
|
79
|
+
try {
|
|
80
|
+
configHash = computeServerHash(definition, environment);
|
|
81
|
+
} catch {
|
|
82
|
+
return false;
|
|
83
|
+
}
|
|
84
|
+
if (!entry || entry.configHash !== configHash)
|
|
85
|
+
return false;
|
|
86
|
+
if (!entry.cachedAt || typeof entry.cachedAt !== "number")
|
|
87
|
+
return false;
|
|
88
|
+
const declaredTtlMs = entry.ttlMs;
|
|
89
|
+
if (typeof declaredTtlMs === "number" && Number.isSafeInteger(declaredTtlMs) && declaredTtlMs >= 0) {
|
|
90
|
+
if (declaredTtlMs === 0)
|
|
91
|
+
return false;
|
|
92
|
+
const ageMs = Date.now() - entry.cachedAt;
|
|
93
|
+
const effectiveMaxAge = maxAgeMs > 0 ? Math.min(maxAgeMs, declaredTtlMs) : declaredTtlMs;
|
|
94
|
+
return ageMs < effectiveMaxAge;
|
|
95
|
+
}
|
|
96
|
+
if (maxAgeMs > 0 && Date.now() - entry.cachedAt > maxAgeMs)
|
|
97
|
+
return false;
|
|
98
|
+
return true;
|
|
99
|
+
}
|
|
100
|
+
export function parseDirectToolSelectors(selectors) {
|
|
101
|
+
const servers = new Set;
|
|
102
|
+
const tools = new Map;
|
|
103
|
+
for (let selector of selectors) {
|
|
104
|
+
selector = selector.replace(/\/+$/, "");
|
|
105
|
+
if (selector.includes("/")) {
|
|
106
|
+
const [server, tool] = selector.split("/", 2);
|
|
107
|
+
if (server && tool) {
|
|
108
|
+
const serverTools = tools.get(server) ?? new Set;
|
|
109
|
+
serverTools.add(tool);
|
|
110
|
+
tools.set(server, serverTools);
|
|
111
|
+
} else if (server) {
|
|
112
|
+
servers.add(server);
|
|
113
|
+
}
|
|
114
|
+
} else if (selector) {
|
|
115
|
+
servers.add(selector);
|
|
116
|
+
}
|
|
117
|
+
}
|
|
118
|
+
return { servers, tools };
|
|
119
|
+
}
|
|
120
|
+
export function getMissingConfiguredDirectToolServers(config, cache, envOverride) {
|
|
121
|
+
const missing = [];
|
|
122
|
+
const globalDirect = config.settings?.directTools;
|
|
123
|
+
const envSelection = envOverride ? parseDirectToolSelectors(envOverride) : null;
|
|
124
|
+
for (const [serverName, definition] of Object.entries(config.mcpServers)) {
|
|
125
|
+
if (isServerDisabled(definition))
|
|
126
|
+
continue;
|
|
127
|
+
const hasDirectTools = envSelection ? envSelection.servers.has(serverName) || envSelection.tools.has(serverName) : definition.directTools !== undefined ? !!definition.directTools : !!globalDirect;
|
|
128
|
+
if (!hasDirectTools)
|
|
129
|
+
continue;
|
|
130
|
+
const serverCache = cache?.servers?.[serverName];
|
|
131
|
+
if (!serverCache || !isServerCacheValid(serverCache, definition)) {
|
|
132
|
+
missing.push(serverName);
|
|
133
|
+
}
|
|
134
|
+
}
|
|
135
|
+
return missing;
|
|
136
|
+
}
|
|
137
|
+
export function reconstructToolMetadata(serverName, entry, prefix, definition, configuredServers, cache, sharedSelectorCandidateIndex) {
|
|
138
|
+
const metadata = [];
|
|
139
|
+
const seenNames = new Set;
|
|
140
|
+
const effectivePrefix = resolveToolPrefix(definition, prefix);
|
|
141
|
+
const hasToolFilters = Array.isArray(definition.includeTools) && definition.includeTools.length > 0 || Array.isArray(definition.excludeTools) && definition.excludeTools.length > 0;
|
|
142
|
+
const selectorCandidateIndex = hasToolFilters ? sharedSelectorCandidateIndex ?? (configuredServers && cache ? createCachedToolSelectorCandidateIndex(configuredServers, cache, prefix) : undefined) : undefined;
|
|
143
|
+
for (const tool of entry.tools ?? []) {
|
|
144
|
+
if (!tool?.name)
|
|
145
|
+
continue;
|
|
146
|
+
if (!isUiToolVisibleToModel(tool.uiVisibility)) {
|
|
147
|
+
continue;
|
|
148
|
+
}
|
|
149
|
+
if (!isToolAllowed(tool.name, serverName, effectivePrefix, definition.includeTools, definition.excludeTools, selectorCandidateIndex)) {
|
|
150
|
+
continue;
|
|
151
|
+
}
|
|
152
|
+
const name = formatToolName(tool.name, serverName, effectivePrefix);
|
|
153
|
+
if (seenNames.has(name)) {
|
|
154
|
+
continue;
|
|
155
|
+
}
|
|
156
|
+
seenNames.add(name);
|
|
157
|
+
metadata.push({
|
|
158
|
+
name,
|
|
159
|
+
originalName: tool.name,
|
|
160
|
+
description: tool.description ?? "",
|
|
161
|
+
...tool.inputSchema !== undefined ? { inputSchema: tool.inputSchema } : {},
|
|
162
|
+
...tool.uiResourceUri !== undefined ? { uiResourceUri: tool.uiResourceUri } : {},
|
|
163
|
+
...tool.uiVisibility !== undefined ? { uiVisibility: tool.uiVisibility } : {},
|
|
164
|
+
...tool.uiStreamMode !== undefined ? { uiStreamMode: tool.uiStreamMode } : {}
|
|
165
|
+
});
|
|
166
|
+
}
|
|
167
|
+
if (definition.exposeResources !== false) {
|
|
168
|
+
for (const resource of entry.resources ?? []) {
|
|
169
|
+
if (!resource?.name || !resource?.uri)
|
|
170
|
+
continue;
|
|
171
|
+
const baseName = `read_${resourceNameToToolName(resource.name)}`;
|
|
172
|
+
if (!isToolAllowed(baseName, serverName, effectivePrefix, definition.includeTools, definition.excludeTools, selectorCandidateIndex)) {
|
|
173
|
+
continue;
|
|
174
|
+
}
|
|
175
|
+
const name = formatToolName(baseName, serverName, effectivePrefix);
|
|
176
|
+
if (seenNames.has(name)) {
|
|
177
|
+
continue;
|
|
178
|
+
}
|
|
179
|
+
seenNames.add(name);
|
|
180
|
+
metadata.push({
|
|
181
|
+
name,
|
|
182
|
+
originalName: baseName,
|
|
183
|
+
description: resource.description ?? `Read resource: ${resource.uri}`,
|
|
184
|
+
resourceUri: resource.uri
|
|
185
|
+
});
|
|
186
|
+
}
|
|
187
|
+
}
|
|
188
|
+
return metadata;
|
|
189
|
+
}
|
|
190
|
+
export function createCachedToolSelectorCandidateIndex(configuredServers, cache, prefix) {
|
|
191
|
+
const candidates = new Set;
|
|
192
|
+
for (const [serverName, definition] of Object.entries(configuredServers)) {
|
|
193
|
+
const entry = cache.servers[serverName];
|
|
194
|
+
if (!entry || !isServerCacheValid(entry, definition) || isServerDisabled(definition))
|
|
195
|
+
continue;
|
|
196
|
+
const effectivePrefix = resolveToolPrefix(definition, prefix);
|
|
197
|
+
for (const tool of entry.tools ?? []) {
|
|
198
|
+
if (!isUiToolVisibleToModel(tool.uiVisibility))
|
|
199
|
+
continue;
|
|
200
|
+
for (const candidate of getToolNameCandidates(tool.name, serverName, effectivePrefix, false))
|
|
201
|
+
candidates.add(candidate);
|
|
202
|
+
}
|
|
203
|
+
if (definition.exposeResources !== false) {
|
|
204
|
+
for (const resource of entry.resources ?? []) {
|
|
205
|
+
const baseName = `read_${resourceNameToToolName(resource.name)}`;
|
|
206
|
+
for (const candidate of getToolNameCandidates(baseName, serverName, effectivePrefix, false))
|
|
207
|
+
candidates.add(candidate);
|
|
208
|
+
}
|
|
209
|
+
}
|
|
210
|
+
}
|
|
211
|
+
return createToolSelectorCandidateIndex(candidates);
|
|
212
|
+
}
|
|
213
|
+
export function serializeTools(tools) {
|
|
214
|
+
return tools.filter((t) => t?.name).map((t) => {
|
|
215
|
+
const uiResourceUri = tryGetToolUiResourceUri(t);
|
|
216
|
+
const uiVisibility = extractUiToolVisibility(t._meta);
|
|
217
|
+
const uiStreamMode = extractToolUiStreamMode(t._meta);
|
|
218
|
+
return {
|
|
219
|
+
name: t.name,
|
|
220
|
+
...t.description !== undefined ? { description: t.description } : {},
|
|
221
|
+
...t.inputSchema !== undefined ? { inputSchema: t.inputSchema } : {},
|
|
222
|
+
...uiResourceUri !== undefined ? { uiResourceUri } : {},
|
|
223
|
+
...uiVisibility !== undefined ? { uiVisibility } : {},
|
|
224
|
+
...uiStreamMode !== undefined ? { uiStreamMode } : {}
|
|
225
|
+
};
|
|
226
|
+
});
|
|
227
|
+
}
|
|
228
|
+
export function serializeResources(resources) {
|
|
229
|
+
return resources.filter((r) => r?.name && r?.uri).map((r) => ({
|
|
230
|
+
uri: r.uri,
|
|
231
|
+
name: r.name,
|
|
232
|
+
...r.description !== undefined ? { description: r.description } : {}
|
|
233
|
+
}));
|
|
234
|
+
}
|
|
235
|
+
export function serializePrompts(prompts) {
|
|
236
|
+
return (prompts ?? []).filter((prompt) => prompt?.name).map((prompt) => ({
|
|
237
|
+
name: prompt.name,
|
|
238
|
+
...prompt.title !== undefined ? { title: prompt.title } : {},
|
|
239
|
+
...prompt.description !== undefined ? { description: prompt.description } : {},
|
|
240
|
+
...Array.isArray(prompt.arguments) ? {
|
|
241
|
+
arguments: prompt.arguments.filter((argument) => argument?.name).map((argument) => ({
|
|
242
|
+
name: argument.name,
|
|
243
|
+
...argument.description !== undefined ? { description: argument.description } : {},
|
|
244
|
+
...argument.required !== undefined ? { required: argument.required } : {}
|
|
245
|
+
}))
|
|
246
|
+
} : {}
|
|
247
|
+
}));
|
|
248
|
+
}
|
|
249
|
+
export function reconstructPromptMetadata(serverName, prompts, prefix, definition) {
|
|
250
|
+
const effectivePrefix = resolveToolPrefix(definition, prefix);
|
|
251
|
+
return (prompts ?? []).filter((prompt) => prompt?.name).map((prompt) => {
|
|
252
|
+
const args = Array.isArray(prompt.arguments) ? prompt.arguments.filter((argument) => argument?.name).map((argument) => ({
|
|
253
|
+
name: argument.name,
|
|
254
|
+
...argument.description !== undefined ? { description: argument.description } : {},
|
|
255
|
+
...argument.required !== undefined ? { required: argument.required } : {}
|
|
256
|
+
})) : [];
|
|
257
|
+
return {
|
|
258
|
+
serverName,
|
|
259
|
+
originalName: prompt.name,
|
|
260
|
+
commandName: formatPromptCommandName(prompt.name, serverName, effectivePrefix),
|
|
261
|
+
...prompt.title !== undefined ? { title: prompt.title } : {},
|
|
262
|
+
description: prompt.description ?? "",
|
|
263
|
+
arguments: args
|
|
264
|
+
};
|
|
265
|
+
});
|
|
266
|
+
}
|
|
267
|
+
function stableStringify(value) {
|
|
268
|
+
if (value === null || value === undefined || typeof value !== "object") {
|
|
269
|
+
const serialized = JSON.stringify(value);
|
|
270
|
+
return serialized === undefined ? "undefined" : serialized;
|
|
271
|
+
}
|
|
272
|
+
if (Array.isArray(value)) {
|
|
273
|
+
return `[${value.map((v) => stableStringify(v)).join(",")}]`;
|
|
274
|
+
}
|
|
275
|
+
const obj = value;
|
|
276
|
+
const keys = Object.keys(obj).sort();
|
|
277
|
+
return `{${keys.map((k) => `${JSON.stringify(k)}:${stableStringify(obj[k])}`).join(",")}}`;
|
|
278
|
+
}
|
|
279
|
+
function tryGetToolUiResourceUri(tool) {
|
|
280
|
+
try {
|
|
281
|
+
return getToolUiResourceUri({ _meta: tool._meta });
|
|
282
|
+
} catch {
|
|
283
|
+
return;
|
|
284
|
+
}
|
|
285
|
+
}
|