@jingyi0605/codingns 0.3.0 → 0.3.5

package/dist/server/modules/sessions/session-history-service.js

@@ -1090,39 +1090,52 @@ export class SessionHistoryService {
             providerSessionId: snapshot.providerSessionId,
             rawStoreRef: snapshot.rawStoreRef
         });
-        const currentBinding = this.sessionBindingRepository.findBySessionId(sessionId);
-        const timestamp = nowIso();
-        const duplicateBinding = this.findSameWorkspaceBindingDuplicate(sessionId, workspaceId, resolvedSnapshot);
-        this.db.transaction(() => {
-            if (duplicateBinding) {
-                // 运行时链路显式指定了当前 sessionId，就应该由当前会话接管同工作区里的重复底层会话。
-                // 否则后续事件重放或后台发现补录都会持续撞 UNIQUE(provider, provider_session_id)。
-                this.mergeSessionIntoTarget({
-                    workspaceId,
-                    targetSessionId: sessionId,
-                    sourceSessionId: duplicateBinding.sessionId,
-                    provider: resolvedSnapshot.provider,
-                    timestamp
-                });
+        // discovery 和 runtime 回填会并发命中这里；如果在事务外先看重复，再事务内写入，
+        // 中间就会留下一个竞态窗口，最后直接撞 UNIQUE(provider, provider_session_id)。
+        for (let attempt = 0; attempt < 2; attempt += 1) {
+            try {
+                this.db.transaction(() => {
+                    const currentBinding = this.sessionBindingRepository.findBySessionId(sessionId);
+                    const timestamp = nowIso();
+                    const duplicateBinding = this.findSameWorkspaceBindingDuplicate(sessionId, workspaceId, resolvedSnapshot);
+                    if (duplicateBinding) {
+                        // 运行时链路显式指定了当前 sessionId，就应该由当前会话接管同工作区里的重复底层会话。
+                        // 否则后续事件重放或后台发现补录都会持续撞 UNIQUE(provider, provider_session_id)。
+                        this.mergeSessionIntoTarget({
+                            workspaceId,
+                            targetSessionId: sessionId,
+                            sourceSessionId: duplicateBinding.sessionId,
+                            provider: resolvedSnapshot.provider,
+                            timestamp
+                        });
+                    }
+                    const currentIndex = this.sessionIndexRepository.findIndexRecordBySessionId(sessionId);
+                    this.sessionBindingRepository.upsert({
+                        sessionId,
+                        workspaceId,
+                        provider: resolvedSnapshot.provider,
+                        providerSessionId: resolvedSnapshot.providerSessionId,
+                        rawStoreRef: resolvedSnapshot.rawStoreRef,
+                        createdAt: pickEarlierIso(currentBinding?.createdAt ?? null, duplicateBinding?.createdAt ?? null)
+                            ?? timestamp,
+                        updatedAt: timestamp
+                    });
+                    if (currentIndex) {
+                        this.sessionIndexRepository.upsert({
+                            ...currentIndex,
+                            updatedAt: timestamp
+                        });
+                    }
+                })();
+                return;
             }
-            const currentIndex = this.sessionIndexRepository.findIndexRecordBySessionId(sessionId);
-            this.sessionBindingRepository.upsert({
-                sessionId,
-                workspaceId,
-                provider: resolvedSnapshot.provider,
-                providerSessionId: resolvedSnapshot.providerSessionId,
-                rawStoreRef: resolvedSnapshot.rawStoreRef,
-                createdAt: pickEarlierIso(currentBinding?.createdAt ?? null, duplicateBinding?.createdAt ?? null)
-                    ?? timestamp,
-                updatedAt: timestamp
-            });
-            if (currentIndex) {
-                this.sessionIndexRepository.upsert({
-                    ...currentIndex,
-                    updatedAt: timestamp
-                });
+            catch (error) {
+                if (attempt === 0 && isSessionBindingProviderUniqueConflict(error)) {
+                    continue;
+                }
+                throw error;
             }
-        })();
+        }
     }
     async runDiscoverWorkspaceSessions(workspaceId, userId, refreshStateMode = "inline") {
         const startedAt = Date.now();
@@ -1162,7 +1175,9 @@ export class SessionHistoryService {
             const claimedPendingSessionIds = new Set();
             const persistPass1Transaction = this.db.transaction((batch) => {
                 for (const session of batch) {
-                    const exactExisting = this.sessionBindingRepository.findByProviderSession(session.provider, session.providerSessionId) ?? this.sessionBindingRepository.findByRawStoreRef(session.provider, session.rawStoreRef);
+                    const exactExisting = this.sessionBindingRepository.findByProviderSession(session.provider, session.providerSessionId) ?? (shouldMatchSessionBindingByRawStoreRef(session.provider)
+                        ? this.sessionBindingRepository.findByRawStoreRef(session.provider, session.rawStoreRef)
+                        : null);
                     if (exactExisting && exactExisting.workspaceId !== workspaceId) {
                         continue;
                     }
@@ -1405,12 +1420,13 @@ export class SessionHistoryService {
             ? this.sessionSyncService.readRecentHistory(provider, providerSessionId, rawStoreRef, knownTotalMessageCount, limit)
             : this.sessionSyncService.readHistory(provider, providerSessionId, rawStoreRef, cursor, limit, direction);
         return historyTask
-            .then((page) => {
-            const messagesWithAttachments = this.sessionMessageAttachmentService.enrichMessages(sessionId, page.messages);
+            .then(async (page) => {
+            const sanitizedPage = await this.sanitizeForkHistoryPage(sessionId, page, cursor, direction);
+            const messagesWithAttachments = this.sessionMessageAttachmentService.enrichMessages(sessionId, sanitizedPage.messages);
             const messages = this.enrichMessagesWithOrigin(sessionId, messagesWithAttachments);
             this.persistSessionChangedFiles(sessionId, messages);
             return {
-                ...page,
+                ...sanitizedPage,
                 messages
             };
         })
@@ -1429,6 +1445,56 @@ export class SessionHistoryService {
     enrichMessagesWithOrigin(sessionId, messages) {
         return this.resolveMessageOrigins(sessionId, messages);
     }
+    async sanitizeForkHistoryPage(sessionId, page, cursor, direction) {
+        if (direction !== "forward" || cursor !== null || page.messages.length === 0) {
+            return page;
+        }
+        const forkRecord = this.sessionForkRepository.findBySessionId(sessionId);
+        if (!forkRecord
+            || forkRecord.forkSourceType !== "message"
+            || !forkRecord.forkSourceMessageId) {
+            return page;
+        }
+        const childSession = this.sessionIndexRepository.findIndexRecordBySessionId(sessionId);
+        const childCreatedAt = childSession?.createdAt?.trim() || null;
+        if (!childCreatedAt) {
+            return page;
+        }
+        const parentBinding = this.getBindingOrThrow(forkRecord.forkSourceSessionId);
+        const inheritedMessages = await this.readForkSourceMessages(forkRecord.forkSourceSessionId, parentBinding, "message", forkRecord.forkSourceMessageId, null);
+        const expectedInheritedCount = inheritedMessages.length;
+        if (expectedInheritedCount <= 0) {
+            return page;
+        }
+        const parentMessages = await this.readForkSourceMessages(forkRecord.forkSourceSessionId, parentBinding, "session", null, null);
+        let leakedInheritedCount = countCommonHistoryPrefixLength(page.messages.slice(expectedInheritedCount), parentMessages.slice(expectedInheritedCount));
+        if (leakedInheritedCount <= 0) {
+            for (let index = expectedInheritedCount; index < page.messages.length; index += 1) {
+                const message = page.messages[index];
+                if (!message || message.timestamp > childCreatedAt) {
+                    break;
+                }
+                leakedInheritedCount += 1;
+            }
+        }
+        if (forkRecord.inheritedPrefixMessageCount !== expectedInheritedCount) {
+            this.sessionForkRepository.upsert({
+                ...forkRecord,
+                inheritedPrefixMessageCount: expectedInheritedCount
+            });
+        }
+        if (leakedInheritedCount <= 0) {
+            return page;
+        }
+        return {
+            ...page,
+            messages: [
+                ...page.messages.slice(0, expectedInheritedCount),
+                ...page.messages.slice(expectedInheritedCount + leakedInheritedCount)
+            ],
+            total: Math.max(0, page.total - leakedInheritedCount)
+        };
+    }
     resolveMessageOrigins(sessionId, messages) {
         const originRepository = this.sessionMessageOriginRepository;
         if (!originRepository || messages.length === 0) {
@@ -1683,7 +1749,13 @@ export class SessionHistoryService {
         return workspace;
     }
     getSessionListItemOrThrow(sessionId, userId) {
-        const item = this.sessionIndexRepository.findBySessionId(sessionId, userId);
+        const canonicalSessionId = this.resolveCanonicalSessionId(sessionId, userId);
+        const item = this.findSessionListItem(canonicalSessionId, sessionId, userId)
+            ?? this.repairMissingSessionListItem(canonicalSessionId, userId)
+            ?? (canonicalSessionId === sessionId
+                ? null
+                : this.repairMissingSessionListItem(sessionId, userId))
+            ?? this.findSessionListItem(canonicalSessionId, sessionId, userId);
         if (!item) {
             throw new AppError({
                 statusCode: 500,
@@ -1697,6 +1769,75 @@ export class SessionHistoryService {
         }
         return this.sessionIndexRepository.findBySessionId(aliasTargetSessionId, userId) ?? item;
     }
+    findSessionListItem(canonicalSessionId, sessionId, userId) {
+        return (this.sessionIndexRepository.findBySessionId(canonicalSessionId, userId)
+            ?? (canonicalSessionId === sessionId
+                ? null
+                : this.sessionIndexRepository.findBySessionId(sessionId, userId)));
+    }
+    repairMissingSessionListItem(sessionId, userId) {
+        const binding = this.sessionBindingRepository.findBySessionId(sessionId);
+        if (!binding) {
+            return null;
+        }
+        const existingIndex = this.sessionIndexRepository.findIndexRecordBySessionId(sessionId);
+        const existingSnapshot = this.sessionStatusSnapshotRepository.findBySessionId(sessionId);
+        const existingState = this.sessionStateRepository.findBySessionAndUser(sessionId, userId);
+        const timestamp = nowIso();
+        const fallbackLastMessageAt = existingIndex?.lastMessageAt
+            ?? existingState?.lastEventAt
+            ?? existingSnapshot?.lastSyncAt
+            ?? null;
+        const fallbackCreatedAt = pickEarlierIso(binding.createdAt, existingIndex?.createdAt ?? null)
+            ?? timestamp;
+        this.db.transaction(() => {
+            this.sessionIndexRepository.upsert({
+                sessionId,
+                workspaceId: binding.workspaceId,
+                provider: binding.provider,
+                parentSessionId: existingIndex?.parentSessionId ?? this.sessionForkRepository.findBySessionId(sessionId)?.parentSessionId ?? null,
+                sessionKind: existingIndex?.sessionKind ?? "default",
+                annotationSourceMessageId: existingIndex?.annotationSourceMessageId ?? null,
+                annotationSourceText: existingIndex?.annotationSourceText ?? null,
+                isSubagent: existingIndex?.isSubagent ?? false,
+                subagentLabel: existingIndex?.subagentLabel ?? null,
+                title: existingIndex?.title?.trim()
+                    || buildRecoveredSessionTitle(binding.provider, binding.providerSessionId),
+                messageCount: existingIndex?.messageCount ?? 0,
+                isArchived: existingIndex?.isArchived ?? false,
+                lastMessageAt: fallbackLastMessageAt,
+                createdAt: fallbackCreatedAt,
+                updatedAt: timestamp
+            });
+            if (!existingSnapshot) {
+                this.sessionStatusSnapshotRepository.upsert({
+                    sessionId,
+                    syncStatus: "idle",
+                    syncCursor: null,
+                    lastSyncAt: fallbackLastMessageAt,
+                    lastErrorCode: null,
+                    lastErrorDetail: null,
+                    resumedAt: null,
+                    updatedAt: timestamp
+                });
+            }
+            if (!existingState) {
+                this.sessionStateRepository.upsert({
+                    sessionId,
+                    userId,
+                    runningState: inferRecoveredSessionRunningState(binding),
+                    activitySource: inferRecoveredSessionActivitySource(binding),
+                    favorite: false,
+                    lastEventAt: shouldRecoverSessionAsActive(binding) ? (binding.updatedAt || timestamp) : fallbackLastMessageAt,
+                    completedAt: null,
+                    lastSeenAt: null,
+                    updatedAt: timestamp
+                });
+            }
+        })();
+        console.warn(`[session-history] repaired missing session index for ${sessionId} (${binding.provider})`);
+        return this.sessionIndexRepository.findBySessionId(sessionId, userId);
+    }
     resolveCanonicalSessionId(sessionId, userId) {
         if (userId) {
             const item = this.sessionIndexRepository.findBySessionId(sessionId, userId);
@@ -1812,7 +1953,9 @@ export class SessionHistoryService {
         if (isPendingBindingValue(snapshot.providerSessionId)) {
             return null;
         }
-        const existing = this.sessionBindingRepository.findByProviderSession(snapshot.provider, snapshot.providerSessionId) ?? this.sessionBindingRepository.findByRawStoreRef(snapshot.provider, snapshot.rawStoreRef);
+        const existing = this.sessionBindingRepository.findByProviderSession(snapshot.provider, snapshot.providerSessionId) ?? (shouldMatchSessionBindingByRawStoreRef(snapshot.provider)
+            ? this.sessionBindingRepository.findByRawStoreRef(snapshot.provider, snapshot.rawStoreRef)
+            : null);
         if (!existing || existing.sessionId === sessionId) {
             return null;
         }
@@ -1825,11 +1968,22 @@ export class SessionHistoryService {
         if (input.targetSessionId === input.sourceSessionId) {
             return;
         }
-        const targetBinding = this.sessionBindingRepository.findBySessionId(input.targetSessionId);
         const sourceBinding = this.sessionBindingRepository.findBySessionId(input.sourceSessionId);
-        if (!targetBinding || !sourceBinding) {
+        if (!sourceBinding) {
             return;
         }
+        const targetBinding = this.sessionBindingRepository.findBySessionId(input.targetSessionId);
+        if (!targetBinding) {
+            this.sessionBindingRepository.upsert({
+                sessionId: input.targetSessionId,
+                workspaceId: input.workspaceId,
+                provider: input.provider,
+                providerSessionId: buildPendingBindingValue(input.provider, input.targetSessionId),
+                rawStoreRef: buildPendingBindingValue(input.provider, input.targetSessionId),
+                createdAt: sourceBinding.createdAt,
+                updatedAt: input.timestamp
+            });
+        }
         const targetIndex = this.sessionIndexRepository.findIndexRecordBySessionId(input.targetSessionId);
         const sourceIndex = this.sessionIndexRepository.findIndexRecordBySessionId(input.sourceSessionId);
         const targetSnapshot = this.sessionStatusSnapshotRepository.findBySessionId(input.targetSessionId);
@@ -2215,6 +2369,7 @@ function buildInspectionActivityObservation(sessionId, inspection, observedAt) {
         source: hasInspectionEvidence(inspection) ? "inferred_log" : "unknown",
         confidence: "weak",
         detail: inspection.errorDetail,
+        interruptSource: null,
         errorCode: inspection.errorCode,
         observedAt: inspection.completedAtCandidate ?? inspection.lastEventAt ?? observedAt
     };
@@ -2448,6 +2603,35 @@ function normalizeSessionBindingSnapshot(sessionId, snapshot) {
         rawStoreRef: buildPendingBindingValue("claude-code", sessionId)
     };
 }
+function countCommonHistoryPrefixLength(left, right) {
+    const maxLength = Math.min(left.length, right.length);
+    let count = 0;
+    for (; count < maxLength; count += 1) {
+        if (!areHistoryMessagesEquivalent(left[count], right[count])) {
+            break;
+        }
+    }
+    return count;
+}
+function areHistoryMessagesEquivalent(left, right) {
+    if (!left || !right) {
+        return false;
+    }
+    if (left.messageId && right.messageId) {
+        if (left.messageId === right.messageId) {
+            return true;
+        }
+    }
+    if (left.rawRef && right.rawRef) {
+        if (left.rawRef === right.rawRef) {
+            return true;
+        }
+    }
+    return left.role === right.role
+        && left.kind === right.kind
+        && left.content === right.content
+        && left.timestamp === right.timestamp;
+}
 function shouldSkipClaudePendingBinding(binding) {
     if (binding.provider !== "claude-code") {
         return false;
@@ -2460,9 +2644,24 @@ function shouldSkipClaudePendingBinding(binding) {
 function isPendingBindingValue(value) {
     return value.trim().toLowerCase().startsWith("pending://");
 }
+function isSessionBindingProviderUniqueConflict(error) {
+    if (!(error instanceof Error)) {
+        return false;
+    }
+    return error.message.includes("UNIQUE constraint failed: session_bindings.provider, session_bindings.provider_session_id");
+}
 function buildPendingBindingValue(provider, sessionId) {
     return `pending://${provider}/${sessionId}`;
 }
+function shouldRecoverSessionAsActive(binding) {
+    return isPendingBindingValue(binding.providerSessionId) || isPendingBindingValue(binding.rawStoreRef);
+}
+function inferRecoveredSessionRunningState(binding) {
+    return shouldRecoverSessionAsActive(binding) ? "starting" : "idle";
+}
+function inferRecoveredSessionActivitySource(binding) {
+    return shouldRecoverSessionAsActive(binding) ? "runtime" : "none";
+}
 function buildAliasBindingValue(provider, targetSessionId, sourceSessionId) {
     return `alias://${provider}/${targetSessionId}/${sourceSessionId}`;
 }
@@ -2746,6 +2945,9 @@ function isLegacyCodingNsRolloutSession(providerSessionId, rawStoreRef) {
 function shouldRemoveMissingSyntheticCodexSession(rawStoreRef) {
     return isSyntheticCodexRawStoreRef(rawStoreRef) && !existsSync(rawStoreRef);
 }
+function shouldMatchSessionBindingByRawStoreRef(provider) {
+    return provider !== "codex";
+}
 function resolveSessionListTitle(provider, existingTitle, fallbackContent, parentTitle = null) {
     const normalizedExistingTitle = existingTitle?.trim() ?? "";
     const normalizedParentTitle = parentTitle?.trim() ?? "";
@@ -2768,6 +2970,24 @@ function buildUserMessageTitle(content, fallbackTitle) {
     const title = content.trim().replace(/\s+/g, " ");
     return title.slice(0, 48) || fallbackTitle;
 }
+function buildRecoveredSessionTitle(provider, providerSessionId) {
+    if (isPendingBindingValue(providerSessionId)) {
+        return "新会话";
+    }
+    const normalizedProvider = provider.trim().toLowerCase();
+    const providerLabel = normalizedProvider === "claude-code"
+        ? "Claude"
+        : normalizedProvider === "codex"
+            ? "Codex"
+            : normalizedProvider === "gemini"
+                ? "Gemini"
+                : normalizedProvider === "kimi"
+                    ? "Kimi"
+                    : normalizedProvider === "opencode"
+                        ? "OpenCode"
+                        : provider;
+    return `${providerLabel} 会话 ${providerSessionId.slice(0, 8)}`;
+}
 function resolvePersistedSessionTitle(provider, discoveredTitle, existingTitle, parentTitle = null) {
     const nextTitle = discoveredTitle.trim();
     const currentTitle = existingTitle?.trim() ?? "";