factory-factory 0.3.8 → 0.3.10

package/dist/src/backend/routers/websocket/post-run-logs.handler.test.js

@@ -0,0 +1,140 @@
+import { EventEmitter } from 'node:events';
+import { beforeEach, describe, expect, it, vi } from 'vitest';
+import { WS_READY_STATE } from '../../../backend/constants/websocket.js';
+import { createPostRunLogsUpgradeHandler, postRunLogsConnections } from './post-run-logs.handler.js';
+class MockWebSocket extends EventEmitter {
+    readyState = WS_READY_STATE.OPEN;
+    send = vi.fn();
+}
+function createLogger() {
+    return {
+        debug: vi.fn(),
+        info: vi.fn(),
+        warn: vi.fn(),
+        error: vi.fn(),
+    };
+}
+function createWssFromQueue(queue) {
+    return {
+        handleUpgrade: vi.fn((_request, _socket, _head, callback) => {
+            const ws = queue.shift();
+            if (!ws) {
+                throw new Error('No mock websocket available');
+            }
+            callback(ws);
+        }),
+    };
+}
+describe('createPostRunLogsUpgradeHandler', () => {
+    beforeEach(() => {
+        postRunLogsConnections.clear();
+        vi.clearAllMocks();
+    });
+    it('rejects upgrades when workspaceId is missing', () => {
+        const logger = createLogger();
+        const runScriptService = {
+            getPostRunOutputBuffer: vi.fn(() => ''),
+            subscribeToPostRunOutput: vi.fn(),
+        };
+        const appContext = {
+            services: {
+                createLogger: vi.fn(() => logger),
+                runScriptService,
+            },
+        };
+        const handler = createPostRunLogsUpgradeHandler(appContext);
+        const request = {};
+        const socket = { write: vi.fn(), destroy: vi.fn() };
+        const wss = { handleUpgrade: vi.fn() };
+        handler(request, socket, Buffer.alloc(0), new URL('http://localhost/post-run-logs'), wss, new WeakMap());
+        expect(logger.warn).toHaveBeenCalledWith('Post-run logs WebSocket missing workspaceId');
+        expect(socket.write).toHaveBeenCalledWith(expect.stringContaining('400 Bad Request'));
+        expect(socket.destroy).toHaveBeenCalledTimes(1);
+        expect(wss.handleUpgrade).not.toHaveBeenCalled();
+    });
+    it('streams buffered/live output and cleans up subscription on close', () => {
+        const workspaceId = 'workspace-1';
+        const logger = createLogger();
+        const unsubscribe = vi.fn();
+        let outputCallback;
+        const runScriptService = {
+            getPostRunOutputBuffer: vi.fn(() => 'buffered post-run logs\n'),
+            subscribeToPostRunOutput: vi.fn((_id, callback) => {
+                outputCallback = callback;
+                return unsubscribe;
+            }),
+        };
+        const appContext = {
+            services: {
+                createLogger: vi.fn(() => logger),
+                runScriptService,
+            },
+        };
+        const ws = new MockWebSocket();
+        const handler = createPostRunLogsUpgradeHandler(appContext);
+        const request = {};
+        const socket = { write: vi.fn(), destroy: vi.fn() };
+        const wss = createWssFromQueue([ws]);
+        const wsAliveMap = new WeakMap();
+        handler(request, socket, Buffer.alloc(0), new URL(`http://localhost/post-run-logs?workspaceId=${workspaceId}`), wss, wsAliveMap);
+        expect(runScriptService.getPostRunOutputBuffer).toHaveBeenCalledWith(workspaceId);
+        expect(runScriptService.subscribeToPostRunOutput).toHaveBeenCalledWith(workspaceId, expect.any(Function));
+        expect(ws.send).toHaveBeenCalledWith(JSON.stringify({ type: 'output', data: 'buffered post-run logs\n' }));
+        expect(postRunLogsConnections.get(workspaceId)?.has(ws)).toBe(true);
+        if (!outputCallback) {
+            throw new Error('Expected output callback to be registered');
+        }
+        outputCallback('live output');
+        expect(ws.send).toHaveBeenCalledWith(JSON.stringify({ type: 'output', data: 'live output' }));
+        const sendsBeforeClosed = ws.send.mock.calls.length;
+        ws.readyState = WS_READY_STATE.CLOSED;
+        outputCallback('ignored output');
+        expect(ws.send).toHaveBeenCalledTimes(sendsBeforeClosed);
+        ws.emit('error', new Error('post-run socket failed'));
+        expect(logger.error).toHaveBeenCalledWith('Post-run logs WebSocket error', expect.objectContaining({ message: 'post-run socket failed' }));
+        ws.emit('close');
+        expect(unsubscribe).toHaveBeenCalledTimes(1);
+        expect(postRunLogsConnections.has(workspaceId)).toBe(false);
+    });
+    it('tracks multiple connections and only deletes workspace entry after the last close', () => {
+        const workspaceId = 'workspace-2';
+        const logger = createLogger();
+        const unsubscribeFirst = vi.fn();
+        const unsubscribeSecond = vi.fn();
+        let subscribeCount = 0;
+        const runScriptService = {
+            getPostRunOutputBuffer: vi.fn(() => ''),
+            subscribeToPostRunOutput: vi.fn(() => {
+                subscribeCount += 1;
+                return subscribeCount === 1 ? unsubscribeFirst : unsubscribeSecond;
+            }),
+        };
+        const appContext = {
+            services: {
+                createLogger: vi.fn(() => logger),
+                runScriptService,
+            },
+        };
+        const ws1 = new MockWebSocket();
+        const ws2 = new MockWebSocket();
+        const handler = createPostRunLogsUpgradeHandler(appContext);
+        const request = {};
+        const socket = { write: vi.fn(), destroy: vi.fn() };
+        const wss = createWssFromQueue([ws1, ws2]);
+        const wsAliveMap = new WeakMap();
+        const url = new URL(`http://localhost/post-run-logs?workspaceId=${workspaceId}`);
+        handler(request, socket, Buffer.alloc(0), url, wss, wsAliveMap);
+        handler(request, socket, Buffer.alloc(0), url, wss, wsAliveMap);
+        expect(runScriptService.getPostRunOutputBuffer).toHaveBeenCalledTimes(2);
+        expect(ws1.send).not.toHaveBeenCalled();
+        expect(ws2.send).not.toHaveBeenCalled();
+        expect(postRunLogsConnections.get(workspaceId)?.size).toBe(2);
+        ws1.emit('close');
+        expect(unsubscribeFirst).toHaveBeenCalledTimes(1);
+        expect(postRunLogsConnections.get(workspaceId)?.size).toBe(1);
+        ws2.emit('close');
+        expect(unsubscribeSecond).toHaveBeenCalledTimes(1);
+        expect(postRunLogsConnections.has(workspaceId)).toBe(false);
+    });
+});
+//# sourceMappingURL=post-run-logs.handler.test.js.map

package/dist/src/backend/routers/websocket/post-run-logs.handler.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"post-run-logs.handler.test.js","sourceRoot":"","sources":["../../../../../src/backend/routers/websocket/post-run-logs.handler.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAG3C,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAG9D,OAAO,EAAE,cAAc,EAAE,MAAM,+BAA+B,CAAC;AAC/D,OAAO,EAAE,+BAA+B,EAAE,sBAAsB,EAAE,MAAM,yBAAyB,CAAC;AAElG,MAAM,aAAc,SAAQ,YAAY;IACtC,UAAU,GAAW,cAAc,CAAC,IAAI,CAAC;IACzC,IAAI,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;CAChB;AAED,SAAS,YAAY;IACnB,OAAO;QACL,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;QACd,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;QACb,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;QACb,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;KACf,CAAC;AACJ,CAAC;AAED,SAAS,kBAAkB,CAAC,KAAsB;IAChD,OAAO;QACL,aAAa,EAAE,EAAE,CAAC,EAAE,CAClB,CACE,QAAyB,EACzB,OAAe,EACf,KAAa,EACb,QAAqC,EACrC,EAAE;YACF,MAAM,EAAE,GAAG,KAAK,CAAC,KAAK,EAAE,CAAC;YACzB,IAAI,CAAC,EAAE,EAAE,CAAC;gBACR,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;YACjD,CAAC;YACD,QAAQ,CAAC,EAA0B,CAAC,CAAC;QACvC,CAAC,CACF;KAC4B,CAAC;AAClC,CAAC;AAED,QAAQ,CAAC,iCAAiC,EAAE,GAAG,EAAE;IAC/C,UAAU,CAAC,GAAG,EAAE;QACd,sBAAsB,CAAC,KAAK,EAAE,CAAC;QAC/B,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,MAAM,GAAG,YAAY,EAAE,CAAC;QAC9B,MAAM,gBAAgB,GAAG;YACvB,sBAAsB,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC;YACvC,wBAAwB,EAAE,EAAE,CAAC,EAAE,EAAE;SAClC,CAAC;QACF,MAAM,UAAU,GAAG;YACjB,QAAQ,EAAE;gBACR,YAAY,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC;gBACjC,gBAAgB;aACjB;SACuB,CAAC;QAE3B,MAAM,OAAO,GAAG,+BAA+B,CAAC,UAAU,CAAC,CAAC;QAC5D,MAAM,OAAO,GAAG,EAAqB,CAAC;QACtC,MAAM,MAAM,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,EAAuB,CAAC;QACzE,MAAM,GAAG,GAAG,EAAE,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE,EAAgC,CAAC;QAErE,OAAO,CACL,OAAO,EACP,MAAM,EACN,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EACf,IAAI,GAAG,CAAC,gCAAgC,CAAC,EACzC,GAAG,EACH,IAAI,OAAO,EAAsB,CAClC,CAAC;QAEF,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,6CAA6C,CAAC,CAAC;QACxF,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAAC,gBAAgB,CAAC,iBAAiB,CAAC,CAAC,CAAC;QACtF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAChD,MAAM,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kEAAkE,EAAE,GAAG,EAAE;QAC1E,MAAM,WAAW,GAAG,aAAa,CAAC;QAClC,MAAM,MAAM,GAAG,YAAY,EAAE,CAAC;QAC9B,MAAM,WAAW,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAC5B,IAAI,cAAoD,CAAC;QACzD,MAAM,gBAAgB,GAAG;YACvB,sBAAsB,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,0BAA0B,CAAC;YAC/D,wBAAwB,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,GAAW,EAAE,QAAgC,EAAE,EAAE;gBAChF,cAAc,GAAG,QAAQ,CAAC;gBAC1B,OAAO,WAAW,CAAC;YACrB,CAAC,CAAC;SACH,CAAC;QACF,MAAM,UAAU,GAAG;YACjB,QAAQ,EAAE;gBACR,YAAY,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC;gBACjC,gBAAgB;aACjB;SACuB,CAAC;QAE3B,MAAM,EAAE,GAAG,IAAI,aAAa,EAAE,CAAC;QAC/B,MAAM,OAAO,GAAG,+BAA+B,CAAC,UAAU,CAAC,CAAC;QAC5D,MAAM,OAAO,GAAG,EAAqB,CAAC;QACtC,MAAM,MAAM,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,EAAuB,CAAC;QACzE,MAAM,GAAG,GAAG,kBAAkB,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QACrC,MAAM,UAAU,GAAG,IAAI,OAAO,EAAsB,CAAC;QAErD,OAAO,CACL,OAAO,EACP,MAAM,EACN,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EACf,IAAI,GAAG,CAAC,8CAA8C,WAAW,EAAE,CAAC,EACpE,GAAG,EACH,UAAU,CACX,CAAC;QAEF,MAAM,CAAC,gBAAgB,CAAC,sBAAsB,CAAC,CAAC,oBAAoB,CAAC,WAAW,CAAC,CAAC;QAClF,MAAM,CAAC,gBAAgB,CAAC,wBAAwB,CAAC,CAAC,oBAAoB,CACpE,WAAW,EACX,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CACrB,CAAC;QACF,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAClC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,0BAA0B,EAAE,CAAC,CACrE,CAAC;QACF,MAAM,CAAC,sBAAsB,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,GAAG,CAAC,EAA0B,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAE5F,IAAI,CAAC,cAAc,EAAE,CAAC;YACpB,MAAM,IAAI,KAAK,CAAC,2CAA2C,CAAC,CAAC;QAC/D,CAAC;QAED,cAAc,CAAC,aAAa,CAAC,CAAC;QAC9B,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC,CAAC,CAAC;QAE9F,MAAM,iBAAiB,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC;QACpD,EAAE,CAAC,UAAU,GAAG,cAAc,CAAC,MAAM,CAAC;QACtC,cAAc,CAAC,gBAAgB,CAAC,CAAC;QACjC,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,qBAAqB,CAAC,iBAAiB,CAAC,CAAC;QAEzD,EAAE,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC,CAAC;QACtD,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,oBAAoB,CACvC,+BAA+B,EAC/B,MAAM,CAAC,gBAAgB,CAAC,EAAE,OAAO,EAAE,wBAAwB,EAAE,CAAC,CAC/D,CAAC;QAEF,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACjB,MAAM,CAAC,WAAW,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC7C,MAAM,CAAC,sBAAsB,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mFAAmF,EAAE,GAAG,EAAE;QAC3F,MAAM,WAAW,GAAG,aAAa,CAAC;QAClC,MAAM,MAAM,GAAG,YAAY,EAAE,CAAC;QAC9B,MAAM,gBAAgB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACjC,MAAM,iBAAiB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAClC,IAAI,cAAc,GAAG,CAAC,CAAC;QACvB,MAAM,gBAAgB,GAAG;YACvB,sBAAsB,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC;YACvC,wBAAwB,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE;gBACnC,cAAc,IAAI,CAAC,CAAC;gBACpB,OAAO,cAAc,KAAK,CAAC,CAAC,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,iBAAiB,CAAC;YACrE,CAAC,CAAC;SACH,CAAC;QACF,MAAM,UAAU,GAAG;YACjB,QAAQ,EAAE;gBACR,YAAY,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC;gBACjC,gBAAgB;aACjB;SACuB,CAAC;QAE3B,MAAM,GAAG,GAAG,IAAI,aAAa,EAAE,CAAC;QAChC,MAAM,GAAG,GAAG,IAAI,aAAa,EAAE,CAAC;QAChC,MAAM,OAAO,GAAG,+BAA+B,CAAC,UAAU,CAAC,CAAC;QAC5D,MAAM,OAAO,GAAG,EAAqB,CAAC;QACtC,MAAM,MAAM,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,EAAuB,CAAC;QACzE,MAAM,GAAG,GAAG,kBAAkB,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;QAC3C,MAAM,UAAU,GAAG,IAAI,OAAO,EAAsB,CAAC;QACrD,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,8CAA8C,WAAW,EAAE,CAAC,CAAC;QAEjF,OAAO,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,GAAG,EAAE,UAAU,CAAC,CAAC;QAChE,OAAO,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,GAAG,EAAE,UAAU,CAAC,CAAC;QAEhE,MAAM,CAAC,gBAAgB,CAAC,sBAAsB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACzE,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACxC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACxC,MAAM,CAAC,sBAAsB,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAE9D,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAClB,MAAM,CAAC,gBAAgB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAClD,MAAM,CAAC,sBAAsB,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAE9D,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAClB,MAAM,CAAC,iBAAiB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACnD,MAAM,CAAC,sBAAsB,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/backend/routers/websocket/setup-terminal.handler.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"setup-terminal.handler.d.ts","sourceRoot":"","sources":["../../../../../src/backend/routers/websocket/setup-terminal.handler.ts"],"names":[],"mappings":"AAQA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,WAAW,CAAC;AAGjD,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAE1C,OAAO,KAAK,EAAE,SAAS,EAAE,eAAe,EAAE,MAAM,IAAI,CAAC;AACrD,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAwFxD,wBAAgB,iCAAiC,CAAC,UAAU,EAAE,UAAU,IAIpE,SAAS,eAAe,EACxB,QAAQ,MAAM,EACd,MAAM,MAAM,EACZ,MAAM,GAAG,EACT,KAAK,eAAe,EACpB,YAAY,OAAO,CAAC,SAAS,EAAE,OAAO,CAAC,KACtC,IAAI,CAgDR"}
+{"version":3,"file":"setup-terminal.handler.d.ts","sourceRoot":"","sources":["../../../../../src/backend/routers/websocket/setup-terminal.handler.ts"],"names":[],"mappings":"AAQA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,WAAW,CAAC;AAGjD,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAE1C,OAAO,KAAK,EAAE,SAAS,EAAE,eAAe,EAAE,MAAM,IAAI,CAAC;AACrD,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAuHxD,wBAAgB,iCAAiC,CAAC,UAAU,EAAE,UAAU,IAKpE,SAAS,eAAe,EACxB,QAAQ,MAAM,EACd,MAAM,MAAM,EACZ,MAAM,GAAG,EACT,KAAK,eAAe,EACpB,YAAY,OAAO,CAAC,SAAS,EAAE,OAAO,CAAC,KACtC,IAAI,CAiDR"}

package/dist/src/backend/routers/websocket/setup-terminal.handler.js

@@ -1,13 +1,15 @@
 import { createRequire } from 'node:module';
 import { homedir, tmpdir } from 'node:os';
 import { WS_READY_STATE } from '../../../backend/constants/websocket.js';
+import { toError } from '../../../backend/lib/error-utils.js';
+import { SetupTerminalMessageSchema, } from '../../../backend/schemas/websocket/index.js';
 import { toMessageString } from './message-utils.js';
 import { markWebSocketAlive } from './upgrade-utils.js';
 const require = createRequire(import.meta.url);
 function getNodePty() {
     return require('node-pty');
 }
-function handleCreate(ws, message, state, logger) {
+function handleCreate(ws, message, state, logger, configService) {
     if (state.pty) {
         ws.send(JSON.stringify({ type: 'error', message: 'Terminal already exists' }));
         return;
@@ -15,7 +17,7 @@ function handleCreate(ws, message, state, logger) {
     const cols = message.cols ?? 80;
     const rows = message.rows ?? 24;
     const cwd = homedir() || tmpdir();
-    const shellPath = process.env.SHELL || '/bin/bash';
+    const shellPath = configService.getShellPath();
     logger.info('Creating setup terminal', { cwd, shell: shellPath, cols, rows });
     const nodePty = getNodePty();
     state.pty = nodePty.spawn(shellPath, [], {
@@ -24,7 +26,7 @@ function handleCreate(ws, message, state, logger) {
         rows,
         cwd,
         env: {
-            ...process.env,
+            ...configService.getChildProcessEnv(),
             TERM: 'xterm-256color',
             COLORTERM: 'truecolor',
         },
@@ -44,12 +46,12 @@ function handleCreate(ws, message, state, logger) {
     ws.send(JSON.stringify({ type: 'created' }));
 }
 function handleInput(message, state) {
-    if (state.pty && message.data) {
+    if (state.pty) {
         state.pty.write(message.data);
     }
 }
 function handleResize(message, state) {
-    if (state.pty && message.cols && message.rows) {
+    if (state.pty) {
         state.pty.resize(message.cols, message.rows);
     }
 }
@@ -58,8 +60,25 @@ function handlePing(ws) {
         ws.send(JSON.stringify({ type: 'pong' }));
     }
 }
+function parseSetupTerminalMessage(data, logger) {
+    const rawMessage = JSON.parse(toMessageString(data));
+    const parseResult = SetupTerminalMessageSchema.safeParse(rawMessage);
+    if (!parseResult.success) {
+        logger.warn('Invalid setup terminal message format', {
+            errors: parseResult.error.issues,
+        });
+        return null;
+    }
+    return parseResult.data;
+}
+function sendSocketError(ws, message) {
+    if (ws.readyState === WS_READY_STATE.OPEN) {
+        ws.send(JSON.stringify({ type: 'error', message }));
+    }
+}
 export function createSetupTerminalUpgradeHandler(appContext) {
     const logger = appContext.services.createLogger('setup-terminal-handler');
+    const { configService } = appContext.services;
     return function handleSetupTerminalUpgrade(request, socket, head, _url, wss, wsAliveMap) {
         wss.handleUpgrade(request, socket, head, (ws) => {
             logger.info('Setup terminal WebSocket connected');
@@ -67,11 +86,14 @@ export function createSetupTerminalUpgradeHandler(appContext) {
             const state = { pty: null };
             ws.on('message', (data) => {
                 try {
-                    const raw = JSON.parse(toMessageString(data));
-                    const message = raw;
+                    const message = parseSetupTerminalMessage(data, logger);
+                    if (!message) {
+                        sendSocketError(ws, 'Invalid message format');
+                        return;
+                    }
                     switch (message.type) {
                         case 'create':
-                            handleCreate(ws, message, state, logger);
+                            handleCreate(ws, message, state, logger, configService);
                             break;
                         case 'input':
                             handleInput(message, state);
@@ -85,11 +107,9 @@ export function createSetupTerminalUpgradeHandler(appContext) {
                     }
                 }
                 catch (error) {
-                    const err = error;
+                    const err = toError(error);
                     logger.error('Error in setup terminal', err);
-                    if (ws.readyState === WS_READY_STATE.OPEN) {
-                        ws.send(JSON.stringify({ type: 'error', message: err.message }));
-                    }
+                    sendSocketError(ws, err.message);
                 }
             });
             ws.on('close', () => {

package/dist/src/backend/routers/websocket/setup-terminal.handler.js.map

@@ -1 +1 @@
-{"version":3,"file":"setup-terminal.handler.js","sourceRoot":"","sources":["../../../../../src/backend/routers/websocket/setup-terminal.handler.ts"],"names":[],"mappings":"AASA,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAC5C,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AAK1C,OAAO,EAAE,cAAc,EAAE,MAAM,+BAA+B,CAAC;AAC/D,OAAO,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAC;AAClD,OAAO,EAAE,kBAAkB,EAAE,MAAM,iBAAiB,CAAC;AAErD,MAAM,OAAO,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAE/C,SAAS,UAAU;IACjB,OAAO,OAAO,CAAC,UAAU,CAAC,CAAC;AAC7B,CAAC;AAaD,SAAS,YAAY,CACnB,EAAa,EACb,OAA6B,EAC7B,KAAyB,EACzB,MAA0D;IAE1D,IAAI,KAAK,CAAC,GAAG,EAAE,CAAC;QACd,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,yBAAyB,EAAE,CAAC,CAAC,CAAC;QAC/E,OAAO;IACT,CAAC;IAED,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,IAAI,EAAE,CAAC;IAChC,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,IAAI,EAAE,CAAC;IAChC,MAAM,GAAG,GAAG,OAAO,EAAE,IAAI,MAAM,EAAE,CAAC;IAClC,MAAM,SAAS,GAAG,OAAO,CAAC,GAAG,CAAC,KAAK,IAAI,WAAW,CAAC;IAEnD,MAAM,CAAC,IAAI,CAAC,yBAAyB,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;IAE9E,MAAM,OAAO,GAAG,UAAU,EAAE,CAAC;IAC7B,KAAK,CAAC,GAAG,GAAG,OAAO,CAAC,KAAK,CAAC,SAAS,EAAE,EAAE,EAAE;QACvC,IAAI,EAAE,gBAAgB;QACtB,IAAI;QACJ,IAAI;QACJ,GAAG;QACH,GAAG,EAAE;YACH,GAAG,OAAO,CAAC,GAAG;YACd,IAAI,EAAE,gBAAgB;YACtB,SAAS,EAAE,WAAW;SACvB;KACF,CAAC,CAAC;IAEH,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,MAAc,EAAE,EAAE;QAClC,IAAI,EAAE,CAAC,UAAU,KAAK,cAAc,CAAC,IAAI,EAAE,CAAC;YAC1C,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;QAC5D,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,EAAE,QAAQ,EAAwB,EAAE,EAAE;QACtD,MAAM,CAAC,IAAI,CAAC,uBAAuB,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC;QACnD,IAAI,EAAE,CAAC,UAAU,KAAK,cAAc,CAAC,IAAI,EAAE,CAAC;YAC1C,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC;QACtD,CAAC;QACD,KAAK,CAAC,GAAG,GAAG,IAAI,CAAC;IACnB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC;AAC/C,CAAC;AAED,SAAS,WAAW,CAAC,OAA6B,EAAE,KAAyB;IAC3E,IAAI,KAAK,CAAC,GAAG,IAAI,OAAO,CAAC,IAAI,EAAE,CAAC;QAC9B,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IAChC,CAAC;AACH,CAAC;AAED,SAAS,YAAY,CAAC,OAA6B,EAAE,KAAyB;IAC5E,IAAI,KAAK,CAAC,GAAG,IAAI,OAAO,CAAC,IAAI,IAAI,OAAO,CAAC,IAAI,EAAE,CAAC;QAC9C,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;IAC/C,CAAC;AACH,CAAC;AAED,SAAS,UAAU,CAAC,EAAa;IAC/B,IAAI,EAAE,CAAC,UAAU,KAAK,cAAc,CAAC,IAAI,EAAE,CAAC;QAC1C,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;IAC5C,CAAC;AACH,CAAC;AAED,MAAM,UAAU,iCAAiC,CAAC,UAAsB;IACtE,MAAM,MAAM,GAAG,UAAU,CAAC,QAAQ,CAAC,YAAY,CAAC,wBAAwB,CAAC,CAAC;IAE1E,OAAO,SAAS,0BAA0B,CACxC,OAAwB,EACxB,MAAc,EACd,IAAY,EACZ,IAAS,EACT,GAAoB,EACpB,UAAuC;QAEvC,GAAG,CAAC,aAAa,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,EAAE,EAAE,EAAE;YAC9C,MAAM,CAAC,IAAI,CAAC,oCAAoC,CAAC,CAAC;YAClD,kBAAkB,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;YAEnC,MAAM,KAAK,GAAuB,EAAE,GAAG,EAAE,IAAI,EAAE,CAAC;YAEhD,EAAE,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,IAAI,EAAE,EAAE;gBACxB,IAAI,CAAC;oBACH,MAAM,GAAG,GAAY,IAAI,CAAC,KAAK,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC;oBACvD,MAAM,OAAO,GAAG,GAA2B,CAAC;oBAE5C,QAAQ,OAAO,CAAC,IAAI,EAAE,CAAC;wBACrB,KAAK,QAAQ;4BACX,YAAY,CAAC,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC;4BACzC,MAAM;wBACR,KAAK,OAAO;4BACV,WAAW,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;4BAC5B,MAAM;wBACR,KAAK,QAAQ;4BACX,YAAY,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;4BAC7B,MAAM;wBACR,KAAK,MAAM;4BACT,UAAU,CAAC,EAAE,CAAC,CAAC;4BACf,MAAM;oBACV,CAAC;gBACH,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,MAAM,GAAG,GAAG,KAAc,CAAC;oBAC3B,MAAM,CAAC,KAAK,CAAC,yBAAyB,EAAE,GAAG,CAAC,CAAC;oBAC7C,IAAI,EAAE,CAAC,UAAU,KAAK,cAAc,CAAC,IAAI,EAAE,CAAC;wBAC1C,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;oBACnE,CAAC;gBACH,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;gBAClB,MAAM,CAAC,IAAI,CAAC,iCAAiC,CAAC,CAAC;gBAC/C,IAAI,KAAK,CAAC,GAAG,EAAE,CAAC;oBACd,KAAK,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC;oBACjB,KAAK,CAAC,GAAG,GAAG,IAAI,CAAC;gBACnB,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;gBACvB,MAAM,CAAC,KAAK,CAAC,gCAAgC,EAAE,KAAK,CAAC,CAAC;YACxD,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC;AACJ,CAAC"}
+{"version":3,"file":"setup-terminal.handler.js","sourceRoot":"","sources":["../../../../../src/backend/routers/websocket/setup-terminal.handler.ts"],"names":[],"mappings":"AASA,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAC5C,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AAK1C,OAAO,EAAE,cAAc,EAAE,MAAM,+BAA+B,CAAC;AAC/D,OAAO,EAAE,OAAO,EAAE,MAAM,2BAA2B,CAAC;AACpD,OAAO,EAEL,0BAA0B,GAC3B,MAAM,6BAA6B,CAAC;AACrC,OAAO,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAC;AAClD,OAAO,EAAE,kBAAkB,EAAE,MAAM,iBAAiB,CAAC;AAErD,MAAM,OAAO,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAE/C,SAAS,UAAU;IACjB,OAAO,OAAO,CAAC,UAAU,CAAC,CAAC;AAC7B,CAAC;AASD,SAAS,YAAY,CACnB,EAAa,EACb,OAA+D,EAC/D,KAAyB,EACzB,MAA2B,EAC3B,aAAyC;IAEzC,IAAI,KAAK,CAAC,GAAG,EAAE,CAAC;QACd,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,yBAAyB,EAAE,CAAC,CAAC,CAAC;QAC/E,OAAO;IACT,CAAC;IAED,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,IAAI,EAAE,CAAC;IAChC,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,IAAI,EAAE,CAAC;IAChC,MAAM,GAAG,GAAG,OAAO,EAAE,IAAI,MAAM,EAAE,CAAC;IAClC,MAAM,SAAS,GAAG,aAAa,CAAC,YAAY,EAAE,CAAC;IAE/C,MAAM,CAAC,IAAI,CAAC,yBAAyB,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;IAE9E,MAAM,OAAO,GAAG,UAAU,EAAE,CAAC;IAC7B,KAAK,CAAC,GAAG,GAAG,OAAO,CAAC,KAAK,CAAC,SAAS,EAAE,EAAE,EAAE;QACvC,IAAI,EAAE,gBAAgB;QACtB,IAAI;QACJ,IAAI;QACJ,GAAG;QACH,GAAG,EAAE;YACH,GAAG,aAAa,CAAC,kBAAkB,EAAE;YACrC,IAAI,EAAE,gBAAgB;YACtB,SAAS,EAAE,WAAW;SACvB;KACF,CAAC,CAAC;IAEH,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,MAAc,EAAE,EAAE;QAClC,IAAI,EAAE,CAAC,UAAU,KAAK,cAAc,CAAC,IAAI,EAAE,CAAC;YAC1C,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;QAC5D,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,EAAE,QAAQ,EAAwB,EAAE,EAAE;QACtD,MAAM,CAAC,IAAI,CAAC,uBAAuB,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC;QACnD,IAAI,EAAE,CAAC,UAAU,KAAK,cAAc,CAAC,IAAI,EAAE,CAAC;YAC1C,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC;QACtD,CAAC;QACD,KAAK,CAAC,GAAG,GAAG,IAAI,CAAC;IACnB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC;AAC/C,CAAC;AAED,SAAS,WAAW,CAClB,OAA8D,EAC9D,KAAyB;IAEzB,IAAI,KAAK,CAAC,GAAG,EAAE,CAAC;QACd,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IAChC,CAAC;AACH,CAAC;AAED,SAAS,YAAY,CACnB,OAA+D,EAC/D,KAAyB;IAEzB,IAAI,KAAK,CAAC,GAAG,EAAE,CAAC;QACd,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;IAC/C,CAAC;AACH,CAAC;AAED,SAAS,UAAU,CAAC,EAAa;IAC/B,IAAI,EAAE,CAAC,UAAU,KAAK,cAAc,CAAC,IAAI,EAAE,CAAC;QAC1C,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;IAC5C,CAAC;AACH,CAAC;AAED,SAAS,yBAAyB,CAChC,IAAa,EACb,MAA2B;IAE3B,MAAM,UAAU,GAAY,IAAI,CAAC,KAAK,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC;IAC9D,MAAM,WAAW,GAAG,0BAA0B,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;IAErE,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC;QACzB,MAAM,CAAC,IAAI,CAAC,uCAAuC,EAAE;YACnD,MAAM,EAAE,WAAW,CAAC,KAAK,CAAC,MAAM;SACjC,CAAC,CAAC;QACH,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO,WAAW,CAAC,IAAI,CAAC;AAC1B,CAAC;AAED,SAAS,eAAe,CAAC,EAAa,EAAE,OAAe;IACrD,IAAI,EAAE,CAAC,UAAU,KAAK,cAAc,CAAC,IAAI,EAAE,CAAC;QAC1C,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC;IACtD,CAAC;AACH,CAAC;AAED,MAAM,UAAU,iCAAiC,CAAC,UAAsB;IACtE,MAAM,MAAM,GAAG,UAAU,CAAC,QAAQ,CAAC,YAAY,CAAC,wBAAwB,CAAC,CAAC;IAC1E,MAAM,EAAE,aAAa,EAAE,GAAG,UAAU,CAAC,QAAQ,CAAC;IAE9C,OAAO,SAAS,0BAA0B,CACxC,OAAwB,EACxB,MAAc,EACd,IAAY,EACZ,IAAS,EACT,GAAoB,EACpB,UAAuC;QAEvC,GAAG,CAAC,aAAa,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,EAAE,EAAE,EAAE;YAC9C,MAAM,CAAC,IAAI,CAAC,oCAAoC,CAAC,CAAC;YAClD,kBAAkB,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;YAEnC,MAAM,KAAK,GAAuB,EAAE,GAAG,EAAE,IAAI,EAAE,CAAC;YAEhD,EAAE,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,IAAI,EAAE,EAAE;gBACxB,IAAI,CAAC;oBACH,MAAM,OAAO,GAAG,yBAAyB,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;oBACxD,IAAI,CAAC,OAAO,EAAE,CAAC;wBACb,eAAe,CAAC,EAAE,EAAE,wBAAwB,CAAC,CAAC;wBAC9C,OAAO;oBACT,CAAC;oBAED,QAAQ,OAAO,CAAC,IAAI,EAAE,CAAC;wBACrB,KAAK,QAAQ;4BACX,YAAY,CAAC,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,aAAa,CAAC,CAAC;4BACxD,MAAM;wBACR,KAAK,OAAO;4BACV,WAAW,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;4BAC5B,MAAM;wBACR,KAAK,QAAQ;4BACX,YAAY,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;4BAC7B,MAAM;wBACR,KAAK,MAAM;4BACT,UAAU,CAAC,EAAE,CAAC,CAAC;4BACf,MAAM;oBACV,CAAC;gBACH,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,MAAM,GAAG,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC;oBAC3B,MAAM,CAAC,KAAK,CAAC,yBAAyB,EAAE,GAAG,CAAC,CAAC;oBAC7C,eAAe,CAAC,EAAE,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC;gBACnC,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;gBAClB,MAAM,CAAC,IAAI,CAAC,iCAAiC,CAAC,CAAC;gBAC/C,IAAI,KAAK,CAAC,GAAG,EAAE,CAAC;oBACd,KAAK,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC;oBACjB,KAAK,CAAC,GAAG,GAAG,IAAI,CAAC;gBACnB,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;gBACvB,MAAM,CAAC,KAAK,CAAC,gCAAgC,EAAE,KAAK,CAAC,CAAC;YACxD,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC;AACJ,CAAC"}

package/dist/src/backend/routers/websocket/setup-terminal.handler.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=setup-terminal.handler.test.d.ts.map

package/dist/src/backend/routers/websocket/setup-terminal.handler.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"setup-terminal.handler.test.d.ts","sourceRoot":"","sources":["../../../../../src/backend/routers/websocket/setup-terminal.handler.test.ts"],"names":[],"mappings":""}

package/dist/src/backend/routers/websocket/setup-terminal.handler.test.js

@@ -0,0 +1,170 @@
+import { EventEmitter } from 'node:events';
+import { beforeEach, describe, expect, it, vi } from 'vitest';
+import { WS_READY_STATE } from '../../../backend/constants/websocket.js';
+const mockPtyWrite = vi.fn();
+const mockPtyResize = vi.fn();
+const mockPtyKill = vi.fn();
+const mockPtySpawn = vi.fn();
+let onDataCallback = null;
+let onExitCallback = null;
+vi.mock('node:module', () => ({
+    createRequire: () => (id) => {
+        if (id === 'node-pty') {
+            return {
+                spawn: mockPtySpawn,
+            };
+        }
+        throw new Error(`Unexpected require: ${id}`);
+    },
+}));
+import { createSetupTerminalUpgradeHandler } from './setup-terminal.handler.js';
+class MockWebSocket extends EventEmitter {
+    readyState = WS_READY_STATE.OPEN;
+    send = vi.fn();
+}
+function createLogger() {
+    return {
+        debug: vi.fn(),
+        info: vi.fn(),
+        warn: vi.fn(),
+        error: vi.fn(),
+    };
+}
+function createMockPty() {
+    return {
+        onData: vi.fn((callback) => {
+            onDataCallback = callback;
+        }),
+        onExit: vi.fn((callback) => {
+            onExitCallback = callback;
+        }),
+        write: mockPtyWrite,
+        resize: mockPtyResize,
+        kill: mockPtyKill,
+    };
+}
+function createWss(ws) {
+    return {
+        handleUpgrade: vi.fn((_request, _socket, _head, callback) => callback(ws)),
+    };
+}
+describe('createSetupTerminalUpgradeHandler', () => {
+    beforeEach(() => {
+        vi.clearAllMocks();
+        onDataCallback = null;
+        onExitCallback = null;
+        mockPtySpawn.mockReturnValue(createMockPty());
+    });
+    it('creates terminal and routes lifecycle messages', () => {
+        const logger = createLogger();
+        const configService = {
+            getShellPath: vi.fn(() => '/bin/zsh'),
+            getChildProcessEnv: vi.fn(() => ({ PATH: '/usr/bin' })),
+        };
+        const appContext = {
+            services: {
+                createLogger: vi.fn(() => logger),
+                configService,
+            },
+        };
+        const handler = createSetupTerminalUpgradeHandler(appContext);
+        const ws = new MockWebSocket();
+        const wss = createWss(ws);
+        const request = {};
+        const socket = { write: vi.fn(), destroy: vi.fn() };
+        const wsAliveMap = new WeakMap();
+        handler(request, socket, Buffer.alloc(0), new URL('http://localhost/setup-terminal'), wss, wsAliveMap);
+        ws.emit('message', JSON.stringify({ type: 'create', cols: 120, rows: 40 }));
+        expect(mockPtySpawn).toHaveBeenCalledWith('/bin/zsh', [], expect.objectContaining({
+            name: 'xterm-256color',
+            cols: 120,
+            rows: 40,
+            env: expect.objectContaining({
+                PATH: '/usr/bin',
+                TERM: 'xterm-256color',
+                COLORTERM: 'truecolor',
+            }),
+        }));
+        expect(ws.send).toHaveBeenCalledWith(JSON.stringify({ type: 'created' }));
+        onDataCallback?.('hello from shell');
+        expect(ws.send).toHaveBeenCalledWith(JSON.stringify({ type: 'output', data: 'hello from shell' }));
+        ws.emit('message', JSON.stringify({ type: 'input', data: 'echo hi\n' }));
+        expect(mockPtyWrite).toHaveBeenCalledWith('echo hi\n');
+        ws.emit('message', JSON.stringify({ type: 'resize', cols: 150, rows: 50 }));
+        expect(mockPtyResize).toHaveBeenCalledWith(150, 50);
+        ws.emit('message', JSON.stringify({ type: 'ping' }));
+        expect(ws.send).toHaveBeenCalledWith(JSON.stringify({ type: 'pong' }));
+        ws.emit('message', JSON.stringify({ type: 'create' }));
+        expect(ws.send).toHaveBeenCalledWith(JSON.stringify({ type: 'error', message: 'Terminal already exists' }));
+        onExitCallback?.({ exitCode: 9 });
+        expect(ws.send).toHaveBeenCalledWith(JSON.stringify({ type: 'exit', exitCode: 9 }));
+        ws.emit('message', JSON.stringify({ type: 'create' }));
+        expect(mockPtySpawn).toHaveBeenCalledTimes(2);
+    });
+    it('rejects messages that fail schema validation', () => {
+        const logger = createLogger();
+        const configService = {
+            getShellPath: vi.fn(() => '/bin/zsh'),
+            getChildProcessEnv: vi.fn(() => ({})),
+        };
+        const appContext = {
+            services: {
+                createLogger: vi.fn(() => logger),
+                configService,
+            },
+        };
+        const handler = createSetupTerminalUpgradeHandler(appContext);
+        const ws = new MockWebSocket();
+        const wss = createWss(ws);
+        const request = {};
+        const socket = { write: vi.fn(), destroy: vi.fn() };
+        const wsAliveMap = new WeakMap();
+        handler(request, socket, Buffer.alloc(0), new URL('http://localhost/setup-terminal'), wss, wsAliveMap);
+        ws.emit('message', JSON.stringify({ type: 'resize', cols: '120', rows: 40 }));
+        expect(ws.send).toHaveBeenCalledWith(JSON.stringify({ type: 'error', message: 'Invalid message format' }));
+        expect(logger.warn).toHaveBeenCalledWith('Invalid setup terminal message format', {
+            errors: expect.any(Array),
+        });
+    });
+    it('reports runtime parsing/spawn failures and cleans up on close', () => {
+        const logger = createLogger();
+        const configService = {
+            getShellPath: vi.fn(() => '/bin/zsh'),
+            getChildProcessEnv: vi.fn(() => ({})),
+        };
+        const appContext = {
+            services: {
+                createLogger: vi.fn(() => logger),
+                configService,
+            },
+        };
+        const handler = createSetupTerminalUpgradeHandler(appContext);
+        const ws = new MockWebSocket();
+        const wss = createWss(ws);
+        const request = {};
+        const socket = { write: vi.fn(), destroy: vi.fn() };
+        const wsAliveMap = new WeakMap();
+        handler(request, socket, Buffer.alloc(0), new URL('http://localhost/setup-terminal'), wss, wsAliveMap);
+        ws.emit('message', '{');
+        expect(logger.error).toHaveBeenCalledWith('Error in setup terminal', expect.any(SyntaxError));
+        expect(ws.send).toHaveBeenCalledWith(expect.stringMatching(/^{"type":"error","message":".+"}$/));
+        mockPtySpawn.mockImplementationOnce(() => {
+            throw new Error('spawn failed');
+        });
+        ws.emit('message', JSON.stringify({ type: 'create' }));
+        expect(logger.error).toHaveBeenCalledWith('Error in setup terminal', expect.objectContaining({ message: 'spawn failed' }));
+        expect(ws.send).toHaveBeenCalledWith(JSON.stringify({ type: 'error', message: 'spawn failed' }));
+        ws.emit('message', JSON.stringify({ type: 'create' }));
+        ws.emit('close');
+        expect(mockPtyKill).toHaveBeenCalledTimes(1);
+        ws.emit('error', new Error('socket error'));
+        expect(logger.error).toHaveBeenCalledWith('Setup terminal WebSocket error', expect.objectContaining({ message: 'socket error' }));
+        ws.readyState = WS_READY_STATE.CLOSED;
+        const sentBeforeClosed = ws.send.mock.calls.length;
+        ws.emit('message', JSON.stringify({ type: 'ping' }));
+        ws.emit('message', JSON.stringify({ type: 'resize', cols: '120', rows: 40 }));
+        onDataCallback?.('ignored output');
+        expect(ws.send).toHaveBeenCalledTimes(sentBeforeClosed);
+    });
+});
+//# sourceMappingURL=setup-terminal.handler.test.js.map

package/dist/src/backend/routers/websocket/setup-terminal.handler.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"setup-terminal.handler.test.js","sourceRoot":"","sources":["../../../../../src/backend/routers/websocket/setup-terminal.handler.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAG3C,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAG9D,OAAO,EAAE,cAAc,EAAE,MAAM,+BAA+B,CAAC;AAE/D,MAAM,YAAY,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;AAC7B,MAAM,aAAa,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;AAC9B,MAAM,WAAW,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;AAC5B,MAAM,YAAY,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;AAE7B,IAAI,cAAc,GAAoC,IAAI,CAAC;AAC3D,IAAI,cAAc,GAAmD,IAAI,CAAC;AAE1E,EAAE,CAAC,IAAI,CAAC,aAAa,EAAE,GAAG,EAAE,CAAC,CAAC;IAC5B,aAAa,EAAE,GAAG,EAAE,CAAC,CAAC,EAAU,EAAE,EAAE;QAClC,IAAI,EAAE,KAAK,UAAU,EAAE,CAAC;YACtB,OAAO;gBACL,KAAK,EAAE,YAAY;aACpB,CAAC;QACJ,CAAC;QACD,MAAM,IAAI,KAAK,CAAC,uBAAuB,EAAE,EAAE,CAAC,CAAC;IAC/C,CAAC;CACF,CAAC,CAAC,CAAC;AAEJ,OAAO,EAAE,iCAAiC,EAAE,MAAM,0BAA0B,CAAC;AAE7E,MAAM,aAAc,SAAQ,YAAY;IACtC,UAAU,GAAW,cAAc,CAAC,IAAI,CAAC;IACzC,IAAI,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;CAChB;AAED,SAAS,YAAY;IACnB,OAAO;QACL,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;QACd,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;QACb,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;QACb,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;KACf,CAAC;AACJ,CAAC;AAED,SAAS,aAAa;IACpB,OAAO;QACL,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,QAAgC,EAAE,EAAE;YACjD,cAAc,GAAG,QAAQ,CAAC;QAC5B,CAAC,CAAC;QACF,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,QAA+C,EAAE,EAAE;YAChE,cAAc,GAAG,QAAQ,CAAC;QAC5B,CAAC,CAAC;QACF,KAAK,EAAE,YAAY;QACnB,MAAM,EAAE,aAAa;QACrB,IAAI,EAAE,WAAW;KAClB,CAAC;AACJ,CAAC;AAED,SAAS,SAAS,CAAC,EAAiB;IAClC,OAAO;QACL,aAAa,EAAE,EAAE,CAAC,EAAE,CAClB,CACE,QAAyB,EACzB,OAAe,EACf,KAAa,EACb,QAAqC,EACrC,EAAE,CAAC,QAAQ,CAAC,EAA0B,CAAC,CAC1C;KAC4B,CAAC;AAClC,CAAC;AAED,QAAQ,CAAC,mCAAmC,EAAE,GAAG,EAAE;IACjD,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,cAAc,GAAG,IAAI,CAAC;QACtB,cAAc,GAAG,IAAI,CAAC;QACtB,YAAY,CAAC,eAAe,CAAC,aAAa,EAAE,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;QACxD,MAAM,MAAM,GAAG,YAAY,EAAE,CAAC;QAC9B,MAAM,aAAa,GAAG;YACpB,YAAY,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC;YACrC,kBAAkB,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,CAAC;SACxD,CAAC;QACF,MAAM,UAAU,GAAG;YACjB,QAAQ,EAAE;gBACR,YAAY,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC;gBACjC,aAAa;aACd;SACuB,CAAC;QAE3B,MAAM,OAAO,GAAG,iCAAiC,CAAC,UAAU,CAAC,CAAC;QAC9D,MAAM,EAAE,GAAG,IAAI,aAAa,EAAE,CAAC;QAC/B,MAAM,GAAG,GAAG,SAAS,CAAC,EAAE,CAAC,CAAC;QAC1B,MAAM,OAAO,GAAG,EAAqB,CAAC;QACtC,MAAM,MAAM,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,EAAuB,CAAC;QACzE,MAAM,UAAU,GAAG,IAAI,OAAO,EAAsB,CAAC;QAErD,OAAO,CACL,OAAO,EACP,MAAM,EACN,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EACf,IAAI,GAAG,CAAC,iCAAiC,CAAC,EAC1C,GAAG,EACH,UAAU,CACX,CAAC;QAEF,EAAE,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;QAE5E,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CACvC,UAAU,EACV,EAAE,EACF,MAAM,CAAC,gBAAgB,CAAC;YACtB,IAAI,EAAE,gBAAgB;YACtB,IAAI,EAAE,GAAG;YACT,IAAI,EAAE,EAAE;YACR,GAAG,EAAE,MAAM,CAAC,gBAAgB,CAAC;gBAC3B,IAAI,EAAE,UAAU;gBAChB,IAAI,EAAE,gBAAgB;gBACtB,SAAS,EAAE,WAAW;aACvB,CAAC;SACH,CAAC,CACH,CAAC;QACF,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC;QAE1E,cAAc,EAAE,CAAC,kBAAkB,CAAC,CAAC;QACrC,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAClC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,kBAAkB,EAAE,CAAC,CAC7D,CAAC;QAEF,EAAE,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC;QACzE,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,WAAW,CAAC,CAAC;QAEvD,EAAE,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;QAC5E,MAAM,CAAC,aAAa,CAAC,CAAC,oBAAoB,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;QAEpD,EAAE,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;QACrD,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;QAEvE,EAAE,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC;QACvD,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAClC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,yBAAyB,EAAE,CAAC,CACtE,CAAC;QAEF,cAAc,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC;QAClC,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QAEpF,EAAE,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC;QACvD,MAAM,CAAC,YAAY,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,MAAM,GAAG,YAAY,EAAE,CAAC;QAC9B,MAAM,aAAa,GAAG;YACpB,YAAY,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC;YACrC,kBAAkB,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;SACtC,CAAC;QACF,MAAM,UAAU,GAAG;YACjB,QAAQ,EAAE;gBACR,YAAY,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC;gBACjC,aAAa;aACd;SACuB,CAAC;QAE3B,MAAM,OAAO,GAAG,iCAAiC,CAAC,UAAU,CAAC,CAAC;QAC9D,MAAM,EAAE,GAAG,IAAI,aAAa,EAAE,CAAC;QAC/B,MAAM,GAAG,GAAG,SAAS,CAAC,EAAE,CAAC,CAAC;QAC1B,MAAM,OAAO,GAAG,EAAqB,CAAC;QACtC,MAAM,MAAM,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,EAAuB,CAAC;QACzE,MAAM,UAAU,GAAG,IAAI,OAAO,EAAsB,CAAC;QAErD,OAAO,CACL,OAAO,EACP,MAAM,EACN,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EACf,IAAI,GAAG,CAAC,iCAAiC,CAAC,EAC1C,GAAG,EACH,UAAU,CACX,CAAC;QAEF,EAAE,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;QAE9E,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAClC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,wBAAwB,EAAE,CAAC,CACrE,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,uCAAuC,EAAE;YAChF,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC;SAC1B,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;QACvE,MAAM,MAAM,GAAG,YAAY,EAAE,CAAC;QAC9B,MAAM,aAAa,GAAG;YACpB,YAAY,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC;YACrC,kBAAkB,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;SACtC,CAAC;QACF,MAAM,UAAU,GAAG;YACjB,QAAQ,EAAE;gBACR,YAAY,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC;gBACjC,aAAa;aACd;SACuB,CAAC;QAE3B,MAAM,OAAO,GAAG,iCAAiC,CAAC,UAAU,CAAC,CAAC;QAC9D,MAAM,EAAE,GAAG,IAAI,aAAa,EAAE,CAAC;QAC/B,MAAM,GAAG,GAAG,SAAS,CAAC,EAAE,CAAC,CAAC;QAC1B,MAAM,OAAO,GAAG,EAAqB,CAAC;QACtC,MAAM,MAAM,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,EAAuB,CAAC;QACzE,MAAM,UAAU,GAAG,IAAI,OAAO,EAAsB,CAAC;QAErD,OAAO,CACL,OAAO,EACP,MAAM,EACN,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EACf,IAAI,GAAG,CAAC,iCAAiC,CAAC,EAC1C,GAAG,EACH,UAAU,CACX,CAAC;QAEF,EAAE,CAAC,IAAI,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC;QACxB,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAAC,yBAAyB,EAAE,MAAM,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,CAAC;QAC9F,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAClC,MAAM,CAAC,cAAc,CAAC,mCAAmC,CAAC,CAC3D,CAAC;QAEF,YAAY,CAAC,sBAAsB,CAAC,GAAG,EAAE;YACvC,MAAM,IAAI,KAAK,CAAC,cAAc,CAAC,CAAC;QAClC,CAAC,CAAC,CAAC;QACH,EAAE,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC;QACvD,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,oBAAoB,CACvC,yBAAyB,EACzB,MAAM,CAAC,gBAAgB,CAAC,EAAE,OAAO,EAAE,cAAc,EAAE,CAAC,CACrD,CAAC;QACF,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAClC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,cAAc,EAAE,CAAC,CAC3D,CAAC;QAEF,EAAE,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC;QACvD,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACjB,MAAM,CAAC,WAAW,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAE7C,EAAE,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC;QAC5C,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,oBAAoB,CACvC,gCAAgC,EAChC,MAAM,CAAC,gBAAgB,CAAC,EAAE,OAAO,EAAE,cAAc,EAAE,CAAC,CACrD,CAAC;QAEF,EAAE,CAAC,UAAU,GAAG,cAAc,CAAC,MAAM,CAAC;QACtC,MAAM,gBAAgB,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC;QACnD,EAAE,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;QACrD,EAAE,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;QAC9E,cAAc,EAAE,CAAC,gBAAgB,CAAC,CAAC;QACnC,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,qBAAqB,CAAC,gBAAgB,CAAC,CAAC;IAC1D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/backend/routers/websocket/snapshots.handler.d.ts

@@ -1,9 +1,20 @@
 import type { IncomingMessage } from 'node:http';
 import type { Duplex } from 'node:stream';
 import type { WebSocket, WebSocketServer } from 'ws';
-import { type AppContext } from '../../../backend/app-context.js';
+import type { AppContext } from '../../../backend/app-context.js';
 export type SnapshotConnectionsMap = Map<string, Set<WebSocket>>;
 export declare const snapshotConnections: SnapshotConnectionsMap;
-export declare function createSnapshotsUpgradeHandler(appContext: AppContext): (request: IncomingMessage, socket: Duplex, head: Buffer, url: URL, wss: WebSocketServer, wsAliveMap: WeakMap<WebSocket, boolean>) => void;
-export declare const handleSnapshotsUpgrade: (request: IncomingMessage, socket: Duplex, head: Buffer, url: URL, wss: WebSocketServer, wsAliveMap: WeakMap<WebSocket, boolean>) => void;
+declare class SnapshotStoreSubscriptionState {
+    private active;
+    private snapshotChangedListener;
+    private snapshotRemovedListener;
+    ensure(connections: SnapshotConnectionsMap, logger: ReturnType<AppContext['services']['createLogger']>): void;
+    reset(): void;
+}
+export declare function resetSnapshotsHandlerStateForTests(): void;
+export declare function createSnapshotsUpgradeHandler(appContext: AppContext, options?: {
+    connections?: SnapshotConnectionsMap;
+    subscriptionState?: SnapshotStoreSubscriptionState;
+}): (request: IncomingMessage, socket: Duplex, head: Buffer, url: URL, wss: WebSocketServer, wsAliveMap: WeakMap<WebSocket, boolean>) => void;
+export {};
 //# sourceMappingURL=snapshots.handler.d.ts.map

package/dist/src/backend/routers/websocket/snapshots.handler.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"snapshots.handler.d.ts","sourceRoot":"","sources":["../../../../../src/backend/routers/websocket/snapshots.handler.ts"],"names":[],"mappings":"AAQA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,WAAW,CAAC;AACjD,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,KAAK,EAAE,SAAS,EAAE,eAAe,EAAE,MAAM,IAAI,CAAC;AACrD,OAAO,EAAE,KAAK,UAAU,EAAoB,MAAM,uBAAuB,CAAC;AAkB1E,MAAM,MAAM,sBAAsB,GAAG,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC;AAMjE,eAAO,MAAM,mBAAmB,EAAE,sBAAkC,CAAC;AA6DrE,wBAAgB,6BAA6B,CAAC,UAAU,EAAE,UAAU,IAMhE,SAAS,eAAe,EACxB,QAAQ,MAAM,EACd,MAAM,MAAM,EACZ,KAAK,GAAG,EACR,KAAK,eAAe,EACpB,YAAY,OAAO,CAAC,SAAS,EAAE,OAAO,CAAC,KACtC,IAAI,CA4CR;AAED,eAAO,MAAM,sBAAsB,YApDtB,eAAe,UAChB,MAAM,QACR,MAAM,OACP,GAAG,OACH,eAAe,cACR,OAAO,CAAC,SAAS,EAAE,OAAO,CAAC,KACtC,IA8CkF,CAAC"}
+{"version":3,"file":"snapshots.handler.d.ts","sourceRoot":"","sources":["../../../../../src/backend/routers/websocket/snapshots.handler.ts"],"names":[],"mappings":"AAQA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,WAAW,CAAC;AACjD,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,KAAK,EAAE,SAAS,EAAE,eAAe,EAAE,MAAM,IAAI,CAAC;AACrD,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAmBxD,MAAM,MAAM,sBAAsB,GAAG,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC;AAMjE,eAAO,MAAM,mBAAmB,EAAE,sBAAkC,CAAC;AAMrE,cAAM,8BAA8B;IAClC,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,uBAAuB,CAAwD;IACvF,OAAO,CAAC,uBAAuB,CAAwD;IAEvF,MAAM,CACJ,WAAW,EAAE,sBAAsB,EACnC,MAAM,EAAE,UAAU,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,cAAc,CAAC,CAAC,GACzD,IAAI;IA2DP,KAAK,IAAI,IAAI;CAiBd;AAQD,wBAAgB,kCAAkC,IAAI,IAAI,CAGzD;AAMD,wBAAgB,6BAA6B,CAC3C,UAAU,EAAE,UAAU,EACtB,OAAO,GAAE;IACP,WAAW,CAAC,EAAE,sBAAsB,CAAC;IACrC,iBAAiB,CAAC,EAAE,8BAA8B,CAAC;CAC/C,IAOJ,SAAS,eAAe,EACxB,QAAQ,MAAM,EACd,MAAM,MAAM,EACZ,KAAK,GAAG,EACR,KAAK,eAAe,EACpB,YAAY,OAAO,CAAC,SAAS,EAAE,OAAO,CAAC,KACtC,IAAI,CAgDR"}