@slopus/rig-codemode-codex 0.0.0-bootstrap.0

package/THIRD-PARTY-NOTICES.md

@@ -0,0 +1,9 @@
+# Third-party notices
+
+This package includes source code from OpenAI Codex at commit
+`37eef7baccebaeb00e42a88b323054cdbfe418c5`. That source is licensed under the
+Apache License 2.0; see `LICENSE-CODEX`.
+
+The native binary statically links the Rust dependencies recorded in
+`native/Cargo.lock`, including `rusty_v8`. Consumers can inspect the lockfile
+for the exact dependency versions used by this package.

package/UPSTREAM.md

@@ -0,0 +1,26 @@
+# Upstream source
+
+The native runtime in `native/code-mode`, `native/code-mode-protocol`, and
+`native/code-mode-host` is copied from OpenAI Codex:
+
+- Repository: https://github.com/openai/codex
+- Commit: `37eef7baccebaeb00e42a88b323054cdbfe418c5`
+- Commit date: `2026-07-21T22:16:27Z`
+- Retrieved: `2026-07-21`
+- Source directory: `codex-rs`
+- Upstream workspace version: `0.0.0`
+- Rust toolchain at that commit: `1.95.0`
+- V8 crate at that commit: `149.2.0`
+- License: Apache-2.0
+
+The copied code-mode crates are unmodified. `native/protocol` contains only the
+upstream `codex-rs/protocol/src/tool_name.rs` type required by those crates,
+plus a minimal crate manifest and re-export. The smaller protocol slice keeps
+this standalone build independent from the rest of Codex.
+
+The package's TypeScript process client, npm staging scripts, tests, and release
+workflow are original integration code.
+
+`release/rusty-v8-checksums` pins the archive and generated-binding hashes from
+the upstream `rusty-v8-v149.2.0` release. Release builds use these checked-in
+manifests rather than trusting checksums downloaded beside the payloads.

package/dist/CodeMode.d.ts

@@ -0,0 +1,13 @@
+import { CodeModeSession } from "./CodeModeSession.js";
+import type { CodeModeOptions, CodeModeSessionOptions } from "./types.js";
+export declare class CodeMode {
+    private readonly client;
+    private closed;
+    private readonly pendingSessionIds;
+    private readonly sessions;
+    private constructor();
+    static create(options?: CodeModeOptions): Promise<CodeMode>;
+    createSession(options?: CodeModeSessionOptions): Promise<CodeModeSession>;
+    close(): Promise<void>;
+}
+//# sourceMappingURL=CodeMode.d.ts.map

package/dist/CodeMode.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"CodeMode.d.ts","sourceRoot":"","sources":["../sources/CodeMode.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AAGvD,OAAO,KAAK,EAAE,eAAe,EAAE,sBAAsB,EAAE,MAAM,YAAY,CAAC;AAE1E,qBAAa,QAAQ;IACjB,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAa;IACpC,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,QAAQ,CAAC,iBAAiB,CAAqB;IACvD,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAsC;IAE/D,OAAO;WAkBM,MAAM,CAAC,OAAO,GAAE,eAAoB,GAAG,OAAO,CAAC,QAAQ,CAAC;IAM/D,aAAa,CAAC,OAAO,GAAE,sBAA2B,GAAG,OAAO,CAAC,eAAe,CAAC;IA4B7E,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;CAe/B"}

package/dist/CodeMode.js

@@ -0,0 +1,64 @@
+import { randomUUID } from "node:crypto";
+import { CodeModeSession } from "./CodeModeSession.js";
+import { HostClient } from "./HostClient.js";
+import { resolveCodeModeBinary } from "./resolveCodeModeBinary.js";
+export class CodeMode {
+    client;
+    closed = false;
+    pendingSessionIds = new Set();
+    sessions = new Map();
+    constructor(options) {
+        this.client = new HostClient(resolveCodeModeBinary(options.binaryPath), options.env, options.sandbox ?? "auto", async (message, signal) => {
+            const session = this.sessions.get(message.sessionId);
+            if (session === undefined) {
+                throw new Error(`Code Mode requested a delegate for unknown session ${message.sessionId}.`);
+            }
+            return session.handleDelegate(message, signal);
+        }, (sessionId, cellId) => this.sessions.get(sessionId)?.handleCellClosed(cellId));
+    }
+    static async create(options = {}) {
+        const codeMode = new CodeMode(options);
+        await codeMode.client.connect();
+        return codeMode;
+    }
+    async createSession(options = {}) {
+        if (this.closed) {
+            throw new Error("Code Mode host is closed.");
+        }
+        const sessionId = options.sessionId ?? `session-${randomUUID()}`;
+        if (this.sessions.has(sessionId) || this.pendingSessionIds.has(sessionId)) {
+            throw new Error(`Code Mode session ${sessionId} is already open.`);
+        }
+        this.pendingSessionIds.add(sessionId);
+        try {
+            const response = await this.client.openSession(sessionId);
+            if (response.type !== "session/ready") {
+                throw new Error(`Expected session/ready, received ${response.type}.`);
+            }
+            if (this.closed) {
+                await this.client.shutdownSession(sessionId).catch(() => undefined);
+                throw new Error("Code Mode host is closed.");
+            }
+            const session = new CodeModeSession(this.client, sessionId, options, () => this.sessions.delete(sessionId));
+            this.sessions.set(sessionId, session);
+            return session;
+        }
+        finally {
+            this.pendingSessionIds.delete(sessionId);
+        }
+    }
+    async close() {
+        if (this.closed) {
+            return;
+        }
+        this.closed = true;
+        const sessions = [...this.sessions.values()];
+        const results = await Promise.allSettled(sessions.map((session) => session.close()));
+        await this.client.close();
+        const failure = results.find((result) => result.status === "rejected");
+        if (failure !== undefined) {
+            throw failure.reason;
+        }
+    }
+}
+//# sourceMappingURL=CodeMode.js.map

package/dist/CodeMode.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"CodeMode.js","sourceRoot":"","sources":["../sources/CodeMode.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAEzC,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,OAAO,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AAGnE,MAAM,OAAO,QAAQ;IACA,MAAM,CAAa;IAC5B,MAAM,GAAG,KAAK,CAAC;IACN,iBAAiB,GAAG,IAAI,GAAG,EAAU,CAAC;IACtC,QAAQ,GAAG,IAAI,GAAG,EAA2B,CAAC;IAE/D,YAAoB,OAAwB;QACxC,IAAI,CAAC,MAAM,GAAG,IAAI,UAAU,CACxB,qBAAqB,CAAC,OAAO,CAAC,UAAU,CAAC,EACzC,OAAO,CAAC,GAAG,EACX,OAAO,CAAC,OAAO,IAAI,MAAM,EACzB,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE;YACtB,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;YACrD,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;gBACxB,MAAM,IAAI,KAAK,CACX,sDAAsD,OAAO,CAAC,SAAS,GAAG,CAC7E,CAAC;YACN,CAAC;YACD,OAAO,OAAO,CAAC,cAAc,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QACnD,CAAC,EACD,CAAC,SAAS,EAAE,MAAM,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,gBAAgB,CAAC,MAAM,CAAC,CAChF,CAAC;IACN,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,UAA2B,EAAE;QAC7C,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC,OAAO,CAAC,CAAC;QACvC,MAAM,QAAQ,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;QAChC,OAAO,QAAQ,CAAC;IACpB,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,UAAkC,EAAE;QACpD,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YACd,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;QACjD,CAAC;QACD,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,WAAW,UAAU,EAAE,EAAE,CAAC;QACjE,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;YACxE,MAAM,IAAI,KAAK,CAAC,qBAAqB,SAAS,mBAAmB,CAAC,CAAC;QACvE,CAAC;QACD,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACtC,IAAI,CAAC;YACD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC;YAC1D,IAAI,QAAQ,CAAC,IAAI,KAAK,eAAe,EAAE,CAAC;gBACpC,MAAM,IAAI,KAAK,CAAC,oCAAoC,QAAQ,CAAC,IAAI,GAAG,CAAC,CAAC;YAC1E,CAAC;YACD,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;gBACd,MAAM,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;gBACpE,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;YACjD,CAAC;YACD,MAAM,OAAO,GAAG,IAAI,eAAe,CAAC,IAAI,CAAC,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,GAAG,EAAE,CACtE,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,SAAS,CAAC,CAClC,CAAC;YACF,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;YACtC,OAAO,OAAO,CAAC;QACnB,CAAC;gBAAS,CAAC;YACP,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAC7C,CAAC;IACL,CAAC;IAED,KAAK,CAAC,KAAK;QACP,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YACd,OAAO;QACX,CAAC;QACD,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACnB,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;QAC7C,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QACrF,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;QAC1B,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,CACxB,CAAC,MAAM,EAAmC,EAAE,CAAC,MAAM,CAAC,MAAM,KAAK,UAAU,CAC5E,CAAC;QACF,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;YACxB,MAAM,OAAO,CAAC,MAAM,CAAC;QACzB,CAAC;IACL,CAAC;CACJ"}

package/dist/CodeModeHostCommand.d.ts

@@ -0,0 +1,15 @@
+import type { CodeModeSandboxMode } from "./types.js";
+export interface CodeModeHostCommand {
+    readonly args: readonly string[];
+    readonly command: string;
+    readonly cwd?: string;
+}
+export interface CreateCodeModeHostCommandOptions {
+    readonly binaryPath: string;
+    readonly env: NodeJS.ProcessEnv;
+    readonly platform?: NodeJS.Platform;
+    readonly resolveSandboxExecutable?: (platform: NodeJS.Platform, env: NodeJS.ProcessEnv) => string | undefined;
+    readonly sandbox: CodeModeSandboxMode;
+    readonly systemPathExists?: (path: string) => boolean;
+}
+//# sourceMappingURL=CodeModeHostCommand.d.ts.map

package/dist/CodeModeHostCommand.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"CodeModeHostCommand.d.ts","sourceRoot":"","sources":["../sources/CodeModeHostCommand.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,YAAY,CAAC;AAEtD,MAAM,WAAW,mBAAmB;IAChC,QAAQ,CAAC,IAAI,EAAE,SAAS,MAAM,EAAE,CAAC;IACjC,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC;IACzB,QAAQ,CAAC,GAAG,CAAC,EAAE,MAAM,CAAC;CACzB;AAED,MAAM,WAAW,gCAAgC;IAC7C,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC;IAC5B,QAAQ,CAAC,GAAG,EAAE,MAAM,CAAC,UAAU,CAAC;IAChC,QAAQ,CAAC,QAAQ,CAAC,EAAE,MAAM,CAAC,QAAQ,CAAC;IACpC,QAAQ,CAAC,wBAAwB,CAAC,EAAE,CAChC,QAAQ,EAAE,MAAM,CAAC,QAAQ,EACzB,GAAG,EAAE,MAAM,CAAC,UAAU,KACrB,MAAM,GAAG,SAAS,CAAC;IACxB,QAAQ,CAAC,OAAO,EAAE,mBAAmB,CAAC;IACtC,QAAQ,CAAC,gBAAgB,CAAC,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,OAAO,CAAC;CACzD"}

package/dist/CodeModeHostCommand.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=CodeModeHostCommand.js.map

package/dist/CodeModeHostCommand.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"CodeModeHostCommand.js","sourceRoot":"","sources":["../sources/CodeModeHostCommand.ts"],"names":[],"mappings":""}

package/dist/CodeModeSession.d.ts

@@ -0,0 +1,38 @@
+import type { HostClient } from "./HostClient.js";
+import type { HostMessage } from "./protocol.js";
+import type { CodeModeRunOptions, CodeModeRunResult, CodeModeSessionOptions, CodeModeResponse, JsonValue } from "./types.js";
+type DelegateRequestMessage = Extract<HostMessage, {
+    readonly type: "delegate/request";
+}>;
+export declare class CodeModeSession {
+    private readonly client;
+    readonly id: string;
+    private readonly options;
+    private readonly onClose;
+    private readonly cellTools;
+    private closed;
+    private readonly defaultTools;
+    constructor(client: HostClient, id: string, options: CodeModeSessionOptions, onClose: () => void);
+    execute(source: string, options?: CodeModeRunOptions): Promise<CodeModeResponse>;
+    wait(cellId: string, yieldTimeMs?: number, signal?: AbortSignal): Promise<CodeModeResponse>;
+    terminate(cellId: string): Promise<CodeModeResponse>;
+    run(source: string, options?: CodeModeRunOptions): Promise<CodeModeRunResult>;
+    close(): Promise<void>;
+    handleDelegate(message: DelegateRequestMessage, signal: AbortSignal): Promise<{
+        readonly type: "notification/delivered";
+    } | {
+        readonly type: "tool/result";
+        readonly result: JsonValue;
+    }>;
+    handleCellClosed(cellId: string): void;
+    private assertOpen;
+    private indexTools;
+    private mergeTools;
+    private throwIfAborted;
+    private toolKey;
+    private toWireTool;
+    private unwrapWait;
+    private validateTools;
+}
+export {};
+//# sourceMappingURL=CodeModeSession.d.ts.map

package/dist/CodeModeSession.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"CodeModeSession.d.ts","sourceRoot":"","sources":["../sources/CodeModeSession.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAClD,OAAO,KAAK,EAAE,WAAW,EAAuC,MAAM,eAAe,CAAC;AAEtF,OAAO,KAAK,EACR,kBAAkB,EAClB,iBAAiB,EAEjB,sBAAsB,EAGtB,gBAAgB,EAChB,SAAS,EACZ,MAAM,YAAY,CAAC;AAEpB,KAAK,sBAAsB,GAAG,OAAO,CAAC,WAAW,EAAE;IAAE,QAAQ,CAAC,IAAI,EAAE,kBAAkB,CAAA;CAAE,CAAC,CAAC;AAE1F,qBAAa,eAAe;IAMpB,OAAO,CAAC,QAAQ,CAAC,MAAM;IACvB,QAAQ,CAAC,EAAE,EAAE,MAAM;IACnB,OAAO,CAAC,QAAQ,CAAC,OAAO;IACxB,OAAO,CAAC,QAAQ,CAAC,OAAO;IAR5B,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAgD;IAC1E,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,QAAQ,CAAC,YAAY,CAA0B;gBAGlC,MAAM,EAAE,UAAU,EAC1B,EAAE,EAAE,MAAM,EACF,OAAO,EAAE,sBAAsB,EAC/B,OAAO,EAAE,MAAM,IAAI;IAMlC,OAAO,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,GAAE,kBAAuB,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAsCpF,IAAI,CACN,MAAM,EAAE,MAAM,EACd,WAAW,SAAS,EACpB,MAAM,CAAC,EAAE,WAAW,GACrB,OAAO,CAAC,gBAAgB,CAAC;IAQtB,SAAS,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAKpD,GAAG,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,GAAE,kBAAuB,GAAG,OAAO,CAAC,iBAAiB,CAAC;IA0CjF,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAgBtB,cAAc,CAChB,OAAO,EAAE,sBAAsB,EAC/B,MAAM,EAAE,WAAW,GACpB,OAAO,CACJ;QAAE,QAAQ,CAAC,IAAI,EAAE,wBAAwB,CAAA;KAAE,GAC3C;QAAE,QAAQ,CAAC,IAAI,EAAE,aAAa,CAAC;QAAC,QAAQ,CAAC,MAAM,EAAE,SAAS,CAAA;KAAE,CACjE;IAkCD,gBAAgB,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;IAKtC,OAAO,CAAC,UAAU;IAMlB,OAAO,CAAC,UAAU;IASlB,OAAO,CAAC,UAAU;IAWlB,OAAO,CAAC,cAAc;IAQtB,OAAO,CAAC,OAAO;IAOf,OAAO,CAAC,UAAU;IAYlB,OAAO,CAAC,UAAU;IAIlB,OAAO,CAAC,aAAa;CAgBxB"}

package/dist/CodeModeSession.js

@@ -0,0 +1,207 @@
+import { randomUUID } from "node:crypto";
+import { toCodeModeResponse } from "./toCodeModeResponse.js";
+export class CodeModeSession {
+    client;
+    id;
+    options;
+    onClose;
+    cellTools = new Map();
+    closed = false;
+    defaultTools;
+    constructor(client, id, options, onClose) {
+        this.client = client;
+        this.id = id;
+        this.options = options;
+        this.onClose = onClose;
+        this.defaultTools = options.tools ?? [];
+        this.validateTools(this.defaultTools);
+    }
+    async execute(source, options = {}) {
+        this.assertOpen();
+        this.throwIfAborted(options.signal);
+        const tools = this.mergeTools(options.tools);
+        this.validateTools(tools);
+        let cellId;
+        const abort = () => {
+            if (cellId !== undefined) {
+                void this.client.terminate(this.id, cellId).catch(() => undefined);
+            }
+        };
+        options.signal?.addEventListener("abort", abort, { once: true });
+        try {
+            const response = await this.client.execute(this.id, {
+                tool_call_id: options.toolCallId ?? randomUUID(),
+                enabled_tools: tools.map((tool) => this.toWireTool(tool)),
+                source,
+                yield_time_ms: options.yieldTimeMs ?? null,
+                max_output_tokens: null,
+            }, (startedCellId) => {
+                cellId = startedCellId;
+                this.cellTools.set(startedCellId, this.indexTools(tools));
+                if (options.signal?.aborted === true) {
+                    abort();
+                }
+            }, options.signal);
+            this.throwIfAborted(options.signal);
+            return toCodeModeResponse(response);
+        }
+        finally {
+            options.signal?.removeEventListener("abort", abort);
+        }
+    }
+    async wait(cellId, yieldTimeMs = 10_000, signal) {
+        this.assertOpen();
+        this.throwIfAborted(signal);
+        const outcome = await this.client.wait(this.id, cellId, yieldTimeMs, signal);
+        this.throwIfAborted(signal);
+        return this.unwrapWait(outcome);
+    }
+    async terminate(cellId) {
+        this.assertOpen();
+        return this.unwrapWait(await this.client.terminate(this.id, cellId));
+    }
+    async run(source, options = {}) {
+        const responses = [];
+        const contentItems = [];
+        let response = await this.execute(source, options);
+        for (;;) {
+            responses.push(response);
+            contentItems.push(...response.contentItems);
+            if (response.state !== "yielded") {
+                const text = contentItems
+                    .filter((item) => item.type === "input_text")
+                    .map((item) => item.text)
+                    .join("\n");
+                return response.errorText === undefined
+                    ? {
+                        state: response.state,
+                        cellId: response.cellId,
+                        contentItems,
+                        responses,
+                        text,
+                    }
+                    : {
+                        state: response.state,
+                        cellId: response.cellId,
+                        contentItems,
+                        errorText: response.errorText,
+                        responses,
+                        text,
+                    };
+            }
+            response = await this.wait(response.cellId, options.yieldTimeMs ?? 10_000, options.signal);
+        }
+    }
+    async close() {
+        if (this.closed) {
+            return;
+        }
+        this.closed = true;
+        try {
+            const response = await this.client.shutdownSession(this.id);
+            if (response.type !== "session/closed") {
+                throw new Error(`Expected session/closed, received ${response.type}.`);
+            }
+        }
+        finally {
+            this.cellTools.clear();
+            this.onClose();
+        }
+    }
+    async handleDelegate(message, signal) {
+        if (message.request.type === "notification/send") {
+            await this.options.onNotification?.({
+                callId: message.request.callId,
+                cellId: message.request.cellId,
+                text: message.request.text,
+            });
+            return { type: "notification/delivered" };
+        }
+        const invocation = message.request.invocation;
+        const tools = this.cellTools.get(invocation.cell_id);
+        const tool = tools?.get(this.toolKey(invocation.tool_name));
+        if (tool === undefined) {
+            throw new Error(`Code Mode requested an unknown tool: ${this.toolKey(invocation.tool_name)}`);
+        }
+        const context = {
+            cellId: invocation.cell_id,
+            runtimeToolCallId: invocation.runtime_tool_call_id,
+            signal,
+            toolKind: invocation.tool_kind,
+            toolName: invocation.tool_name.namespace === null
+                ? { name: invocation.tool_name.name }
+                : {
+                    name: invocation.tool_name.name,
+                    namespace: invocation.tool_name.namespace,
+                },
+        };
+        const result = await tool.execute(invocation.input ?? undefined, context);
+        return { type: "tool/result", result };
+    }
+    handleCellClosed(cellId) {
+        this.cellTools.delete(cellId);
+        this.options.onCellClosed?.(cellId);
+    }
+    assertOpen() {
+        if (this.closed) {
+            throw new Error(`Code Mode session ${this.id} is closed.`);
+        }
+    }
+    indexTools(tools) {
+        return new Map(tools.map((tool) => [
+            this.toolKey(tool.toolName ?? { name: tool.name, namespace: null }),
+            tool,
+        ]));
+    }
+    mergeTools(overrides) {
+        if (overrides === undefined) {
+            return this.defaultTools;
+        }
+        const merged = new Map(this.defaultTools.map((tool) => [tool.name, tool]));
+        for (const tool of overrides) {
+            merged.set(tool.name, tool);
+        }
+        return [...merged.values()];
+    }
+    throwIfAborted(signal) {
+        if (signal?.aborted === true) {
+            throw signal.reason instanceof Error
+                ? signal.reason
+                : new DOMException("Code Mode execution aborted.", "AbortError");
+        }
+    }
+    toolKey(toolName) {
+        return `${toolName.namespace ?? ""}\u0000${toolName.name}`;
+    }
+    toWireTool(tool) {
+        const toolName = tool.toolName ?? { name: tool.name };
+        return {
+            name: tool.name,
+            tool_name: { name: toolName.name, namespace: toolName.namespace ?? null },
+            description: tool.description ?? "",
+            kind: tool.kind ?? "function",
+            input_schema: tool.inputSchema ?? null,
+            output_schema: tool.outputSchema ?? null,
+        };
+    }
+    unwrapWait(outcome) {
+        return toCodeModeResponse("LiveCell" in outcome ? outcome.LiveCell : outcome.MissingCell);
+    }
+    validateTools(tools) {
+        const globalNames = new Set();
+        const toolNames = new Set();
+        for (const tool of tools) {
+            if (tool.name.trim() === "") {
+                throw new Error("Code Mode tool names must not be empty.");
+            }
+            if (!globalNames.add(tool.name)) {
+                throw new Error(`Duplicate Code Mode global tool name: ${tool.name}`);
+            }
+            const key = this.toolKey(tool.toolName ?? { name: tool.name });
+            if (!toolNames.add(key)) {
+                throw new Error(`Duplicate Code Mode tool identity: ${key.replace("\u0000", "")}`);
+            }
+        }
+    }
+}
+//# sourceMappingURL=CodeModeSession.js.map

package/dist/CodeModeSession.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"CodeModeSession.js","sourceRoot":"","sources":["../sources/CodeModeSession.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAIzC,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAc7D,MAAM,OAAO,eAAe;IAMH;IACR;IACQ;IACA;IARJ,SAAS,GAAG,IAAI,GAAG,EAAqC,CAAC;IAClE,MAAM,GAAG,KAAK,CAAC;IACN,YAAY,CAA0B;IAEvD,YACqB,MAAkB,EAC1B,EAAU,EACF,OAA+B,EAC/B,OAAmB;QAHnB,WAAM,GAAN,MAAM,CAAY;QAC1B,OAAE,GAAF,EAAE,CAAQ;QACF,YAAO,GAAP,OAAO,CAAwB;QAC/B,YAAO,GAAP,OAAO,CAAY;QAEpC,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,KAAK,IAAI,EAAE,CAAC;QACxC,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IAC1C,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAAc,EAAE,UAA8B,EAAE;QAC1D,IAAI,CAAC,UAAU,EAAE,CAAC;QAClB,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QACpC,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAC7C,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;QAC1B,IAAI,MAA0B,CAAC;QAC/B,MAAM,KAAK,GAAG,GAAG,EAAE;YACf,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;gBACvB,KAAK,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;YACvE,CAAC;QACL,CAAC,CAAC;QACF,OAAO,CAAC,MAAM,EAAE,gBAAgB,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QACjE,IAAI,CAAC;YACD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CACtC,IAAI,CAAC,EAAE,EACP;gBACI,YAAY,EAAE,OAAO,CAAC,UAAU,IAAI,UAAU,EAAE;gBAChD,aAAa,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;gBACzD,MAAM;gBACN,aAAa,EAAE,OAAO,CAAC,WAAW,IAAI,IAAI;gBAC1C,iBAAiB,EAAE,IAAI;aAC1B,EACD,CAAC,aAAa,EAAE,EAAE;gBACd,MAAM,GAAG,aAAa,CAAC;gBACvB,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,aAAa,EAAE,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC;gBAC1D,IAAI,OAAO,CAAC,MAAM,EAAE,OAAO,KAAK,IAAI,EAAE,CAAC;oBACnC,KAAK,EAAE,CAAC;gBACZ,CAAC;YACL,CAAC,EACD,OAAO,CAAC,MAAM,CACjB,CAAC;YACF,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YACpC,OAAO,kBAAkB,CAAC,QAAQ,CAAC,CAAC;QACxC,CAAC;gBAAS,CAAC;YACP,OAAO,CAAC,MAAM,EAAE,mBAAmB,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;QACxD,CAAC;IACL,CAAC;IAED,KAAK,CAAC,IAAI,CACN,MAAc,EACd,WAAW,GAAG,MAAM,EACpB,MAAoB;QAEpB,IAAI,CAAC,UAAU,EAAE,CAAC;QAClB,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;QAC5B,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,CAAC,CAAC;QAC7E,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;QAC5B,OAAO,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;IACpC,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,MAAc;QAC1B,IAAI,CAAC,UAAU,EAAE,CAAC;QAClB,OAAO,IAAI,CAAC,UAAU,CAAC,MAAM,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC;IACzE,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,MAAc,EAAE,UAA8B,EAAE;QACtD,MAAM,SAAS,GAAuB,EAAE,CAAC;QACzC,MAAM,YAAY,GAA0B,EAAE,CAAC;QAC/C,IAAI,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QACnD,SAAS,CAAC;YACN,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACzB,YAAY,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,YAAY,CAAC,CAAC;YAC5C,IAAI,QAAQ,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;gBAC/B,MAAM,IAAI,GAAG,YAAY;qBACpB,MAAM,CACH,CACI,IAAI,EACkE,EAAE,CACxE,IAAI,CAAC,IAAI,KAAK,YAAY,CACjC;qBACA,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;qBACxB,IAAI,CAAC,IAAI,CAAC,CAAC;gBAChB,OAAO,QAAQ,CAAC,SAAS,KAAK,SAAS;oBACnC,CAAC,CAAC;wBACI,KAAK,EAAE,QAAQ,CAAC,KAAK;wBACrB,MAAM,EAAE,QAAQ,CAAC,MAAM;wBACvB,YAAY;wBACZ,SAAS;wBACT,IAAI;qBACP;oBACH,CAAC,CAAC;wBACI,KAAK,EAAE,QAAQ,CAAC,KAAK;wBACrB,MAAM,EAAE,QAAQ,CAAC,MAAM;wBACvB,YAAY;wBACZ,SAAS,EAAE,QAAQ,CAAC,SAAS;wBAC7B,SAAS;wBACT,IAAI;qBACP,CAAC;YACZ,CAAC;YACD,QAAQ,GAAG,MAAM,IAAI,CAAC,IAAI,CACtB,QAAQ,CAAC,MAAM,EACf,OAAO,CAAC,WAAW,IAAI,MAAM,EAC7B,OAAO,CAAC,MAAM,CACjB,CAAC;QACN,CAAC;IACL,CAAC;IAED,KAAK,CAAC,KAAK;QACP,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YACd,OAAO;QACX,CAAC;QACD,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACnB,IAAI,CAAC;YACD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAC5D,IAAI,QAAQ,CAAC,IAAI,KAAK,gBAAgB,EAAE,CAAC;gBACrC,MAAM,IAAI,KAAK,CAAC,qCAAqC,QAAQ,CAAC,IAAI,GAAG,CAAC,CAAC;YAC3E,CAAC;QACL,CAAC;gBAAS,CAAC;YACP,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;YACvB,IAAI,CAAC,OAAO,EAAE,CAAC;QACnB,CAAC;IACL,CAAC;IAED,KAAK,CAAC,cAAc,CAChB,OAA+B,EAC/B,MAAmB;QAKnB,IAAI,OAAO,CAAC,OAAO,CAAC,IAAI,KAAK,mBAAmB,EAAE,CAAC;YAC/C,MAAM,IAAI,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC;gBAChC,MAAM,EAAE,OAAO,CAAC,OAAO,CAAC,MAAM;gBAC9B,MAAM,EAAE,OAAO,CAAC,OAAO,CAAC,MAAM;gBAC9B,IAAI,EAAE,OAAO,CAAC,OAAO,CAAC,IAAI;aAC7B,CAAC,CAAC;YACH,OAAO,EAAE,IAAI,EAAE,wBAAwB,EAAE,CAAC;QAC9C,CAAC;QACD,MAAM,UAAU,GAAG,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC;QAC9C,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;QACrD,MAAM,IAAI,GAAG,KAAK,EAAE,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC;QAC5D,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;YACrB,MAAM,IAAI,KAAK,CACX,wCAAwC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAC/E,CAAC;QACN,CAAC;QACD,MAAM,OAAO,GAAwB;YACjC,MAAM,EAAE,UAAU,CAAC,OAAO;YAC1B,iBAAiB,EAAE,UAAU,CAAC,oBAAoB;YAClD,MAAM;YACN,QAAQ,EAAE,UAAU,CAAC,SAAS;YAC9B,QAAQ,EACJ,UAAU,CAAC,SAAS,CAAC,SAAS,KAAK,IAAI;gBACnC,CAAC,CAAC,EAAE,IAAI,EAAE,UAAU,CAAC,SAAS,CAAC,IAAI,EAAE;gBACrC,CAAC,CAAC;oBACI,IAAI,EAAE,UAAU,CAAC,SAAS,CAAC,IAAI;oBAC/B,SAAS,EAAE,UAAU,CAAC,SAAS,CAAC,SAAS;iBAC5C;SACd,CAAC;QACF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,KAAK,IAAI,SAAS,EAAE,OAAO,CAAC,CAAC;QAC1E,OAAO,EAAE,IAAI,EAAE,aAAa,EAAE,MAAM,EAAE,CAAC;IAC3C,CAAC;IAED,gBAAgB,CAAC,MAAc;QAC3B,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAC9B,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,CAAC,MAAM,CAAC,CAAC;IACxC,CAAC;IAEO,UAAU;QACd,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YACd,MAAM,IAAI,KAAK,CAAC,qBAAqB,IAAI,CAAC,EAAE,aAAa,CAAC,CAAC;QAC/D,CAAC;IACL,CAAC;IAEO,UAAU,CAAC,KAA8B;QAC7C,OAAO,IAAI,GAAG,CACV,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC;YAChB,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC;YACnE,IAAI;SACP,CAAC,CACL,CAAC;IACN,CAAC;IAEO,UAAU,CAAC,SAA8C;QAC7D,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;YAC1B,OAAO,IAAI,CAAC,YAAY,CAAC;QAC7B,CAAC;QACD,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC;QAC3E,KAAK,MAAM,IAAI,IAAI,SAAS,EAAE,CAAC;YAC3B,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;QAChC,CAAC;QACD,OAAO,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;IAChC,CAAC;IAEO,cAAc,CAAC,MAA+B;QAClD,IAAI,MAAM,EAAE,OAAO,KAAK,IAAI,EAAE,CAAC;YAC3B,MAAM,MAAM,CAAC,MAAM,YAAY,KAAK;gBAChC,CAAC,CAAC,MAAM,CAAC,MAAM;gBACf,CAAC,CAAC,IAAI,YAAY,CAAC,8BAA8B,EAAE,YAAY,CAAC,CAAC;QACzE,CAAC;IACL,CAAC;IAEO,OAAO,CAAC,QAGf;QACG,OAAO,GAAG,QAAQ,CAAC,SAAS,IAAI,EAAE,SAAS,QAAQ,CAAC,IAAI,EAAE,CAAC;IAC/D,CAAC;IAEO,UAAU,CAAC,IAAkB;QACjC,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC;QACtD,OAAO;YACH,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,SAAS,EAAE,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,SAAS,EAAE,QAAQ,CAAC,SAAS,IAAI,IAAI,EAAE;YACzE,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,EAAE;YACnC,IAAI,EAAE,IAAI,CAAC,IAAI,IAAI,UAAU;YAC7B,YAAY,EAAE,IAAI,CAAC,WAAW,IAAI,IAAI;YACtC,aAAa,EAAE,IAAI,CAAC,YAAY,IAAI,IAAI;SAC3C,CAAC;IACN,CAAC;IAEO,UAAU,CAAC,OAAwB;QACvC,OAAO,kBAAkB,CAAC,UAAU,IAAI,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;IAC9F,CAAC;IAEO,aAAa,CAAC,KAA8B;QAChD,MAAM,WAAW,GAAG,IAAI,GAAG,EAAU,CAAC;QACtC,MAAM,SAAS,GAAG,IAAI,GAAG,EAAU,CAAC;QACpC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACvB,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;gBAC1B,MAAM,IAAI,KAAK,CAAC,yCAAyC,CAAC,CAAC;YAC/D,CAAC;YACD,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC9B,MAAM,IAAI,KAAK,CAAC,yCAAyC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;YAC1E,CAAC;YACD,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;YAC/D,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;gBACtB,MAAM,IAAI,KAAK,CAAC,sCAAsC,GAAG,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC;YACvF,CAAC;QACL,CAAC;IACL,CAAC;CACJ"}

package/dist/Deferred.d.ts

@@ -0,0 +1,7 @@
+export declare class Deferred<T> {
+    readonly promise: Promise<T>;
+    resolve: (value: T | PromiseLike<T>) => void;
+    reject: (reason?: unknown) => void;
+    constructor();
+}
+//# sourceMappingURL=Deferred.d.ts.map

package/dist/Deferred.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"Deferred.d.ts","sourceRoot":"","sources":["../sources/Deferred.ts"],"names":[],"mappings":"AAAA,qBAAa,QAAQ,CAAC,CAAC;IACnB,QAAQ,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC;IAC7B,OAAO,EAAG,CAAC,KAAK,EAAE,CAAC,GAAG,WAAW,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC;IAC9C,MAAM,EAAG,CAAC,MAAM,CAAC,EAAE,OAAO,KAAK,IAAI,CAAC;;CAQvC"}

package/dist/Deferred.js

@@ -0,0 +1,12 @@
+export class Deferred {
+    promise;
+    resolve;
+    reject;
+    constructor() {
+        this.promise = new Promise((resolve, reject) => {
+            this.resolve = resolve;
+            this.reject = reject;
+        });
+    }
+}
+//# sourceMappingURL=Deferred.js.map

package/dist/Deferred.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"Deferred.js","sourceRoot":"","sources":["../sources/Deferred.ts"],"names":[],"mappings":"AAAA,MAAM,OAAO,QAAQ;IACR,OAAO,CAAa;IAC7B,OAAO,CAAuC;IAC9C,MAAM,CAA8B;IAEpC;QACI,IAAI,CAAC,OAAO,GAAG,IAAI,OAAO,CAAI,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YAC9C,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;YACvB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACzB,CAAC,CAAC,CAAC;IACP,CAAC;CACJ"}

package/dist/FramedProcess.d.ts

@@ -0,0 +1,21 @@
+import type { ChildProcessWithoutNullStreams } from "node:child_process";
+import { type ClientMessage, type HostMessage } from "./protocol.js";
+export declare class FramedProcess {
+    private readonly child;
+    private readonly onMessage;
+    private readonly onFailure;
+    private buffer;
+    private closed;
+    private readonly exitPromise;
+    private failure;
+    private stderr;
+    private writeTail;
+    constructor(child: ChildProcessWithoutNullStreams, onMessage: (message: HostMessage) => void, onFailure: (error: Error) => void);
+    send(message: ClientMessage): Promise<void>;
+    close(): Promise<void>;
+    kill(): void;
+    private fail;
+    private read;
+    private readStderr;
+}
+//# sourceMappingURL=FramedProcess.d.ts.map

package/dist/FramedProcess.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"FramedProcess.d.ts","sourceRoot":"","sources":["../sources/FramedProcess.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,8BAA8B,EAAE,MAAM,oBAAoB,CAAC;AAGzE,OAAO,EAAmB,KAAK,aAAa,EAAE,KAAK,WAAW,EAAE,MAAM,eAAe,CAAC;AAItF,qBAAa,aAAa;IASlB,OAAO,CAAC,QAAQ,CAAC,KAAK;IACtB,OAAO,CAAC,QAAQ,CAAC,SAAS;IAC1B,OAAO,CAAC,QAAQ,CAAC,SAAS;IAV9B,OAAO,CAAC,MAAM,CAAmB;IACjC,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAgB;IAC5C,OAAO,CAAC,OAAO,CAAoB;IACnC,OAAO,CAAC,MAAM,CAAmB;IACjC,OAAO,CAAC,SAAS,CAAqB;gBAGjB,KAAK,EAAE,8BAA8B,EACrC,SAAS,EAAE,CAAC,OAAO,EAAE,WAAW,KAAK,IAAI,EACzC,SAAS,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI;IA0BtD,IAAI,CAAC,OAAO,EAAE,aAAa,GAAG,OAAO,CAAC,IAAI,CAAC;IA8BrC,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAoB5B,IAAI,IAAI,IAAI;IAIZ,OAAO,CAAC,IAAI;IASZ,OAAO,CAAC,IAAI;IAyBZ,OAAO,CAAC,UAAU;CAMrB"}

package/dist/FramedProcess.js

@@ -0,0 +1,125 @@
+import { assertJsonValue } from "./assertJsonValue.js";
+import { MAX_FRAME_BYTES } from "./protocol.js";
+const MAX_STDERR_BYTES = 64 * 1024;
+export class FramedProcess {
+    child;
+    onMessage;
+    onFailure;
+    buffer = Buffer.alloc(0);
+    closed = false;
+    exitPromise;
+    failure;
+    stderr = Buffer.alloc(0);
+    writeTail = Promise.resolve();
+    constructor(child, onMessage, onFailure) {
+        this.child = child;
+        this.onMessage = onMessage;
+        this.onFailure = onFailure;
+        this.child.stdout.on("data", (chunk) => this.read(chunk));
+        this.child.stderr.on("data", (chunk) => this.readStderr(chunk));
+        this.child.stdin.on("error", (error) => this.fail(error));
+        this.child.stdout.on("error", (error) => this.fail(error));
+        this.child.stderr.on("error", (error) => this.fail(error));
+        this.child.once("error", (error) => this.fail(error));
+        this.exitPromise = new Promise((resolve) => {
+            this.child.once("exit", (code, signal) => {
+                if (!this.closed) {
+                    const suffix = this.stderr.length === 0 ? "" : `\n${this.stderr.toString("utf8")}`;
+                    this.fail(new Error(signal === null
+                        ? `Code Mode host exited with code ${String(code)}.${suffix}`
+                        : `Code Mode host exited from signal ${signal}.${suffix}`));
+                }
+                resolve();
+            });
+        });
+    }
+    send(message) {
+        if (this.failure !== undefined) {
+            return Promise.reject(this.failure);
+        }
+        assertJsonValue(message);
+        const payload = Buffer.from(JSON.stringify(message), "utf8");
+        if (payload.length > MAX_FRAME_BYTES) {
+            return Promise.reject(new Error(`Code Mode frame exceeds ${String(MAX_FRAME_BYTES)} bytes.`));
+        }
+        const header = Buffer.allocUnsafe(4);
+        header.writeUInt32LE(payload.length);
+        const frame = Buffer.concat([header, payload]);
+        const sent = this.writeTail.then(() => new Promise((resolve, reject) => {
+            this.child.stdin.write(frame, (error) => {
+                if (error === null || error === undefined) {
+                    resolve();
+                }
+                else {
+                    reject(error);
+                }
+            });
+        }));
+        this.writeTail = sent.catch(() => undefined);
+        return sent;
+    }
+    async close() {
+        if (this.closed) {
+            return this.exitPromise;
+        }
+        this.closed = true;
+        const timeout = setTimeout(() => this.child.kill("SIGKILL"), 6_000);
+        timeout.unref();
+        try {
+            await this.writeTail;
+            this.child.stdin.end();
+            await this.exitPromise;
+        }
+        catch (error) {
+            this.child.kill("SIGKILL");
+            await this.exitPromise;
+            throw error;
+        }
+        finally {
+            clearTimeout(timeout);
+        }
+    }
+    kill() {
+        this.child.kill();
+    }
+    fail(error) {
+        if (this.failure !== undefined || this.closed) {
+            return;
+        }
+        this.failure = error;
+        this.onFailure(error);
+        this.child.kill("SIGKILL");
+    }
+    read(chunk) {
+        if (this.failure !== undefined) {
+            return;
+        }
+        this.buffer = Buffer.concat([this.buffer, chunk]);
+        while (this.buffer.length >= 4) {
+            const length = this.buffer.readUInt32LE(0);
+            if (length > MAX_FRAME_BYTES) {
+                this.fail(new Error(`Code Mode frame exceeds ${String(MAX_FRAME_BYTES)} bytes.`));
+                return;
+            }
+            if (this.buffer.length < length + 4) {
+                return;
+            }
+            const payload = this.buffer.subarray(4, length + 4);
+            this.buffer = this.buffer.subarray(length + 4);
+            try {
+                this.onMessage(JSON.parse(payload.toString("utf8")));
+            }
+            catch (error) {
+                this.fail(error instanceof Error ? error : new Error(String(error)));
+                return;
+            }
+        }
+    }
+    readStderr(chunk) {
+        this.stderr = Buffer.concat([this.stderr, chunk]);
+        if (this.stderr.length > MAX_STDERR_BYTES) {
+            this.stderr = this.stderr.subarray(this.stderr.length - MAX_STDERR_BYTES);
+        }
+    }
+}
+//# sourceMappingURL=FramedProcess.js.map

package/dist/FramedProcess.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"FramedProcess.js","sourceRoot":"","sources":["../sources/FramedProcess.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EAAE,eAAe,EAAwC,MAAM,eAAe,CAAC;AAEtF,MAAM,gBAAgB,GAAG,EAAE,GAAG,IAAI,CAAC;AAEnC,MAAM,OAAO,aAAa;IASD;IACA;IACA;IAVb,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IACzB,MAAM,GAAG,KAAK,CAAC;IACN,WAAW,CAAgB;IACpC,OAAO,CAAoB;IAC3B,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IACzB,SAAS,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC;IAEtC,YACqB,KAAqC,EACrC,SAAyC,EACzC,SAAiC;QAFjC,UAAK,GAAL,KAAK,CAAgC;QACrC,cAAS,GAAT,SAAS,CAAgC;QACzC,cAAS,GAAT,SAAS,CAAwB;QAElD,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAa,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QAClE,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAa,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC;QACxE,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QAC1D,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QAC3D,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QAC3D,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QACtD,IAAI,CAAC,WAAW,GAAG,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YACvC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE;gBACrC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;oBACf,MAAM,MAAM,GACR,IAAI,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;oBACxE,IAAI,CAAC,IAAI,CACL,IAAI,KAAK,CACL,MAAM,KAAK,IAAI;wBACX,CAAC,CAAC,mCAAmC,MAAM,CAAC,IAAI,CAAC,IAAI,MAAM,EAAE;wBAC7D,CAAC,CAAC,qCAAqC,MAAM,IAAI,MAAM,EAAE,CAChE,CACJ,CAAC;gBACN,CAAC;gBACD,OAAO,EAAE,CAAC;YACd,CAAC,CAAC,CAAC;QACP,CAAC,CAAC,CAAC;IACP,CAAC;IAED,IAAI,CAAC,OAAsB;QACvB,IAAI,IAAI,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;YAC7B,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACxC,CAAC;QACD,eAAe,CAAC,OAAO,CAAC,CAAC;QACzB,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,EAAE,MAAM,CAAC,CAAC;QAC7D,IAAI,OAAO,CAAC,MAAM,GAAG,eAAe,EAAE,CAAC;YACnC,OAAO,OAAO,CAAC,MAAM,CACjB,IAAI,KAAK,CAAC,2BAA2B,MAAM,CAAC,eAAe,CAAC,SAAS,CAAC,CACzE,CAAC;QACN,CAAC;QACD,MAAM,MAAM,GAAG,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;QACrC,MAAM,CAAC,aAAa,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QACrC,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC;QAC/C,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAC5B,GAAG,EAAE,CACD,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YAClC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;gBACpC,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;oBACxC,OAAO,EAAE,CAAC;gBACd,CAAC;qBAAM,CAAC;oBACJ,MAAM,CAAC,KAAK,CAAC,CAAC;gBAClB,CAAC;YACL,CAAC,CAAC,CAAC;QACP,CAAC,CAAC,CACT,CAAC;QACF,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QAC7C,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,KAAK;QACP,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YACd,OAAO,IAAI,CAAC,WAAW,CAAC;QAC5B,CAAC;QACD,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACnB,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,CAAC,CAAC;QACpE,OAAO,CAAC,KAAK,EAAE,CAAC;QAChB,IAAI,CAAC;YACD,MAAM,IAAI,CAAC,SAAS,CAAC;YACrB,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC;YACvB,MAAM,IAAI,CAAC,WAAW,CAAC;QAC3B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC3B,MAAM,IAAI,CAAC,WAAW,CAAC;YACvB,MAAM,KAAK,CAAC;QAChB,CAAC;gBAAS,CAAC;YACP,YAAY,CAAC,OAAO,CAAC,CAAC;QAC1B,CAAC;IACL,CAAC;IAED,IAAI;QACA,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;IACtB,CAAC;IAEO,IAAI,CAAC,KAAY;QACrB,IAAI,IAAI,CAAC,OAAO,KAAK,SAAS,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAC5C,OAAO;QACX,CAAC;QACD,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC;QACrB,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QACtB,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAC/B,CAAC;IAEO,IAAI,CAAC,KAAa;QACtB,IAAI,IAAI,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;YAC7B,OAAO;QACX,CAAC;QACD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC,CAAC;QAClD,OAAO,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;YAC7B,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC3C,IAAI,MAAM,GAAG,eAAe,EAAE,CAAC;gBAC3B,IAAI,CAAC,IAAI,CAAC,IAAI,KAAK,CAAC,2BAA2B,MAAM,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC,CAAC;gBAClF,OAAO;YACX,CAAC;YACD,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,MAAM,GAAG,CAAC,EAAE,CAAC;gBAClC,OAAO;YACX,CAAC;YACD,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,EAAE,MAAM,GAAG,CAAC,CAAC,CAAC;YACpD,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAC/C,IAAI,CAAC;gBACD,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAgB,CAAC,CAAC;YACxE,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACb,IAAI,CAAC,IAAI,CAAC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;gBACrE,OAAO;YACX,CAAC;QACL,CAAC;IACL,CAAC;IAEO,UAAU,CAAC,KAAa;QAC5B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC,CAAC;QAClD,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,gBAAgB,EAAE,CAAC;YACxC,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,gBAAgB,CAAC,CAAC;QAC9E,CAAC;IACL,CAAC;CACJ"}