@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,455 @@
|
|
|
1
|
+
import { execFile } from "node:child_process";
|
|
2
|
+
import { randomUUID } from "node:crypto";
|
|
3
|
+
import net from "node:net";
|
|
4
|
+
import { UrlElicitationRequiredError } from "@modelcontextprotocol/client";
|
|
5
|
+
import {
|
|
6
|
+
createUiModelContextUpdate,
|
|
7
|
+
extractUiPromptText,
|
|
8
|
+
UI_STREAM_HOST_CONTEXT_KEY,
|
|
9
|
+
UI_STREAM_REQUEST_META_KEY,
|
|
10
|
+
UI_STREAM_STRUCTURED_CONTENT_KEY
|
|
11
|
+
} from "./types.js";
|
|
12
|
+
import { logger } from "./logger.js";
|
|
13
|
+
import { startUiServer } from "./ui-server.js";
|
|
14
|
+
import { isGlimpseAvailable, openGlimpseWindow } from "./glimpse-ui.js";
|
|
15
|
+
import { combineAbortSignals, isAbortError } from "./runtime-owner.js";
|
|
16
|
+
import { throwIfAborted } from "./abort.js";
|
|
17
|
+
import { InputRequiredNeedsUiError } from "./errors.js";
|
|
18
|
+
let activeGlimpseWindow = null;
|
|
19
|
+
export function summarizeUiSessionResult(uiSession) {
|
|
20
|
+
if (!uiSession) {
|
|
21
|
+
return {
|
|
22
|
+
message: "Interactive UI was unavailable; returning the tool result inline.",
|
|
23
|
+
uiOpen: false
|
|
24
|
+
};
|
|
25
|
+
}
|
|
26
|
+
if (!uiSession.windowOpen) {
|
|
27
|
+
const action = uiSession.reused ? "Updated the suppressed MCP UI session." : "MCP UI window was suppressed.";
|
|
28
|
+
return {
|
|
29
|
+
message: `${action} Open manually: ${uiSession.url}`,
|
|
30
|
+
uiOpen: false,
|
|
31
|
+
uiViewer: uiSession.viewer,
|
|
32
|
+
uiUrl: uiSession.url
|
|
33
|
+
};
|
|
34
|
+
}
|
|
35
|
+
return {
|
|
36
|
+
message: uiSession.reused ? "Updated the open UI." : "Interactive UI is open. I'll respond to your prompts and intents as you interact with it.",
|
|
37
|
+
uiOpen: true,
|
|
38
|
+
uiViewer: uiSession.viewer,
|
|
39
|
+
uiUrl: uiSession.url
|
|
40
|
+
};
|
|
41
|
+
}
|
|
42
|
+
const MAX_COMPLETED_SESSIONS = 10;
|
|
43
|
+
function withStreamEnvelope(result, streamId, sequence) {
|
|
44
|
+
if (!streamId) {
|
|
45
|
+
return result;
|
|
46
|
+
}
|
|
47
|
+
const structuredContent = result.structuredContent && typeof result.structuredContent === "object" && !Array.isArray(result.structuredContent) ? { ...result.structuredContent } : {};
|
|
48
|
+
const rawEnvelope = structuredContent[UI_STREAM_STRUCTURED_CONTENT_KEY];
|
|
49
|
+
const envelope = rawEnvelope && typeof rawEnvelope === "object" && !Array.isArray(rawEnvelope) ? { ...rawEnvelope } : {
|
|
50
|
+
frameType: "final",
|
|
51
|
+
phase: "settled",
|
|
52
|
+
status: result.isError ? "error" : "ok"
|
|
53
|
+
};
|
|
54
|
+
structuredContent[UI_STREAM_STRUCTURED_CONTENT_KEY] = {
|
|
55
|
+
...envelope,
|
|
56
|
+
streamId,
|
|
57
|
+
sequence
|
|
58
|
+
};
|
|
59
|
+
return {
|
|
60
|
+
...result,
|
|
61
|
+
structuredContent
|
|
62
|
+
};
|
|
63
|
+
}
|
|
64
|
+
async function openInBrowser(state, url, signal) {
|
|
65
|
+
throwIfAborted(signal);
|
|
66
|
+
try {
|
|
67
|
+
await state.openBrowser(url);
|
|
68
|
+
throwIfAborted(signal);
|
|
69
|
+
return null;
|
|
70
|
+
} catch (error) {
|
|
71
|
+
if (isAbortError(error, signal))
|
|
72
|
+
throw error;
|
|
73
|
+
const message = error instanceof Error ? error.message : String(error);
|
|
74
|
+
if (state.owner?.isActive() === false)
|
|
75
|
+
return null;
|
|
76
|
+
state.ui?.notify(`MCP UI browser open failed: ${message}`, "warning");
|
|
77
|
+
return message;
|
|
78
|
+
}
|
|
79
|
+
}
|
|
80
|
+
function isRemoteSession() {
|
|
81
|
+
return Boolean(process.env.SSH_CONNECTION || process.env.SSH_TTY);
|
|
82
|
+
}
|
|
83
|
+
function hasActiveRemoteLogin() {
|
|
84
|
+
return new Promise((resolve) => {
|
|
85
|
+
execFile("who", { encoding: "utf8", timeout: 2000 }, (error, out) => {
|
|
86
|
+
if (error) {
|
|
87
|
+
resolve(false);
|
|
88
|
+
return;
|
|
89
|
+
}
|
|
90
|
+
resolve(/\(.+\)\s*$/m.test(out));
|
|
91
|
+
});
|
|
92
|
+
});
|
|
93
|
+
}
|
|
94
|
+
function probeMoshiGateway() {
|
|
95
|
+
return new Promise((resolve) => {
|
|
96
|
+
const socket = net.connect({ host: "127.0.0.1", port: 24543 });
|
|
97
|
+
const done = (ok) => {
|
|
98
|
+
socket.destroy();
|
|
99
|
+
resolve(ok);
|
|
100
|
+
};
|
|
101
|
+
socket.setTimeout(300);
|
|
102
|
+
socket.once("connect", () => done(true));
|
|
103
|
+
socket.once("timeout", () => done(false));
|
|
104
|
+
socket.once("error", () => done(false));
|
|
105
|
+
});
|
|
106
|
+
}
|
|
107
|
+
function remoteAccessHint(opts) {
|
|
108
|
+
const lines = [
|
|
109
|
+
opts.openError !== null ? "Couldn't open MCP UI here. Open it from your own device:" : opts.openedOnHost ? "MCP UI opened on this host. If you're controlling this session remotely, open it from your own device:" : "This looks like a remote session - if no MCP UI appeared, open it from your own device:",
|
|
110
|
+
` ${opts.url}`
|
|
111
|
+
];
|
|
112
|
+
if (opts.openError !== null) {
|
|
113
|
+
lines.push(`Browser launch failed: ${opts.openError}`);
|
|
114
|
+
}
|
|
115
|
+
if (opts.moshi) {
|
|
116
|
+
lines.push(`Moshi: tap the preview button in the terminal title bar and pick this MCP UI server (it must reach ports ${opts.port} and ${opts.proxyPort}).`);
|
|
117
|
+
}
|
|
118
|
+
lines.push(`SSH: run \`ssh -L ${opts.port}:127.0.0.1:${opts.port} -L ${opts.proxyPort}:127.0.0.1:${opts.proxyPort} <this-host>\` on your local machine, then open the URL above.`, "mosh can't forward ports - run that ssh command in a separate terminal.");
|
|
119
|
+
return lines.join(`
|
|
120
|
+
`);
|
|
121
|
+
}
|
|
122
|
+
export async function maybeStartUiSession(state, request) {
|
|
123
|
+
const log = logger.child({
|
|
124
|
+
component: "UiSession",
|
|
125
|
+
server: request.serverName,
|
|
126
|
+
tool: request.toolName
|
|
127
|
+
});
|
|
128
|
+
const runtimeSignal = combineAbortSignals(state.owner?.signal, request.signal) ?? new AbortController().signal;
|
|
129
|
+
try {
|
|
130
|
+
throwIfAborted(runtimeSignal);
|
|
131
|
+
if (state.uiServer && state.uiServer.serverName === request.serverName && state.uiServer.toolName === request.toolName) {
|
|
132
|
+
const existingHandle = state.uiServer;
|
|
133
|
+
const streamMode = request.streamMode;
|
|
134
|
+
const streamId = streamMode ? randomUUID() : undefined;
|
|
135
|
+
const streamToken = streamMode ? randomUUID() : undefined;
|
|
136
|
+
let active = true;
|
|
137
|
+
let nextStreamSequence = 0;
|
|
138
|
+
const cleanupStreamListener = () => {
|
|
139
|
+
if (streamToken) {
|
|
140
|
+
state.manager.removeUiStreamListener(streamToken);
|
|
141
|
+
}
|
|
142
|
+
};
|
|
143
|
+
existingHandle.sendToolInput(request.toolArgs);
|
|
144
|
+
if (streamToken) {
|
|
145
|
+
state.manager.registerUiStreamListener(streamToken, (serverName, notification) => {
|
|
146
|
+
if (!active || state.uiServer !== existingHandle)
|
|
147
|
+
return;
|
|
148
|
+
if (serverName !== request.serverName)
|
|
149
|
+
return;
|
|
150
|
+
nextStreamSequence += 1;
|
|
151
|
+
existingHandle.sendResultPatch(withStreamEnvelope(notification.result, streamId, nextStreamSequence));
|
|
152
|
+
});
|
|
153
|
+
}
|
|
154
|
+
return {
|
|
155
|
+
serverName: request.serverName,
|
|
156
|
+
toolName: request.toolName,
|
|
157
|
+
reused: true,
|
|
158
|
+
...streamId !== undefined ? { streamId } : {},
|
|
159
|
+
...streamToken !== undefined ? { streamToken } : {},
|
|
160
|
+
...streamMode !== undefined ? { streamMode } : {},
|
|
161
|
+
...streamToken ? { requestMeta: { [UI_STREAM_REQUEST_META_KEY]: streamToken } } : {},
|
|
162
|
+
url: existingHandle.url,
|
|
163
|
+
viewer: existingHandle.viewer ?? "browser",
|
|
164
|
+
windowOpen: existingHandle.windowOpen ?? true,
|
|
165
|
+
isActive: () => active && state.uiServer === existingHandle,
|
|
166
|
+
sendToolResult: (result) => {
|
|
167
|
+
if (!active || state.uiServer !== existingHandle)
|
|
168
|
+
return;
|
|
169
|
+
nextStreamSequence += 1;
|
|
170
|
+
existingHandle.sendToolResult(withStreamEnvelope(result, streamId, nextStreamSequence));
|
|
171
|
+
},
|
|
172
|
+
sendResultPatch: (result) => {
|
|
173
|
+
if (!active || state.uiServer !== existingHandle)
|
|
174
|
+
return;
|
|
175
|
+
nextStreamSequence += 1;
|
|
176
|
+
existingHandle.sendResultPatch(withStreamEnvelope(result, streamId, nextStreamSequence));
|
|
177
|
+
},
|
|
178
|
+
sendToolCancelled: (reason) => {
|
|
179
|
+
if (!active || state.uiServer !== existingHandle)
|
|
180
|
+
return;
|
|
181
|
+
nextStreamSequence += 1;
|
|
182
|
+
existingHandle.sendToolResult(withStreamEnvelope({
|
|
183
|
+
isError: true,
|
|
184
|
+
content: [{ type: "text", text: reason }]
|
|
185
|
+
}, streamId, nextStreamSequence));
|
|
186
|
+
},
|
|
187
|
+
close: () => {
|
|
188
|
+
active = false;
|
|
189
|
+
cleanupStreamListener();
|
|
190
|
+
}
|
|
191
|
+
};
|
|
192
|
+
}
|
|
193
|
+
const resource = await state.uiResourceHandler.readUiResource(request.serverName, request.uiResourceUri, {
|
|
194
|
+
config: state.config,
|
|
195
|
+
signal: runtimeSignal,
|
|
196
|
+
onNeedsAuth: request.onNeedsAuth
|
|
197
|
+
});
|
|
198
|
+
throwIfAborted(runtimeSignal);
|
|
199
|
+
if (state.uiServer) {
|
|
200
|
+
state.uiServer.close("replaced");
|
|
201
|
+
state.uiServer = null;
|
|
202
|
+
}
|
|
203
|
+
if (activeGlimpseWindow) {
|
|
204
|
+
activeGlimpseWindow.close();
|
|
205
|
+
activeGlimpseWindow = null;
|
|
206
|
+
}
|
|
207
|
+
const streamMode = request.streamMode;
|
|
208
|
+
const streamId = streamMode ? randomUUID() : undefined;
|
|
209
|
+
const streamToken = streamMode ? randomUUID() : undefined;
|
|
210
|
+
const hostContext = streamMode && streamId ? {
|
|
211
|
+
[UI_STREAM_HOST_CONTEXT_KEY]: {
|
|
212
|
+
mode: streamMode,
|
|
213
|
+
streamId,
|
|
214
|
+
intermediateResultPatches: streamMode === "stream-first",
|
|
215
|
+
partialInput: false
|
|
216
|
+
}
|
|
217
|
+
} : undefined;
|
|
218
|
+
let active = true;
|
|
219
|
+
let nextStreamSequence = 0;
|
|
220
|
+
let handle;
|
|
221
|
+
const resourceListenerToken = randomUUID();
|
|
222
|
+
const cleanupListeners = () => {
|
|
223
|
+
if (streamToken) {
|
|
224
|
+
state.manager.removeUiStreamListener(streamToken);
|
|
225
|
+
}
|
|
226
|
+
state.manager.removeResourceUpdatedListener?.(resourceListenerToken);
|
|
227
|
+
};
|
|
228
|
+
handle = await startUiServer({
|
|
229
|
+
serverName: request.serverName,
|
|
230
|
+
toolName: request.toolName,
|
|
231
|
+
toolArgs: streamMode === "stream-first" ? {} : request.toolArgs,
|
|
232
|
+
resource,
|
|
233
|
+
manager: state.manager,
|
|
234
|
+
config: state.config,
|
|
235
|
+
state,
|
|
236
|
+
...request.onNeedsAuth ? { onNeedsAuth: request.onNeedsAuth } : {},
|
|
237
|
+
consentManager: state.consentManager,
|
|
238
|
+
...hostContext !== undefined ? { hostContext } : {},
|
|
239
|
+
onMessage: (params) => {
|
|
240
|
+
const prompt = extractUiPromptText(params);
|
|
241
|
+
if (prompt) {
|
|
242
|
+
if (state.sendMessage) {
|
|
243
|
+
state.sendMessage({
|
|
244
|
+
customType: "mcp-ui-prompt",
|
|
245
|
+
content: [{ type: "text", text: `User sent prompt from ${request.serverName} UI: "${prompt}"` }],
|
|
246
|
+
display: `\uD83D\uDCAC UI Prompt: ${prompt}`,
|
|
247
|
+
details: { server: request.serverName, tool: request.toolName, prompt }
|
|
248
|
+
}, { triggerTurn: true });
|
|
249
|
+
log.debug("Triggered agent turn for UI prompt", { prompt: prompt.slice(0, 50) });
|
|
250
|
+
}
|
|
251
|
+
} else if (params.type === "intent" || params.intent) {
|
|
252
|
+
const intent = params.intent ?? "";
|
|
253
|
+
const intentParams = params.params;
|
|
254
|
+
if (intent && state.sendMessage) {
|
|
255
|
+
const paramsStr = intentParams ? ` ${JSON.stringify(intentParams)}` : "";
|
|
256
|
+
state.sendMessage({
|
|
257
|
+
customType: "mcp-ui-intent",
|
|
258
|
+
content: [{ type: "text", text: `User triggered intent from ${request.serverName} UI: ${intent}${paramsStr}` }],
|
|
259
|
+
display: `\uD83C\uDFAF UI Intent: ${intent}`,
|
|
260
|
+
details: { server: request.serverName, tool: request.toolName, intent, params: intentParams }
|
|
261
|
+
}, { triggerTurn: true });
|
|
262
|
+
log.debug("Triggered agent turn for UI intent", { intent });
|
|
263
|
+
}
|
|
264
|
+
} else if (params.type === "notify" || params.message) {
|
|
265
|
+
const text = params.message ?? "";
|
|
266
|
+
if (text && state.ui) {
|
|
267
|
+
state.ui.notify(`[${request.serverName}] ${text}`, "info");
|
|
268
|
+
}
|
|
269
|
+
}
|
|
270
|
+
},
|
|
271
|
+
onContextUpdate: (params) => {
|
|
272
|
+
const update = createUiModelContextUpdate(params);
|
|
273
|
+
log.debug("Model context update from UI", {
|
|
274
|
+
hasContent: !!params.content,
|
|
275
|
+
hasStructured: !!params.structuredContent,
|
|
276
|
+
hasUpdate: !!update
|
|
277
|
+
});
|
|
278
|
+
if (update && state.sendMessage) {
|
|
279
|
+
state.sendMessage({
|
|
280
|
+
customType: "mcp-ui-context",
|
|
281
|
+
content: [{ type: "text", text: `User submitted model context from ${request.serverName} UI:
|
|
282
|
+
${update.summary}` }],
|
|
283
|
+
display: "UI Context submitted",
|
|
284
|
+
details: { server: request.serverName, tool: request.toolName, context: update }
|
|
285
|
+
}, { triggerTurn: true });
|
|
286
|
+
}
|
|
287
|
+
},
|
|
288
|
+
onComplete: (reason) => {
|
|
289
|
+
active = false;
|
|
290
|
+
cleanupListeners();
|
|
291
|
+
if (state.uiServer === handle) {
|
|
292
|
+
const messages = handle.getSessionMessages();
|
|
293
|
+
const stream = handle.getStreamSummary();
|
|
294
|
+
const hasContent = messages.prompts.length > 0 || messages.intents.length > 0 || messages.notifications.length > 0 || messages.contexts.length > 0 || !!stream;
|
|
295
|
+
if (hasContent) {
|
|
296
|
+
state.completedUiSessions.push({
|
|
297
|
+
serverName: handle.serverName,
|
|
298
|
+
toolName: handle.toolName,
|
|
299
|
+
completedAt: new Date,
|
|
300
|
+
reason,
|
|
301
|
+
messages,
|
|
302
|
+
...stream !== undefined ? { stream } : {}
|
|
303
|
+
});
|
|
304
|
+
while (state.completedUiSessions.length > MAX_COMPLETED_SESSIONS) {
|
|
305
|
+
state.completedUiSessions.shift();
|
|
306
|
+
}
|
|
307
|
+
log.debug("Session completed", {
|
|
308
|
+
reason,
|
|
309
|
+
prompts: messages.prompts.length,
|
|
310
|
+
intents: messages.intents.length,
|
|
311
|
+
notifications: messages.notifications.length,
|
|
312
|
+
contexts: messages.contexts.length,
|
|
313
|
+
streamFrames: stream?.frames ?? 0
|
|
314
|
+
});
|
|
315
|
+
}
|
|
316
|
+
state.uiServer = null;
|
|
317
|
+
if (activeGlimpseWindow) {
|
|
318
|
+
activeGlimpseWindow.close();
|
|
319
|
+
activeGlimpseWindow = null;
|
|
320
|
+
}
|
|
321
|
+
}
|
|
322
|
+
}
|
|
323
|
+
});
|
|
324
|
+
if (state.owner?.isActive() === false || runtimeSignal.aborted) {
|
|
325
|
+
handle.close("runtime_owner_stopped");
|
|
326
|
+
throwIfAborted(runtimeSignal);
|
|
327
|
+
throw new Error("MCP UI session became stale before registration");
|
|
328
|
+
}
|
|
329
|
+
if (streamToken) {
|
|
330
|
+
state.manager.registerUiStreamListener(streamToken, (serverName, notification) => {
|
|
331
|
+
if (!active || state.uiServer !== handle)
|
|
332
|
+
return;
|
|
333
|
+
if (serverName !== request.serverName)
|
|
334
|
+
return;
|
|
335
|
+
nextStreamSequence += 1;
|
|
336
|
+
handle.sendResultPatch(withStreamEnvelope(notification.result, streamId, nextStreamSequence));
|
|
337
|
+
});
|
|
338
|
+
}
|
|
339
|
+
state.manager.registerResourceUpdatedListener?.(resourceListenerToken, request.serverName, request.uiResourceUri, (_serverName, uri) => {
|
|
340
|
+
if (!active || state.uiServer !== handle)
|
|
341
|
+
return;
|
|
342
|
+
handle.sendResourceUpdated(uri);
|
|
343
|
+
});
|
|
344
|
+
state.uiServer = handle;
|
|
345
|
+
const viewerPref = process.env.MCP_UI_VIEWER?.toLowerCase();
|
|
346
|
+
const uiSuppressed = viewerPref === "none" || viewerPref === "off" || viewerPref === "disabled";
|
|
347
|
+
let viewer = "browser";
|
|
348
|
+
let windowOpen = true;
|
|
349
|
+
const remoteByEnv = isRemoteSession();
|
|
350
|
+
if (uiSuppressed) {
|
|
351
|
+
viewer = "suppressed";
|
|
352
|
+
windowOpen = false;
|
|
353
|
+
state.ui?.notify(`MCP UI window suppressed (MCP_UI_VIEWER=${viewerPref}). Open manually: ${handle.url}
|
|
354
|
+
` + `If this session is remote, run ssh -L ${handle.port}:127.0.0.1:${handle.port} -L ${handle.proxyPort}:127.0.0.1:${handle.proxyPort} <this-host> first.`, "info");
|
|
355
|
+
log.info("Suppressing MCP UI window (MCP_UI_VIEWER=" + viewerPref + ")", { url: handle.url });
|
|
356
|
+
} else {
|
|
357
|
+
const remoteLikely = remoteByEnv || await hasActiveRemoteLogin();
|
|
358
|
+
const emitRemoteHint = async (openError, openedOnHost = false) => {
|
|
359
|
+
state.ui?.notify(remoteAccessHint({
|
|
360
|
+
url: handle.url,
|
|
361
|
+
port: handle.port,
|
|
362
|
+
proxyPort: handle.proxyPort,
|
|
363
|
+
moshi: await probeMoshiGateway(),
|
|
364
|
+
openError,
|
|
365
|
+
openedOnHost
|
|
366
|
+
}), openError === null ? "info" : "warning");
|
|
367
|
+
};
|
|
368
|
+
const glimpseDetected = !remoteByEnv && isGlimpseAvailable();
|
|
369
|
+
const useGlimpse = !remoteByEnv && (viewerPref === "glimpse" || viewerPref !== "browser" && glimpseDetected);
|
|
370
|
+
if (useGlimpse) {
|
|
371
|
+
try {
|
|
372
|
+
const glimpseHtml = `<!DOCTYPE html><html><head><meta charset="utf-8"><style>body{margin:0;padding:0;width:100vw;height:100vh;overflow:hidden}iframe{width:100%;height:100%;border:none}</style></head><body><iframe src="${handle.url}"></iframe></body></html>`;
|
|
373
|
+
const glimpseWindow = await openGlimpseWindow(glimpseHtml, {
|
|
374
|
+
title: `MCP · ${request.serverName} · ${request.toolName}`,
|
|
375
|
+
width: 1000,
|
|
376
|
+
height: 800,
|
|
377
|
+
onClosed: () => {
|
|
378
|
+
if (active)
|
|
379
|
+
handle.close("glimpse-closed");
|
|
380
|
+
}
|
|
381
|
+
});
|
|
382
|
+
if (state.owner?.isActive() === false || runtimeSignal.aborted) {
|
|
383
|
+
glimpseWindow.close();
|
|
384
|
+
throwIfAborted(runtimeSignal);
|
|
385
|
+
throw new Error("MCP Glimpse window became stale before registration");
|
|
386
|
+
}
|
|
387
|
+
activeGlimpseWindow = glimpseWindow;
|
|
388
|
+
viewer = "glimpse";
|
|
389
|
+
if (remoteLikely) {
|
|
390
|
+
await emitRemoteHint(null, true);
|
|
391
|
+
}
|
|
392
|
+
} catch (error) {
|
|
393
|
+
log.debug("Glimpse unavailable, using browser", {
|
|
394
|
+
error: error instanceof Error ? error.message : String(error)
|
|
395
|
+
});
|
|
396
|
+
const openError = await openInBrowser(state, handle.url, runtimeSignal);
|
|
397
|
+
if (openError !== null || remoteLikely) {
|
|
398
|
+
await emitRemoteHint(openError);
|
|
399
|
+
}
|
|
400
|
+
viewer = "browser";
|
|
401
|
+
}
|
|
402
|
+
} else {
|
|
403
|
+
const openError = await openInBrowser(state, handle.url, runtimeSignal);
|
|
404
|
+
if (openError !== null || remoteLikely) {
|
|
405
|
+
await emitRemoteHint(openError);
|
|
406
|
+
}
|
|
407
|
+
}
|
|
408
|
+
}
|
|
409
|
+
throwIfAborted(runtimeSignal);
|
|
410
|
+
handle.viewer = viewer;
|
|
411
|
+
handle.windowOpen = windowOpen;
|
|
412
|
+
return {
|
|
413
|
+
serverName: request.serverName,
|
|
414
|
+
toolName: request.toolName,
|
|
415
|
+
reused: false,
|
|
416
|
+
...streamId !== undefined ? { streamId } : {},
|
|
417
|
+
...streamToken !== undefined ? { streamToken } : {},
|
|
418
|
+
...streamMode !== undefined ? { streamMode } : {},
|
|
419
|
+
...streamToken ? { requestMeta: { [UI_STREAM_REQUEST_META_KEY]: streamToken } } : {},
|
|
420
|
+
url: handle.url,
|
|
421
|
+
viewer,
|
|
422
|
+
windowOpen,
|
|
423
|
+
isActive: () => active && state.uiServer === handle,
|
|
424
|
+
sendToolResult: (result) => {
|
|
425
|
+
if (!active || state.uiServer !== handle)
|
|
426
|
+
return;
|
|
427
|
+
nextStreamSequence += 1;
|
|
428
|
+
handle.sendToolResult(withStreamEnvelope(result, streamId, nextStreamSequence));
|
|
429
|
+
},
|
|
430
|
+
sendResultPatch: (result) => {
|
|
431
|
+
if (!active || state.uiServer !== handle)
|
|
432
|
+
return;
|
|
433
|
+
nextStreamSequence += 1;
|
|
434
|
+
handle.sendResultPatch(withStreamEnvelope(result, streamId, nextStreamSequence));
|
|
435
|
+
},
|
|
436
|
+
sendToolCancelled: (reason) => {
|
|
437
|
+
if (!active || state.uiServer !== handle)
|
|
438
|
+
return;
|
|
439
|
+
handle.sendToolCancelled(reason);
|
|
440
|
+
},
|
|
441
|
+
close: (reason) => {
|
|
442
|
+
active = false;
|
|
443
|
+
cleanupListeners();
|
|
444
|
+
handle.close(reason);
|
|
445
|
+
}
|
|
446
|
+
};
|
|
447
|
+
} catch (error) {
|
|
448
|
+
if (error instanceof UrlElicitationRequiredError || error instanceof InputRequiredNeedsUiError || isAbortError(error, runtimeSignal))
|
|
449
|
+
throw error;
|
|
450
|
+
const message = error instanceof Error ? error.message : String(error);
|
|
451
|
+
log.error("Failed to start UI session", error instanceof Error ? error : undefined);
|
|
452
|
+
state.ui?.notify(`MCP UI unavailable for ${request.toolName} (${request.serverName}): ${message}`, "warning");
|
|
453
|
+
return null;
|
|
454
|
+
}
|
|
455
|
+
}
|
|
@@ -0,0 +1,58 @@
|
|
|
1
|
+
import { z } from "zod";
|
|
2
|
+
export const UI_STREAM_HOST_CONTEXT_KEY = "dm-mcp-adapter/stream";
|
|
3
|
+
export const UI_STREAM_REQUEST_META_KEY = "dm-mcp-adapter/stream-token";
|
|
4
|
+
export const UI_STREAM_RESULT_PATCH_METHOD = "notifications/dm-mcp-adapter/ui-result-patch";
|
|
5
|
+
export const SERVER_STREAM_RESULT_PATCH_METHOD = "notifications/dm-mcp-adapter/result-patch";
|
|
6
|
+
export const UI_STREAM_STRUCTURED_CONTENT_KEY = "dm-mcp-adapter/stream";
|
|
7
|
+
export const uiStreamModeSchema = z.enum(["eager", "stream-first"]);
|
|
8
|
+
export const visualizationStreamPhaseSchema = z.enum(["shell", "narrative", "structure", "detail", "settled"]);
|
|
9
|
+
export const visualizationStreamFrameTypeSchema = z.enum(["patch", "checkpoint", "final"]);
|
|
10
|
+
export const visualizationStreamStatusSchema = z.enum(["ok", "error"]);
|
|
11
|
+
const looseRecordSchema = z.record(z.string(), z.unknown());
|
|
12
|
+
const looseArraySchema = z.array(z.unknown());
|
|
13
|
+
export const uiStreamHostContextSchema = z.object({
|
|
14
|
+
mode: uiStreamModeSchema,
|
|
15
|
+
streamId: z.string().min(1),
|
|
16
|
+
intermediateResultPatches: z.boolean(),
|
|
17
|
+
partialInput: z.boolean()
|
|
18
|
+
});
|
|
19
|
+
export const visualizationStreamEnvelopeSchema = z.object({
|
|
20
|
+
streamId: z.string().min(1),
|
|
21
|
+
sequence: z.number().int().nonnegative(),
|
|
22
|
+
frameType: visualizationStreamFrameTypeSchema,
|
|
23
|
+
phase: visualizationStreamPhaseSchema,
|
|
24
|
+
status: visualizationStreamStatusSchema,
|
|
25
|
+
message: z.string().optional(),
|
|
26
|
+
spec: looseRecordSchema.optional(),
|
|
27
|
+
checkpoint: looseRecordSchema.optional()
|
|
28
|
+
});
|
|
29
|
+
export const uiStreamCallToolResultSchema = z.object({
|
|
30
|
+
content: looseArraySchema.optional(),
|
|
31
|
+
structuredContent: looseRecordSchema.optional(),
|
|
32
|
+
isError: z.boolean().optional(),
|
|
33
|
+
_meta: looseRecordSchema.optional()
|
|
34
|
+
}).passthrough();
|
|
35
|
+
export const uiStreamResultPatchNotificationSchema = z.object({
|
|
36
|
+
method: z.literal(UI_STREAM_RESULT_PATCH_METHOD),
|
|
37
|
+
params: uiStreamCallToolResultSchema
|
|
38
|
+
});
|
|
39
|
+
export const serverStreamResultPatchNotificationSchema = z.object({
|
|
40
|
+
method: z.literal(SERVER_STREAM_RESULT_PATCH_METHOD),
|
|
41
|
+
params: z.object({
|
|
42
|
+
streamToken: z.string().min(1),
|
|
43
|
+
result: uiStreamCallToolResultSchema
|
|
44
|
+
})
|
|
45
|
+
});
|
|
46
|
+
export function getUiStreamHostContext(hostContext) {
|
|
47
|
+
const candidate = hostContext?.[UI_STREAM_HOST_CONTEXT_KEY];
|
|
48
|
+
const parsed = uiStreamHostContextSchema.safeParse(candidate);
|
|
49
|
+
return parsed.success ? parsed.data : undefined;
|
|
50
|
+
}
|
|
51
|
+
export function getVisualizationStreamEnvelope(structuredContent) {
|
|
52
|
+
if (!structuredContent || typeof structuredContent !== "object" || Array.isArray(structuredContent)) {
|
|
53
|
+
return;
|
|
54
|
+
}
|
|
55
|
+
const candidate = structuredContent[UI_STREAM_STRUCTURED_CONTENT_KEY];
|
|
56
|
+
const parsed = visualizationStreamEnvelopeSchema.safeParse(candidate);
|
|
57
|
+
return parsed.success ? parsed.data : undefined;
|
|
58
|
+
}
|
|
@@ -0,0 +1,26 @@
|
|
|
1
|
+
export function extractUiToolVisibility(meta) {
|
|
2
|
+
if (!meta || typeof meta !== "object")
|
|
3
|
+
return;
|
|
4
|
+
const ui = meta.ui;
|
|
5
|
+
if (!ui || typeof ui !== "object" || Array.isArray(ui))
|
|
6
|
+
return;
|
|
7
|
+
const visibility = ui.visibility;
|
|
8
|
+
if (visibility === undefined)
|
|
9
|
+
return;
|
|
10
|
+
if (!Array.isArray(visibility))
|
|
11
|
+
return [];
|
|
12
|
+
const values = [];
|
|
13
|
+
for (const entry of visibility) {
|
|
14
|
+
if (entry !== "model" && entry !== "app")
|
|
15
|
+
return [];
|
|
16
|
+
if (!values.includes(entry))
|
|
17
|
+
values.push(entry);
|
|
18
|
+
}
|
|
19
|
+
return values;
|
|
20
|
+
}
|
|
21
|
+
export function isUiToolVisibleToModel(visibility) {
|
|
22
|
+
return visibility === undefined || visibility.includes("model");
|
|
23
|
+
}
|
|
24
|
+
export function isUiToolCallableByApp(visibility) {
|
|
25
|
+
return visibility === undefined || visibility.includes("app");
|
|
26
|
+
}
|
|
@@ -0,0 +1,83 @@
|
|
|
1
|
+
import { createConnection } from "node:net";
|
|
2
|
+
import { ReadBuffer, serializeMessage } from "@modelcontextprotocol/client";
|
|
3
|
+
|
|
4
|
+
export class UnixSocketClientTransport {
|
|
5
|
+
socketPath;
|
|
6
|
+
socket;
|
|
7
|
+
readBuffer = new ReadBuffer;
|
|
8
|
+
onclose;
|
|
9
|
+
onerror;
|
|
10
|
+
onmessage;
|
|
11
|
+
constructor(socketPath) {
|
|
12
|
+
this.socketPath = socketPath;
|
|
13
|
+
}
|
|
14
|
+
async start() {
|
|
15
|
+
if (this.socket) {
|
|
16
|
+
throw new Error("UnixSocketClientTransport already started");
|
|
17
|
+
}
|
|
18
|
+
await new Promise((resolve, reject) => {
|
|
19
|
+
const socket = createConnection(this.socketPath);
|
|
20
|
+
this.socket = socket;
|
|
21
|
+
let connected = false;
|
|
22
|
+
socket.once("connect", () => {
|
|
23
|
+
connected = true;
|
|
24
|
+
resolve();
|
|
25
|
+
});
|
|
26
|
+
socket.on("data", (chunk) => {
|
|
27
|
+
try {
|
|
28
|
+
this.readBuffer.append(chunk);
|
|
29
|
+
while (true) {
|
|
30
|
+
const message = this.readBuffer.readMessage();
|
|
31
|
+
if (message === null)
|
|
32
|
+
break;
|
|
33
|
+
this.onmessage?.(message);
|
|
34
|
+
}
|
|
35
|
+
} catch (error) {
|
|
36
|
+
const cause = error instanceof Error ? error : new Error(String(error));
|
|
37
|
+
this.onerror?.(cause);
|
|
38
|
+
this.close();
|
|
39
|
+
}
|
|
40
|
+
});
|
|
41
|
+
socket.on("error", (error) => {
|
|
42
|
+
if (!connected)
|
|
43
|
+
reject(error);
|
|
44
|
+
this.onerror?.(error);
|
|
45
|
+
});
|
|
46
|
+
socket.on("close", () => {
|
|
47
|
+
if (this.socket === socket)
|
|
48
|
+
this.socket = undefined;
|
|
49
|
+
this.readBuffer.clear();
|
|
50
|
+
this.onclose?.();
|
|
51
|
+
});
|
|
52
|
+
});
|
|
53
|
+
}
|
|
54
|
+
async close() {
|
|
55
|
+
const socket = this.socket;
|
|
56
|
+
this.socket = undefined;
|
|
57
|
+
this.readBuffer.clear();
|
|
58
|
+
if (!socket || socket.destroyed)
|
|
59
|
+
return;
|
|
60
|
+
await new Promise((resolve) => {
|
|
61
|
+
const timeout = setTimeout(() => socket.destroy(), 2000);
|
|
62
|
+
timeout.unref();
|
|
63
|
+
socket.once("close", () => {
|
|
64
|
+
clearTimeout(timeout);
|
|
65
|
+
resolve();
|
|
66
|
+
});
|
|
67
|
+
socket.end();
|
|
68
|
+
});
|
|
69
|
+
}
|
|
70
|
+
async send(message) {
|
|
71
|
+
const socket = this.socket;
|
|
72
|
+
if (!socket || socket.destroyed)
|
|
73
|
+
throw new Error("Unix socket is not connected");
|
|
74
|
+
await new Promise((resolve, reject) => {
|
|
75
|
+
socket.write(serializeMessage(message), (error) => {
|
|
76
|
+
if (error)
|
|
77
|
+
reject(error);
|
|
78
|
+
else
|
|
79
|
+
resolve();
|
|
80
|
+
});
|
|
81
|
+
});
|
|
82
|
+
}
|
|
83
|
+
}
|