feat(web): add WebSocket transport for message event streaming with SSE fallback (#764)

* feat: add websocket message stream transport * fix: avoid false missing session directories in sidebar * fix: re-probe project root session directories * refactor: use button group for message stream transport * fix: resolve chat input hook dependency warning --------- Co-authored-by: Bohdan Triapitsyn <artmore@protonmail.com>
2026-04-17 11:07:26 +03:00
parent fd8972a7d9
commit bee9d19f3a
26 changed files with 1515 additions and 163 deletions
@@ -0,0 +1,89 @@
+import { describe, expect, it } from 'bun:test';
+
+import {
+  MESSAGE_STREAM_DIRECTORY_WS_PATH,
+  MESSAGE_STREAM_GLOBAL_WS_PATH,
+  parseSseEventEnvelope,
+  sendMessageStreamWsEvent,
+  sendMessageStreamWsFrame,
+} from './protocol.js';
+
+describe('event stream protocol helpers', () => {
+  it('exports stable websocket paths', () => {
+    expect(MESSAGE_STREAM_GLOBAL_WS_PATH).toBe('/api/global/event/ws');
+    expect(MESSAGE_STREAM_DIRECTORY_WS_PATH).toBe('/api/event/ws');
+  });
+
+  it('parses wrapped SSE payloads with event id and directory', () => {
+    const envelope = parseSseEventEnvelope(
+      'id: evt-1\n' +
+      'event: message\n' +
+      'data: {"directory":"/tmp/project","payload":{"type":"session.updated"}}\n'
+    );
+
+    expect(envelope).toEqual({
+      eventId: 'evt-1',
+      directory: '/tmp/project',
+      payload: { type: 'session.updated' },
+    });
+  });
+
+  it('derives directory from payload properties when not wrapped', () => {
+    const envelope = parseSseEventEnvelope(
+      'data: {"type":"openchamber:notification","properties":{"directory":"/tmp/project"}}\n'
+    );
+
+    expect(envelope).toEqual({
+      eventId: null,
+      directory: '/tmp/project',
+      payload: {
+        type: 'openchamber:notification',
+        properties: { directory: '/tmp/project' },
+      },
+    });
+  });
+
+  it('returns null for malformed SSE blocks', () => {
+    expect(parseSseEventEnvelope('event: message\n')).toBeNull();
+    expect(parseSseEventEnvelope('data: {oops}\n')).toBeNull();
+  });
+
+  it('serializes generic websocket frames', () => {
+    let rawPayload = null;
+    const socket = {
+      readyState: 1,
+      send(payload) {
+        rawPayload = payload;
+      },
+    };
+
+    const sent = sendMessageStreamWsFrame(socket, { type: 'ready' });
+
+    expect(sent).toBe(true);
+    expect(rawPayload).toBe('{"type":"ready"}');
+  });
+
+  it('serializes event frames with routing metadata', () => {
+    let rawPayload = null;
+    const socket = {
+      readyState: 1,
+      send(payload) {
+        rawPayload = payload;
+      },
+    };
+
+    const sent = sendMessageStreamWsEvent(
+      socket,
+      { type: 'openchamber:heartbeat', timestamp: 1 },
+      { eventId: 'evt-2', directory: '/tmp/project' }
+    );
+
+    expect(sent).toBe(true);
+    expect(JSON.parse(rawPayload)).toEqual({
+      type: 'event',
+      payload: { type: 'openchamber:heartbeat', timestamp: 1 },
+      eventId: 'evt-2',
+      directory: '/tmp/project',
+    });
+  });
+});