@ccpocket/bridge 1.67.2 → 1.67.4

package/dist/websocket.js

@@ -999,13 +999,58 @@ export class BridgeWebSocketServer {
         }
         this.seedCodexCanonicalUserTurnUuidMap(session, history);
         let nextSeq = canonicalEntries.at(-1)?.seq ?? 0;
+        const canonicalAssistantContentsByUser = new Map();
+        let canonicalUserKey = null;
+        for (const entry of canonicalEntries) {
+            if (entry.message.type === "user_input") {
+                canonicalUserKey = entry.message.userMessageUuid
+                    ? `user:${entry.message.userMessageUuid}`
+                    : null;
+                continue;
+            }
+            if (entry.message.type !== "assistant" || !canonicalUserKey)
+                continue;
+            const contents = canonicalAssistantContentsByUser.get(canonicalUserKey);
+            const contentKey = this.historyValueKey(entry.message.message.content);
+            if (contents)
+                contents.push(contentKey);
+            else
+                canonicalAssistantContentsByUser.set(canonicalUserKey, [contentKey]);
+        }
+        let pendingCanonicalAssistantContents = [];
         const retainedMessages = [];
         const retainedEntries = [];
         for (const entry of liveEntries) {
             if (!this.shouldRetainCodexLiveHistoryMessage(entry.message))
                 continue;
             const keys = this.codexHistoryMessageIdentityKeys(entry.message);
-            if (keys.some((key) => canonicalKeys.has(key)))
+            const matchesCanonical = keys.some((key) => canonicalKeys.has(key));
+            if (entry.message.type === "user_input") {
+                const userKey = entry.message.userMessageUuid
+                    ? `user:${entry.message.userMessageUuid}`
+                    : null;
+                pendingCanonicalAssistantContents =
+                    matchesCanonical && userKey
+                        ? [...(canonicalAssistantContentsByUser.get(userKey) ?? [])]
+                        : [];
+            }
+            if (entry.message.type === "assistant") {
+                const contentKey = this.historyValueKey(entry.message.message.content);
+                const contentIndex = pendingCanonicalAssistantContents.indexOf(contentKey);
+                if (matchesCanonical) {
+                    if (contentIndex !== -1) {
+                        pendingCanonicalAssistantContents.splice(contentIndex, 1);
+                    }
+                    continue;
+                }
+                const matchesCanonicalTurnAssistant = contentIndex !== -1;
+                if (matchesCanonicalTurnAssistant) {
+                    pendingCanonicalAssistantContents.splice(contentIndex, 1);
+                }
+                if (matchesCanonicalTurnAssistant)
+                    continue;
+            }
+            if (matchesCanonical)
                 continue;
             const seq = ++nextSeq;
             entry.message.historySeq = seq;
@@ -1114,6 +1159,7 @@ export class BridgeWebSocketServer {
             });
             this.sendCodexCurrentSettings(ws, sessionId, session);
             this.sendCodexQueueState(ws, sessionId, session);
+            this.sendCodexGoalState(ws, sessionId, session);
             if (options.includeCachedCommands) {
                 this.sendCachedCommands(ws, sessionId, session);
             }
@@ -1144,6 +1190,7 @@ export class BridgeWebSocketServer {
                 sessionId,
             });
             this.sendCodexQueueState(ws, sessionId, session);
+            this.sendCodexGoalState(ws, sessionId, session);
             this.sendCachedCommands(ws, sessionId, session);
             return true;
         }
@@ -2991,6 +3038,7 @@ export class BridgeWebSocketServer {
                     });
                     if (session.provider === "codex") {
                         this.sendCodexQueueState(ws, msg.sessionId, session);
+                        this.sendCodexGoalState(ws, msg.sessionId, session);
                     }
                     this.sendCachedCommands(ws, msg.sessionId, session);
                 }
@@ -3028,6 +3076,7 @@ export class BridgeWebSocketServer {
                     });
                     this.sendCodexCurrentSettings(ws, msg.sessionId, session);
                     this.sendCodexQueueState(ws, msg.sessionId, session);
+                    this.sendCodexGoalState(ws, msg.sessionId, session);
                     break;
                 }
                 if (session) {
@@ -5653,6 +5702,15 @@ export class BridgeWebSocketServer {
                 : [],
         });
     }
+    sendCodexGoalState(ws, sessionId, session) {
+        if (session.codexGoal === undefined)
+            return;
+        this.send(ws, {
+            type: "goal_state",
+            sessionId,
+            goal: session.codexGoal,
+        });
+    }
     sendCachedCommands(ws, sessionId, session) {
         // Restore command metadata when the original init/supported_commands event
         // has fallen out of the bounded in-memory history.