Bohdan Triapitsyn
e892346c6b
Canonicalize session message/part materialization across load, prefetch, reconnect, and recovery paths so OpenChamber restores session snapshots through one consistent merge flow. Preserve live assistant streaming text when stale or delayed snapshots arrive, while still replacing optimistic user parts with confirmed server snapshots to avoid duplicated user messages. Narrow recovery triggers to explicit incomplete snapshot signals instead of broad session-event fallbacks, reducing unnecessary session refetches during active streaming. Keep turn windowing aligned with parented assistant replies and add regression coverage for materialization gaps, stale snapshot protection, optimistic user replacement, reconnect recovery, and turn grouping.
113 lines
4.0 KiB
TypeScript
113 lines
4.0 KiB
TypeScript
/**
|
|
* Sync refs — imperative access to sync state from non-React code.
|
|
*
|
|
* SyncProvider sets these refs on mount. Store actions (session-ui-store,
|
|
* session-actions) use them to read child-store domain data without hooks.
|
|
*/
|
|
|
|
import type { OpencodeClient } from "@opencode-ai/sdk/v2/client"
|
|
import type { ChildStoreManager } from "./child-store"
|
|
import { getSessionMaterializationStatus } from "./materialization"
|
|
import type { State } from "./types"
|
|
|
|
let _sdk: OpencodeClient | null = null
|
|
let _childStores: ChildStoreManager | null = null
|
|
let _directory: string = ""
|
|
let _registerSessionDirectory: ((sessionID: string, directory: string) => void) | null = null
|
|
|
|
export function setSyncRefs(
|
|
sdk: OpencodeClient,
|
|
childStores: ChildStoreManager,
|
|
directory: string,
|
|
registerSessionDirectory?: (sessionID: string, directory: string) => void,
|
|
) {
|
|
_sdk = sdk
|
|
_childStores = childStores
|
|
_directory = directory
|
|
if (registerSessionDirectory) {
|
|
_registerSessionDirectory = registerSessionDirectory
|
|
}
|
|
}
|
|
|
|
/** Pre-register a session→directory mapping in the routing index.
|
|
* Called from session-actions when creating sessions so SSE events
|
|
* arriving before session.created can be routed correctly. */
|
|
export function registerSessionDirectory(sessionID: string, directory: string) {
|
|
_registerSessionDirectory?.(sessionID, directory)
|
|
}
|
|
|
|
export function getSyncSDK(): OpencodeClient {
|
|
if (!_sdk) throw new Error("SDK not initialized — is SyncProvider mounted?")
|
|
return _sdk
|
|
}
|
|
|
|
export function getSyncChildStores(): ChildStoreManager {
|
|
if (!_childStores) throw new Error("ChildStoreManager not initialized — is SyncProvider mounted?")
|
|
return _childStores
|
|
}
|
|
|
|
export function getSyncDirectory(): string {
|
|
return _directory
|
|
}
|
|
|
|
/** Read current directory's child store state. Returns undefined if not bootstrapped. */
|
|
export function getDirectoryState(directory?: string): State | undefined {
|
|
const stores = _childStores
|
|
if (!stores) return undefined
|
|
const dir = directory || _directory
|
|
if (!dir) return undefined
|
|
return stores.getState(dir)
|
|
}
|
|
|
|
/** Read sessions from current directory's child store */
|
|
export function getSyncSessions(directory?: string) {
|
|
return getDirectoryState(directory)?.session ?? []
|
|
}
|
|
|
|
/** Read sessions across all initialized child stores */
|
|
export function getAllSyncSessions() {
|
|
const stores = _childStores
|
|
if (!stores) return []
|
|
|
|
const deduped = new Map<string, State["session"][number]>()
|
|
for (const store of stores.children.values()) {
|
|
for (const session of store.getState().session) {
|
|
if (!session?.id) continue
|
|
deduped.set(session.id, session)
|
|
}
|
|
}
|
|
return Array.from(deduped.values())
|
|
}
|
|
|
|
/** Read messages for a session from current directory's child store */
|
|
export function getSyncMessages(sessionId: string, directory?: string) {
|
|
return getDirectoryState(directory)?.message[sessionId] ?? []
|
|
}
|
|
|
|
/** Read renderability of a session snapshot from current directory's child store */
|
|
export function getSyncSessionMaterializationStatus(sessionId: string, directory?: string) {
|
|
const state = getDirectoryState(directory)
|
|
if (!state) return { hasMessages: false, renderable: false, missingPartMessageIDs: [] }
|
|
return getSessionMaterializationStatus(state, sessionId)
|
|
}
|
|
|
|
/** Read parts for a message from current directory's child store */
|
|
export function getSyncParts(messageId: string, directory?: string) {
|
|
return getDirectoryState(directory)?.part[messageId] ?? []
|
|
}
|
|
|
|
/** Read session status from current directory's child store */
|
|
export function getSyncSessionStatus(sessionId: string, directory?: string) {
|
|
return getDirectoryState(directory)?.session_status[sessionId]
|
|
}
|
|
|
|
/** Read permissions for a session from current directory's child store */
|
|
export function getSyncPermissions(sessionId: string, directory?: string) {
|
|
return getDirectoryState(directory)?.permission[sessionId] ?? []
|
|
}
|
|
|
|
/** Read questions for a session from current directory's child store */
|
|
export function getSyncQuestions(sessionId: string, directory?: string) {
|
|
return getDirectoryState(directory)?.question[sessionId] ?? []
|
|
}
|