@akon-labs/nexus-cli 0.1.8 → 0.1.10

package/dist/extensions/mcp/tool-approval.js

@@ -0,0 +1,156 @@
+import { createHash, randomUUID } from "node:crypto";
+import { abortable } from "./abort.js";
+import { combineAbortSignals } from "./runtime-owner.js";
+import { getToolNameCandidates, MCP_TOOL_APPROVAL_REQUEST_EVENT, matchesToolPattern, resolveToolPrefix, } from "./types.js";
+import { sanitizeTerminalText } from "./utils.js";
+function stableStringify(value) {
+    if (value === null || value === undefined || typeof value !== "object") {
+        const serialized = JSON.stringify(value);
+        return serialized === undefined ? "undefined" : serialized;
+    }
+    if (Array.isArray(value)) {
+        return `[${value.map((item) => stableStringify(item)).join(",")}]`;
+    }
+    const object = value;
+    return `{${Object.keys(object)
+        .sort()
+        .map((key) => `${JSON.stringify(key)}:${stableStringify(object[key])}`)
+        .join(",")}}`;
+}
+export function isToolCallApprovalRequired(config, serverName, toolMeta, toolMetadata) {
+    const definition = config.mcpServers[serverName];
+    const serverApproval = definition?.approveTools;
+    const approval = serverApproval !== undefined ? serverApproval : config.settings?.approveTools;
+    if (approval === true)
+        return true;
+    if (!Array.isArray(approval) || approval.length === 0)
+        return false;
+    const prefix = resolveToolPrefix(definition, config.settings?.toolPrefix);
+    const currentCandidates = getToolNameCandidates(toolMeta.originalName, serverName, prefix, false);
+    if (serverApproval !== undefined) {
+        if (matchesToolPattern(currentCandidates, approval))
+            return true;
+        if (!toolMetadata)
+            return matchesToolPattern(getToolNameCandidates(toolMeta.originalName, serverName, prefix), approval);
+        const legacyCandidates = getToolNameCandidates(toolMeta.originalName, serverName, prefix);
+        const legacyEmittedName = [...currentCandidates]
+            .find((candidate) => candidate !== toolMeta.originalName)
+            ?.replace(/-/g, "_");
+        if (legacyEmittedName)
+            legacyCandidates.add(legacyEmittedName);
+        for (const candidate of currentCandidates)
+            legacyCandidates.delete(candidate);
+        const otherCurrentCandidates = new Set();
+        for (const tool of toolMetadata.get(serverName) ?? []) {
+            for (const candidate of getToolNameCandidates(tool.originalName, serverName, prefix, false)) {
+                otherCurrentCandidates.add(candidate);
+            }
+        }
+        for (const candidate of currentCandidates)
+            otherCurrentCandidates.delete(candidate);
+        return approval.some((pattern) => matchesToolPattern(legacyCandidates, [pattern]) && !matchesToolPattern(otherCurrentCandidates, [pattern]));
+    }
+    if (matchesToolPattern(currentCandidates, approval))
+        return true;
+    if (!toolMetadata)
+        return false;
+    const legacyCandidates = getToolNameCandidates(toolMeta.originalName, serverName, prefix);
+    const legacyEmittedName = [...currentCandidates]
+        .find((candidate) => candidate !== toolMeta.originalName)
+        ?.replace(/-/g, "_");
+    if (legacyEmittedName)
+        legacyCandidates.add(legacyEmittedName);
+    for (const candidate of currentCandidates)
+        legacyCandidates.delete(candidate);
+    const otherCurrentCandidates = new Set();
+    for (const [name, metadata] of toolMetadata) {
+        const otherPrefix = resolveToolPrefix(config.mcpServers[name], config.settings?.toolPrefix);
+        for (const tool of metadata) {
+            for (const candidate of getToolNameCandidates(tool.originalName, name, otherPrefix, false)) {
+                otherCurrentCandidates.add(candidate);
+            }
+        }
+    }
+    for (const candidate of currentCandidates)
+        otherCurrentCandidates.delete(candidate);
+    return approval.some((pattern) => matchesToolPattern(legacyCandidates, [pattern]) && !matchesToolPattern(otherCurrentCandidates, [pattern]));
+}
+function isMcpToolApprovalDecision(value) {
+    return value === "allow_once" || value === "allow_for_session" || value === "deny" || value === "abstain";
+}
+async function requestBrokerApproval(state, serverName, toolMeta, args, origin, signal) {
+    if (!state.approvalEvents)
+        return "abstain";
+    let acceptingClaim = true;
+    let handler;
+    const request = {
+        requestId: randomUUID(),
+        serverName,
+        originalToolName: toolMeta.originalName,
+        prefixedToolName: toolMeta.name,
+        args: args ?? {},
+        origin,
+        ...(signal !== undefined ? { signal } : {}),
+        claim(candidate) {
+            if (!acceptingClaim || handler)
+                return false;
+            handler = candidate;
+            return true;
+        },
+    };
+    state.approvalEvents.emit(MCP_TOOL_APPROVAL_REQUEST_EVENT, request);
+    acceptingClaim = false;
+    if (!handler)
+        return "abstain";
+    try {
+        const decision = await abortable(Promise.resolve().then(handler), signal);
+        return isMcpToolApprovalDecision(decision) ? decision : "deny";
+    }
+    catch (error) {
+        if (signal?.aborted)
+            throw error;
+        return "deny";
+    }
+}
+export async function ensureToolCallApproved(state, serverName, toolMeta, args, signal, origin = toolMeta.resourceUri ? "resource" : "proxy", approvalMetadata) {
+    const argsHash = createHash("sha256")
+        .update(stableStringify(args ?? {}))
+        .digest("hex");
+    const cacheKey = `${serverName}\u0000${toolMeta.originalName}\u0000${argsHash}`;
+    if (!state.approvedToolCalls)
+        state.approvedToolCalls = new Map();
+    const approvedToolCalls = state.approvedToolCalls;
+    if (approvedToolCalls.has(cacheKey)) {
+        return { ok: true };
+    }
+    const brokerDecision = await requestBrokerApproval(state, serverName, toolMeta, args, origin, signal);
+    if (brokerDecision === "allow_once")
+        return { ok: true };
+    if (brokerDecision === "allow_for_session") {
+        approvedToolCalls.set(cacheKey, true);
+        return { ok: true };
+    }
+    if (brokerDecision === "deny")
+        return { ok: false, reason: "denied" };
+    if (!isToolCallApprovalRequired(state.config, serverName, toolMeta, approvalMetadata ?? state.toolMetadata)) {
+        return { ok: true };
+    }
+    if (!state.ui) {
+        return { ok: false, reason: "approval_required_headless" };
+    }
+    const json = JSON.stringify(args ?? {}, null, 2);
+    const sanitized = sanitizeTerminalText(json);
+    const preview = sanitized.length > 500 ? `${sanitized.slice(0, 500)}...` : sanitized;
+    const title = `MCP: ${sanitizeTerminalText(serverName)} wants to run ${sanitizeTerminalText(toolMeta.originalName)}`;
+    const ownedSignal = combineAbortSignals(state.owner?.signal, signal);
+    const decision = await abortable(state.ui.select(`${title}\n\nArguments:\n${preview}`, ["Allow once", "Allow for session", "Deny"]), ownedSignal);
+    if (decision === "Allow once") {
+        return { ok: true };
+    }
+    if (decision === "Allow for session") {
+        approvedToolCalls.set(cacheKey, true);
+        return { ok: true };
+    }
+    return { ok: false, reason: "denied" };
+}
+//# sourceMappingURL=tool-approval.js.map

package/dist/extensions/mcp/tool-approval.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-approval.js","sourceRoot":"","sources":["../../../src/extensions/mcp/tool-approval.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACrD,OAAO,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AACvC,OAAO,EAAE,mBAAmB,EAAE,MAAM,oBAAoB,CAAC;AAEzD,OAAO,EACN,qBAAqB,EACrB,+BAA+B,EAM/B,kBAAkB,EAClB,iBAAiB,GAEjB,MAAM,YAAY,CAAC;AACpB,OAAO,EAAE,oBAAoB,EAAE,MAAM,YAAY,CAAC;AAIlD,SAAS,eAAe,CAAC,KAAc,EAAU;IAChD,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,SAAS,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QACxE,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QACzC,OAAO,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,UAAU,CAAC;IAC5D,CAAC;IACD,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QAC1B,OAAO,IAAI,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;IACpE,CAAC;IACD,MAAM,MAAM,GAAG,KAAgC,CAAC;IAChD,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC;SAC5B,IAAI,EAAE;SACN,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,eAAe,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;SACtE,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;AAAA,CACf;AAED,MAAM,UAAU,0BAA0B,CACzC,MAAiB,EACjB,UAAkB,EAClB,QAA4C,EAC5C,YAA2D,EACjD;IACV,MAAM,UAAU,GAAG,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;IACjD,MAAM,cAAc,GAAG,UAAU,EAAE,YAAY,CAAC;IAChD,MAAM,QAAQ,GAAG,cAAc,KAAK,SAAS,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,EAAE,YAAY,CAAC;IAE/F,IAAI,QAAQ,KAAK,IAAI;QAAE,OAAO,IAAI,CAAC;IACnC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,KAAK,CAAC;IAEpE,MAAM,MAAM,GAAG,iBAAiB,CAAC,UAAU,EAAE,MAAM,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;IAC1E,MAAM,iBAAiB,GAAG,qBAAqB,CAAC,QAAQ,CAAC,YAAY,EAAE,UAAU,EAAE,MAAM,EAAE,KAAK,CAAC,CAAC;IAClG,IAAI,cAAc,KAAK,SAAS,EAAE,CAAC;QAClC,IAAI,kBAAkB,CAAC,iBAAiB,EAAE,QAAQ,CAAC;YAAE,OAAO,IAAI,CAAC;QACjE,IAAI,CAAC,YAAY;YAChB,OAAO,kBAAkB,CAAC,qBAAqB,CAAC,QAAQ,CAAC,YAAY,EAAE,UAAU,EAAE,MAAM,CAAC,EAAE,QAAQ,CAAC,CAAC;QACvG,MAAM,gBAAgB,GAAG,qBAAqB,CAAC,QAAQ,CAAC,YAAY,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;QAC1F,MAAM,iBAAiB,GAAG,CAAC,GAAG,iBAAiB,CAAC;aAC9C,IAAI,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,KAAK,QAAQ,CAAC,YAAY,CAAC;YACzD,EAAE,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;QACtB,IAAI,iBAAiB;YAAE,gBAAgB,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAC;QAC/D,KAAK,MAAM,SAAS,IAAI,iBAAiB;YAAE,gBAAgB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAC9E,MAAM,sBAAsB,GAAG,IAAI,GAAG,EAAU,CAAC;QACjD,KAAK,MAAM,IAAI,IAAI,YAAY,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,EAAE,EAAE,CAAC;YACvD,KAAK,MAAM,SAAS,IAAI,qBAAqB,CAAC,IAAI,CAAC,YAAY,EAAE,UAAU,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,CAAC;gBAC7F,sBAAsB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YACvC,CAAC;QACF,CAAC;QACD,KAAK,MAAM,SAAS,IAAI,iBAAiB;YAAE,sBAAsB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QACpF,OAAO,QAAQ,CAAC,IAAI,CACnB,CAAC,OAAO,EAAE,EAAE,CACX,kBAAkB,CAAC,gBAAgB,EAAE,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,sBAAsB,EAAE,CAAC,OAAO,CAAC,CAAC,CAC1G,CAAC;IACH,CAAC;IAED,IAAI,kBAAkB,CAAC,iBAAiB,EAAE,QAAQ,CAAC;QAAE,OAAO,IAAI,CAAC;IACjE,IAAI,CAAC,YAAY;QAAE,OAAO,KAAK,CAAC;IAEhC,MAAM,gBAAgB,GAAG,qBAAqB,CAAC,QAAQ,CAAC,YAAY,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;IAC1F,MAAM,iBAAiB,GAAG,CAAC,GAAG,iBAAiB,CAAC;SAC9C,IAAI,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,KAAK,QAAQ,CAAC,YAAY,CAAC;QACzD,EAAE,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;IACtB,IAAI,iBAAiB;QAAE,gBAAgB,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAC;IAC/D,KAAK,MAAM,SAAS,IAAI,iBAAiB;QAAE,gBAAgB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;IAC9E,MAAM,sBAAsB,GAAG,IAAI,GAAG,EAAU,CAAC;IACjD,KAAK,MAAM,CAAC,IAAI,EAAE,QAAQ,CAAC,IAAI,YAAY,EAAE,CAAC;QAC7C,MAAM,WAAW,GAAG,iBAAiB,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,MAAM,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;QAC5F,KAAK,MAAM,IAAI,IAAI,QAAQ,EAAE,CAAC;YAC7B,KAAK,MAAM,SAAS,IAAI,qBAAqB,CAAC,IAAI,CAAC,YAAY,EAAE,IAAI,EAAE,WAAW,EAAE,KAAK,CAAC,EAAE,CAAC;gBAC5F,sBAAsB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YACvC,CAAC;QACF,CAAC;IACF,CAAC;IACD,KAAK,MAAM,SAAS,IAAI,iBAAiB;QAAE,sBAAsB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;IAEpF,OAAO,QAAQ,CAAC,IAAI,CACnB,CAAC,OAAO,EAAE,EAAE,CACX,kBAAkB,CAAC,gBAAgB,EAAE,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,sBAAsB,EAAE,CAAC,OAAO,CAAC,CAAC,CAC1G,CAAC;AAAA,CACF;AAED,SAAS,yBAAyB,CAAC,KAAc,EAAoC;IACpF,OAAO,KAAK,KAAK,YAAY,IAAI,KAAK,KAAK,mBAAmB,IAAI,KAAK,KAAK,MAAM,IAAI,KAAK,KAAK,SAAS,CAAC;AAAA,CAC1G;AAED,KAAK,UAAU,qBAAqB,CACnC,KAAwB,EACxB,UAAkB,EAClB,QAAsB,EACtB,IAAyC,EACzC,MAA6B,EAC7B,MAAoB,EACe;IACnC,IAAI,CAAC,KAAK,CAAC,cAAc;QAAE,OAAO,SAAS,CAAC;IAE5C,IAAI,cAAc,GAAG,IAAI,CAAC;IAC1B,IAAI,OAA2C,CAAC;IAChD,MAAM,OAAO,GAA2B;QACvC,SAAS,EAAE,UAAU,EAAE;QACvB,UAAU;QACV,gBAAgB,EAAE,QAAQ,CAAC,YAAY;QACvC,gBAAgB,EAAE,QAAQ,CAAC,IAAI;QAC/B,IAAI,EAAE,IAAI,IAAI,EAAE;QAChB,MAAM;QACN,GAAG,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC3C,KAAK,CAAC,SAAiC,EAAE;YACxC,IAAI,CAAC,cAAc,IAAI,OAAO;gBAAE,OAAO,KAAK,CAAC;YAC7C,OAAO,GAAG,SAAS,CAAC;YACpB,OAAO,IAAI,CAAC;QAAA,CACZ;KACD,CAAC;IAEF,KAAK,CAAC,cAAc,CAAC,IAAI,CAAC,+BAA+B,EAAE,OAAO,CAAC,CAAC;IACpE,cAAc,GAAG,KAAK,CAAC;IACvB,IAAI,CAAC,OAAO;QAAE,OAAO,SAAS,CAAC;IAE/B,IAAI,CAAC;QACJ,MAAM,QAAQ,GAAG,MAAM,SAAS,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,MAAM,CAAC,CAAC;QAC1E,OAAO,yBAAyB,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC;IAChE,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAChB,IAAI,MAAM,EAAE,OAAO;YAAE,MAAM,KAAK,CAAC;QACjC,OAAO,MAAM,CAAC;IACf,CAAC;AAAA,CACD;AAED,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAC3C,KAAwB,EACxB,UAAkB,EAClB,QAAsB,EACtB,IAAyC,EACzC,MAAoB,EACpB,MAAM,GAA0B,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,OAAO,EAC3E,gBAA+D,EAC7B;IAClC,MAAM,QAAQ,GAAG,UAAU,CAAC,QAAQ,CAAC;SACnC,MAAM,CAAC,eAAe,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;SACnC,MAAM,CAAC,KAAK,CAAC,CAAC;IAChB,MAAM,QAAQ,GAAG,GAAG,UAAU,SAAS,QAAQ,CAAC,YAAY,SAAS,QAAQ,EAAE,CAAC;IAChF,IAAI,CAAC,KAAK,CAAC,iBAAiB;QAAE,KAAK,CAAC,iBAAiB,GAAG,IAAI,GAAG,EAAgB,CAAC;IAChF,MAAM,iBAAiB,GAAG,KAAK,CAAC,iBAAiB,CAAC;IAClD,IAAI,iBAAiB,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;QACrC,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC;IACrB,CAAC;IAED,MAAM,cAAc,GAAG,MAAM,qBAAqB,CAAC,KAAK,EAAE,UAAU,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;IACtG,IAAI,cAAc,KAAK,YAAY;QAAE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC;IACzD,IAAI,cAAc,KAAK,mBAAmB,EAAE,CAAC;QAC5C,iBAAiB,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QACtC,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC;IACrB,CAAC;IACD,IAAI,cAAc,KAAK,MAAM;QAAE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC;IAEtE,IAAI,CAAC,0BAA0B,CAAC,KAAK,CAAC,MAAM,EAAE,UAAU,EAAE,QAAQ,EAAE,gBAAgB,IAAI,KAAK,CAAC,YAAY,CAAC,EAAE,CAAC;QAC7G,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC;IACrB,CAAC;IAED,IAAI,CAAC,KAAK,CAAC,EAAE,EAAE,CAAC;QACf,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,4BAA4B,EAAE,CAAC;IAC5D,CAAC;IAED,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,IAAI,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;IACjD,MAAM,SAAS,GAAG,oBAAoB,CAAC,IAAI,CAAC,CAAC;IAC7C,MAAM,OAAO,GAAG,SAAS,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC,CAAC,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC;IACrF,MAAM,KAAK,GAAG,QAAQ,oBAAoB,CAAC,UAAU,CAAC,iBAAiB,oBAAoB,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,CAAC;IACrH,MAAM,WAAW,GAAG,mBAAmB,CAAC,KAAK,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;IACrE,MAAM,QAAQ,GAAG,MAAM,SAAS,CAC/B,KAAK,CAAC,EAAE,CAAC,MAAM,CAAC,GAAG,KAAK,mBAAmB,OAAO,EAAE,EAAE,CAAC,YAAY,EAAE,mBAAmB,EAAE,MAAM,CAAC,CAAC,EAClG,WAAW,CACX,CAAC;IAEF,IAAI,QAAQ,KAAK,YAAY,EAAE,CAAC;QAC/B,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC;IACrB,CAAC;IACD,IAAI,QAAQ,KAAK,mBAAmB,EAAE,CAAC;QACtC,iBAAiB,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QACtC,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC;IACrB,CAAC;IACD,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC;AAAA,CACvC","sourcesContent":["import { createHash, randomUUID } from \"node:crypto\";\nimport { abortable } from \"./abort.ts\";\nimport { combineAbortSignals } from \"./runtime-owner.ts\";\nimport type { McpExtensionState } from \"./state.ts\";\nimport {\n\tgetToolNameCandidates,\n\tMCP_TOOL_APPROVAL_REQUEST_EVENT,\n\ttype McpConfig,\n\ttype McpToolApprovalDecision,\n\ttype McpToolApprovalHandler,\n\ttype McpToolApprovalOrigin,\n\ttype McpToolApprovalRequest,\n\tmatchesToolPattern,\n\tresolveToolPrefix,\n\ttype ToolMetadata,\n} from \"./types.ts\";\nimport { sanitizeTerminalText } from \"./utils.ts\";\n\nexport type ToolCallApprovalResult = { ok: true } | { ok: false; reason: \"denied\" | \"approval_required_headless\" };\n\nfunction stableStringify(value: unknown): string {\n\tif (value === null || value === undefined || typeof value !== \"object\") {\n\t\tconst serialized = JSON.stringify(value);\n\t\treturn serialized === undefined ? \"undefined\" : serialized;\n\t}\n\tif (Array.isArray(value)) {\n\t\treturn `[${value.map((item) => stableStringify(item)).join(\",\")}]`;\n\t}\n\tconst object = value as Record<string, unknown>;\n\treturn `{${Object.keys(object)\n\t\t.sort()\n\t\t.map((key) => `${JSON.stringify(key)}:${stableStringify(object[key])}`)\n\t\t.join(\",\")}}`;\n}\n\nexport function isToolCallApprovalRequired(\n\tconfig: McpConfig,\n\tserverName: string,\n\ttoolMeta: Pick<ToolMetadata, \"originalName\">,\n\ttoolMetadata?: ReadonlyMap<string, readonly ToolMetadata[]>,\n): boolean {\n\tconst definition = config.mcpServers[serverName];\n\tconst serverApproval = definition?.approveTools;\n\tconst approval = serverApproval !== undefined ? serverApproval : config.settings?.approveTools;\n\n\tif (approval === true) return true;\n\tif (!Array.isArray(approval) || approval.length === 0) return false;\n\n\tconst prefix = resolveToolPrefix(definition, config.settings?.toolPrefix);\n\tconst currentCandidates = getToolNameCandidates(toolMeta.originalName, serverName, prefix, false);\n\tif (serverApproval !== undefined) {\n\t\tif (matchesToolPattern(currentCandidates, approval)) return true;\n\t\tif (!toolMetadata)\n\t\t\treturn matchesToolPattern(getToolNameCandidates(toolMeta.originalName, serverName, prefix), approval);\n\t\tconst legacyCandidates = getToolNameCandidates(toolMeta.originalName, serverName, prefix);\n\t\tconst legacyEmittedName = [...currentCandidates]\n\t\t\t.find((candidate) => candidate !== toolMeta.originalName)\n\t\t\t?.replace(/-/g, \"_\");\n\t\tif (legacyEmittedName) legacyCandidates.add(legacyEmittedName);\n\t\tfor (const candidate of currentCandidates) legacyCandidates.delete(candidate);\n\t\tconst otherCurrentCandidates = new Set<string>();\n\t\tfor (const tool of toolMetadata.get(serverName) ?? []) {\n\t\t\tfor (const candidate of getToolNameCandidates(tool.originalName, serverName, prefix, false)) {\n\t\t\t\totherCurrentCandidates.add(candidate);\n\t\t\t}\n\t\t}\n\t\tfor (const candidate of currentCandidates) otherCurrentCandidates.delete(candidate);\n\t\treturn approval.some(\n\t\t\t(pattern) =>\n\t\t\t\tmatchesToolPattern(legacyCandidates, [pattern]) && !matchesToolPattern(otherCurrentCandidates, [pattern]),\n\t\t);\n\t}\n\n\tif (matchesToolPattern(currentCandidates, approval)) return true;\n\tif (!toolMetadata) return false;\n\n\tconst legacyCandidates = getToolNameCandidates(toolMeta.originalName, serverName, prefix);\n\tconst legacyEmittedName = [...currentCandidates]\n\t\t.find((candidate) => candidate !== toolMeta.originalName)\n\t\t?.replace(/-/g, \"_\");\n\tif (legacyEmittedName) legacyCandidates.add(legacyEmittedName);\n\tfor (const candidate of currentCandidates) legacyCandidates.delete(candidate);\n\tconst otherCurrentCandidates = new Set<string>();\n\tfor (const [name, metadata] of toolMetadata) {\n\t\tconst otherPrefix = resolveToolPrefix(config.mcpServers[name], config.settings?.toolPrefix);\n\t\tfor (const tool of metadata) {\n\t\t\tfor (const candidate of getToolNameCandidates(tool.originalName, name, otherPrefix, false)) {\n\t\t\t\totherCurrentCandidates.add(candidate);\n\t\t\t}\n\t\t}\n\t}\n\tfor (const candidate of currentCandidates) otherCurrentCandidates.delete(candidate);\n\n\treturn approval.some(\n\t\t(pattern) =>\n\t\t\tmatchesToolPattern(legacyCandidates, [pattern]) && !matchesToolPattern(otherCurrentCandidates, [pattern]),\n\t);\n}\n\nfunction isMcpToolApprovalDecision(value: unknown): value is McpToolApprovalDecision {\n\treturn value === \"allow_once\" || value === \"allow_for_session\" || value === \"deny\" || value === \"abstain\";\n}\n\nasync function requestBrokerApproval(\n\tstate: McpExtensionState,\n\tserverName: string,\n\ttoolMeta: ToolMetadata,\n\targs: Record<string, unknown> | undefined,\n\torigin: McpToolApprovalOrigin,\n\tsignal?: AbortSignal,\n): Promise<McpToolApprovalDecision> {\n\tif (!state.approvalEvents) return \"abstain\";\n\n\tlet acceptingClaim = true;\n\tlet handler: McpToolApprovalHandler | undefined;\n\tconst request: McpToolApprovalRequest = {\n\t\trequestId: randomUUID(),\n\t\tserverName,\n\t\toriginalToolName: toolMeta.originalName,\n\t\tprefixedToolName: toolMeta.name,\n\t\targs: args ?? {},\n\t\torigin,\n\t\t...(signal !== undefined ? { signal } : {}),\n\t\tclaim(candidate: McpToolApprovalHandler) {\n\t\t\tif (!acceptingClaim || handler) return false;\n\t\t\thandler = candidate;\n\t\t\treturn true;\n\t\t},\n\t};\n\n\tstate.approvalEvents.emit(MCP_TOOL_APPROVAL_REQUEST_EVENT, request);\n\tacceptingClaim = false;\n\tif (!handler) return \"abstain\";\n\n\ttry {\n\t\tconst decision = await abortable(Promise.resolve().then(handler), signal);\n\t\treturn isMcpToolApprovalDecision(decision) ? decision : \"deny\";\n\t} catch (error) {\n\t\tif (signal?.aborted) throw error;\n\t\treturn \"deny\";\n\t}\n}\n\nexport async function ensureToolCallApproved(\n\tstate: McpExtensionState,\n\tserverName: string,\n\ttoolMeta: ToolMetadata,\n\targs: Record<string, unknown> | undefined,\n\tsignal?: AbortSignal,\n\torigin: McpToolApprovalOrigin = toolMeta.resourceUri ? \"resource\" : \"proxy\",\n\tapprovalMetadata?: ReadonlyMap<string, readonly ToolMetadata[]>,\n): Promise<ToolCallApprovalResult> {\n\tconst argsHash = createHash(\"sha256\")\n\t\t.update(stableStringify(args ?? {}))\n\t\t.digest(\"hex\");\n\tconst cacheKey = `${serverName}\\u0000${toolMeta.originalName}\\u0000${argsHash}`;\n\tif (!state.approvedToolCalls) state.approvedToolCalls = new Map<string, true>();\n\tconst approvedToolCalls = state.approvedToolCalls;\n\tif (approvedToolCalls.has(cacheKey)) {\n\t\treturn { ok: true };\n\t}\n\n\tconst brokerDecision = await requestBrokerApproval(state, serverName, toolMeta, args, origin, signal);\n\tif (brokerDecision === \"allow_once\") return { ok: true };\n\tif (brokerDecision === \"allow_for_session\") {\n\t\tapprovedToolCalls.set(cacheKey, true);\n\t\treturn { ok: true };\n\t}\n\tif (brokerDecision === \"deny\") return { ok: false, reason: \"denied\" };\n\n\tif (!isToolCallApprovalRequired(state.config, serverName, toolMeta, approvalMetadata ?? state.toolMetadata)) {\n\t\treturn { ok: true };\n\t}\n\n\tif (!state.ui) {\n\t\treturn { ok: false, reason: \"approval_required_headless\" };\n\t}\n\n\tconst json = JSON.stringify(args ?? {}, null, 2);\n\tconst sanitized = sanitizeTerminalText(json);\n\tconst preview = sanitized.length > 500 ? `${sanitized.slice(0, 500)}...` : sanitized;\n\tconst title = `MCP: ${sanitizeTerminalText(serverName)} wants to run ${sanitizeTerminalText(toolMeta.originalName)}`;\n\tconst ownedSignal = combineAbortSignals(state.owner?.signal, signal);\n\tconst decision = await abortable(\n\t\tstate.ui.select(`${title}\\n\\nArguments:\\n${preview}`, [\"Allow once\", \"Allow for session\", \"Deny\"]),\n\t\townedSignal,\n\t);\n\n\tif (decision === \"Allow once\") {\n\t\treturn { ok: true };\n\t}\n\tif (decision === \"Allow for session\") {\n\t\tapprovedToolCalls.set(cacheKey, true);\n\t\treturn { ok: true };\n\t}\n\treturn { ok: false, reason: \"denied\" };\n}\n"]}

package/dist/extensions/mcp/tool-metadata.d.ts

@@ -0,0 +1,11 @@
+import type { McpExtensionState } from "./state.ts";
+import type { McpResource, McpTool, ServerEntry, ToolMetadata, ToolPrefix } from "./types.ts";
+export declare function buildToolMetadata(tools: McpTool[], resources: McpResource[], definition: ServerEntry, serverName: string, prefix: ToolPrefix, configuredServers?: Record<string, ServerEntry>, knownMetadata?: Map<string, ToolMetadata[]>, includeMissingConfiguredCandidates?: boolean): {
+    metadata: ToolMetadata[];
+    failedTools: string[];
+};
+export declare function getToolNames(state: McpExtensionState, serverName: string): string[];
+export declare function totalToolCount(state: McpExtensionState): number;
+export declare function findToolByName(metadata: ToolMetadata[] | undefined, toolName: string): ToolMetadata | undefined;
+export declare function formatSchema(schema: unknown, indent?: string): string;
+//# sourceMappingURL=tool-metadata.d.ts.map

package/dist/extensions/mcp/tool-metadata.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-metadata.d.ts","sourceRoot":"","sources":["../../../src/extensions/mcp/tool-metadata.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AACpD,OAAO,KAAK,EAAE,WAAW,EAAE,OAAO,EAAE,WAAW,EAAE,YAAY,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAY9F,wBAAgB,iBAAiB,CAChC,KAAK,EAAE,OAAO,EAAE,EAChB,SAAS,EAAE,WAAW,EAAE,EACxB,UAAU,EAAE,WAAW,EACvB,UAAU,EAAE,MAAM,EAClB,MAAM,EAAE,UAAU,EAClB,iBAAiB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,WAAW,CAAC,EAC/C,aAAa,CAAC,EAAE,GAAG,CAAC,MAAM,EAAE,YAAY,EAAE,CAAC,EAC3C,kCAAkC,UAAQ,GACxC;IAAE,QAAQ,EAAE,YAAY,EAAE,CAAC;IAAC,WAAW,EAAE,MAAM,EAAE,CAAA;CAAE,CA2JrD;AAED,wBAAgB,YAAY,CAAC,KAAK,EAAE,iBAAiB,EAAE,UAAU,EAAE,MAAM,GAAG,MAAM,EAAE,CAEnF;AAED,wBAAgB,cAAc,CAAC,KAAK,EAAE,iBAAiB,GAAG,MAAM,CAM/D;AAED,wBAAgB,cAAc,CAAC,QAAQ,EAAE,YAAY,EAAE,GAAG,SAAS,EAAE,QAAQ,EAAE,MAAM,GAAG,YAAY,GAAG,SAAS,CAM/G;AAED,wBAAgB,YAAY,CAAC,MAAM,EAAE,OAAO,EAAE,MAAM,SAAO,GAAG,MAAM,CAmCnE","sourcesContent":["import { resourceNameToToolName } from \"./resource-tools.ts\";\nimport type { McpExtensionState } from \"./state.ts\";\nimport type { McpResource, McpTool, ServerEntry, ToolMetadata, ToolPrefix } from \"./types.ts\";\nimport {\n\tcreateToolSelectorCandidateIndex,\n\tformatToolName,\n\tgetToolNameCandidates,\n\tisToolAllowed,\n\tresolveToolPrefix,\n} from \"./types.ts\";\nimport { getToolUiResourceUri } from \"./ui-app-bridge-helpers.ts\";\nimport { extractUiToolVisibility, isUiToolVisibleToModel } from \"./ui-tool-visibility.ts\";\nimport { extractToolUiStreamMode } from \"./utils.ts\";\n\nexport function buildToolMetadata(\n\ttools: McpTool[],\n\tresources: McpResource[],\n\tdefinition: ServerEntry,\n\tserverName: string,\n\tprefix: ToolPrefix,\n\tconfiguredServers?: Record<string, ServerEntry>,\n\tknownMetadata?: Map<string, ToolMetadata[]>,\n\tincludeMissingConfiguredCandidates = false,\n): { metadata: ToolMetadata[]; failedTools: string[] } {\n\tconst metadata: ToolMetadata[] = [];\n\tconst failedTools: string[] = [];\n\tconst seenNames = new Set<string>();\n\tconst effectivePrefix = resolveToolPrefix(definition, prefix);\n\tconst hasToolFilters =\n\t\t(Array.isArray(definition.includeTools) && definition.includeTools.length > 0) ||\n\t\t(Array.isArray(definition.excludeTools) && definition.excludeTools.length > 0);\n\tconst selectorCandidateIndex =\n\t\thasToolFilters && configuredServers\n\t\t\t? (() => {\n\t\t\t\t\tconst candidates = new Set<string>();\n\t\t\t\t\tconst additionalCandidatesByToolName = new Map<string, Set<string>>();\n\t\t\t\t\tconst evaluatedToolNames = new Set<string>();\n\t\t\t\t\tconst addCandidates = (\n\t\t\t\t\t\ttarget: Set<string>,\n\t\t\t\t\t\toriginalName: string,\n\t\t\t\t\t\tcandidateServerName: string,\n\t\t\t\t\t\tcandidatePrefix: ToolPrefix,\n\t\t\t\t\t) => {\n\t\t\t\t\t\tfor (const candidate of getToolNameCandidates(\n\t\t\t\t\t\t\toriginalName,\n\t\t\t\t\t\t\tcandidateServerName,\n\t\t\t\t\t\t\tcandidatePrefix,\n\t\t\t\t\t\t\tfalse,\n\t\t\t\t\t\t))\n\t\t\t\t\t\t\ttarget.add(candidate);\n\t\t\t\t\t};\n\n\t\t\t\t\tfor (const tool of tools) {\n\t\t\t\t\t\tif (!tool?.name) continue;\n\t\t\t\t\t\tevaluatedToolNames.add(tool.name);\n\t\t\t\t\t\taddCandidates(candidates, tool.name, serverName, effectivePrefix);\n\t\t\t\t\t}\n\t\t\t\t\tif (definition.exposeResources !== false) {\n\t\t\t\t\t\tfor (const resource of resources) {\n\t\t\t\t\t\t\tconst baseName = `read_${resourceNameToToolName(resource.name)}`;\n\t\t\t\t\t\t\tevaluatedToolNames.add(baseName);\n\t\t\t\t\t\t\tif (resource?.name && resource?.uri)\n\t\t\t\t\t\t\t\taddCandidates(candidates, baseName, serverName, effectivePrefix);\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t\tfor (const [otherServerName, otherDefinition] of Object.entries(configuredServers)) {\n\t\t\t\t\t\tif (otherServerName === serverName) continue;\n\t\t\t\t\t\tconst otherPrefix = resolveToolPrefix(otherDefinition, prefix);\n\t\t\t\t\t\tconst knownTools = knownMetadata?.get(otherServerName);\n\t\t\t\t\t\tif (knownTools) {\n\t\t\t\t\t\t\tfor (const tool of knownTools) {\n\t\t\t\t\t\t\t\tcandidates.add(tool.name);\n\t\t\t\t\t\t\t\taddCandidates(candidates, tool.originalName, otherServerName, otherPrefix);\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t} else if (!knownMetadata || includeMissingConfiguredCandidates) {\n\t\t\t\t\t\t\tfor (const toolName of evaluatedToolNames) {\n\t\t\t\t\t\t\t\tlet additionalCandidates = additionalCandidatesByToolName.get(toolName);\n\t\t\t\t\t\t\t\tif (!additionalCandidates) {\n\t\t\t\t\t\t\t\t\tadditionalCandidates = new Set<string>();\n\t\t\t\t\t\t\t\t\tadditionalCandidatesByToolName.set(toolName, additionalCandidates);\n\t\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\t\taddCandidates(additionalCandidates, toolName, otherServerName, otherPrefix);\n\t\t\t\t\t\t\t\tif (includeMissingConfiguredCandidates) {\n\t\t\t\t\t\t\t\t\tfor (const candidate of getToolNameCandidates(\n\t\t\t\t\t\t\t\t\t\ttoolName,\n\t\t\t\t\t\t\t\t\t\totherServerName,\n\t\t\t\t\t\t\t\t\t\totherPrefix,\n\t\t\t\t\t\t\t\t\t\tfalse,\n\t\t\t\t\t\t\t\t\t)) {\n\t\t\t\t\t\t\t\t\t\tadditionalCandidates.add(candidate.replace(/-/g, \"_\"));\n\t\t\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t\treturn createToolSelectorCandidateIndex(candidates, additionalCandidatesByToolName);\n\t\t\t\t})()\n\t\t\t: undefined;\n\n\tfor (const tool of tools) {\n\t\tif (!tool?.name) {\n\t\t\tfailedTools.push(\"(unnamed)\");\n\t\t\tcontinue;\n\t\t}\n\t\tif (\n\t\t\t!isToolAllowed(\n\t\t\t\ttool.name,\n\t\t\t\tserverName,\n\t\t\t\teffectivePrefix,\n\t\t\t\tdefinition.includeTools,\n\t\t\t\tdefinition.excludeTools,\n\t\t\t\tselectorCandidateIndex,\n\t\t\t)\n\t\t) {\n\t\t\tcontinue;\n\t\t}\n\n\t\tconst name = formatToolName(tool.name, serverName, effectivePrefix);\n\t\tif (seenNames.has(name)) {\n\t\t\tcontinue;\n\t\t}\n\n\t\tconst uiVisibility = extractUiToolVisibility(tool._meta);\n\t\tif (!isUiToolVisibleToModel(uiVisibility)) {\n\t\t\tcontinue;\n\t\t}\n\t\tseenNames.add(name);\n\n\t\tlet uiResourceUri: string | undefined;\n\t\ttry {\n\t\t\tuiResourceUri = getToolUiResourceUri({ _meta: tool._meta });\n\t\t} catch {\n\t\t\tfailedTools.push(tool.name);\n\t\t}\n\t\tconst uiStreamMode = extractToolUiStreamMode(tool._meta);\n\t\tmetadata.push({\n\t\t\tname,\n\t\t\toriginalName: tool.name,\n\t\t\tdescription: tool.description ?? \"\",\n\t\t\t...(tool.inputSchema !== undefined ? { inputSchema: tool.inputSchema } : {}),\n\t\t\t...(uiResourceUri !== undefined ? { uiResourceUri } : {}),\n\t\t\t...(uiVisibility !== undefined ? { uiVisibility } : {}),\n\t\t\t...(uiStreamMode !== undefined ? { uiStreamMode } : {}),\n\t\t});\n\t}\n\n\tif (definition.exposeResources !== false) {\n\t\tfor (const resource of resources) {\n\t\t\tconst baseName = `read_${resourceNameToToolName(resource.name)}`;\n\t\t\tif (\n\t\t\t\t!isToolAllowed(\n\t\t\t\t\tbaseName,\n\t\t\t\t\tserverName,\n\t\t\t\t\teffectivePrefix,\n\t\t\t\t\tdefinition.includeTools,\n\t\t\t\t\tdefinition.excludeTools,\n\t\t\t\t\tselectorCandidateIndex,\n\t\t\t\t)\n\t\t\t) {\n\t\t\t\tcontinue;\n\t\t\t}\n\n\t\t\tconst name = formatToolName(baseName, serverName, effectivePrefix);\n\t\t\tif (seenNames.has(name)) {\n\t\t\t\tcontinue;\n\t\t\t}\n\t\t\tseenNames.add(name);\n\n\t\t\tmetadata.push({\n\t\t\t\tname,\n\t\t\t\toriginalName: baseName,\n\t\t\t\tdescription: resource.description ?? `Read resource: ${resource.uri}`,\n\t\t\t\tresourceUri: resource.uri,\n\t\t\t});\n\t\t}\n\t}\n\n\treturn { metadata, failedTools };\n}\n\nexport function getToolNames(state: McpExtensionState, serverName: string): string[] {\n\treturn state.toolMetadata.get(serverName)?.map((m) => m.name) ?? [];\n}\n\nexport function totalToolCount(state: McpExtensionState): number {\n\tlet count = 0;\n\tfor (const metadata of state.toolMetadata.values()) {\n\t\tcount += metadata.length;\n\t}\n\treturn count;\n}\n\nexport function findToolByName(metadata: ToolMetadata[] | undefined, toolName: string): ToolMetadata | undefined {\n\tif (!metadata) return undefined;\n\tconst exact = metadata.find((m) => m.name === toolName);\n\tif (exact) return exact;\n\tconst normalized = toolName.replace(/-/g, \"_\");\n\treturn metadata.find((m) => m.name.replace(/-/g, \"_\") === normalized);\n}\n\nexport function formatSchema(schema: unknown, indent = \"  \"): string {\n\tif (!schema || typeof schema !== \"object\" || Array.isArray(schema)) {\n\t\treturn `${indent}(no schema)`;\n\t}\n\n\tconst s = schema as Record<string, unknown>;\n\n\tif (s.type === \"object\" && s.properties && typeof s.properties === \"object\" && !Array.isArray(s.properties)) {\n\t\tconst props = s.properties as Record<string, unknown>;\n\t\tconst required = Array.isArray(s.required)\n\t\t\t? s.required.filter((name): name is string => typeof name === \"string\")\n\t\t\t: [];\n\n\t\tif (Object.keys(props).length === 0) {\n\t\t\treturn `${indent}(no parameters)`;\n\t\t}\n\n\t\tconst lines: string[] = [];\n\t\tfor (const [name, propSchema] of Object.entries(props)) {\n\t\t\tlines.push(...formatProperty(name, propSchema, required.includes(name), indent));\n\t\t}\n\t\treturn lines.join(\"\\n\");\n\t}\n\n\tconst lines = formatNestedSchema(s, indent);\n\tif (lines.length > 0) {\n\t\treturn lines.join(\"\\n\");\n\t}\n\n\tconst typeStr = formatType(s);\n\tif (typeStr) {\n\t\treturn `${indent}(${typeStr})`;\n\t}\n\n\treturn `${indent}(complex schema)`;\n}\n\nfunction formatProperty(name: string, schema: unknown, required: boolean, indent: string): string[] {\n\tif (!schema || typeof schema !== \"object\" || Array.isArray(schema)) {\n\t\treturn [`${indent}${name}${required ? \" *required*\" : \"\"}`];\n\t}\n\n\tconst s = schema as Record<string, unknown>;\n\tconst parts = [`${indent}${name}`];\n\tconst typeStr = formatType(s);\n\tif (typeStr) parts.push(`(${typeStr})`);\n\tif (required) parts.push(\"*required*\");\n\tappendSchemaAnnotations(parts, s);\n\n\treturn [parts.join(\" \"), ...formatNestedSchema(s, `${indent}  `)];\n}\n\nfunction formatNestedSchema(schema: Record<string, unknown>, indent: string): string[] {\n\tconst lines: string[] = [];\n\n\tif (Array.isArray(schema.anyOf)) {\n\t\tlines.push(...formatVariants(\"anyOf\", schema.anyOf, indent));\n\t}\n\tif (Array.isArray(schema.oneOf)) {\n\t\tlines.push(...formatVariants(\"oneOf\", schema.oneOf, indent));\n\t}\n\tif (schema.items !== undefined) {\n\t\tlines.push(...formatProperty(\"items\", schema.items, false, indent));\n\t}\n\tif (schema.properties && typeof schema.properties === \"object\" && !Array.isArray(schema.properties)) {\n\t\tconst required = Array.isArray(schema.required)\n\t\t\t? schema.required.filter((name): name is string => typeof name === \"string\")\n\t\t\t: [];\n\t\tfor (const [name, propSchema] of Object.entries(schema.properties as Record<string, unknown>)) {\n\t\t\tlines.push(...formatProperty(name, propSchema, required.includes(name), indent));\n\t\t}\n\t}\n\n\treturn lines;\n}\n\nfunction formatVariants(keyword: \"anyOf\" | \"oneOf\", variants: unknown[], indent: string): string[] {\n\tconst lines = [`${indent}${keyword}:`];\n\n\tfor (const variant of variants) {\n\t\tif (!variant || typeof variant !== \"object\" || Array.isArray(variant)) {\n\t\t\tlines.push(`${indent}  - ${JSON.stringify(variant)}`);\n\t\t\tcontinue;\n\t\t}\n\n\t\tconst s = variant as Record<string, unknown>;\n\t\tconst typeStr = formatType(s) || \"schema\";\n\t\tconst parts = [`${indent}  - ${typeStr}`];\n\t\tappendSchemaAnnotations(parts, s);\n\t\tlines.push(parts.join(\" \"));\n\t\tlines.push(...formatNestedSchema(s, `${indent}    `));\n\t}\n\n\treturn lines;\n}\n\nfunction formatType(schema: Record<string, unknown>): string {\n\tif (Object.hasOwn(schema, \"const\")) {\n\t\treturn `const ${JSON.stringify(schema.const)}`;\n\t}\n\n\tif (Array.isArray(schema.enum)) {\n\t\treturn `enum: ${schema.enum.map((v) => JSON.stringify(v)).join(\", \")}`;\n\t}\n\n\tif (Array.isArray(schema.type)) {\n\t\treturn schema.type.map((type) => String(type)).join(\" | \");\n\t}\n\n\tif (schema.type) {\n\t\treturn String(schema.type);\n\t}\n\n\tif (schema.properties && typeof schema.properties === \"object\" && !Array.isArray(schema.properties)) {\n\t\treturn \"object\";\n\t}\n\n\tif (schema.items !== undefined) {\n\t\treturn \"array\";\n\t}\n\n\treturn \"\";\n}\n\nfunction appendSchemaAnnotations(parts: string[], schema: Record<string, unknown>): void {\n\tif (schema.description && typeof schema.description === \"string\") {\n\t\tparts.push(`- ${schema.description}`);\n\t}\n\n\tfor (const key of [\n\t\t\"minLength\",\n\t\t\"maxLength\",\n\t\t\"minimum\",\n\t\t\"maximum\",\n\t\t\"minItems\",\n\t\t\"maxItems\",\n\t\t\"format\",\n\t\t\"pattern\",\n\t] as const) {\n\t\tif (schema[key] !== undefined) {\n\t\t\tparts.push(`[${key}: ${JSON.stringify(schema[key])}]`);\n\t\t}\n\t}\n\n\tif (schema.default !== undefined) {\n\t\tparts.push(`[default: ${JSON.stringify(schema.default)}]`);\n\t}\n}\n"]}

package/dist/extensions/mcp/tool-metadata.js

@@ -0,0 +1,264 @@
+import { resourceNameToToolName } from "./resource-tools.js";
+import { createToolSelectorCandidateIndex, formatToolName, getToolNameCandidates, isToolAllowed, resolveToolPrefix, } from "./types.js";
+import { getToolUiResourceUri } from "./ui-app-bridge-helpers.js";
+import { extractUiToolVisibility, isUiToolVisibleToModel } from "./ui-tool-visibility.js";
+import { extractToolUiStreamMode } from "./utils.js";
+export function buildToolMetadata(tools, resources, definition, serverName, prefix, configuredServers, knownMetadata, includeMissingConfiguredCandidates = false) {
+    const metadata = [];
+    const failedTools = [];
+    const seenNames = new Set();
+    const effectivePrefix = resolveToolPrefix(definition, prefix);
+    const hasToolFilters = (Array.isArray(definition.includeTools) && definition.includeTools.length > 0) ||
+        (Array.isArray(definition.excludeTools) && definition.excludeTools.length > 0);
+    const selectorCandidateIndex = hasToolFilters && configuredServers
+        ? (() => {
+            const candidates = new Set();
+            const additionalCandidatesByToolName = new Map();
+            const evaluatedToolNames = new Set();
+            const addCandidates = (target, originalName, candidateServerName, candidatePrefix) => {
+                for (const candidate of getToolNameCandidates(originalName, candidateServerName, candidatePrefix, false))
+                    target.add(candidate);
+            };
+            for (const tool of tools) {
+                if (!tool?.name)
+                    continue;
+                evaluatedToolNames.add(tool.name);
+                addCandidates(candidates, tool.name, serverName, effectivePrefix);
+            }
+            if (definition.exposeResources !== false) {
+                for (const resource of resources) {
+                    const baseName = `read_${resourceNameToToolName(resource.name)}`;
+                    evaluatedToolNames.add(baseName);
+                    if (resource?.name && resource?.uri)
+                        addCandidates(candidates, baseName, serverName, effectivePrefix);
+                }
+            }
+            for (const [otherServerName, otherDefinition] of Object.entries(configuredServers)) {
+                if (otherServerName === serverName)
+                    continue;
+                const otherPrefix = resolveToolPrefix(otherDefinition, prefix);
+                const knownTools = knownMetadata?.get(otherServerName);
+                if (knownTools) {
+                    for (const tool of knownTools) {
+                        candidates.add(tool.name);
+                        addCandidates(candidates, tool.originalName, otherServerName, otherPrefix);
+                    }
+                }
+                else if (!knownMetadata || includeMissingConfiguredCandidates) {
+                    for (const toolName of evaluatedToolNames) {
+                        let additionalCandidates = additionalCandidatesByToolName.get(toolName);
+                        if (!additionalCandidates) {
+                            additionalCandidates = new Set();
+                            additionalCandidatesByToolName.set(toolName, additionalCandidates);
+                        }
+                        addCandidates(additionalCandidates, toolName, otherServerName, otherPrefix);
+                        if (includeMissingConfiguredCandidates) {
+                            for (const candidate of getToolNameCandidates(toolName, otherServerName, otherPrefix, false)) {
+                                additionalCandidates.add(candidate.replace(/-/g, "_"));
+                            }
+                        }
+                    }
+                }
+            }
+            return createToolSelectorCandidateIndex(candidates, additionalCandidatesByToolName);
+        })()
+        : undefined;
+    for (const tool of tools) {
+        if (!tool?.name) {
+            failedTools.push("(unnamed)");
+            continue;
+        }
+        if (!isToolAllowed(tool.name, serverName, effectivePrefix, definition.includeTools, definition.excludeTools, selectorCandidateIndex)) {
+            continue;
+        }
+        const name = formatToolName(tool.name, serverName, effectivePrefix);
+        if (seenNames.has(name)) {
+            continue;
+        }
+        const uiVisibility = extractUiToolVisibility(tool._meta);
+        if (!isUiToolVisibleToModel(uiVisibility)) {
+            continue;
+        }
+        seenNames.add(name);
+        let uiResourceUri;
+        try {
+            uiResourceUri = getToolUiResourceUri({ _meta: tool._meta });
+        }
+        catch {
+            failedTools.push(tool.name);
+        }
+        const uiStreamMode = extractToolUiStreamMode(tool._meta);
+        metadata.push({
+            name,
+            originalName: tool.name,
+            description: tool.description ?? "",
+            ...(tool.inputSchema !== undefined ? { inputSchema: tool.inputSchema } : {}),
+            ...(uiResourceUri !== undefined ? { uiResourceUri } : {}),
+            ...(uiVisibility !== undefined ? { uiVisibility } : {}),
+            ...(uiStreamMode !== undefined ? { uiStreamMode } : {}),
+        });
+    }
+    if (definition.exposeResources !== false) {
+        for (const resource of resources) {
+            const baseName = `read_${resourceNameToToolName(resource.name)}`;
+            if (!isToolAllowed(baseName, serverName, effectivePrefix, definition.includeTools, definition.excludeTools, selectorCandidateIndex)) {
+                continue;
+            }
+            const name = formatToolName(baseName, serverName, effectivePrefix);
+            if (seenNames.has(name)) {
+                continue;
+            }
+            seenNames.add(name);
+            metadata.push({
+                name,
+                originalName: baseName,
+                description: resource.description ?? `Read resource: ${resource.uri}`,
+                resourceUri: resource.uri,
+            });
+        }
+    }
+    return { metadata, failedTools };
+}
+export function getToolNames(state, serverName) {
+    return state.toolMetadata.get(serverName)?.map((m) => m.name) ?? [];
+}
+export function totalToolCount(state) {
+    let count = 0;
+    for (const metadata of state.toolMetadata.values()) {
+        count += metadata.length;
+    }
+    return count;
+}
+export function findToolByName(metadata, toolName) {
+    if (!metadata)
+        return undefined;
+    const exact = metadata.find((m) => m.name === toolName);
+    if (exact)
+        return exact;
+    const normalized = toolName.replace(/-/g, "_");
+    return metadata.find((m) => m.name.replace(/-/g, "_") === normalized);
+}
+export function formatSchema(schema, indent = "  ") {
+    if (!schema || typeof schema !== "object" || Array.isArray(schema)) {
+        return `${indent}(no schema)`;
+    }
+    const s = schema;
+    if (s.type === "object" && s.properties && typeof s.properties === "object" && !Array.isArray(s.properties)) {
+        const props = s.properties;
+        const required = Array.isArray(s.required)
+            ? s.required.filter((name) => typeof name === "string")
+            : [];
+        if (Object.keys(props).length === 0) {
+            return `${indent}(no parameters)`;
+        }
+        const lines = [];
+        for (const [name, propSchema] of Object.entries(props)) {
+            lines.push(...formatProperty(name, propSchema, required.includes(name), indent));
+        }
+        return lines.join("\n");
+    }
+    const lines = formatNestedSchema(s, indent);
+    if (lines.length > 0) {
+        return lines.join("\n");
+    }
+    const typeStr = formatType(s);
+    if (typeStr) {
+        return `${indent}(${typeStr})`;
+    }
+    return `${indent}(complex schema)`;
+}
+function formatProperty(name, schema, required, indent) {
+    if (!schema || typeof schema !== "object" || Array.isArray(schema)) {
+        return [`${indent}${name}${required ? " *required*" : ""}`];
+    }
+    const s = schema;
+    const parts = [`${indent}${name}`];
+    const typeStr = formatType(s);
+    if (typeStr)
+        parts.push(`(${typeStr})`);
+    if (required)
+        parts.push("*required*");
+    appendSchemaAnnotations(parts, s);
+    return [parts.join(" "), ...formatNestedSchema(s, `${indent}  `)];
+}
+function formatNestedSchema(schema, indent) {
+    const lines = [];
+    if (Array.isArray(schema.anyOf)) {
+        lines.push(...formatVariants("anyOf", schema.anyOf, indent));
+    }
+    if (Array.isArray(schema.oneOf)) {
+        lines.push(...formatVariants("oneOf", schema.oneOf, indent));
+    }
+    if (schema.items !== undefined) {
+        lines.push(...formatProperty("items", schema.items, false, indent));
+    }
+    if (schema.properties && typeof schema.properties === "object" && !Array.isArray(schema.properties)) {
+        const required = Array.isArray(schema.required)
+            ? schema.required.filter((name) => typeof name === "string")
+            : [];
+        for (const [name, propSchema] of Object.entries(schema.properties)) {
+            lines.push(...formatProperty(name, propSchema, required.includes(name), indent));
+        }
+    }
+    return lines;
+}
+function formatVariants(keyword, variants, indent) {
+    const lines = [`${indent}${keyword}:`];
+    for (const variant of variants) {
+        if (!variant || typeof variant !== "object" || Array.isArray(variant)) {
+            lines.push(`${indent}  - ${JSON.stringify(variant)}`);
+            continue;
+        }
+        const s = variant;
+        const typeStr = formatType(s) || "schema";
+        const parts = [`${indent}  - ${typeStr}`];
+        appendSchemaAnnotations(parts, s);
+        lines.push(parts.join(" "));
+        lines.push(...formatNestedSchema(s, `${indent}    `));
+    }
+    return lines;
+}
+function formatType(schema) {
+    if (Object.hasOwn(schema, "const")) {
+        return `const ${JSON.stringify(schema.const)}`;
+    }
+    if (Array.isArray(schema.enum)) {
+        return `enum: ${schema.enum.map((v) => JSON.stringify(v)).join(", ")}`;
+    }
+    if (Array.isArray(schema.type)) {
+        return schema.type.map((type) => String(type)).join(" | ");
+    }
+    if (schema.type) {
+        return String(schema.type);
+    }
+    if (schema.properties && typeof schema.properties === "object" && !Array.isArray(schema.properties)) {
+        return "object";
+    }
+    if (schema.items !== undefined) {
+        return "array";
+    }
+    return "";
+}
+function appendSchemaAnnotations(parts, schema) {
+    if (schema.description && typeof schema.description === "string") {
+        parts.push(`- ${schema.description}`);
+    }
+    for (const key of [
+        "minLength",
+        "maxLength",
+        "minimum",
+        "maximum",
+        "minItems",
+        "maxItems",
+        "format",
+        "pattern",
+    ]) {
+        if (schema[key] !== undefined) {
+            parts.push(`[${key}: ${JSON.stringify(schema[key])}]`);
+        }
+    }
+    if (schema.default !== undefined) {
+        parts.push(`[default: ${JSON.stringify(schema.default)}]`);
+    }
+}
+//# sourceMappingURL=tool-metadata.js.map

package/dist/extensions/mcp/tool-metadata.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-metadata.js","sourceRoot":"","sources":["../../../src/extensions/mcp/tool-metadata.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,sBAAsB,EAAE,MAAM,qBAAqB,CAAC;AAG7D,OAAO,EACN,gCAAgC,EAChC,cAAc,EACd,qBAAqB,EACrB,aAAa,EACb,iBAAiB,GACjB,MAAM,YAAY,CAAC;AACpB,OAAO,EAAE,oBAAoB,EAAE,MAAM,4BAA4B,CAAC;AAClE,OAAO,EAAE,uBAAuB,EAAE,sBAAsB,EAAE,MAAM,yBAAyB,CAAC;AAC1F,OAAO,EAAE,uBAAuB,EAAE,MAAM,YAAY,CAAC;AAErD,MAAM,UAAU,iBAAiB,CAChC,KAAgB,EAChB,SAAwB,EACxB,UAAuB,EACvB,UAAkB,EAClB,MAAkB,EAClB,iBAA+C,EAC/C,aAA2C,EAC3C,kCAAkC,GAAG,KAAK,EACY;IACtD,MAAM,QAAQ,GAAmB,EAAE,CAAC;IACpC,MAAM,WAAW,GAAa,EAAE,CAAC;IACjC,MAAM,SAAS,GAAG,IAAI,GAAG,EAAU,CAAC;IACpC,MAAM,eAAe,GAAG,iBAAiB,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;IAC9D,MAAM,cAAc,GACnB,CAAC,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,YAAY,CAAC,IAAI,UAAU,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,CAAC;QAC9E,CAAC,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,YAAY,CAAC,IAAI,UAAU,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IAChF,MAAM,sBAAsB,GAC3B,cAAc,IAAI,iBAAiB;QAClC,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC;YACP,MAAM,UAAU,GAAG,IAAI,GAAG,EAAU,CAAC;YACrC,MAAM,8BAA8B,GAAG,IAAI,GAAG,EAAuB,CAAC;YACtE,MAAM,kBAAkB,GAAG,IAAI,GAAG,EAAU,CAAC;YAC7C,MAAM,aAAa,GAAG,CACrB,MAAmB,EACnB,YAAoB,EACpB,mBAA2B,EAC3B,eAA2B,EAC1B,EAAE,CAAC;gBACJ,KAAK,MAAM,SAAS,IAAI,qBAAqB,CAC5C,YAAY,EACZ,mBAAmB,EACnB,eAAe,EACf,KAAK,CACL;oBACA,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YAAA,CACvB,CAAC;YAEF,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBAC1B,IAAI,CAAC,IAAI,EAAE,IAAI;oBAAE,SAAS;gBAC1B,kBAAkB,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBAClC,aAAa,CAAC,UAAU,EAAE,IAAI,CAAC,IAAI,EAAE,UAAU,EAAE,eAAe,CAAC,CAAC;YACnE,CAAC;YACD,IAAI,UAAU,CAAC,eAAe,KAAK,KAAK,EAAE,CAAC;gBAC1C,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;oBAClC,MAAM,QAAQ,GAAG,QAAQ,sBAAsB,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;oBACjE,kBAAkB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;oBACjC,IAAI,QAAQ,EAAE,IAAI,IAAI,QAAQ,EAAE,GAAG;wBAClC,aAAa,CAAC,UAAU,EAAE,QAAQ,EAAE,UAAU,EAAE,eAAe,CAAC,CAAC;gBACnE,CAAC;YACF,CAAC;YACD,KAAK,MAAM,CAAC,eAAe,EAAE,eAAe,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,EAAE,CAAC;gBACpF,IAAI,eAAe,KAAK,UAAU;oBAAE,SAAS;gBAC7C,MAAM,WAAW,GAAG,iBAAiB,CAAC,eAAe,EAAE,MAAM,CAAC,CAAC;gBAC/D,MAAM,UAAU,GAAG,aAAa,EAAE,GAAG,CAAC,eAAe,CAAC,CAAC;gBACvD,IAAI,UAAU,EAAE,CAAC;oBAChB,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;wBAC/B,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;wBAC1B,aAAa,CAAC,UAAU,EAAE,IAAI,CAAC,YAAY,EAAE,eAAe,EAAE,WAAW,CAAC,CAAC;oBAC5E,CAAC;gBACF,CAAC;qBAAM,IAAI,CAAC,aAAa,IAAI,kCAAkC,EAAE,CAAC;oBACjE,KAAK,MAAM,QAAQ,IAAI,kBAAkB,EAAE,CAAC;wBAC3C,IAAI,oBAAoB,GAAG,8BAA8B,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;wBACxE,IAAI,CAAC,oBAAoB,EAAE,CAAC;4BAC3B,oBAAoB,GAAG,IAAI,GAAG,EAAU,CAAC;4BACzC,8BAA8B,CAAC,GAAG,CAAC,QAAQ,EAAE,oBAAoB,CAAC,CAAC;wBACpE,CAAC;wBACD,aAAa,CAAC,oBAAoB,EAAE,QAAQ,EAAE,eAAe,EAAE,WAAW,CAAC,CAAC;wBAC5E,IAAI,kCAAkC,EAAE,CAAC;4BACxC,KAAK,MAAM,SAAS,IAAI,qBAAqB,CAC5C,QAAQ,EACR,eAAe,EACf,WAAW,EACX,KAAK,CACL,EAAE,CAAC;gCACH,oBAAoB,CAAC,GAAG,CAAC,SAAS,CAAC,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC,CAAC;4BACxD,CAAC;wBACF,CAAC;oBACF,CAAC;gBACF,CAAC;YACF,CAAC;YACD,OAAO,gCAAgC,CAAC,UAAU,EAAE,8BAA8B,CAAC,CAAC;QAAA,CACpF,CAAC,EAAE;QACL,CAAC,CAAC,SAAS,CAAC;IAEd,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QAC1B,IAAI,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC;YACjB,WAAW,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YAC9B,SAAS;QACV,CAAC;QACD,IACC,CAAC,aAAa,CACb,IAAI,CAAC,IAAI,EACT,UAAU,EACV,eAAe,EACf,UAAU,CAAC,YAAY,EACvB,UAAU,CAAC,YAAY,EACvB,sBAAsB,CACtB,EACA,CAAC;YACF,SAAS;QACV,CAAC;QAED,MAAM,IAAI,GAAG,cAAc,CAAC,IAAI,CAAC,IAAI,EAAE,UAAU,EAAE,eAAe,CAAC,CAAC;QACpE,IAAI,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;YACzB,SAAS;QACV,CAAC;QAED,MAAM,YAAY,GAAG,uBAAuB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACzD,IAAI,CAAC,sBAAsB,CAAC,YAAY,CAAC,EAAE,CAAC;YAC3C,SAAS;QACV,CAAC;QACD,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAEpB,IAAI,aAAiC,CAAC;QACtC,IAAI,CAAC;YACJ,aAAa,GAAG,oBAAoB,CAAC,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC;QAC7D,CAAC;QAAC,MAAM,CAAC;YACR,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC7B,CAAC;QACD,MAAM,YAAY,GAAG,uBAAuB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACzD,QAAQ,CAAC,IAAI,CAAC;YACb,IAAI;YACJ,YAAY,EAAE,IAAI,CAAC,IAAI;YACvB,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,EAAE;YACnC,GAAG,CAAC,IAAI,CAAC,WAAW,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC5E,GAAG,CAAC,aAAa,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACzD,GAAG,CAAC,YAAY,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACvD,GAAG,CAAC,YAAY,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACvD,CAAC,CAAC;IACJ,CAAC;IAED,IAAI,UAAU,CAAC,eAAe,KAAK,KAAK,EAAE,CAAC;QAC1C,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;YAClC,MAAM,QAAQ,GAAG,QAAQ,sBAAsB,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;YACjE,IACC,CAAC,aAAa,CACb,QAAQ,EACR,UAAU,EACV,eAAe,EACf,UAAU,CAAC,YAAY,EACvB,UAAU,CAAC,YAAY,EACvB,sBAAsB,CACtB,EACA,CAAC;gBACF,SAAS;YACV,CAAC;YAED,MAAM,IAAI,GAAG,cAAc,CAAC,QAAQ,EAAE,UAAU,EAAE,eAAe,CAAC,CAAC;YACnE,IAAI,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;gBACzB,SAAS;YACV,CAAC;YACD,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YAEpB,QAAQ,CAAC,IAAI,CAAC;gBACb,IAAI;gBACJ,YAAY,EAAE,QAAQ;gBACtB,WAAW,EAAE,QAAQ,CAAC,WAAW,IAAI,kBAAkB,QAAQ,CAAC,GAAG,EAAE;gBACrE,WAAW,EAAE,QAAQ,CAAC,GAAG;aACzB,CAAC,CAAC;QACJ,CAAC;IACF,CAAC;IAED,OAAO,EAAE,QAAQ,EAAE,WAAW,EAAE,CAAC;AAAA,CACjC;AAED,MAAM,UAAU,YAAY,CAAC,KAAwB,EAAE,UAAkB,EAAY;IACpF,OAAO,KAAK,CAAC,YAAY,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;AAAA,CACpE;AAED,MAAM,UAAU,cAAc,CAAC,KAAwB,EAAU;IAChE,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,KAAK,MAAM,QAAQ,IAAI,KAAK,CAAC,YAAY,CAAC,MAAM,EAAE,EAAE,CAAC;QACpD,KAAK,IAAI,QAAQ,CAAC,MAAM,CAAC;IAC1B,CAAC;IACD,OAAO,KAAK,CAAC;AAAA,CACb;AAED,MAAM,UAAU,cAAc,CAAC,QAAoC,EAAE,QAAgB,EAA4B;IAChH,IAAI,CAAC,QAAQ;QAAE,OAAO,SAAS,CAAC;IAChC,MAAM,KAAK,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;IACxD,IAAI,KAAK;QAAE,OAAO,KAAK,CAAC;IACxB,MAAM,UAAU,GAAG,QAAQ,CAAC,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;IAC/C,OAAO,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC,KAAK,UAAU,CAAC,CAAC;AAAA,CACtE;AAED,MAAM,UAAU,YAAY,CAAC,MAAe,EAAE,MAAM,GAAG,IAAI,EAAU;IACpE,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;QACpE,OAAO,GAAG,MAAM,aAAa,CAAC;IAC/B,CAAC;IAED,MAAM,CAAC,GAAG,MAAiC,CAAC;IAE5C,IAAI,CAAC,CAAC,IAAI,KAAK,QAAQ,IAAI,CAAC,CAAC,UAAU,IAAI,OAAO,CAAC,CAAC,UAAU,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,UAAU,CAAC,EAAE,CAAC;QAC7G,MAAM,KAAK,GAAG,CAAC,CAAC,UAAqC,CAAC;QACtD,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,QAAQ,CAAC;YACzC,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,IAAI,EAAkB,EAAE,CAAC,OAAO,IAAI,KAAK,QAAQ,CAAC;YACvE,CAAC,CAAC,EAAE,CAAC;QAEN,IAAI,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACrC,OAAO,GAAG,MAAM,iBAAiB,CAAC;QACnC,CAAC;QAED,MAAM,KAAK,GAAa,EAAE,CAAC;QAC3B,KAAK,MAAM,CAAC,IAAI,EAAE,UAAU,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YACxD,KAAK,CAAC,IAAI,CAAC,GAAG,cAAc,CAAC,IAAI,EAAE,UAAU,EAAE,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC;QAClF,CAAC;QACD,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACzB,CAAC;IAED,MAAM,KAAK,GAAG,kBAAkB,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC;IAC5C,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACtB,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACzB,CAAC;IAED,MAAM,OAAO,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC;IAC9B,IAAI,OAAO,EAAE,CAAC;QACb,OAAO,GAAG,MAAM,IAAI,OAAO,GAAG,CAAC;IAChC,CAAC;IAED,OAAO,GAAG,MAAM,kBAAkB,CAAC;AAAA,CACnC;AAED,SAAS,cAAc,CAAC,IAAY,EAAE,MAAe,EAAE,QAAiB,EAAE,MAAc,EAAY;IACnG,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;QACpE,OAAO,CAAC,GAAG,MAAM,GAAG,IAAI,GAAG,QAAQ,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;IAC7D,CAAC;IAED,MAAM,CAAC,GAAG,MAAiC,CAAC;IAC5C,MAAM,KAAK,GAAG,CAAC,GAAG,MAAM,GAAG,IAAI,EAAE,CAAC,CAAC;IACnC,MAAM,OAAO,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC;IAC9B,IAAI,OAAO;QAAE,KAAK,CAAC,IAAI,CAAC,IAAI,OAAO,GAAG,CAAC,CAAC;IACxC,IAAI,QAAQ;QAAE,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IACvC,uBAAuB,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;IAElC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,GAAG,kBAAkB,CAAC,CAAC,EAAE,GAAG,MAAM,IAAI,CAAC,CAAC,CAAC;AAAA,CAClE;AAED,SAAS,kBAAkB,CAAC,MAA+B,EAAE,MAAc,EAAY;IACtF,MAAM,KAAK,GAAa,EAAE,CAAC;IAE3B,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;QACjC,KAAK,CAAC,IAAI,CAAC,GAAG,cAAc,CAAC,OAAO,EAAE,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC;IAC9D,CAAC;IACD,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;QACjC,KAAK,CAAC,IAAI,CAAC,GAAG,cAAc,CAAC,OAAO,EAAE,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC;IAC9D,CAAC;IACD,IAAI,MAAM,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;QAChC,KAAK,CAAC,IAAI,CAAC,GAAG,cAAc,CAAC,OAAO,EAAE,MAAM,CAAC,KAAK,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC;IACrE,CAAC;IACD,IAAI,MAAM,CAAC,UAAU,IAAI,OAAO,MAAM,CAAC,UAAU,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,UAAU,CAAC,EAAE,CAAC;QACrG,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC;YAC9C,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,IAAI,EAAkB,EAAE,CAAC,OAAO,IAAI,KAAK,QAAQ,CAAC;YAC5E,CAAC,CAAC,EAAE,CAAC;QACN,KAAK,MAAM,CAAC,IAAI,EAAE,UAAU,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,UAAqC,CAAC,EAAE,CAAC;YAC/F,KAAK,CAAC,IAAI,CAAC,GAAG,cAAc,CAAC,IAAI,EAAE,UAAU,EAAE,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC;QAClF,CAAC;IACF,CAAC;IAED,OAAO,KAAK,CAAC;AAAA,CACb;AAED,SAAS,cAAc,CAAC,OAA0B,EAAE,QAAmB,EAAE,MAAc,EAAY;IAClG,MAAM,KAAK,GAAG,CAAC,GAAG,MAAM,GAAG,OAAO,GAAG,CAAC,CAAC;IAEvC,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAChC,IAAI,CAAC,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;YACvE,KAAK,CAAC,IAAI,CAAC,GAAG,MAAM,OAAO,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YACtD,SAAS;QACV,CAAC;QAED,MAAM,CAAC,GAAG,OAAkC,CAAC;QAC7C,MAAM,OAAO,GAAG,UAAU,CAAC,CAAC,CAAC,IAAI,QAAQ,CAAC;QAC1C,MAAM,KAAK,GAAG,CAAC,GAAG,MAAM,OAAO,OAAO,EAAE,CAAC,CAAC;QAC1C,uBAAuB,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QAClC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;QAC5B,KAAK,CAAC,IAAI,CAAC,GAAG,kBAAkB,CAAC,CAAC,EAAE,GAAG,MAAM,MAAM,CAAC,CAAC,CAAC;IACvD,CAAC;IAED,OAAO,KAAK,CAAC;AAAA,CACb;AAED,SAAS,UAAU,CAAC,MAA+B,EAAU;IAC5D,IAAI,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,CAAC;QACpC,OAAO,SAAS,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;IAChD,CAAC;IAED,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;QAChC,OAAO,SAAS,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;IACxE,CAAC;IAED,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;QAChC,OAAO,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC5D,CAAC;IAED,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;QACjB,OAAO,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IAC5B,CAAC;IAED,IAAI,MAAM,CAAC,UAAU,IAAI,OAAO,MAAM,CAAC,UAAU,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,UAAU,CAAC,EAAE,CAAC;QACrG,OAAO,QAAQ,CAAC;IACjB,CAAC;IAED,IAAI,MAAM,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;QAChC,OAAO,OAAO,CAAC;IAChB,CAAC;IAED,OAAO,EAAE,CAAC;AAAA,CACV;AAED,SAAS,uBAAuB,CAAC,KAAe,EAAE,MAA+B,EAAQ;IACxF,IAAI,MAAM,CAAC,WAAW,IAAI,OAAO,MAAM,CAAC,WAAW,KAAK,QAAQ,EAAE,CAAC;QAClE,KAAK,CAAC,IAAI,CAAC,KAAK,MAAM,CAAC,WAAW,EAAE,CAAC,CAAC;IACvC,CAAC;IAED,KAAK,MAAM,GAAG,IAAI;QACjB,WAAW;QACX,WAAW;QACX,SAAS;QACT,SAAS;QACT,UAAU;QACV,UAAU;QACV,QAAQ;QACR,SAAS;KACA,EAAE,CAAC;QACZ,IAAI,MAAM,CAAC,GAAG,CAAC,KAAK,SAAS,EAAE,CAAC;YAC/B,KAAK,CAAC,IAAI,CAAC,IAAI,GAAG,KAAK,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC;QACxD,CAAC;IACF,CAAC;IAED,IAAI,MAAM,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;QAClC,KAAK,CAAC,IAAI,CAAC,aAAa,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;IAC5D,CAAC;AAAA,CACD","sourcesContent":["import { resourceNameToToolName } from \"./resource-tools.ts\";\nimport type { McpExtensionState } from \"./state.ts\";\nimport type { McpResource, McpTool, ServerEntry, ToolMetadata, ToolPrefix } from \"./types.ts\";\nimport {\n\tcreateToolSelectorCandidateIndex,\n\tformatToolName,\n\tgetToolNameCandidates,\n\tisToolAllowed,\n\tresolveToolPrefix,\n} from \"./types.ts\";\nimport { getToolUiResourceUri } from \"./ui-app-bridge-helpers.ts\";\nimport { extractUiToolVisibility, isUiToolVisibleToModel } from \"./ui-tool-visibility.ts\";\nimport { extractToolUiStreamMode } from \"./utils.ts\";\n\nexport function buildToolMetadata(\n\ttools: McpTool[],\n\tresources: McpResource[],\n\tdefinition: ServerEntry,\n\tserverName: string,\n\tprefix: ToolPrefix,\n\tconfiguredServers?: Record<string, ServerEntry>,\n\tknownMetadata?: Map<string, ToolMetadata[]>,\n\tincludeMissingConfiguredCandidates = false,\n): { metadata: ToolMetadata[]; failedTools: string[] } {\n\tconst metadata: ToolMetadata[] = [];\n\tconst failedTools: string[] = [];\n\tconst seenNames = new Set<string>();\n\tconst effectivePrefix = resolveToolPrefix(definition, prefix);\n\tconst hasToolFilters =\n\t\t(Array.isArray(definition.includeTools) && definition.includeTools.length > 0) ||\n\t\t(Array.isArray(definition.excludeTools) && definition.excludeTools.length > 0);\n\tconst selectorCandidateIndex =\n\t\thasToolFilters && configuredServers\n\t\t\t? (() => {\n\t\t\t\t\tconst candidates = new Set<string>();\n\t\t\t\t\tconst additionalCandidatesByToolName = new Map<string, Set<string>>();\n\t\t\t\t\tconst evaluatedToolNames = new Set<string>();\n\t\t\t\t\tconst addCandidates = (\n\t\t\t\t\t\ttarget: Set<string>,\n\t\t\t\t\t\toriginalName: string,\n\t\t\t\t\t\tcandidateServerName: string,\n\t\t\t\t\t\tcandidatePrefix: ToolPrefix,\n\t\t\t\t\t) => {\n\t\t\t\t\t\tfor (const candidate of getToolNameCandidates(\n\t\t\t\t\t\t\toriginalName,\n\t\t\t\t\t\t\tcandidateServerName,\n\t\t\t\t\t\t\tcandidatePrefix,\n\t\t\t\t\t\t\tfalse,\n\t\t\t\t\t\t))\n\t\t\t\t\t\t\ttarget.add(candidate);\n\t\t\t\t\t};\n\n\t\t\t\t\tfor (const tool of tools) {\n\t\t\t\t\t\tif (!tool?.name) continue;\n\t\t\t\t\t\tevaluatedToolNames.add(tool.name);\n\t\t\t\t\t\taddCandidates(candidates, tool.name, serverName, effectivePrefix);\n\t\t\t\t\t}\n\t\t\t\t\tif (definition.exposeResources !== false) {\n\t\t\t\t\t\tfor (const resource of resources) {\n\t\t\t\t\t\t\tconst baseName = `read_${resourceNameToToolName(resource.name)}`;\n\t\t\t\t\t\t\tevaluatedToolNames.add(baseName);\n\t\t\t\t\t\t\tif (resource?.name && resource?.uri)\n\t\t\t\t\t\t\t\taddCandidates(candidates, baseName, serverName, effectivePrefix);\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t\tfor (const [otherServerName, otherDefinition] of Object.entries(configuredServers)) {\n\t\t\t\t\t\tif (otherServerName === serverName) continue;\n\t\t\t\t\t\tconst otherPrefix = resolveToolPrefix(otherDefinition, prefix);\n\t\t\t\t\t\tconst knownTools = knownMetadata?.get(otherServerName);\n\t\t\t\t\t\tif (knownTools) {\n\t\t\t\t\t\t\tfor (const tool of knownTools) {\n\t\t\t\t\t\t\t\tcandidates.add(tool.name);\n\t\t\t\t\t\t\t\taddCandidates(candidates, tool.originalName, otherServerName, otherPrefix);\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t} else if (!knownMetadata || includeMissingConfiguredCandidates) {\n\t\t\t\t\t\t\tfor (const toolName of evaluatedToolNames) {\n\t\t\t\t\t\t\t\tlet additionalCandidates = additionalCandidatesByToolName.get(toolName);\n\t\t\t\t\t\t\t\tif (!additionalCandidates) {\n\t\t\t\t\t\t\t\t\tadditionalCandidates = new Set<string>();\n\t\t\t\t\t\t\t\t\tadditionalCandidatesByToolName.set(toolName, additionalCandidates);\n\t\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\t\taddCandidates(additionalCandidates, toolName, otherServerName, otherPrefix);\n\t\t\t\t\t\t\t\tif (includeMissingConfiguredCandidates) {\n\t\t\t\t\t\t\t\t\tfor (const candidate of getToolNameCandidates(\n\t\t\t\t\t\t\t\t\t\ttoolName,\n\t\t\t\t\t\t\t\t\t\totherServerName,\n\t\t\t\t\t\t\t\t\t\totherPrefix,\n\t\t\t\t\t\t\t\t\t\tfalse,\n\t\t\t\t\t\t\t\t\t)) {\n\t\t\t\t\t\t\t\t\t\tadditionalCandidates.add(candidate.replace(/-/g, \"_\"));\n\t\t\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t\treturn createToolSelectorCandidateIndex(candidates, additionalCandidatesByToolName);\n\t\t\t\t})()\n\t\t\t: undefined;\n\n\tfor (const tool of tools) {\n\t\tif (!tool?.name) {\n\t\t\tfailedTools.push(\"(unnamed)\");\n\t\t\tcontinue;\n\t\t}\n\t\tif (\n\t\t\t!isToolAllowed(\n\t\t\t\ttool.name,\n\t\t\t\tserverName,\n\t\t\t\teffectivePrefix,\n\t\t\t\tdefinition.includeTools,\n\t\t\t\tdefinition.excludeTools,\n\t\t\t\tselectorCandidateIndex,\n\t\t\t)\n\t\t) {\n\t\t\tcontinue;\n\t\t}\n\n\t\tconst name = formatToolName(tool.name, serverName, effectivePrefix);\n\t\tif (seenNames.has(name)) {\n\t\t\tcontinue;\n\t\t}\n\n\t\tconst uiVisibility = extractUiToolVisibility(tool._meta);\n\t\tif (!isUiToolVisibleToModel(uiVisibility)) {\n\t\t\tcontinue;\n\t\t}\n\t\tseenNames.add(name);\n\n\t\tlet uiResourceUri: string | undefined;\n\t\ttry {\n\t\t\tuiResourceUri = getToolUiResourceUri({ _meta: tool._meta });\n\t\t} catch {\n\t\t\tfailedTools.push(tool.name);\n\t\t}\n\t\tconst uiStreamMode = extractToolUiStreamMode(tool._meta);\n\t\tmetadata.push({\n\t\t\tname,\n\t\t\toriginalName: tool.name,\n\t\t\tdescription: tool.description ?? \"\",\n\t\t\t...(tool.inputSchema !== undefined ? { inputSchema: tool.inputSchema } : {}),\n\t\t\t...(uiResourceUri !== undefined ? { uiResourceUri } : {}),\n\t\t\t...(uiVisibility !== undefined ? { uiVisibility } : {}),\n\t\t\t...(uiStreamMode !== undefined ? { uiStreamMode } : {}),\n\t\t});\n\t}\n\n\tif (definition.exposeResources !== false) {\n\t\tfor (const resource of resources) {\n\t\t\tconst baseName = `read_${resourceNameToToolName(resource.name)}`;\n\t\t\tif (\n\t\t\t\t!isToolAllowed(\n\t\t\t\t\tbaseName,\n\t\t\t\t\tserverName,\n\t\t\t\t\teffectivePrefix,\n\t\t\t\t\tdefinition.includeTools,\n\t\t\t\t\tdefinition.excludeTools,\n\t\t\t\t\tselectorCandidateIndex,\n\t\t\t\t)\n\t\t\t) {\n\t\t\t\tcontinue;\n\t\t\t}\n\n\t\t\tconst name = formatToolName(baseName, serverName, effectivePrefix);\n\t\t\tif (seenNames.has(name)) {\n\t\t\t\tcontinue;\n\t\t\t}\n\t\t\tseenNames.add(name);\n\n\t\t\tmetadata.push({\n\t\t\t\tname,\n\t\t\t\toriginalName: baseName,\n\t\t\t\tdescription: resource.description ?? `Read resource: ${resource.uri}`,\n\t\t\t\tresourceUri: resource.uri,\n\t\t\t});\n\t\t}\n\t}\n\n\treturn { metadata, failedTools };\n}\n\nexport function getToolNames(state: McpExtensionState, serverName: string): string[] {\n\treturn state.toolMetadata.get(serverName)?.map((m) => m.name) ?? [];\n}\n\nexport function totalToolCount(state: McpExtensionState): number {\n\tlet count = 0;\n\tfor (const metadata of state.toolMetadata.values()) {\n\t\tcount += metadata.length;\n\t}\n\treturn count;\n}\n\nexport function findToolByName(metadata: ToolMetadata[] | undefined, toolName: string): ToolMetadata | undefined {\n\tif (!metadata) return undefined;\n\tconst exact = metadata.find((m) => m.name === toolName);\n\tif (exact) return exact;\n\tconst normalized = toolName.replace(/-/g, \"_\");\n\treturn metadata.find((m) => m.name.replace(/-/g, \"_\") === normalized);\n}\n\nexport function formatSchema(schema: unknown, indent = \"  \"): string {\n\tif (!schema || typeof schema !== \"object\" || Array.isArray(schema)) {\n\t\treturn `${indent}(no schema)`;\n\t}\n\n\tconst s = schema as Record<string, unknown>;\n\n\tif (s.type === \"object\" && s.properties && typeof s.properties === \"object\" && !Array.isArray(s.properties)) {\n\t\tconst props = s.properties as Record<string, unknown>;\n\t\tconst required = Array.isArray(s.required)\n\t\t\t? s.required.filter((name): name is string => typeof name === \"string\")\n\t\t\t: [];\n\n\t\tif (Object.keys(props).length === 0) {\n\t\t\treturn `${indent}(no parameters)`;\n\t\t}\n\n\t\tconst lines: string[] = [];\n\t\tfor (const [name, propSchema] of Object.entries(props)) {\n\t\t\tlines.push(...formatProperty(name, propSchema, required.includes(name), indent));\n\t\t}\n\t\treturn lines.join(\"\\n\");\n\t}\n\n\tconst lines = formatNestedSchema(s, indent);\n\tif (lines.length > 0) {\n\t\treturn lines.join(\"\\n\");\n\t}\n\n\tconst typeStr = formatType(s);\n\tif (typeStr) {\n\t\treturn `${indent}(${typeStr})`;\n\t}\n\n\treturn `${indent}(complex schema)`;\n}\n\nfunction formatProperty(name: string, schema: unknown, required: boolean, indent: string): string[] {\n\tif (!schema || typeof schema !== \"object\" || Array.isArray(schema)) {\n\t\treturn [`${indent}${name}${required ? \" *required*\" : \"\"}`];\n\t}\n\n\tconst s = schema as Record<string, unknown>;\n\tconst parts = [`${indent}${name}`];\n\tconst typeStr = formatType(s);\n\tif (typeStr) parts.push(`(${typeStr})`);\n\tif (required) parts.push(\"*required*\");\n\tappendSchemaAnnotations(parts, s);\n\n\treturn [parts.join(\" \"), ...formatNestedSchema(s, `${indent}  `)];\n}\n\nfunction formatNestedSchema(schema: Record<string, unknown>, indent: string): string[] {\n\tconst lines: string[] = [];\n\n\tif (Array.isArray(schema.anyOf)) {\n\t\tlines.push(...formatVariants(\"anyOf\", schema.anyOf, indent));\n\t}\n\tif (Array.isArray(schema.oneOf)) {\n\t\tlines.push(...formatVariants(\"oneOf\", schema.oneOf, indent));\n\t}\n\tif (schema.items !== undefined) {\n\t\tlines.push(...formatProperty(\"items\", schema.items, false, indent));\n\t}\n\tif (schema.properties && typeof schema.properties === \"object\" && !Array.isArray(schema.properties)) {\n\t\tconst required = Array.isArray(schema.required)\n\t\t\t? schema.required.filter((name): name is string => typeof name === \"string\")\n\t\t\t: [];\n\t\tfor (const [name, propSchema] of Object.entries(schema.properties as Record<string, unknown>)) {\n\t\t\tlines.push(...formatProperty(name, propSchema, required.includes(name), indent));\n\t\t}\n\t}\n\n\treturn lines;\n}\n\nfunction formatVariants(keyword: \"anyOf\" | \"oneOf\", variants: unknown[], indent: string): string[] {\n\tconst lines = [`${indent}${keyword}:`];\n\n\tfor (const variant of variants) {\n\t\tif (!variant || typeof variant !== \"object\" || Array.isArray(variant)) {\n\t\t\tlines.push(`${indent}  - ${JSON.stringify(variant)}`);\n\t\t\tcontinue;\n\t\t}\n\n\t\tconst s = variant as Record<string, unknown>;\n\t\tconst typeStr = formatType(s) || \"schema\";\n\t\tconst parts = [`${indent}  - ${typeStr}`];\n\t\tappendSchemaAnnotations(parts, s);\n\t\tlines.push(parts.join(\" \"));\n\t\tlines.push(...formatNestedSchema(s, `${indent}    `));\n\t}\n\n\treturn lines;\n}\n\nfunction formatType(schema: Record<string, unknown>): string {\n\tif (Object.hasOwn(schema, \"const\")) {\n\t\treturn `const ${JSON.stringify(schema.const)}`;\n\t}\n\n\tif (Array.isArray(schema.enum)) {\n\t\treturn `enum: ${schema.enum.map((v) => JSON.stringify(v)).join(\", \")}`;\n\t}\n\n\tif (Array.isArray(schema.type)) {\n\t\treturn schema.type.map((type) => String(type)).join(\" | \");\n\t}\n\n\tif (schema.type) {\n\t\treturn String(schema.type);\n\t}\n\n\tif (schema.properties && typeof schema.properties === \"object\" && !Array.isArray(schema.properties)) {\n\t\treturn \"object\";\n\t}\n\n\tif (schema.items !== undefined) {\n\t\treturn \"array\";\n\t}\n\n\treturn \"\";\n}\n\nfunction appendSchemaAnnotations(parts: string[], schema: Record<string, unknown>): void {\n\tif (schema.description && typeof schema.description === \"string\") {\n\t\tparts.push(`- ${schema.description}`);\n\t}\n\n\tfor (const key of [\n\t\t\"minLength\",\n\t\t\"maxLength\",\n\t\t\"minimum\",\n\t\t\"maximum\",\n\t\t\"minItems\",\n\t\t\"maxItems\",\n\t\t\"format\",\n\t\t\"pattern\",\n\t] as const) {\n\t\tif (schema[key] !== undefined) {\n\t\t\tparts.push(`[${key}: ${JSON.stringify(schema[key])}]`);\n\t\t}\n\t}\n\n\tif (schema.default !== undefined) {\n\t\tparts.push(`[default: ${JSON.stringify(schema.default)}]`);\n\t}\n}\n"]}

package/dist/extensions/mcp/tool-registrar.d.ts

@@ -0,0 +1,21 @@
+import type { ContentBlock, McpContent } from "./types.ts";
+type McpResourceContent = {
+    uri: string;
+    text?: string | undefined;
+    blob?: string | undefined;
+    mimeType?: string | undefined;
+    [key: string]: unknown;
+};
+export declare function cleanupMaterializedBinaryResources(scope?: object): void;
+/**
+ * Transform MCP content types to Nexus content blocks.
+ */
+export declare function transformMcpResourceContents(contents: McpResourceContent[], scope?: object): ContentBlock[];
+export declare function transformMcpContent(content: McpContent[], scope?: object): ContentBlock[];
+/**
+ * Resolve a tool result's content blocks, falling back to structuredContent
+ * when content is empty.
+ */
+export declare function resolveMcpResultContent(result: Record<string, unknown>, scope?: object): ContentBlock[];
+export {};
+//# sourceMappingURL=tool-registrar.d.ts.map

package/dist/extensions/mcp/tool-registrar.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-registrar.d.ts","sourceRoot":"","sources":["../../../src/extensions/mcp/tool-registrar.ts"],"names":[],"mappings":"AAOA,OAAO,KAAK,EAAE,YAAY,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAkD3D,KAAK,kBAAkB,GAAG;IACzB,GAAG,EAAE,MAAM,CAAC;IACZ,IAAI,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC1B,IAAI,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC1B,QAAQ,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC9B,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;CACvB,CAAC;AA8CF,wBAAgB,kCAAkC,CAAC,KAAK,CAAC,EAAE,MAAM,GAAG,IAAI,CAYvE;AA8DD;;GAEG;AACH,wBAAgB,4BAA4B,CAAC,QAAQ,EAAE,kBAAkB,EAAE,EAAE,KAAK,CAAC,EAAE,MAAM,GAAG,YAAY,EAAE,CAO3G;AAED,wBAAgB,mBAAmB,CAAC,OAAO,EAAE,UAAU,EAAE,EAAE,KAAK,CAAC,EAAE,MAAM,GAAG,YAAY,EAAE,CA2CzF;AAED;;;GAGG;AACH,wBAAgB,uBAAuB,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,KAAK,CAAC,EAAE,MAAM,GAAG,YAAY,EAAE,CASvG","sourcesContent":["// tool-registrar.ts - MCP content transformation\n// NOTE: Tools are NOT registered with Nexus - only the unified `mcp` proxy tool is registered.\n// This keeps the LLM context small (1 tool instead of 100s).\n\nimport { mkdtempSync, rmSync, writeFileSync } from \"node:fs\";\nimport { tmpdir } from \"node:os\";\nimport { join } from \"node:path\";\nimport type { ContentBlock, McpContent } from \"./types.ts\";\n\nconst MAX_BINARY_RESOURCE_BYTES = 10 * 1024 * 1024;\nconst MAX_SESSION_RESOURCE_BYTES = 100 * 1024 * 1024;\nconst MAX_SESSION_RESOURCE_FILES = 10_000; // Bounds metadata from empty or tiny resources.\nconst CLEANUP_RETRY_DELAY_MS = 30_000;\nconst MAX_CLEANUP_RETRY_ATTEMPTS = 3;\ntype MaterializedResourceSession = {\n\tdirectory: string | undefined;\n\tbytes: number;\n\tfiles: number;\n\tsequence: number;\n};\n\nfunction createMaterializedResourceSession(): MaterializedResourceSession {\n\treturn {\n\t\tdirectory: undefined,\n\t\tbytes: 0,\n\t\tfiles: 0,\n\t\tsequence: 0,\n\t};\n}\n\nconst defaultMaterializedResourceSession = createMaterializedResourceSession();\nconst scopedMaterializedResourceSessions = new WeakMap<object, MaterializedResourceSession>();\nconst pendingCleanupDirectories = new Set<string>();\nconst cleanupRetryAttempts = new Map<string, number>();\nlet pendingCleanupRetry: ReturnType<typeof setTimeout> | undefined;\n\nfunction isAbortedScope(scope: object | undefined): boolean {\n\treturn !!scope && \"aborted\" in scope && (scope as { aborted?: unknown }).aborted === true;\n}\n\nfunction getMaterializedResourceSession(scope?: object): MaterializedResourceSession | undefined {\n\tif (isAbortedScope(scope)) return undefined;\n\tif (!scope) return defaultMaterializedResourceSession;\n\tlet session = scopedMaterializedResourceSessions.get(scope);\n\tif (!session) {\n\t\tsession = createMaterializedResourceSession();\n\t\tscopedMaterializedResourceSessions.set(scope, session);\n\t}\n\treturn session;\n}\n\ntype BinaryResource = {\n\turi?: string | undefined;\n\ttext?: string | undefined;\n\tmimeType?: string | undefined;\n\tblob: string;\n};\ntype McpResourceContent = {\n\turi: string;\n\ttext?: string | undefined;\n\tblob?: string | undefined;\n\tmimeType?: string | undefined;\n\t[key: string]: unknown;\n};\n\nfunction hasRetryableCleanupDirectory(): boolean {\n\tfor (const directory of pendingCleanupDirectories) {\n\t\tif ((cleanupRetryAttempts.get(directory) ?? 0) < MAX_CLEANUP_RETRY_ATTEMPTS) return true;\n\t}\n\treturn false;\n}\n\nfunction schedulePendingCleanupRetry(): void {\n\tif (pendingCleanupRetry || !hasRetryableCleanupDirectory()) return;\n\tfor (const directory of pendingCleanupDirectories) {\n\t\tconst attempts = cleanupRetryAttempts.get(directory) ?? 0;\n\t\tif (attempts < MAX_CLEANUP_RETRY_ATTEMPTS) cleanupRetryAttempts.set(directory, attempts + 1);\n\t}\n\tpendingCleanupRetry = setTimeout(() => {\n\t\tpendingCleanupRetry = undefined;\n\t\ttry {\n\t\t\tdrainPendingCleanupDirectories();\n\t\t} catch {\n\t\t\t// drainPendingCleanupDirectories already retained the paths and rescheduled another retry.\n\t\t}\n\t}, CLEANUP_RETRY_DELAY_MS);\n}\n\nfunction drainPendingCleanupDirectories(): void {\n\tconst failures: unknown[] = [];\n\tfor (const directory of Array.from(pendingCleanupDirectories)) {\n\t\ttry {\n\t\t\trmSync(directory, { recursive: true, force: true });\n\t\t\tpendingCleanupDirectories.delete(directory);\n\t\t\tcleanupRetryAttempts.delete(directory);\n\t\t} catch (error) {\n\t\t\tfailures.push(error);\n\t\t}\n\t}\n\tif (pendingCleanupDirectories.size === 0 && pendingCleanupRetry) {\n\t\tclearTimeout(pendingCleanupRetry);\n\t\tpendingCleanupRetry = undefined;\n\t}\n\tif (failures.length > 0) {\n\t\tschedulePendingCleanupRetry();\n\t\tthrow new AggregateError(failures, \"Failed to clean materialized MCP resources\");\n\t}\n}\n\nexport function cleanupMaterializedBinaryResources(scope?: object): void {\n\tconst session = scope ? scopedMaterializedResourceSessions.get(scope) : defaultMaterializedResourceSession;\n\tif (session?.directory) pendingCleanupDirectories.add(session.directory);\n\tif (session) {\n\t\tsession.directory = undefined;\n\t\tsession.bytes = 0;\n\t\tsession.files = 0;\n\t\tsession.sequence = 0;\n\t\tif (scope) scopedMaterializedResourceSessions.delete(scope);\n\t}\n\n\tdrainPendingCleanupDirectories();\n}\n\nfunction replaceBlob(resource: BinaryResource, text: string): string {\n\tdelete (resource as Partial<BinaryResource>).blob;\n\tresource.text = text;\n\treturn text;\n}\n\nfunction omitBinaryResource(resource: BinaryResource, reason: string): string {\n\treturn replaceBlob(\n\t\tresource,\n\t\t[\n\t\t\t`[Resource: ${resource.uri ?? \"(no URI)\"}]`,\n\t\t\t`Binary content omitted: ${reason}`,\n\t\t\t`MIME type: ${resource.mimeType ?? \"application/octet-stream\"}`,\n\t\t].join(\"\\n\"),\n\t);\n}\n\nfunction materializeBinaryResource(resource: BinaryResource, scope?: object): string {\n\tconst session = getMaterializedResourceSession(scope);\n\tif (!session) return omitBinaryResource(resource, \"runtime stopped\");\n\tconst decodedBytes = Buffer.byteLength(resource.blob, \"base64\");\n\tif (decodedBytes > MAX_BINARY_RESOURCE_BYTES) {\n\t\treturn omitBinaryResource(resource, \"decoded size exceeds 10 MiB\");\n\t}\n\tif (session.bytes + decodedBytes > MAX_SESSION_RESOURCE_BYTES || session.files >= MAX_SESSION_RESOURCE_FILES) {\n\t\treturn omitBinaryResource(resource, \"session resource limit reached\");\n\t}\n\n\ttry {\n\t\tsession.directory ??= mkdtempSync(join(tmpdir(), \"nexus-mcp-resource-\"));\n\t} catch {\n\t\treturn omitBinaryResource(resource, \"could not be saved\");\n\t}\n\n\tconst filePath = join(session.directory, `resource-${++session.sequence}.bin`);\n\tsession.bytes += decodedBytes;\n\tsession.files += 1;\n\ttry {\n\t\twriteFileSync(filePath, Buffer.from(resource.blob, \"base64\"), { flag: \"wx\", mode: 0o600 });\n\t} catch {\n\t\ttry {\n\t\t\trmSync(filePath, { force: true });\n\t\t\tsession.bytes -= decodedBytes;\n\t\t\tsession.files -= 1;\n\t\t} catch {\n\t\t\t// Keep the reservation when a partial file cannot be removed.\n\t\t}\n\t\treturn omitBinaryResource(resource, \"could not be saved\");\n\t}\n\n\treturn replaceBlob(\n\t\tresource,\n\t\t[\n\t\t\t`[Resource: ${resource.uri ?? \"(no URI)\"}]`,\n\t\t\t`Binary content saved to ${filePath}`,\n\t\t\t`MIME type: ${resource.mimeType ?? \"application/octet-stream\"}`,\n\t\t].join(\"\\n\"),\n\t);\n}\n\n/**\n * Transform MCP content types to Nexus content blocks.\n */\nexport function transformMcpResourceContents(contents: McpResourceContent[], scope?: object): ContentBlock[] {\n\treturn contents.map((resource) => {\n\t\tif (typeof resource.text === \"string\") return { type: \"text\" as const, text: resource.text };\n\t\tif (typeof resource.blob === \"string\")\n\t\t\treturn { type: \"text\" as const, text: materializeBinaryResource(resource as BinaryResource, scope) };\n\t\treturn { type: \"text\" as const, text: JSON.stringify(resource) };\n\t});\n}\n\nexport function transformMcpContent(content: McpContent[], scope?: object): ContentBlock[] {\n\treturn content.map((c) => {\n\t\tif (c.type === \"text\") {\n\t\t\treturn { type: \"text\" as const, text: c.text ?? \"\" };\n\t\t}\n\t\tif (c.type === \"image\") {\n\t\t\treturn {\n\t\t\t\ttype: \"image\" as const,\n\t\t\t\tdata: c.data ?? \"\",\n\t\t\t\tmimeType: c.mimeType ?? \"image/png\",\n\t\t\t};\n\t\t}\n\t\tif (c.type === \"resource\") {\n\t\t\tconst resourceUri = c.resource?.uri ?? \"(no URI)\";\n\t\t\tif (c.resource && \"blob\" in c.resource && typeof c.resource.blob === \"string\") {\n\t\t\t\tconst binaryResource = c.resource as typeof c.resource & { mimeType?: string; blob: string };\n\t\t\t\treturn {\n\t\t\t\t\ttype: \"text\" as const,\n\t\t\t\t\ttext: materializeBinaryResource(binaryResource, scope),\n\t\t\t\t};\n\t\t\t}\n\t\t\tconst resourceContent = c.resource?.text ?? (c.resource ? JSON.stringify(c.resource) : \"(no content)\");\n\t\t\treturn {\n\t\t\t\ttype: \"text\" as const,\n\t\t\t\ttext: `[Resource: ${resourceUri}]\\n${resourceContent}`,\n\t\t\t};\n\t\t}\n\t\tif (c.type === \"resource_link\") {\n\t\t\tconst linkName = c.name ?? c.uri ?? \"unknown\";\n\t\t\tconst linkUri = c.uri ?? \"(no URI)\";\n\t\t\treturn {\n\t\t\t\ttype: \"text\" as const,\n\t\t\t\ttext: `[Resource Link: ${linkName}]\\nURI: ${linkUri}`,\n\t\t\t};\n\t\t}\n\t\tif (c.type === \"audio\") {\n\t\t\treturn {\n\t\t\t\ttype: \"text\" as const,\n\t\t\t\ttext: `[Audio content: ${c.mimeType ?? \"audio/*\"}]`,\n\t\t\t};\n\t\t}\n\t\treturn { type: \"text\" as const, text: JSON.stringify(c) };\n\t});\n}\n\n/**\n * Resolve a tool result's content blocks, falling back to structuredContent\n * when content is empty.\n */\nexport function resolveMcpResultContent(result: Record<string, unknown>, scope?: object): ContentBlock[] {\n\tconst blocks = transformMcpContent((Array.isArray(result.content) ? result.content : []) as McpContent[], scope);\n\tif (blocks.length > 0) return blocks;\n\n\tif (result.structuredContent !== undefined && result.structuredContent !== null) {\n\t\treturn [{ type: \"text\" as const, text: stringifyStructuredContent(result.structuredContent) }];\n\t}\n\n\treturn [];\n}\n\nfunction stringifyStructuredContent(value: unknown): string {\n\ttry {\n\t\treturn JSON.stringify(value, null, 2) ?? String(value);\n\t} catch {\n\t\treturn String(value);\n\t}\n}\n"]}