@ccpocket/bridge 1.48.0 → 1.49.0

package/dist/websocket.js

@@ -38,6 +38,10 @@ const CODEX_MODELS = [
     "gpt-5.3-codex",
     "gpt-5.3-codex-spark",
 ];
+const OPT_IN_SERVER_MESSAGES = new Set([
+    "conversation_queue",
+    "prompt_history_status",
+]);
 // ---- Codex mode mapping helpers ----
 /** Map unified PermissionMode to Codex approval_policy.
  *  Only "bypassPermissions" maps to "never"; all others use "on-request". */
@@ -166,6 +170,7 @@ export class BridgeWebSocketServer {
     worktreeStore;
     pushRelay;
     promptHistoryBackup;
+    promptHistoryStore;
     recentSessionsRequestId = 0;
     debugEvents = new Map();
     notifiedPermissionToolUses = new Map();
@@ -181,7 +186,7 @@ export class BridgeWebSocketServer {
     platform;
     clientSupportedServerMessages = new WeakMap();
     constructor(options) {
-        const { server, apiKey, allowedDirs, imageStore, galleryStore, projectHistory, debugTraceStore, recordingStore, firebaseAuth, promptHistoryBackup, platform, } = options;
+        const { server, apiKey, allowedDirs, imageStore, galleryStore, projectHistory, debugTraceStore, recordingStore, firebaseAuth, promptHistoryBackup, promptHistoryStore, platform, } = options;
         this.apiKey = apiKey ?? null;
         this.allowedDirs = allowedDirs ?? [];
         this.imageStore = imageStore ?? null;
@@ -192,6 +197,7 @@ export class BridgeWebSocketServer {
         this.worktreeStore = new WorktreeStore();
         this.pushRelay = new PushRelayClient({ firebaseAuth });
         this.promptHistoryBackup = promptHistoryBackup ?? null;
+        this.promptHistoryStore = promptHistoryStore ?? null;
         this.platform = platform ?? process.platform;
         this.archiveStore = new ArchiveStore();
         void this.debugTraceStore.init().catch((err) => {
@@ -544,6 +550,7 @@ export class BridgeWebSocketServer {
     async handleClientMessage(msg, ws) {
         if (msg.type === "client_capabilities") {
             this.clientSupportedServerMessages.set(ws, new Set(msg.supportedServerMessages ?? []));
+            this.sendPromptHistoryStatus(ws);
             return;
         }
         const incomingSessionId = this.extractSessionIdFromClientMessage(msg);
@@ -3205,6 +3212,144 @@ export class BridgeWebSocketServer {
                 });
                 break;
             }
+            case "record_prompt_history": {
+                if (!this.promptHistoryStore) {
+                    this.send(ws, {
+                        type: "prompt_history_mutation_result",
+                        success: false,
+                        error: "Prompt history store not available",
+                    });
+                    break;
+                }
+                try {
+                    const entry = await this.promptHistoryStore.record({
+                        text: msg.text,
+                        projectPath: msg.projectPath,
+                        clientId: msg.clientId,
+                        clientName: msg.clientName,
+                        sessionId: msg.sessionId,
+                        usedAt: msg.usedAt,
+                    });
+                    this.send(ws, {
+                        type: "prompt_history_mutation_result",
+                        success: true,
+                        bridgeInstanceId: this.promptHistoryStore.bridgeInstanceId,
+                        revision: this.promptHistoryStore.revision,
+                        entry,
+                    });
+                    this.broadcastPromptHistoryStatus();
+                }
+                catch (err) {
+                    this.send(ws, {
+                        type: "prompt_history_mutation_result",
+                        success: false,
+                        error: err instanceof Error ? err.message : String(err),
+                    });
+                }
+                break;
+            }
+            case "sync_prompt_history": {
+                if (!this.promptHistoryStore) {
+                    this.send(ws, {
+                        type: "prompt_history_sync_result",
+                        success: false,
+                        error: "Prompt history store not available",
+                    });
+                    break;
+                }
+                try {
+                    if (msg.entries?.length) {
+                        await this.promptHistoryStore.mergeClientEntries(msg.entries);
+                    }
+                    this.send(ws, {
+                        type: "prompt_history_sync_result",
+                        success: true,
+                        bridgeInstanceId: this.promptHistoryStore.bridgeInstanceId,
+                        revision: this.promptHistoryStore.revision,
+                        syncedAt: new Date().toISOString(),
+                        fullSnapshot: true,
+                        entries: this.promptHistoryStore.list(msg.includeDeleted ?? true),
+                    });
+                    this.broadcastPromptHistoryStatus();
+                }
+                catch (err) {
+                    this.send(ws, {
+                        type: "prompt_history_sync_result",
+                        success: false,
+                        error: err instanceof Error ? err.message : String(err),
+                    });
+                }
+                break;
+            }
+            case "mutate_prompt_history": {
+                if (!this.promptHistoryStore) {
+                    this.send(ws, {
+                        type: "prompt_history_mutation_result",
+                        success: false,
+                        error: "Prompt history store not available",
+                    });
+                    break;
+                }
+                try {
+                    const entry = await this.promptHistoryStore.mutate({
+                        id: msg.id,
+                        text: msg.text,
+                        projectPath: msg.projectPath,
+                        action: msg.action,
+                        isFavorite: msg.isFavorite,
+                        updatedAt: msg.updatedAt,
+                    });
+                    this.send(ws, {
+                        type: "prompt_history_mutation_result",
+                        success: entry != null,
+                        bridgeInstanceId: this.promptHistoryStore.bridgeInstanceId,
+                        revision: this.promptHistoryStore.revision,
+                        entry: entry ?? undefined,
+                        error: entry == null ? "Prompt not found" : undefined,
+                    });
+                    if (entry)
+                        this.broadcastPromptHistoryStatus();
+                }
+                catch (err) {
+                    this.send(ws, {
+                        type: "prompt_history_mutation_result",
+                        success: false,
+                        error: err instanceof Error ? err.message : String(err),
+                    });
+                }
+                break;
+            }
+            case "import_prompt_history_v1": {
+                if (!this.promptHistoryStore) {
+                    this.send(ws, {
+                        type: "prompt_history_sync_result",
+                        success: false,
+                        error: "Prompt history store not available",
+                    });
+                    break;
+                }
+                try {
+                    const result = await this.promptHistoryStore.importEntries(msg.entries, msg.clientId, msg.clientName);
+                    this.send(ws, {
+                        type: "prompt_history_sync_result",
+                        success: true,
+                        bridgeInstanceId: this.promptHistoryStore.bridgeInstanceId,
+                        revision: this.promptHistoryStore.revision,
+                        syncedAt: new Date().toISOString(),
+                        fullSnapshot: true,
+                        entries: result.entries,
+                    });
+                    this.broadcastPromptHistoryStatus();
+                }
+                catch (err) {
+                    this.send(ws, {
+                        type: "prompt_history_sync_result",
+                        success: false,
+                        error: err instanceof Error ? err.message : String(err),
+                    });
+                }
+                break;
+            }
             case "rename_session": {
                 const name = msg.name || null;
                 await this.handleRenameSession(ws, msg.sessionId, name, msg);
@@ -3308,6 +3453,19 @@ export class BridgeWebSocketServer {
             bridgeVersion: getPackageVersion(),
         });
     }
+    sendPromptHistoryStatus(ws) {
+        if (!this.promptHistoryStore)
+            return;
+        const entries = this.promptHistoryStore.list(true);
+        const updatedAt = entries.reduce((latest, entry) => !latest || entry.updatedAt > latest ? entry.updatedAt : latest, undefined);
+        this.send(ws, {
+            type: "prompt_history_status",
+            bridgeInstanceId: this.promptHistoryStore.bridgeInstanceId,
+            revision: this.promptHistoryStore.revision,
+            entryCount: entries.filter((entry) => !entry.deletedAt).length,
+            updatedAt,
+        });
+    }
     /** Broadcast session list to all connected clients. */
     broadcastSessionList() {
         this.pruneDebugEvents();
@@ -3323,6 +3481,19 @@ export class BridgeWebSocketServer {
             bridgeVersion: getPackageVersion(),
         });
     }
+    broadcastPromptHistoryStatus() {
+        if (!this.promptHistoryStore)
+            return;
+        const entries = this.promptHistoryStore.list(true);
+        const updatedAt = entries.reduce((latest, entry) => !latest || entry.updatedAt > latest ? entry.updatedAt : latest, undefined);
+        this.broadcast({
+            type: "prompt_history_status",
+            bridgeInstanceId: this.promptHistoryStore.bridgeInstanceId,
+            revision: this.promptHistoryStore.revision,
+            entryCount: entries.filter((entry) => !entry.deletedAt).length,
+            updatedAt,
+        });
+    }
     broadcastSessionMessage(sessionId, msg) {
         this.maybeSendPushNotification(sessionId, msg);
         this.recordDebugEvent(sessionId, {
@@ -3681,11 +3852,10 @@ export class BridgeWebSocketServer {
         }
     }
     shouldSendToClient(ws, msg) {
-        if (msg.type !== "conversation_queue")
+        const type = typeof msg.type === "string" ? msg.type : "";
+        if (!OPT_IN_SERVER_MESSAGES.has(type))
             return true;
-        return (this.clientSupportedServerMessages
-            .get(ws)
-            ?.has("conversation_queue") ?? false);
+        return (this.clientSupportedServerMessages.get(ws)?.has(type) ?? false);
     }
     hasInputConflictSince(sessionId, baseSeq) {
         const delta = this.sessionManager.getHistorySince(sessionId, baseSeq);