wind-agent-cli 1.37.2

package/dist-server/server/modules/websocket/services/chat-run-registry.service.js

@@ -0,0 +1,266 @@
+import path from 'node:path';
+import { projectsDb, sessionsDb } from '../../../modules/database/index.js';
+import { generateDisplayName } from '../../../modules/projects/index.js';
+import { ChatSessionWriter } from '../../../modules/websocket/services/chat-session-writer.service.js';
+import { connectedClients, WS_OPEN_STATE } from '../../../modules/websocket/services/websocket-state.service.js';
+/**
+ * How long a completed run stays available for replay. Covers the window
+ * between a run finishing and the client refreshing history over REST (for
+ * example when the browser tab was asleep while the run completed).
+ */
+const COMPLETED_RUN_RETENTION_MS = 5 * 60 * 1000;
+/**
+ * Upper bound on buffered events per run so a very long tool-heavy run cannot
+ * grow memory unbounded. When exceeded, the oldest events are dropped —
+ * a reconnecting client whose `lastSeq` predates the buffer falls back to a
+ * REST history refresh, which is always the authoritative source.
+ */
+const MAX_BUFFERED_EVENTS_PER_RUN = 5000;
+/**
+ * Active and recently-completed runs keyed by app session id.
+ *
+ * This map is the single in-memory source of truth for "is something running
+ * for this session" — the chat websocket handler, abort path, and subscribe
+ * path all consult it instead of asking each provider runtime individually.
+ */
+const runs = new Map();
+async function broadcastCanonicalSessionUpsert(appSessionId) {
+    const row = sessionsDb.getSessionById(appSessionId);
+    if (!row || row.isArchived) {
+        return;
+    }
+    const projectPath = row.project_path;
+    const project = projectPath ? projectsDb.getProjectPath(projectPath) : null;
+    const displayName = project?.custom_project_name?.trim()
+        ? project.custom_project_name
+        : await generateDisplayName(path.basename(projectPath ?? '') || (projectPath ?? ''), projectPath);
+    const payload = JSON.stringify({
+        kind: 'session_upserted',
+        sessionId: row.session_id,
+        providerSessionId: row.provider_session_id,
+        provider: row.provider,
+        session: {
+            id: row.session_id,
+            summary: row.custom_name || '',
+            messageCount: 0,
+            lastActivity: row.updated_at ?? row.created_at ?? new Date().toISOString(),
+        },
+        project: project
+            ? {
+                projectId: project.project_id,
+                path: project.project_path,
+                fullPath: project.project_path,
+                displayName,
+                isStarred: Boolean(project.isStarred),
+            }
+            : null,
+        timestamp: new Date().toISOString(),
+    });
+    connectedClients.forEach((client) => {
+        if (client.readyState === WS_OPEN_STATE) {
+            client.send(payload);
+        }
+    });
+}
+function evictRunLater(appSessionId) {
+    const timer = setTimeout(() => {
+        const run = runs.get(appSessionId);
+        if (run && run.status === 'completed') {
+            runs.delete(appSessionId);
+        }
+    }, COMPLETED_RUN_RETENTION_MS);
+    // Never keep the process alive just to evict a buffered run.
+    timer.unref?.();
+}
+/**
+ * Decorates one outbound live event for a run and records it in the event log.
+ *
+ * Responsibilities:
+ * 1. Remap `sessionId` (and `actualSessionId` on `complete`) to the stable
+ *    app session id — provider-native ids never leave the backend.
+ * 2. Assign the next `seq` so clients can detect/replay gaps.
+ * 3. Buffer the event for `chat.subscribe` replay.
+ * 4. Flip the run to `completed` when the terminal `complete` event passes by.
+ */
+function decorateAndRecordEvent(run, message) {
+    // Exactly-one-complete contract: when a run is aborted the chat handler
+    // emits the terminal `complete` immediately, but the killed runtime may
+    // still emit its own `complete` from its exit handler moments later.
+    // Whichever arrives first wins; the duplicate is dropped here.
+    if (message.kind === 'complete' && run.status === 'completed') {
+        return null;
+    }
+    run.lastSeq += 1;
+    const outbound = {
+        ...message,
+        sessionId: run.appSessionId,
+        seq: run.lastSeq,
+    };
+    if (message.kind === 'complete') {
+        // The provider may report its own id here; the frontend only ever knows
+        // the app id, so the "actual" id is by definition the app id as well.
+        outbound.actualSessionId = run.appSessionId;
+        run.status = 'completed';
+        run.completedAt = Date.now();
+        evictRunLater(run.appSessionId);
+    }
+    run.events.push(outbound);
+    if (run.events.length > MAX_BUFFERED_EVENTS_PER_RUN) {
+        run.events.splice(0, run.events.length - MAX_BUFFERED_EVENTS_PER_RUN);
+    }
+    return outbound;
+}
+/**
+ * Records the provider-native session id for a run and persists the
+ * app-id-to-provider-id mapping so history fetches and future resumes can
+ * address the provider transcript.
+ *
+ * Called from the gateway writer when the runtime either calls
+ * `setSessionId(...)` or emits its `session_created` event — whichever
+ * happens first wins; later calls with the same id are no-ops.
+ */
+function recordProviderSessionId(run, providerSessionId) {
+    if (!providerSessionId || run.providerSessionId === providerSessionId) {
+        return;
+    }
+    run.providerSessionId = providerSessionId;
+    try {
+        sessionsDb.assignProviderSessionId(run.appSessionId, providerSessionId);
+        void broadcastCanonicalSessionUpsert(run.appSessionId).catch((error) => {
+            const message = error instanceof Error ? error.message : String(error);
+            console.error('[ChatRunRegistry] Failed to broadcast canonical session mapping', {
+                appSessionId: run.appSessionId,
+                providerSessionId,
+                error: message,
+            });
+        });
+    }
+    catch (error) {
+        const message = error instanceof Error ? error.message : String(error);
+        console.error('[ChatRunRegistry] Failed to persist provider session id mapping', {
+            appSessionId: run.appSessionId,
+            providerSessionId,
+            error: message,
+        });
+    }
+}
+/**
+ * Registry of live provider runs keyed by the stable app session id.
+ *
+ * The registry is what makes the websocket protocol provider-independent:
+ * every run gets a `ChatSessionWriter` that remaps provider-native session
+ * ids to the app id, assigns `seq` numbers, and buffers events for replay —
+ * regardless of which provider runtime produced them.
+ */
+export const chatRunRegistry = {
+    /**
+     * Starts tracking a run and returns it, or `null` when a run is already in
+     * progress for the session (callers must reject the duplicate send).
+     */
+    startRun(input) {
+        const existing = runs.get(input.appSessionId);
+        if (existing && existing.status === 'running') {
+            return null;
+        }
+        const run = {
+            appSessionId: input.appSessionId,
+            provider: input.provider,
+            providerSessionId: input.providerSessionId,
+            status: 'running',
+            lastSeq: 0,
+            events: [],
+            writer: null,
+            startedAt: Date.now(),
+            completedAt: null,
+        };
+        run.writer = new ChatSessionWriter({
+            connection: input.connection,
+            userId: input.userId,
+            provider: input.provider,
+            providerSessionId: input.providerSessionId,
+            onProviderSessionId: (providerSessionId) => {
+                recordProviderSessionId(run, providerSessionId);
+            },
+            decorateOutboundEvent: (message) => decorateAndRecordEvent(run, message),
+        });
+        runs.set(input.appSessionId, run);
+        return run;
+    },
+    getRun(appSessionId) {
+        return runs.get(appSessionId);
+    },
+    isProcessing(appSessionId) {
+        return runs.get(appSessionId)?.status === 'running';
+    },
+    listRunningRuns() {
+        return Array.from(runs.values())
+            .filter((run) => run.status === 'running')
+            .map((run) => ({
+            sessionId: run.appSessionId,
+            provider: run.provider,
+            startedAt: run.startedAt,
+            lastSeq: run.lastSeq,
+        }));
+    },
+    /**
+     * Re-attaches a run's outbound stream to a (new) websocket connection.
+     *
+     * This is the generic replacement for the Claude-only writer reconnect:
+     * after a page refresh the new socket subscribes and immediately starts
+     * receiving the still-running stream, for every provider.
+     */
+    attachConnection(appSessionId, connection) {
+        const run = runs.get(appSessionId);
+        if (!run) {
+            return false;
+        }
+        run.writer.updateWebSocket(connection);
+        return true;
+    },
+    /**
+     * Returns buffered events with `seq` greater than `afterSeq` for replay.
+     *
+     * An empty array with `run.lastSeq > afterSeq` not covered by the buffer
+     * means the buffer was truncated; the client should refresh over REST.
+     */
+    replayEvents(appSessionId, afterSeq) {
+        const run = runs.get(appSessionId);
+        if (!run) {
+            return [];
+        }
+        return run.events.filter((event) => typeof event.seq === 'number' && event.seq > afterSeq);
+    },
+    /**
+     * Emits a synthetic terminal `complete` if (and only if) the run is still
+     * marked running. Used when a provider runtime throws or resolves without
+     * having produced its own terminal event, and by the abort path.
+     */
+    completeRun(appSessionId, opts) {
+        const run = runs.get(appSessionId);
+        if (!run || run.status !== 'running') {
+            return;
+        }
+        run.writer.sendComplete(opts);
+    },
+    /**
+     * Safety-net variant of `completeRun` scoped to one specific run: a no-op
+     * unless `run` is still the session's current, running run. A runtime
+     * promise can resolve after its own `complete` already streamed AND a new
+     * run has replaced it in the registry (a queued message sends within
+     * milliseconds of the previous turn ending) — the session-keyed
+     * `completeRun` would terminate that newer run.
+     */
+    completeRunIfCurrent(run, opts) {
+        if (runs.get(run.appSessionId) !== run || run.status !== 'running') {
+            return;
+        }
+        run.writer.sendComplete(opts);
+    },
+    /**
+     * Test-only escape hatch: clears every tracked run.
+     */
+    clearAll() {
+        runs.clear();
+    },
+};
+//# sourceMappingURL=chat-run-registry.service.js.map

package/dist-server/server/modules/websocket/services/chat-run-registry.service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chat-run-registry.service.js","sourceRoot":"","sources":["../../../../../server/modules/websocket/services/chat-run-registry.service.ts"],"names":[],"mappings":"AAAA,OAAO,IAAI,MAAM,WAAW,CAAC;AAE7B,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,6BAA6B,CAAC;AACrE,OAAO,EAAE,mBAAmB,EAAE,MAAM,6BAA6B,CAAC;AAClE,OAAO,EAAE,iBAAiB,EAAE,MAAM,6DAA6D,CAAC;AAChG,OAAO,EAAE,gBAAgB,EAAE,aAAa,EAAE,MAAM,yDAAyD,CAAC;AAmC1G;;;;GAIG;AACH,MAAM,0BAA0B,GAAG,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;AAEjD;;;;;GAKG;AACH,MAAM,2BAA2B,GAAG,IAAI,CAAC;AAEzC;;;;;;GAMG;AACH,MAAM,IAAI,GAAG,IAAI,GAAG,EAAmB,CAAC;AAExC,KAAK,UAAU,+BAA+B,CAAC,YAAoB;IACjE,MAAM,GAAG,GAAG,UAAU,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;IACpD,IAAI,CAAC,GAAG,IAAI,GAAG,CAAC,UAAU,EAAE,CAAC;QAC3B,OAAO;IACT,CAAC;IAED,MAAM,WAAW,GAAG,GAAG,CAAC,YAAY,CAAC;IACrC,MAAM,OAAO,GAAG,WAAW,CAAC,CAAC,CAAC,UAAU,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;IAC5E,MAAM,WAAW,GAAG,OAAO,EAAE,mBAAmB,EAAE,IAAI,EAAE;QACtD,CAAC,CAAC,OAAO,CAAC,mBAAmB;QAC7B,CAAC,CAAC,MAAM,mBAAmB,CAAC,IAAI,CAAC,QAAQ,CAAC,WAAW,IAAI,EAAE,CAAC,IAAI,CAAC,WAAW,IAAI,EAAE,CAAC,EAAE,WAAW,CAAC,CAAC;IAEpG,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC;QAC7B,IAAI,EAAE,kBAAkB;QACxB,SAAS,EAAE,GAAG,CAAC,UAAU;QACzB,iBAAiB,EAAE,GAAG,CAAC,mBAAmB;QAC1C,QAAQ,EAAE,GAAG,CAAC,QAAQ;QACtB,OAAO,EAAE;YACP,EAAE,EAAE,GAAG,CAAC,UAAU;YAClB,OAAO,EAAE,GAAG,CAAC,WAAW,IAAI,EAAE;YAC9B,YAAY,EAAE,CAAC;YACf,YAAY,EAAE,GAAG,CAAC,UAAU,IAAI,GAAG,CAAC,UAAU,IAAI,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SAC3E;QACD,OAAO,EAAE,OAAO;YACd,CAAC,CAAC;gBACA,SAAS,EAAE,OAAO,CAAC,UAAU;gBAC7B,IAAI,EAAE,OAAO,CAAC,YAAY;gBAC1B,QAAQ,EAAE,OAAO,CAAC,YAAY;gBAC9B,WAAW;gBACX,SAAS,EAAE,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC;aACtC;YACD,CAAC,CAAC,IAAI;QACR,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;KACpC,CAAC,CAAC;IAEH,gBAAgB,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;QAClC,IAAI,MAAM,CAAC,UAAU,KAAK,aAAa,EAAE,CAAC;YACxC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACvB,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,aAAa,CAAC,YAAoB;IACzC,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE;QAC5B,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QACnC,IAAI,GAAG,IAAI,GAAG,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;YACtC,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;QAC5B,CAAC;IACH,CAAC,EAAE,0BAA0B,CAAC,CAAC;IAE/B,6DAA6D;IAC7D,KAAK,CAAC,KAAK,EAAE,EAAE,CAAC;AAClB,CAAC;AAED;;;;;;;;;GASG;AACH,SAAS,sBAAsB,CAAC,GAAY,EAAE,OAA0B;IACtE,wEAAwE;IACxE,wEAAwE;IACxE,qEAAqE;IACrE,+DAA+D;IAC/D,IAAI,OAAO,CAAC,IAAI,KAAK,UAAU,IAAI,GAAG,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;QAC9D,OAAO,IAAI,CAAC;IACd,CAAC;IAED,GAAG,CAAC,OAAO,IAAI,CAAC,CAAC;IAEjB,MAAM,QAAQ,GAAsB;QAClC,GAAG,OAAO;QACV,SAAS,EAAE,GAAG,CAAC,YAAY;QAC3B,GAAG,EAAE,GAAG,CAAC,OAAO;KACjB,CAAC;IAEF,IAAI,OAAO,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;QAChC,wEAAwE;QACxE,sEAAsE;QACtE,QAAQ,CAAC,eAAe,GAAG,GAAG,CAAC,YAAY,CAAC;QAC5C,GAAG,CAAC,MAAM,GAAG,WAAW,CAAC;QACzB,GAAG,CAAC,WAAW,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,aAAa,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;IAClC,CAAC;IAED,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC1B,IAAI,GAAG,CAAC,MAAM,CAAC,MAAM,GAAG,2BAA2B,EAAE,CAAC;QACpD,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,EAAE,GAAG,CAAC,MAAM,CAAC,MAAM,GAAG,2BAA2B,CAAC,CAAC;IACxE,CAAC;IAED,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED;;;;;;;;GAQG;AACH,SAAS,uBAAuB,CAAC,GAAY,EAAE,iBAAyB;IACtE,IAAI,CAAC,iBAAiB,IAAI,GAAG,CAAC,iBAAiB,KAAK,iBAAiB,EAAE,CAAC;QACtE,OAAO;IACT,CAAC;IAED,GAAG,CAAC,iBAAiB,GAAG,iBAAiB,CAAC;IAE1C,IAAI,CAAC;QACH,UAAU,CAAC,uBAAuB,CAAC,GAAG,CAAC,YAAY,EAAE,iBAAiB,CAAC,CAAC;QACxE,KAAK,+BAA+B,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;YACrE,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACvE,OAAO,CAAC,KAAK,CAAC,iEAAiE,EAAE;gBAC/E,YAAY,EAAE,GAAG,CAAC,YAAY;gBAC9B,iBAAiB;gBACjB,KAAK,EAAE,OAAO;aACf,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACvE,OAAO,CAAC,KAAK,CAAC,iEAAiE,EAAE;YAC/E,YAAY,EAAE,GAAG,CAAC,YAAY;YAC9B,iBAAiB;YACjB,KAAK,EAAE,OAAO;SACf,CAAC,CAAC;IACL,CAAC;AACH,CAAC;AAED;;;;;;;GAOG;AACH,MAAM,CAAC,MAAM,eAAe,GAAG;IAC7B;;;OAGG;IACH,QAAQ,CAAC,KAMR;QACC,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QAC9C,IAAI,QAAQ,IAAI,QAAQ,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;YAC9C,OAAO,IAAI,CAAC;QACd,CAAC;QAED,MAAM,GAAG,GAAY;YACnB,YAAY,EAAE,KAAK,CAAC,YAAY;YAChC,QAAQ,EAAE,KAAK,CAAC,QAAQ;YACxB,iBAAiB,EAAE,KAAK,CAAC,iBAAiB;YAC1C,MAAM,EAAE,SAAS;YACjB,OAAO,EAAE,CAAC;YACV,MAAM,EAAE,EAAE;YACV,MAAM,EAAE,IAAoC;YAC5C,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,WAAW,EAAE,IAAI;SAClB,CAAC;QAEF,GAAG,CAAC,MAAM,GAAG,IAAI,iBAAiB,CAAC;YACjC,UAAU,EAAE,KAAK,CAAC,UAAU;YAC5B,MAAM,EAAE,KAAK,CAAC,MAAM;YACpB,QAAQ,EAAE,KAAK,CAAC,QAAQ;YACxB,iBAAiB,EAAE,KAAK,CAAC,iBAAiB;YAC1C,mBAAmB,EAAE,CAAC,iBAAiB,EAAE,EAAE;gBACzC,uBAAuB,CAAC,GAAG,EAAE,iBAAiB,CAAC,CAAC;YAClD,CAAC;YACD,qBAAqB,EAAE,CAAC,OAAO,EAAE,EAAE,CAAC,sBAAsB,CAAC,GAAG,EAAE,OAAO,CAAC;SACzE,CAAC,CAAC;QAEH,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,YAAY,EAAE,GAAG,CAAC,CAAC;QAClC,OAAO,GAAG,CAAC;IACb,CAAC;IAED,MAAM,CAAC,YAAoB;QACzB,OAAO,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;IAChC,CAAC;IAED,YAAY,CAAC,YAAoB;QAC/B,OAAO,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE,MAAM,KAAK,SAAS,CAAC;IACtD,CAAC;IAED,eAAe;QAMb,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;aAC7B,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,MAAM,KAAK,SAAS,CAAC;aACzC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;YACb,SAAS,EAAE,GAAG,CAAC,YAAY;YAC3B,QAAQ,EAAE,GAAG,CAAC,QAAQ;YACtB,SAAS,EAAE,GAAG,CAAC,SAAS;YACxB,OAAO,EAAE,GAAG,CAAC,OAAO;SACrB,CAAC,CAAC,CAAC;IACR,CAAC;IAED;;;;;;OAMG;IACH,gBAAgB,CAAC,YAAoB,EAAE,UAAoC;QACzE,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QACnC,IAAI,CAAC,GAAG,EAAE,CAAC;YACT,OAAO,KAAK,CAAC;QACf,CAAC;QAED,GAAG,CAAC,MAAM,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;QACvC,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;;;;OAKG;IACH,YAAY,CAAC,YAAoB,EAAE,QAAgB;QACjD,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QACnC,IAAI,CAAC,GAAG,EAAE,CAAC;YACT,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,OAAO,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,OAAO,KAAK,CAAC,GAAG,KAAK,QAAQ,IAAI,KAAK,CAAC,GAAG,GAAG,QAAQ,CAAC,CAAC;IAC7F,CAAC;IAED;;;;OAIG;IACH,WAAW,CAAC,YAAoB,EAAE,IAA6C;QAC7E,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QACnC,IAAI,CAAC,GAAG,IAAI,GAAG,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;YACrC,OAAO;QACT,CAAC;QAED,GAAG,CAAC,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;IAChC,CAAC;IAED;;;;;;;OAOG;IACH,oBAAoB,CAAC,GAAY,EAAE,IAA6C;QAC9E,IAAI,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,YAAY,CAAC,KAAK,GAAG,IAAI,GAAG,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;YACnE,OAAO;QACT,CAAC;QAED,GAAG,CAAC,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;IAChC,CAAC;IAED;;OAEG;IACH,QAAQ;QACN,IAAI,CAAC,KAAK,EAAE,CAAC;IACf,CAAC;CACF,CAAC"}

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

@@ -0,0 +1,105 @@
+import { WS_OPEN_STATE } from '../../../modules/websocket/services/websocket-state.service.js';
+import { createCompleteMessage, readObjectRecord } from '../../../shared/utils.js';
+/**
+ * Gateway writer handed to provider runtimes instead of a raw websocket writer.
+ *
+ * It exposes the exact same surface as `WebSocketWriter` (`send`,
+ * `setSessionId`, `getSessionId`, `updateWebSocket`, `userId`,
+ * `isWebSocketWriter`) so the provider runtime adapters need zero changes —
+ * but everything that flows through
+ * it is translated from the provider's world into the app's protocol:
+ *
+ * - `session_created` events are swallowed and turned into a provider-id
+ *   mapping; the frontend never learns provider-native ids.
+ * - every other event gets `sessionId` remapped to the app session id and a
+ *   per-run `seq` assigned before being forwarded.
+ * - `setSessionId(...)` calls (used by runtimes to label captured ids) are
+ *   intercepted and recorded as the provider-id mapping as well.
+ */
+export class ChatSessionWriter {
+    ws;
+    userId;
+    /**
+     * Some runtimes feature-detect their writer with this flag; keep it so the
+     * gateway writer is a drop-in replacement for `WebSocketWriter`.
+     */
+    isWebSocketWriter = true;
+    options;
+    /**
+     * The provider-native session id as the runtime knows it. Kept locally
+     * (besides the registry) because runtimes read it back via `getSessionId()`
+     * to label their own outgoing events — those labels are remapped on send
+     * anyway, but the runtime-visible value must stay provider-native.
+     */
+    providerSessionId;
+    constructor(options) {
+        this.options = options;
+        this.ws = options.connection;
+        this.userId = options.userId;
+        this.providerSessionId = options.providerSessionId;
+    }
+    send(data) {
+        const record = readObjectRecord(data);
+        if (!record || typeof record.kind !== 'string') {
+            // Provider runtimes only emit kind-based normalized messages. Anything
+            // else indicates a programming error; drop it rather than leaking an
+            // un-remapped payload to the client.
+            console.error('[ChatSessionWriter] Dropping non-normalized outbound payload', data);
+            return;
+        }
+        const message = record;
+        if (message.kind === 'session_created') {
+            const announcedId = typeof message.newSessionId === 'string' && message.newSessionId
+                ? message.newSessionId
+                : message.sessionId;
+            if (announcedId) {
+                this.captureProviderSessionId(announcedId);
+            }
+            // Swallowed on purpose: the frontend already has the stable app session
+            // id, so there is no client-side handoff to perform anymore.
+            return;
+        }
+        const outbound = this.options.decorateOutboundEvent(message);
+        if (outbound) {
+            this.forward(outbound);
+        }
+    }
+    /**
+     * Emits the synthetic terminal `complete` for runs that ended without one
+     * (runtime crash before completing, or user abort).
+     */
+    sendComplete(opts) {
+        const message = createCompleteMessage({
+            provider: this.options.provider,
+            sessionId: this.providerSessionId,
+            exitCode: opts.exitCode,
+            aborted: opts.aborted,
+        });
+        const outbound = this.options.decorateOutboundEvent(message);
+        if (outbound) {
+            this.forward(outbound);
+        }
+    }
+    updateWebSocket(newConnection) {
+        this.ws = newConnection;
+    }
+    setSessionId(sessionId) {
+        this.captureProviderSessionId(sessionId);
+    }
+    getSessionId() {
+        return this.providerSessionId;
+    }
+    captureProviderSessionId(providerSessionId) {
+        if (!providerSessionId || this.providerSessionId === providerSessionId) {
+            return;
+        }
+        this.providerSessionId = providerSessionId;
+        this.options.onProviderSessionId(providerSessionId);
+    }
+    forward(message) {
+        if (this.ws.readyState === WS_OPEN_STATE) {
+            this.ws.send(JSON.stringify(message));
+        }
+    }
+}
+//# sourceMappingURL=chat-session-writer.service.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"chat-session-writer.service.js","sourceRoot":"","sources":["../../../../../server/modules/websocket/services/chat-session-writer.service.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,yDAAyD,CAAC;AAMxF,OAAO,EAAE,qBAAqB,EAAE,gBAAgB,EAAE,MAAM,mBAAmB,CAAC;AAuB5E;;;;;;;;;;;;;;;GAeG;AACH,MAAM,OAAO,iBAAiB;IAC5B,EAAE,CAA2B;IAC7B,MAAM,CAAyB;IAC/B;;;OAGG;IACH,iBAAiB,GAAG,IAAI,CAAC;IAER,OAAO,CAA2B;IACnD;;;;;OAKG;IACK,iBAAiB,CAAgB;IAEzC,YAAY,OAAiC;QAC3C,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,EAAE,GAAG,OAAO,CAAC,UAAU,CAAC;QAC7B,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC7B,IAAI,CAAC,iBAAiB,GAAG,OAAO,CAAC,iBAAiB,CAAC;IACrD,CAAC;IAED,IAAI,CAAC,IAAa;QAChB,MAAM,MAAM,GAAG,gBAAgB,CAAC,IAAI,CAAC,CAAC;QACtC,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC/C,uEAAuE;YACvE,qEAAqE;YACrE,qCAAqC;YACrC,OAAO,CAAC,KAAK,CAAC,8DAA8D,EAAE,IAAI,CAAC,CAAC;YACpF,OAAO;QACT,CAAC;QAED,MAAM,OAAO,GAAG,MAA2B,CAAC;QAE5C,IAAI,OAAO,CAAC,IAAI,KAAK,iBAAiB,EAAE,CAAC;YACvC,MAAM,WAAW,GACf,OAAO,OAAO,CAAC,YAAY,KAAK,QAAQ,IAAI,OAAO,CAAC,YAAY;gBAC9D,CAAC,CAAC,OAAO,CAAC,YAAY;gBACtB,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC;YACxB,IAAI,WAAW,EAAE,CAAC;gBAChB,IAAI,CAAC,wBAAwB,CAAC,WAAW,CAAC,CAAC;YAC7C,CAAC;YACD,wEAAwE;YACxE,6DAA6D;YAC7D,OAAO;QACT,CAAC;QAED,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC;QAC7D,IAAI,QAAQ,EAAE,CAAC;YACb,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QACzB,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,YAAY,CAAC,IAA6C;QACxD,MAAM,OAAO,GAAG,qBAAqB,CAAC;YACpC,QAAQ,EAAE,IAAI,CAAC,OAAO,CAAC,QAAQ;YAC/B,SAAS,EAAE,IAAI,CAAC,iBAAiB;YACjC,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,OAAO,EAAE,IAAI,CAAC,OAAO;SACtB,CAAC,CAAC;QACH,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC;QAC7D,IAAI,QAAQ,EAAE,CAAC;YACb,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QACzB,CAAC;IACH,CAAC;IAED,eAAe,CAAC,aAAuC;QACrD,IAAI,CAAC,EAAE,GAAG,aAAa,CAAC;IAC1B,CAAC;IAED,YAAY,CAAC,SAAiB;QAC5B,IAAI,CAAC,wBAAwB,CAAC,SAAS,CAAC,CAAC;IAC3C,CAAC;IAED,YAAY;QACV,OAAO,IAAI,CAAC,iBAAiB,CAAC;IAChC,CAAC;IAEO,wBAAwB,CAAC,iBAAyB;QACxD,IAAI,CAAC,iBAAiB,IAAI,IAAI,CAAC,iBAAiB,KAAK,iBAAiB,EAAE,CAAC;YACvE,OAAO;QACT,CAAC;QAED,IAAI,CAAC,iBAAiB,GAAG,iBAAiB,CAAC;QAC3C,IAAI,CAAC,OAAO,CAAC,mBAAmB,CAAC,iBAAiB,CAAC,CAAC;IACtD,CAAC;IAEO,OAAO,CAAC,OAA0B;QACxC,IAAI,IAAI,CAAC,EAAE,CAAC,UAAU,KAAK,aAAa,EAAE,CAAC;YACzC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC;QACxC,CAAC;IACH,CAAC;CACF"}