@piecemaker-legal/piecemaker 1.0.0

package/dist-server/server/modules/websocket/services/websocket-writer.service.js

@@ -0,0 +1,32 @@
+import { WS_OPEN_STATE } from '../../../modules/websocket/services/websocket-state.service.js';
+/**
+ * Thin transport adapter that gives WebSocket connections the same interface as
+ * SSE writers used by API routes (`send`, `setSessionId`, `getSessionId`).
+ */
+export class WebSocketWriter {
+    ws;
+    sessionId;
+    userId;
+    isWebSocketWriter;
+    constructor(ws, userId = null) {
+        this.ws = ws;
+        this.sessionId = null;
+        this.userId = userId;
+        this.isWebSocketWriter = true;
+    }
+    send(data) {
+        if (this.ws.readyState === WS_OPEN_STATE) {
+            this.ws.send(JSON.stringify(data));
+        }
+    }
+    updateWebSocket(newRawWs) {
+        this.ws = newRawWs;
+    }
+    setSessionId(sessionId) {
+        this.sessionId = sessionId;
+    }
+    getSessionId() {
+        return this.sessionId;
+    }
+}
+//# sourceMappingURL=websocket-writer.service.js.map

package/dist-server/server/modules/websocket/services/websocket-writer.service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"websocket-writer.service.js","sourceRoot":"","sources":["../../../../../server/modules/websocket/services/websocket-writer.service.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,yDAAyD,CAAC;AAGxF;;;GAGG;AACH,MAAM,OAAO,eAAe;IAC1B,EAAE,CAA2B;IAC7B,SAAS,CAAgB;IACzB,MAAM,CAAyB;IAC/B,iBAAiB,CAAU;IAE3B,YAAY,EAA4B,EAAE,SAAiC,IAAI;QAC7E,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACb,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QACtB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,iBAAiB,GAAG,IAAI,CAAC;IAChC,CAAC;IAED,IAAI,CAAC,IAAa;QAChB,IAAI,IAAI,CAAC,EAAE,CAAC,UAAU,KAAK,aAAa,EAAE,CAAC;YACzC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC;QACrC,CAAC;IACH,CAAC;IAED,eAAe,CAAC,QAAkC;QAChD,IAAI,CAAC,EAAE,GAAG,QAAQ,CAAC;IACrB,CAAC;IAED,YAAY,CAAC,SAAiB;QAC5B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC7B,CAAC;IAED,YAAY;QACV,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;CACF"}

package/dist-server/server/modules/websocket/tests/chat-attachment-filter.test.js

@@ -0,0 +1,40 @@
+import assert from 'node:assert/strict';
+import os from 'node:os';
+import path from 'node:path';
+import test from 'node:test';
+import { filterAttachmentsToUploadStore, filterImagesToUploadStore, } from '../../../modules/websocket/services/chat-websocket.service.js';
+const STORE = path.join(os.tmpdir(), 'cloudcli-assets-store');
+test('images inside the upload store pass through', () => {
+    const inside = path.join(STORE, 'shot.png');
+    const result = filterImagesToUploadStore([{ path: inside, name: 'shot.png', mimeType: 'image/png' }], STORE);
+    assert.equal(result.length, 1);
+    assert.equal(result[0].path, inside);
+});
+test('bare filenames are anchored inside the store', () => {
+    const result = filterImagesToUploadStore(['shot.png'], STORE);
+    assert.equal(result.length, 1);
+});
+test('paths outside the store, traversal, and subdirs are dropped', () => {
+    const result = filterImagesToUploadStore([
+        { path: 'C:/Users/victim/.ssh/id_rsa' },
+        { path: '/etc/passwd' },
+        { path: '../outside.png' },
+        { path: path.join(STORE, '..', 'escaped.png') },
+        { path: path.join(STORE, 'nested', 'deep.png') },
+        { path: STORE }, // the store folder itself is not a file
+    ], STORE);
+    assert.deepEqual(result, []);
+});
+test('malformed payloads yield no attachments', () => {
+    assert.deepEqual(filterImagesToUploadStore(undefined, STORE), []);
+    assert.deepEqual(filterImagesToUploadStore('nope', STORE), []);
+    assert.deepEqual(filterImagesToUploadStore([{ name: 'no-path' }, 42], STORE), []);
+});
+test('general files inside the upload store preserve their metadata', () => {
+    const inside = path.join(STORE, 'brief.pdf');
+    const result = filterAttachmentsToUploadStore([{ path: inside, name: 'brief.pdf', mimeType: 'application/pdf', size: 2048 }], STORE);
+    assert.deepEqual(result, [
+        { path: inside, name: 'brief.pdf', mimeType: 'application/pdf', size: 2048 },
+    ]);
+});
+//# sourceMappingURL=chat-attachment-filter.test.js.map

package/dist-server/server/modules/websocket/tests/chat-attachment-filter.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chat-attachment-filter.test.js","sourceRoot":"","sources":["../../../../../server/modules/websocket/tests/chat-attachment-filter.test.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,oBAAoB,CAAC;AACxC,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,IAAI,MAAM,WAAW,CAAC;AAE7B,OAAO,EACL,8BAA8B,EAC9B,yBAAyB,GAC1B,MAAM,wDAAwD,CAAC;AAEhE,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,uBAAuB,CAAC,CAAC;AAE9D,IAAI,CAAC,6CAA6C,EAAE,GAAG,EAAE;IACvD,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;IAC5C,MAAM,MAAM,GAAG,yBAAyB,CACtC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,WAAW,EAAE,CAAC,EAC3D,KAAK,CACN,CAAC;IACF,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;IAC/B,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;AACvC,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,8CAA8C,EAAE,GAAG,EAAE;IACxD,MAAM,MAAM,GAAG,yBAAyB,CAAC,CAAC,UAAU,CAAC,EAAE,KAAK,CAAC,CAAC;IAC9D,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;AACjC,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,6DAA6D,EAAE,GAAG,EAAE;IACvE,MAAM,MAAM,GAAG,yBAAyB,CACtC;QACE,EAAE,IAAI,EAAE,6BAA6B,EAAE;QACvC,EAAE,IAAI,EAAE,aAAa,EAAE;QACvB,EAAE,IAAI,EAAE,gBAAgB,EAAE;QAC1B,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,EAAE,aAAa,CAAC,EAAE;QAC/C,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,QAAQ,EAAE,UAAU,CAAC,EAAE;QAChD,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,wCAAwC;KAC1D,EACD,KAAK,CACN,CAAC;IACF,MAAM,CAAC,SAAS,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;AAC/B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,yCAAyC,EAAE,GAAG,EAAE;IACnD,MAAM,CAAC,SAAS,CAAC,yBAAyB,CAAC,SAAS,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC;IAClE,MAAM,CAAC,SAAS,CAAC,yBAAyB,CAAC,MAAM,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC;IAC/D,MAAM,CAAC,SAAS,CAAC,yBAAyB,CAAC,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC;AACpF,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,+DAA+D,EAAE,GAAG,EAAE;IACzE,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;IAC7C,MAAM,MAAM,GAAG,8BAA8B,CAC3C,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,QAAQ,EAAE,iBAAiB,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,EAC9E,KAAK,CACN,CAAC;IAEF,MAAM,CAAC,SAAS,CAAC,MAAM,EAAE;QACvB,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,QAAQ,EAAE,iBAAiB,EAAE,IAAI,EAAE,IAAI,EAAE;KAC7E,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist-server/server/modules/websocket/tests/chat-edit-send.test.js

@@ -0,0 +1,263 @@
+import assert from 'node:assert/strict';
+import { EventEmitter } from 'node:events';
+import { mkdtemp, rm, writeFile } from 'node:fs/promises';
+import os from 'node:os';
+import path from 'node:path';
+import test from 'node:test';
+import { closeConnection, initializeDatabase, sessionsDb } from '../../../modules/database/index.js';
+import { sessionsService } from '../../../modules/providers/index.js';
+import { handleChatConnection } from '../../../modules/websocket/services/chat-websocket.service.js';
+import { chatRunRegistry } from '../../../modules/websocket/services/chat-run-registry.service.js';
+import { connectedClients } from '../../../modules/websocket/services/websocket-state.service.js';
+const SESSION_ID = 'edit-session';
+function createFakeSocket() {
+    const socket = new EventEmitter();
+    socket.readyState = 1;
+    socket.frames = [];
+    socket.send = (data) => socket.frames.push(JSON.parse(data));
+    return socket;
+}
+/** Two turns and an edit of the second, which is the transcript the gateway reads. */
+const TRANSCRIPT_ROWS = [
+    {
+        type: 'user', uuid: 'e-u1', parentUuid: null, sessionId: SESSION_ID,
+        timestamp: '2026-08-23T10:00:00.000Z',
+        message: { role: 'user', content: [{ type: 'text', text: 'first' }] },
+    },
+    {
+        type: 'assistant', uuid: 'e-a1', parentUuid: 'e-u1', sessionId: SESSION_ID,
+        timestamp: '2026-08-23T10:00:01.000Z',
+        message: { role: 'assistant', model: 'claude-opus-5', content: [{ type: 'text', text: 'reply' }] },
+    },
+    {
+        type: 'user', uuid: 'e-u2', parentUuid: 'e-a1', sessionId: SESSION_ID,
+        timestamp: '2026-08-23T10:00:02.000Z',
+        message: { role: 'user', content: [{ type: 'text', text: 'second' }] },
+    },
+];
+/**
+ * The same two turns as a Codex rollout. Codex rows carry no id of their own,
+ * so the edit anchor is the enclosing turn — which is also the unit its fork
+ * endpoint cuts at.
+ */
+const CODEX_TRANSCRIPT_ROWS = [
+    { type: 'event_msg', payload: { type: 'task_started', turn_id: 'turn-a' } },
+    { type: 'event_msg', payload: { type: 'user_message', message: 'first' } },
+    { type: 'event_msg', payload: { type: 'task_complete', turn_id: 'turn-a' } },
+    { type: 'event_msg', payload: { type: 'task_started', turn_id: 'turn-b' } },
+    { type: 'event_msg', payload: { type: 'user_message', message: 'second' } },
+    { type: 'event_msg', payload: { type: 'task_complete', turn_id: 'turn-b' } },
+];
+/**
+ * Set by a test that needs a run to still be in flight when the next frame
+ * arrives; the stub runtime returns immediately otherwise. Released when the
+ * test ends, so no stub run is left pending for the rest of the file.
+ */
+let holdRun = null;
+let releaseHeldRun = null;
+function holdTheNextRun() {
+    holdRun = new Promise((resolve) => { releaseHeldRun = resolve; });
+}
+async function withGateway(provider, runTest, rows = TRANSCRIPT_ROWS) {
+    const previousDatabasePath = process.env.DATABASE_PATH;
+    const tempDirectory = await mkdtemp(path.join(os.tmpdir(), 'chat-edit-send-'));
+    const transcriptPath = path.join(tempDirectory, `${SESSION_ID}.jsonl`);
+    await writeFile(transcriptPath, `${rows.map((row) => JSON.stringify(row)).join('\n')}\n`, 'utf8');
+    closeConnection();
+    process.env.DATABASE_PATH = path.join(tempDirectory, 'auth.db');
+    await initializeDatabase();
+    const runs = [];
+    const socket = createFakeSocket();
+    try {
+        const now = new Date().toISOString();
+        sessionsDb.createSession(SESSION_ID, provider, tempDirectory, 'Edit session', now, now, transcriptPath);
+        handleChatConnection(socket, { user: { id: 1 } }, {
+            runtime: {
+                hasRuntime: () => true,
+                run: async (runProvider, command, options) => {
+                    runs.push({ provider: runProvider, command, options });
+                    if (holdRun) {
+                        await holdRun;
+                    }
+                },
+            },
+        });
+        await runTest({ socket, runs });
+    }
+    finally {
+        releaseHeldRun?.();
+        releaseHeldRun = null;
+        holdRun = null;
+        connectedClients.clear();
+        chatRunRegistry.clearAll();
+        closeConnection();
+        if (previousDatabasePath === undefined) {
+            delete process.env.DATABASE_PATH;
+        }
+        else {
+            process.env.DATABASE_PATH = previousDatabasePath;
+        }
+        await rm(tempDirectory, { recursive: true, force: true });
+    }
+}
+/** The handler is async and the socket listener does not await it. */
+const settle = () => new Promise((resolve) => { setTimeout(resolve, 30); });
+test('an edit resumes through the turn before the one being replaced', async () => {
+    await withGateway('claude', async ({ socket, runs }) => {
+        socket.emit('message', JSON.stringify({
+            type: 'chat.edit-send',
+            sessionId: SESSION_ID,
+            anchorId: 'e-u2',
+            content: 'a better second prompt',
+        }));
+        await settle();
+        assert.equal(runs.length, 1);
+        assert.equal(runs[0].command, 'a better second prompt');
+        // Inclusive of the row it names, so this is the last turn KEPT.
+        assert.equal(runs[0].options.resumeAnchorId, 'e-a1');
+        assert.equal(runs[0].options.resumeFromScratch, false);
+    });
+});
+test('editing the first prompt starts the conversation over', async () => {
+    await withGateway('claude', async ({ socket, runs }) => {
+        socket.emit('message', JSON.stringify({
+            type: 'chat.edit-send',
+            sessionId: SESSION_ID,
+            anchorId: 'e-u1',
+            content: 'a better first prompt',
+        }));
+        await settle();
+        assert.equal(runs.length, 1);
+        assert.equal(runs[0].options.resumeAnchorId, undefined);
+        assert.equal(runs[0].options.resumeFromScratch, true);
+    });
+});
+test('every subscribed client is told to drop the superseded turns', async () => {
+    await withGateway('claude', async ({ socket }) => {
+        socket.emit('message', JSON.stringify({
+            type: 'chat.edit-send',
+            sessionId: SESSION_ID,
+            anchorId: 'e-u2',
+            content: 'replacement',
+        }));
+        await settle();
+        const truncation = socket.frames.find((frame) => frame.kind === 'history_truncated');
+        assert.ok(truncation, 'a history_truncated frame is emitted');
+        assert.equal(truncation?.anchorId, 'e-u2');
+        assert.equal(truncation?.sessionId, SESSION_ID);
+        // Sequenced like every other run event, so a reconnecting tab replays it.
+        assert.equal(typeof truncation?.seq, 'number');
+    });
+});
+test('an edit without an anchor is refused', async () => {
+    await withGateway('claude', async ({ socket, runs }) => {
+        socket.emit('message', JSON.stringify({
+            type: 'chat.edit-send',
+            sessionId: SESSION_ID,
+            content: 'no anchor',
+        }));
+        await settle();
+        assert.equal(runs.length, 0);
+        assert.equal(socket.frames.at(-1)?.code, 'ANCHOR_REQUIRED');
+    });
+});
+test('an anchor the transcript does not hold is refused', async () => {
+    await withGateway('claude', async ({ socket, runs }) => {
+        socket.emit('message', JSON.stringify({
+            type: 'chat.edit-send',
+            sessionId: SESSION_ID,
+            anchorId: 'not-in-transcript',
+            content: 'replacement',
+        }));
+        await settle();
+        assert.equal(runs.length, 0);
+        assert.equal(socket.frames.at(-1)?.code, 'ANCHOR_NOT_FOUND');
+    });
+});
+test('a provider that cannot re-run from a point is refused rather than sending a new message', async () => {
+    await withGateway('cursor', async ({ socket, runs }) => {
+        socket.emit('message', JSON.stringify({
+            type: 'chat.edit-send',
+            sessionId: SESSION_ID,
+            anchorId: 'e-u2',
+            content: 'replacement',
+        }));
+        await settle();
+        assert.equal(runs.length, 0);
+        assert.equal(socket.frames.at(-1)?.code, 'EDIT_NOT_SUPPORTED');
+    });
+});
+test('a refused send never rewinds the conversation', async () => {
+    await withGateway('codex', async ({ socket, runs }) => {
+        // The rewind moves the session onto a different provider transcript and
+        // cannot be undone, so a send the gateway is about to refuse must not
+        // reach it. A run already in flight is the realistic way that happens: a
+        // scheduled message, or another device.
+        const realRewind = sessionsService.rewindSessionForEdit;
+        let rewound = false;
+        sessionsService.rewindSessionForEdit = async () => { rewound = true; };
+        holdTheNextRun();
+        try {
+            socket.emit('message', JSON.stringify({
+                type: 'chat.send',
+                sessionId: SESSION_ID,
+                content: 'a turn that is already running',
+            }));
+            await settle();
+            assert.equal(runs.length, 1);
+            socket.emit('message', JSON.stringify({
+                type: 'chat.edit-send',
+                sessionId: SESSION_ID,
+                anchorId: 'turn-b',
+                content: 'an edit that arrives too late',
+            }));
+            await settle();
+        }
+        finally {
+            sessionsService.rewindSessionForEdit = realRewind;
+        }
+        assert.equal(rewound, false);
+        assert.equal(runs.length, 1);
+        assert.equal(socket.frames.at(-1)?.code, 'RUN_IN_PROGRESS');
+    }, CODEX_TRANSCRIPT_ROWS);
+});
+test('a provider that has to branch to rewind is rewound before the run, not during it', async () => {
+    await withGateway('codex', async ({ socket, runs }) => {
+        // The rewind itself belongs to the provider and is covered there; what
+        // this asserts is the gateway's half — that a provider which reports it
+        // rewound gets an ordinary run instead of one carrying a resume anchor its
+        // runtime would not know what to do with.
+        const realRewind = sessionsService.rewindSessionForEdit;
+        const rewindCalls = [];
+        let truncatedBeforeRewind = false;
+        sessionsService.rewindSessionForEdit = async (sessionId, keepThroughId) => {
+            truncatedBeforeRewind = socket.frames.some((frame) => frame.kind === 'history_truncated');
+            rewindCalls.push({ sessionId, keepThroughId });
+        };
+        try {
+            socket.emit('message', JSON.stringify({
+                type: 'chat.edit-send',
+                sessionId: SESSION_ID,
+                anchorId: 'turn-b',
+                content: 'a better second prompt',
+            }));
+            await settle();
+        }
+        finally {
+            sessionsService.rewindSessionForEdit = realRewind;
+        }
+        // Resolved from the real Codex rollout: the turn before the edited one.
+        assert.deepEqual(rewindCalls, [{ sessionId: SESSION_ID, keepThroughId: 'turn-a' }]);
+        assert.equal(runs.length, 1);
+        assert.equal(runs[0].command, 'a better second prompt');
+        assert.equal(runs[0].options.resumeAnchorId, undefined);
+        assert.equal(runs[0].options.resumeFromScratch, undefined);
+        // Clients still hear about the cut; how it was made is the server's
+        // business. And they hear about it first: a rewind that branches waits on
+        // a spawned process, and holding the frame until it returned left the
+        // replaced message on screen for about a second.
+        assert.ok(socket.frames.some((frame) => frame.kind === 'history_truncated'));
+        assert.equal(truncatedBeforeRewind, true);
+    }, CODEX_TRANSCRIPT_ROWS);
+});
+//# sourceMappingURL=chat-edit-send.test.js.map

package/dist-server/server/modules/websocket/tests/chat-edit-send.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chat-edit-send.test.js","sourceRoot":"","sources":["../../../../../server/modules/websocket/tests/chat-edit-send.test.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,oBAAoB,CAAC;AACxC,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAC3C,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAC1D,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,IAAI,MAAM,WAAW,CAAC;AAE7B,OAAO,EAAE,eAAe,EAAE,kBAAkB,EAAE,UAAU,EAAE,MAAM,6BAA6B,CAAC;AAC9F,OAAO,EAAE,eAAe,EAAE,MAAM,8BAA8B,CAAC;AAC/D,OAAO,EAAE,oBAAoB,EAAE,MAAM,wDAAwD,CAAC;AAC9F,OAAO,EAAE,eAAe,EAAE,MAAM,2DAA2D,CAAC;AAC5F,OAAO,EAAE,gBAAgB,EAAE,MAAM,yDAAyD,CAAC;AAE3F,MAAM,UAAU,GAAG,cAAc,CAAC;AAElC,SAAS,gBAAgB;IACvB,MAAM,MAAM,GAAG,IAAI,YAAY,EAI9B,CAAC;IACF,MAAM,CAAC,UAAU,GAAG,CAAC,CAAC;IACtB,MAAM,CAAC,MAAM,GAAG,EAAE,CAAC;IACnB,MAAM,CAAC,IAAI,GAAG,CAAC,IAAY,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAA4B,CAAC,CAAC;IAChG,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,sFAAsF;AACtF,MAAM,eAAe,GAAG;IACtB;QACE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,UAAU,EAAE,IAAI,EAAE,SAAS,EAAE,UAAU;QACnE,SAAS,EAAE,0BAA0B;QACrC,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE;KACtE;IACD;QACE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,SAAS,EAAE,UAAU;QAC1E,SAAS,EAAE,0BAA0B;QACrC,OAAO,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,KAAK,EAAE,eAAe,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE;KACnG;IACD;QACE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,SAAS,EAAE,UAAU;QACrE,SAAS,EAAE,0BAA0B;QACrC,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,EAAE;KACvE;CACF,CAAC;AAEF;;;;GAIG;AACH,MAAM,qBAAqB,GAAG;IAC5B,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,cAAc,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE;IAC3E,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,cAAc,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE;IAC1E,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,eAAe,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE;IAC5E,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,cAAc,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE;IAC3E,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,cAAc,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE;IAC3E,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,eAAe,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE;CAC7E,CAAC;AAIF;;;;GAIG;AACH,IAAI,OAAO,GAAyB,IAAI,CAAC;AACzC,IAAI,cAAc,GAAwB,IAAI,CAAC;AAE/C,SAAS,cAAc;IACrB,OAAO,GAAG,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE,GAAG,cAAc,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;AAC1E,CAAC;AAED,KAAK,UAAU,WAAW,CACxB,QAAgB,EAChB,OAGmB,EACnB,OAAkB,eAAe;IAEjC,MAAM,oBAAoB,GAAG,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC;IACvD,MAAM,aAAa,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,iBAAiB,CAAC,CAAC,CAAC;IAC/E,MAAM,cAAc,GAAG,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,GAAG,UAAU,QAAQ,CAAC,CAAC;IACvE,MAAM,SAAS,CAAC,cAAc,EAAE,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IAElG,eAAe,EAAE,CAAC;IAClB,OAAO,CAAC,GAAG,CAAC,aAAa,GAAG,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,SAAS,CAAC,CAAC;IAChE,MAAM,kBAAkB,EAAE,CAAC;IAE3B,MAAM,IAAI,GAAc,EAAE,CAAC;IAC3B,MAAM,MAAM,GAAG,gBAAgB,EAAE,CAAC;IAElC,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QACrC,UAAU,CAAC,aAAa,CAAC,UAAU,EAAE,QAAQ,EAAE,aAAa,EAAE,cAAc,EAAE,GAAG,EAAE,GAAG,EAAE,cAAc,CAAC,CAAC;QAExG,oBAAoB,CAClB,MAAe,EACf,EAAE,IAAI,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAW,EAC5B;YACE,OAAO,EAAE;gBACP,UAAU,EAAE,GAAG,EAAE,CAAC,IAAI;gBACtB,GAAG,EAAE,KAAK,EAAE,WAAmB,EAAE,OAAe,EAAE,OAAgC,EAAE,EAAE;oBACpF,IAAI,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,WAAW,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;oBACvD,IAAI,OAAO,EAAE,CAAC;wBACZ,MAAM,OAAO,CAAC;oBAChB,CAAC;gBACH,CAAC;aACO;SACX,CACF,CAAC;QAEF,MAAM,OAAO,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;IAClC,CAAC;YAAS,CAAC;QACT,cAAc,EAAE,EAAE,CAAC;QACnB,cAAc,GAAG,IAAI,CAAC;QACtB,OAAO,GAAG,IAAI,CAAC;QACf,gBAAgB,CAAC,KAAK,EAAE,CAAC;QACzB,eAAe,CAAC,QAAQ,EAAE,CAAC;QAC3B,eAAe,EAAE,CAAC;QAClB,IAAI,oBAAoB,KAAK,SAAS,EAAE,CAAC;YACvC,OAAO,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC;QACnC,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,GAAG,CAAC,aAAa,GAAG,oBAAoB,CAAC;QACnD,CAAC;QACD,MAAM,EAAE,CAAC,aAAa,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAC5D,CAAC;AACH,CAAC;AAED,sEAAsE;AACtE,MAAM,MAAM,GAAG,GAAG,EAAE,CAAC,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,GAAG,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AAE5E,IAAI,CAAC,gEAAgE,EAAE,KAAK,IAAI,EAAE;IAChF,MAAM,WAAW,CAAC,QAAQ,EAAE,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE;QACrD,MAAM,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC;YACpC,IAAI,EAAE,gBAAgB;YACtB,SAAS,EAAE,UAAU;YACrB,QAAQ,EAAE,MAAM;YAChB,OAAO,EAAE,wBAAwB;SAClC,CAAC,CAAC,CAAC;QACJ,MAAM,MAAM,EAAE,CAAC;QAEf,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QAC7B,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,wBAAwB,CAAC,CAAC;QACxD,gEAAgE;QAChE,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,cAAc,EAAE,MAAM,CAAC,CAAC;QACrD,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,iBAAiB,EAAE,KAAK,CAAC,CAAC;IACzD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;IACvE,MAAM,WAAW,CAAC,QAAQ,EAAE,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE;QACrD,MAAM,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC;YACpC,IAAI,EAAE,gBAAgB;YACtB,SAAS,EAAE,UAAU;YACrB,QAAQ,EAAE,MAAM;YAChB,OAAO,EAAE,uBAAuB;SACjC,CAAC,CAAC,CAAC;QACJ,MAAM,MAAM,EAAE,CAAC;QAEf,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QAC7B,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,cAAc,EAAE,SAAS,CAAC,CAAC;QACxD,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,iBAAiB,EAAE,IAAI,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;IAC9E,MAAM,WAAW,CAAC,QAAQ,EAAE,KAAK,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE;QAC/C,MAAM,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC;YACpC,IAAI,EAAE,gBAAgB;YACtB,SAAS,EAAE,UAAU;YACrB,QAAQ,EAAE,MAAM;YAChB,OAAO,EAAE,aAAa;SACvB,CAAC,CAAC,CAAC;QACJ,MAAM,MAAM,EAAE,CAAC;QAEf,MAAM,UAAU,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,mBAAmB,CAAC,CAAC;QACrF,MAAM,CAAC,EAAE,CAAC,UAAU,EAAE,sCAAsC,CAAC,CAAC;QAC9D,MAAM,CAAC,KAAK,CAAC,UAAU,EAAE,QAAQ,EAAE,MAAM,CAAC,CAAC;QAC3C,MAAM,CAAC,KAAK,CAAC,UAAU,EAAE,SAAS,EAAE,UAAU,CAAC,CAAC;QAChD,0EAA0E;QAC1E,MAAM,CAAC,KAAK,CAAC,OAAO,UAAU,EAAE,GAAG,EAAE,QAAQ,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;IACtD,MAAM,WAAW,CAAC,QAAQ,EAAE,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE;QACrD,MAAM,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC;YACpC,IAAI,EAAE,gBAAgB;YACtB,SAAS,EAAE,UAAU;YACrB,OAAO,EAAE,WAAW;SACrB,CAAC,CAAC,CAAC;QACJ,MAAM,MAAM,EAAE,CAAC;QAEf,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QAC7B,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,iBAAiB,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;IACnE,MAAM,WAAW,CAAC,QAAQ,EAAE,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE;QACrD,MAAM,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC;YACpC,IAAI,EAAE,gBAAgB;YACtB,SAAS,EAAE,UAAU;YACrB,QAAQ,EAAE,mBAAmB;YAC7B,OAAO,EAAE,aAAa;SACvB,CAAC,CAAC,CAAC;QACJ,MAAM,MAAM,EAAE,CAAC;QAEf,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QAC7B,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,kBAAkB,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,yFAAyF,EAAE,KAAK,IAAI,EAAE;IACzG,MAAM,WAAW,CAAC,QAAQ,EAAE,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE;QACrD,MAAM,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC;YACpC,IAAI,EAAE,gBAAgB;YACtB,SAAS,EAAE,UAAU;YACrB,QAAQ,EAAE,MAAM;YAChB,OAAO,EAAE,aAAa;SACvB,CAAC,CAAC,CAAC;QACJ,MAAM,MAAM,EAAE,CAAC;QAEf,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QAC7B,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,oBAAoB,CAAC,CAAC;IACjE,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;IAC/D,MAAM,WAAW,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE;QACpD,wEAAwE;QACxE,sEAAsE;QACtE,yEAAyE;QACzE,wCAAwC;QACxC,MAAM,UAAU,GAAG,eAAe,CAAC,oBAAoB,CAAC;QACxD,IAAI,OAAO,GAAG,KAAK,CAAC;QACpB,eAAe,CAAC,oBAAoB,GAAG,KAAK,IAAI,EAAE,GAAG,OAAO,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;QACvE,cAAc,EAAE,CAAC;QAEjB,IAAI,CAAC;YACH,MAAM,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC;gBACpC,IAAI,EAAE,WAAW;gBACjB,SAAS,EAAE,UAAU;gBACrB,OAAO,EAAE,gCAAgC;aAC1C,CAAC,CAAC,CAAC;YACJ,MAAM,MAAM,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;YAE7B,MAAM,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC;gBACpC,IAAI,EAAE,gBAAgB;gBACtB,SAAS,EAAE,UAAU;gBACrB,QAAQ,EAAE,QAAQ;gBAClB,OAAO,EAAE,+BAA+B;aACzC,CAAC,CAAC,CAAC;YACJ,MAAM,MAAM,EAAE,CAAC;QACjB,CAAC;gBAAS,CAAC;YACT,eAAe,CAAC,oBAAoB,GAAG,UAAU,CAAC;QACpD,CAAC;QAED,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;QAC7B,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QAC7B,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,iBAAiB,CAAC,CAAC;IAC9D,CAAC,EAAE,qBAAqB,CAAC,CAAC;AAC5B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,kFAAkF,EAAE,KAAK,IAAI,EAAE;IAClG,MAAM,WAAW,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE;QACpD,uEAAuE;QACvE,wEAAwE;QACxE,2EAA2E;QAC3E,0CAA0C;QAC1C,MAAM,UAAU,GAAG,eAAe,CAAC,oBAAoB,CAAC;QACxD,MAAM,WAAW,GAAc,EAAE,CAAC;QAClC,IAAI,qBAAqB,GAAG,KAAK,CAAC;QAClC,eAAe,CAAC,oBAAoB,GAAG,KAAK,EAAE,SAAiB,EAAE,aAA4B,EAAE,EAAE;YAC/F,qBAAqB,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,mBAAmB,CAAC,CAAC;YAC1F,WAAW,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC,CAAC;QACjD,CAAC,CAAC;QAEF,IAAI,CAAC;YACH,MAAM,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC;gBACpC,IAAI,EAAE,gBAAgB;gBACtB,SAAS,EAAE,UAAU;gBACrB,QAAQ,EAAE,QAAQ;gBAClB,OAAO,EAAE,wBAAwB;aAClC,CAAC,CAAC,CAAC;YACJ,MAAM,MAAM,EAAE,CAAC;QACjB,CAAC;gBAAS,CAAC;YACT,eAAe,CAAC,oBAAoB,GAAG,UAAU,CAAC;QACpD,CAAC;QAED,wEAAwE;QACxE,MAAM,CAAC,SAAS,CAAC,WAAW,EAAE,CAAC,EAAE,SAAS,EAAE,UAAU,EAAE,aAAa,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC;QACpF,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QAC7B,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,wBAAwB,CAAC,CAAC;QACxD,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,cAAc,EAAE,SAAS,CAAC,CAAC;QACxD,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,iBAAiB,EAAE,SAAS,CAAC,CAAC;QAE3D,oEAAoE;QACpE,0EAA0E;QAC1E,sEAAsE;QACtE,iDAAiD;QACjD,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,mBAAmB,CAAC,CAAC,CAAC;QAC7E,MAAM,CAAC,KAAK,CAAC,qBAAqB,EAAE,IAAI,CAAC,CAAC;IAC5C,CAAC,EAAE,qBAAqB,CAAC,CAAC;AAC5B,CAAC,CAAC,CAAC"}

package/dist-server/server/modules/websocket/tests/chat-run-registry.test.js

@@ -0,0 +1,275 @@
+import assert from 'node:assert/strict';
+import { mkdtemp, rm } from 'node:fs/promises';
+import { tmpdir } from 'node:os';
+import path from 'node:path';
+import test from 'node:test';
+import { closeConnection, initializeDatabase, sessionsDb } from '../../../modules/database/index.js';
+import { chatRunRegistry } from '../../../modules/websocket/services/chat-run-registry.service.js';
+import { connectedClients } from '../../../modules/websocket/services/websocket-state.service.js';
+/**
+ * Minimal stand-in for a websocket connection: collects every JSON frame the
+ * gateway writer forwards so assertions can inspect the outbound protocol.
+ */
+class FakeConnection {
+    readyState = 1; // WS_OPEN_STATE
+    frames = [];
+    send(data) {
+        this.frames.push(JSON.parse(data));
+    }
+}
+async function withIsolatedDatabase(runTest) {
+    const previousDatabasePath = process.env.DATABASE_PATH;
+    const tempDirectory = await mkdtemp(path.join(tmpdir(), 'chat-run-registry-'));
+    const databasePath = path.join(tempDirectory, 'auth.db');
+    closeConnection();
+    process.env.DATABASE_PATH = databasePath;
+    await initializeDatabase();
+    try {
+        await runTest();
+    }
+    finally {
+        connectedClients.clear();
+        chatRunRegistry.clearAll();
+        closeConnection();
+        if (previousDatabasePath === undefined) {
+            delete process.env.DATABASE_PATH;
+        }
+        else {
+            process.env.DATABASE_PATH = previousDatabasePath;
+        }
+        await rm(tempDirectory, { recursive: true, force: true });
+    }
+}
+test('live events are remapped to the app session id and sequenced', async () => {
+    await withIsolatedDatabase(() => {
+        sessionsDb.createAppSession('app-run-1', 'claude', '/workspace/demo');
+        const connection = new FakeConnection();
+        const run = chatRunRegistry.startRun({
+            appSessionId: 'app-run-1',
+            provider: 'claude',
+            providerSessionId: null,
+            connection,
+            userId: 'user-1',
+        });
+        assert.ok(run);
+        run.writer.send({ kind: 'stream_delta', provider: 'claude', sessionId: 'provider-id-9', content: 'hello' });
+        run.writer.send({ kind: 'text', provider: 'claude', sessionId: 'provider-id-9', content: 'hello world' });
+        assert.equal(connection.frames.length, 2);
+        assert.equal(connection.frames[0]?.sessionId, 'app-run-1');
+        assert.equal(connection.frames[0]?.seq, 1);
+        assert.equal(connection.frames[1]?.sessionId, 'app-run-1');
+        assert.equal(connection.frames[1]?.seq, 2);
+    });
+});
+test('session_created is swallowed and persisted as the provider-id mapping', async () => {
+    await withIsolatedDatabase(async () => {
+        sessionsDb.createAppSession('app-run-2', 'cursor', '/workspace/demo');
+        const connection = new FakeConnection();
+        connectedClients.add(connection);
+        const run = chatRunRegistry.startRun({
+            appSessionId: 'app-run-2',
+            provider: 'cursor',
+            providerSessionId: null,
+            connection,
+            userId: null,
+        });
+        assert.ok(run);
+        run.writer.send({
+            kind: 'session_created',
+            provider: 'cursor',
+            sessionId: 'cursor-native-7',
+            newSessionId: 'cursor-native-7',
+        });
+        // The upsert is broadcast without blocking the run: resolving the owning
+        // project's display name is async, so let that settle before asserting.
+        await new Promise((resolve) => { setTimeout(resolve, 0); });
+        // The provider-native event itself is never forwarded...
+        const sessionUpserts = connection.frames.filter((frame) => frame.kind === 'session_upserted');
+        assert.equal(sessionUpserts.length, 1);
+        assert.equal(sessionUpserts[0]?.sessionId, 'app-run-2');
+        assert.equal(sessionUpserts[0]?.providerSessionId, 'cursor-native-7');
+        // ...but the canonical mapping is recorded and persisted in the database.
+        assert.equal(run.providerSessionId, 'cursor-native-7');
+        assert.equal(sessionsDb.getSessionById('app-run-2')?.provider_session_id, 'cursor-native-7');
+    });
+});
+test('complete marks the run finished and duplicate completes are dropped', async () => {
+    await withIsolatedDatabase(() => {
+        sessionsDb.createAppSession('app-run-3', 'codex', '/workspace/demo');
+        const connection = new FakeConnection();
+        const run = chatRunRegistry.startRun({
+            appSessionId: 'app-run-3',
+            provider: 'codex',
+            providerSessionId: null,
+            connection,
+            userId: null,
+        });
+        assert.ok(run);
+        run.writer.send({ kind: 'complete', provider: 'codex', sessionId: 'native-3', exitCode: 0 });
+        // Late duplicate from a killed runtime's exit handler.
+        run.writer.send({ kind: 'complete', provider: 'codex', sessionId: 'native-3', exitCode: 1 });
+        const completes = connection.frames.filter((frame) => frame.kind === 'complete');
+        assert.equal(completes.length, 1);
+        assert.equal(completes[0]?.actualSessionId, 'app-run-3');
+        assert.equal(chatRunRegistry.isProcessing('app-run-3'), false);
+        // completeRun is also a no-op once the run already completed.
+        chatRunRegistry.completeRun('app-run-3', { exitCode: 1 });
+        assert.equal(connection.frames.filter((frame) => frame.kind === 'complete').length, 1);
+    });
+});
+test('a finished run\'s safety net cannot complete the session\'s next run', async () => {
+    await withIsolatedDatabase(() => {
+        sessionsDb.createAppSession('app-run-9', 'codex', '/workspace/demo');
+        const connection = new FakeConnection();
+        const firstRun = chatRunRegistry.startRun({
+            appSessionId: 'app-run-9',
+            provider: 'codex',
+            providerSessionId: null,
+            connection,
+            userId: null,
+        });
+        assert.ok(firstRun);
+        firstRun.writer.send({ kind: 'complete', provider: 'codex', sessionId: 'native-9', exitCode: 0 });
+        // A queued message starts the next run before the first run's runtime
+        // promise settles (the chat handler's `finally` hasn't executed yet).
+        const secondRun = chatRunRegistry.startRun({
+            appSessionId: 'app-run-9',
+            provider: 'codex',
+            providerSessionId: null,
+            connection,
+            userId: null,
+        });
+        assert.ok(secondRun);
+        // First run's safety net fires late: it must not touch the new run.
+        chatRunRegistry.completeRunIfCurrent(firstRun, { exitCode: 1 });
+        assert.equal(chatRunRegistry.isProcessing('app-run-9'), true);
+        assert.equal(connection.frames.filter((frame) => frame.kind === 'complete').length, 1);
+        // The second run's own safety net still works while it is current.
+        chatRunRegistry.completeRunIfCurrent(secondRun, { exitCode: 1 });
+        assert.equal(chatRunRegistry.isProcessing('app-run-9'), false);
+        assert.equal(connection.frames.filter((frame) => frame.kind === 'complete').length, 2);
+    });
+});
+test('listRunningRuns returns only currently running app sessions', async () => {
+    await withIsolatedDatabase(() => {
+        sessionsDb.createAppSession('app-run-7', 'claude', '/workspace/demo');
+        sessionsDb.createAppSession('app-run-8', 'codex', '/workspace/demo');
+        const connection = new FakeConnection();
+        const completedRun = chatRunRegistry.startRun({
+            appSessionId: 'app-run-7',
+            provider: 'claude',
+            providerSessionId: null,
+            connection,
+            userId: null,
+        });
+        assert.ok(completedRun);
+        const runningRun = chatRunRegistry.startRun({
+            appSessionId: 'app-run-8',
+            provider: 'codex',
+            providerSessionId: null,
+            connection,
+            userId: null,
+        });
+        assert.ok(runningRun);
+        chatRunRegistry.completeRun('app-run-7', { exitCode: 0 });
+        const runningSessions = chatRunRegistry.listRunningRuns();
+        assert.deepEqual(runningSessions.map((session) => session.sessionId), ['app-run-8']);
+        assert.equal(runningSessions[0]?.provider, 'codex');
+    });
+});
+test('replayEvents returns only events after the requested seq', async () => {
+    await withIsolatedDatabase(() => {
+        sessionsDb.createAppSession('app-run-4', 'claude', '/workspace/demo');
+        const connection = new FakeConnection();
+        const run = chatRunRegistry.startRun({
+            appSessionId: 'app-run-4',
+            provider: 'claude',
+            providerSessionId: null,
+            connection,
+            userId: null,
+        });
+        assert.ok(run);
+        run.writer.send({ kind: 'stream_delta', provider: 'claude', sessionId: 'x', content: 'a' });
+        run.writer.send({ kind: 'stream_delta', provider: 'claude', sessionId: 'x', content: 'b' });
+        run.writer.send({ kind: 'stream_delta', provider: 'claude', sessionId: 'x', content: 'c' });
+        const replayed = chatRunRegistry.replayEvents('app-run-4', 1);
+        assert.deepEqual(replayed.map((event) => event.content), ['b', 'c']);
+        assert.deepEqual(replayed.map((event) => event.seq), [2, 3]);
+    });
+});
+test('attachConnection adds a socket without cutting off the ones already watching', async () => {
+    await withIsolatedDatabase(() => {
+        sessionsDb.createAppSession('app-run-5', 'opencode', '/workspace/demo');
+        const firstConnection = new FakeConnection();
+        const run = chatRunRegistry.startRun({
+            appSessionId: 'app-run-5',
+            provider: 'opencode',
+            providerSessionId: null,
+            connection: firstConnection,
+            userId: null,
+        });
+        assert.ok(run);
+        run.writer.send({ kind: 'stream_delta', provider: 'opencode', sessionId: 'o', content: 'before' });
+        // A second tab on the same session subscribes mid-run.
+        const secondConnection = new FakeConnection();
+        assert.equal(chatRunRegistry.attachConnection('app-run-5', secondConnection), true);
+        run.writer.send({ kind: 'stream_delta', provider: 'opencode', sessionId: 'o', content: 'after' });
+        assert.deepEqual(firstConnection.frames.map((frame) => frame.content), ['before', 'after']);
+        assert.deepEqual(secondConnection.frames.map((frame) => frame.content), ['after']);
+    });
+});
+test('a refreshed tab stops receiving once its old socket is closed', async () => {
+    await withIsolatedDatabase(() => {
+        sessionsDb.createAppSession('app-run-5b', 'opencode', '/workspace/demo');
+        const staleConnection = new FakeConnection();
+        const run = chatRunRegistry.startRun({
+            appSessionId: 'app-run-5b',
+            provider: 'opencode',
+            providerSessionId: null,
+            connection: staleConnection,
+            userId: null,
+        });
+        assert.ok(run);
+        // The page reloads: the original socket closes and the fresh one subscribes.
+        staleConnection.readyState = 3;
+        const reloadedConnection = new FakeConnection();
+        assert.equal(chatRunRegistry.attachConnection('app-run-5b', reloadedConnection), true);
+        run.writer.send({ kind: 'stream_delta', provider: 'opencode', sessionId: 'o', content: 'after' });
+        run.writer.send({ kind: 'stream_delta', provider: 'opencode', sessionId: 'o', content: 'later' });
+        assert.deepEqual(staleConnection.frames, []);
+        assert.deepEqual(reloadedConnection.frames.map((frame) => frame.content), ['after', 'later']);
+    });
+});
+test('startRun rejects a second concurrent run for the same session', async () => {
+    await withIsolatedDatabase(() => {
+        sessionsDb.createAppSession('app-run-6', 'opencode', '/workspace/demo');
+        const connection = new FakeConnection();
+        const first = chatRunRegistry.startRun({
+            appSessionId: 'app-run-6',
+            provider: 'opencode',
+            providerSessionId: null,
+            connection,
+            userId: null,
+        });
+        assert.ok(first);
+        const second = chatRunRegistry.startRun({
+            appSessionId: 'app-run-6',
+            provider: 'opencode',
+            providerSessionId: null,
+            connection,
+            userId: null,
+        });
+        assert.equal(second, null);
+        // After the run finishes a new one is allowed again.
+        chatRunRegistry.completeRun('app-run-6', { exitCode: 0 });
+        const third = chatRunRegistry.startRun({
+            appSessionId: 'app-run-6',
+            provider: 'opencode',
+            providerSessionId: null,
+            connection,
+            userId: null,
+        });
+        assert.ok(third);
+    });
+});
+//# sourceMappingURL=chat-run-registry.test.js.map