@markusylisiurunen/tau 0.2.129 → 0.3.1

package/dist/transport/stdio_session_transport.d.ts

@@ -0,0 +1,38 @@
+import type { ChildProcessWithoutNullStreams } from "node:child_process";
+import { type SessionProtocolInitializeParams, type SessionProtocolMethod, type SessionProtocolParamsByMethod, type SessionProtocolReadyMessage, type SessionProtocolResultByMethod } from "../protocol/session_protocol.js";
+import type { SessionProtocolDeltaListener, SessionProtocolEphemeralListener, SessionProtocolTransport } from "./session_transport.js";
+export type SessionProtocolSpawnedProcess = ChildProcessWithoutNullStreams;
+export declare class StdioSessionProtocolTransport implements SessionProtocolTransport {
+    private readonly process;
+    private readonly deltaListeners;
+    private readonly ephemeralListeners;
+    private readonly pendingRequests;
+    private readonly readyDeferred;
+    private readonly exitDeferred;
+    private readyValue?;
+    private stdoutBuffer;
+    private stderrBuffer;
+    private connectPromise?;
+    private closePromise?;
+    private isConnected;
+    private isClosed;
+    private isExited;
+    private fatalError?;
+    constructor(process: SessionProtocolSpawnedProcess);
+    get ready(): SessionProtocolReadyMessage;
+    connect(initializeParams: SessionProtocolInitializeParams, timeoutMs: number): Promise<void>;
+    private establishConnection;
+    onDelta(listener: SessionProtocolDeltaListener): () => void;
+    onEphemeral(listener: SessionProtocolEphemeralListener): () => void;
+    request<M extends SessionProtocolMethod>(method: M, params: SessionProtocolParamsByMethod[M]): Promise<SessionProtocolResultByMethod[M]>;
+    close(): Promise<void>;
+    private readonly handleStdoutData;
+    private readonly handleStderrData;
+    private readonly handleProcessError;
+    private readonly handleProcessExit;
+    private readonly handleProcessClose;
+    private onProcessTerminated;
+    private handleSessionProtocolLine;
+    private failTransport;
+    private rejectReadyIfPending;
+}

package/dist/transport/stdio_session_transport.js

@@ -0,0 +1,243 @@
+import { parseSessionProtocolOutgoingLine, } from "../protocol/session_protocol.js";
+import { TauProcessError, TauTransportError } from "./errors.js";
+import { PendingSessionProtocolRequests, } from "./pending_session_protocol_requests.js";
+import { createDeferred, handleSessionProtocolTransportParseFailure, handleSessionProtocolTransportResponse, notifySessionProtocolDeltaListeners, notifySessionProtocolEphemeralListeners, waitForPromiseOrTimeout, withTimeout, } from "./session_protocol_transport_helpers.js";
+const CLOSE_TIMEOUT_MS = 2_000;
+export class StdioSessionProtocolTransport {
+    process;
+    deltaListeners = new Set();
+    ephemeralListeners = new Set();
+    pendingRequests = new PendingSessionProtocolRequests();
+    readyDeferred = createDeferred();
+    exitDeferred = createDeferred();
+    readyValue;
+    stdoutBuffer = "";
+    stderrBuffer = "";
+    connectPromise;
+    closePromise;
+    isConnected = false;
+    isClosed = false;
+    isExited = false;
+    fatalError;
+    constructor(process) {
+        this.process = process;
+        this.process.stdout.setEncoding("utf8");
+        this.process.stderr.setEncoding("utf8");
+        this.process.stdout.on("data", this.handleStdoutData);
+        this.process.stderr.on("data", this.handleStderrData);
+        this.process.on("error", this.handleProcessError);
+        this.process.on("exit", this.handleProcessExit);
+        this.process.on("close", this.handleProcessClose);
+    }
+    get ready() {
+        if (!this.readyValue) {
+            throw new TauTransportError("tau sdk client is not connected");
+        }
+        return this.readyValue;
+    }
+    async connect(initializeParams, timeoutMs) {
+        if (this.isClosed) {
+            throw new TauTransportError("tau sdk client is closed");
+        }
+        if (this.fatalError) {
+            throw this.fatalError;
+        }
+        if (this.isConnected) {
+            return;
+        }
+        this.connectPromise ??= this.establishConnection(initializeParams, timeoutMs);
+        await this.connectPromise;
+    }
+    async establishConnection(initializeParams, timeoutMs) {
+        const ready = await withTimeout(this.readyDeferred.promise, timeoutMs, new TauTransportError(`timed out waiting for session protocol ready message after ${timeoutMs}ms`));
+        this.readyValue = ready;
+        await this.request("initialize", initializeParams);
+        this.isConnected = true;
+    }
+    onDelta(listener) {
+        this.deltaListeners.add(listener);
+        return () => {
+            this.deltaListeners.delete(listener);
+        };
+    }
+    onEphemeral(listener) {
+        this.ephemeralListeners.add(listener);
+        return () => {
+            this.ephemeralListeners.delete(listener);
+        };
+    }
+    request(method, params) {
+        if (this.isClosed) {
+            return Promise.reject(new TauTransportError("tau sdk client is closed"));
+        }
+        if (this.fatalError) {
+            return Promise.reject(this.fatalError);
+        }
+        let pending;
+        try {
+            pending = this.pendingRequests.create(method, params);
+        }
+        catch (error) {
+            return Promise.reject(error);
+        }
+        const payload = JSON.stringify(pending.request);
+        const requestId = pending.request.id;
+        try {
+            this.process.stdin.write(`${payload}\n`, "utf8", (error) => {
+                if (!error) {
+                    return;
+                }
+                this.pendingRequests.reject(requestId, new TauTransportError("failed to write request to tau rpc process", { cause: error }));
+            });
+        }
+        catch (error) {
+            this.pendingRequests.reject(requestId, new TauTransportError("failed to write request to tau rpc process", { cause: error }));
+        }
+        return pending.promise;
+    }
+    async close() {
+        if (this.closePromise) {
+            return this.closePromise;
+        }
+        this.isClosed = true;
+        const closeError = new TauTransportError("tau sdk client was closed");
+        this.pendingRequests.rejectAll(closeError);
+        this.rejectReadyIfPending(closeError);
+        this.deltaListeners.clear();
+        this.ephemeralListeners.clear();
+        this.process.stdout.off("data", this.handleStdoutData);
+        this.process.stderr.off("data", this.handleStderrData);
+        this.closePromise = (async () => {
+            if (!this.isExited) {
+                try {
+                    this.process.stdin.end();
+                }
+                catch {
+                    // ignore transport shutdown errors
+                }
+                try {
+                    this.process.kill("SIGTERM");
+                }
+                catch {
+                    // ignore transport shutdown errors
+                }
+                await waitForExitOrTimeout(this.exitDeferred.promise, CLOSE_TIMEOUT_MS);
+                if (!this.isExited) {
+                    try {
+                        this.process.kill("SIGKILL");
+                    }
+                    catch {
+                        // ignore transport shutdown errors
+                    }
+                    await waitForExitOrTimeout(this.exitDeferred.promise, CLOSE_TIMEOUT_MS);
+                }
+            }
+            this.process.off("error", this.handleProcessError);
+            this.process.off("exit", this.handleProcessExit);
+            this.process.off("close", this.handleProcessClose);
+        })();
+        return this.closePromise;
+    }
+    handleStdoutData = (chunk) => {
+        this.stdoutBuffer += chunk.toString();
+        while (true) {
+            const nextLineIndex = this.stdoutBuffer.indexOf("\n");
+            if (nextLineIndex === -1) {
+                break;
+            }
+            const line = this.stdoutBuffer.slice(0, nextLineIndex).trim();
+            this.stdoutBuffer = this.stdoutBuffer.slice(nextLineIndex + 1);
+            if (!line) {
+                continue;
+            }
+            this.handleSessionProtocolLine(line);
+        }
+    };
+    handleStderrData = (chunk) => {
+        this.stderrBuffer += chunk.toString();
+    };
+    handleProcessError = (error) => {
+        this.failTransport(new TauProcessError("tau rpc process transport failure", {
+            stderr: this.stderrBuffer,
+            cause: error,
+        }));
+    };
+    handleProcessExit = (code, signal) => {
+        this.onProcessTerminated({ code, signal });
+    };
+    handleProcessClose = (code, signal) => {
+        this.onProcessTerminated({ code, signal });
+    };
+    onProcessTerminated(exit) {
+        if (!this.isExited) {
+            this.isExited = true;
+            this.exitDeferred.resolve(exit);
+        }
+        if (this.isClosed) {
+            return;
+        }
+        this.isClosed = true;
+        const message = [
+            "tau rpc process exited unexpectedly",
+            `code=${exit.code ?? "null"}`,
+            `signal=${exit.signal ?? "null"}`,
+        ].join(" ");
+        this.failTransport(new TauProcessError(message, {
+            exitCode: exit.code,
+            signal: exit.signal,
+            stderr: this.stderrBuffer,
+        }));
+    }
+    handleSessionProtocolLine(line) {
+        const parsed = parseSessionProtocolOutgoingLine(line);
+        if (!parsed.ok) {
+            handleSessionProtocolTransportParseFailure({
+                failure: parsed,
+                pendingRequests: this.pendingRequests,
+                failTransport: (error) => this.failTransport(error),
+                malformedJsonPeer: "tau rpc process",
+                invalidPayloadPeer: "tau process",
+            });
+            return;
+        }
+        const message = parsed.message;
+        if (message.type === "ready") {
+            if (!this.readyValue) {
+                this.readyValue = message;
+                this.readyDeferred.resolve(message);
+            }
+            return;
+        }
+        if (message.type === "session.delta") {
+            notifySessionProtocolDeltaListeners(this.deltaListeners, message, {
+                ignoreListenerErrors: true,
+            });
+            return;
+        }
+        if (message.type === "session.ephemeral") {
+            notifySessionProtocolEphemeralListeners(this.ephemeralListeners, message, {
+                ignoreListenerErrors: true,
+            });
+            return;
+        }
+        handleSessionProtocolTransportResponse(message, this.pendingRequests, (error) => this.failTransport(error));
+    }
+    failTransport(error) {
+        if (this.fatalError) {
+            return;
+        }
+        this.fatalError = error;
+        this.pendingRequests.rejectAll(error);
+        this.rejectReadyIfPending(error);
+    }
+    rejectReadyIfPending(error) {
+        if (this.readyValue || !this.connectPromise) {
+            return;
+        }
+        this.readyDeferred.reject(error);
+    }
+}
+async function waitForExitOrTimeout(exitPromise, timeoutMs) {
+    await waitForPromiseOrTimeout(exitPromise, timeoutMs);
+}
+//# sourceMappingURL=stdio_session_transport.js.map

package/dist/transport/stdio_session_transport.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"stdio_session_transport.js","sourceRoot":"","sources":["../../src/transport/stdio_session_transport.ts"],"names":[],"mappings":"AAEA,OAAO,EACL,gCAAgC,GAMjC,MAAM,iCAAiC,CAAC;AACzC,OAAO,EAAE,eAAe,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC;AACjE,OAAO,EAEL,8BAA8B,GAC/B,MAAM,wCAAwC,CAAC;AAChD,OAAO,EACL,cAAc,EACd,0CAA0C,EAC1C,sCAAsC,EACtC,mCAAmC,EACnC,uCAAuC,EACvC,uBAAuB,EACvB,WAAW,GACZ,MAAM,yCAAyC,CAAC;AAcjD,MAAM,gBAAgB,GAAG,KAAK,CAAC;AAE/B,MAAM,OAAO,6BAA6B;IACvB,OAAO,CAAC;IACR,cAAc,GAAG,IAAI,GAAG,EAAgC,CAAC;IACzD,kBAAkB,GAAG,IAAI,GAAG,EAAoC,CAAC;IACjE,eAAe,GAAG,IAAI,8BAA8B,EAAE,CAAC;IACvD,aAAa,GAAG,cAAc,EAA+B,CAAC;IAC9D,YAAY,GAAG,cAAc,EAAmB,CAAC;IAE1D,UAAU,CAA+B;IACzC,YAAY,GAAG,EAAE,CAAC;IAClB,YAAY,GAAG,EAAE,CAAC;IAClB,cAAc,CAAiB;IAC/B,YAAY,CAAiB;IAC7B,WAAW,GAAG,KAAK,CAAC;IACpB,QAAQ,GAAG,KAAK,CAAC;IACjB,QAAQ,GAAG,KAAK,CAAC;IACjB,UAAU,CAAS;IAE3B,YAAY,OAAsC;QAChD,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QAEvB,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;QACxC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;QAExC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,IAAI,CAAC,gBAAgB,CAAC,CAAC;QACtD,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,IAAI,CAAC,gBAAgB,CAAC,CAAC;QACtD,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,EAAE,IAAI,CAAC,kBAAkB,CAAC,CAAC;QAClD,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,MAAM,EAAE,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAChD,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,EAAE,IAAI,CAAC,kBAAkB,CAAC,CAAC;IACpD,CAAC;IAED,IAAI,KAAK;QACP,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;YACrB,MAAM,IAAI,iBAAiB,CAAC,iCAAiC,CAAC,CAAC;QACjE,CAAC;QAED,OAAO,IAAI,CAAC,UAAU,CAAC;IACzB,CAAC;IAED,KAAK,CAAC,OAAO,CACX,gBAAiD,EACjD,SAAiB;QAEjB,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClB,MAAM,IAAI,iBAAiB,CAAC,0BAA0B,CAAC,CAAC;QAC1D,CAAC;QAED,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,MAAM,IAAI,CAAC,UAAU,CAAC;QACxB,CAAC;QAED,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,OAAO;QACT,CAAC;QAED,IAAI,CAAC,cAAc,KAAK,IAAI,CAAC,mBAAmB,CAAC,gBAAgB,EAAE,SAAS,CAAC,CAAC;QAC9E,MAAM,IAAI,CAAC,cAAc,CAAC;IAC5B,CAAC;IAEO,KAAK,CAAC,mBAAmB,CAC/B,gBAAiD,EACjD,SAAiB;QAEjB,MAAM,KAAK,GAAG,MAAM,WAAW,CAC7B,IAAI,CAAC,aAAa,CAAC,OAAO,EAC1B,SAAS,EACT,IAAI,iBAAiB,CACnB,8DAA8D,SAAS,IAAI,CAC5E,CACF,CAAC;QAEF,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;QACxB,MAAM,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,gBAAgB,CAAC,CAAC;QACnD,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;IAC1B,CAAC;IAED,OAAO,CAAC,QAAsC;QAC5C,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAClC,OAAO,GAAG,EAAE;YACV,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QACvC,CAAC,CAAC;IACJ,CAAC;IAED,WAAW,CAAC,QAA0C;QACpD,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACtC,OAAO,GAAG,EAAE;YACV,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC3C,CAAC,CAAC;IACJ,CAAC;IAED,OAAO,CACL,MAAS,EACT,MAAwC;QAExC,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClB,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,iBAAiB,CAAC,0BAA0B,CAAC,CAAC,CAAC;QAC3E,CAAC;QAED,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACzC,CAAC;QAED,IAAI,OAAyC,CAAC;QAC9C,IAAI,CAAC;YACH,OAAO,GAAG,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACxD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC/B,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAChD,MAAM,SAAS,GAAG,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QACrC,IAAI,CAAC;YACH,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,OAAO,IAAI,EAAE,MAAM,EAAE,CAAC,KAAK,EAAE,EAAE;gBACzD,IAAI,CAAC,KAAK,EAAE,CAAC;oBACX,OAAO;gBACT,CAAC;gBAED,IAAI,CAAC,eAAe,CAAC,MAAM,CACzB,SAAS,EACT,IAAI,iBAAiB,CAAC,4CAA4C,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CACtF,CAAC;YACJ,CAAC,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,eAAe,CAAC,MAAM,CACzB,SAAS,EACT,IAAI,iBAAiB,CAAC,4CAA4C,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CACtF,CAAC;QACJ,CAAC;QAED,OAAO,OAAO,CAAC,OAAO,CAAC;IACzB,CAAC;IAED,KAAK,CAAC,KAAK;QACT,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACtB,OAAO,IAAI,CAAC,YAAY,CAAC;QAC3B,CAAC;QAED,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;QAErB,MAAM,UAAU,GAAG,IAAI,iBAAiB,CAAC,2BAA2B,CAAC,CAAC;QACtE,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QAC3C,IAAI,CAAC,oBAAoB,CAAC,UAAU,CAAC,CAAC;QACtC,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,CAAC;QAC5B,IAAI,CAAC,kBAAkB,CAAC,KAAK,EAAE,CAAC;QAEhC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC,gBAAgB,CAAC,CAAC;QACvD,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAEvD,IAAI,CAAC,YAAY,GAAG,CAAC,KAAK,IAAI,EAAE;YAC9B,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACnB,IAAI,CAAC;oBACH,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC;gBAC3B,CAAC;gBAAC,MAAM,CAAC;oBACP,mCAAmC;gBACrC,CAAC;gBAED,IAAI,CAAC;oBACH,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBAC/B,CAAC;gBAAC,MAAM,CAAC;oBACP,mCAAmC;gBACrC,CAAC;gBAED,MAAM,oBAAoB,CAAC,IAAI,CAAC,YAAY,CAAC,OAAO,EAAE,gBAAgB,CAAC,CAAC;gBAExE,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;oBACnB,IAAI,CAAC;wBACH,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;oBAC/B,CAAC;oBAAC,MAAM,CAAC;wBACP,mCAAmC;oBACrC,CAAC;oBAED,MAAM,oBAAoB,CAAC,IAAI,CAAC,YAAY,CAAC,OAAO,EAAE,gBAAgB,CAAC,CAAC;gBAC1E,CAAC;YACH,CAAC;YAED,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI,CAAC,kBAAkB,CAAC,CAAC;YACnD,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC,iBAAiB,CAAC,CAAC;YACjD,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI,CAAC,kBAAkB,CAAC,CAAC;QACrD,CAAC,CAAC,EAAE,CAAC;QAEL,OAAO,IAAI,CAAC,YAAY,CAAC;IAC3B,CAAC;IAEgB,gBAAgB,GAAG,CAAC,KAAsB,EAAQ,EAAE;QACnE,IAAI,CAAC,YAAY,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;QAEtC,OAAO,IAAI,EAAE,CAAC;YACZ,MAAM,aAAa,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YACtD,IAAI,aAAa,KAAK,CAAC,CAAC,EAAE,CAAC;gBACzB,MAAM;YACR,CAAC;YAED,MAAM,IAAI,GAAG,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,EAAE,aAAa,CAAC,CAAC,IAAI,EAAE,CAAC;YAC9D,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,aAAa,GAAG,CAAC,CAAC,CAAC;YAE/D,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,SAAS;YACX,CAAC;YAED,IAAI,CAAC,yBAAyB,CAAC,IAAI,CAAC,CAAC;QACvC,CAAC;IACH,CAAC,CAAC;IAEe,gBAAgB,GAAG,CAAC,KAAsB,EAAQ,EAAE;QACnE,IAAI,CAAC,YAAY,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;IACxC,CAAC,CAAC;IAEe,kBAAkB,GAAG,CAAC,KAAY,EAAQ,EAAE;QAC3D,IAAI,CAAC,aAAa,CAChB,IAAI,eAAe,CAAC,mCAAmC,EAAE;YACvD,MAAM,EAAE,IAAI,CAAC,YAAY;YACzB,KAAK,EAAE,KAAK;SACb,CAAC,CACH,CAAC;IACJ,CAAC,CAAC;IAEe,iBAAiB,GAAG,CACnC,IAAmB,EACnB,MAA6B,EACvB,EAAE;QACR,IAAI,CAAC,mBAAmB,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;IAC7C,CAAC,CAAC;IAEe,kBAAkB,GAAG,CACpC,IAAmB,EACnB,MAA6B,EACvB,EAAE;QACR,IAAI,CAAC,mBAAmB,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;IAC7C,CAAC,CAAC;IAEM,mBAAmB,CAAC,IAAqB;QAC/C,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACnB,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;YACrB,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAClC,CAAC;QAED,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClB,OAAO;QACT,CAAC;QAED,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;QAErB,MAAM,OAAO,GAAG;YACd,qCAAqC;YACrC,QAAQ,IAAI,CAAC,IAAI,IAAI,MAAM,EAAE;YAC7B,UAAU,IAAI,CAAC,MAAM,IAAI,MAAM,EAAE;SAClC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAEZ,IAAI,CAAC,aAAa,CAChB,IAAI,eAAe,CAAC,OAAO,EAAE;YAC3B,QAAQ,EAAE,IAAI,CAAC,IAAI;YACnB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,MAAM,EAAE,IAAI,CAAC,YAAY;SAC1B,CAAC,CACH,CAAC;IACJ,CAAC;IAEO,yBAAyB,CAAC,IAAY;QAC5C,MAAM,MAAM,GAAG,gCAAgC,CAAC,IAAI,CAAC,CAAC;QACtD,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,CAAC;YACf,0CAA0C,CAAC;gBACzC,OAAO,EAAE,MAAM;gBACf,eAAe,EAAE,IAAI,CAAC,eAAe;gBACrC,aAAa,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC;gBACnD,iBAAiB,EAAE,iBAAiB;gBACpC,kBAAkB,EAAE,aAAa;aAClC,CAAC,CAAC;YACH,OAAO;QACT,CAAC;QAED,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC;QAC/B,IAAI,OAAO,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;YAC7B,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;gBACrB,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC;gBAC1B,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YACtC,CAAC;YACD,OAAO;QACT,CAAC;QAED,IAAI,OAAO,CAAC,IAAI,KAAK,eAAe,EAAE,CAAC;YACrC,mCAAmC,CAAC,IAAI,CAAC,cAAc,EAAE,OAAO,EAAE;gBAChE,oBAAoB,EAAE,IAAI;aAC3B,CAAC,CAAC;YACH,OAAO;QACT,CAAC;QAED,IAAI,OAAO,CAAC,IAAI,KAAK,mBAAmB,EAAE,CAAC;YACzC,uCAAuC,CAAC,IAAI,CAAC,kBAAkB,EAAE,OAAO,EAAE;gBACxE,oBAAoB,EAAE,IAAI;aAC3B,CAAC,CAAC;YACH,OAAO;QACT,CAAC;QAED,sCAAsC,CAAC,OAAO,EAAE,IAAI,CAAC,eAAe,EAAE,CAAC,KAAK,EAAE,EAAE,CAC9E,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,CAC1B,CAAC;IACJ,CAAC;IAEO,aAAa,CAAC,KAAY;QAChC,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,OAAO;QACT,CAAC;QAED,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;QACxB,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QACtC,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;IACnC,CAAC;IAEO,oBAAoB,CAAC,KAAY;QACvC,IAAI,IAAI,CAAC,UAAU,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC;YAC5C,OAAO;QACT,CAAC;QAED,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IACnC,CAAC;CACF;AAED,KAAK,UAAU,oBAAoB,CACjC,WAAqC,EACrC,SAAiB;IAEjB,MAAM,uBAAuB,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC;AACxD,CAAC"}

package/dist/transport/websocket_session_transport.d.ts

@@ -0,0 +1,52 @@
+import { type SessionProtocolInitializeParams, type SessionProtocolMethod, type SessionProtocolParamsByMethod, type SessionProtocolReadyMessage, type SessionProtocolResultByMethod } from "../protocol/session_protocol.js";
+import type { SessionProtocolDeltaListener, SessionProtocolEphemeralListener, SessionProtocolTransport } from "./session_transport.js";
+type WebSocketEventName = "open" | "message" | "error" | "close";
+type WebSocketLike = {
+    readonly readyState: number;
+    send(data: string): void;
+    close(code?: number, reason?: string): void;
+    addEventListener(type: WebSocketEventName, listener: (event: unknown) => void): void;
+    removeEventListener(type: WebSocketEventName, listener: (event: unknown) => void): void;
+};
+export type WebSocketSessionProtocolTransportOptions = {
+    url: string;
+    authToken?: string;
+    webSocketFactory?: (url: string) => WebSocketLike;
+};
+export declare class WebSocketSessionProtocolTransport implements SessionProtocolTransport {
+    private readonly url;
+    private readonly authToken?;
+    private readonly webSocketFactory?;
+    private readonly deltaListeners;
+    private readonly ephemeralListeners;
+    private readonly pendingRequests;
+    private readonly readyDeferred;
+    private readonly openDeferred;
+    private readonly closeDeferred;
+    private socket?;
+    private readyValue?;
+    private connectPromise?;
+    private closePromise?;
+    private isConnected;
+    private isClosed;
+    private isSocketOpen;
+    private fatalError?;
+    constructor(options: WebSocketSessionProtocolTransportOptions);
+    get ready(): SessionProtocolReadyMessage;
+    connect(initializeParams: SessionProtocolInitializeParams, timeoutMs: number): Promise<void>;
+    private establishConnection;
+    private createSocket;
+    request<M extends SessionProtocolMethod>(method: M, params: SessionProtocolParamsByMethod[M]): Promise<SessionProtocolResultByMethod[M]>;
+    onDelta(listener: SessionProtocolDeltaListener): () => void;
+    onEphemeral(listener: SessionProtocolEphemeralListener): () => void;
+    close(): Promise<void>;
+    private readonly handleOpen;
+    private readonly handleMessage;
+    private readonly handleError;
+    private readonly handleClose;
+    private handleSessionProtocolPayload;
+    private failTransport;
+    private rejectReadyIfPending;
+    private removeSocketListeners;
+}
+export {};

package/dist/transport/websocket_session_transport.js

@@ -0,0 +1,235 @@
+import { parseSessionProtocolOutgoingLine, } from "../protocol/session_protocol.js";
+import { TauTransportError } from "./errors.js";
+import { PendingSessionProtocolRequests, } from "./pending_session_protocol_requests.js";
+import { createDeferred, handleSessionProtocolTransportParseFailure, handleSessionProtocolTransportResponse, notifySessionProtocolDeltaListeners, notifySessionProtocolEphemeralListeners, waitForPromiseOrTimeout, withTimeout, } from "./session_protocol_transport_helpers.js";
+const WEB_SOCKET_OPEN = 1;
+const WEB_SOCKET_CLOSING = 2;
+const WEB_SOCKET_CLOSED = 3;
+const WEB_SOCKET_CLOSE_TIMEOUT_MS = 2_000;
+export class WebSocketSessionProtocolTransport {
+    url;
+    authToken;
+    webSocketFactory;
+    deltaListeners = new Set();
+    ephemeralListeners = new Set();
+    pendingRequests = new PendingSessionProtocolRequests();
+    readyDeferred = createDeferred();
+    openDeferred = createDeferred();
+    closeDeferred = createDeferred();
+    socket;
+    readyValue;
+    connectPromise;
+    closePromise;
+    isConnected = false;
+    isClosed = false;
+    isSocketOpen = false;
+    fatalError;
+    constructor(options) {
+        this.url = options.url;
+        this.authToken = options.authToken;
+        this.webSocketFactory = options.webSocketFactory;
+    }
+    get ready() {
+        if (!this.readyValue) {
+            throw new TauTransportError("tau websocket transport is not connected");
+        }
+        return this.readyValue;
+    }
+    async connect(initializeParams, timeoutMs) {
+        if (this.isClosed) {
+            throw new TauTransportError("tau websocket transport is closed");
+        }
+        if (this.fatalError) {
+            throw this.fatalError;
+        }
+        if (this.isConnected) {
+            return;
+        }
+        this.connectPromise ??= this.establishConnection(initializeParams, timeoutMs);
+        await this.connectPromise;
+    }
+    async establishConnection(initializeParams, timeoutMs) {
+        const socket = this.createSocket();
+        this.socket = socket;
+        socket.addEventListener("open", this.handleOpen);
+        socket.addEventListener("message", this.handleMessage);
+        socket.addEventListener("error", this.handleError);
+        socket.addEventListener("close", this.handleClose);
+        await withTimeout(this.openDeferred.promise, timeoutMs, new TauTransportError(`timed out opening websocket after ${timeoutMs}ms`));
+        const ready = await withTimeout(this.readyDeferred.promise, timeoutMs, new TauTransportError(`timed out waiting for session protocol ready message after ${timeoutMs}ms`));
+        this.readyValue = ready;
+        await this.request("initialize", initializeParams);
+        this.isConnected = true;
+    }
+    createSocket() {
+        const url = withAuthToken(this.url, this.authToken);
+        if (this.webSocketFactory) {
+            return this.webSocketFactory(url);
+        }
+        const WebSocketConstructor = globalThis.WebSocket;
+        if (!WebSocketConstructor) {
+            throw new TauTransportError("global WebSocket is not available in this runtime");
+        }
+        return new WebSocketConstructor(url);
+    }
+    request(method, params) {
+        if (this.isClosed) {
+            return Promise.reject(new TauTransportError("tau websocket transport is closed"));
+        }
+        if (this.fatalError) {
+            return Promise.reject(this.fatalError);
+        }
+        const socket = this.socket;
+        if (!socket || socket.readyState !== WEB_SOCKET_OPEN) {
+            return Promise.reject(new TauTransportError("tau websocket transport is not open"));
+        }
+        let pending;
+        try {
+            pending = this.pendingRequests.create(method, params);
+        }
+        catch (error) {
+            return Promise.reject(error);
+        }
+        try {
+            socket.send(JSON.stringify(pending.request));
+        }
+        catch (error) {
+            this.pendingRequests.reject(pending.request.id, new TauTransportError("failed to write request to tau websocket", { cause: error }));
+        }
+        return pending.promise;
+    }
+    onDelta(listener) {
+        this.deltaListeners.add(listener);
+        return () => {
+            this.deltaListeners.delete(listener);
+        };
+    }
+    onEphemeral(listener) {
+        this.ephemeralListeners.add(listener);
+        return () => {
+            this.ephemeralListeners.delete(listener);
+        };
+    }
+    async close() {
+        if (this.closePromise) {
+            return this.closePromise;
+        }
+        this.isClosed = true;
+        const closeError = new TauTransportError("tau websocket transport was closed");
+        this.pendingRequests.rejectAll(closeError);
+        this.rejectReadyIfPending(closeError);
+        this.deltaListeners.clear();
+        this.ephemeralListeners.clear();
+        this.closePromise = (async () => {
+            const socket = this.socket;
+            if (socket) {
+                if (socket.readyState !== WEB_SOCKET_CLOSING && socket.readyState !== WEB_SOCKET_CLOSED) {
+                    socket.close(1000, "client closed");
+                }
+                if (socket.readyState === WEB_SOCKET_OPEN || socket.readyState === WEB_SOCKET_CLOSING) {
+                    await waitForCloseOrTimeout(this.closeDeferred.promise, WEB_SOCKET_CLOSE_TIMEOUT_MS);
+                }
+                this.removeSocketListeners(socket);
+            }
+        })();
+        return this.closePromise;
+    }
+    handleOpen = () => {
+        this.isSocketOpen = true;
+        this.openDeferred.resolve();
+    };
+    handleMessage = (event) => {
+        const data = readMessageEventData(event);
+        if (typeof data !== "string") {
+            this.failTransport(new TauTransportError("received non-text websocket message"));
+            return;
+        }
+        this.handleSessionProtocolPayload(data);
+    };
+    handleError = (event) => {
+        this.failTransport(new TauTransportError("tau websocket transport failure", { cause: event }));
+    };
+    handleClose = () => {
+        this.closeDeferred.resolve();
+        if (this.isClosed) {
+            return;
+        }
+        this.isClosed = true;
+        this.failTransport(new TauTransportError("tau websocket closed unexpectedly"));
+    };
+    handleSessionProtocolPayload(payload) {
+        const parsed = parseSessionProtocolOutgoingLine(payload);
+        if (!parsed.ok) {
+            handleSessionProtocolTransportParseFailure({
+                failure: parsed,
+                pendingRequests: this.pendingRequests,
+                failTransport: (error) => this.failTransport(error),
+                malformedJsonPeer: "tau websocket",
+                invalidPayloadPeer: "tau websocket",
+            });
+            return;
+        }
+        const message = parsed.message;
+        if (message.type === "ready") {
+            if (!this.readyValue) {
+                this.readyValue = message;
+                this.readyDeferred.resolve(message);
+            }
+            return;
+        }
+        if (message.type === "session.delta") {
+            notifySessionProtocolDeltaListeners(this.deltaListeners, message, {
+                ignoreListenerErrors: true,
+            });
+            return;
+        }
+        if (message.type === "session.ephemeral") {
+            notifySessionProtocolEphemeralListeners(this.ephemeralListeners, message, {
+                ignoreListenerErrors: true,
+            });
+            return;
+        }
+        handleSessionProtocolTransportResponse(message, this.pendingRequests, (error) => this.failTransport(error));
+    }
+    failTransport(error) {
+        if (this.fatalError) {
+            return;
+        }
+        this.fatalError = error;
+        this.pendingRequests.rejectAll(error);
+        this.rejectReadyIfPending(error);
+    }
+    rejectReadyIfPending(error) {
+        if (this.readyValue || !this.connectPromise) {
+            return;
+        }
+        if (this.isSocketOpen) {
+            this.readyDeferred.reject(error);
+        }
+        this.openDeferred.reject(error);
+    }
+    removeSocketListeners(socket) {
+        socket.removeEventListener("open", this.handleOpen);
+        socket.removeEventListener("message", this.handleMessage);
+        socket.removeEventListener("error", this.handleError);
+        socket.removeEventListener("close", this.handleClose);
+    }
+}
+function withAuthToken(url, authToken) {
+    if (authToken === undefined) {
+        return url;
+    }
+    const parsed = new URL(url);
+    parsed.searchParams.set("tau_token", authToken);
+    return parsed.toString();
+}
+function readMessageEventData(event) {
+    if (typeof event === "object" && event !== null && "data" in event) {
+        return event.data;
+    }
+    return undefined;
+}
+async function waitForCloseOrTimeout(closePromise, timeoutMs) {
+    await waitForPromiseOrTimeout(closePromise, timeoutMs);
+}
+//# sourceMappingURL=websocket_session_transport.js.map

package/dist/transport/websocket_session_transport.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"websocket_session_transport.js","sourceRoot":"","sources":["../../src/transport/websocket_session_transport.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,gCAAgC,GAMjC,MAAM,iCAAiC,CAAC;AACzC,OAAO,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC;AAChD,OAAO,EAEL,8BAA8B,GAC/B,MAAM,wCAAwC,CAAC;AAChD,OAAO,EACL,cAAc,EACd,0CAA0C,EAC1C,sCAAsC,EACtC,mCAAmC,EACnC,uCAAuC,EACvC,uBAAuB,EACvB,WAAW,GACZ,MAAM,yCAAyC,CAAC;AAuBjD,MAAM,eAAe,GAAG,CAAC,CAAC;AAC1B,MAAM,kBAAkB,GAAG,CAAC,CAAC;AAC7B,MAAM,iBAAiB,GAAG,CAAC,CAAC;AAC5B,MAAM,2BAA2B,GAAG,KAAK,CAAC;AAE1C,MAAM,OAAO,iCAAiC;IAC3B,GAAG,CAAS;IACZ,SAAS,CAAU;IACnB,gBAAgB,CAAkC;IAClD,cAAc,GAAG,IAAI,GAAG,EAAgC,CAAC;IACzD,kBAAkB,GAAG,IAAI,GAAG,EAAoC,CAAC;IACjE,eAAe,GAAG,IAAI,8BAA8B,EAAE,CAAC;IACvD,aAAa,GAAG,cAAc,EAA+B,CAAC;IAC9D,YAAY,GAAG,cAAc,EAAQ,CAAC;IACtC,aAAa,GAAG,cAAc,EAAQ,CAAC;IAEhD,MAAM,CAAiB;IACvB,UAAU,CAA+B;IACzC,cAAc,CAAiB;IAC/B,YAAY,CAAiB;IAC7B,WAAW,GAAG,KAAK,CAAC;IACpB,QAAQ,GAAG,KAAK,CAAC;IACjB,YAAY,GAAG,KAAK,CAAC;IACrB,UAAU,CAAS;IAE3B,YAAY,OAAiD;QAC3D,IAAI,CAAC,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC;QACvB,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;QACnC,IAAI,CAAC,gBAAgB,GAAG,OAAO,CAAC,gBAAgB,CAAC;IACnD,CAAC;IAED,IAAI,KAAK;QACP,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;YACrB,MAAM,IAAI,iBAAiB,CAAC,0CAA0C,CAAC,CAAC;QAC1E,CAAC;QAED,OAAO,IAAI,CAAC,UAAU,CAAC;IACzB,CAAC;IAED,KAAK,CAAC,OAAO,CACX,gBAAiD,EACjD,SAAiB;QAEjB,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClB,MAAM,IAAI,iBAAiB,CAAC,mCAAmC,CAAC,CAAC;QACnE,CAAC;QAED,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,MAAM,IAAI,CAAC,UAAU,CAAC;QACxB,CAAC;QAED,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,OAAO;QACT,CAAC;QAED,IAAI,CAAC,cAAc,KAAK,IAAI,CAAC,mBAAmB,CAAC,gBAAgB,EAAE,SAAS,CAAC,CAAC;QAC9E,MAAM,IAAI,CAAC,cAAc,CAAC;IAC5B,CAAC;IAEO,KAAK,CAAC,mBAAmB,CAC/B,gBAAiD,EACjD,SAAiB;QAEjB,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,EAAE,CAAC;QACnC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,MAAM,CAAC,gBAAgB,CAAC,MAAM,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;QACjD,MAAM,CAAC,gBAAgB,CAAC,SAAS,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC;QACvD,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;QACnD,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;QAEnD,MAAM,WAAW,CACf,IAAI,CAAC,YAAY,CAAC,OAAO,EACzB,SAAS,EACT,IAAI,iBAAiB,CAAC,qCAAqC,SAAS,IAAI,CAAC,CAC1E,CAAC;QACF,MAAM,KAAK,GAAG,MAAM,WAAW,CAC7B,IAAI,CAAC,aAAa,CAAC,OAAO,EAC1B,SAAS,EACT,IAAI,iBAAiB,CACnB,8DAA8D,SAAS,IAAI,CAC5E,CACF,CAAC;QAEF,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;QACxB,MAAM,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,gBAAgB,CAAC,CAAC;QACnD,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;IAC1B,CAAC;IAEO,YAAY;QAClB,MAAM,GAAG,GAAG,aAAa,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;QACpD,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC1B,OAAO,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC;QACpC,CAAC;QAED,MAAM,oBAAoB,GAAG,UAAU,CAAC,SAAS,CAAC;QAClD,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC1B,MAAM,IAAI,iBAAiB,CAAC,mDAAmD,CAAC,CAAC;QACnF,CAAC;QAED,OAAO,IAAI,oBAAoB,CAAC,GAAG,CAAkB,CAAC;IACxD,CAAC;IAED,OAAO,CACL,MAAS,EACT,MAAwC;QAExC,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClB,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,iBAAiB,CAAC,mCAAmC,CAAC,CAAC,CAAC;QACpF,CAAC;QAED,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACzC,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;QAC3B,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC,UAAU,KAAK,eAAe,EAAE,CAAC;YACrD,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,iBAAiB,CAAC,qCAAqC,CAAC,CAAC,CAAC;QACtF,CAAC;QAED,IAAI,OAAyC,CAAC;QAC9C,IAAI,CAAC;YACH,OAAO,GAAG,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACxD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC/B,CAAC;QAED,IAAI,CAAC;YACH,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC;QAC/C,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,eAAe,CAAC,MAAM,CACzB,OAAO,CAAC,OAAO,CAAC,EAAE,EAClB,IAAI,iBAAiB,CAAC,0CAA0C,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CACpF,CAAC;QACJ,CAAC;QAED,OAAO,OAAO,CAAC,OAAO,CAAC;IACzB,CAAC;IAED,OAAO,CAAC,QAAsC;QAC5C,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAClC,OAAO,GAAG,EAAE;YACV,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QACvC,CAAC,CAAC;IACJ,CAAC;IAED,WAAW,CAAC,QAA0C;QACpD,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACtC,OAAO,GAAG,EAAE;YACV,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC3C,CAAC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,KAAK;QACT,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACtB,OAAO,IAAI,CAAC,YAAY,CAAC;QAC3B,CAAC;QAED,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;QAErB,MAAM,UAAU,GAAG,IAAI,iBAAiB,CAAC,oCAAoC,CAAC,CAAC;QAC/E,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QAC3C,IAAI,CAAC,oBAAoB,CAAC,UAAU,CAAC,CAAC;QACtC,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,CAAC;QAC5B,IAAI,CAAC,kBAAkB,CAAC,KAAK,EAAE,CAAC;QAEhC,IAAI,CAAC,YAAY,GAAG,CAAC,KAAK,IAAI,EAAE;YAC9B,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;YAC3B,IAAI,MAAM,EAAE,CAAC;gBACX,IAAI,MAAM,CAAC,UAAU,KAAK,kBAAkB,IAAI,MAAM,CAAC,UAAU,KAAK,iBAAiB,EAAE,CAAC;oBACxF,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC;gBACtC,CAAC;gBACD,IAAI,MAAM,CAAC,UAAU,KAAK,eAAe,IAAI,MAAM,CAAC,UAAU,KAAK,kBAAkB,EAAE,CAAC;oBACtF,MAAM,qBAAqB,CAAC,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,2BAA2B,CAAC,CAAC;gBACvF,CAAC;gBACD,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC;YACrC,CAAC;QACH,CAAC,CAAC,EAAE,CAAC;QAEL,OAAO,IAAI,CAAC,YAAY,CAAC;IAC3B,CAAC;IAEgB,UAAU,GAAG,GAAS,EAAE;QACvC,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;QACzB,IAAI,CAAC,YAAY,CAAC,OAAO,EAAE,CAAC;IAC9B,CAAC,CAAC;IAEe,aAAa,GAAG,CAAC,KAAc,EAAQ,EAAE;QACxD,MAAM,IAAI,GAAG,oBAAoB,CAAC,KAAK,CAAC,CAAC;QACzC,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC7B,IAAI,CAAC,aAAa,CAAC,IAAI,iBAAiB,CAAC,qCAAqC,CAAC,CAAC,CAAC;YACjF,OAAO;QACT,CAAC;QAED,IAAI,CAAC,4BAA4B,CAAC,IAAI,CAAC,CAAC;IAC1C,CAAC,CAAC;IAEe,WAAW,GAAG,CAAC,KAAc,EAAQ,EAAE;QACtD,IAAI,CAAC,aAAa,CAAC,IAAI,iBAAiB,CAAC,iCAAiC,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC;IACjG,CAAC,CAAC;IAEe,WAAW,GAAG,GAAS,EAAE;QACxC,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,CAAC;QAC7B,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClB,OAAO;QACT,CAAC;QAED,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;QACrB,IAAI,CAAC,aAAa,CAAC,IAAI,iBAAiB,CAAC,mCAAmC,CAAC,CAAC,CAAC;IACjF,CAAC,CAAC;IAEM,4BAA4B,CAAC,OAAe;QAClD,MAAM,MAAM,GAAG,gCAAgC,CAAC,OAAO,CAAC,CAAC;QACzD,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,CAAC;YACf,0CAA0C,CAAC;gBACzC,OAAO,EAAE,MAAM;gBACf,eAAe,EAAE,IAAI,CAAC,eAAe;gBACrC,aAAa,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC;gBACnD,iBAAiB,EAAE,eAAe;gBAClC,kBAAkB,EAAE,eAAe;aACpC,CAAC,CAAC;YACH,OAAO;QACT,CAAC;QAED,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC;QAC/B,IAAI,OAAO,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;YAC7B,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;gBACrB,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC;gBAC1B,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YACtC,CAAC;YACD,OAAO;QACT,CAAC;QAED,IAAI,OAAO,CAAC,IAAI,KAAK,eAAe,EAAE,CAAC;YACrC,mCAAmC,CAAC,IAAI,CAAC,cAAc,EAAE,OAAO,EAAE;gBAChE,oBAAoB,EAAE,IAAI;aAC3B,CAAC,CAAC;YACH,OAAO;QACT,CAAC;QAED,IAAI,OAAO,CAAC,IAAI,KAAK,mBAAmB,EAAE,CAAC;YACzC,uCAAuC,CAAC,IAAI,CAAC,kBAAkB,EAAE,OAAO,EAAE;gBACxE,oBAAoB,EAAE,IAAI;aAC3B,CAAC,CAAC;YACH,OAAO;QACT,CAAC;QAED,sCAAsC,CAAC,OAAO,EAAE,IAAI,CAAC,eAAe,EAAE,CAAC,KAAK,EAAE,EAAE,CAC9E,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,CAC1B,CAAC;IACJ,CAAC;IAEO,aAAa,CAAC,KAAY;QAChC,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,OAAO;QACT,CAAC;QAED,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;QACxB,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QACtC,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;IACnC,CAAC;IAEO,oBAAoB,CAAC,KAAY;QACvC,IAAI,IAAI,CAAC,UAAU,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC;YAC5C,OAAO;QACT,CAAC;QAED,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACtB,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACnC,CAAC;QACD,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAClC,CAAC;IAEO,qBAAqB,CAAC,MAAqB;QACjD,MAAM,CAAC,mBAAmB,CAAC,MAAM,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;QACpD,MAAM,CAAC,mBAAmB,CAAC,SAAS,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC;QAC1D,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;QACtD,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;IACxD,CAAC;CACF;AAED,SAAS,aAAa,CAAC,GAAW,EAAE,SAA6B;IAC/D,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;QAC5B,OAAO,GAAG,CAAC;IACb,CAAC;IAED,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC,CAAC;IAC5B,MAAM,CAAC,YAAY,CAAC,GAAG,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC;IAChD,OAAO,MAAM,CAAC,QAAQ,EAAE,CAAC;AAC3B,CAAC;AAED,SAAS,oBAAoB,CAAC,KAAc;IAC1C,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,MAAM,IAAI,KAAK,EAAE,CAAC;QACnE,OAAO,KAAK,CAAC,IAAI,CAAC;IACpB,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,KAAK,UAAU,qBAAqB,CAClC,YAA2B,EAC3B,SAAiB;IAEjB,MAAM,uBAAuB,CAAC,YAAY,EAAE,SAAS,CAAC,CAAC;AACzD,CAAC"}

package/dist/tui/chat_controller/assistant_clipboard.js

@@ -0,0 +1,39 @@
+import { extractAllFencedCodeBlocks, extractAssistantText } from "../../core/utils/messages.js";
+import { copyTextToClipboard } from "../clipboard.js";
+export async function copyAssistantTextToClipboard(options) {
+    if (!options.message) {
+        options.view.addSystemMessage("no assistant message to copy yet.", "warn");
+        return;
+    }
+    const text = extractAssistantText(options.message);
+    if (!text.trim()) {
+        options.view.addSystemMessage("last assistant message was empty.", "warn");
+        return;
+    }
+    try {
+        await copyTextToClipboard(text);
+        options.view.addSystemMessage("copied last assistant message to clipboard.", "success");
+    }
+    catch (error) {
+        options.view.addSystemMessage(`clipboard copy failed: ${error.message}`, "error");
+    }
+}
+export async function copyAssistantCodeToClipboard(options) {
+    if (!options.message) {
+        options.view.addSystemMessage("no assistant message to copy yet.", "warn");
+        return;
+    }
+    const code = extractAllFencedCodeBlocks(extractAssistantText(options.message));
+    if (!code) {
+        options.view.addSystemMessage("no code block to copy yet.", "warn");
+        return;
+    }
+    try {
+        await copyTextToClipboard(code);
+        options.view.addSystemMessage("copied all code blocks to clipboard.", "success");
+    }
+    catch (error) {
+        options.view.addSystemMessage(`clipboard copy failed: ${error.message}`, "error");
+    }
+}
+//# sourceMappingURL=assistant_clipboard.js.map

package/dist/tui/chat_controller/assistant_clipboard.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"assistant_clipboard.js","sourceRoot":"","sources":["../../../src/tui/chat_controller/assistant_clipboard.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,0BAA0B,EAAE,oBAAoB,EAAE,MAAM,8BAA8B,CAAC;AAEhG,OAAO,EAAE,mBAAmB,EAAE,MAAM,iBAAiB,CAAC;AAEtD,MAAM,CAAC,KAAK,UAAU,4BAA4B,CAAC,OAGlD;IACC,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;QACrB,OAAO,CAAC,IAAI,CAAC,gBAAgB,CAAC,mCAAmC,EAAE,MAAM,CAAC,CAAC;QAC3E,OAAO;IACT,CAAC;IAED,MAAM,IAAI,GAAG,oBAAoB,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IACnD,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC;QACjB,OAAO,CAAC,IAAI,CAAC,gBAAgB,CAAC,mCAAmC,EAAE,MAAM,CAAC,CAAC;QAC3E,OAAO;IACT,CAAC;IAED,IAAI,CAAC;QACH,MAAM,mBAAmB,CAAC,IAAI,CAAC,CAAC;QAChC,OAAO,CAAC,IAAI,CAAC,gBAAgB,CAAC,6CAA6C,EAAE,SAAS,CAAC,CAAC;IAC1F,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,IAAI,CAAC,gBAAgB,CAAC,0BAA2B,KAAe,CAAC,OAAO,EAAE,EAAE,OAAO,CAAC,CAAC;IAC/F,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,4BAA4B,CAAC,OAGlD;IACC,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;QACrB,OAAO,CAAC,IAAI,CAAC,gBAAgB,CAAC,mCAAmC,EAAE,MAAM,CAAC,CAAC;QAC3E,OAAO;IACT,CAAC;IAED,MAAM,IAAI,GAAG,0BAA0B,CAAC,oBAAoB,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC;IAC/E,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,OAAO,CAAC,IAAI,CAAC,gBAAgB,CAAC,4BAA4B,EAAE,MAAM,CAAC,CAAC;QACpE,OAAO;IACT,CAAC;IAED,IAAI,CAAC;QACH,MAAM,mBAAmB,CAAC,IAAI,CAAC,CAAC;QAChC,OAAO,CAAC,IAAI,CAAC,gBAAgB,CAAC,sCAAsC,EAAE,SAAS,CAAC,CAAC;IACnF,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,IAAI,CAAC,gBAAgB,CAAC,0BAA2B,KAAe,CAAC,OAAO,EAAE,EAAE,OAAO,CAAC,CAAC;IAC/F,CAAC;AACH,CAAC"}