@earendil-works/pi-server 0.84.0

package/dist/testing/client.js

@@ -0,0 +1,118 @@
+import { once } from "node:events";
+import { createConnection } from "node:net";
+import { encodeClientMessage, PROTOCOL_VERSION, ServerMessageDecoder, } from "@earendil-works/pi-protocol";
+import { Deferred } from "./service.js";
+export class ProtocolTestClient {
+    messages = [];
+    channel;
+    decoder = new ServerMessageDecoder();
+    waiters = new Set();
+    closedDeferred = new Deferred();
+    requestSequence = 0;
+    closedValue = false;
+    constructor(channel) {
+        this.channel = channel;
+    }
+    get closed() {
+        return this.closedValue;
+    }
+    hello(version = PROTOCOL_VERSION) {
+        const response = this.next((message) => message.type === "hello" || message.type === "hello_error");
+        void this.sendMessage({ type: "hello", version });
+        return response;
+    }
+    async request(command, id = `request-${++this.requestSequence}`) {
+        const response = this.next((message) => message.type === "response" && message.id === id);
+        await this.sendMessage({ type: "request", id, request: command });
+        return (await response);
+    }
+    sendMessage(message) {
+        return this.channel.send(encodeClientMessage(message));
+    }
+    sendBytes(chunk) {
+        return this.channel.send(chunk);
+    }
+    sendFragmentedMessage(message, splitAt) {
+        return this.channel.sendFragmented(encodeClientMessage(message), splitAt);
+    }
+    next(predicate) {
+        return this.nextFrom(0, predicate);
+    }
+    nextFrom(index, predicate) {
+        const existing = this.messages.slice(index).find(predicate);
+        if (existing)
+            return Promise.resolve(existing);
+        if (this.closedValue)
+            return Promise.reject(new Error("Wire client is closed"));
+        return new Promise((resolve, reject) => this.waiters.add({ predicate, resolve, reject }));
+    }
+    waitForClose() {
+        return this.closedValue ? Promise.resolve() : this.closedDeferred.promise;
+    }
+    close() {
+        return this.channel.close();
+    }
+    receive(chunk) {
+        try {
+            for (const message of this.decoder.push(chunk)) {
+                this.messages.push(message);
+                for (const waiter of this.waiters) {
+                    if (!waiter.predicate(message))
+                        continue;
+                    this.waiters.delete(waiter);
+                    waiter.resolve(message);
+                }
+            }
+        }
+        catch (error) {
+            this.fail(error instanceof Error ? error : new Error(String(error)));
+        }
+    }
+    markClosed() {
+        if (this.closedValue)
+            return;
+        this.closedValue = true;
+        this.closedDeferred.resolve(undefined);
+        this.fail(new Error("Wire connection closed"));
+    }
+    fail(error) {
+        for (const waiter of this.waiters)
+            waiter.reject(error);
+        this.waiters.clear();
+    }
+}
+export async function connectUnixTestClient(path) {
+    const socket = createConnection(path);
+    await once(socket, "connect");
+    const client = new ProtocolTestClient({
+        send: (chunk) => writeSocket(socket, chunk),
+        async sendFragmented(chunk, splitAt) {
+            await writeSocket(socket, chunk.subarray(0, splitAt));
+            await writeSocket(socket, chunk.subarray(splitAt));
+        },
+        async close() {
+            if (socket.destroyed)
+                return;
+            const closed = once(socket, "close");
+            socket.destroy();
+            await closed;
+        },
+    });
+    socket.on("data", (chunk) => {
+        client.receive(new Uint8Array(chunk.buffer, chunk.byteOffset, chunk.byteLength));
+    });
+    socket.on("error", (error) => client.fail(error));
+    socket.once("close", () => client.markClosed());
+    return client;
+}
+function writeSocket(socket, chunk) {
+    return new Promise((resolve, reject) => {
+        socket.write(chunk, (error) => {
+            if (error)
+                reject(error);
+            else
+                resolve();
+        });
+    });
+}
+//# sourceMappingURL=client.js.map

package/dist/testing/client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"client.js","sourceRoot":"","sources":["../../src/testing/client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,aAAa,CAAC;AACnC,OAAO,EAAE,gBAAgB,EAAe,MAAM,UAAU,CAAC;AACzD,OAAO,EAGN,mBAAmB,EACnB,gBAAgB,EAGhB,oBAAoB,GACpB,MAAM,6BAA6B,CAAC;AACrC,OAAO,EAAE,QAAQ,EAAE,MAAM,cAAc,CAAC;AAcxC,MAAM,OAAO,kBAAkB;IACrB,QAAQ,GAAoB,EAAE,CAAC;IACvB,OAAO,CAAc;IACrB,OAAO,GAAG,IAAI,oBAAoB,EAAE,CAAC;IACrC,OAAO,GAAG,IAAI,GAAG,EAAiB,CAAC;IACnC,cAAc,GAAG,IAAI,QAAQ,EAAQ,CAAC;IAC/C,eAAe,GAAG,CAAC,CAAC;IACpB,WAAW,GAAG,KAAK,CAAC;IAE5B,YAAY,OAAoB,EAAE;QACjC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IAAA,CACvB;IAED,IAAI,MAAM,GAAY;QACrB,OAAO,IAAI,CAAC,WAAW,CAAC;IAAA,CACxB;IAED,KAAK,CAAC,OAAO,GAAW,gBAAgB,EAA0B;QACjE,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,OAAO,IAAI,OAAO,CAAC,IAAI,KAAK,aAAa,CAAC,CAAC;QACpG,KAAK,IAAI,CAAC,WAAW,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;QAClD,OAAO,QAAQ,CAAC;IAAA,CAChB;IAED,KAAK,CAAC,OAAO,CAAC,OAAgB,EAAE,EAAE,GAAG,WAAW,EAAE,IAAI,CAAC,eAAe,EAAE,EAA6B;QACpG,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CACzB,CAAC,OAAO,EAA+B,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,UAAU,IAAI,OAAO,CAAC,EAAE,KAAK,EAAE,CAC1F,CAAC;QACF,MAAM,IAAI,CAAC,WAAW,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;QAClE,OAAO,CAAC,MAAM,QAAQ,CAAqB,CAAC;IAAA,CAC5C;IAED,WAAW,CAAC,OAAsB,EAAiB;QAClD,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC,CAAC;IAAA,CACvD;IAED,SAAS,CAAC,KAAiB,EAAiB;QAC3C,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAAA,CAChC;IAED,qBAAqB,CAAC,OAAsB,EAAE,OAAe,EAAiB;QAC7E,OAAO,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,mBAAmB,CAAC,OAAO,CAAC,EAAE,OAAO,CAAC,CAAC;IAAA,CAC1E;IAED,IAAI,CAAC,SAA8C,EAA0B;QAC5E,OAAO,IAAI,CAAC,QAAQ,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;IAAA,CACnC;IAED,QAAQ,CAAC,KAAa,EAAE,SAA8C,EAA0B;QAC/F,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC5D,IAAI,QAAQ;YAAE,OAAO,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAC/C,IAAI,IAAI,CAAC,WAAW;YAAE,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC,CAAC;QAChF,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;IAAA,CAC1F;IAED,YAAY,GAAkB;QAC7B,OAAO,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC;IAAA,CAC1E;IAED,KAAK,GAAkB;QACtB,OAAO,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;IAAA,CAC5B;IAED,OAAO,CAAC,KAAiB,EAAQ;QAChC,IAAI,CAAC;YACJ,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;gBAChD,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBAC5B,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;oBACnC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,OAAO,CAAC;wBAAE,SAAS;oBACzC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;oBAC5B,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;gBACzB,CAAC;YACF,CAAC;QACF,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,IAAI,CAAC,IAAI,CAAC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACtE,CAAC;IAAA,CACD;IAED,UAAU,GAAS;QAClB,IAAI,IAAI,CAAC,WAAW;YAAE,OAAO;QAC7B,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;QACxB,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACvC,IAAI,CAAC,IAAI,CAAC,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC,CAAC;IAAA,CAC/C;IAED,IAAI,CAAC,KAAY,EAAQ;QACxB,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO;YAAE,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACxD,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;IAAA,CACrB;CACD;AAED,MAAM,CAAC,KAAK,UAAU,qBAAqB,CAAC,IAAY,EAA+B;IACtF,MAAM,MAAM,GAAG,gBAAgB,CAAC,IAAI,CAAC,CAAC;IACtC,MAAM,IAAI,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;IAC9B,MAAM,MAAM,GAAG,IAAI,kBAAkB,CAAC;QACrC,IAAI,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,WAAW,CAAC,MAAM,EAAE,KAAK,CAAC;QAC3C,KAAK,CAAC,cAAc,CAAC,KAAK,EAAE,OAAO,EAAE;YACpC,MAAM,WAAW,CAAC,MAAM,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC,CAAC;YACtD,MAAM,WAAW,CAAC,MAAM,EAAE,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC;QAAA,CACnD;QACD,KAAK,CAAC,KAAK,GAAG;YACb,IAAI,MAAM,CAAC,SAAS;gBAAE,OAAO;YAC7B,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YACrC,MAAM,CAAC,OAAO,EAAE,CAAC;YACjB,MAAM,MAAM,CAAC;QAAA,CACb;KACD,CAAC,CAAC;IACH,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC;QAC5B,MAAM,CAAC,OAAO,CAAC,IAAI,UAAU,CAAC,KAAK,CAAC,MAAM,EAAE,KAAK,CAAC,UAAU,EAAE,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC;IAAA,CACjF,CAAC,CAAC;IACH,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;IAClD,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC;IAChD,OAAO,MAAM,CAAC;AAAA,CACd;AAED,SAAS,WAAW,CAAC,MAAc,EAAE,KAAiB,EAAiB;IACtE,OAAO,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,CAAC;QAC7C,MAAM,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC;YAC9B,IAAI,KAAK;gBAAE,MAAM,CAAC,KAAK,CAAC,CAAC;;gBACpB,OAAO,EAAE,CAAC;QAAA,CACf,CAAC,CAAC;IAAA,CACH,CAAC,CAAC;AAAA,CACH","sourcesContent":["import { once } from \"node:events\";\nimport { createConnection, type Socket } from \"node:net\";\nimport {\n\ttype ClientMessage,\n\ttype Command,\n\tencodeClientMessage,\n\tPROTOCOL_VERSION,\n\ttype ResponseEnvelope,\n\ttype ServerMessage,\n\tServerMessageDecoder,\n} from \"@earendil-works/pi-protocol\";\nimport { Deferred } from \"./service.ts\";\n\ninterface MessageWaiter {\n\tpredicate: (message: ServerMessage) => boolean;\n\tresolve: (message: ServerMessage) => void;\n\treject: (error: Error) => void;\n}\n\nexport interface WireChannel {\n\tsend(chunk: Uint8Array): Promise<void>;\n\tsendFragmented(chunk: Uint8Array, splitAt: number): Promise<void>;\n\tclose(): Promise<void>;\n}\n\nexport class ProtocolTestClient {\n\treadonly messages: ServerMessage[] = [];\n\tprivate readonly channel: WireChannel;\n\tprivate readonly decoder = new ServerMessageDecoder();\n\tprivate readonly waiters = new Set<MessageWaiter>();\n\tprivate readonly closedDeferred = new Deferred<void>();\n\tprivate requestSequence = 0;\n\tprivate closedValue = false;\n\n\tconstructor(channel: WireChannel) {\n\t\tthis.channel = channel;\n\t}\n\n\tget closed(): boolean {\n\t\treturn this.closedValue;\n\t}\n\n\thello(version: number = PROTOCOL_VERSION): Promise<ServerMessage> {\n\t\tconst response = this.next((message) => message.type === \"hello\" || message.type === \"hello_error\");\n\t\tvoid this.sendMessage({ type: \"hello\", version });\n\t\treturn response;\n\t}\n\n\tasync request(command: Command, id = `request-${++this.requestSequence}`): Promise<ResponseEnvelope> {\n\t\tconst response = this.next(\n\t\t\t(message): message is ResponseEnvelope => message.type === \"response\" && message.id === id,\n\t\t);\n\t\tawait this.sendMessage({ type: \"request\", id, request: command });\n\t\treturn (await response) as ResponseEnvelope;\n\t}\n\n\tsendMessage(message: ClientMessage): Promise<void> {\n\t\treturn this.channel.send(encodeClientMessage(message));\n\t}\n\n\tsendBytes(chunk: Uint8Array): Promise<void> {\n\t\treturn this.channel.send(chunk);\n\t}\n\n\tsendFragmentedMessage(message: ClientMessage, splitAt: number): Promise<void> {\n\t\treturn this.channel.sendFragmented(encodeClientMessage(message), splitAt);\n\t}\n\n\tnext(predicate: (message: ServerMessage) => boolean): Promise<ServerMessage> {\n\t\treturn this.nextFrom(0, predicate);\n\t}\n\n\tnextFrom(index: number, predicate: (message: ServerMessage) => boolean): Promise<ServerMessage> {\n\t\tconst existing = this.messages.slice(index).find(predicate);\n\t\tif (existing) return Promise.resolve(existing);\n\t\tif (this.closedValue) return Promise.reject(new Error(\"Wire client is closed\"));\n\t\treturn new Promise((resolve, reject) => this.waiters.add({ predicate, resolve, reject }));\n\t}\n\n\twaitForClose(): Promise<void> {\n\t\treturn this.closedValue ? Promise.resolve() : this.closedDeferred.promise;\n\t}\n\n\tclose(): Promise<void> {\n\t\treturn this.channel.close();\n\t}\n\n\treceive(chunk: Uint8Array): void {\n\t\ttry {\n\t\t\tfor (const message of this.decoder.push(chunk)) {\n\t\t\t\tthis.messages.push(message);\n\t\t\t\tfor (const waiter of this.waiters) {\n\t\t\t\t\tif (!waiter.predicate(message)) continue;\n\t\t\t\t\tthis.waiters.delete(waiter);\n\t\t\t\t\twaiter.resolve(message);\n\t\t\t\t}\n\t\t\t}\n\t\t} catch (error) {\n\t\t\tthis.fail(error instanceof Error ? error : new Error(String(error)));\n\t\t}\n\t}\n\n\tmarkClosed(): void {\n\t\tif (this.closedValue) return;\n\t\tthis.closedValue = true;\n\t\tthis.closedDeferred.resolve(undefined);\n\t\tthis.fail(new Error(\"Wire connection closed\"));\n\t}\n\n\tfail(error: Error): void {\n\t\tfor (const waiter of this.waiters) waiter.reject(error);\n\t\tthis.waiters.clear();\n\t}\n}\n\nexport async function connectUnixTestClient(path: string): Promise<ProtocolTestClient> {\n\tconst socket = createConnection(path);\n\tawait once(socket, \"connect\");\n\tconst client = new ProtocolTestClient({\n\t\tsend: (chunk) => writeSocket(socket, chunk),\n\t\tasync sendFragmented(chunk, splitAt) {\n\t\t\tawait writeSocket(socket, chunk.subarray(0, splitAt));\n\t\t\tawait writeSocket(socket, chunk.subarray(splitAt));\n\t\t},\n\t\tasync close() {\n\t\t\tif (socket.destroyed) return;\n\t\t\tconst closed = once(socket, \"close\");\n\t\t\tsocket.destroy();\n\t\t\tawait closed;\n\t\t},\n\t});\n\tsocket.on(\"data\", (chunk) => {\n\t\tclient.receive(new Uint8Array(chunk.buffer, chunk.byteOffset, chunk.byteLength));\n\t});\n\tsocket.on(\"error\", (error) => client.fail(error));\n\tsocket.once(\"close\", () => client.markClosed());\n\treturn client;\n}\n\nfunction writeSocket(socket: Socket, chunk: Uint8Array): Promise<void> {\n\treturn new Promise<void>((resolve, reject) => {\n\t\tsocket.write(chunk, (error) => {\n\t\t\tif (error) reject(error);\n\t\t\telse resolve();\n\t\t});\n\t});\n}\n"]}

package/dist/testing/index.d.ts

@@ -0,0 +1,6 @@
+export type { WireChannel } from "./client.ts";
+export { connectUnixTestClient, ProtocolTestClient } from "./client.ts";
+export type { TestServer, TestServerOptions } from "./server.ts";
+export { createTestServer } from "./server.ts";
+export { Deferred, TEST_MODEL, TestServerService, TestSessionRuntime } from "./service.ts";
+//# sourceMappingURL=index.d.ts.map

package/dist/testing/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/testing/index.ts"],"names":[],"mappings":"AAAA,YAAY,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC/C,OAAO,EAAE,qBAAqB,EAAE,kBAAkB,EAAE,MAAM,aAAa,CAAC;AACxE,YAAY,EAAE,UAAU,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC;AACjE,OAAO,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAC/C,OAAO,EAAE,QAAQ,EAAE,UAAU,EAAE,iBAAiB,EAAE,kBAAkB,EAAE,MAAM,cAAc,CAAC","sourcesContent":["export type { WireChannel } from \"./client.ts\";\nexport { connectUnixTestClient, ProtocolTestClient } from \"./client.ts\";\nexport type { TestServer, TestServerOptions } from \"./server.ts\";\nexport { createTestServer } from \"./server.ts\";\nexport { Deferred, TEST_MODEL, TestServerService, TestSessionRuntime } from \"./service.ts\";\n"]}

package/dist/testing/index.js

@@ -0,0 +1,4 @@
+export { connectUnixTestClient, ProtocolTestClient } from "./client.js";
+export { createTestServer } from "./server.js";
+export { Deferred, TEST_MODEL, TestServerService, TestSessionRuntime } from "./service.js";
+//# sourceMappingURL=index.js.map

package/dist/testing/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/testing/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,qBAAqB,EAAE,kBAAkB,EAAE,MAAM,aAAa,CAAC;AAExE,OAAO,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAC/C,OAAO,EAAE,QAAQ,EAAE,UAAU,EAAE,iBAAiB,EAAE,kBAAkB,EAAE,MAAM,cAAc,CAAC","sourcesContent":["export type { WireChannel } from \"./client.ts\";\nexport { connectUnixTestClient, ProtocolTestClient } from \"./client.ts\";\nexport type { TestServer, TestServerOptions } from \"./server.ts\";\nexport { createTestServer } from \"./server.ts\";\nexport { Deferred, TEST_MODEL, TestServerService, TestSessionRuntime } from \"./service.ts\";\n"]}

package/dist/testing/server.d.ts

@@ -0,0 +1,12 @@
+import { PiServer } from "../server.ts";
+import type { PiServerOptions, PiServerService } from "../types.ts";
+export interface TestServerOptions extends PiServerOptions {
+    service?: PiServerService;
+}
+export interface TestServer {
+    server: PiServer;
+    service: PiServerService;
+}
+/** Create an unstarted PiServer with deterministic defaults for transport conformance tests. */
+export declare function createTestServer(options: TestServerOptions): TestServer;
+//# sourceMappingURL=server.d.ts.map

package/dist/testing/server.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"server.d.ts","sourceRoot":"","sources":["../../src/testing/server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,cAAc,CAAC;AACxC,OAAO,KAAK,EAAE,eAAe,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AAGpE,MAAM,WAAW,iBAAkB,SAAQ,eAAe;IACzD,OAAO,CAAC,EAAE,eAAe,CAAC;CAC1B;AAED,MAAM,WAAW,UAAU;IAC1B,MAAM,EAAE,QAAQ,CAAC;IACjB,OAAO,EAAE,eAAe,CAAC;CACzB;AAED,gGAAgG;AAChG,wBAAgB,gBAAgB,CAAC,OAAO,EAAE,iBAAiB,GAAG,UAAU,CAYvE","sourcesContent":["import { PiServer } from \"../server.ts\";\nimport type { PiServerOptions, PiServerService } from \"../types.ts\";\nimport { TestServerService } from \"./service.ts\";\n\nexport interface TestServerOptions extends PiServerOptions {\n\tservice?: PiServerService;\n}\n\nexport interface TestServer {\n\tserver: PiServer;\n\tservice: PiServerService;\n}\n\n/** Create an unstarted PiServer with deterministic defaults for transport conformance tests. */\nexport function createTestServer(options: TestServerOptions): TestServer {\n\tconst service = options.service ?? new TestServerService();\n\treturn {\n\t\tserver: new PiServer(service, {\n\t\t\tlisteners: options.listeners,\n\t\t\tmaxFrameLength: options.maxFrameLength,\n\t\t\thandshakeTimeoutMs: options.handshakeTimeoutMs,\n\t\t\tserverId: options.serverId,\n\t\t\tonError: options.onError,\n\t\t}),\n\t\tservice,\n\t};\n}\n"]}

package/dist/testing/server.js

@@ -0,0 +1,17 @@
+import { PiServer } from "../server.js";
+import { TestServerService } from "./service.js";
+/** Create an unstarted PiServer with deterministic defaults for transport conformance tests. */
+export function createTestServer(options) {
+    const service = options.service ?? new TestServerService();
+    return {
+        server: new PiServer(service, {
+            listeners: options.listeners,
+            maxFrameLength: options.maxFrameLength,
+            handshakeTimeoutMs: options.handshakeTimeoutMs,
+            serverId: options.serverId,
+            onError: options.onError,
+        }),
+        service,
+    };
+}
+//# sourceMappingURL=server.js.map

package/dist/testing/server.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"server.js","sourceRoot":"","sources":["../../src/testing/server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,cAAc,CAAC;AAExC,OAAO,EAAE,iBAAiB,EAAE,MAAM,cAAc,CAAC;AAWjD,gGAAgG;AAChG,MAAM,UAAU,gBAAgB,CAAC,OAA0B,EAAc;IACxE,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,IAAI,IAAI,iBAAiB,EAAE,CAAC;IAC3D,OAAO;QACN,MAAM,EAAE,IAAI,QAAQ,CAAC,OAAO,EAAE;YAC7B,SAAS,EAAE,OAAO,CAAC,SAAS;YAC5B,cAAc,EAAE,OAAO,CAAC,cAAc;YACtC,kBAAkB,EAAE,OAAO,CAAC,kBAAkB;YAC9C,QAAQ,EAAE,OAAO,CAAC,QAAQ;YAC1B,OAAO,EAAE,OAAO,CAAC,OAAO;SACxB,CAAC;QACF,OAAO;KACP,CAAC;AAAA,CACF","sourcesContent":["import { PiServer } from \"../server.ts\";\nimport type { PiServerOptions, PiServerService } from \"../types.ts\";\nimport { TestServerService } from \"./service.ts\";\n\nexport interface TestServerOptions extends PiServerOptions {\n\tservice?: PiServerService;\n}\n\nexport interface TestServer {\n\tserver: PiServer;\n\tservice: PiServerService;\n}\n\n/** Create an unstarted PiServer with deterministic defaults for transport conformance tests. */\nexport function createTestServer(options: TestServerOptions): TestServer {\n\tconst service = options.service ?? new TestServerService();\n\treturn {\n\t\tserver: new PiServer(service, {\n\t\t\tlisteners: options.listeners,\n\t\t\tmaxFrameLength: options.maxFrameLength,\n\t\t\thandshakeTimeoutMs: options.handshakeTimeoutMs,\n\t\t\tserverId: options.serverId,\n\t\t\tonError: options.onError,\n\t\t}),\n\t\tservice,\n\t};\n}\n"]}

package/dist/testing/service.d.ts

@@ -0,0 +1,59 @@
+import type { ModelMetadata, ModelRef, SessionMetadata, SessionPhase, SessionSnapshot, ThinkingLevel, TranscriptProgress } from "@earendil-works/pi-protocol";
+import { PiServerError } from "../errors.ts";
+import type { CreateSessionOptions, PiServerService, PiSessionRuntime, PiSessionRuntimeEvent, PromptInput } from "../types.ts";
+export declare const TEST_MODEL: ModelMetadata;
+export declare class Deferred<T> {
+    readonly promise: Promise<T>;
+    private resolvePromise;
+    constructor();
+    resolve(value: T): void;
+}
+interface StoredSession {
+    snapshot: SessionSnapshot;
+}
+export declare class TestSessionRuntime implements PiSessionRuntime {
+    readonly disposed: Deferred<void>;
+    disposeCount: number;
+    readonly steers: PromptInput[];
+    private readonly stored;
+    private readonly onDispose;
+    private readonly listeners;
+    private pendingPrompt?;
+    constructor(stored: StoredSession, onDispose: () => void);
+    snapshot(): SessionSnapshot;
+    getPhase(): SessionPhase;
+    prompt(input: PromptInput): Promise<void>;
+    steer(input: PromptInput): Promise<void>;
+    abort(): Promise<void>;
+    setModel(model: ModelRef): Promise<void>;
+    setThinking(thinkingLevel: ThinkingLevel): Promise<void>;
+    subscribe(listener: (event: PiSessionRuntimeEvent) => void): () => void;
+    dispose(): Promise<void>;
+    setPhase(phase: SessionPhase): void;
+    finishPrompt(): void;
+    emitProgress(progress: TranscriptProgress): void;
+    emitError(error: PiServerError): void;
+    emitSnapshot(): void;
+    private update;
+}
+interface ListDelay {
+    entered: Deferred<void>;
+    release: Deferred<void>;
+}
+export declare class TestServerService implements PiServerService {
+    readonly sessions: Map<string, StoredSession>;
+    readonly runtimes: Map<string, TestSessionRuntime[]>;
+    readonly locked: Set<string>;
+    lastCreatedId?: string;
+    private nextListDelay?;
+    listSessions(): Promise<SessionMetadata[]>;
+    listModels(): Promise<ModelMetadata[]>;
+    createSession(options: CreateSessionOptions): Promise<PiSessionRuntime>;
+    openSession(sessionId: string): Promise<PiSessionRuntime>;
+    seed(id?: string, name?: string, cwd?: string, model?: ModelRef, thinkingLevel?: ThinkingLevel): void;
+    delayNextList(): ListDelay;
+    latestRuntime(id: string): TestSessionRuntime;
+    private acquire;
+}
+export {};
+//# sourceMappingURL=service.d.ts.map

package/dist/testing/service.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"service.d.ts","sourceRoot":"","sources":["../../src/testing/service.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACX,aAAa,EACb,QAAQ,EACR,eAAe,EACf,YAAY,EACZ,eAAe,EACf,aAAa,EACb,kBAAkB,EAClB,MAAM,6BAA6B,CAAC;AACrC,OAAO,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAC7C,OAAO,KAAK,EACX,oBAAoB,EACpB,eAAe,EACf,gBAAgB,EAChB,qBAAqB,EACrB,WAAW,EACX,MAAM,aAAa,CAAC;AAErB,eAAO,MAAM,UAAU,EAAE,aAYxB,CAAC;AAEF,qBAAa,QAAQ,CAAC,CAAC;IACtB,QAAQ,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC;IAC7B,OAAO,CAAC,cAAc,CAAsB;IAE5C,cAIC;IAED,OAAO,CAAC,KAAK,EAAE,CAAC,GAAG,IAAI,CAEtB;CACD;AAED,UAAU,aAAa;IACtB,QAAQ,EAAE,eAAe,CAAC;CAC1B;AAED,qBAAa,kBAAmB,YAAW,gBAAgB;IAC1D,QAAQ,CAAC,QAAQ,iBAAwB;IACzC,YAAY,SAAK;IACjB,QAAQ,CAAC,MAAM,EAAE,WAAW,EAAE,CAAM;IACpC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAgB;IACvC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAa;IACvC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAqD;IAC/E,OAAO,CAAC,aAAa,CAAC,CAAiE;IAEvF,YAAY,MAAM,EAAE,aAAa,EAAE,SAAS,EAAE,MAAM,IAAI,EAGvD;IAED,QAAQ,IAAI,eAAe,CAE1B;IAED,QAAQ,IAAI,YAAY,CAEvB;IAEK,MAAM,CAAC,KAAK,EAAE,WAAW,GAAG,OAAO,CAAC,IAAI,CAAC,CA0C9C;IAEK,KAAK,CAAC,KAAK,EAAE,WAAW,GAAG,OAAO,CAAC,IAAI,CAAC,CAe7C;IAEK,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAG3B;IAEK,QAAQ,CAAC,KAAK,EAAE,QAAQ,GAAG,OAAO,CAAC,IAAI,CAAC,CAG7C;IAEK,WAAW,CAAC,aAAa,EAAE,aAAa,GAAG,OAAO,CAAC,IAAI,CAAC,CAG7D;IAED,SAAS,CAAC,QAAQ,EAAE,CAAC,KAAK,EAAE,qBAAqB,KAAK,IAAI,GAAG,MAAM,IAAI,CAGtE;IAEK,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC,CAI7B;IAED,QAAQ,CAAC,KAAK,EAAE,YAAY,GAAG,IAAI,CAElC;IAED,YAAY,IAAI,IAAI,CAGnB;IAED,YAAY,CAAC,QAAQ,EAAE,kBAAkB,GAAG,IAAI,CAE/C;IAED,SAAS,CAAC,KAAK,EAAE,aAAa,GAAG,IAAI,CAEpC;IAED,YAAY,IAAI,IAAI,CAEnB;IAED,OAAO,CAAC,MAAM;CASd;AAED,UAAU,SAAS;IAClB,OAAO,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC;IACxB,OAAO,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC;CACxB;AAED,qBAAa,iBAAkB,YAAW,eAAe;IACxD,QAAQ,CAAC,QAAQ,6BAAoC;IACrD,QAAQ,CAAC,QAAQ,oCAA2C;IAC5D,QAAQ,CAAC,MAAM,cAAqB;IACpC,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,OAAO,CAAC,aAAa,CAAC,CAAY;IAE5B,YAAY,IAAI,OAAO,CAAC,eAAe,EAAE,CAAC,CAc/C;IAEK,UAAU,IAAI,OAAO,CAAC,aAAa,EAAE,CAAC,CAE3C;IAEK,aAAa,CAAC,OAAO,EAAE,oBAAoB,GAAG,OAAO,CAAC,gBAAgB,CAAC,CAK5E;IAEK,WAAW,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,CAAC,CAI9D;IAED,IAAI,CACH,EAAE,SAAc,EAChB,IAAI,SAAkB,EACtB,GAAG,SAA+B,EAClC,KAAK,GAAE,QAA+D,EACtE,aAAa,GAAE,aAAqB,GAClC,IAAI,CAmBN;IAED,aAAa,IAAI,SAAS,CAIzB;IAED,aAAa,CAAC,EAAE,EAAE,MAAM,GAAG,kBAAkB,CAI5C;IAED,OAAO,CAAC,OAAO;CAUf","sourcesContent":["import type {\n\tModelMetadata,\n\tModelRef,\n\tSessionMetadata,\n\tSessionPhase,\n\tSessionSnapshot,\n\tThinkingLevel,\n\tTranscriptProgress,\n} from \"@earendil-works/pi-protocol\";\nimport { PiServerError } from \"../errors.ts\";\nimport type {\n\tCreateSessionOptions,\n\tPiServerService,\n\tPiSessionRuntime,\n\tPiSessionRuntimeEvent,\n\tPromptInput,\n} from \"../types.ts\";\n\nexport const TEST_MODEL: ModelMetadata = {\n\tprovider: \"test\",\n\tid: \"small\",\n\tname: \"Test Small\",\n\tapi: \"test-api\",\n\treasoning: true,\n\tinput: [\"text\", \"image\"],\n\tcontextWindow: 16_000,\n\tmaxTokens: 2_000,\n\tcost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },\n\tsupportedThinkingLevels: [\"off\", \"medium\", \"high\"],\n\tauthenticated: true,\n};\n\nexport class Deferred<T> {\n\treadonly promise: Promise<T>;\n\tprivate resolvePromise!: (value: T) => void;\n\n\tconstructor() {\n\t\tthis.promise = new Promise<T>((resolve) => {\n\t\t\tthis.resolvePromise = resolve;\n\t\t});\n\t}\n\n\tresolve(value: T): void {\n\t\tthis.resolvePromise(value);\n\t}\n}\n\ninterface StoredSession {\n\tsnapshot: SessionSnapshot;\n}\n\nexport class TestSessionRuntime implements PiSessionRuntime {\n\treadonly disposed = new Deferred<void>();\n\tdisposeCount = 0;\n\treadonly steers: PromptInput[] = [];\n\tprivate readonly stored: StoredSession;\n\tprivate readonly onDispose: () => void;\n\tprivate readonly listeners = new Set<(event: PiSessionRuntimeEvent) => void>();\n\tprivate pendingPrompt?: { input: PromptInput; done: Deferred<\"complete\" | \"aborted\"> };\n\n\tconstructor(stored: StoredSession, onDispose: () => void) {\n\t\tthis.stored = stored;\n\t\tthis.onDispose = onDispose;\n\t}\n\n\tsnapshot(): SessionSnapshot {\n\t\treturn structuredClone(this.stored.snapshot);\n\t}\n\n\tgetPhase(): SessionPhase {\n\t\treturn this.stored.snapshot.phase;\n\t}\n\n\tasync prompt(input: PromptInput): Promise<void> {\n\t\tif (this.getPhase() !== \"idle\") throw new PiServerError(\"busy\", \"A prompt is already running\");\n\t\tconst done = new Deferred<\"complete\" | \"aborted\">();\n\t\tthis.pendingPrompt = { input, done };\n\t\tthis.update({\n\t\t\tphase: \"turn\",\n\t\t\ttranscript: [\n\t\t\t\t...this.stored.snapshot.transcript,\n\t\t\t\t{\n\t\t\t\t\tid: `user-${this.stored.snapshot.revision + 1}`,\n\t\t\t\t\trole: \"user\",\n\t\t\t\t\tcontent: [{ type: \"text\", text: input.text }],\n\t\t\t\t\ttimestamp: this.stored.snapshot.revision + 1,\n\t\t\t\t},\n\t\t\t],\n\t\t});\n\t\tconst outcome = await done.promise;\n\t\tconst assistant =\n\t\t\toutcome === \"complete\"\n\t\t\t\t? {\n\t\t\t\t\t\tid: `assistant-${this.stored.snapshot.revision + 1}`,\n\t\t\t\t\t\trole: \"assistant\" as const,\n\t\t\t\t\t\tcontent: [{ type: \"text\" as const, text: `reply:${input.text}` }],\n\t\t\t\t\t\tstatus: \"complete\" as const,\n\t\t\t\t\t\tmodel: this.stored.snapshot.model,\n\t\t\t\t\t\tstopReason: \"stop\" as const,\n\t\t\t\t\t\ttimestamp: this.stored.snapshot.revision + 1,\n\t\t\t\t\t}\n\t\t\t\t: {\n\t\t\t\t\t\tid: `assistant-${this.stored.snapshot.revision + 1}`,\n\t\t\t\t\t\trole: \"assistant\" as const,\n\t\t\t\t\t\tcontent: [{ type: \"text\" as const, text: \"\" }],\n\t\t\t\t\t\tstatus: \"aborted\" as const,\n\t\t\t\t\t\tmodel: this.stored.snapshot.model,\n\t\t\t\t\t\tstopReason: \"aborted\" as const,\n\t\t\t\t\t\ttimestamp: this.stored.snapshot.revision + 1,\n\t\t\t\t\t};\n\t\tthis.update({\n\t\t\tphase: \"idle\",\n\t\t\ttranscript: [...this.stored.snapshot.transcript, assistant],\n\t\t});\n\t\tthis.pendingPrompt = undefined;\n\t}\n\n\tasync steer(input: PromptInput): Promise<void> {\n\t\tif (this.getPhase() === \"idle\") throw new PiServerError(\"busy\", \"There is no active prompt to steer\");\n\t\tthis.steers.push(input);\n\t\tthis.update({\n\t\t\tqueuedSteerCount: this.stored.snapshot.queuedSteerCount + 1,\n\t\t\tqueuedSteer: [\n\t\t\t\t...this.stored.snapshot.queuedSteer,\n\t\t\t\t{\n\t\t\t\t\tid: `steer-${this.stored.snapshot.revision + 1}`,\n\t\t\t\t\trole: \"user\",\n\t\t\t\t\tcontent: [{ type: \"text\", text: input.text }],\n\t\t\t\t\ttimestamp: this.stored.snapshot.revision + 1,\n\t\t\t\t},\n\t\t\t],\n\t\t});\n\t}\n\n\tasync abort(): Promise<void> {\n\t\tif (!this.pendingPrompt) throw new PiServerError(\"busy\", \"There is no active prompt to abort\");\n\t\tthis.pendingPrompt.done.resolve(\"aborted\");\n\t}\n\n\tasync setModel(model: ModelRef): Promise<void> {\n\t\tif (this.getPhase() !== \"idle\") throw new PiServerError(\"busy\", \"Session is busy\");\n\t\tthis.update({ model });\n\t}\n\n\tasync setThinking(thinkingLevel: ThinkingLevel): Promise<void> {\n\t\tif (this.getPhase() !== \"idle\") throw new PiServerError(\"busy\", \"Session is busy\");\n\t\tthis.update({ thinkingLevel });\n\t}\n\n\tsubscribe(listener: (event: PiSessionRuntimeEvent) => void): () => void {\n\t\tthis.listeners.add(listener);\n\t\treturn () => this.listeners.delete(listener);\n\t}\n\n\tasync dispose(): Promise<void> {\n\t\tthis.disposeCount += 1;\n\t\tthis.onDispose();\n\t\tthis.disposed.resolve(undefined);\n\t}\n\n\tsetPhase(phase: SessionPhase): void {\n\t\tthis.stored.snapshot = { ...this.stored.snapshot, phase };\n\t}\n\n\tfinishPrompt(): void {\n\t\tif (!this.pendingPrompt) throw new Error(\"No prompt is pending\");\n\t\tthis.pendingPrompt.done.resolve(\"complete\");\n\t}\n\n\temitProgress(progress: TranscriptProgress): void {\n\t\tfor (const listener of this.listeners) listener({ type: \"progress\", progress });\n\t}\n\n\temitError(error: PiServerError): void {\n\t\tfor (const listener of this.listeners) listener({ type: \"error\", error });\n\t}\n\n\temitSnapshot(): void {\n\t\tfor (const listener of this.listeners) listener({ type: \"snapshot\" });\n\t}\n\n\tprivate update(updates: Partial<SessionSnapshot>): void {\n\t\tthis.stored.snapshot = {\n\t\t\t...this.stored.snapshot,\n\t\t\t...updates,\n\t\t\trevision: this.stored.snapshot.revision + 1,\n\t\t\tupdatedAt: this.stored.snapshot.updatedAt + 1,\n\t\t};\n\t\tthis.emitSnapshot();\n\t}\n}\n\ninterface ListDelay {\n\tentered: Deferred<void>;\n\trelease: Deferred<void>;\n}\n\nexport class TestServerService implements PiServerService {\n\treadonly sessions = new Map<string, StoredSession>();\n\treadonly runtimes = new Map<string, TestSessionRuntime[]>();\n\treadonly locked = new Set<string>();\n\tlastCreatedId?: string;\n\tprivate nextListDelay?: ListDelay;\n\n\tasync listSessions(): Promise<SessionMetadata[]> {\n\t\tconst delay = this.nextListDelay;\n\t\tif (delay) {\n\t\t\tthis.nextListDelay = undefined;\n\t\t\tdelay.entered.resolve(undefined);\n\t\t\tawait delay.release.promise;\n\t\t}\n\t\treturn [...this.sessions.values()].map(({ snapshot }) => ({\n\t\t\tid: snapshot.id,\n\t\t\tcreatedAt: snapshot.createdAt,\n\t\t\tupdatedAt: snapshot.updatedAt,\n\t\t\tsessionName: snapshot.name,\n\t\t\tcwd: snapshot.cwd,\n\t\t}));\n\t}\n\n\tasync listModels(): Promise<ModelMetadata[]> {\n\t\treturn [TEST_MODEL];\n\t}\n\n\tasync createSession(options: CreateSessionOptions): Promise<PiSessionRuntime> {\n\t\tthis.lastCreatedId = options.id;\n\t\tif (this.sessions.has(options.id)) throw new PiServerError(\"session_locked\", \"Session already exists\");\n\t\tthis.seed(options.id, options.name, options.cwd, options.model, options.thinkingLevel);\n\t\treturn this.acquire(options.id);\n\t}\n\n\tasync openSession(sessionId: string): Promise<PiSessionRuntime> {\n\t\tif (!this.sessions.has(sessionId)) throw new PiServerError(\"not_found\", `Unknown session: ${sessionId}`);\n\t\tif (this.locked.has(sessionId)) throw new PiServerError(\"session_locked\", `Session is locked: ${sessionId}`);\n\t\treturn this.acquire(sessionId);\n\t}\n\n\tseed(\n\t\tid = \"session-1\",\n\t\tname = `Session ${id}`,\n\t\tcwd = \"/tmp/pi-server-conformance\",\n\t\tmodel: ModelRef = { provider: TEST_MODEL.provider, id: TEST_MODEL.id },\n\t\tthinkingLevel: ThinkingLevel = \"off\",\n\t): void {\n\t\tthis.sessions.set(id, {\n\t\t\tsnapshot: {\n\t\t\t\tid,\n\t\t\t\tname,\n\t\t\t\tcwd,\n\t\t\t\tcreatedAt: 1,\n\t\t\t\tupdatedAt: 1,\n\t\t\t\tphase: \"idle\",\n\t\t\t\tmodel,\n\t\t\t\tthinkingLevel,\n\t\t\t\tattached: false,\n\t\t\t\tlocked: false,\n\t\t\t\trevision: 0,\n\t\t\t\ttranscript: [],\n\t\t\t\tqueuedSteer: [],\n\t\t\t\tqueuedSteerCount: 0,\n\t\t\t},\n\t\t});\n\t}\n\n\tdelayNextList(): ListDelay {\n\t\tconst delay = { entered: new Deferred<void>(), release: new Deferred<void>() };\n\t\tthis.nextListDelay = delay;\n\t\treturn delay;\n\t}\n\n\tlatestRuntime(id: string): TestSessionRuntime {\n\t\tconst runtimes = this.runtimes.get(id);\n\t\tif (!runtimes?.length) throw new Error(`No runtime for ${id}`);\n\t\treturn runtimes.at(-1)!;\n\t}\n\n\tprivate acquire(id: string): TestSessionRuntime {\n\t\tconst stored = this.sessions.get(id);\n\t\tif (!stored) throw new Error(`Unknown session: ${id}`);\n\t\tthis.locked.add(id);\n\t\tconst runtime = new TestSessionRuntime(stored, () => this.locked.delete(id));\n\t\tconst runtimes = this.runtimes.get(id) ?? [];\n\t\truntimes.push(runtime);\n\t\tthis.runtimes.set(id, runtimes);\n\t\treturn runtime;\n\t}\n}\n"]}

package/dist/testing/service.js

@@ -0,0 +1,240 @@
+import { PiServerError } from "../errors.js";
+export const TEST_MODEL = {
+    provider: "test",
+    id: "small",
+    name: "Test Small",
+    api: "test-api",
+    reasoning: true,
+    input: ["text", "image"],
+    contextWindow: 16_000,
+    maxTokens: 2_000,
+    cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
+    supportedThinkingLevels: ["off", "medium", "high"],
+    authenticated: true,
+};
+export class Deferred {
+    promise;
+    resolvePromise;
+    constructor() {
+        this.promise = new Promise((resolve) => {
+            this.resolvePromise = resolve;
+        });
+    }
+    resolve(value) {
+        this.resolvePromise(value);
+    }
+}
+export class TestSessionRuntime {
+    disposed = new Deferred();
+    disposeCount = 0;
+    steers = [];
+    stored;
+    onDispose;
+    listeners = new Set();
+    pendingPrompt;
+    constructor(stored, onDispose) {
+        this.stored = stored;
+        this.onDispose = onDispose;
+    }
+    snapshot() {
+        return structuredClone(this.stored.snapshot);
+    }
+    getPhase() {
+        return this.stored.snapshot.phase;
+    }
+    async prompt(input) {
+        if (this.getPhase() !== "idle")
+            throw new PiServerError("busy", "A prompt is already running");
+        const done = new Deferred();
+        this.pendingPrompt = { input, done };
+        this.update({
+            phase: "turn",
+            transcript: [
+                ...this.stored.snapshot.transcript,
+                {
+                    id: `user-${this.stored.snapshot.revision + 1}`,
+                    role: "user",
+                    content: [{ type: "text", text: input.text }],
+                    timestamp: this.stored.snapshot.revision + 1,
+                },
+            ],
+        });
+        const outcome = await done.promise;
+        const assistant = outcome === "complete"
+            ? {
+                id: `assistant-${this.stored.snapshot.revision + 1}`,
+                role: "assistant",
+                content: [{ type: "text", text: `reply:${input.text}` }],
+                status: "complete",
+                model: this.stored.snapshot.model,
+                stopReason: "stop",
+                timestamp: this.stored.snapshot.revision + 1,
+            }
+            : {
+                id: `assistant-${this.stored.snapshot.revision + 1}`,
+                role: "assistant",
+                content: [{ type: "text", text: "" }],
+                status: "aborted",
+                model: this.stored.snapshot.model,
+                stopReason: "aborted",
+                timestamp: this.stored.snapshot.revision + 1,
+            };
+        this.update({
+            phase: "idle",
+            transcript: [...this.stored.snapshot.transcript, assistant],
+        });
+        this.pendingPrompt = undefined;
+    }
+    async steer(input) {
+        if (this.getPhase() === "idle")
+            throw new PiServerError("busy", "There is no active prompt to steer");
+        this.steers.push(input);
+        this.update({
+            queuedSteerCount: this.stored.snapshot.queuedSteerCount + 1,
+            queuedSteer: [
+                ...this.stored.snapshot.queuedSteer,
+                {
+                    id: `steer-${this.stored.snapshot.revision + 1}`,
+                    role: "user",
+                    content: [{ type: "text", text: input.text }],
+                    timestamp: this.stored.snapshot.revision + 1,
+                },
+            ],
+        });
+    }
+    async abort() {
+        if (!this.pendingPrompt)
+            throw new PiServerError("busy", "There is no active prompt to abort");
+        this.pendingPrompt.done.resolve("aborted");
+    }
+    async setModel(model) {
+        if (this.getPhase() !== "idle")
+            throw new PiServerError("busy", "Session is busy");
+        this.update({ model });
+    }
+    async setThinking(thinkingLevel) {
+        if (this.getPhase() !== "idle")
+            throw new PiServerError("busy", "Session is busy");
+        this.update({ thinkingLevel });
+    }
+    subscribe(listener) {
+        this.listeners.add(listener);
+        return () => this.listeners.delete(listener);
+    }
+    async dispose() {
+        this.disposeCount += 1;
+        this.onDispose();
+        this.disposed.resolve(undefined);
+    }
+    setPhase(phase) {
+        this.stored.snapshot = { ...this.stored.snapshot, phase };
+    }
+    finishPrompt() {
+        if (!this.pendingPrompt)
+            throw new Error("No prompt is pending");
+        this.pendingPrompt.done.resolve("complete");
+    }
+    emitProgress(progress) {
+        for (const listener of this.listeners)
+            listener({ type: "progress", progress });
+    }
+    emitError(error) {
+        for (const listener of this.listeners)
+            listener({ type: "error", error });
+    }
+    emitSnapshot() {
+        for (const listener of this.listeners)
+            listener({ type: "snapshot" });
+    }
+    update(updates) {
+        this.stored.snapshot = {
+            ...this.stored.snapshot,
+            ...updates,
+            revision: this.stored.snapshot.revision + 1,
+            updatedAt: this.stored.snapshot.updatedAt + 1,
+        };
+        this.emitSnapshot();
+    }
+}
+export class TestServerService {
+    sessions = new Map();
+    runtimes = new Map();
+    locked = new Set();
+    lastCreatedId;
+    nextListDelay;
+    async listSessions() {
+        const delay = this.nextListDelay;
+        if (delay) {
+            this.nextListDelay = undefined;
+            delay.entered.resolve(undefined);
+            await delay.release.promise;
+        }
+        return [...this.sessions.values()].map(({ snapshot }) => ({
+            id: snapshot.id,
+            createdAt: snapshot.createdAt,
+            updatedAt: snapshot.updatedAt,
+            sessionName: snapshot.name,
+            cwd: snapshot.cwd,
+        }));
+    }
+    async listModels() {
+        return [TEST_MODEL];
+    }
+    async createSession(options) {
+        this.lastCreatedId = options.id;
+        if (this.sessions.has(options.id))
+            throw new PiServerError("session_locked", "Session already exists");
+        this.seed(options.id, options.name, options.cwd, options.model, options.thinkingLevel);
+        return this.acquire(options.id);
+    }
+    async openSession(sessionId) {
+        if (!this.sessions.has(sessionId))
+            throw new PiServerError("not_found", `Unknown session: ${sessionId}`);
+        if (this.locked.has(sessionId))
+            throw new PiServerError("session_locked", `Session is locked: ${sessionId}`);
+        return this.acquire(sessionId);
+    }
+    seed(id = "session-1", name = `Session ${id}`, cwd = "/tmp/pi-server-conformance", model = { provider: TEST_MODEL.provider, id: TEST_MODEL.id }, thinkingLevel = "off") {
+        this.sessions.set(id, {
+            snapshot: {
+                id,
+                name,
+                cwd,
+                createdAt: 1,
+                updatedAt: 1,
+                phase: "idle",
+                model,
+                thinkingLevel,
+                attached: false,
+                locked: false,
+                revision: 0,
+                transcript: [],
+                queuedSteer: [],
+                queuedSteerCount: 0,
+            },
+        });
+    }
+    delayNextList() {
+        const delay = { entered: new Deferred(), release: new Deferred() };
+        this.nextListDelay = delay;
+        return delay;
+    }
+    latestRuntime(id) {
+        const runtimes = this.runtimes.get(id);
+        if (!runtimes?.length)
+            throw new Error(`No runtime for ${id}`);
+        return runtimes.at(-1);
+    }
+    acquire(id) {
+        const stored = this.sessions.get(id);
+        if (!stored)
+            throw new Error(`Unknown session: ${id}`);
+        this.locked.add(id);
+        const runtime = new TestSessionRuntime(stored, () => this.locked.delete(id));
+        const runtimes = this.runtimes.get(id) ?? [];
+        runtimes.push(runtime);
+        this.runtimes.set(id, runtimes);
+        return runtime;
+    }
+}
+//# sourceMappingURL=service.js.map

package/dist/testing/service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"service.js","sourceRoot":"","sources":["../../src/testing/service.ts"],"names":[],"mappings":"AASA,OAAO,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAS7C,MAAM,CAAC,MAAM,UAAU,GAAkB;IACxC,QAAQ,EAAE,MAAM;IAChB,EAAE,EAAE,OAAO;IACX,IAAI,EAAE,YAAY;IAClB,GAAG,EAAE,UAAU;IACf,SAAS,EAAE,IAAI;IACf,KAAK,EAAE,CAAC,MAAM,EAAE,OAAO,CAAC;IACxB,aAAa,EAAE,MAAM;IACrB,SAAS,EAAE,KAAK;IAChB,IAAI,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE;IAC1D,uBAAuB,EAAE,CAAC,KAAK,EAAE,QAAQ,EAAE,MAAM,CAAC;IAClD,aAAa,EAAE,IAAI;CACnB,CAAC;AAEF,MAAM,OAAO,QAAQ;IACX,OAAO,CAAa;IACrB,cAAc,CAAsB;IAE5C,cAAc;QACb,IAAI,CAAC,OAAO,GAAG,IAAI,OAAO,CAAI,CAAC,OAAO,EAAE,EAAE,CAAC;YAC1C,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC;QAAA,CAC9B,CAAC,CAAC;IAAA,CACH;IAED,OAAO,CAAC,KAAQ,EAAQ;QACvB,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;IAAA,CAC3B;CACD;AAMD,MAAM,OAAO,kBAAkB;IACrB,QAAQ,GAAG,IAAI,QAAQ,EAAQ,CAAC;IACzC,YAAY,GAAG,CAAC,CAAC;IACR,MAAM,GAAkB,EAAE,CAAC;IACnB,MAAM,CAAgB;IACtB,SAAS,CAAa;IACtB,SAAS,GAAG,IAAI,GAAG,EAA0C,CAAC;IACvE,aAAa,CAAkE;IAEvF,YAAY,MAAqB,EAAE,SAAqB,EAAE;QACzD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAAA,CAC3B;IAED,QAAQ,GAAoB;QAC3B,OAAO,eAAe,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IAAA,CAC7C;IAED,QAAQ,GAAiB;QACxB,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC;IAAA,CAClC;IAED,KAAK,CAAC,MAAM,CAAC,KAAkB,EAAiB;QAC/C,IAAI,IAAI,CAAC,QAAQ,EAAE,KAAK,MAAM;YAAE,MAAM,IAAI,aAAa,CAAC,MAAM,EAAE,6BAA6B,CAAC,CAAC;QAC/F,MAAM,IAAI,GAAG,IAAI,QAAQ,EAA0B,CAAC;QACpD,IAAI,CAAC,aAAa,GAAG,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;QACrC,IAAI,CAAC,MAAM,CAAC;YACX,KAAK,EAAE,MAAM;YACb,UAAU,EAAE;gBACX,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,UAAU;gBAClC;oBACC,EAAE,EAAE,QAAQ,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,GAAG,CAAC,EAAE;oBAC/C,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC;oBAC7C,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,GAAG,CAAC;iBAC5C;aACD;SACD,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC;QACnC,MAAM,SAAS,GACd,OAAO,KAAK,UAAU;YACrB,CAAC,CAAC;gBACA,EAAE,EAAE,aAAa,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,GAAG,CAAC,EAAE;gBACpD,IAAI,EAAE,WAAoB;gBAC1B,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,SAAS,KAAK,CAAC,IAAI,EAAE,EAAE,CAAC;gBACjE,MAAM,EAAE,UAAmB;gBAC3B,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK;gBACjC,UAAU,EAAE,MAAe;gBAC3B,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,GAAG,CAAC;aAC5C;YACF,CAAC,CAAC;gBACA,EAAE,EAAE,aAAa,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,GAAG,CAAC,EAAE;gBACpD,IAAI,EAAE,WAAoB;gBAC1B,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;gBAC9C,MAAM,EAAE,SAAkB;gBAC1B,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK;gBACjC,UAAU,EAAE,SAAkB;gBAC9B,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,GAAG,CAAC;aAC5C,CAAC;QACL,IAAI,CAAC,MAAM,CAAC;YACX,KAAK,EAAE,MAAM;YACb,UAAU,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,UAAU,EAAE,SAAS,CAAC;SAC3D,CAAC,CAAC;QACH,IAAI,CAAC,aAAa,GAAG,SAAS,CAAC;IAAA,CAC/B;IAED,KAAK,CAAC,KAAK,CAAC,KAAkB,EAAiB;QAC9C,IAAI,IAAI,CAAC,QAAQ,EAAE,KAAK,MAAM;YAAE,MAAM,IAAI,aAAa,CAAC,MAAM,EAAE,oCAAoC,CAAC,CAAC;QACtG,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACxB,IAAI,CAAC,MAAM,CAAC;YACX,gBAAgB,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,gBAAgB,GAAG,CAAC;YAC3D,WAAW,EAAE;gBACZ,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,WAAW;gBACnC;oBACC,EAAE,EAAE,SAAS,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,GAAG,CAAC,EAAE;oBAChD,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC;oBAC7C,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,GAAG,CAAC;iBAC5C;aACD;SACD,CAAC,CAAC;IAAA,CACH;IAED,KAAK,CAAC,KAAK,GAAkB;QAC5B,IAAI,CAAC,IAAI,CAAC,aAAa;YAAE,MAAM,IAAI,aAAa,CAAC,MAAM,EAAE,oCAAoC,CAAC,CAAC;QAC/F,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IAAA,CAC3C;IAED,KAAK,CAAC,QAAQ,CAAC,KAAe,EAAiB;QAC9C,IAAI,IAAI,CAAC,QAAQ,EAAE,KAAK,MAAM;YAAE,MAAM,IAAI,aAAa,CAAC,MAAM,EAAE,iBAAiB,CAAC,CAAC;QACnF,IAAI,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;IAAA,CACvB;IAED,KAAK,CAAC,WAAW,CAAC,aAA4B,EAAiB;QAC9D,IAAI,IAAI,CAAC,QAAQ,EAAE,KAAK,MAAM;YAAE,MAAM,IAAI,aAAa,CAAC,MAAM,EAAE,iBAAiB,CAAC,CAAC;QACnF,IAAI,CAAC,MAAM,CAAC,EAAE,aAAa,EAAE,CAAC,CAAC;IAAA,CAC/B;IAED,SAAS,CAAC,QAAgD,EAAc;QACvE,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC7B,OAAO,GAAG,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IAAA,CAC7C;IAED,KAAK,CAAC,OAAO,GAAkB;QAC9B,IAAI,CAAC,YAAY,IAAI,CAAC,CAAC;QACvB,IAAI,CAAC,SAAS,EAAE,CAAC;QACjB,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IAAA,CACjC;IAED,QAAQ,CAAC,KAAmB,EAAQ;QACnC,IAAI,CAAC,MAAM,CAAC,QAAQ,GAAG,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,KAAK,EAAE,CAAC;IAAA,CAC1D;IAED,YAAY,GAAS;QACpB,IAAI,CAAC,IAAI,CAAC,aAAa;YAAE,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;QACjE,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;IAAA,CAC5C;IAED,YAAY,CAAC,QAA4B,EAAQ;QAChD,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,SAAS;YAAE,QAAQ,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,CAAC,CAAC;IAAA,CAChF;IAED,SAAS,CAAC,KAAoB,EAAQ;QACrC,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,SAAS;YAAE,QAAQ,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;IAAA,CAC1E;IAED,YAAY,GAAS;QACpB,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,SAAS;YAAE,QAAQ,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,CAAC;IAAA,CACtE;IAEO,MAAM,CAAC,OAAiC,EAAQ;QACvD,IAAI,CAAC,MAAM,CAAC,QAAQ,GAAG;YACtB,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ;YACvB,GAAG,OAAO;YACV,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,GAAG,CAAC;YAC3C,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,SAAS,GAAG,CAAC;SAC7C,CAAC;QACF,IAAI,CAAC,YAAY,EAAE,CAAC;IAAA,CACpB;CACD;AAOD,MAAM,OAAO,iBAAiB;IACpB,QAAQ,GAAG,IAAI,GAAG,EAAyB,CAAC;IAC5C,QAAQ,GAAG,IAAI,GAAG,EAAgC,CAAC;IACnD,MAAM,GAAG,IAAI,GAAG,EAAU,CAAC;IACpC,aAAa,CAAU;IACf,aAAa,CAAa;IAElC,KAAK,CAAC,YAAY,GAA+B;QAChD,MAAM,KAAK,GAAG,IAAI,CAAC,aAAa,CAAC;QACjC,IAAI,KAAK,EAAE,CAAC;YACX,IAAI,CAAC,aAAa,GAAG,SAAS,CAAC;YAC/B,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;YACjC,MAAM,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC;QAC7B,CAAC;QACD,OAAO,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC;YACzD,EAAE,EAAE,QAAQ,CAAC,EAAE;YACf,SAAS,EAAE,QAAQ,CAAC,SAAS;YAC7B,SAAS,EAAE,QAAQ,CAAC,SAAS;YAC7B,WAAW,EAAE,QAAQ,CAAC,IAAI;YAC1B,GAAG,EAAE,QAAQ,CAAC,GAAG;SACjB,CAAC,CAAC,CAAC;IAAA,CACJ;IAED,KAAK,CAAC,UAAU,GAA6B;QAC5C,OAAO,CAAC,UAAU,CAAC,CAAC;IAAA,CACpB;IAED,KAAK,CAAC,aAAa,CAAC,OAA6B,EAA6B;QAC7E,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC,EAAE,CAAC;QAChC,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;YAAE,MAAM,IAAI,aAAa,CAAC,gBAAgB,EAAE,wBAAwB,CAAC,CAAC;QACvG,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,GAAG,EAAE,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC,aAAa,CAAC,CAAC;QACvF,OAAO,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAAA,CAChC;IAED,KAAK,CAAC,WAAW,CAAC,SAAiB,EAA6B;QAC/D,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC;YAAE,MAAM,IAAI,aAAa,CAAC,WAAW,EAAE,oBAAoB,SAAS,EAAE,CAAC,CAAC;QACzG,IAAI,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC;YAAE,MAAM,IAAI,aAAa,CAAC,gBAAgB,EAAE,sBAAsB,SAAS,EAAE,CAAC,CAAC;QAC7G,OAAO,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IAAA,CAC/B;IAED,IAAI,CACH,EAAE,GAAG,WAAW,EAChB,IAAI,GAAG,WAAW,EAAE,EAAE,EACtB,GAAG,GAAG,4BAA4B,EAClC,KAAK,GAAa,EAAE,QAAQ,EAAE,UAAU,CAAC,QAAQ,EAAE,EAAE,EAAE,UAAU,CAAC,EAAE,EAAE,EACtE,aAAa,GAAkB,KAAK,EAC7B;QACP,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,EAAE;YACrB,QAAQ,EAAE;gBACT,EAAE;gBACF,IAAI;gBACJ,GAAG;gBACH,SAAS,EAAE,CAAC;gBACZ,SAAS,EAAE,CAAC;gBACZ,KAAK,EAAE,MAAM;gBACb,KAAK;gBACL,aAAa;gBACb,QAAQ,EAAE,KAAK;gBACf,MAAM,EAAE,KAAK;gBACb,QAAQ,EAAE,CAAC;gBACX,UAAU,EAAE,EAAE;gBACd,WAAW,EAAE,EAAE;gBACf,gBAAgB,EAAE,CAAC;aACnB;SACD,CAAC,CAAC;IAAA,CACH;IAED,aAAa,GAAc;QAC1B,MAAM,KAAK,GAAG,EAAE,OAAO,EAAE,IAAI,QAAQ,EAAQ,EAAE,OAAO,EAAE,IAAI,QAAQ,EAAQ,EAAE,CAAC;QAC/E,IAAI,CAAC,aAAa,GAAG,KAAK,CAAC;QAC3B,OAAO,KAAK,CAAC;IAAA,CACb;IAED,aAAa,CAAC,EAAU,EAAsB;QAC7C,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACvC,IAAI,CAAC,QAAQ,EAAE,MAAM;YAAE,MAAM,IAAI,KAAK,CAAC,kBAAkB,EAAE,EAAE,CAAC,CAAC;QAC/D,OAAO,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,CAAE,CAAC;IAAA,CACxB;IAEO,OAAO,CAAC,EAAU,EAAsB;QAC/C,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACrC,IAAI,CAAC,MAAM;YAAE,MAAM,IAAI,KAAK,CAAC,oBAAoB,EAAE,EAAE,CAAC,CAAC;QACvD,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACpB,MAAM,OAAO,GAAG,IAAI,kBAAkB,CAAC,MAAM,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;QAC7E,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC;QAC7C,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACvB,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,EAAE,QAAQ,CAAC,CAAC;QAChC,OAAO,OAAO,CAAC;IAAA,CACf;CACD","sourcesContent":["import type {\n\tModelMetadata,\n\tModelRef,\n\tSessionMetadata,\n\tSessionPhase,\n\tSessionSnapshot,\n\tThinkingLevel,\n\tTranscriptProgress,\n} from \"@earendil-works/pi-protocol\";\nimport { PiServerError } from \"../errors.ts\";\nimport type {\n\tCreateSessionOptions,\n\tPiServerService,\n\tPiSessionRuntime,\n\tPiSessionRuntimeEvent,\n\tPromptInput,\n} from \"../types.ts\";\n\nexport const TEST_MODEL: ModelMetadata = {\n\tprovider: \"test\",\n\tid: \"small\",\n\tname: \"Test Small\",\n\tapi: \"test-api\",\n\treasoning: true,\n\tinput: [\"text\", \"image\"],\n\tcontextWindow: 16_000,\n\tmaxTokens: 2_000,\n\tcost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },\n\tsupportedThinkingLevels: [\"off\", \"medium\", \"high\"],\n\tauthenticated: true,\n};\n\nexport class Deferred<T> {\n\treadonly promise: Promise<T>;\n\tprivate resolvePromise!: (value: T) => void;\n\n\tconstructor() {\n\t\tthis.promise = new Promise<T>((resolve) => {\n\t\t\tthis.resolvePromise = resolve;\n\t\t});\n\t}\n\n\tresolve(value: T): void {\n\t\tthis.resolvePromise(value);\n\t}\n}\n\ninterface StoredSession {\n\tsnapshot: SessionSnapshot;\n}\n\nexport class TestSessionRuntime implements PiSessionRuntime {\n\treadonly disposed = new Deferred<void>();\n\tdisposeCount = 0;\n\treadonly steers: PromptInput[] = [];\n\tprivate readonly stored: StoredSession;\n\tprivate readonly onDispose: () => void;\n\tprivate readonly listeners = new Set<(event: PiSessionRuntimeEvent) => void>();\n\tprivate pendingPrompt?: { input: PromptInput; done: Deferred<\"complete\" | \"aborted\"> };\n\n\tconstructor(stored: StoredSession, onDispose: () => void) {\n\t\tthis.stored = stored;\n\t\tthis.onDispose = onDispose;\n\t}\n\n\tsnapshot(): SessionSnapshot {\n\t\treturn structuredClone(this.stored.snapshot);\n\t}\n\n\tgetPhase(): SessionPhase {\n\t\treturn this.stored.snapshot.phase;\n\t}\n\n\tasync prompt(input: PromptInput): Promise<void> {\n\t\tif (this.getPhase() !== \"idle\") throw new PiServerError(\"busy\", \"A prompt is already running\");\n\t\tconst done = new Deferred<\"complete\" | \"aborted\">();\n\t\tthis.pendingPrompt = { input, done };\n\t\tthis.update({\n\t\t\tphase: \"turn\",\n\t\t\ttranscript: [\n\t\t\t\t...this.stored.snapshot.transcript,\n\t\t\t\t{\n\t\t\t\t\tid: `user-${this.stored.snapshot.revision + 1}`,\n\t\t\t\t\trole: \"user\",\n\t\t\t\t\tcontent: [{ type: \"text\", text: input.text }],\n\t\t\t\t\ttimestamp: this.stored.snapshot.revision + 1,\n\t\t\t\t},\n\t\t\t],\n\t\t});\n\t\tconst outcome = await done.promise;\n\t\tconst assistant =\n\t\t\toutcome === \"complete\"\n\t\t\t\t? {\n\t\t\t\t\t\tid: `assistant-${this.stored.snapshot.revision + 1}`,\n\t\t\t\t\t\trole: \"assistant\" as const,\n\t\t\t\t\t\tcontent: [{ type: \"text\" as const, text: `reply:${input.text}` }],\n\t\t\t\t\t\tstatus: \"complete\" as const,\n\t\t\t\t\t\tmodel: this.stored.snapshot.model,\n\t\t\t\t\t\tstopReason: \"stop\" as const,\n\t\t\t\t\t\ttimestamp: this.stored.snapshot.revision + 1,\n\t\t\t\t\t}\n\t\t\t\t: {\n\t\t\t\t\t\tid: `assistant-${this.stored.snapshot.revision + 1}`,\n\t\t\t\t\t\trole: \"assistant\" as const,\n\t\t\t\t\t\tcontent: [{ type: \"text\" as const, text: \"\" }],\n\t\t\t\t\t\tstatus: \"aborted\" as const,\n\t\t\t\t\t\tmodel: this.stored.snapshot.model,\n\t\t\t\t\t\tstopReason: \"aborted\" as const,\n\t\t\t\t\t\ttimestamp: this.stored.snapshot.revision + 1,\n\t\t\t\t\t};\n\t\tthis.update({\n\t\t\tphase: \"idle\",\n\t\t\ttranscript: [...this.stored.snapshot.transcript, assistant],\n\t\t});\n\t\tthis.pendingPrompt = undefined;\n\t}\n\n\tasync steer(input: PromptInput): Promise<void> {\n\t\tif (this.getPhase() === \"idle\") throw new PiServerError(\"busy\", \"There is no active prompt to steer\");\n\t\tthis.steers.push(input);\n\t\tthis.update({\n\t\t\tqueuedSteerCount: this.stored.snapshot.queuedSteerCount + 1,\n\t\t\tqueuedSteer: [\n\t\t\t\t...this.stored.snapshot.queuedSteer,\n\t\t\t\t{\n\t\t\t\t\tid: `steer-${this.stored.snapshot.revision + 1}`,\n\t\t\t\t\trole: \"user\",\n\t\t\t\t\tcontent: [{ type: \"text\", text: input.text }],\n\t\t\t\t\ttimestamp: this.stored.snapshot.revision + 1,\n\t\t\t\t},\n\t\t\t],\n\t\t});\n\t}\n\n\tasync abort(): Promise<void> {\n\t\tif (!this.pendingPrompt) throw new PiServerError(\"busy\", \"There is no active prompt to abort\");\n\t\tthis.pendingPrompt.done.resolve(\"aborted\");\n\t}\n\n\tasync setModel(model: ModelRef): Promise<void> {\n\t\tif (this.getPhase() !== \"idle\") throw new PiServerError(\"busy\", \"Session is busy\");\n\t\tthis.update({ model });\n\t}\n\n\tasync setThinking(thinkingLevel: ThinkingLevel): Promise<void> {\n\t\tif (this.getPhase() !== \"idle\") throw new PiServerError(\"busy\", \"Session is busy\");\n\t\tthis.update({ thinkingLevel });\n\t}\n\n\tsubscribe(listener: (event: PiSessionRuntimeEvent) => void): () => void {\n\t\tthis.listeners.add(listener);\n\t\treturn () => this.listeners.delete(listener);\n\t}\n\n\tasync dispose(): Promise<void> {\n\t\tthis.disposeCount += 1;\n\t\tthis.onDispose();\n\t\tthis.disposed.resolve(undefined);\n\t}\n\n\tsetPhase(phase: SessionPhase): void {\n\t\tthis.stored.snapshot = { ...this.stored.snapshot, phase };\n\t}\n\n\tfinishPrompt(): void {\n\t\tif (!this.pendingPrompt) throw new Error(\"No prompt is pending\");\n\t\tthis.pendingPrompt.done.resolve(\"complete\");\n\t}\n\n\temitProgress(progress: TranscriptProgress): void {\n\t\tfor (const listener of this.listeners) listener({ type: \"progress\", progress });\n\t}\n\n\temitError(error: PiServerError): void {\n\t\tfor (const listener of this.listeners) listener({ type: \"error\", error });\n\t}\n\n\temitSnapshot(): void {\n\t\tfor (const listener of this.listeners) listener({ type: \"snapshot\" });\n\t}\n\n\tprivate update(updates: Partial<SessionSnapshot>): void {\n\t\tthis.stored.snapshot = {\n\t\t\t...this.stored.snapshot,\n\t\t\t...updates,\n\t\t\trevision: this.stored.snapshot.revision + 1,\n\t\t\tupdatedAt: this.stored.snapshot.updatedAt + 1,\n\t\t};\n\t\tthis.emitSnapshot();\n\t}\n}\n\ninterface ListDelay {\n\tentered: Deferred<void>;\n\trelease: Deferred<void>;\n}\n\nexport class TestServerService implements PiServerService {\n\treadonly sessions = new Map<string, StoredSession>();\n\treadonly runtimes = new Map<string, TestSessionRuntime[]>();\n\treadonly locked = new Set<string>();\n\tlastCreatedId?: string;\n\tprivate nextListDelay?: ListDelay;\n\n\tasync listSessions(): Promise<SessionMetadata[]> {\n\t\tconst delay = this.nextListDelay;\n\t\tif (delay) {\n\t\t\tthis.nextListDelay = undefined;\n\t\t\tdelay.entered.resolve(undefined);\n\t\t\tawait delay.release.promise;\n\t\t}\n\t\treturn [...this.sessions.values()].map(({ snapshot }) => ({\n\t\t\tid: snapshot.id,\n\t\t\tcreatedAt: snapshot.createdAt,\n\t\t\tupdatedAt: snapshot.updatedAt,\n\t\t\tsessionName: snapshot.name,\n\t\t\tcwd: snapshot.cwd,\n\t\t}));\n\t}\n\n\tasync listModels(): Promise<ModelMetadata[]> {\n\t\treturn [TEST_MODEL];\n\t}\n\n\tasync createSession(options: CreateSessionOptions): Promise<PiSessionRuntime> {\n\t\tthis.lastCreatedId = options.id;\n\t\tif (this.sessions.has(options.id)) throw new PiServerError(\"session_locked\", \"Session already exists\");\n\t\tthis.seed(options.id, options.name, options.cwd, options.model, options.thinkingLevel);\n\t\treturn this.acquire(options.id);\n\t}\n\n\tasync openSession(sessionId: string): Promise<PiSessionRuntime> {\n\t\tif (!this.sessions.has(sessionId)) throw new PiServerError(\"not_found\", `Unknown session: ${sessionId}`);\n\t\tif (this.locked.has(sessionId)) throw new PiServerError(\"session_locked\", `Session is locked: ${sessionId}`);\n\t\treturn this.acquire(sessionId);\n\t}\n\n\tseed(\n\t\tid = \"session-1\",\n\t\tname = `Session ${id}`,\n\t\tcwd = \"/tmp/pi-server-conformance\",\n\t\tmodel: ModelRef = { provider: TEST_MODEL.provider, id: TEST_MODEL.id },\n\t\tthinkingLevel: ThinkingLevel = \"off\",\n\t): void {\n\t\tthis.sessions.set(id, {\n\t\t\tsnapshot: {\n\t\t\t\tid,\n\t\t\t\tname,\n\t\t\t\tcwd,\n\t\t\t\tcreatedAt: 1,\n\t\t\t\tupdatedAt: 1,\n\t\t\t\tphase: \"idle\",\n\t\t\t\tmodel,\n\t\t\t\tthinkingLevel,\n\t\t\t\tattached: false,\n\t\t\t\tlocked: false,\n\t\t\t\trevision: 0,\n\t\t\t\ttranscript: [],\n\t\t\t\tqueuedSteer: [],\n\t\t\t\tqueuedSteerCount: 0,\n\t\t\t},\n\t\t});\n\t}\n\n\tdelayNextList(): ListDelay {\n\t\tconst delay = { entered: new Deferred<void>(), release: new Deferred<void>() };\n\t\tthis.nextListDelay = delay;\n\t\treturn delay;\n\t}\n\n\tlatestRuntime(id: string): TestSessionRuntime {\n\t\tconst runtimes = this.runtimes.get(id);\n\t\tif (!runtimes?.length) throw new Error(`No runtime for ${id}`);\n\t\treturn runtimes.at(-1)!;\n\t}\n\n\tprivate acquire(id: string): TestSessionRuntime {\n\t\tconst stored = this.sessions.get(id);\n\t\tif (!stored) throw new Error(`Unknown session: ${id}`);\n\t\tthis.locked.add(id);\n\t\tconst runtime = new TestSessionRuntime(stored, () => this.locked.delete(id));\n\t\tconst runtimes = this.runtimes.get(id) ?? [];\n\t\truntimes.push(runtime);\n\t\tthis.runtimes.set(id, runtimes);\n\t\treturn runtime;\n\t}\n}\n"]}

package/dist/transports/unix/index.d.ts

@@ -0,0 +1,4 @@
+export { createUnixListener } from "./listener.ts";
+export { createUnixServer } from "./preset.ts";
+export type { UnixListenerOptions, UnixServerOptions } from "./types.ts";
+//# sourceMappingURL=index.d.ts.map

package/dist/transports/unix/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/transports/unix/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,MAAM,eAAe,CAAC;AACnD,OAAO,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAC/C,YAAY,EAAE,mBAAmB,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC","sourcesContent":["export { createUnixListener } from \"./listener.ts\";\nexport { createUnixServer } from \"./preset.ts\";\nexport type { UnixListenerOptions, UnixServerOptions } from \"./types.ts\";\n"]}

package/dist/transports/unix/index.js

@@ -0,0 +1,3 @@
+export { createUnixListener } from "./listener.js";
+export { createUnixServer } from "./preset.js";
+//# sourceMappingURL=index.js.map

package/dist/transports/unix/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/transports/unix/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,MAAM,eAAe,CAAC;AACnD,OAAO,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC","sourcesContent":["export { createUnixListener } from \"./listener.ts\";\nexport { createUnixServer } from \"./preset.ts\";\nexport type { UnixListenerOptions, UnixServerOptions } from \"./types.ts\";\n"]}

package/dist/transports/unix/listener.d.ts

@@ -0,0 +1,25 @@
+import { type Socket } from "node:net";
+import type { ByteConnection } from "../../connection.ts";
+import type { PiServerListener } from "../../listener.ts";
+import type { UnixListenerOptions } from "./types.ts";
+export declare function validateUnixSocketPath(path: string, description?: string): void;
+/** @internal Exported only for transport-level verification. */
+export declare class UnixByteConnection implements ByteConnection {
+    private readonly socket;
+    private readonly gracefulCloseTimeoutMs;
+    private readonly maxPendingBytes;
+    private pendingBytes;
+    private closedValue;
+    private closing;
+    private writeTail;
+    private closePromise?;
+    private resolveClose?;
+    constructor(socket: Socket, gracefulCloseTimeoutMs: number, maxPendingBytes: number);
+    get closed(): boolean;
+    send(chunk: Uint8Array): Promise<void>;
+    close(finalChunk?: Uint8Array): Promise<void>;
+    markClosed(): void;
+    private write;
+}
+export declare function createUnixListener(options: UnixListenerOptions): PiServerListener;
+//# sourceMappingURL=listener.d.ts.map