@earendil-works/pi-server 0.84.0

package/dist/sessions.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"sessions.d.ts","sourceRoot":"","sources":["../src/sessions.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,OAAO,EAAE,aAAa,EAAE,eAAe,EAAmB,MAAM,6BAA6B,CAAC;AAC5G,OAAO,KAAK,EAAE,cAAc,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAC;AAEvE,OAAO,KAAK,EAAwB,eAAe,EAA2C,MAAM,YAAY,CAAC;AAajH,UAAU,yBAAyB;IAClC,OAAO,EAAE,eAAe,CAAC;IACzB,SAAS,EAAE,MAAM,OAAO,CAAC;IACzB,WAAW,EAAE,CAAC,UAAU,EAAE,eAAe,EAAE,OAAO,EAAE,aAAa,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC;IACvF,eAAe,EAAE,CAAC,UAAU,EAAE,cAAc,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAC/D,UAAU,EAAE,CAAC,UAAU,EAAE,eAAe,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAC3D,uBAAuB,EAAE,MAAM,IAAI,CAAC;IACpC,WAAW,EAAE,CAAC,KAAK,EAAE,OAAO,KAAK,IAAI,CAAC;CACtC;AAYD,qBAAa,kBAAkB;IAC9B,OAAO,CAAC,QAAQ,CAAC,OAAO,CAA4B;IACpD,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAkC;IAC/D,OAAO,CAAC,QAAQ,CAAC,eAAe,CAA2C;IAE3E,YAAY,OAAO,EAAE,yBAAyB,EAE7C;IAEK,cAAc,CAAC,UAAU,EAAE,eAAe,EAAE,OAAO,EAAE,OAAO;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OAyEjE;IAEK,UAAU,CAAC,UAAU,EAAE,eAAe,GAAG,OAAO,CAAC,IAAI,CAAC,CAU3D;IAEK,YAAY,IAAI,OAAO,CAAC,eAAe,EAAE,CAAC,CAgB/C;IAEK,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAiB3B;YAEa,YAAY;YAeZ,OAAO;YAuBP,MAAM;IAuCpB,OAAO,CAAC,kBAAkB;YAiBZ,SAAS;YAWT,kBAAkB;IAahC,OAAO,CAAC,aAAa;YAIP,iBAAiB;YAOjB,MAAM;IASpB,OAAO,CAAC,eAAe;IAWvB,OAAO,CAAC,oBAAoB;YAId,YAAY;CAsB1B","sourcesContent":["import { randomUUID } from \"node:crypto\";\nimport type { Command, EventEnvelope, SessionMetadata, SessionSnapshot } from \"@earendil-works/pi-protocol\";\nimport type { ByteConnection, ConnectionState } from \"./connection.ts\";\nimport { PiServerError } from \"./errors.ts\";\nimport type { CreateSessionOptions, PiServerService, PiSessionRuntime, PiSessionRuntimeEvent } from \"./types.ts\";\n\ninterface LiveSession {\n\tid: string;\n\truntime: PiSessionRuntime;\n\tconnections: Set<ConnectionState>;\n\tunsubscribe: () => void;\n\toperationCount: number;\n\tready: boolean;\n\tterminal: boolean;\n\tdisposing?: Promise<void>;\n}\n\ninterface LiveSessionManagerOptions {\n\tservice: PiServerService;\n\tisClosing: () => boolean;\n\tsendMessage: (connection: ConnectionState, message: EventEnvelope) => Promise<boolean>;\n\tcloseConnection: (connection: ByteConnection) => Promise<void>;\n\tdisconnect: (connection: ConnectionState) => Promise<void>;\n\tbroadcastServerSnapshot: () => void;\n\treportError: (error: unknown) => void;\n}\n\nfunction toMetadata(snapshot: SessionSnapshot): SessionMetadata {\n\treturn {\n\t\tid: snapshot.id,\n\t\tcreatedAt: snapshot.createdAt,\n\t\tupdatedAt: snapshot.updatedAt,\n\t\tsessionName: snapshot.name,\n\t\tcwd: snapshot.cwd,\n\t};\n}\n\nexport class LiveSessionManager {\n\tprivate readonly options: LiveSessionManagerOptions;\n\tprivate readonly liveSessions = new Map<string, LiveSession>();\n\tprivate readonly openingSessions = new Map<string, Promise<LiveSession>>();\n\n\tconstructor(options: LiveSessionManagerOptions) {\n\t\tthis.options = options;\n\t}\n\n\tasync executeCommand(connection: ConnectionState, command: Command) {\n\t\tswitch (command.command) {\n\t\t\tcase \"list\":\n\t\t\t\treturn { command: \"list\" as const, sessions: await this.listMetadata() };\n\t\t\tcase \"create\": {\n\t\t\t\tconst id = randomUUID();\n\t\t\t\tconst options: CreateSessionOptions = {\n\t\t\t\t\tid,\n\t\t\t\t\tcwd: command.cwd,\n\t\t\t\t\tname: command.name,\n\t\t\t\t\tmodel: command.model,\n\t\t\t\t\tthinkingLevel: command.thinkingLevel,\n\t\t\t\t};\n\t\t\t\tconst live = await this.acquire(id, () => this.options.service.createSession(options));\n\t\t\t\tawait this.attach(connection, live);\n\t\t\t\tconst session = this.forConnection(await this.broadcastSnapshot(live), connection);\n\t\t\t\tthis.options.broadcastServerSnapshot();\n\t\t\t\treturn { command: \"create\" as const, session };\n\t\t\t}\n\t\t\tcase \"attach\": {\n\t\t\t\tconst live = await this.acquire(command.sessionId, () =>\n\t\t\t\t\tthis.options.service.openSession(command.sessionId),\n\t\t\t\t);\n\t\t\t\tawait this.attach(connection, live);\n\t\t\t\tconst session = this.forConnection(await this.broadcastSnapshot(live), connection);\n\t\t\t\tthis.options.broadcastServerSnapshot();\n\t\t\t\treturn { command: \"attach\" as const, session };\n\t\t\t}\n\t\t\tcase \"detach\": {\n\t\t\t\tconst live = this.liveSessions.get(command.sessionId);\n\t\t\t\tif (connection.sessionIds.has(command.sessionId)) {\n\t\t\t\t\tconnection.sessionIds.delete(command.sessionId);\n\t\t\t\t\tif (live) {\n\t\t\t\t\t\tlive.connections.delete(connection);\n\t\t\t\t\t\tif (live.connections.size > 0 && !live.terminal && !live.disposing) {\n\t\t\t\t\t\t\tawait this.broadcastSnapshot(live);\n\t\t\t\t\t\t}\n\t\t\t\t\t\tawait this.maybeDispose(live);\n\t\t\t\t\t}\n\t\t\t\t\tthis.options.broadcastServerSnapshot();\n\t\t\t\t}\n\t\t\t\treturn { command: \"detach\" as const, sessionId: command.sessionId };\n\t\t\t}\n\t\t\tcase \"prompt\": {\n\t\t\t\tconst live = this.requireAttached(connection, command.sessionId);\n\t\t\t\tconst session = await this.runOperation(connection, live, () =>\n\t\t\t\t\tlive.runtime.prompt({ text: command.text }),\n\t\t\t\t);\n\t\t\t\treturn { command: \"prompt\" as const, session };\n\t\t\t}\n\t\t\tcase \"steer\": {\n\t\t\t\tconst live = this.requireAttached(connection, command.sessionId);\n\t\t\t\tconst session = await this.runOperation(connection, live, () => live.runtime.steer({ text: command.text }));\n\t\t\t\treturn { command: \"steer\" as const, session };\n\t\t\t}\n\t\t\tcase \"abort\": {\n\t\t\t\tconst live = this.requireAttached(connection, command.sessionId);\n\t\t\t\tconst session = await this.runOperation(connection, live, () => live.runtime.abort());\n\t\t\t\treturn { command: \"abort\" as const, session };\n\t\t\t}\n\t\t\tcase \"set_model\": {\n\t\t\t\tconst live = this.requireAttached(connection, command.sessionId);\n\t\t\t\tconst session = await this.runOperation(connection, live, () => live.runtime.setModel(command.model));\n\t\t\t\treturn { command: \"set_model\" as const, session };\n\t\t\t}\n\t\t\tcase \"set_thinking\": {\n\t\t\t\tconst live = this.requireAttached(connection, command.sessionId);\n\t\t\t\tconst session = await this.runOperation(connection, live, () =>\n\t\t\t\t\tlive.runtime.setThinking(command.thinkingLevel),\n\t\t\t\t);\n\t\t\t\treturn { command: \"set_thinking\" as const, session };\n\t\t\t}\n\t\t}\n\t}\n\n\tasync disconnect(connection: ConnectionState): Promise<void> {\n\t\tconst sessions = [...connection.sessionIds]\n\t\t\t.map((id) => this.liveSessions.get(id))\n\t\t\t.filter((live): live is LiveSession => live !== undefined);\n\t\tconnection.sessionIds.clear();\n\t\tfor (const live of sessions) live.connections.delete(connection);\n\t\tconst results = await Promise.allSettled(sessions.map((live) => this.maybeDispose(live)));\n\t\tfor (const result of results) {\n\t\t\tif (result.status === \"rejected\") this.options.reportError(result.reason);\n\t\t}\n\t}\n\n\tasync listMetadata(): Promise<SessionMetadata[]> {\n\t\tconst stored = await this.options.service.listSessions();\n\t\tconst liveSnapshots = await Promise.all(\n\t\t\t[...this.liveSessions.values()]\n\t\t\t\t.filter((live) => !live.disposing)\n\t\t\t\t.map(async (live) => [live.id, await this.normalizedSnapshot(live)] as const),\n\t\t);\n\t\tconst liveById = new Map(liveSnapshots);\n\t\tconst metadata = stored.map((item) => {\n\t\t\tconst snapshot = liveById.get(item.id);\n\t\t\tif (!snapshot) return item;\n\t\t\tliveById.delete(item.id);\n\t\t\treturn { ...item, ...toMetadata(snapshot) };\n\t\t});\n\t\tfor (const snapshot of liveById.values()) metadata.push(toMetadata(snapshot));\n\t\treturn metadata;\n\t}\n\n\tasync close(): Promise<void> {\n\t\tconst openingResults = await Promise.allSettled([...this.openingSessions.values()]);\n\t\tfor (const result of openingResults) {\n\t\t\tif (result.status === \"rejected\") this.options.reportError(result.reason);\n\t\t}\n\t\tconst sessions = [...this.liveSessions.values()];\n\t\tthis.liveSessions.clear();\n\t\tawait Promise.all(\n\t\t\tsessions.map(async (live) => {\n\t\t\t\tif (live.disposing) {\n\t\t\t\t\tawait live.disposing;\n\t\t\t\t\treturn;\n\t\t\t\t}\n\t\t\t\tlive.unsubscribe();\n\t\t\t\tawait live.runtime.dispose();\n\t\t\t}),\n\t\t);\n\t}\n\n\tprivate async runOperation(\n\t\tconnection: ConnectionState,\n\t\tlive: LiveSession,\n\t\toperation: () => Promise<void>,\n\t): Promise<SessionSnapshot> {\n\t\tlive.operationCount += 1;\n\t\ttry {\n\t\t\tawait operation();\n\t\t\treturn this.forConnection(await this.broadcastSnapshot(live), connection);\n\t\t} finally {\n\t\t\tlive.operationCount -= 1;\n\t\t\tthis.scheduleMaybeDispose(live);\n\t\t}\n\t}\n\n\tprivate async acquire(id: string, acquireRuntime: () => Promise<PiSessionRuntime>): Promise<LiveSession> {\n\t\tfor (;;) {\n\t\t\tconst existing = this.liveSessions.get(id);\n\t\t\tif (existing) {\n\t\t\t\tif (existing.terminal) throw new PiServerError(\"session_locked\", `Session runtime is terminating: ${id}`);\n\t\t\t\tif (existing.disposing) {\n\t\t\t\t\tawait existing.disposing;\n\t\t\t\t\tcontinue;\n\t\t\t\t}\n\t\t\t\treturn existing;\n\t\t\t}\n\t\t\tconst opening = this.openingSessions.get(id);\n\t\t\tif (opening) return opening;\n\t\t\tconst pending = this.create(id, acquireRuntime);\n\t\t\tthis.openingSessions.set(id, pending);\n\t\t\ttry {\n\t\t\t\treturn await pending;\n\t\t\t} finally {\n\t\t\t\tif (this.openingSessions.get(id) === pending) this.openingSessions.delete(id);\n\t\t\t}\n\t\t}\n\t}\n\n\tprivate async create(id: string, acquireRuntime: () => Promise<PiSessionRuntime>): Promise<LiveSession> {\n\t\tconst runtime = await acquireRuntime();\n\t\tif (this.options.isClosing()) {\n\t\t\tawait runtime.dispose();\n\t\t\tthrow new Error(\"PiServer closed while acquiring a session runtime\");\n\t\t}\n\t\tlet live: LiveSession | undefined;\n\t\ttry {\n\t\t\tconst snapshot = await runtime.snapshot();\n\t\t\tif (snapshot.id !== id) {\n\t\t\t\tthrow new PiServerError(\n\t\t\t\t\t\"invalid_request\",\n\t\t\t\t\t`Service returned session ${snapshot.id} for server-assigned session ${id}`,\n\t\t\t\t);\n\t\t\t}\n\t\t\tlive = {\n\t\t\t\tid,\n\t\t\t\truntime,\n\t\t\t\tconnections: new Set(),\n\t\t\t\tunsubscribe: () => {},\n\t\t\t\toperationCount: 0,\n\t\t\t\tready: false,\n\t\t\t\tterminal: false,\n\t\t\t};\n\t\t\tlive.unsubscribe = runtime.subscribe((event) => this.handleRuntimeEvent(live!, event));\n\t\t\tthis.liveSessions.set(id, live);\n\t\t\tlive.ready = true;\n\t\t\treturn live;\n\t\t} catch (error) {\n\t\t\tif (live) live.unsubscribe();\n\t\t\ttry {\n\t\t\t\tawait runtime.dispose();\n\t\t\t} catch (disposeError) {\n\t\t\t\tthis.options.reportError(disposeError);\n\t\t\t}\n\t\t\tthrow error;\n\t\t}\n\t}\n\n\tprivate handleRuntimeEvent(live: LiveSession, event: PiSessionRuntimeEvent): void {\n\t\tif (event.type === \"error\") {\n\t\t\tvoid this.terminate(live, event.error).catch((error: unknown) => this.options.reportError(error));\n\t\t\treturn;\n\t\t}\n\t\tif (event.type === \"progress\") {\n\t\t\tconst envelope: EventEnvelope = {\n\t\t\t\ttype: \"event\",\n\t\t\t\tevent: { type: \"session_progress\", sessionId: live.id, progress: event.progress },\n\t\t\t};\n\t\t\tfor (const connection of live.connections) void this.options.sendMessage(connection, envelope);\n\t\t} else {\n\t\t\tvoid this.broadcastSnapshot(live).catch((error: unknown) => this.options.reportError(error));\n\t\t}\n\t\tthis.scheduleMaybeDispose(live);\n\t}\n\n\tprivate async terminate(live: LiveSession, error: PiServerError): Promise<void> {\n\t\tif (live.terminal) return;\n\t\tlive.terminal = true;\n\t\tthis.options.reportError(error);\n\t\tlive.unsubscribe();\n\t\tconst connections = [...live.connections];\n\t\tawait Promise.all(connections.map((connection) => this.options.closeConnection(connection.connection)));\n\t\tawait Promise.all(connections.map((connection) => this.options.disconnect(connection)));\n\t\tawait this.maybeDispose(live);\n\t}\n\n\tprivate async normalizedSnapshot(live: LiveSession): Promise<SessionSnapshot> {\n\t\tconst snapshot = await live.runtime.snapshot();\n\t\tif (snapshot.id !== live.id) {\n\t\t\tthrow new PiServerError(\"invalid_request\", `Runtime session ID changed from ${live.id} to ${snapshot.id}`);\n\t\t}\n\t\treturn {\n\t\t\t...snapshot,\n\t\t\tphase: live.runtime.getPhase(),\n\t\t\tattached: live.connections.size > 0,\n\t\t\tlocked: true,\n\t\t};\n\t}\n\n\tprivate forConnection(snapshot: SessionSnapshot, connection: ConnectionState): SessionSnapshot {\n\t\treturn { ...snapshot, attached: connection.sessionIds.has(snapshot.id) };\n\t}\n\n\tprivate async broadcastSnapshot(live: LiveSession): Promise<SessionSnapshot> {\n\t\tconst snapshot = await this.normalizedSnapshot(live);\n\t\tconst envelope: EventEnvelope = { type: \"event\", event: { type: \"session_snapshot\", snapshot } };\n\t\tfor (const connection of live.connections) void this.options.sendMessage(connection, envelope);\n\t\treturn snapshot;\n\t}\n\n\tprivate async attach(connection: ConnectionState, live: LiveSession): Promise<void> {\n\t\tif (connection.disconnected || connection.stage !== \"ready\" || connection.connection.closed) {\n\t\t\tawait this.maybeDispose(live);\n\t\t\tthrow new PiServerError(\"invalid_request\", \"Connection closed while attaching to a session\");\n\t\t}\n\t\tconnection.sessionIds.add(live.id);\n\t\tlive.connections.add(connection);\n\t}\n\n\tprivate requireAttached(connection: ConnectionState, sessionId: string): LiveSession {\n\t\tif (!connection.sessionIds.has(sessionId)) {\n\t\t\tthrow new PiServerError(\"invalid_request\", `Connection is not attached to session ${sessionId}`);\n\t\t}\n\t\tconst live = this.liveSessions.get(sessionId);\n\t\tif (!live || live.terminal || live.disposing) {\n\t\t\tthrow new PiServerError(\"not_found\", `Session is not live: ${sessionId}`);\n\t\t}\n\t\treturn live;\n\t}\n\n\tprivate scheduleMaybeDispose(live: LiveSession): void {\n\t\tvoid this.maybeDispose(live).catch((error: unknown) => this.options.reportError(error));\n\t}\n\n\tprivate async maybeDispose(live: LiveSession): Promise<void> {\n\t\tif (\n\t\t\tthis.options.isClosing() ||\n\t\t\t!live.ready ||\n\t\t\tlive.disposing ||\n\t\t\tlive.connections.size > 0 ||\n\t\t\tlive.operationCount > 0 ||\n\t\t\t(!live.terminal && live.runtime.getPhase() !== \"idle\")\n\t\t) {\n\t\t\treturn live.disposing;\n\t\t}\n\t\tlive.unsubscribe();\n\t\tlive.disposing = (async () => {\n\t\t\ttry {\n\t\t\t\tawait live.runtime.dispose();\n\t\t\t} finally {\n\t\t\t\tif (this.liveSessions.get(live.id) === live) this.liveSessions.delete(live.id);\n\t\t\t}\n\t\t})();\n\t\tawait live.disposing;\n\t\tif (!this.options.isClosing()) this.options.broadcastServerSnapshot();\n\t}\n}\n"]}

package/dist/sessions.js

@@ -0,0 +1,305 @@
+import { randomUUID } from "node:crypto";
+import { PiServerError } from "./errors.js";
+function toMetadata(snapshot) {
+    return {
+        id: snapshot.id,
+        createdAt: snapshot.createdAt,
+        updatedAt: snapshot.updatedAt,
+        sessionName: snapshot.name,
+        cwd: snapshot.cwd,
+    };
+}
+export class LiveSessionManager {
+    options;
+    liveSessions = new Map();
+    openingSessions = new Map();
+    constructor(options) {
+        this.options = options;
+    }
+    async executeCommand(connection, command) {
+        switch (command.command) {
+            case "list":
+                return { command: "list", sessions: await this.listMetadata() };
+            case "create": {
+                const id = randomUUID();
+                const options = {
+                    id,
+                    cwd: command.cwd,
+                    name: command.name,
+                    model: command.model,
+                    thinkingLevel: command.thinkingLevel,
+                };
+                const live = await this.acquire(id, () => this.options.service.createSession(options));
+                await this.attach(connection, live);
+                const session = this.forConnection(await this.broadcastSnapshot(live), connection);
+                this.options.broadcastServerSnapshot();
+                return { command: "create", session };
+            }
+            case "attach": {
+                const live = await this.acquire(command.sessionId, () => this.options.service.openSession(command.sessionId));
+                await this.attach(connection, live);
+                const session = this.forConnection(await this.broadcastSnapshot(live), connection);
+                this.options.broadcastServerSnapshot();
+                return { command: "attach", session };
+            }
+            case "detach": {
+                const live = this.liveSessions.get(command.sessionId);
+                if (connection.sessionIds.has(command.sessionId)) {
+                    connection.sessionIds.delete(command.sessionId);
+                    if (live) {
+                        live.connections.delete(connection);
+                        if (live.connections.size > 0 && !live.terminal && !live.disposing) {
+                            await this.broadcastSnapshot(live);
+                        }
+                        await this.maybeDispose(live);
+                    }
+                    this.options.broadcastServerSnapshot();
+                }
+                return { command: "detach", sessionId: command.sessionId };
+            }
+            case "prompt": {
+                const live = this.requireAttached(connection, command.sessionId);
+                const session = await this.runOperation(connection, live, () => live.runtime.prompt({ text: command.text }));
+                return { command: "prompt", session };
+            }
+            case "steer": {
+                const live = this.requireAttached(connection, command.sessionId);
+                const session = await this.runOperation(connection, live, () => live.runtime.steer({ text: command.text }));
+                return { command: "steer", session };
+            }
+            case "abort": {
+                const live = this.requireAttached(connection, command.sessionId);
+                const session = await this.runOperation(connection, live, () => live.runtime.abort());
+                return { command: "abort", session };
+            }
+            case "set_model": {
+                const live = this.requireAttached(connection, command.sessionId);
+                const session = await this.runOperation(connection, live, () => live.runtime.setModel(command.model));
+                return { command: "set_model", session };
+            }
+            case "set_thinking": {
+                const live = this.requireAttached(connection, command.sessionId);
+                const session = await this.runOperation(connection, live, () => live.runtime.setThinking(command.thinkingLevel));
+                return { command: "set_thinking", session };
+            }
+        }
+    }
+    async disconnect(connection) {
+        const sessions = [...connection.sessionIds]
+            .map((id) => this.liveSessions.get(id))
+            .filter((live) => live !== undefined);
+        connection.sessionIds.clear();
+        for (const live of sessions)
+            live.connections.delete(connection);
+        const results = await Promise.allSettled(sessions.map((live) => this.maybeDispose(live)));
+        for (const result of results) {
+            if (result.status === "rejected")
+                this.options.reportError(result.reason);
+        }
+    }
+    async listMetadata() {
+        const stored = await this.options.service.listSessions();
+        const liveSnapshots = await Promise.all([...this.liveSessions.values()]
+            .filter((live) => !live.disposing)
+            .map(async (live) => [live.id, await this.normalizedSnapshot(live)]));
+        const liveById = new Map(liveSnapshots);
+        const metadata = stored.map((item) => {
+            const snapshot = liveById.get(item.id);
+            if (!snapshot)
+                return item;
+            liveById.delete(item.id);
+            return { ...item, ...toMetadata(snapshot) };
+        });
+        for (const snapshot of liveById.values())
+            metadata.push(toMetadata(snapshot));
+        return metadata;
+    }
+    async close() {
+        const openingResults = await Promise.allSettled([...this.openingSessions.values()]);
+        for (const result of openingResults) {
+            if (result.status === "rejected")
+                this.options.reportError(result.reason);
+        }
+        const sessions = [...this.liveSessions.values()];
+        this.liveSessions.clear();
+        await Promise.all(sessions.map(async (live) => {
+            if (live.disposing) {
+                await live.disposing;
+                return;
+            }
+            live.unsubscribe();
+            await live.runtime.dispose();
+        }));
+    }
+    async runOperation(connection, live, operation) {
+        live.operationCount += 1;
+        try {
+            await operation();
+            return this.forConnection(await this.broadcastSnapshot(live), connection);
+        }
+        finally {
+            live.operationCount -= 1;
+            this.scheduleMaybeDispose(live);
+        }
+    }
+    async acquire(id, acquireRuntime) {
+        for (;;) {
+            const existing = this.liveSessions.get(id);
+            if (existing) {
+                if (existing.terminal)
+                    throw new PiServerError("session_locked", `Session runtime is terminating: ${id}`);
+                if (existing.disposing) {
+                    await existing.disposing;
+                    continue;
+                }
+                return existing;
+            }
+            const opening = this.openingSessions.get(id);
+            if (opening)
+                return opening;
+            const pending = this.create(id, acquireRuntime);
+            this.openingSessions.set(id, pending);
+            try {
+                return await pending;
+            }
+            finally {
+                if (this.openingSessions.get(id) === pending)
+                    this.openingSessions.delete(id);
+            }
+        }
+    }
+    async create(id, acquireRuntime) {
+        const runtime = await acquireRuntime();
+        if (this.options.isClosing()) {
+            await runtime.dispose();
+            throw new Error("PiServer closed while acquiring a session runtime");
+        }
+        let live;
+        try {
+            const snapshot = await runtime.snapshot();
+            if (snapshot.id !== id) {
+                throw new PiServerError("invalid_request", `Service returned session ${snapshot.id} for server-assigned session ${id}`);
+            }
+            live = {
+                id,
+                runtime,
+                connections: new Set(),
+                unsubscribe: () => { },
+                operationCount: 0,
+                ready: false,
+                terminal: false,
+            };
+            live.unsubscribe = runtime.subscribe((event) => this.handleRuntimeEvent(live, event));
+            this.liveSessions.set(id, live);
+            live.ready = true;
+            return live;
+        }
+        catch (error) {
+            if (live)
+                live.unsubscribe();
+            try {
+                await runtime.dispose();
+            }
+            catch (disposeError) {
+                this.options.reportError(disposeError);
+            }
+            throw error;
+        }
+    }
+    handleRuntimeEvent(live, event) {
+        if (event.type === "error") {
+            void this.terminate(live, event.error).catch((error) => this.options.reportError(error));
+            return;
+        }
+        if (event.type === "progress") {
+            const envelope = {
+                type: "event",
+                event: { type: "session_progress", sessionId: live.id, progress: event.progress },
+            };
+            for (const connection of live.connections)
+                void this.options.sendMessage(connection, envelope);
+        }
+        else {
+            void this.broadcastSnapshot(live).catch((error) => this.options.reportError(error));
+        }
+        this.scheduleMaybeDispose(live);
+    }
+    async terminate(live, error) {
+        if (live.terminal)
+            return;
+        live.terminal = true;
+        this.options.reportError(error);
+        live.unsubscribe();
+        const connections = [...live.connections];
+        await Promise.all(connections.map((connection) => this.options.closeConnection(connection.connection)));
+        await Promise.all(connections.map((connection) => this.options.disconnect(connection)));
+        await this.maybeDispose(live);
+    }
+    async normalizedSnapshot(live) {
+        const snapshot = await live.runtime.snapshot();
+        if (snapshot.id !== live.id) {
+            throw new PiServerError("invalid_request", `Runtime session ID changed from ${live.id} to ${snapshot.id}`);
+        }
+        return {
+            ...snapshot,
+            phase: live.runtime.getPhase(),
+            attached: live.connections.size > 0,
+            locked: true,
+        };
+    }
+    forConnection(snapshot, connection) {
+        return { ...snapshot, attached: connection.sessionIds.has(snapshot.id) };
+    }
+    async broadcastSnapshot(live) {
+        const snapshot = await this.normalizedSnapshot(live);
+        const envelope = { type: "event", event: { type: "session_snapshot", snapshot } };
+        for (const connection of live.connections)
+            void this.options.sendMessage(connection, envelope);
+        return snapshot;
+    }
+    async attach(connection, live) {
+        if (connection.disconnected || connection.stage !== "ready" || connection.connection.closed) {
+            await this.maybeDispose(live);
+            throw new PiServerError("invalid_request", "Connection closed while attaching to a session");
+        }
+        connection.sessionIds.add(live.id);
+        live.connections.add(connection);
+    }
+    requireAttached(connection, sessionId) {
+        if (!connection.sessionIds.has(sessionId)) {
+            throw new PiServerError("invalid_request", `Connection is not attached to session ${sessionId}`);
+        }
+        const live = this.liveSessions.get(sessionId);
+        if (!live || live.terminal || live.disposing) {
+            throw new PiServerError("not_found", `Session is not live: ${sessionId}`);
+        }
+        return live;
+    }
+    scheduleMaybeDispose(live) {
+        void this.maybeDispose(live).catch((error) => this.options.reportError(error));
+    }
+    async maybeDispose(live) {
+        if (this.options.isClosing() ||
+            !live.ready ||
+            live.disposing ||
+            live.connections.size > 0 ||
+            live.operationCount > 0 ||
+            (!live.terminal && live.runtime.getPhase() !== "idle")) {
+            return live.disposing;
+        }
+        live.unsubscribe();
+        live.disposing = (async () => {
+            try {
+                await live.runtime.dispose();
+            }
+            finally {
+                if (this.liveSessions.get(live.id) === live)
+                    this.liveSessions.delete(live.id);
+            }
+        })();
+        await live.disposing;
+        if (!this.options.isClosing())
+            this.options.broadcastServerSnapshot();
+    }
+}
+//# sourceMappingURL=sessions.js.map

package/dist/sessions.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"sessions.js","sourceRoot":"","sources":["../src/sessions.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAGzC,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAwB5C,SAAS,UAAU,CAAC,QAAyB,EAAmB;IAC/D,OAAO;QACN,EAAE,EAAE,QAAQ,CAAC,EAAE;QACf,SAAS,EAAE,QAAQ,CAAC,SAAS;QAC7B,SAAS,EAAE,QAAQ,CAAC,SAAS;QAC7B,WAAW,EAAE,QAAQ,CAAC,IAAI;QAC1B,GAAG,EAAE,QAAQ,CAAC,GAAG;KACjB,CAAC;AAAA,CACF;AAED,MAAM,OAAO,kBAAkB;IACb,OAAO,CAA4B;IACnC,YAAY,GAAG,IAAI,GAAG,EAAuB,CAAC;IAC9C,eAAe,GAAG,IAAI,GAAG,EAAgC,CAAC;IAE3E,YAAY,OAAkC,EAAE;QAC/C,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IAAA,CACvB;IAED,KAAK,CAAC,cAAc,CAAC,UAA2B,EAAE,OAAgB,EAAE;QACnE,QAAQ,OAAO,CAAC,OAAO,EAAE,CAAC;YACzB,KAAK,MAAM;gBACV,OAAO,EAAE,OAAO,EAAE,MAAe,EAAE,QAAQ,EAAE,MAAM,IAAI,CAAC,YAAY,EAAE,EAAE,CAAC;YAC1E,KAAK,QAAQ,EAAE,CAAC;gBACf,MAAM,EAAE,GAAG,UAAU,EAAE,CAAC;gBACxB,MAAM,OAAO,GAAyB;oBACrC,EAAE;oBACF,GAAG,EAAE,OAAO,CAAC,GAAG;oBAChB,IAAI,EAAE,OAAO,CAAC,IAAI;oBAClB,KAAK,EAAE,OAAO,CAAC,KAAK;oBACpB,aAAa,EAAE,OAAO,CAAC,aAAa;iBACpC,CAAC;gBACF,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC;gBACvF,MAAM,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC;gBACpC,MAAM,OAAO,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,EAAE,UAAU,CAAC,CAAC;gBACnF,IAAI,CAAC,OAAO,CAAC,uBAAuB,EAAE,CAAC;gBACvC,OAAO,EAAE,OAAO,EAAE,QAAiB,EAAE,OAAO,EAAE,CAAC;YAChD,CAAC;YACD,KAAK,QAAQ,EAAE,CAAC;gBACf,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,EAAE,GAAG,EAAE,CACvD,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC,SAAS,CAAC,CACnD,CAAC;gBACF,MAAM,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC;gBACpC,MAAM,OAAO,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,EAAE,UAAU,CAAC,CAAC;gBACnF,IAAI,CAAC,OAAO,CAAC,uBAAuB,EAAE,CAAC;gBACvC,OAAO,EAAE,OAAO,EAAE,QAAiB,EAAE,OAAO,EAAE,CAAC;YAChD,CAAC;YACD,KAAK,QAAQ,EAAE,CAAC;gBACf,MAAM,IAAI,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;gBACtD,IAAI,UAAU,CAAC,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC;oBAClD,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;oBAChD,IAAI,IAAI,EAAE,CAAC;wBACV,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;wBACpC,IAAI,IAAI,CAAC,WAAW,CAAC,IAAI,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;4BACpE,MAAM,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;wBACpC,CAAC;wBACD,MAAM,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;oBAC/B,CAAC;oBACD,IAAI,CAAC,OAAO,CAAC,uBAAuB,EAAE,CAAC;gBACxC,CAAC;gBACD,OAAO,EAAE,OAAO,EAAE,QAAiB,EAAE,SAAS,EAAE,OAAO,CAAC,SAAS,EAAE,CAAC;YACrE,CAAC;YACD,KAAK,QAAQ,EAAE,CAAC;gBACf,MAAM,IAAI,GAAG,IAAI,CAAC,eAAe,CAAC,UAAU,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC;gBACjE,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,UAAU,EAAE,IAAI,EAAE,GAAG,EAAE,CAC9D,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,OAAO,CAAC,IAAI,EAAE,CAAC,CAC3C,CAAC;gBACF,OAAO,EAAE,OAAO,EAAE,QAAiB,EAAE,OAAO,EAAE,CAAC;YAChD,CAAC;YACD,KAAK,OAAO,EAAE,CAAC;gBACd,MAAM,IAAI,GAAG,IAAI,CAAC,eAAe,CAAC,UAAU,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC;gBACjE,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,UAAU,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;gBAC5G,OAAO,EAAE,OAAO,EAAE,OAAgB,EAAE,OAAO,EAAE,CAAC;YAC/C,CAAC;YACD,KAAK,OAAO,EAAE,CAAC;gBACd,MAAM,IAAI,GAAG,IAAI,CAAC,eAAe,CAAC,UAAU,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC;gBACjE,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,UAAU,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC;gBACtF,OAAO,EAAE,OAAO,EAAE,OAAgB,EAAE,OAAO,EAAE,CAAC;YAC/C,CAAC;YACD,KAAK,WAAW,EAAE,CAAC;gBAClB,MAAM,IAAI,GAAG,IAAI,CAAC,eAAe,CAAC,UAAU,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC;gBACjE,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,UAAU,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC;gBACtG,OAAO,EAAE,OAAO,EAAE,WAAoB,EAAE,OAAO,EAAE,CAAC;YACnD,CAAC;YACD,KAAK,cAAc,EAAE,CAAC;gBACrB,MAAM,IAAI,GAAG,IAAI,CAAC,eAAe,CAAC,UAAU,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC;gBACjE,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,UAAU,EAAE,IAAI,EAAE,GAAG,EAAE,CAC9D,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC,aAAa,CAAC,CAC/C,CAAC;gBACF,OAAO,EAAE,OAAO,EAAE,cAAuB,EAAE,OAAO,EAAE,CAAC;YACtD,CAAC;QACF,CAAC;IAAA,CACD;IAED,KAAK,CAAC,UAAU,CAAC,UAA2B,EAAiB;QAC5D,MAAM,QAAQ,GAAG,CAAC,GAAG,UAAU,CAAC,UAAU,CAAC;aACzC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;aACtC,MAAM,CAAC,CAAC,IAAI,EAAuB,EAAE,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC;QAC5D,UAAU,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;QAC9B,KAAK,MAAM,IAAI,IAAI,QAAQ;YAAE,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;QACjE,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC1F,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;YAC9B,IAAI,MAAM,CAAC,MAAM,KAAK,UAAU;gBAAE,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAC3E,CAAC;IAAA,CACD;IAED,KAAK,CAAC,YAAY,GAA+B;QAChD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,YAAY,EAAE,CAAC;QACzD,MAAM,aAAa,GAAG,MAAM,OAAO,CAAC,GAAG,CACtC,CAAC,GAAG,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC;aAC7B,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC;aACjC,GAAG,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,MAAM,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAU,CAAC,CAC9E,CAAC;QACF,MAAM,QAAQ,GAAG,IAAI,GAAG,CAAC,aAAa,CAAC,CAAC;QACxC,MAAM,QAAQ,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC;YACrC,MAAM,QAAQ,GAAG,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACvC,IAAI,CAAC,QAAQ;gBAAE,OAAO,IAAI,CAAC;YAC3B,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACzB,OAAO,EAAE,GAAG,IAAI,EAAE,GAAG,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QAAA,CAC5C,CAAC,CAAC;QACH,KAAK,MAAM,QAAQ,IAAI,QAAQ,CAAC,MAAM,EAAE;YAAE,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC;QAC9E,OAAO,QAAQ,CAAC;IAAA,CAChB;IAED,KAAK,CAAC,KAAK,GAAkB;QAC5B,MAAM,cAAc,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,CAAC,GAAG,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QACpF,KAAK,MAAM,MAAM,IAAI,cAAc,EAAE,CAAC;YACrC,IAAI,MAAM,CAAC,MAAM,KAAK,UAAU;gBAAE,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAC3E,CAAC;QACD,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC,CAAC;QACjD,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC;QAC1B,MAAM,OAAO,CAAC,GAAG,CAChB,QAAQ,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE,CAAC;YAC5B,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;gBACpB,MAAM,IAAI,CAAC,SAAS,CAAC;gBACrB,OAAO;YACR,CAAC;YACD,IAAI,CAAC,WAAW,EAAE,CAAC;YACnB,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;QAAA,CAC7B,CAAC,CACF,CAAC;IAAA,CACF;IAEO,KAAK,CAAC,YAAY,CACzB,UAA2B,EAC3B,IAAiB,EACjB,SAA8B,EACH;QAC3B,IAAI,CAAC,cAAc,IAAI,CAAC,CAAC;QACzB,IAAI,CAAC;YACJ,MAAM,SAAS,EAAE,CAAC;YAClB,OAAO,IAAI,CAAC,aAAa,CAAC,MAAM,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,EAAE,UAAU,CAAC,CAAC;QAC3E,CAAC;gBAAS,CAAC;YACV,IAAI,CAAC,cAAc,IAAI,CAAC,CAAC;YACzB,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC;QACjC,CAAC;IAAA,CACD;IAEO,KAAK,CAAC,OAAO,CAAC,EAAU,EAAE,cAA+C,EAAwB;QACxG,SAAS,CAAC;YACT,MAAM,QAAQ,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YAC3C,IAAI,QAAQ,EAAE,CAAC;gBACd,IAAI,QAAQ,CAAC,QAAQ;oBAAE,MAAM,IAAI,aAAa,CAAC,gBAAgB,EAAE,mCAAmC,EAAE,EAAE,CAAC,CAAC;gBAC1G,IAAI,QAAQ,CAAC,SAAS,EAAE,CAAC;oBACxB,MAAM,QAAQ,CAAC,SAAS,CAAC;oBACzB,SAAS;gBACV,CAAC;gBACD,OAAO,QAAQ,CAAC;YACjB,CAAC;YACD,MAAM,OAAO,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YAC7C,IAAI,OAAO;gBAAE,OAAO,OAAO,CAAC;YAC5B,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC;YAChD,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;YACtC,IAAI,CAAC;gBACJ,OAAO,MAAM,OAAO,CAAC;YACtB,CAAC;oBAAS,CAAC;gBACV,IAAI,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,EAAE,CAAC,KAAK,OAAO;oBAAE,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YAC/E,CAAC;QACF,CAAC;IAAA,CACD;IAEO,KAAK,CAAC,MAAM,CAAC,EAAU,EAAE,cAA+C,EAAwB;QACvG,MAAM,OAAO,GAAG,MAAM,cAAc,EAAE,CAAC;QACvC,IAAI,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC;YAC9B,MAAM,OAAO,CAAC,OAAO,EAAE,CAAC;YACxB,MAAM,IAAI,KAAK,CAAC,mDAAmD,CAAC,CAAC;QACtE,CAAC;QACD,IAAI,IAA6B,CAAC;QAClC,IAAI,CAAC;YACJ,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,QAAQ,EAAE,CAAC;YAC1C,IAAI,QAAQ,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC;gBACxB,MAAM,IAAI,aAAa,CACtB,iBAAiB,EACjB,4BAA4B,QAAQ,CAAC,EAAE,gCAAgC,EAAE,EAAE,CAC3E,CAAC;YACH,CAAC;YACD,IAAI,GAAG;gBACN,EAAE;gBACF,OAAO;gBACP,WAAW,EAAE,IAAI,GAAG,EAAE;gBACtB,WAAW,EAAE,GAAG,EAAE,CAAC,EAAC,CAAC;gBACrB,cAAc,EAAE,CAAC;gBACjB,KAAK,EAAE,KAAK;gBACZ,QAAQ,EAAE,KAAK;aACf,CAAC;YACF,IAAI,CAAC,WAAW,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,kBAAkB,CAAC,IAAK,EAAE,KAAK,CAAC,CAAC,CAAC;YACvF,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;YAChC,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;YAClB,OAAO,IAAI,CAAC;QACb,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,IAAI,IAAI;gBAAE,IAAI,CAAC,WAAW,EAAE,CAAC;YAC7B,IAAI,CAAC;gBACJ,MAAM,OAAO,CAAC,OAAO,EAAE,CAAC;YACzB,CAAC;YAAC,OAAO,YAAY,EAAE,CAAC;gBACvB,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC;YACxC,CAAC;YACD,MAAM,KAAK,CAAC;QACb,CAAC;IAAA,CACD;IAEO,kBAAkB,CAAC,IAAiB,EAAE,KAA4B,EAAQ;QACjF,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;YAC5B,KAAK,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC;YAClG,OAAO;QACR,CAAC;QACD,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;YAC/B,MAAM,QAAQ,GAAkB;gBAC/B,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE,EAAE,IAAI,EAAE,kBAAkB,EAAE,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAE;aACjF,CAAC;YACF,KAAK,MAAM,UAAU,IAAI,IAAI,CAAC,WAAW;gBAAE,KAAK,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QAChG,CAAC;aAAM,CAAC;YACP,KAAK,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC;QAC9F,CAAC;QACD,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC;IAAA,CAChC;IAEO,KAAK,CAAC,SAAS,CAAC,IAAiB,EAAE,KAAoB,EAAiB;QAC/E,IAAI,IAAI,CAAC,QAAQ;YAAE,OAAO;QAC1B,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;QACrB,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;QAChC,IAAI,CAAC,WAAW,EAAE,CAAC;QACnB,MAAM,WAAW,GAAG,CAAC,GAAG,IAAI,CAAC,WAAW,CAAC,CAAC;QAC1C,MAAM,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;QACxG,MAAM,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;QACxF,MAAM,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;IAAA,CAC9B;IAEO,KAAK,CAAC,kBAAkB,CAAC,IAAiB,EAA4B;QAC7E,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC;QAC/C,IAAI,QAAQ,CAAC,EAAE,KAAK,IAAI,CAAC,EAAE,EAAE,CAAC;YAC7B,MAAM,IAAI,aAAa,CAAC,iBAAiB,EAAE,mCAAmC,IAAI,CAAC,EAAE,OAAO,QAAQ,CAAC,EAAE,EAAE,CAAC,CAAC;QAC5G,CAAC;QACD,OAAO;YACN,GAAG,QAAQ;YACX,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE;YAC9B,QAAQ,EAAE,IAAI,CAAC,WAAW,CAAC,IAAI,GAAG,CAAC;YACnC,MAAM,EAAE,IAAI;SACZ,CAAC;IAAA,CACF;IAEO,aAAa,CAAC,QAAyB,EAAE,UAA2B,EAAmB;QAC9F,OAAO,EAAE,GAAG,QAAQ,EAAE,QAAQ,EAAE,UAAU,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC,EAAE,CAAC;IAAA,CACzE;IAEO,KAAK,CAAC,iBAAiB,CAAC,IAAiB,EAA4B;QAC5E,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC;QACrD,MAAM,QAAQ,GAAkB,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,kBAAkB,EAAE,QAAQ,EAAE,EAAE,CAAC;QACjG,KAAK,MAAM,UAAU,IAAI,IAAI,CAAC,WAAW;YAAE,KAAK,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QAC/F,OAAO,QAAQ,CAAC;IAAA,CAChB;IAEO,KAAK,CAAC,MAAM,CAAC,UAA2B,EAAE,IAAiB,EAAiB;QACnF,IAAI,UAAU,CAAC,YAAY,IAAI,UAAU,CAAC,KAAK,KAAK,OAAO,IAAI,UAAU,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC;YAC7F,MAAM,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;YAC9B,MAAM,IAAI,aAAa,CAAC,iBAAiB,EAAE,gDAAgD,CAAC,CAAC;QAC9F,CAAC;QACD,UAAU,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACnC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;IAAA,CACjC;IAEO,eAAe,CAAC,UAA2B,EAAE,SAAiB,EAAe;QACpF,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;YAC3C,MAAM,IAAI,aAAa,CAAC,iBAAiB,EAAE,yCAAyC,SAAS,EAAE,CAAC,CAAC;QAClG,CAAC;QACD,MAAM,IAAI,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC9C,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YAC9C,MAAM,IAAI,aAAa,CAAC,WAAW,EAAE,wBAAwB,SAAS,EAAE,CAAC,CAAC;QAC3E,CAAC;QACD,OAAO,IAAI,CAAC;IAAA,CACZ;IAEO,oBAAoB,CAAC,IAAiB,EAAQ;QACrD,KAAK,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC;IAAA,CACxF;IAEO,KAAK,CAAC,YAAY,CAAC,IAAiB,EAAiB;QAC5D,IACC,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE;YACxB,CAAC,IAAI,CAAC,KAAK;YACX,IAAI,CAAC,SAAS;YACd,IAAI,CAAC,WAAW,CAAC,IAAI,GAAG,CAAC;YACzB,IAAI,CAAC,cAAc,GAAG,CAAC;YACvB,CAAC,CAAC,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,KAAK,MAAM,CAAC,EACrD,CAAC;YACF,OAAO,IAAI,CAAC,SAAS,CAAC;QACvB,CAAC;QACD,IAAI,CAAC,WAAW,EAAE,CAAC;QACnB,IAAI,CAAC,SAAS,GAAG,CAAC,KAAK,IAAI,EAAE,CAAC;YAC7B,IAAI,CAAC;gBACJ,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;YAC9B,CAAC;oBAAS,CAAC;gBACV,IAAI,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,KAAK,IAAI;oBAAE,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAChF,CAAC;QAAA,CACD,CAAC,EAAE,CAAC;QACL,MAAM,IAAI,CAAC,SAAS,CAAC;QACrB,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE;YAAE,IAAI,CAAC,OAAO,CAAC,uBAAuB,EAAE,CAAC;IAAA,CACtE;CACD","sourcesContent":["import { randomUUID } from \"node:crypto\";\nimport type { Command, EventEnvelope, SessionMetadata, SessionSnapshot } from \"@earendil-works/pi-protocol\";\nimport type { ByteConnection, ConnectionState } from \"./connection.ts\";\nimport { PiServerError } from \"./errors.ts\";\nimport type { CreateSessionOptions, PiServerService, PiSessionRuntime, PiSessionRuntimeEvent } from \"./types.ts\";\n\ninterface LiveSession {\n\tid: string;\n\truntime: PiSessionRuntime;\n\tconnections: Set<ConnectionState>;\n\tunsubscribe: () => void;\n\toperationCount: number;\n\tready: boolean;\n\tterminal: boolean;\n\tdisposing?: Promise<void>;\n}\n\ninterface LiveSessionManagerOptions {\n\tservice: PiServerService;\n\tisClosing: () => boolean;\n\tsendMessage: (connection: ConnectionState, message: EventEnvelope) => Promise<boolean>;\n\tcloseConnection: (connection: ByteConnection) => Promise<void>;\n\tdisconnect: (connection: ConnectionState) => Promise<void>;\n\tbroadcastServerSnapshot: () => void;\n\treportError: (error: unknown) => void;\n}\n\nfunction toMetadata(snapshot: SessionSnapshot): SessionMetadata {\n\treturn {\n\t\tid: snapshot.id,\n\t\tcreatedAt: snapshot.createdAt,\n\t\tupdatedAt: snapshot.updatedAt,\n\t\tsessionName: snapshot.name,\n\t\tcwd: snapshot.cwd,\n\t};\n}\n\nexport class LiveSessionManager {\n\tprivate readonly options: LiveSessionManagerOptions;\n\tprivate readonly liveSessions = new Map<string, LiveSession>();\n\tprivate readonly openingSessions = new Map<string, Promise<LiveSession>>();\n\n\tconstructor(options: LiveSessionManagerOptions) {\n\t\tthis.options = options;\n\t}\n\n\tasync executeCommand(connection: ConnectionState, command: Command) {\n\t\tswitch (command.command) {\n\t\t\tcase \"list\":\n\t\t\t\treturn { command: \"list\" as const, sessions: await this.listMetadata() };\n\t\t\tcase \"create\": {\n\t\t\t\tconst id = randomUUID();\n\t\t\t\tconst options: CreateSessionOptions = {\n\t\t\t\t\tid,\n\t\t\t\t\tcwd: command.cwd,\n\t\t\t\t\tname: command.name,\n\t\t\t\t\tmodel: command.model,\n\t\t\t\t\tthinkingLevel: command.thinkingLevel,\n\t\t\t\t};\n\t\t\t\tconst live = await this.acquire(id, () => this.options.service.createSession(options));\n\t\t\t\tawait this.attach(connection, live);\n\t\t\t\tconst session = this.forConnection(await this.broadcastSnapshot(live), connection);\n\t\t\t\tthis.options.broadcastServerSnapshot();\n\t\t\t\treturn { command: \"create\" as const, session };\n\t\t\t}\n\t\t\tcase \"attach\": {\n\t\t\t\tconst live = await this.acquire(command.sessionId, () =>\n\t\t\t\t\tthis.options.service.openSession(command.sessionId),\n\t\t\t\t);\n\t\t\t\tawait this.attach(connection, live);\n\t\t\t\tconst session = this.forConnection(await this.broadcastSnapshot(live), connection);\n\t\t\t\tthis.options.broadcastServerSnapshot();\n\t\t\t\treturn { command: \"attach\" as const, session };\n\t\t\t}\n\t\t\tcase \"detach\": {\n\t\t\t\tconst live = this.liveSessions.get(command.sessionId);\n\t\t\t\tif (connection.sessionIds.has(command.sessionId)) {\n\t\t\t\t\tconnection.sessionIds.delete(command.sessionId);\n\t\t\t\t\tif (live) {\n\t\t\t\t\t\tlive.connections.delete(connection);\n\t\t\t\t\t\tif (live.connections.size > 0 && !live.terminal && !live.disposing) {\n\t\t\t\t\t\t\tawait this.broadcastSnapshot(live);\n\t\t\t\t\t\t}\n\t\t\t\t\t\tawait this.maybeDispose(live);\n\t\t\t\t\t}\n\t\t\t\t\tthis.options.broadcastServerSnapshot();\n\t\t\t\t}\n\t\t\t\treturn { command: \"detach\" as const, sessionId: command.sessionId };\n\t\t\t}\n\t\t\tcase \"prompt\": {\n\t\t\t\tconst live = this.requireAttached(connection, command.sessionId);\n\t\t\t\tconst session = await this.runOperation(connection, live, () =>\n\t\t\t\t\tlive.runtime.prompt({ text: command.text }),\n\t\t\t\t);\n\t\t\t\treturn { command: \"prompt\" as const, session };\n\t\t\t}\n\t\t\tcase \"steer\": {\n\t\t\t\tconst live = this.requireAttached(connection, command.sessionId);\n\t\t\t\tconst session = await this.runOperation(connection, live, () => live.runtime.steer({ text: command.text }));\n\t\t\t\treturn { command: \"steer\" as const, session };\n\t\t\t}\n\t\t\tcase \"abort\": {\n\t\t\t\tconst live = this.requireAttached(connection, command.sessionId);\n\t\t\t\tconst session = await this.runOperation(connection, live, () => live.runtime.abort());\n\t\t\t\treturn { command: \"abort\" as const, session };\n\t\t\t}\n\t\t\tcase \"set_model\": {\n\t\t\t\tconst live = this.requireAttached(connection, command.sessionId);\n\t\t\t\tconst session = await this.runOperation(connection, live, () => live.runtime.setModel(command.model));\n\t\t\t\treturn { command: \"set_model\" as const, session };\n\t\t\t}\n\t\t\tcase \"set_thinking\": {\n\t\t\t\tconst live = this.requireAttached(connection, command.sessionId);\n\t\t\t\tconst session = await this.runOperation(connection, live, () =>\n\t\t\t\t\tlive.runtime.setThinking(command.thinkingLevel),\n\t\t\t\t);\n\t\t\t\treturn { command: \"set_thinking\" as const, session };\n\t\t\t}\n\t\t}\n\t}\n\n\tasync disconnect(connection: ConnectionState): Promise<void> {\n\t\tconst sessions = [...connection.sessionIds]\n\t\t\t.map((id) => this.liveSessions.get(id))\n\t\t\t.filter((live): live is LiveSession => live !== undefined);\n\t\tconnection.sessionIds.clear();\n\t\tfor (const live of sessions) live.connections.delete(connection);\n\t\tconst results = await Promise.allSettled(sessions.map((live) => this.maybeDispose(live)));\n\t\tfor (const result of results) {\n\t\t\tif (result.status === \"rejected\") this.options.reportError(result.reason);\n\t\t}\n\t}\n\n\tasync listMetadata(): Promise<SessionMetadata[]> {\n\t\tconst stored = await this.options.service.listSessions();\n\t\tconst liveSnapshots = await Promise.all(\n\t\t\t[...this.liveSessions.values()]\n\t\t\t\t.filter((live) => !live.disposing)\n\t\t\t\t.map(async (live) => [live.id, await this.normalizedSnapshot(live)] as const),\n\t\t);\n\t\tconst liveById = new Map(liveSnapshots);\n\t\tconst metadata = stored.map((item) => {\n\t\t\tconst snapshot = liveById.get(item.id);\n\t\t\tif (!snapshot) return item;\n\t\t\tliveById.delete(item.id);\n\t\t\treturn { ...item, ...toMetadata(snapshot) };\n\t\t});\n\t\tfor (const snapshot of liveById.values()) metadata.push(toMetadata(snapshot));\n\t\treturn metadata;\n\t}\n\n\tasync close(): Promise<void> {\n\t\tconst openingResults = await Promise.allSettled([...this.openingSessions.values()]);\n\t\tfor (const result of openingResults) {\n\t\t\tif (result.status === \"rejected\") this.options.reportError(result.reason);\n\t\t}\n\t\tconst sessions = [...this.liveSessions.values()];\n\t\tthis.liveSessions.clear();\n\t\tawait Promise.all(\n\t\t\tsessions.map(async (live) => {\n\t\t\t\tif (live.disposing) {\n\t\t\t\t\tawait live.disposing;\n\t\t\t\t\treturn;\n\t\t\t\t}\n\t\t\t\tlive.unsubscribe();\n\t\t\t\tawait live.runtime.dispose();\n\t\t\t}),\n\t\t);\n\t}\n\n\tprivate async runOperation(\n\t\tconnection: ConnectionState,\n\t\tlive: LiveSession,\n\t\toperation: () => Promise<void>,\n\t): Promise<SessionSnapshot> {\n\t\tlive.operationCount += 1;\n\t\ttry {\n\t\t\tawait operation();\n\t\t\treturn this.forConnection(await this.broadcastSnapshot(live), connection);\n\t\t} finally {\n\t\t\tlive.operationCount -= 1;\n\t\t\tthis.scheduleMaybeDispose(live);\n\t\t}\n\t}\n\n\tprivate async acquire(id: string, acquireRuntime: () => Promise<PiSessionRuntime>): Promise<LiveSession> {\n\t\tfor (;;) {\n\t\t\tconst existing = this.liveSessions.get(id);\n\t\t\tif (existing) {\n\t\t\t\tif (existing.terminal) throw new PiServerError(\"session_locked\", `Session runtime is terminating: ${id}`);\n\t\t\t\tif (existing.disposing) {\n\t\t\t\t\tawait existing.disposing;\n\t\t\t\t\tcontinue;\n\t\t\t\t}\n\t\t\t\treturn existing;\n\t\t\t}\n\t\t\tconst opening = this.openingSessions.get(id);\n\t\t\tif (opening) return opening;\n\t\t\tconst pending = this.create(id, acquireRuntime);\n\t\t\tthis.openingSessions.set(id, pending);\n\t\t\ttry {\n\t\t\t\treturn await pending;\n\t\t\t} finally {\n\t\t\t\tif (this.openingSessions.get(id) === pending) this.openingSessions.delete(id);\n\t\t\t}\n\t\t}\n\t}\n\n\tprivate async create(id: string, acquireRuntime: () => Promise<PiSessionRuntime>): Promise<LiveSession> {\n\t\tconst runtime = await acquireRuntime();\n\t\tif (this.options.isClosing()) {\n\t\t\tawait runtime.dispose();\n\t\t\tthrow new Error(\"PiServer closed while acquiring a session runtime\");\n\t\t}\n\t\tlet live: LiveSession | undefined;\n\t\ttry {\n\t\t\tconst snapshot = await runtime.snapshot();\n\t\t\tif (snapshot.id !== id) {\n\t\t\t\tthrow new PiServerError(\n\t\t\t\t\t\"invalid_request\",\n\t\t\t\t\t`Service returned session ${snapshot.id} for server-assigned session ${id}`,\n\t\t\t\t);\n\t\t\t}\n\t\t\tlive = {\n\t\t\t\tid,\n\t\t\t\truntime,\n\t\t\t\tconnections: new Set(),\n\t\t\t\tunsubscribe: () => {},\n\t\t\t\toperationCount: 0,\n\t\t\t\tready: false,\n\t\t\t\tterminal: false,\n\t\t\t};\n\t\t\tlive.unsubscribe = runtime.subscribe((event) => this.handleRuntimeEvent(live!, event));\n\t\t\tthis.liveSessions.set(id, live);\n\t\t\tlive.ready = true;\n\t\t\treturn live;\n\t\t} catch (error) {\n\t\t\tif (live) live.unsubscribe();\n\t\t\ttry {\n\t\t\t\tawait runtime.dispose();\n\t\t\t} catch (disposeError) {\n\t\t\t\tthis.options.reportError(disposeError);\n\t\t\t}\n\t\t\tthrow error;\n\t\t}\n\t}\n\n\tprivate handleRuntimeEvent(live: LiveSession, event: PiSessionRuntimeEvent): void {\n\t\tif (event.type === \"error\") {\n\t\t\tvoid this.terminate(live, event.error).catch((error: unknown) => this.options.reportError(error));\n\t\t\treturn;\n\t\t}\n\t\tif (event.type === \"progress\") {\n\t\t\tconst envelope: EventEnvelope = {\n\t\t\t\ttype: \"event\",\n\t\t\t\tevent: { type: \"session_progress\", sessionId: live.id, progress: event.progress },\n\t\t\t};\n\t\t\tfor (const connection of live.connections) void this.options.sendMessage(connection, envelope);\n\t\t} else {\n\t\t\tvoid this.broadcastSnapshot(live).catch((error: unknown) => this.options.reportError(error));\n\t\t}\n\t\tthis.scheduleMaybeDispose(live);\n\t}\n\n\tprivate async terminate(live: LiveSession, error: PiServerError): Promise<void> {\n\t\tif (live.terminal) return;\n\t\tlive.terminal = true;\n\t\tthis.options.reportError(error);\n\t\tlive.unsubscribe();\n\t\tconst connections = [...live.connections];\n\t\tawait Promise.all(connections.map((connection) => this.options.closeConnection(connection.connection)));\n\t\tawait Promise.all(connections.map((connection) => this.options.disconnect(connection)));\n\t\tawait this.maybeDispose(live);\n\t}\n\n\tprivate async normalizedSnapshot(live: LiveSession): Promise<SessionSnapshot> {\n\t\tconst snapshot = await live.runtime.snapshot();\n\t\tif (snapshot.id !== live.id) {\n\t\t\tthrow new PiServerError(\"invalid_request\", `Runtime session ID changed from ${live.id} to ${snapshot.id}`);\n\t\t}\n\t\treturn {\n\t\t\t...snapshot,\n\t\t\tphase: live.runtime.getPhase(),\n\t\t\tattached: live.connections.size > 0,\n\t\t\tlocked: true,\n\t\t};\n\t}\n\n\tprivate forConnection(snapshot: SessionSnapshot, connection: ConnectionState): SessionSnapshot {\n\t\treturn { ...snapshot, attached: connection.sessionIds.has(snapshot.id) };\n\t}\n\n\tprivate async broadcastSnapshot(live: LiveSession): Promise<SessionSnapshot> {\n\t\tconst snapshot = await this.normalizedSnapshot(live);\n\t\tconst envelope: EventEnvelope = { type: \"event\", event: { type: \"session_snapshot\", snapshot } };\n\t\tfor (const connection of live.connections) void this.options.sendMessage(connection, envelope);\n\t\treturn snapshot;\n\t}\n\n\tprivate async attach(connection: ConnectionState, live: LiveSession): Promise<void> {\n\t\tif (connection.disconnected || connection.stage !== \"ready\" || connection.connection.closed) {\n\t\t\tawait this.maybeDispose(live);\n\t\t\tthrow new PiServerError(\"invalid_request\", \"Connection closed while attaching to a session\");\n\t\t}\n\t\tconnection.sessionIds.add(live.id);\n\t\tlive.connections.add(connection);\n\t}\n\n\tprivate requireAttached(connection: ConnectionState, sessionId: string): LiveSession {\n\t\tif (!connection.sessionIds.has(sessionId)) {\n\t\t\tthrow new PiServerError(\"invalid_request\", `Connection is not attached to session ${sessionId}`);\n\t\t}\n\t\tconst live = this.liveSessions.get(sessionId);\n\t\tif (!live || live.terminal || live.disposing) {\n\t\t\tthrow new PiServerError(\"not_found\", `Session is not live: ${sessionId}`);\n\t\t}\n\t\treturn live;\n\t}\n\n\tprivate scheduleMaybeDispose(live: LiveSession): void {\n\t\tvoid this.maybeDispose(live).catch((error: unknown) => this.options.reportError(error));\n\t}\n\n\tprivate async maybeDispose(live: LiveSession): Promise<void> {\n\t\tif (\n\t\t\tthis.options.isClosing() ||\n\t\t\t!live.ready ||\n\t\t\tlive.disposing ||\n\t\t\tlive.connections.size > 0 ||\n\t\t\tlive.operationCount > 0 ||\n\t\t\t(!live.terminal && live.runtime.getPhase() !== \"idle\")\n\t\t) {\n\t\t\treturn live.disposing;\n\t\t}\n\t\tlive.unsubscribe();\n\t\tlive.disposing = (async () => {\n\t\t\ttry {\n\t\t\t\tawait live.runtime.dispose();\n\t\t\t} finally {\n\t\t\t\tif (this.liveSessions.get(live.id) === live) this.liveSessions.delete(live.id);\n\t\t\t}\n\t\t})();\n\t\tawait live.disposing;\n\t\tif (!this.options.isClosing()) this.options.broadcastServerSnapshot();\n\t}\n}\n"]}

package/dist/snapshots.d.ts

@@ -0,0 +1,24 @@
+import { type EventEnvelope, type ModelMetadata, type ServerSnapshot, type SessionMetadata } from "@earendil-works/pi-protocol";
+import type { ConnectionState } from "./connection.ts";
+import type { PiServerService } from "./types.ts";
+interface ServerSnapshotPublisherOptions {
+    serverId: string;
+    service: PiServerService;
+    connections: Set<ConnectionState>;
+    isClosing: () => boolean;
+    listSessions: () => Promise<SessionMetadata[]>;
+    sendMessage: (connection: ConnectionState, message: EventEnvelope) => Promise<boolean>;
+    reportError: (error: unknown) => void;
+}
+export declare class ServerSnapshotPublisher {
+    private readonly options;
+    private revision;
+    private broadcastQueue;
+    constructor(options: ServerSnapshotPublisherOptions);
+    get currentRevision(): number;
+    get(models?: ModelMetadata[]): Promise<ServerSnapshot>;
+    broadcast(): Promise<void>;
+    private performBroadcast;
+}
+export {};
+//# sourceMappingURL=snapshots.d.ts.map

package/dist/snapshots.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"snapshots.d.ts","sourceRoot":"","sources":["../src/snapshots.ts"],"names":[],"mappings":"AAAA,OAAO,EACN,KAAK,aAAa,EAClB,KAAK,aAAa,EAElB,KAAK,cAAc,EACnB,KAAK,eAAe,EACpB,MAAM,6BAA6B,CAAC;AACrC,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAC;AACvD,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AAElD,UAAU,8BAA8B;IACvC,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,eAAe,CAAC;IACzB,WAAW,EAAE,GAAG,CAAC,eAAe,CAAC,CAAC;IAClC,SAAS,EAAE,MAAM,OAAO,CAAC;IACzB,YAAY,EAAE,MAAM,OAAO,CAAC,eAAe,EAAE,CAAC,CAAC;IAC/C,WAAW,EAAE,CAAC,UAAU,EAAE,eAAe,EAAE,OAAO,EAAE,aAAa,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC;IACvF,WAAW,EAAE,CAAC,KAAK,EAAE,OAAO,KAAK,IAAI,CAAC;CACtC;AAED,qBAAa,uBAAuB;IACnC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAiC;IACzD,OAAO,CAAC,QAAQ,CAAK;IACrB,OAAO,CAAC,cAAc,CAAoC;IAE1D,YAAY,OAAO,EAAE,8BAA8B,EAElD;IAED,IAAI,eAAe,IAAI,MAAM,CAE5B;IAEK,GAAG,CAAC,MAAM,CAAC,EAAE,aAAa,EAAE,GAAG,OAAO,CAAC,cAAc,CAAC,CAQ3D;IAED,SAAS,IAAI,OAAO,CAAC,IAAI,CAAC,CAIzB;YAEa,gBAAgB;CAY9B","sourcesContent":["import {\n\ttype EventEnvelope,\n\ttype ModelMetadata,\n\tPROTOCOL_VERSION,\n\ttype ServerSnapshot,\n\ttype SessionMetadata,\n} from \"@earendil-works/pi-protocol\";\nimport type { ConnectionState } from \"./connection.ts\";\nimport type { PiServerService } from \"./types.ts\";\n\ninterface ServerSnapshotPublisherOptions {\n\tserverId: string;\n\tservice: PiServerService;\n\tconnections: Set<ConnectionState>;\n\tisClosing: () => boolean;\n\tlistSessions: () => Promise<SessionMetadata[]>;\n\tsendMessage: (connection: ConnectionState, message: EventEnvelope) => Promise<boolean>;\n\treportError: (error: unknown) => void;\n}\n\nexport class ServerSnapshotPublisher {\n\tprivate readonly options: ServerSnapshotPublisherOptions;\n\tprivate revision = 0;\n\tprivate broadcastQueue: Promise<void> = Promise.resolve();\n\n\tconstructor(options: ServerSnapshotPublisherOptions) {\n\t\tthis.options = options;\n\t}\n\n\tget currentRevision(): number {\n\t\treturn this.revision;\n\t}\n\n\tasync get(models?: ModelMetadata[]): Promise<ServerSnapshot> {\n\t\treturn {\n\t\t\tserverId: this.options.serverId,\n\t\t\tprotocolVersion: PROTOCOL_VERSION,\n\t\t\trevision: this.revision,\n\t\t\tsessions: await this.options.listSessions(),\n\t\t\tmodels: models ?? (await this.options.service.listModels()),\n\t\t};\n\t}\n\n\tbroadcast(): Promise<void> {\n\t\tconst broadcast = this.broadcastQueue.then(() => this.performBroadcast());\n\t\tthis.broadcastQueue = broadcast.catch((error: unknown) => this.options.reportError(error));\n\t\treturn broadcast;\n\t}\n\n\tprivate async performBroadcast(): Promise<void> {\n\t\tconst readyConnections = [...this.options.connections].filter(\n\t\t\t(connection) => connection.stage === \"ready\" && !connection.disconnected,\n\t\t);\n\t\tif (readyConnections.length === 0 || this.options.isClosing()) return;\n\t\tconst revision = ++this.revision;\n\t\tconst models = await this.options.service.listModels();\n\t\tconst current = await this.get(models);\n\t\tconst snapshot: ServerSnapshot = { ...current, revision };\n\t\tconst envelope: EventEnvelope = { type: \"event\", event: { type: \"server_snapshot\", snapshot } };\n\t\tfor (const connection of readyConnections) await this.options.sendMessage(connection, envelope);\n\t}\n}\n"]}

package/dist/snapshots.js

@@ -0,0 +1,39 @@
+import { PROTOCOL_VERSION, } from "@earendil-works/pi-protocol";
+export class ServerSnapshotPublisher {
+    options;
+    revision = 0;
+    broadcastQueue = Promise.resolve();
+    constructor(options) {
+        this.options = options;
+    }
+    get currentRevision() {
+        return this.revision;
+    }
+    async get(models) {
+        return {
+            serverId: this.options.serverId,
+            protocolVersion: PROTOCOL_VERSION,
+            revision: this.revision,
+            sessions: await this.options.listSessions(),
+            models: models ?? (await this.options.service.listModels()),
+        };
+    }
+    broadcast() {
+        const broadcast = this.broadcastQueue.then(() => this.performBroadcast());
+        this.broadcastQueue = broadcast.catch((error) => this.options.reportError(error));
+        return broadcast;
+    }
+    async performBroadcast() {
+        const readyConnections = [...this.options.connections].filter((connection) => connection.stage === "ready" && !connection.disconnected);
+        if (readyConnections.length === 0 || this.options.isClosing())
+            return;
+        const revision = ++this.revision;
+        const models = await this.options.service.listModels();
+        const current = await this.get(models);
+        const snapshot = { ...current, revision };
+        const envelope = { type: "event", event: { type: "server_snapshot", snapshot } };
+        for (const connection of readyConnections)
+            await this.options.sendMessage(connection, envelope);
+    }
+}
+//# sourceMappingURL=snapshots.js.map

package/dist/snapshots.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"snapshots.js","sourceRoot":"","sources":["../src/snapshots.ts"],"names":[],"mappings":"AAAA,OAAO,EAGN,gBAAgB,GAGhB,MAAM,6BAA6B,CAAC;AAcrC,MAAM,OAAO,uBAAuB;IAClB,OAAO,CAAiC;IACjD,QAAQ,GAAG,CAAC,CAAC;IACb,cAAc,GAAkB,OAAO,CAAC,OAAO,EAAE,CAAC;IAE1D,YAAY,OAAuC,EAAE;QACpD,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IAAA,CACvB;IAED,IAAI,eAAe,GAAW;QAC7B,OAAO,IAAI,CAAC,QAAQ,CAAC;IAAA,CACrB;IAED,KAAK,CAAC,GAAG,CAAC,MAAwB,EAA2B;QAC5D,OAAO;YACN,QAAQ,EAAE,IAAI,CAAC,OAAO,CAAC,QAAQ;YAC/B,eAAe,EAAE,gBAAgB;YACjC,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,QAAQ,EAAE,MAAM,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE;YAC3C,MAAM,EAAE,MAAM,IAAI,CAAC,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;SAC3D,CAAC;IAAA,CACF;IAED,SAAS,GAAkB;QAC1B,MAAM,SAAS,GAAG,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,gBAAgB,EAAE,CAAC,CAAC;QAC1E,IAAI,CAAC,cAAc,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC;QAC3F,OAAO,SAAS,CAAC;IAAA,CACjB;IAEO,KAAK,CAAC,gBAAgB,GAAkB;QAC/C,MAAM,gBAAgB,GAAG,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,MAAM,CAC5D,CAAC,UAAU,EAAE,EAAE,CAAC,UAAU,CAAC,KAAK,KAAK,OAAO,IAAI,CAAC,UAAU,CAAC,YAAY,CACxE,CAAC;QACF,IAAI,gBAAgB,CAAC,MAAM,KAAK,CAAC,IAAI,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE;YAAE,OAAO;QACtE,MAAM,QAAQ,GAAG,EAAE,IAAI,CAAC,QAAQ,CAAC;QACjC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;QACvD,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACvC,MAAM,QAAQ,GAAmB,EAAE,GAAG,OAAO,EAAE,QAAQ,EAAE,CAAC;QAC1D,MAAM,QAAQ,GAAkB,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,iBAAiB,EAAE,QAAQ,EAAE,EAAE,CAAC;QAChG,KAAK,MAAM,UAAU,IAAI,gBAAgB;YAAE,MAAM,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;IAAA,CAChG;CACD","sourcesContent":["import {\n\ttype EventEnvelope,\n\ttype ModelMetadata,\n\tPROTOCOL_VERSION,\n\ttype ServerSnapshot,\n\ttype SessionMetadata,\n} from \"@earendil-works/pi-protocol\";\nimport type { ConnectionState } from \"./connection.ts\";\nimport type { PiServerService } from \"./types.ts\";\n\ninterface ServerSnapshotPublisherOptions {\n\tserverId: string;\n\tservice: PiServerService;\n\tconnections: Set<ConnectionState>;\n\tisClosing: () => boolean;\n\tlistSessions: () => Promise<SessionMetadata[]>;\n\tsendMessage: (connection: ConnectionState, message: EventEnvelope) => Promise<boolean>;\n\treportError: (error: unknown) => void;\n}\n\nexport class ServerSnapshotPublisher {\n\tprivate readonly options: ServerSnapshotPublisherOptions;\n\tprivate revision = 0;\n\tprivate broadcastQueue: Promise<void> = Promise.resolve();\n\n\tconstructor(options: ServerSnapshotPublisherOptions) {\n\t\tthis.options = options;\n\t}\n\n\tget currentRevision(): number {\n\t\treturn this.revision;\n\t}\n\n\tasync get(models?: ModelMetadata[]): Promise<ServerSnapshot> {\n\t\treturn {\n\t\t\tserverId: this.options.serverId,\n\t\t\tprotocolVersion: PROTOCOL_VERSION,\n\t\t\trevision: this.revision,\n\t\t\tsessions: await this.options.listSessions(),\n\t\t\tmodels: models ?? (await this.options.service.listModels()),\n\t\t};\n\t}\n\n\tbroadcast(): Promise<void> {\n\t\tconst broadcast = this.broadcastQueue.then(() => this.performBroadcast());\n\t\tthis.broadcastQueue = broadcast.catch((error: unknown) => this.options.reportError(error));\n\t\treturn broadcast;\n\t}\n\n\tprivate async performBroadcast(): Promise<void> {\n\t\tconst readyConnections = [...this.options.connections].filter(\n\t\t\t(connection) => connection.stage === \"ready\" && !connection.disconnected,\n\t\t);\n\t\tif (readyConnections.length === 0 || this.options.isClosing()) return;\n\t\tconst revision = ++this.revision;\n\t\tconst models = await this.options.service.listModels();\n\t\tconst current = await this.get(models);\n\t\tconst snapshot: ServerSnapshot = { ...current, revision };\n\t\tconst envelope: EventEnvelope = { type: \"event\", event: { type: \"server_snapshot\", snapshot } };\n\t\tfor (const connection of readyConnections) await this.options.sendMessage(connection, envelope);\n\t}\n}\n"]}

package/dist/testing/client.d.ts

@@ -0,0 +1,31 @@
+import { type ClientMessage, type Command, type ResponseEnvelope, type ServerMessage } from "@earendil-works/pi-protocol";
+export interface WireChannel {
+    send(chunk: Uint8Array): Promise<void>;
+    sendFragmented(chunk: Uint8Array, splitAt: number): Promise<void>;
+    close(): Promise<void>;
+}
+export declare class ProtocolTestClient {
+    readonly messages: ServerMessage[];
+    private readonly channel;
+    private readonly decoder;
+    private readonly waiters;
+    private readonly closedDeferred;
+    private requestSequence;
+    private closedValue;
+    constructor(channel: WireChannel);
+    get closed(): boolean;
+    hello(version?: number): Promise<ServerMessage>;
+    request(command: Command, id?: string): Promise<ResponseEnvelope>;
+    sendMessage(message: ClientMessage): Promise<void>;
+    sendBytes(chunk: Uint8Array): Promise<void>;
+    sendFragmentedMessage(message: ClientMessage, splitAt: number): Promise<void>;
+    next(predicate: (message: ServerMessage) => boolean): Promise<ServerMessage>;
+    nextFrom(index: number, predicate: (message: ServerMessage) => boolean): Promise<ServerMessage>;
+    waitForClose(): Promise<void>;
+    close(): Promise<void>;
+    receive(chunk: Uint8Array): void;
+    markClosed(): void;
+    fail(error: Error): void;
+}
+export declare function connectUnixTestClient(path: string): Promise<ProtocolTestClient>;
+//# sourceMappingURL=client.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../../src/testing/client.ts"],"names":[],"mappings":"AAEA,OAAO,EACN,KAAK,aAAa,EAClB,KAAK,OAAO,EAGZ,KAAK,gBAAgB,EACrB,KAAK,aAAa,EAElB,MAAM,6BAA6B,CAAC;AASrC,MAAM,WAAW,WAAW;IAC3B,IAAI,CAAC,KAAK,EAAE,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACvC,cAAc,CAAC,KAAK,EAAE,UAAU,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAClE,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;CACvB;AAED,qBAAa,kBAAkB;IAC9B,QAAQ,CAAC,QAAQ,EAAE,aAAa,EAAE,CAAM;IACxC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAc;IACtC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAA8B;IACtD,OAAO,CAAC,QAAQ,CAAC,OAAO,CAA4B;IACpD,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAwB;IACvD,OAAO,CAAC,eAAe,CAAK;IAC5B,OAAO,CAAC,WAAW,CAAS;IAE5B,YAAY,OAAO,EAAE,WAAW,EAE/B;IAED,IAAI,MAAM,IAAI,OAAO,CAEpB;IAED,KAAK,CAAC,OAAO,GAAE,MAAyB,GAAG,OAAO,CAAC,aAAa,CAAC,CAIhE;IAEK,OAAO,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,SAAsC,GAAG,OAAO,CAAC,gBAAgB,CAAC,CAMnG;IAED,WAAW,CAAC,OAAO,EAAE,aAAa,GAAG,OAAO,CAAC,IAAI,CAAC,CAEjD;IAED,SAAS,CAAC,KAAK,EAAE,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC,CAE1C;IAED,qBAAqB,CAAC,OAAO,EAAE,aAAa,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAE5E;IAED,IAAI,CAAC,SAAS,EAAE,CAAC,OAAO,EAAE,aAAa,KAAK,OAAO,GAAG,OAAO,CAAC,aAAa,CAAC,CAE3E;IAED,QAAQ,CAAC,KAAK,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,OAAO,EAAE,aAAa,KAAK,OAAO,GAAG,OAAO,CAAC,aAAa,CAAC,CAK9F;IAED,YAAY,IAAI,OAAO,CAAC,IAAI,CAAC,CAE5B;IAED,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAErB;IAED,OAAO,CAAC,KAAK,EAAE,UAAU,GAAG,IAAI,CAa/B;IAED,UAAU,IAAI,IAAI,CAKjB;IAED,IAAI,CAAC,KAAK,EAAE,KAAK,GAAG,IAAI,CAGvB;CACD;AAED,wBAAsB,qBAAqB,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,kBAAkB,CAAC,CAsBrF","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"]}