@ccpocket/bridge 1.86.2 → 1.88.0

package/dist/websocket.js

@@ -1,3 +1,4 @@
+import { performanceMessage } from "./performance-mode.js";
 import { createHash, randomUUID } from "node:crypto";
 import { execFile, execFileSync } from "node:child_process";
 import { existsSync } from "node:fs";
@@ -104,6 +105,7 @@ const CODEX_USER_TURN_UUID_RE = /^codex:user-turn:(\d+)$/;
 const OPT_IN_SERVER_MESSAGES = new Set([
     "conversation_queue",
     "goal_state",
+    "codex_recovery_state",
     "guardian_approval",
     "prompt_history_status",
     "projects",
@@ -571,6 +573,16 @@ export class BridgeWebSocketServer {
     deltaBatchMaxChars;
     deltaBatches = new Map();
     platform;
+    deliveryPreferences = new WeakMap();
+    activityTimes = new WeakMap();
+    performanceEnabled(ws, msg) {
+        const prefs = this.deliveryPreferences.get(ws);
+        if (!prefs || typeof msg.sessionId !== "string")
+            return false;
+        return Object.hasOwn(prefs.sessions, msg.sessionId)
+            ? prefs.sessions[msg.sessionId]
+            : prefs.enabled;
+    }
     clientSupportedServerMessages = new WeakMap();
     clientProtocolVersions = new WeakMap();
     rejectedProtocolClients = new WeakSet();
@@ -618,7 +630,15 @@ export class BridgeWebSocketServer {
         else {
             console.log("[ws] Push relay enabled (Firebase Anonymous Auth)");
         }
-        this.wss = new WebSocketServer({ server });
+        this.wss = new WebSocketServer({
+            server,
+            perMessageDeflate: {
+                serverNoContextTakeover: true,
+                clientNoContextTakeover: true,
+                threshold: 1024,
+                zlibDeflateOptions: { level: 3 },
+            },
+        });
         this.sessionManager = new SessionManager((sessionId, msg) => {
             this.broadcastSessionMessage(sessionId, msg);
         }, imageStore, galleryStore, 
@@ -1954,6 +1974,13 @@ export class BridgeWebSocketServer {
             }
             this.clientProtocolVersions.set(ws, selectedProtocolVersion);
             this.clientSupportedServerMessages.set(ws, new Set(msg.supportedServerMessages ?? []));
+            this.deliveryPreferences.set(ws, {
+                enabled: msg.performanceMode ?? false,
+                sessions: msg.sessionPerformanceModes ?? {},
+            });
+            if (msg.deliveryRevision !== undefined) {
+                this.send(ws, { type: "performance_mode_state", deliveryRevision: msg.deliveryRevision });
+            }
             this.sendPromptHistoryStatus(ws);
             return;
         }
@@ -3104,6 +3131,21 @@ export class BridgeWebSocketServer {
                 console.log(`[ws] set_codex_speed(codex): serviceTier=${serviceTier}`);
                 break;
             }
+            case "set_codex_recovery":
+            case "cancel_codex_recovery": {
+                const session = this.resolveSession(msg.sessionId);
+                if (!session || session.provider !== "codex") {
+                    this.send(ws, { type: "error", sessionId: msg.sessionId, errorCode: "codex_recovery_unsupported", message: "Automatic recovery requires an active Codex session." });
+                    break;
+                }
+                const process = session.process;
+                if (msg.type === "set_codex_recovery")
+                    process.setRecoveryEnabled(msg.enabled);
+                else
+                    process.cancelRecovery();
+                this.send(ws, { type: "codex_recovery_state", sessionId: session.id, recovery: process.getRecoveryState() });
+                break;
+            }
             case "get_goal": {
                 const session = this.resolveSession(msg.sessionId);
                 if (!session || session.provider !== "codex") {
@@ -6549,6 +6591,7 @@ export class BridgeWebSocketServer {
             protocolVersion: BRIDGE_PROTOCOL_MAX_VERSION,
             minimumProtocolVersion: BRIDGE_PROTOCOL_MIN_VERSION,
             protocolCapabilities: [
+                "performance_mode_v1",
                 "project_request_correlation_v1",
                 "session_context_v1",
             ],
@@ -6587,6 +6630,7 @@ export class BridgeWebSocketServer {
             protocolVersion: BRIDGE_PROTOCOL_MAX_VERSION,
             minimumProtocolVersion: BRIDGE_PROTOCOL_MIN_VERSION,
             protocolCapabilities: [
+                "performance_mode_v1",
                 "project_request_correlation_v1",
                 "session_context_v1",
             ],
@@ -6620,12 +6664,17 @@ export class BridgeWebSocketServer {
     broadcastSessionMessage(sessionId, msg, exclude) {
         if (this.shouldBatchDelta(msg, exclude)) {
             this.trackSessionMessage(sessionId, msg);
-            const chunks = this.splitDeltaText(msg.text);
+            let chunks;
             for (const client of this.wss.clients) {
                 if (client.readyState !== WebSocket.OPEN)
                     continue;
                 if (!this.shouldSendToClient(client, msg))
                     continue;
+                if (msg.type === "thinking_delta" && this.performanceEnabled(client, { sessionId })) {
+                    this.send(client, { ...msg, sessionId });
+                    continue;
+                }
+                chunks ??= this.splitDeltaText(msg.text);
                 this.queueDeltaForClient(client, sessionId, msg.type, chunks);
             }
             return;
@@ -6717,7 +6766,7 @@ export class BridgeWebSocketServer {
         if (client.readyState !== WebSocket.OPEN)
             return;
         for (const msg of batch.messages) {
-            client.send(JSON.stringify({ ...msg, sessionId }));
+            this.send(client, { ...msg, sessionId });
         }
     }
     discardClientDeltaBatches(client) {
@@ -6858,9 +6907,9 @@ export class BridgeWebSocketServer {
         }
     }
     async listWorkspaceFilteredCodexThreads(msg, matchesWorkspace, limit) {
-        const activeProcess = this.getActiveCodexProcess();
-        const process = activeProcess ?? (await this.createStandaloneCodexProcess(undefined));
-        const isStandalone = activeProcess === null;
+        // A busy app-server can be blocked on a large thread/read or a running turn.
+        // Discovery must remain independent so one session cannot hide the list.
+        const process = await this.createStandaloneCodexProcess(undefined);
         try {
             const archivedIds = this.archiveStore.archivedIds();
             const matchingThreads = [];
@@ -6892,12 +6941,19 @@ export class BridgeWebSocketServer {
                 }
                 cursor = result.nextCursor;
             } while (matchingThreads.length < limit && cursor != null);
-            const indexedById = await getCodexSessionIndexMetadata(matchingThreads.map((thread) => thread.id));
-            return matchingThreads.map((thread) => this.enrichRecentSessionWorkspace(codexThreadToRecentSession(thread, indexedById.get(thread.id))));
+            const [indexedById, threadNames] = await Promise.all([
+                getCodexSessionIndexMetadata(matchingThreads.map((thread) => thread.id)),
+                loadCodexSessionNames(),
+            ]);
+            return matchingThreads.map((thread) => this.enrichRecentSessionWorkspace({
+                ...codexThreadToRecentSession(thread, indexedById.get(thread.id)),
+                ...(threadNames.get(thread.id)
+                    ? { name: threadNames.get(thread.id) }
+                    : {}),
+            }));
         }
         finally {
-            if (isStandalone)
-                process.stop();
+            process.stop();
         }
     }
     async listRecentSessionsRaw(msg) {
@@ -7319,9 +7375,7 @@ export class BridgeWebSocketServer {
     async listRecentCodexThreads(msg) {
         const limit = msg.limit ?? 20;
         const offset = msg.offset ?? 0;
-        const process = this.getActiveCodexProcess() ??
-            (await this.createStandaloneCodexProcess(msg.projectPath));
-        const isStandalone = process !== this.getActiveCodexProcess();
+        const process = await this.createStandaloneCodexProcess(msg.projectPath);
         try {
             const archivedIds = this.archiveStore.archivedIds();
             const visibleThreads = [];
@@ -7349,17 +7403,23 @@ export class BridgeWebSocketServer {
                 hasServerMore = cursor != null;
             } while (visibleThreads.length < targetCount && cursor != null);
             const pageThreads = visibleThreads.slice(offset, offset + limit);
-            const indexedById = await getCodexSessionIndexMetadata(pageThreads.map((thread) => thread.id));
-            const sessions = pageThreads.map((thread) => codexThreadToRecentSession(thread, indexedById.get(thread.id)));
+            const [indexedById, threadNames] = await Promise.all([
+                getCodexSessionIndexMetadata(pageThreads.map((thread) => thread.id)),
+                loadCodexSessionNames(),
+            ]);
+            const sessions = pageThreads.map((thread) => ({
+                ...codexThreadToRecentSession(thread, indexedById.get(thread.id)),
+                ...(threadNames.get(thread.id)
+                    ? { name: threadNames.get(thread.id) }
+                    : {}),
+            }));
             return {
                 sessions,
                 hasMore: hasServerMore || visibleThreads.length > offset + limit,
             };
         }
         finally {
-            if (isStandalone) {
-                process.stop();
-            }
+            process.stop();
         }
     }
     async createStandaloneCodexProcess(projectPath) {
@@ -7623,6 +7683,28 @@ export class BridgeWebSocketServer {
     prepareServerMessageForClient(ws, msg) {
         if (!this.shouldSendToClient(ws, msg))
             return null;
+        if (this.performanceEnabled(ws, msg)) {
+            const projected = performanceMessage(msg);
+            if (projected) {
+                msg = projected;
+            }
+            else {
+                const sessionId = msg.sessionId;
+                const times = this.activityTimes.get(ws) ?? new Map();
+                this.activityTimes.set(ws, times);
+                const now = Date.now();
+                if (now - (times.get(sessionId) ?? 0) < 1000)
+                    return null;
+                times.set(sessionId, now);
+                const historySeq = msg.historySeq;
+                return {
+                    type: "session_activity",
+                    sessionId,
+                    historySeq: typeof historySeq === "number" ? historySeq : undefined,
+                    at: new Date(now).toISOString(),
+                };
+            }
+        }
         if (!("messages" in msg) || !Array.isArray(msg.messages))
             return msg;
         const messages = msg.messages;
@@ -7699,6 +7781,9 @@ export class BridgeWebSocketServer {
         });
     }
     sendCodexGoalState(ws, sessionId, session) {
+        const recovery = session.process.getRecoveryState?.();
+        if (recovery)
+            this.send(ws, { type: "codex_recovery_state", sessionId, recovery });
         if (session.codexGoal === undefined)
             return;
         this.send(ws, {