@herbertgao/pi-extensions 2026.9.10 → 2026.9.12

package/node_modules/pi-mcp-adapter/dist/request-headers-command.js

@@ -0,0 +1,312 @@
+import { spawn, spawnSync } from "node:child_process";
+import { randomUUID } from "node:crypto";
+import { interpolateEnvVars } from "./utils.js";
+const DEFAULT_TIMEOUT_MS = 10_000;
+const MAX_OUTPUT_BYTES = 64 * 1024;
+const USE_PROCESS_GROUP = process.platform !== "win32";
+const CLEANUP_TOKEN_ENV = "PI_MCP_REQUEST_HEADERS_CLEANUP_TOKEN";
+// Busy hosts can exceed spawnSync's 1 MiB default when `ps axeww` dumps each
+// process environment. Keep discovery below a bounded cap instead of letting
+// `ps` get SIGTERM'd and reporting a false cleanup failure.
+const PS_MAX_BUFFER_BYTES = 8 * 1024 * 1024;
+function isNoSuchProcessError(error) {
+    return typeof error === "object" && error !== null && "code" in error && error.code === "ESRCH";
+}
+function runPosixPs(args) {
+    if (process.env.PI_MCP_ADAPTER_TEST_FAIL_PS === "1")
+        return { status: 1, signal: null, stdout: "" };
+    const result = spawnSync("ps", args, { encoding: "utf8", maxBuffer: PS_MAX_BUFFER_BYTES });
+    return { status: result.status, signal: result.signal, stdout: result.stdout };
+}
+function psFailureReason(result) {
+    return result.status === null
+        ? `ps was killed by signal ${result.signal ?? "unknown"}`
+        : `ps exited with code ${result.status}`;
+}
+function collectPosixProcessPids(rootPid, cleanupToken) {
+    const result = runPosixPs(["axeww", "-o", "pid=,ppid=,command="]);
+    if (result.status !== 0) {
+        throw new Error(`HTTP request headers command cleanup failed: ${psFailureReason(result)}`);
+    }
+    const childrenByParent = new Map();
+    const processPids = new Set();
+    const needle = cleanupToken ? `${CLEANUP_TOKEN_ENV}=${cleanupToken}` : undefined;
+    for (const line of result.stdout.split("\n")) {
+        const match = /^\s*(\d+)\s+(\d+)(?:\s+(.*))?$/.exec(line);
+        if (!match)
+            continue;
+        const pid = Number(match[1]);
+        const ppid = Number(match[2]);
+        if (!Number.isInteger(pid) || !Number.isInteger(ppid))
+            continue;
+        const children = childrenByParent.get(ppid);
+        if (children)
+            children.push(pid);
+        else
+            childrenByParent.set(ppid, [pid]);
+        if (needle && pid !== process.pid && match[3]?.includes(needle))
+            processPids.add(pid);
+    }
+    const stack = [...(childrenByParent.get(rootPid) ?? [])];
+    while (stack.length > 0) {
+        const pid = stack.pop();
+        processPids.add(pid);
+        stack.push(...(childrenByParent.get(pid) ?? []));
+    }
+    return [...processPids];
+}
+function assertPosixProcessDiscoveryAvailable() {
+    const result = runPosixPs(["axeww", "-o", "pid=,ppid=,command="]);
+    if (result.status !== 0) {
+        throw new Error(`HTTP request headers command cleanup failed: ${psFailureReason(result)}`);
+    }
+}
+function isTaskkillNoSuchProcess(result) {
+    return `${result.stdout ?? ""}\n${result.stderr ?? ""}`.toLowerCase().includes("not found");
+}
+function signalPid(pid, signal) {
+    try {
+        process.kill(pid, signal);
+    }
+    catch (error) {
+        if (!isNoSuchProcessError(error))
+            throw error;
+    }
+}
+function signalProcessGroup(pid, signal) {
+    try {
+        process.kill(-pid, signal);
+    }
+    catch (error) {
+        if (!isNoSuchProcessError(error))
+            throw error;
+    }
+}
+function killRequestHeadersCommand(child, trackedPosixDescendantPids = new Set(), cleanupToken) {
+    if (process.platform === "win32" && child.pid !== undefined) {
+        const result = spawnSync("taskkill", ["/pid", String(child.pid), "/T", "/F"], {
+            encoding: "utf8",
+            windowsHide: true,
+        });
+        if (result.status === 0 || result.status === 128 || isTaskkillNoSuchProcess(result))
+            return;
+        throw new Error(`HTTP request headers command cleanup failed: taskkill exited with code ${result.status ?? "unknown"}`);
+    }
+    if (USE_PROCESS_GROUP && child.pid !== undefined) {
+        const frozenPids = new Set();
+        let cleanupError;
+        try {
+            signalProcessGroup(child.pid, "SIGSTOP");
+            for (const pid of trackedPosixDescendantPids) {
+                signalPid(pid, "SIGSTOP");
+                frozenPids.add(pid);
+            }
+            let stablePasses = 0;
+            for (let pass = 0; pass < 16; pass++) {
+                const candidates = collectPosixProcessPids(child.pid, cleanupToken);
+                const newPids = candidates.filter(pid => !frozenPids.has(pid));
+                if (newPids.length === 0) {
+                    stablePasses++;
+                    if (stablePasses >= 2)
+                        return;
+                    continue;
+                }
+                stablePasses = 0;
+                for (const pid of newPids) {
+                    signalPid(pid, "SIGSTOP");
+                    frozenPids.add(pid);
+                }
+            }
+            cleanupError = new Error("HTTP request headers command cleanup failed: descendant process tree did not stabilize");
+        }
+        catch (error) {
+            cleanupError = error instanceof Error ? error : new Error(String(error));
+        }
+        finally {
+            signalProcessGroup(child.pid, "SIGKILL");
+            for (const pid of frozenPids)
+                signalPid(pid, "SIGKILL");
+        }
+        if (cleanupError)
+            throw cleanupError;
+        return;
+    }
+    child.kill("SIGKILL");
+}
+function resolvedCommand(config) {
+    if (!config || typeof config !== "object" || Array.isArray(config)) {
+        throw new Error("HTTP request headers command must be an object");
+    }
+    if (typeof config.command !== "string" || config.command.trim() === "") {
+        throw new Error("HTTP request headers command requires a non-empty command");
+    }
+    if (config.args !== undefined && (!Array.isArray(config.args) || config.args.some(arg => typeof arg !== "string"))) {
+        throw new Error("HTTP request headers command args must be strings");
+    }
+    if (config.env !== undefined && (typeof config.env !== "object"
+        || Array.isArray(config.env)
+        || Object.values(config.env).some(value => typeof value !== "string"))) {
+        throw new Error("HTTP request headers command env values must be strings");
+    }
+    const timeoutMs = config.timeoutMs ?? DEFAULT_TIMEOUT_MS;
+    if (!Number.isInteger(timeoutMs) || timeoutMs <= 0 || timeoutMs > 60_000) {
+        throw new Error("HTTP request headers command timeoutMs must be an integer between 1 and 60000");
+    }
+    return {
+        command: interpolateEnvVars(config.command),
+        args: (config.args ?? []).map(interpolateEnvVars),
+        env: {
+            ...process.env,
+            ...Object.fromEntries(Object.entries(config.env ?? {}).map(([key, value]) => [key, interpolateEnvVars(value)])),
+        },
+        timeoutMs,
+    };
+}
+async function invokeRequestHeadersCommand(config, envelope, signal) {
+    const resolved = resolvedCommand(config);
+    if (USE_PROCESS_GROUP)
+        assertPosixProcessDiscoveryAvailable();
+    return new Promise((resolve, reject) => {
+        let stdout = Buffer.alloc(0);
+        let settled = false;
+        const cleanupToken = randomUUID();
+        const child = spawn(resolved.command, resolved.args, {
+            env: { ...resolved.env, [CLEANUP_TOKEN_ENV]: cleanupToken },
+            stdio: ["pipe", "pipe", "ignore"],
+            windowsHide: true,
+            detached: USE_PROCESS_GROUP,
+        });
+        const trackedPosixDescendantPids = new Set();
+        let trackingError;
+        const trackPosixDescendants = () => {
+            if (!USE_PROCESS_GROUP || child.pid === undefined || settled || trackingError)
+                return;
+            try {
+                for (const pid of collectPosixProcessPids(child.pid, cleanupToken))
+                    trackedPosixDescendantPids.add(pid);
+            }
+            catch (error) {
+                trackingError = error instanceof Error ? error : new Error(String(error));
+            }
+        };
+        const descendantTracker = USE_PROCESS_GROUP ? setInterval(trackPosixDescendants, 50) : undefined;
+        descendantTracker?.unref();
+        const finish = (result) => {
+            if (settled)
+                return;
+            settled = true;
+            clearTimeout(timer);
+            if (descendantTracker)
+                clearInterval(descendantTracker);
+            signal.removeEventListener("abort", abort);
+            if (result.status === "error")
+                reject(result.error);
+            else
+                resolve(result.headers);
+        };
+        const finishAfterKill = (result) => {
+            try {
+                killRequestHeadersCommand(child, trackedPosixDescendantPids, cleanupToken);
+                finish(result);
+            }
+            catch (cleanupError) {
+                finish({ status: "error", error: cleanupError instanceof Error ? cleanupError : new Error(String(cleanupError)) });
+            }
+        };
+        const failAfterKill = (message) => {
+            finishAfterKill({ status: "error", error: trackingError ?? new Error(message) });
+        };
+        const abort = () => {
+            failAfterKill("HTTP request headers command aborted");
+        };
+        const timer = setTimeout(() => {
+            failAfterKill(`HTTP request headers command timed out after ${resolved.timeoutMs}ms`);
+        }, resolved.timeoutMs);
+        signal.addEventListener("abort", abort, { once: true });
+        if (signal.aborted) {
+            abort();
+            return;
+        }
+        child.on("error", () => finish({ status: "error", error: new Error("HTTP request headers command failed to start") }));
+        child.stdout.on("data", (chunk) => {
+            if (settled)
+                return;
+            stdout = Buffer.concat([stdout, Buffer.from(chunk)]);
+            if (stdout.byteLength > MAX_OUTPUT_BYTES) {
+                failAfterKill("HTTP request headers command output exceeded 64 KiB");
+            }
+        });
+        child.on("close", code => {
+            if (settled)
+                return;
+            if (trackingError) {
+                failAfterKill(trackingError.message);
+                return;
+            }
+            if (code !== 0) {
+                failAfterKill(`HTTP request headers command exited with code ${code ?? "unknown"}`);
+                return;
+            }
+            let parsed;
+            try {
+                parsed = JSON.parse(stdout.toString("utf8"));
+            }
+            catch {
+                finishAfterKill({ status: "error", error: new Error("HTTP request headers command returned invalid JSON") });
+                return;
+            }
+            if (!parsed || typeof parsed !== "object" || Array.isArray(parsed)) {
+                finishAfterKill({ status: "error", error: new Error("HTTP request headers command must return a JSON object") });
+                return;
+            }
+            const entries = Object.entries(parsed);
+            if (entries.some(([, value]) => typeof value !== "string")) {
+                finishAfterKill({ status: "error", error: new Error("HTTP request headers command values must be strings") });
+                return;
+            }
+            try {
+                finishAfterKill({ status: "success", headers: new Headers(entries) });
+            }
+            catch {
+                finishAfterKill({ status: "error", error: new Error("HTTP request headers command returned an invalid header") });
+            }
+        });
+        child.stdin.on("error", () => { });
+        child.stdin.end(JSON.stringify(envelope));
+    });
+}
+/** Wrap fetch so a trusted command can derive headers from the exact request. */
+export function createRequestHeadersCommandFetch(config, delegate = globalThis.fetch) {
+    // Validate static configuration before the first request.
+    resolvedCommand(config);
+    return async (input, init) => {
+        const request = new Request(input, init);
+        const signal = init?.signal === null
+            ? request.signal
+            : init?.signal ?? (input instanceof Request ? input.signal : request.signal);
+        const body = Buffer.from(await request.clone().arrayBuffer());
+        const derived = await invokeRequestHeadersCommand(config, {
+            version: 1,
+            method: request.method.toUpperCase(),
+            url: request.url,
+            bodyBase64: body.toString("base64"),
+        }, signal);
+        const headers = new Headers(request.headers);
+        derived.forEach((value, name) => headers.set(name, value));
+        return delegate(new URL(request.url), {
+            method: request.method,
+            headers,
+            ...(request.method === "GET" || request.method === "HEAD" ? {} : { body }),
+            signal,
+            cache: request.cache,
+            credentials: request.credentials,
+            integrity: request.integrity,
+            keepalive: request.keepalive,
+            mode: request.mode,
+            redirect: request.redirect,
+            referrer: request.referrer,
+            referrerPolicy: request.referrerPolicy,
+        });
+    };
+}
+//# sourceMappingURL=request-headers-command.js.map

package/node_modules/pi-mcp-adapter/dist/request-headers-command.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"request-headers-command.js","sourceRoot":"","sources":["../request-headers-command.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,SAAS,EAAqB,MAAM,oBAAoB,CAAC;AACzE,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAGzC,OAAO,EAAE,kBAAkB,EAAE,MAAM,YAAY,CAAC;AAEhD,MAAM,kBAAkB,GAAG,MAAM,CAAC;AAClC,MAAM,gBAAgB,GAAG,EAAE,GAAG,IAAI,CAAC;AACnC,MAAM,iBAAiB,GAAG,OAAO,CAAC,QAAQ,KAAK,OAAO,CAAC;AACvD,MAAM,iBAAiB,GAAG,sCAAsC,CAAC;AACjE,6EAA6E;AAC7E,6EAA6E;AAC7E,4DAA4D;AAC5D,MAAM,mBAAmB,GAAG,CAAC,GAAG,IAAI,GAAG,IAAI,CAAC;AAE5C,SAAS,oBAAoB,CAAC,KAAc;IAC1C,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,MAAM,IAAI,KAAK,IAAK,KAA+B,CAAC,IAAI,KAAK,OAAO,CAAC;AAC7H,CAAC;AAED,SAAS,UAAU,CAAC,IAAc;IAChC,IAAI,OAAO,CAAC,GAAG,CAAC,2BAA2B,KAAK,GAAG;QAAE,OAAO,EAAE,MAAM,EAAE,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC;IACpG,MAAM,MAAM,GAAG,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,SAAS,EAAE,mBAAmB,EAAE,CAAC,CAAC;IAC3F,OAAO,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC;AACjF,CAAC;AAED,SAAS,eAAe,CAAC,MAAgE;IACvF,OAAO,MAAM,CAAC,MAAM,KAAK,IAAI;QAC3B,CAAC,CAAC,2BAA2B,MAAM,CAAC,MAAM,IAAI,SAAS,EAAE;QACzD,CAAC,CAAC,uBAAuB,MAAM,CAAC,MAAM,EAAE,CAAC;AAC7C,CAAC;AAED,SAAS,uBAAuB,CAAC,OAAe,EAAE,YAAqB;IACrE,MAAM,MAAM,GAAG,UAAU,CAAC,CAAC,OAAO,EAAE,IAAI,EAAE,qBAAqB,CAAC,CAAC,CAAC;IAClE,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACxB,MAAM,IAAI,KAAK,CAAC,gDAAgD,eAAe,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IAC7F,CAAC;IAED,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAAoB,CAAC;IACrD,MAAM,WAAW,GAAG,IAAI,GAAG,EAAU,CAAC;IACtC,MAAM,MAAM,GAAG,YAAY,CAAC,CAAC,CAAC,GAAG,iBAAiB,IAAI,YAAY,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;IACjF,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;QAC7C,MAAM,KAAK,GAAG,gCAAgC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC1D,IAAI,CAAC,KAAK;YAAE,SAAS;QACrB,MAAM,GAAG,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;QAC7B,MAAM,IAAI,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;QAC9B,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC;YAAE,SAAS;QAChE,MAAM,QAAQ,GAAG,gBAAgB,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAC5C,IAAI,QAAQ;YAAE,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;;YAC5B,gBAAgB,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;QACvC,IAAI,MAAM,IAAI,GAAG,KAAK,OAAO,CAAC,GAAG,IAAI,KAAK,CAAC,CAAC,CAAC,EAAE,QAAQ,CAAC,MAAM,CAAC;YAAE,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;IACxF,CAAC;IAED,MAAM,KAAK,GAAG,CAAC,GAAG,CAAC,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;IACzD,OAAO,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACxB,MAAM,GAAG,GAAG,KAAK,CAAC,GAAG,EAAG,CAAC;QACzB,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACrB,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,gBAAgB,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;IACnD,CAAC;IACD,OAAO,CAAC,GAAG,WAAW,CAAC,CAAC;AAC1B,CAAC;AAED,SAAS,oCAAoC;IAC3C,MAAM,MAAM,GAAG,UAAU,CAAC,CAAC,OAAO,EAAE,IAAI,EAAE,qBAAqB,CAAC,CAAC,CAAC;IAClE,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACxB,MAAM,IAAI,KAAK,CAAC,gDAAgD,eAAe,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IAC7F,CAAC;AACH,CAAC;AAED,SAAS,uBAAuB,CAAC,MAAoC;IACnE,OAAO,GAAG,MAAM,CAAC,MAAM,IAAI,EAAE,KAAK,MAAM,CAAC,MAAM,IAAI,EAAE,EAAE,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;AAC9F,CAAC;AAED,SAAS,SAAS,CAAC,GAAW,EAAE,MAAsB;IACpD,IAAI,CAAC;QACH,OAAO,CAAC,IAAI,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;IAC5B,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC;YAAE,MAAM,KAAK,CAAC;IAChD,CAAC;AACH,CAAC;AAED,SAAS,kBAAkB,CAAC,GAAW,EAAE,MAAsB;IAC7D,IAAI,CAAC;QACH,OAAO,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;IAC7B,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC;YAAE,MAAM,KAAK,CAAC;IAChD,CAAC;AACH,CAAC;AAED,SAAS,yBAAyB,CAAC,KAAmB,EAAE,6BAA6B,IAAI,GAAG,EAAU,EAAE,YAAqB;IAC3H,IAAI,OAAO,CAAC,QAAQ,KAAK,OAAO,IAAI,KAAK,CAAC,GAAG,KAAK,SAAS,EAAE,CAAC;QAC5D,MAAM,MAAM,GAAG,SAAS,CAAC,UAAU,EAAE,CAAC,MAAM,EAAE,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,EAAE;YAC5E,QAAQ,EAAE,MAAM;YAChB,WAAW,EAAE,IAAI;SAClB,CAAC,CAAC;QACH,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,IAAI,MAAM,CAAC,MAAM,KAAK,GAAG,IAAI,uBAAuB,CAAC,MAAM,CAAC;YAAE,OAAO;QAC5F,MAAM,IAAI,KAAK,CAAC,0EAA0E,MAAM,CAAC,MAAM,IAAI,SAAS,EAAE,CAAC,CAAC;IAC1H,CAAC;IAED,IAAI,iBAAiB,IAAI,KAAK,CAAC,GAAG,KAAK,SAAS,EAAE,CAAC;QACjD,MAAM,UAAU,GAAG,IAAI,GAAG,EAAU,CAAC;QACrC,IAAI,YAA+B,CAAC;QACpC,IAAI,CAAC;YACH,kBAAkB,CAAC,KAAK,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;YACzC,KAAK,MAAM,GAAG,IAAI,0BAA0B,EAAE,CAAC;gBAC7C,SAAS,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;gBAC1B,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YACtB,CAAC;YACD,IAAI,YAAY,GAAG,CAAC,CAAC;YACrB,KAAK,IAAI,IAAI,GAAG,CAAC,EAAE,IAAI,GAAG,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC;gBACrC,MAAM,UAAU,GAAG,uBAAuB,CAAC,KAAK,CAAC,GAAG,EAAE,YAAY,CAAC,CAAC;gBACpE,MAAM,OAAO,GAAG,UAAU,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC;gBAC/D,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;oBACzB,YAAY,EAAE,CAAC;oBACf,IAAI,YAAY,IAAI,CAAC;wBAAE,OAAO;oBAC9B,SAAS;gBACX,CAAC;gBACD,YAAY,GAAG,CAAC,CAAC;gBACjB,KAAK,MAAM,GAAG,IAAI,OAAO,EAAE,CAAC;oBAC1B,SAAS,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;oBAC1B,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;gBACtB,CAAC;YACH,CAAC;YACD,YAAY,GAAG,IAAI,KAAK,CAAC,wFAAwF,CAAC,CAAC;QACrH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;QAC3E,CAAC;gBAAS,CAAC;YACT,kBAAkB,CAAC,KAAK,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;YACzC,KAAK,MAAM,GAAG,IAAI,UAAU;gBAAE,SAAS,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;QAC1D,CAAC;QACD,IAAI,YAAY;YAAE,MAAM,YAAY,CAAC;QACrC,OAAO;IACT,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;AACxB,CAAC;AAaD,SAAS,eAAe,CAAC,MAAiC;IAMxD,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;QACnE,MAAM,IAAI,KAAK,CAAC,gDAAgD,CAAC,CAAC;IACpE,CAAC;IACD,IAAI,OAAO,MAAM,CAAC,OAAO,KAAK,QAAQ,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;QACvE,MAAM,IAAI,KAAK,CAAC,2DAA2D,CAAC,CAAC;IAC/E,CAAC;IACD,IAAI,MAAM,CAAC,IAAI,KAAK,SAAS,IAAI,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,OAAO,GAAG,KAAK,QAAQ,CAAC,CAAC,EAAE,CAAC;QACnH,MAAM,IAAI,KAAK,CAAC,mDAAmD,CAAC,CAAC;IACvE,CAAC;IACD,IAAI,MAAM,CAAC,GAAG,KAAK,SAAS,IAAI,CAC9B,OAAO,MAAM,CAAC,GAAG,KAAK,QAAQ;WAC3B,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC;WACzB,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,OAAO,KAAK,KAAK,QAAQ,CAAC,CACtE,EAAE,CAAC;QACF,MAAM,IAAI,KAAK,CAAC,yDAAyD,CAAC,CAAC;IAC7E,CAAC;IACD,MAAM,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,kBAAkB,CAAC;IACzD,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,IAAI,SAAS,IAAI,CAAC,IAAI,SAAS,GAAG,MAAM,EAAE,CAAC;QACzE,MAAM,IAAI,KAAK,CAAC,+EAA+E,CAAC,CAAC;IACnG,CAAC;IACD,OAAO;QACL,OAAO,EAAE,kBAAkB,CAAC,MAAM,CAAC,OAAO,CAAC;QAC3C,IAAI,EAAE,CAAC,MAAM,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,kBAAkB,CAAC;QACjD,GAAG,EAAE;YACH,GAAG,OAAO,CAAC,GAAG;YACd,GAAG,MAAM,CAAC,WAAW,CACnB,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,EAAE,kBAAkB,CAAC,KAAK,CAAC,CAAC,CAAC,CACzF;SACF;QACD,SAAS;KACV,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,2BAA2B,CACxC,MAAiC,EACjC,QAAoC,EACpC,MAAmB;IAEnB,MAAM,QAAQ,GAAG,eAAe,CAAC,MAAM,CAAC,CAAC;IACzC,IAAI,iBAAiB;QAAE,oCAAoC,EAAE,CAAC;IAC9D,OAAO,IAAI,OAAO,CAAU,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QAC9C,IAAI,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC7B,IAAI,OAAO,GAAG,KAAK,CAAC;QACpB,MAAM,YAAY,GAAG,UAAU,EAAE,CAAC;QAClC,MAAM,KAAK,GAAG,KAAK,CAAC,QAAQ,CAAC,OAAO,EAAE,QAAQ,CAAC,IAAI,EAAE;YACnD,GAAG,EAAE,EAAE,GAAG,QAAQ,CAAC,GAAG,EAAE,CAAC,iBAAiB,CAAC,EAAE,YAAY,EAAE;YAC3D,KAAK,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC;YACjC,WAAW,EAAE,IAAI;YACjB,QAAQ,EAAE,iBAAiB;SAC5B,CAAC,CAAC;QAEH,MAAM,0BAA0B,GAAG,IAAI,GAAG,EAAU,CAAC;QACrD,IAAI,aAAgC,CAAC;QACrC,MAAM,qBAAqB,GAAG,GAAG,EAAE;YACjC,IAAI,CAAC,iBAAiB,IAAI,KAAK,CAAC,GAAG,KAAK,SAAS,IAAI,OAAO,IAAI,aAAa;gBAAE,OAAO;YACtF,IAAI,CAAC;gBACH,KAAK,MAAM,GAAG,IAAI,uBAAuB,CAAC,KAAK,CAAC,GAAG,EAAE,YAAY,CAAC;oBAAE,0BAA0B,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAC1G,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,aAAa,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;YAC5E,CAAC;QACH,CAAC,CAAC;QACF,MAAM,iBAAiB,GAAG,iBAAiB,CAAC,CAAC,CAAC,WAAW,CAAC,qBAAqB,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QACjG,iBAAiB,EAAE,KAAK,EAAE,CAAC;QAE3B,MAAM,MAAM,GAAG,CAAC,MAAqB,EAAE,EAAE;YACvC,IAAI,OAAO;gBAAE,OAAO;YACpB,OAAO,GAAG,IAAI,CAAC;YACf,YAAY,CAAC,KAAK,CAAC,CAAC;YACpB,IAAI,iBAAiB;gBAAE,aAAa,CAAC,iBAAiB,CAAC,CAAC;YACxD,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;YAC3C,IAAI,MAAM,CAAC,MAAM,KAAK,OAAO;gBAAE,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;;gBAC/C,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QAC/B,CAAC,CAAC;QACF,MAAM,eAAe,GAAG,CAAC,MAAqB,EAAE,EAAE;YAChD,IAAI,CAAC;gBACH,yBAAyB,CAAC,KAAK,EAAE,0BAA0B,EAAE,YAAY,CAAC,CAAC;gBAC3E,MAAM,CAAC,MAAM,CAAC,CAAC;YACjB,CAAC;YAAC,OAAO,YAAY,EAAE,CAAC;gBACtB,MAAM,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,YAAY,KAAK,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,EAAE,CAAC,CAAC;YACrH,CAAC;QACH,CAAC,CAAC;QACF,MAAM,aAAa,GAAG,CAAC,OAAe,EAAE,EAAE;YACxC,eAAe,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,aAAa,IAAI,IAAI,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QACnF,CAAC,CAAC;QACF,MAAM,KAAK,GAAG,GAAG,EAAE;YACjB,aAAa,CAAC,sCAAsC,CAAC,CAAC;QACxD,CAAC,CAAC;QACF,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE;YAC5B,aAAa,CAAC,gDAAgD,QAAQ,CAAC,SAAS,IAAI,CAAC,CAAC;QACxF,CAAC,EAAE,QAAQ,CAAC,SAAS,CAAC,CAAC;QAEvB,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QACxD,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;YACnB,KAAK,EAAE,CAAC;YACR,OAAO;QACT,CAAC;QAED,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,KAAK,CAAC,8CAA8C,CAAC,EAAE,CAAC,CAAC,CAAC;QACvH,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAsB,EAAE,EAAE;YACjD,IAAI,OAAO;gBAAE,OAAO;YACpB,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;YACrD,IAAI,MAAM,CAAC,UAAU,GAAG,gBAAgB,EAAE,CAAC;gBACzC,aAAa,CAAC,qDAAqD,CAAC,CAAC;YACvE,CAAC;QACH,CAAC,CAAC,CAAC;QACH,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,IAAI,CAAC,EAAE;YACvB,IAAI,OAAO;gBAAE,OAAO;YACpB,IAAI,aAAa,EAAE,CAAC;gBAClB,aAAa,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;gBACrC,OAAO;YACT,CAAC;YACD,IAAI,IAAI,KAAK,CAAC,EAAE,CAAC;gBACf,aAAa,CAAC,iDAAiD,IAAI,IAAI,SAAS,EAAE,CAAC,CAAC;gBACpF,OAAO;YACT,CAAC;YACD,IAAI,MAAe,CAAC;YACpB,IAAI,CAAC;gBACH,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC;YAC/C,CAAC;YAAC,MAAM,CAAC;gBACP,eAAe,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,KAAK,CAAC,oDAAoD,CAAC,EAAE,CAAC,CAAC;gBAC7G,OAAO;YACT,CAAC;YACD,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;gBACnE,eAAe,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,KAAK,CAAC,wDAAwD,CAAC,EAAE,CAAC,CAAC;gBACjH,OAAO;YACT,CAAC;YACD,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YACvC,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,OAAO,KAAK,KAAK,QAAQ,CAAC,EAAE,CAAC;gBAC3D,eAAe,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,KAAK,CAAC,qDAAqD,CAAC,EAAE,CAAC,CAAC;gBAC9G,OAAO;YACT,CAAC;YACD,IAAI,CAAC;gBACH,eAAe,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,IAAI,OAAO,CAAC,OAAkC,CAAC,EAAE,CAAC,CAAC;YACnG,CAAC;YAAC,MAAM,CAAC;gBACP,eAAe,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,KAAK,CAAC,yDAAyD,CAAC,EAAE,CAAC,CAAC;YACpH,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QAClC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;AACL,CAAC;AAED,iFAAiF;AACjF,MAAM,UAAU,gCAAgC,CAC9C,MAAiC,EACjC,WAAsB,UAAU,CAAC,KAAK;IAEtC,0DAA0D;IAC1D,eAAe,CAAC,MAAM,CAAC,CAAC;IACxB,OAAO,KAAK,EAAE,KAAwB,EAAE,IAAkB,EAAqB,EAAE;QAC/E,MAAM,OAAO,GAAG,IAAI,OAAO,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;QACzC,MAAM,MAAM,GAAG,IAAI,EAAE,MAAM,KAAK,IAAI;YAClC,CAAC,CAAC,OAAO,CAAC,MAAM;YAChB,CAAC,CAAC,IAAI,EAAE,MAAM,IAAI,CAAC,KAAK,YAAY,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAC/E,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,OAAO,CAAC,KAAK,EAAE,CAAC,WAAW,EAAE,CAAC,CAAC;QAC9D,MAAM,OAAO,GAAG,MAAM,2BAA2B,CAAC,MAAM,EAAE;YACxD,OAAO,EAAE,CAAC;YACV,MAAM,EAAE,OAAO,CAAC,MAAM,CAAC,WAAW,EAAE;YACpC,GAAG,EAAE,OAAO,CAAC,GAAG;YAChB,UAAU,EAAE,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC;SACpC,EAAE,MAAM,CAAC,CAAC;QACX,MAAM,OAAO,GAAG,IAAI,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAC7C,OAAO,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC;QAC3D,OAAO,QAAQ,CAAC,IAAI,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE;YACpC,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,OAAO;YACP,GAAG,CAAC,OAAO,CAAC,MAAM,KAAK,KAAK,IAAI,OAAO,CAAC,MAAM,KAAK,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC;YAC1E,MAAM;YACN,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,WAAW,EAAE,OAAO,CAAC,WAAW;YAChC,SAAS,EAAE,OAAO,CAAC,SAAS;YAC5B,SAAS,EAAE,OAAO,CAAC,SAAS;YAC5B,IAAI,EAAE,OAAO,CAAC,IAAI;YAClB,QAAQ,EAAE,OAAO,CAAC,QAAQ;YAC1B,QAAQ,EAAE,OAAO,CAAC,QAAQ;YAC1B,cAAc,EAAE,OAAO,CAAC,cAAc;SACvC,CAAC,CAAC;IACL,CAAC,CAAC;AACJ,CAAC"}

package/node_modules/pi-mcp-adapter/dist/runtime-owner.d.ts

@@ -0,0 +1,13 @@
+import type { ExtensionUIContext } from "@earendil-works/pi-coding-agent";
+export interface McpRuntimeOwner {
+    readonly signal: AbortSignal;
+    isActive(): boolean;
+    addCleanup(cleanup: () => void | Promise<void>): void;
+    stop(reason?: string): Promise<void>;
+    throwIfInactive(): void;
+}
+export declare function createMcpRuntimeOwner(): McpRuntimeOwner;
+export declare function combineAbortSignals(...signals: Array<AbortSignal | undefined>): AbortSignal | undefined;
+/** Fence session-bound UI calls after the owning extension runtime stops. */
+export declare function createOwnedUi(ui: ExtensionUIContext, owner: McpRuntimeOwner): ExtensionUIContext;
+export declare function isAbortError(error: unknown, signal?: AbortSignal): boolean;

package/node_modules/pi-mcp-adapter/dist/runtime-owner.js

@@ -0,0 +1,97 @@
+import { formatTerminalError } from "./utils.js";
+export function createMcpRuntimeOwner() {
+    const controller = new AbortController();
+    const cleanups = [];
+    let stopPromise;
+    const reportCleanupFailure = (error, late) => {
+        console.error(`MCP: ${late ? "late " : ""}runtime cleanup failed: ${formatTerminalError(error)}`);
+    };
+    return {
+        signal: controller.signal,
+        isActive: () => !controller.signal.aborted,
+        addCleanup: cleanup => {
+            if (controller.signal.aborted) {
+                void Promise.resolve().then(cleanup).catch(error => reportCleanupFailure(error, true));
+                return;
+            }
+            cleanups.push(cleanup);
+        },
+        stop: (reason = "MCP extension runtime stopped") => {
+            if (stopPromise)
+                return stopPromise;
+            controller.abort(new Error(reason));
+            const pendingCleanups = cleanups.splice(0).reverse().map(cleanup => Promise.resolve().then(cleanup));
+            stopPromise = Promise.allSettled(pendingCleanups).then(results => {
+                const failures = results.flatMap(result => result.status === "rejected" ? [result.reason] : []);
+                if (failures.length > 0) {
+                    const aggregate = new AggregateError(failures, "MCP runtime cleanup failed");
+                    console.error(`MCP: runtime cleanup failed: ${formatTerminalError(aggregate)}`);
+                    throw aggregate;
+                }
+            });
+            return stopPromise;
+        },
+        throwIfInactive: () => controller.signal.throwIfAborted(),
+    };
+}
+export function combineAbortSignals(...signals) {
+    const active = signals.filter((signal) => signal !== undefined);
+    if (active.length === 0)
+        return undefined;
+    if (active.length === 1)
+        return active[0];
+    if (typeof AbortSignal.any === "function")
+        return AbortSignal.any(active);
+    // AbortSignal.any was added after Node 20.0.0, our minimum supported version.
+    const controller = new AbortController();
+    const alreadyAborted = active.find(signal => signal.aborted);
+    if (alreadyAborted) {
+        controller.abort(alreadyAborted.reason);
+        return controller.signal;
+    }
+    const onAbort = (event) => {
+        for (const signal of active)
+            signal.removeEventListener("abort", onAbort);
+        controller.abort(event.target.reason);
+    };
+    for (const signal of active)
+        signal.addEventListener("abort", onAbort, { once: true });
+    return controller.signal;
+}
+/** Fence session-bound UI calls after the owning extension runtime stops. */
+export function createOwnedUi(ui, owner) {
+    const proxies = new WeakMap();
+    const wrap = (value) => {
+        if ((typeof value !== "object" || value === null) && typeof value !== "function") {
+            return value;
+        }
+        const object = value;
+        const existing = proxies.get(object);
+        if (existing)
+            return existing;
+        const proxy = new Proxy(object, {
+            get(target, property, receiver) {
+                if (!owner.isActive())
+                    return undefined;
+                const member = Reflect.get(target, property, receiver);
+                if (typeof member === "function") {
+                    return (...args) => {
+                        if (!owner.isActive())
+                            return undefined;
+                        return Reflect.apply(member, target, args);
+                    };
+                }
+                return owner.isActive() ? wrap(member) : undefined;
+            },
+        });
+        proxies.set(object, proxy);
+        return proxy;
+    };
+    return wrap(ui);
+}
+export function isAbortError(error, signal) {
+    if (signal?.aborted)
+        return true;
+    return error instanceof Error && (error.name === "AbortError" || error.message === "MCP extension runtime stopped");
+}
+//# sourceMappingURL=runtime-owner.js.map

package/node_modules/pi-mcp-adapter/dist/runtime-owner.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"runtime-owner.js","sourceRoot":"","sources":["../runtime-owner.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,mBAAmB,EAAE,MAAM,YAAY,CAAC;AAUjD,MAAM,UAAU,qBAAqB;IACnC,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;IACzC,MAAM,QAAQ,GAAsC,EAAE,CAAC;IACvD,IAAI,WAAsC,CAAC;IAE3C,MAAM,oBAAoB,GAAG,CAAC,KAAc,EAAE,IAAa,EAAE,EAAE;QAC7D,OAAO,CAAC,KAAK,CAAC,QAAQ,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,2BAA2B,mBAAmB,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IACpG,CAAC,CAAC;IAEF,OAAO;QACL,MAAM,EAAE,UAAU,CAAC,MAAM;QACzB,QAAQ,EAAE,GAAG,EAAE,CAAC,CAAC,UAAU,CAAC,MAAM,CAAC,OAAO;QAC1C,UAAU,EAAE,OAAO,CAAC,EAAE;YACpB,IAAI,UAAU,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBAC9B,KAAK,OAAO,CAAC,OAAO,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC,oBAAoB,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,CAAC;gBACvF,OAAO;YACT,CAAC;YACD,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACzB,CAAC;QACD,IAAI,EAAE,CAAC,MAAM,GAAG,+BAA+B,EAAE,EAAE;YACjD,IAAI,WAAW;gBAAE,OAAO,WAAW,CAAC;YACpC,UAAU,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;YACpC,MAAM,eAAe,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CACjE,OAAO,CAAC,OAAO,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAChC,CAAC;YACF,WAAW,GAAG,OAAO,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE;gBAC/D,MAAM,QAAQ,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM,KAAK,UAAU,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBAChG,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACxB,MAAM,SAAS,GAAG,IAAI,cAAc,CAAC,QAAQ,EAAE,4BAA4B,CAAC,CAAC;oBAC7E,OAAO,CAAC,KAAK,CAAC,gCAAgC,mBAAmB,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;oBAChF,MAAM,SAAS,CAAC;gBAClB,CAAC;YACH,CAAC,CAAC,CAAC;YACH,OAAO,WAAW,CAAC;QACrB,CAAC;QACD,eAAe,EAAE,GAAG,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,cAAc,EAAE;KAC1D,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,GAAG,OAAuC;IAC5E,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,MAAM,EAAyB,EAAE,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC;IACvF,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,SAAS,CAAC;IAC1C,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,MAAM,CAAC,CAAC,CAAC,CAAC;IAC1C,IAAI,OAAO,WAAW,CAAC,GAAG,KAAK,UAAU;QAAE,OAAO,WAAW,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAE1E,8EAA8E;IAC9E,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;IACzC,MAAM,cAAc,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAC7D,IAAI,cAAc,EAAE,CAAC;QACnB,UAAU,CAAC,KAAK,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;QACxC,OAAO,UAAU,CAAC,MAAM,CAAC;IAC3B,CAAC;IAED,MAAM,OAAO,GAAG,CAAC,KAAY,EAAE,EAAE;QAC/B,KAAK,MAAM,MAAM,IAAI,MAAM;YAAE,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QAC1E,UAAU,CAAC,KAAK,CAAE,KAAK,CAAC,MAAsB,CAAC,MAAM,CAAC,CAAC;IACzD,CAAC,CAAC;IACF,KAAK,MAAM,MAAM,IAAI,MAAM;QAAE,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;IACvF,OAAO,UAAU,CAAC,MAAM,CAAC;AAC3B,CAAC;AAED,6EAA6E;AAC7E,MAAM,UAAU,aAAa,CAAC,EAAsB,EAAE,KAAsB;IAC1E,MAAM,OAAO,GAAG,IAAI,OAAO,EAAkB,CAAC;IAC9C,MAAM,IAAI,GAAG,CAAC,KAAc,EAAW,EAAE;QACvC,IAAI,CAAC,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,CAAC,IAAI,OAAO,KAAK,KAAK,UAAU,EAAE,CAAC;YACjF,OAAO,KAAK,CAAC;QACf,CAAC;QACD,MAAM,MAAM,GAAG,KAAe,CAAC;QAC/B,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACrC,IAAI,QAAQ;YAAE,OAAO,QAAQ,CAAC;QAE9B,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,MAAM,EAAE;YAC9B,GAAG,CAAC,MAAM,EAAE,QAAQ,EAAE,QAAQ;gBAC5B,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE;oBAAE,OAAO,SAAS,CAAC;gBACxC,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC;gBACvD,IAAI,OAAO,MAAM,KAAK,UAAU,EAAE,CAAC;oBACjC,OAAO,CAAC,GAAG,IAAe,EAAE,EAAE;wBAC5B,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE;4BAAE,OAAO,SAAS,CAAC;wBACxC,OAAO,OAAO,CAAC,KAAK,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC;oBAC7C,CAAC,CAAC;gBACJ,CAAC;gBACD,OAAO,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;YACrD,CAAC;SACF,CAAC,CAAC;QACH,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;QAC3B,OAAO,KAAK,CAAC;IACf,CAAC,CAAC;IACF,OAAO,IAAI,CAAC,EAAE,CAAuB,CAAC;AACxC,CAAC;AAED,MAAM,UAAU,YAAY,CAAC,KAAc,EAAE,MAAoB;IAC/D,IAAI,MAAM,EAAE,OAAO;QAAE,OAAO,IAAI,CAAC;IACjC,OAAO,KAAK,YAAY,KAAK,IAAI,CAAC,KAAK,CAAC,IAAI,KAAK,YAAY,IAAI,KAAK,CAAC,OAAO,KAAK,+BAA+B,CAAC,CAAC;AACtH,CAAC"}

package/node_modules/pi-mcp-adapter/dist/sampling-handler.d.ts

@@ -0,0 +1,17 @@
+import type { Api, Model } from "@earendil-works/pi-ai";
+import type { ExtensionUIContext, ModelRegistry } from "@earendil-works/pi-coding-agent";
+import type { Client } from "@modelcontextprotocol/client";
+import { type CreateMessageRequest, type CreateMessageResult } from "@modelcontextprotocol/client";
+export type SamplingUIContext = Pick<ExtensionUIContext, "confirm">;
+export type SamplingModelRegistry = Pick<ModelRegistry, "getAvailable" | "complete">;
+export interface SamplingHandlerOptions {
+    serverName: string;
+    autoApprove: boolean;
+    ui?: SamplingUIContext;
+    modelRegistry: SamplingModelRegistry;
+    getCurrentModel: () => Model<Api> | undefined;
+    getSignal: () => AbortSignal | undefined;
+}
+export type ServerSamplingConfig = Omit<SamplingHandlerOptions, "serverName">;
+export declare function registerSamplingHandler(client: Client, options: SamplingHandlerOptions): void;
+export declare function handleSamplingRequest(options: SamplingHandlerOptions, request: CreateMessageRequest): Promise<CreateMessageResult>;