@earendil-works/pi-session-backend-sqlite-node 0.84.0 → 0.84.2

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (67) hide show
  1. package/CHANGELOG.md +11 -1
  2. package/README.md +10 -3
  3. package/dist/index.d.ts.map +1 -1
  4. package/dist/index.js +10 -3
  5. package/dist/index.js.map +1 -1
  6. package/dist/sqlite/branch-cache.d.ts.map +1 -1
  7. package/dist/sqlite/branch-cache.js +12 -13
  8. package/dist/sqlite/branch-cache.js.map +1 -1
  9. package/dist/sqlite/index.d.ts +1 -0
  10. package/dist/sqlite/index.d.ts.map +1 -1
  11. package/dist/sqlite/index.js +1 -0
  12. package/dist/sqlite/index.js.map +1 -1
  13. package/dist/sqlite/migrations/001_initial.sql +9 -10
  14. package/dist/sqlite/migrations.d.ts.map +1 -1
  15. package/dist/sqlite/migrations.js +5 -4
  16. package/dist/sqlite/migrations.js.map +1 -1
  17. package/dist/sqlite/repo.d.ts.map +1 -1
  18. package/dist/sqlite/repo.js +60 -46
  19. package/dist/sqlite/repo.js.map +1 -1
  20. package/dist/sqlite/search-backend.d.ts +7 -2
  21. package/dist/sqlite/search-backend.d.ts.map +1 -1
  22. package/dist/sqlite/search-backend.js +60 -25
  23. package/dist/sqlite/search-backend.js.map +1 -1
  24. package/dist/sqlite/sql.d.ts +19 -0
  25. package/dist/sqlite/sql.d.ts.map +1 -0
  26. package/dist/sqlite/sql.js +58 -0
  27. package/dist/sqlite/sql.js.map +1 -0
  28. package/dist/sqlite/storage/branch-entries.d.ts +8 -2
  29. package/dist/sqlite/storage/branch-entries.d.ts.map +1 -1
  30. package/dist/sqlite/storage/branch-entries.js +87 -61
  31. package/dist/sqlite/storage/branch-entries.js.map +1 -1
  32. package/dist/sqlite/storage/branch-tips.d.ts.map +1 -1
  33. package/dist/sqlite/storage/branch-tips.js +8 -11
  34. package/dist/sqlite/storage/branch-tips.js.map +1 -1
  35. package/dist/sqlite/storage/entries.d.ts +7 -2
  36. package/dist/sqlite/storage/entries.d.ts.map +1 -1
  37. package/dist/sqlite/storage/entries.js +22 -23
  38. package/dist/sqlite/storage/entries.js.map +1 -1
  39. package/dist/sqlite/storage/facts.d.ts +1 -0
  40. package/dist/sqlite/storage/facts.d.ts.map +1 -1
  41. package/dist/sqlite/storage/facts.js +26 -29
  42. package/dist/sqlite/storage/facts.js.map +1 -1
  43. package/dist/sqlite/storage/lanes.d.ts +1 -0
  44. package/dist/sqlite/storage/lanes.d.ts.map +1 -1
  45. package/dist/sqlite/storage/lanes.js +41 -52
  46. package/dist/sqlite/storage/lanes.js.map +1 -1
  47. package/dist/sqlite/storage/records.d.ts +2 -2
  48. package/dist/sqlite/storage/records.d.ts.map +1 -1
  49. package/dist/sqlite/storage/records.js +27 -47
  50. package/dist/sqlite/storage/records.js.map +1 -1
  51. package/dist/sqlite/storage/session-sequences.d.ts.map +1 -1
  52. package/dist/sqlite/storage/session-sequences.js +5 -6
  53. package/dist/sqlite/storage/session-sequences.js.map +1 -1
  54. package/dist/sqlite/storage/session-stats.d.ts.map +1 -1
  55. package/dist/sqlite/storage/session-stats.js +14 -19
  56. package/dist/sqlite/storage/session-stats.js.map +1 -1
  57. package/dist/sqlite/storage/sessions.d.ts +2 -2
  58. package/dist/sqlite/storage/sessions.d.ts.map +1 -1
  59. package/dist/sqlite/storage/sessions.js +38 -41
  60. package/dist/sqlite/storage/sessions.js.map +1 -1
  61. package/dist/sqlite/storage/writer-leases.d.ts.map +1 -1
  62. package/dist/sqlite/storage/writer-leases.js +18 -17
  63. package/dist/sqlite/storage/writer-leases.js.map +1 -1
  64. package/dist/sqlite/types.d.ts +1 -0
  65. package/dist/sqlite/types.d.ts.map +1 -1
  66. package/dist/sqlite/types.js.map +1 -1
  67. package/package.json +3 -3
@@ -1 +1 @@
1
- {"version":3,"file":"repo.d.ts","sourceRoot":"","sources":["../../src/sqlite/repo.ts"],"names":[],"mappings":"AACA,OAAO,EAIN,KAAK,WAAW,EAQhB,OAAO,EAEP,KAAK,WAAW,IAAI,iBAAiB,EAGrC,MAAM,+BAA+B,CAAC;AAkEvC,OAAO,KAAK,EAEX,qBAAqB,EACrB,0BAA0B,EAC1B,wBAAwB,EACxB,qBAAqB,EACrB,0BAA0B,EAC1B,MAAM,YAAY,CAAC;AAEpB,MAAM,WAAW,wBAAwB;IACxC,oGAAoG;IACpG,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,4EAA4E;IAC5E,mBAAmB,CAAC,EAAE,MAAM,CAAC;CAC7B;AAED,MAAM,WAAW,8BAA8B;IAC9C,GAAG,EAAE,0BAA0B,CAAC;IAChC,MAAM,EAAE,qBAAqB,CAAC;IAC9B,YAAY,EAAE,MAAM,CAAC;IACrB,WAAW,CAAC,EAAE,wBAAwB,CAAC;CACvC;AAoiBD,qBAAa,uBACZ,YACC,iBAAiB,CAAC,qBAAqB,EAAE,0BAA0B,EAAE,wBAAwB,CAAC,EAC9F,eAAe;IAEhB,OAAO,CAAC,YAAY,CAAqB;IACzC,OAAO,CAAC,QAAQ,CAA6B;IAC7C,OAAO,CAAC,eAAe,CAAsC;IAC7D,OAAO,CAAC,QAAQ,CAAC,UAAU,CAA8B;IACzD,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAmC;IAClE,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAiC;IACzD,OAAO,CAAC,QAAQ,CAAC,YAAY,CAA6B;IAE1D,YAAY,OAAO,EAAE,8BAA8B,EAGlD;YAEa,yBAAyB;IAMvC,OAAO,CAAC,gBAAgB;IAaxB,OAAO,CAAC,YAAY;IAcd,MAAM,CAAC,OAAO,EAAE,0BAA0B,GAAG,OAAO,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC,CAuBzF;IAEK,IAAI,CAAC,QAAQ,EAAE,qBAAqB,GAAG,OAAO,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC,CAEnF;IAED,2FAA2F;IACrF,iBAAiB,CAAC,QAAQ,EAAE,qBAAqB,GAAG,OAAO,CAAC,IAAI,CAAC,CAWtE;IAED,yFAAyF;IACnF,IAAI,CAAC,OAAO,GAAE,wBAA6B,GAAG,OAAO,CAAC,qBAAqB,EAAE,CAAC,CAQnF;IAEK,MAAM,CAAC,QAAQ,EAAE,qBAAqB,GAAG,OAAO,CAAC,IAAI,CAAC,CAqB3D;IAEK,IAAI,CACT,MAAM,EAAE,qBAAqB,EAC7B,OAAO,EAAE,WAAW,GAAG,0BAA0B,GAC/C,OAAO,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC,CA6GzC;IAEK,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAM3B;IAEK,CAAC,MAAM,CAAC,YAAY,CAAC,IAAI,OAAO,CAAC,IAAI,CAAC,CAE3C;YAEa,eAAe;YAQf,WAAW;YAMX,YAAY;CAgB1B","sourcesContent":["import type { FileError, Result } from \"@earendil-works/pi-agent-core\";\nimport {\n\ttype BranchBounds,\n\ttype Entry,\n\ttype EntryQuery,\n\ttype ForkOptions,\n\ttype LaneRecord,\n\ttype LogItem,\n\ttype LogOptions,\n\ttype NewRecord,\n\ttype OperationStartedRecord,\n\ttype ProvisionedEntry,\n\ttype RecordQuery,\n\tSession,\n\tSessionError,\n\ttype SessionRepo as SessionRepository,\n\ttype SessionStats,\n\ttype SessionStorage,\n} from \"@earendil-works/pi-agent-core\";\nimport { uuidv7 } from \"@earendil-works/pi-ai\";\nimport { appendEntryToBranchCache, buildCachedBranch, deleteBranchCache, rebuildBranchCache } from \"./branch-cache.ts\";\nimport { applyMigrations } from \"./migrations.ts\";\nimport { type CachedBranchEntryRow, queryCachedBranchRows, readCachedBranch } from \"./storage/branch-entries.ts\";\nimport { readBranchTipIds } from \"./storage/branch-tips.ts\";\nimport {\n\tdeleteEntryRows,\n\ttype EntryRow,\n\tentryPayload,\n\tidExistsInEntries,\n\tinsertEntryRow,\n\treadEntryRow,\n\treadEntryRows,\n} from \"./storage/entries.ts\";\nimport { appendFact, deleteFactRows, readFactRows, readLatestFact, readLatestLabelFacts } from \"./storage/facts.ts\";\nimport {\n\tcreateInitialLane,\n\tdeleteLaneRows,\n\tfinishLaneOperation,\n\tcreateLane as insertLane,\n\treadLane,\n\treadLaneHead,\n\treadLaneMoveRows,\n\treadLanes,\n\tsetLaneLeaf,\n\tstartLaneOperation,\n\tmoveLane as updateLane,\n} from \"./storage/lanes.ts\";\nimport {\n\tappendRecordRow,\n\tdeleteRecordRows,\n\tidExistsInRecords,\n\treadOpenOperationRows,\n\treadRecordRows,\n} from \"./storage/records.ts\";\nimport {\n\tadvanceSequence,\n\tcreateSequence,\n\tdeleteSequence,\n\tgetNextSequence,\n\tsetNextSequence,\n} from \"./storage/session-sequences.ts\";\nimport {\n\taddUsageToStats,\n\tcreateStats,\n\tdeleteStats,\n\tincrementMessageCount,\n\treadStats,\n} from \"./storage/session-stats.ts\";\nimport {\n\tdecodeSessionMetadata,\n\tdeleteSessionRow,\n\tinsertSessionRow,\n\treadSessionRow,\n\treadSessionRows,\n\ttype SessionRow,\n\tsessionExists,\n} from \"./storage/sessions.ts\";\nimport {\n\tacquireWriterLease,\n\tdeleteWriterLease,\n\treleaseWriterLease,\n\trenewWriterLease,\n\ttype WriterLease,\n} from \"./storage/writer-leases.ts\";\nimport type {\n\tSqliteDatabase,\n\tSqliteDatabaseFactory,\n\tSqliteSessionCreateOptions,\n\tSqliteSessionListOptions,\n\tSqliteSessionMetadata,\n\tSqliteSessionRepositoryEnv,\n} from \"./types.ts\";\n\nexport interface SqliteWriterLeaseOptions {\n\t/** Time without a successful heartbeat before another writer may take over. Default: 30 seconds. */\n\tttlMs?: number;\n\t/** Idle heartbeat cadence. Default: 10 seconds. Must be less than ttlMs. */\n\theartbeatIntervalMs?: number;\n}\n\nexport interface SqliteSessionRepositoryOptions {\n\tenv: SqliteSessionRepositoryEnv;\n\tsqlite: SqliteDatabaseFactory;\n\tdatabasePath: string;\n\twriterLease?: SqliteWriterLeaseOptions;\n}\n\ninterface ResolvedWriterLeaseOptions {\n\tttlMs: number;\n\theartbeatIntervalMs: number;\n}\n\nfunction resolveWriterLeaseOptions(options: SqliteWriterLeaseOptions | undefined): ResolvedWriterLeaseOptions {\n\tconst ttlMs = options?.ttlMs ?? 30_000;\n\tconst heartbeatIntervalMs = options?.heartbeatIntervalMs ?? 10_000;\n\tif (!Number.isSafeInteger(ttlMs) || ttlMs <= 0) throw new RangeError(\"writerLease.ttlMs must be positive\");\n\tif (!Number.isSafeInteger(heartbeatIntervalMs) || heartbeatIntervalMs <= 0 || heartbeatIntervalMs >= ttlMs) {\n\t\tthrow new RangeError(\"writerLease.heartbeatIntervalMs must be positive and less than ttlMs\");\n\t}\n\treturn { ttlMs, heartbeatIntervalMs };\n}\n\nfunction activeWriterError(sessionId: string): SessionError {\n\treturn new SessionError(\"storage\", `SQLite session ${sessionId} already has an active writer`);\n}\n\nfunction lostWriterError(sessionId: string): SessionError {\n\treturn new SessionError(\"storage\", `SQLite session ${sessionId} writer lease was lost`);\n}\n\nfunction claimWriterLease(db: SqliteDatabase, sessionId: string, options: ResolvedWriterLeaseOptions): WriterLease {\n\tconst now = Date.now();\n\tconst lease = acquireWriterLease(db, sessionId, uuidv7(), now, now + options.ttlMs);\n\tif (!lease) throw activeWriterError(sessionId);\n\treturn lease;\n}\n\nclass SerialOperationQueue {\n\tprivate tail: Promise<void> = Promise.resolve();\n\n\tenqueue<T>(operation: () => Promise<T> | T): Promise<T> {\n\t\tconst result = this.tail.then(operation);\n\t\tthis.tail = result.then(\n\t\t\t() => undefined,\n\t\t\t() => undefined,\n\t\t);\n\t\treturn result;\n\t}\n\n\tasync drain(): Promise<void> {\n\t\tawait this.tail;\n\t}\n}\n\nfunction resultOrThrow<T>(result: Result<T, FileError>, message: string): T {\n\tif (!result.ok) {\n\t\tconst code = result.error.code === \"not_found\" ? \"not_found\" : \"storage\";\n\t\tthrow new SessionError(code, `${message}: ${result.error.message}`, result.error);\n\t}\n\treturn result.value;\n}\n\nfunction getParentPath(path: string): string {\n\tconst normalized = path.replace(/[\\\\/]+$/, \"\");\n\tconst lastSlash = Math.max(normalized.lastIndexOf(\"/\"), normalized.lastIndexOf(\"\\\\\"));\n\tif (lastSlash < 0) return \".\";\n\tif (lastSlash === 0) return normalized.slice(0, 1);\n\treturn normalized.slice(0, lastSlash);\n}\n\nfunction configureSqliteDatabase(db: SqliteDatabase): void {\n\tdb.exec(\"PRAGMA journal_mode=WAL\");\n\tdb.exec(\"PRAGMA synchronous=FULL\");\n\tdb.exec(\"PRAGMA busy_timeout=5000\");\n}\n\nfunction timestampToText(timestamp: number): string {\n\treturn new Date(timestamp).toISOString();\n}\n\nfunction timestampFromText(timestamp: string): number {\n\treturn Date.parse(timestamp);\n}\n\nfunction entryRowFromCached(row: CachedBranchEntryRow): EntryRow {\n\treturn { ...row, seq: row.entry_seq, type: row.type as Entry[\"type\"] };\n}\n\nfunction readObjectPayload(row: EntryRow): Record<string, unknown> {\n\tconst payload = JSON.parse(row.payload) as unknown;\n\tif (typeof payload !== \"object\" || payload === null || Array.isArray(payload)) {\n\t\tthrow new Error(\"Payload is not an object\");\n\t}\n\treturn payload as Record<string, unknown>;\n}\n\nfunction decodeEntry(row: EntryRow): Entry {\n\ttry {\n\t\tconst payload = readObjectPayload(row);\n\t\tconst timestamp = timestampFromText(row.timestamp);\n\t\tif (!Number.isFinite(timestamp)) throw new Error(`Invalid timestamp ${row.timestamp}`);\n\t\tconst base = { id: row.id, seq: row.seq, parentId: row.parent_id, timestamp };\n\t\tswitch (row.type) {\n\t\t\tcase \"message\":\n\t\t\t\tif (typeof payload.message !== \"object\" || payload.message === null) throw new Error(\"Missing message\");\n\t\t\t\treturn {\n\t\t\t\t\t...base,\n\t\t\t\t\ttype: \"message\",\n\t\t\t\t\tmessage: payload.message as Extract<Entry, { type: \"message\" }>[\"message\"],\n\t\t\t\t\t...(payload.terminate === true ? { terminate: true as const } : {}),\n\t\t\t\t};\n\t\t\tcase \"model_change\":\n\t\t\t\tif (typeof payload.provider !== \"string\" || typeof payload.modelId !== \"string\") {\n\t\t\t\t\tthrow new Error(\"Invalid model_change payload\");\n\t\t\t\t}\n\t\t\t\treturn { ...base, type: \"model_change\", provider: payload.provider, modelId: payload.modelId };\n\t\t\tcase \"thinking_level_change\":\n\t\t\t\tif (typeof payload.thinkingLevel !== \"string\") throw new Error(\"Invalid thinking_level_change payload\");\n\t\t\t\treturn { ...base, type: \"thinking_level_change\", thinkingLevel: payload.thinkingLevel };\n\t\t\tcase \"active_tools_change\":\n\t\t\t\tif (!Array.isArray(payload.activeToolNames)) throw new Error(\"Invalid active_tools_change payload\");\n\t\t\t\tif (payload.activeToolNames.some((value) => typeof value !== \"string\")) {\n\t\t\t\t\tthrow new Error(\"Invalid active_tools_change payload\");\n\t\t\t\t}\n\t\t\t\treturn { ...base, type: \"active_tools_change\", activeToolNames: payload.activeToolNames };\n\t\t\tcase \"compaction\":\n\t\t\t\tif (\n\t\t\t\t\ttypeof payload.summary !== \"string\" ||\n\t\t\t\t\t!Array.isArray(payload.retainedTail) ||\n\t\t\t\t\ttypeof payload.tokensBefore !== \"number\"\n\t\t\t\t) {\n\t\t\t\t\tthrow new Error(\"Invalid compaction payload\");\n\t\t\t\t}\n\t\t\t\treturn {\n\t\t\t\t\t...base,\n\t\t\t\t\ttype: \"compaction\",\n\t\t\t\t\tsummary: payload.summary,\n\t\t\t\t\tretainedTail: payload.retainedTail as Extract<Entry, { type: \"compaction\" }>[\"retainedTail\"],\n\t\t\t\t\ttokensBefore: payload.tokensBefore,\n\t\t\t\t\t...(Object.hasOwn(payload, \"details\") ? { details: payload.details } : {}),\n\t\t\t\t\t...(Object.hasOwn(payload, \"usage\")\n\t\t\t\t\t\t? { usage: payload.usage as Extract<Entry, { type: \"compaction\" }>[\"usage\"] }\n\t\t\t\t\t\t: {}),\n\t\t\t\t};\n\t\t\tcase \"branch_summary\":\n\t\t\t\tif (typeof payload.fromId !== \"string\" || typeof payload.summary !== \"string\") {\n\t\t\t\t\tthrow new Error(\"Invalid branch_summary payload\");\n\t\t\t\t}\n\t\t\t\treturn {\n\t\t\t\t\t...base,\n\t\t\t\t\ttype: \"branch_summary\",\n\t\t\t\t\tfromId: payload.fromId,\n\t\t\t\t\tsummary: payload.summary,\n\t\t\t\t\t...(Object.hasOwn(payload, \"details\") ? { details: payload.details } : {}),\n\t\t\t\t\t...(Object.hasOwn(payload, \"usage\")\n\t\t\t\t\t\t? { usage: payload.usage as Extract<Entry, { type: \"branch_summary\" }>[\"usage\"] }\n\t\t\t\t\t\t: {}),\n\t\t\t\t};\n\t\t\tcase \"custom\":\n\t\t\t\tif (typeof payload.customType !== \"string\") throw new Error(\"Invalid custom payload\");\n\t\t\t\treturn {\n\t\t\t\t\t...base,\n\t\t\t\t\ttype: \"custom\",\n\t\t\t\t\tcustomType: payload.customType,\n\t\t\t\t\t...(Object.hasOwn(payload, \"data\") ? { data: payload.data } : {}),\n\t\t\t\t};\n\t\t}\n\t} catch (error) {\n\t\tthrow new SessionError(\n\t\t\t\"invalid_entry\",\n\t\t\t`Invalid SQLite session entry ${row.id}: failed to decode entry ${row.id}`,\n\t\t\terror instanceof Error ? error : undefined,\n\t\t);\n\t}\n}\n\nfunction recordRunId(record: NewRecord): string | undefined {\n\treturn record.type === \"operation_started\" ? record.id : \"runId\" in record ? record.runId : undefined;\n}\n\nfunction recordOpKind(record: NewRecord): string | undefined {\n\treturn record.type === \"operation_started\" ? record.intent.kind : undefined;\n}\n\nfunction decodeRecord(row: { seq: number; timestamp: string; payload: string }): LaneRecord {\n\ttry {\n\t\tconst timestamp = timestampFromText(row.timestamp);\n\t\tif (!Number.isFinite(timestamp)) throw new Error(`Invalid timestamp ${row.timestamp}`);\n\t\treturn {\n\t\t\t...(JSON.parse(row.payload) as object),\n\t\t\tseq: row.seq,\n\t\t\ttimestamp,\n\t\t} as LaneRecord;\n\t} catch (error) {\n\t\tthrow new SessionError(\n\t\t\t\"storage\",\n\t\t\t`Invalid SQLite session record at sequence ${row.seq}: failed to decode payload`,\n\t\t\terror instanceof Error ? error : undefined,\n\t\t);\n\t}\n}\n\nfunction validateCachedBranchRows(rows: readonly CachedBranchEntryRow[], query: BranchBounds): void {\n\tif (rows.length === 0) return;\n\tconst path = [...rows].sort((left, right) => left.entry_seq - right.entry_seq);\n\tif (query.stopAtId === undefined && query.stopAtType === undefined && path[0]?.parent_id !== null) {\n\t\tthrow new SessionError(\"invalid_entry\", `Entry ${path[0]?.parent_id} not found`);\n\t}\n\tfor (let index = 1; index < path.length; index++) {\n\t\tconst previous = path[index - 1]!;\n\t\tconst current = path[index]!;\n\t\tif (current.parent_id !== previous.id) {\n\t\t\tthrow new SessionError(\"invalid_entry\", `Entry ${current.parent_id} not found`);\n\t\t}\n\t}\n}\n\nfunction matchesEntryQuery(entry: Entry, query: EntryQuery): boolean {\n\treturn (\n\t\t(query.type === undefined || entry.type === query.type) &&\n\t\t(query.customType === undefined || (entry.type === \"custom\" && entry.customType === query.customType)) &&\n\t\t(query.cursor === undefined ||\n\t\t\t(query.order === \"oldestFirst\" ? entry.seq > query.cursor.afterSeq : entry.seq < query.cursor.afterSeq))\n\t);\n}\n\nfunction assertUnusedId(db: SqliteDatabase, sessionId: string, id: string): void {\n\tif (idExistsInEntries(db, sessionId, id) || idExistsInRecords(db, sessionId, id)) {\n\t\tthrow new SessionError(\"already_exists\", `ID already exists: ${id}`);\n\t}\n}\n\nfunction requireSessionRow(db: SqliteDatabase, sessionId: string): SessionRow {\n\tconst row = readSessionRow(db, sessionId);\n\tif (!row) throw new SessionError(\"not_found\", `Session not found: ${sessionId}`);\n\treturn row;\n}\n\nclass SqliteSessionStorage implements SessionStorage<SqliteSessionMetadata> {\n\tprivate readonly db: SqliteDatabase;\n\tprivate readonly metadata: SqliteSessionMetadata;\n\tprivate readonly lease: WriterLease;\n\tprivate readonly leaseOptions: ResolvedWriterLeaseOptions;\n\tprivate readonly onRelease: () => void;\n\tprivate readonly operations = new SerialOperationQueue();\n\tprivate heartbeatTimer: ReturnType<typeof setTimeout> | undefined;\n\tprivate leaseError: SessionError | undefined;\n\tprivate closing = false;\n\tprivate releasePromise: Promise<void> | undefined;\n\n\tconstructor(\n\t\tdb: SqliteDatabase,\n\t\tmetadata: SqliteSessionMetadata,\n\t\tlease: WriterLease,\n\t\tleaseOptions: ResolvedWriterLeaseOptions,\n\t\tonRelease: () => void,\n\t) {\n\t\tthis.db = db;\n\t\tthis.metadata = metadata;\n\t\tthis.lease = lease;\n\t\tthis.leaseOptions = leaseOptions;\n\t\tthis.onRelease = onRelease;\n\t\tthis.scheduleHeartbeat();\n\t}\n\n\tasync release(): Promise<void> {\n\t\tthis.releasePromise ??= this.finishRelease();\n\t\tawait this.releasePromise;\n\t}\n\n\tprivate async finishRelease(): Promise<void> {\n\t\tthis.closing = true;\n\t\tif (this.heartbeatTimer !== undefined) clearTimeout(this.heartbeatTimer);\n\t\ttry {\n\t\t\tawait this.operations.enqueue(() =>\n\t\t\t\tthis.db.transaction(() => releaseWriterLease(this.db, this.metadata.id, this.lease)),\n\t\t\t);\n\t\t} finally {\n\t\t\tthis.onRelease();\n\t\t}\n\t}\n\n\tprivate enqueueWrite<T>(operation: () => T): Promise<T> {\n\t\tif (this.closing)\n\t\t\treturn Promise.reject(new SessionError(\"storage\", `SQLite session ${this.metadata.id} is closed`));\n\t\treturn this.operations.enqueue(() => {\n\t\t\tif (this.leaseError) throw this.leaseError;\n\t\t\treturn this.db.transaction(() => {\n\t\t\t\tconst now = Date.now();\n\t\t\t\tif (!renewWriterLease(this.db, this.metadata.id, this.lease, now, now + this.leaseOptions.ttlMs)) {\n\t\t\t\t\tthis.leaseError = lostWriterError(this.metadata.id);\n\t\t\t\t\tif (this.heartbeatTimer !== undefined) clearTimeout(this.heartbeatTimer);\n\t\t\t\t\tthrow this.leaseError;\n\t\t\t\t}\n\t\t\t\treturn operation();\n\t\t\t});\n\t\t});\n\t}\n\n\tprivate scheduleHeartbeat(): void {\n\t\tif (this.closing || this.leaseError) return;\n\t\tthis.heartbeatTimer = setTimeout(async () => {\n\t\t\tthis.heartbeatTimer = undefined;\n\t\t\ttry {\n\t\t\t\tawait this.operations.enqueue(() => {\n\t\t\t\t\tif (this.closing || this.leaseError) return;\n\t\t\t\t\tthis.db.transaction(() => {\n\t\t\t\t\t\tconst now = Date.now();\n\t\t\t\t\t\tif (!renewWriterLease(this.db, this.metadata.id, this.lease, now, now + this.leaseOptions.ttlMs)) {\n\t\t\t\t\t\t\tthis.leaseError = lostWriterError(this.metadata.id);\n\t\t\t\t\t\t}\n\t\t\t\t\t});\n\t\t\t\t});\n\t\t\t} catch {\n\t\t\t\t// A transient heartbeat failure is retried. Every write still verifies ownership transactionally.\n\t\t\t} finally {\n\t\t\t\tthis.scheduleHeartbeat();\n\t\t\t}\n\t\t}, this.leaseOptions.heartbeatIntervalMs);\n\t\tthis.heartbeatTimer.unref();\n\t}\n\n\tasync getMetadata(): Promise<SqliteSessionMetadata> {\n\t\treturn decodeSessionMetadata(requireSessionRow(this.db, this.metadata.id), this.metadata.path);\n\t}\n\n\tisForSession(sessionId: string): boolean {\n\t\treturn this.metadata.id === sessionId;\n\t}\n\n\tasync getLanes(): Promise<{ lane: string; leafId: string | null }[]> {\n\t\treturn readLanes(this.db, this.metadata.id).map((row) => ({ lane: row.lane, leafId: row.leaf_id }));\n\t}\n\n\tasync createLane(lane: string, at: string | null): Promise<void> {\n\t\treturn this.enqueueWrite(() => {\n\t\t\tif (readLane(this.db, this.metadata.id, lane)) {\n\t\t\t\tthrow new SessionError(\"already_exists\", `Lane already exists: ${lane}`);\n\t\t\t}\n\t\t\tif (at !== null && !readEntryRow(this.db, this.metadata.id, at)) {\n\t\t\t\tthrow new SessionError(\"not_found\", `Entry not found: ${at}`);\n\t\t\t}\n\t\t\tconst seq = getNextSequence(this.db, this.metadata.id);\n\t\t\tinsertLane(this.db, this.metadata.id, seq, lane, at);\n\t\t\tadvanceSequence(this.db, this.metadata.id, seq);\n\t\t});\n\t}\n\n\tasync moveLane(lane: string, to: string | null): Promise<void> {\n\t\treturn this.enqueueWrite(() => {\n\t\t\tif (!readLane(this.db, this.metadata.id, lane))\n\t\t\t\tthrow new SessionError(\"invalid_lane\", `Lane not found: ${lane}`);\n\t\t\tif (to !== null && !readEntryRow(this.db, this.metadata.id, to)) {\n\t\t\t\tthrow new SessionError(\"not_found\", `Entry not found: ${to}`);\n\t\t\t}\n\t\t\tconst seq = getNextSequence(this.db, this.metadata.id);\n\t\t\tupdateLane(this.db, this.metadata.id, seq, lane, to);\n\t\t\tadvanceSequence(this.db, this.metadata.id, seq);\n\t\t});\n\t}\n\n\tasync appendEntry<TEntry extends Entry>(entry: ProvisionedEntry<TEntry>, lane: string): Promise<TEntry> {\n\t\treturn this.enqueueWrite(() => {\n\t\t\tconst parentId = readLaneHead(this.db, this.metadata.id, lane).leafId;\n\t\t\tassertUnusedId(this.db, this.metadata.id, entry.id);\n\t\t\tconst seq = getNextSequence(this.db, this.metadata.id);\n\t\t\tconst committed = { ...entry, parentId, seq, timestamp: Date.now() } as Entry;\n\t\t\tinsertEntryRow(this.db, this.metadata.id, {\n\t\t\t\tseq,\n\t\t\t\tid: committed.id,\n\t\t\t\tparentId: committed.parentId,\n\t\t\t\ttype: committed.type,\n\t\t\t\ttimestamp: timestampToText(committed.timestamp),\n\t\t\t\tpayload: JSON.stringify(entryPayload(committed)),\n\t\t\t});\n\t\t\tsetLaneLeaf(this.db, this.metadata.id, lane, committed.id);\n\t\t\tappendEntryToBranchCache(\n\t\t\t\tthis.db,\n\t\t\t\tthis.metadata.id,\n\t\t\t\tcommitted.id,\n\t\t\t\tseq,\n\t\t\t\tcommitted.type,\n\t\t\t\tcommitted.type === \"custom\" ? committed.customType : null,\n\t\t\t\tcommitted.parentId,\n\t\t\t);\n\t\t\tif (committed.type === \"message\") incrementMessageCount(this.db, this.metadata.id);\n\t\t\tadvanceSequence(this.db, this.metadata.id, seq);\n\t\t\treturn structuredClone(committed as TEntry);\n\t\t});\n\t}\n\n\tasync appendRecord<TRecord extends LaneRecord>(record: NewRecord<TRecord>): Promise<TRecord>;\n\tasync appendRecord(record: NewRecord): Promise<LaneRecord> {\n\t\treturn this.enqueueWrite(() => {\n\t\t\tif (!readLane(this.db, this.metadata.id, record.lane)) {\n\t\t\t\tthrow new SessionError(\"invalid_lane\", `Lane not found: ${record.lane}`);\n\t\t\t}\n\t\t\tassertUnusedId(this.db, this.metadata.id, record.id);\n\t\t\tconst seq = getNextSequence(this.db, this.metadata.id);\n\t\t\tconst committed: LaneRecord = { ...record, seq, timestamp: Date.now() };\n\t\t\tif (record.type === \"operation_started\") {\n\t\t\t\tstartLaneOperation(this.db, this.metadata.id, record.lane, record.id);\n\t\t\t}\n\t\t\tappendRecordRow(this.db, this.metadata.id, {\n\t\t\t\tseq,\n\t\t\t\tid: record.id,\n\t\t\t\tlane: record.lane,\n\t\t\t\trunId: recordRunId(record),\n\t\t\t\ttype: record.type,\n\t\t\t\topKind: recordOpKind(record),\n\t\t\t\ttimestamp: timestampToText(committed.timestamp),\n\t\t\t\tpayload: JSON.stringify(record),\n\t\t\t});\n\t\t\tif (record.type === \"operation_finished\") {\n\t\t\t\tfinishLaneOperation(this.db, this.metadata.id, record.lane, record.runId);\n\t\t\t}\n\t\t\tif (record.type === \"usage\") addUsageToStats(this.db, this.metadata.id, record.usage);\n\t\t\tadvanceSequence(this.db, this.metadata.id, seq);\n\t\t\treturn structuredClone(committed);\n\t\t});\n\t}\n\n\tasync getEntry(id: string): Promise<Entry | undefined> {\n\t\tconst row = readEntryRow(this.db, this.metadata.id, id);\n\t\treturn row ? decodeEntry(row) : undefined;\n\t}\n\n\tasync findEntries(query: EntryQuery = {}): Promise<Entry[]> {\n\t\tconst rows = readEntryRows(this.db, this.metadata.id, { order: query.order });\n\t\tconst entries = rows.map(decodeEntry).filter((entry) => matchesEntryQuery(entry, query));\n\t\treturn query.limit === undefined ? entries : entries.slice(0, query.limit);\n\t}\n\n\tasync findEntriesOnBranch(query: EntryQuery & BranchBounds & { start: string }): Promise<Entry[]> {\n\t\tconst cached = readCachedBranch(this.db, this.metadata.id, query.start);\n\t\tif (!cached) {\n\t\t\tif (!readEntryRow(this.db, this.metadata.id, query.start))\n\t\t\t\tthrow new SessionError(\"not_found\", `Entry not found: ${query.start}`);\n\t\t\tthrow new SessionError(\"invalid_entry\", `Branch cache missing entry ${query.start}`);\n\t\t}\n\t\tconst rows = queryCachedBranchRows(this.db, this.metadata.id, cached, query);\n\t\tvalidateCachedBranchRows(rows, query);\n\t\tconst entries = rows\n\t\t\t.map(entryRowFromCached)\n\t\t\t.map(decodeEntry)\n\t\t\t.filter((entry) => matchesEntryQuery(entry, query));\n\t\treturn query.limit === undefined ? entries : entries.slice(0, query.limit);\n\t}\n\n\tasync findRecords(query: RecordQuery = {}): Promise<LaneRecord[]> {\n\t\tconst rows = readRecordRows(this.db, this.metadata.id, query);\n\t\treturn rows.map(decodeRecord);\n\t}\n\n\tasync findOpenOperations(lane: string, options?: { limit?: number }): Promise<OperationStartedRecord[]> {\n\t\tconst rows = readOpenOperationRows(this.db, this.metadata.id, lane, options);\n\n\t\treturn rows.map((row) => {\n\t\t\tconst record = decodeRecord(row);\n\t\t\tif (record.type !== \"operation_started\") {\n\t\t\t\tthrow new SessionError(\"storage\", \"Expected operation_started record\");\n\t\t\t}\n\t\t\treturn record;\n\t\t});\n\t}\n\n\tasync getLog(options: LogOptions = {}): Promise<LogItem[]> {\n\t\tconst afterSeq = options.afterSeq ?? 0;\n\t\tconst entryRows = readEntryRows(this.db, this.metadata.id, { afterSeq, order: \"oldestFirst\" });\n\t\tconst recordRows = readRecordRows(this.db, this.metadata.id, { afterSeq });\n\t\tconst laneRows = readLaneMoveRows(this.db, this.metadata.id, { afterSeq });\n\t\tconst factRows = readFactRows(this.db, this.metadata.id, { afterSeq });\n\n\t\tconst log: LogItem[] = [\n\t\t\t...entryRows.map((row) => ({ kind: \"entry\" as const, seq: row.seq, entry: decodeEntry(row) })),\n\t\t\t...recordRows.map((row) => ({ kind: \"record\" as const, seq: row.seq, record: decodeRecord(row) })),\n\t\t\t...laneRows.map((row) => ({ kind: \"lane\" as const, seq: row.seq, lane: row.lane, leafId: row.leaf_id })),\n\t\t\t...factRows.map((row) => {\n\t\t\t\tif (row.kind === \"name\")\n\t\t\t\t\treturn {\n\t\t\t\t\t\tkind: \"fact\" as const,\n\t\t\t\t\t\tseq: row.seq,\n\t\t\t\t\t\tfact: \"name\" as const,\n\t\t\t\t\t\tname: JSON.parse(row.value ?? \"null\") as string,\n\t\t\t\t\t};\n\t\t\t\treturn {\n\t\t\t\t\tkind: \"fact\" as const,\n\t\t\t\t\tseq: row.seq,\n\t\t\t\t\tfact: \"label\" as const,\n\t\t\t\t\ttargetId: row.key ?? \"\",\n\t\t\t\t\tlabel: row.value === null ? undefined : (JSON.parse(row.value) as string),\n\t\t\t\t};\n\t\t\t}),\n\t\t].sort((left, right) => left.seq - right.seq);\n\t\treturn options.limit === undefined ? log : log.slice(0, options.limit);\n\t}\n\n\tasync getName(): Promise<string | undefined> {\n\t\tconst row = readLatestFact(this.db, this.metadata.id, \"name\", null);\n\t\treturn row?.value === undefined || row.value === null ? undefined : (JSON.parse(row.value) as string);\n\t}\n\n\tasync setName(name: string): Promise<void> {\n\t\treturn this.enqueueWrite(() => {\n\t\t\tconst seq = getNextSequence(this.db, this.metadata.id);\n\t\t\tappendFact(this.db, this.metadata.id, seq, \"name\", null, JSON.stringify(name));\n\t\t\tadvanceSequence(this.db, this.metadata.id, seq);\n\t\t});\n\t}\n\n\tasync getLabel(id: string): Promise<string | undefined> {\n\t\tconst row = readLatestFact(this.db, this.metadata.id, \"label\", id);\n\t\treturn row?.value === undefined || row.value === null ? undefined : (JSON.parse(row.value) as string);\n\t}\n\n\tasync setLabel(id: string, label: string | undefined): Promise<void> {\n\t\treturn this.enqueueWrite(() => {\n\t\t\tif (!readEntryRow(this.db, this.metadata.id, id)) {\n\t\t\t\tthrow new SessionError(\"not_found\", `Entry not found: ${id}`);\n\t\t\t}\n\t\t\tconst seq = getNextSequence(this.db, this.metadata.id);\n\t\t\tappendFact(this.db, this.metadata.id, seq, \"label\", id, label === undefined ? null : JSON.stringify(label));\n\t\t\tadvanceSequence(this.db, this.metadata.id, seq);\n\t\t});\n\t}\n\n\tasync getStats(): Promise<SessionStats> {\n\t\treturn readStats(this.db, this.metadata.id);\n\t}\n}\n\nfunction claimStorage(\n\tdb: SqliteDatabase,\n\tmetadata: SqliteSessionMetadata,\n\tleaseOptions: ResolvedWriterLeaseOptions,\n\tonRelease: () => void,\n): SqliteSessionStorage {\n\trequireSessionRow(db, metadata.id);\n\tconst claimed = db.transaction(() => {\n\t\tconst lease = claimWriterLease(db, metadata.id, leaseOptions);\n\t\tconst row = requireSessionRow(db, metadata.id);\n\t\treadLanes(db, metadata.id);\n\t\treturn { lease, row };\n\t});\n\treturn new SqliteSessionStorage(\n\t\tdb,\n\t\tdecodeSessionMetadata(claimed.row, metadata.path),\n\t\tclaimed.lease,\n\t\tleaseOptions,\n\t\tonRelease,\n\t);\n}\n\nexport class SqliteSessionRepository\n\timplements\n\t\tSessionRepository<SqliteSessionMetadata, SqliteSessionCreateOptions, SqliteSessionListOptions>,\n\t\tAsyncDisposable\n{\n\tprivate databasePath: string | undefined;\n\tprivate database: SqliteDatabase | undefined;\n\tprivate databasePromise: Promise<SqliteDatabase> | undefined;\n\tprivate readonly operations = new SerialOperationQueue();\n\tprivate readonly activeStorages = new Set<SqliteSessionStorage>();\n\tprivate readonly options: SqliteSessionRepositoryOptions;\n\tprivate readonly leaseOptions: ResolvedWriterLeaseOptions;\n\n\tconstructor(options: SqliteSessionRepositoryOptions) {\n\t\tthis.options = options;\n\t\tthis.leaseOptions = resolveWriterLeaseOptions(options.writerLease);\n\t}\n\n\tprivate async releaseStoragesForSession(sessionId: string): Promise<void> {\n\t\tfor (const storage of [...this.activeStorages]) {\n\t\t\tif (storage.isForSession(sessionId)) await storage.release();\n\t\t}\n\t}\n\n\tprivate sessionFromLease(\n\t\tdb: SqliteDatabase,\n\t\tmetadata: SqliteSessionMetadata,\n\t\tlease: WriterLease,\n\t): Session<SqliteSessionMetadata> {\n\t\tlet storage: SqliteSessionStorage;\n\t\tstorage = new SqliteSessionStorage(db, metadata, lease, this.leaseOptions, () => {\n\t\t\tthis.activeStorages.delete(storage);\n\t\t});\n\t\tthis.activeStorages.add(storage);\n\t\treturn new Session(storage);\n\t}\n\n\tprivate claimSession(db: SqliteDatabase, metadata: SqliteSessionMetadata): Session<SqliteSessionMetadata> {\n\t\tconst active = [...this.activeStorages].find((storage) => storage.isForSession(metadata.id));\n\t\tif (active) {\n\t\t\treadLanes(db, metadata.id);\n\t\t\treturn new Session(active);\n\t\t}\n\t\tlet storage: SqliteSessionStorage;\n\t\tstorage = claimStorage(db, metadata, this.leaseOptions, () => {\n\t\t\tthis.activeStorages.delete(storage);\n\t\t});\n\t\tthis.activeStorages.add(storage);\n\t\treturn new Session(storage);\n\t}\n\n\tasync create(options: SqliteSessionCreateOptions): Promise<Session<SqliteSessionMetadata>> {\n\t\treturn this.operations.enqueue(async () => {\n\t\t\tconst db = await this.getDatabase();\n\t\t\tconst path = await this.getDatabasePath();\n\t\t\tconst id = options.id ?? uuidv7();\n\t\t\tif (sessionExists(db, id)) throw new SessionError(\"already_exists\", `Session already exists: ${id}`);\n\t\t\tconst createdAt = Date.now();\n\t\t\tconst lease = db.transaction(() => {\n\t\t\t\tinsertSessionRow(db, {\n\t\t\t\t\tid,\n\t\t\t\t\tcreatedAt: timestampToText(createdAt),\n\t\t\t\t\tcwd: options.cwd,\n\t\t\t\t\tparentSessionId: options.parentSessionId,\n\t\t\t\t\tmetadata: options.metadata,\n\t\t\t\t});\n\t\t\t\tcreateSequence(db, id);\n\t\t\t\tcreateStats(db, id);\n\t\t\t\tcreateInitialLane(db, id);\n\t\t\t\treturn claimWriterLease(db, id, this.leaseOptions);\n\t\t\t});\n\t\t\tconst row = requireSessionRow(db, id);\n\t\t\treturn this.sessionFromLease(db, decodeSessionMetadata(row, path), lease);\n\t\t});\n\t}\n\n\tasync open(metadata: SqliteSessionMetadata): Promise<Session<SqliteSessionMetadata>> {\n\t\treturn this.operations.enqueue(async () => this.claimSession(await this.getDatabase(), metadata));\n\t}\n\n\t/** Rebuilds this session's private branch-read cache from canonical entry parent links. */\n\tasync repairBranchCache(metadata: SqliteSessionMetadata): Promise<void> {\n\t\treturn this.operations.enqueue(async () => {\n\t\t\tawait this.releaseStoragesForSession(metadata.id);\n\t\t\tconst db = await this.getDatabase();\n\t\t\tdb.transaction(() => {\n\t\t\t\tconst lease = claimWriterLease(db, metadata.id, this.leaseOptions);\n\t\t\t\trequireSessionRow(db, metadata.id);\n\t\t\t\trebuildBranchCache(db, metadata.id);\n\t\t\t\treleaseWriterLease(db, metadata.id, lease);\n\t\t\t});\n\t\t});\n\t}\n\n\t/** Reads the session catalog without acquiring or renewing per-session writer leases. */\n\tasync list(options: SqliteSessionListOptions = {}): Promise<SqliteSessionMetadata[]> {\n\t\treturn this.operations.enqueue(async () => {\n\t\t\tconst path = await this.getDatabasePath();\n\t\t\tif (!resultOrThrow(await this.options.env.exists(path), `Failed to check database ${path}`)) return [];\n\t\t\tconst db = await this.getDatabase();\n\t\t\tconst rows = readSessionRows(db, options);\n\t\t\treturn rows.map((row) => decodeSessionMetadata(row, path));\n\t\t});\n\t}\n\n\tasync delete(metadata: SqliteSessionMetadata): Promise<void> {\n\t\treturn this.operations.enqueue(async () => {\n\t\t\tawait this.releaseStoragesForSession(metadata.id);\n\t\t\tconst db = await this.getDatabase();\n\t\t\tdb.transaction(() => {\n\t\t\t\tif (!sessionExists(db, metadata.id)) {\n\t\t\t\t\tdeleteWriterLease(db, metadata.id);\n\t\t\t\t\treturn;\n\t\t\t\t}\n\t\t\t\tclaimWriterLease(db, metadata.id, this.leaseOptions);\n\t\t\t\tdeleteBranchCache(db, metadata.id);\n\t\t\t\tdeleteFactRows(db, metadata.id);\n\t\t\t\tdeleteLaneRows(db, metadata.id);\n\t\t\t\tdeleteRecordRows(db, metadata.id);\n\t\t\t\tdeleteEntryRows(db, metadata.id);\n\t\t\t\tdeleteWriterLease(db, metadata.id);\n\t\t\t\tdeleteStats(db, metadata.id);\n\t\t\t\tdeleteSequence(db, metadata.id);\n\t\t\t\tdeleteSessionRow(db, metadata.id);\n\t\t\t});\n\t\t});\n\t}\n\n\tasync fork(\n\t\tsource: SqliteSessionMetadata,\n\t\toptions: ForkOptions & SqliteSessionCreateOptions,\n\t): Promise<Session<SqliteSessionMetadata>> {\n\t\treturn this.operations.enqueue(async () => {\n\t\t\tconst db = await this.getDatabase();\n\t\t\tconst path = await this.getDatabasePath();\n\t\t\tconst sourceMetadata = decodeSessionMetadata(requireSessionRow(db, source.id), path);\n\t\t\tconst id = options.id ?? uuidv7();\n\t\t\tif (sessionExists(db, id)) throw new SessionError(\"already_exists\", `Session already exists: ${id}`);\n\n\t\t\tconst entries: EntryRow[] = [];\n\t\t\tconst lanes: { lane: string; leafId: string | null }[] = [];\n\t\t\tconst branchTips: string[] = [];\n\t\t\tlet branchForkTargetId: string | null = null;\n\n\t\t\tif (options.scope === \"tree\") {\n\t\t\t\tentries.push(...readEntryRows(db, source.id, { order: \"oldestFirst\" }));\n\t\t\t\tlanes.push(...readLanes(db, source.id).map((row) => ({ lane: row.lane, leafId: row.leaf_id })));\n\t\t\t\tbranchTips.push(...readBranchTipIds(db, source.id));\n\t\t\t} else {\n\t\t\t\tconst main = readLane(db, source.id, \"main\");\n\t\t\t\tif (!main) throw new SessionError(\"invalid_lane\", \"Lane not found: main\");\n\t\t\t\tconst selectedEntryId = options.entryId ?? main.leaf_id;\n\t\t\t\tif (selectedEntryId !== null) {\n\t\t\t\t\tconst target = readEntryRow(db, source.id, selectedEntryId);\n\t\t\t\t\tif (!target || target.type !== \"message\") {\n\t\t\t\t\t\tthrow new SessionError(\n\t\t\t\t\t\t\t\"invalid_fork_target\",\n\t\t\t\t\t\t\t`Fork target is not a message entry: ${selectedEntryId}`,\n\t\t\t\t\t\t);\n\t\t\t\t\t}\n\t\t\t\t\tconst position = options.position ?? (options.entryId === undefined ? \"at\" : \"before\");\n\t\t\t\t\tbranchForkTargetId = position === \"at\" ? target.id : target.parent_id;\n\t\t\t\t}\n\t\t\t\tlanes.push({ lane: \"main\", leafId: branchForkTargetId });\n\t\t\t\tif (branchForkTargetId !== null) {\n\t\t\t\t\tconst cached = readCachedBranch(db, source.id, branchForkTargetId);\n\t\t\t\t\tif (!cached) {\n\t\t\t\t\t\tthrow new SessionError(\n\t\t\t\t\t\t\t\"invalid_fork_target\",\n\t\t\t\t\t\t\t`Fork target is not on a cached branch: ${branchForkTargetId}`,\n\t\t\t\t\t\t);\n\t\t\t\t\t}\n\t\t\t\t\tconst rows = queryCachedBranchRows(db, source.id, cached, { order: \"oldestFirst\" });\n\t\t\t\t\tentries.push(...rows.map(entryRowFromCached));\n\t\t\t\t\tbranchTips.push(branchForkTargetId);\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tconst copiedIds = new Set(entries.map((entry) => entry.id));\n\t\t\tconst latestName = readLatestFact(db, source.id, \"name\", null);\n\t\t\tconst latestLabels = readLatestLabelFacts(db, source.id);\n\t\t\tconst labelsToCopy = latestLabels.filter(\n\t\t\t\t(row) => options.scope === \"tree\" || (row.key !== null && copiedIds.has(row.key)),\n\t\t\t);\n\t\t\tconst createdAt = Date.now();\n\t\t\tconst metadata = options.metadata ?? sourceMetadata.metadata;\n\t\t\tlet lease: WriterLease;\n\n\t\t\ttry {\n\t\t\t\tlease = db.transaction(() => {\n\t\t\t\t\tinsertSessionRow(db, {\n\t\t\t\t\t\tid,\n\t\t\t\t\t\tcreatedAt: timestampToText(createdAt),\n\t\t\t\t\t\tcwd: options.cwd,\n\t\t\t\t\t\tparentSessionId: options.parentSessionId ?? source.id,\n\t\t\t\t\t\tmetadata,\n\t\t\t\t\t});\n\t\t\t\t\tcreateSequence(db, id);\n\t\t\t\t\tcreateStats(db, id, entries.filter((entry) => entry.type === \"message\").length);\n\n\t\t\t\t\tlet nextSeq = 1;\n\t\t\t\t\tconst allocateSeq = () => nextSeq++;\n\t\t\t\t\tfor (const entry of entries) {\n\t\t\t\t\t\tinsertEntryRow(db, id, {\n\t\t\t\t\t\t\tseq: allocateSeq(),\n\t\t\t\t\t\t\tid: entry.id,\n\t\t\t\t\t\t\tparentId: entry.parent_id,\n\t\t\t\t\t\t\ttype: entry.type,\n\t\t\t\t\t\t\ttimestamp: entry.timestamp,\n\t\t\t\t\t\t\tpayload: entry.payload,\n\t\t\t\t\t\t});\n\t\t\t\t\t}\n\n\t\t\t\t\tif (options.scope === \"tree\") {\n\t\t\t\t\t\tfor (const lane of lanes) insertLane(db, id, allocateSeq(), lane.lane, lane.leafId);\n\t\t\t\t\t} else {\n\t\t\t\t\t\tcreateInitialLane(db, id, \"main\", branchForkTargetId);\n\t\t\t\t\t}\n\n\t\t\t\t\tif (latestName?.value !== undefined && latestName.value !== null) {\n\t\t\t\t\t\tappendFact(db, id, allocateSeq(), \"name\", null, latestName.value);\n\t\t\t\t\t}\n\t\t\t\t\tfor (const label of labelsToCopy) appendFact(db, id, allocateSeq(), \"label\", label.key, label.value);\n\n\t\t\t\t\tsetNextSequence(db, id, nextSeq);\n\t\t\t\t\tfor (const tip of branchTips) buildCachedBranch(db, id, tip);\n\t\t\t\t\treturn claimWriterLease(db, id, this.leaseOptions);\n\t\t\t\t});\n\t\t\t} catch (error) {\n\t\t\t\tif (error instanceof SessionError) throw error;\n\t\t\t\tthrow new SessionError(\n\t\t\t\t\t\"storage\",\n\t\t\t\t\t`Failed to fork SQLite session ${id}`,\n\t\t\t\t\terror instanceof Error ? error : undefined,\n\t\t\t\t);\n\t\t\t}\n\n\t\t\tconst row = requireSessionRow(db, id);\n\t\t\treturn this.sessionFromLease(db, decodeSessionMetadata(row, path), lease);\n\t\t});\n\t}\n\n\tasync close(): Promise<void> {\n\t\tawait this.operations.drain();\n\t\tfor (const storage of [...this.activeStorages]) await storage.release();\n\t\tif (this.database) this.database.close();\n\t\tthis.database = undefined;\n\t\tthis.databasePromise = undefined;\n\t}\n\n\tasync [Symbol.asyncDispose](): Promise<void> {\n\t\tawait this.close();\n\t}\n\n\tprivate async getDatabasePath(): Promise<string> {\n\t\tthis.databasePath ??= resultOrThrow(\n\t\t\tawait this.options.env.absolutePath(this.options.databasePath),\n\t\t\t`Failed to resolve SQLite sessions database ${this.options.databasePath}`,\n\t\t);\n\t\treturn this.databasePath;\n\t}\n\n\tprivate async getDatabase(): Promise<SqliteDatabase> {\n\t\tif (!this.databasePromise) this.databasePromise = this.openDatabase();\n\t\tthis.database = await this.databasePromise;\n\t\treturn this.database;\n\t}\n\n\tprivate async openDatabase(): Promise<SqliteDatabase> {\n\t\tconst path = await this.getDatabasePath();\n\t\tresultOrThrow(\n\t\t\tawait this.options.env.createDir(getParentPath(path), { recursive: true }),\n\t\t\t`Failed to create SQLite sessions directory ${path}`,\n\t\t);\n\t\tconst db = await this.options.sqlite.open(path);\n\t\ttry {\n\t\t\tconfigureSqliteDatabase(db);\n\t\t\tawait applyMigrations(db);\n\t\t\treturn db;\n\t\t} catch (error) {\n\t\t\tdb.close();\n\t\t\tthrow error;\n\t\t}\n\t}\n}\n"]}
1
+ {"version":3,"file":"repo.d.ts","sourceRoot":"","sources":["../../src/sqlite/repo.ts"],"names":[],"mappings":"AACA,OAAO,EAIN,KAAK,WAAW,EAQhB,OAAO,EAEP,KAAK,WAAW,IAAI,iBAAiB,EAGrC,MAAM,+BAA+B,CAAC;AAmEvC,OAAO,KAAK,EAEX,qBAAqB,EACrB,0BAA0B,EAC1B,wBAAwB,EACxB,qBAAqB,EACrB,0BAA0B,EAC1B,MAAM,YAAY,CAAC;AAEpB,MAAM,WAAW,wBAAwB;IACxC,oGAAoG;IACpG,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,4EAA4E;IAC5E,mBAAmB,CAAC,EAAE,MAAM,CAAC;CAC7B;AAED,MAAM,WAAW,8BAA8B;IAC9C,GAAG,EAAE,0BAA0B,CAAC;IAChC,MAAM,EAAE,qBAAqB,CAAC;IAC9B,YAAY,EAAE,MAAM,CAAC;IACrB,WAAW,CAAC,EAAE,wBAAwB,CAAC;CACvC;AAkjBD,qBAAa,uBACZ,YACC,iBAAiB,CAAC,qBAAqB,EAAE,0BAA0B,EAAE,wBAAwB,CAAC,EAC9F,eAAe;IAEhB,OAAO,CAAC,YAAY,CAAqB;IACzC,OAAO,CAAC,QAAQ,CAA6B;IAC7C,OAAO,CAAC,eAAe,CAAsC;IAC7D,OAAO,CAAC,QAAQ,CAAC,UAAU,CAA8B;IACzD,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAmC;IAClE,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAiC;IACzD,OAAO,CAAC,QAAQ,CAAC,YAAY,CAA6B;IAE1D,YAAY,OAAO,EAAE,8BAA8B,EAGlD;YAEa,yBAAyB;IAMvC,OAAO,CAAC,gBAAgB;IAaxB,OAAO,CAAC,YAAY;IAcd,MAAM,CAAC,OAAO,EAAE,0BAA0B,GAAG,OAAO,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC,CAuBzF;IAEK,IAAI,CAAC,QAAQ,EAAE,qBAAqB,GAAG,OAAO,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC,CAEnF;IAED,2FAA2F;IACrF,iBAAiB,CAAC,QAAQ,EAAE,qBAAqB,GAAG,OAAO,CAAC,IAAI,CAAC,CAWtE;IAED,yFAAyF;IACnF,IAAI,CAAC,OAAO,GAAE,wBAA6B,GAAG,OAAO,CAAC,qBAAqB,EAAE,CAAC,CAQnF;IAEK,MAAM,CAAC,QAAQ,EAAE,qBAAqB,GAAG,OAAO,CAAC,IAAI,CAAC,CAqB3D;IAEK,IAAI,CACT,MAAM,EAAE,qBAAqB,EAC7B,OAAO,EAAE,WAAW,GAAG,0BAA0B,GAC/C,OAAO,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC,CA6GzC;IAEK,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAM3B;IAEK,CAAC,MAAM,CAAC,YAAY,CAAC,IAAI,OAAO,CAAC,IAAI,CAAC,CAE3C;YAEa,eAAe;YAQf,WAAW;YAMX,YAAY;CAgB1B","sourcesContent":["import type { FileError, Result } from \"@earendil-works/pi-agent-core\";\nimport {\n\ttype BranchBounds,\n\ttype Entry,\n\ttype EntryQuery,\n\ttype ForkOptions,\n\ttype LaneRecord,\n\ttype LogItem,\n\ttype LogOptions,\n\ttype NewRecord,\n\ttype OperationStartedRecord,\n\ttype ProvisionedEntry,\n\ttype RecordQuery,\n\tSession,\n\tSessionError,\n\ttype SessionRepo as SessionRepository,\n\ttype SessionStats,\n\ttype SessionStorage,\n} from \"@earendil-works/pi-agent-core\";\nimport { uuidv7 } from \"@earendil-works/pi-ai\";\nimport { appendEntryToBranchCache, buildCachedBranch, deleteBranchCache, rebuildBranchCache } from \"./branch-cache.ts\";\nimport { applyMigrations } from \"./migrations.ts\";\nimport { sql } from \"./sql.ts\";\nimport { type CachedBranchEntryRow, queryCachedBranchRows, readCachedBranch } from \"./storage/branch-entries.ts\";\nimport { readBranchTipIds } from \"./storage/branch-tips.ts\";\nimport {\n\tdeleteEntryRows,\n\ttype EntryRow,\n\tentryPayload,\n\tidExistsInEntries,\n\tinsertEntryRow,\n\treadEntryRow,\n\treadEntryRows,\n} from \"./storage/entries.ts\";\nimport { appendFact, deleteFactRows, readFactRows, readLatestFact, readLatestLabelFacts } from \"./storage/facts.ts\";\nimport {\n\tcreateInitialLane,\n\tdeleteLaneRows,\n\tfinishLaneOperation,\n\tcreateLane as insertLane,\n\treadLane,\n\treadLaneHead,\n\treadLaneMoveRows,\n\treadLanes,\n\tsetLaneLeaf,\n\tstartLaneOperation,\n\tmoveLane as updateLane,\n} from \"./storage/lanes.ts\";\nimport {\n\tappendRecordRow,\n\tdeleteRecordRows,\n\tidExistsInRecords,\n\treadOpenOperationRows,\n\treadRecordRows,\n} from \"./storage/records.ts\";\nimport {\n\tadvanceSequence,\n\tcreateSequence,\n\tdeleteSequence,\n\tgetNextSequence,\n\tsetNextSequence,\n} from \"./storage/session-sequences.ts\";\nimport {\n\taddUsageToStats,\n\tcreateStats,\n\tdeleteStats,\n\tincrementMessageCount,\n\treadStats,\n} from \"./storage/session-stats.ts\";\nimport {\n\tdecodeSessionMetadata,\n\tdeleteSessionRow,\n\tinsertSessionRow,\n\treadSessionRow,\n\treadSessionRows,\n\ttype SessionRow,\n\tsessionExists,\n} from \"./storage/sessions.ts\";\nimport {\n\tacquireWriterLease,\n\tdeleteWriterLease,\n\treleaseWriterLease,\n\trenewWriterLease,\n\ttype WriterLease,\n} from \"./storage/writer-leases.ts\";\nimport type {\n\tSqliteDatabase,\n\tSqliteDatabaseFactory,\n\tSqliteSessionCreateOptions,\n\tSqliteSessionListOptions,\n\tSqliteSessionMetadata,\n\tSqliteSessionRepositoryEnv,\n} from \"./types.ts\";\n\nexport interface SqliteWriterLeaseOptions {\n\t/** Time without a successful heartbeat before another writer may take over. Default: 30 seconds. */\n\tttlMs?: number;\n\t/** Idle heartbeat cadence. Default: 10 seconds. Must be less than ttlMs. */\n\theartbeatIntervalMs?: number;\n}\n\nexport interface SqliteSessionRepositoryOptions {\n\tenv: SqliteSessionRepositoryEnv;\n\tsqlite: SqliteDatabaseFactory;\n\tdatabasePath: string;\n\twriterLease?: SqliteWriterLeaseOptions;\n}\n\ninterface ResolvedWriterLeaseOptions {\n\tttlMs: number;\n\theartbeatIntervalMs: number;\n}\n\nfunction resolveWriterLeaseOptions(options: SqliteWriterLeaseOptions | undefined): ResolvedWriterLeaseOptions {\n\tconst ttlMs = options?.ttlMs ?? 30_000;\n\tconst heartbeatIntervalMs = options?.heartbeatIntervalMs ?? 10_000;\n\tif (!Number.isSafeInteger(ttlMs) || ttlMs <= 0) throw new RangeError(\"writerLease.ttlMs must be positive\");\n\tif (!Number.isSafeInteger(heartbeatIntervalMs) || heartbeatIntervalMs <= 0 || heartbeatIntervalMs >= ttlMs) {\n\t\tthrow new RangeError(\"writerLease.heartbeatIntervalMs must be positive and less than ttlMs\");\n\t}\n\treturn { ttlMs, heartbeatIntervalMs };\n}\n\nfunction activeWriterError(sessionId: string): SessionError {\n\treturn new SessionError(\"storage\", `SQLite session ${sessionId} already has an active writer`);\n}\n\nfunction lostWriterError(sessionId: string): SessionError {\n\treturn new SessionError(\"storage\", `SQLite session ${sessionId} writer lease was lost`);\n}\n\nfunction claimWriterLease(db: SqliteDatabase, sessionId: string, options: ResolvedWriterLeaseOptions): WriterLease {\n\tconst now = Date.now();\n\tconst lease = acquireWriterLease(db, sessionId, uuidv7(), now, now + options.ttlMs);\n\tif (!lease) throw activeWriterError(sessionId);\n\treturn lease;\n}\n\nclass SerialOperationQueue {\n\tprivate tail: Promise<void> = Promise.resolve();\n\n\tenqueue<T>(operation: () => Promise<T> | T): Promise<T> {\n\t\tconst result = this.tail.then(operation);\n\t\tthis.tail = result.then(\n\t\t\t() => undefined,\n\t\t\t() => undefined,\n\t\t);\n\t\treturn result;\n\t}\n\n\tasync drain(): Promise<void> {\n\t\tawait this.tail;\n\t}\n}\n\nfunction resultOrThrow<T>(result: Result<T, FileError>, message: string): T {\n\tif (!result.ok) {\n\t\tconst code = result.error.code === \"not_found\" ? \"not_found\" : \"storage\";\n\t\tthrow new SessionError(code, `${message}: ${result.error.message}`, result.error);\n\t}\n\treturn result.value;\n}\n\nfunction getParentPath(path: string): string {\n\tconst normalized = path.replace(/[\\\\/]+$/, \"\");\n\tconst lastSlash = Math.max(normalized.lastIndexOf(\"/\"), normalized.lastIndexOf(\"\\\\\"));\n\tif (lastSlash < 0) return \".\";\n\tif (lastSlash === 0) return normalized.slice(0, 1);\n\treturn normalized.slice(0, lastSlash);\n}\n\nfunction configureSqliteDatabase(db: SqliteDatabase): void {\n\tsql`PRAGMA journal_mode=WAL`.exec(db);\n\tsql`PRAGMA synchronous=FULL`.exec(db);\n\tsql`PRAGMA busy_timeout=5000`.exec(db);\n}\n\nfunction entryRowFromCached(row: CachedBranchEntryRow): EntryRow {\n\treturn { ...row, seq: row.entry_seq, type: row.type as Entry[\"type\"] };\n}\n\nfunction readObjectPayload(row: EntryRow): Record<string, unknown> {\n\tconst payload = JSON.parse(row.payload) as unknown;\n\tif (typeof payload !== \"object\" || payload === null || Array.isArray(payload)) {\n\t\tthrow new Error(\"Payload is not an object\");\n\t}\n\treturn payload as Record<string, unknown>;\n}\n\nfunction decodeEntry(row: EntryRow): Entry {\n\ttry {\n\t\tconst payload = readObjectPayload(row);\n\t\tconst base = { id: row.id, seq: row.seq, parentId: row.parent_id, timestamp: row.timestamp };\n\t\tswitch (row.type) {\n\t\t\tcase \"message\":\n\t\t\t\tif (typeof payload.message !== \"object\" || payload.message === null) throw new Error(\"Missing message\");\n\t\t\t\treturn {\n\t\t\t\t\t...base,\n\t\t\t\t\ttype: \"message\",\n\t\t\t\t\tmessage: payload.message as Extract<Entry, { type: \"message\" }>[\"message\"],\n\t\t\t\t\t...(payload.terminate === true ? { terminate: true as const } : {}),\n\t\t\t\t};\n\t\t\tcase \"model_change\":\n\t\t\t\tif (typeof payload.provider !== \"string\" || typeof payload.modelId !== \"string\") {\n\t\t\t\t\tthrow new Error(\"Invalid model_change payload\");\n\t\t\t\t}\n\t\t\t\treturn { ...base, type: \"model_change\", provider: payload.provider, modelId: payload.modelId };\n\t\t\tcase \"thinking_level_change\":\n\t\t\t\tif (typeof payload.thinkingLevel !== \"string\") throw new Error(\"Invalid thinking_level_change payload\");\n\t\t\t\treturn { ...base, type: \"thinking_level_change\", thinkingLevel: payload.thinkingLevel };\n\t\t\tcase \"active_tools_change\":\n\t\t\t\tif (!Array.isArray(payload.activeToolNames)) throw new Error(\"Invalid active_tools_change payload\");\n\t\t\t\tif (payload.activeToolNames.some((value) => typeof value !== \"string\")) {\n\t\t\t\t\tthrow new Error(\"Invalid active_tools_change payload\");\n\t\t\t\t}\n\t\t\t\treturn { ...base, type: \"active_tools_change\", activeToolNames: payload.activeToolNames };\n\t\t\tcase \"compaction\":\n\t\t\t\tif (\n\t\t\t\t\ttypeof payload.summary !== \"string\" ||\n\t\t\t\t\t!Array.isArray(payload.retainedTail) ||\n\t\t\t\t\ttypeof payload.tokensBefore !== \"number\"\n\t\t\t\t) {\n\t\t\t\t\tthrow new Error(\"Invalid compaction payload\");\n\t\t\t\t}\n\t\t\t\treturn {\n\t\t\t\t\t...base,\n\t\t\t\t\ttype: \"compaction\",\n\t\t\t\t\tsummary: payload.summary,\n\t\t\t\t\tretainedTail: payload.retainedTail as Extract<Entry, { type: \"compaction\" }>[\"retainedTail\"],\n\t\t\t\t\ttokensBefore: payload.tokensBefore,\n\t\t\t\t\t...(Object.hasOwn(payload, \"details\") ? { details: payload.details } : {}),\n\t\t\t\t\t...(Object.hasOwn(payload, \"usage\")\n\t\t\t\t\t\t? { usage: payload.usage as Extract<Entry, { type: \"compaction\" }>[\"usage\"] }\n\t\t\t\t\t\t: {}),\n\t\t\t\t};\n\t\t\tcase \"branch_summary\":\n\t\t\t\tif (typeof payload.fromId !== \"string\" || typeof payload.summary !== \"string\") {\n\t\t\t\t\tthrow new Error(\"Invalid branch_summary payload\");\n\t\t\t\t}\n\t\t\t\treturn {\n\t\t\t\t\t...base,\n\t\t\t\t\ttype: \"branch_summary\",\n\t\t\t\t\tfromId: payload.fromId,\n\t\t\t\t\tsummary: payload.summary,\n\t\t\t\t\t...(Object.hasOwn(payload, \"details\") ? { details: payload.details } : {}),\n\t\t\t\t\t...(Object.hasOwn(payload, \"usage\")\n\t\t\t\t\t\t? { usage: payload.usage as Extract<Entry, { type: \"branch_summary\" }>[\"usage\"] }\n\t\t\t\t\t\t: {}),\n\t\t\t\t};\n\t\t\tcase \"custom\":\n\t\t\t\tif (typeof payload.customType !== \"string\") throw new Error(\"Invalid custom payload\");\n\t\t\t\treturn {\n\t\t\t\t\t...base,\n\t\t\t\t\ttype: \"custom\",\n\t\t\t\t\tcustomType: payload.customType,\n\t\t\t\t\t...(Object.hasOwn(payload, \"data\") ? { data: payload.data } : {}),\n\t\t\t\t};\n\t\t}\n\t} catch (error) {\n\t\tthrow new SessionError(\n\t\t\t\"invalid_entry\",\n\t\t\t`Invalid SQLite session entry ${row.id}: failed to decode entry ${row.id}`,\n\t\t\terror instanceof Error ? error : undefined,\n\t\t);\n\t}\n}\n\nfunction recordRunId(record: NewRecord): string | undefined {\n\treturn record.type === \"operation_started\" ? record.id : \"runId\" in record ? record.runId : undefined;\n}\n\nfunction recordOpKind(record: NewRecord): string | undefined {\n\treturn record.type === \"operation_started\" ? record.intent.kind : undefined;\n}\n\nfunction decodeRecord(row: { seq: number; timestamp: number; payload: string }): LaneRecord {\n\ttry {\n\t\treturn {\n\t\t\t...(JSON.parse(row.payload) as object),\n\t\t\tseq: row.seq,\n\t\t\ttimestamp: row.timestamp,\n\t\t} as LaneRecord;\n\t} catch (error) {\n\t\tthrow new SessionError(\n\t\t\t\"storage\",\n\t\t\t`Invalid SQLite session record at sequence ${row.seq}: failed to decode payload`,\n\t\t\terror instanceof Error ? error : undefined,\n\t\t);\n\t}\n}\n\nfunction validateCachedBranchRows(rows: readonly CachedBranchEntryRow[], query: BranchBounds & EntryQuery): void {\n\tif (rows.length === 0 || query.type !== undefined || query.customType !== undefined) return;\n\tconst path = [...rows].sort((left, right) => left.entry_seq - right.entry_seq);\n\tconst shouldIncludeRoot =\n\t\tquery.stopAtId === undefined &&\n\t\tquery.stopAtType === undefined &&\n\t\tquery.cursor === undefined &&\n\t\t(query.order === \"oldestFirst\" || query.limit === undefined);\n\tif (shouldIncludeRoot && path[0]?.parent_id !== null) {\n\t\tthrow new SessionError(\"invalid_entry\", `Entry ${path[0]?.parent_id} not found`);\n\t}\n\tfor (let index = 1; index < path.length; index++) {\n\t\tconst previous = path[index - 1]!;\n\t\tconst current = path[index]!;\n\t\tif (current.parent_id !== previous.id) {\n\t\t\tthrow new SessionError(\"invalid_entry\", `Entry ${current.parent_id} not found`);\n\t\t}\n\t}\n}\n\nfunction matchesEntryQuery(entry: Entry, query: EntryQuery): boolean {\n\treturn (\n\t\t(query.type === undefined || entry.type === query.type) &&\n\t\t(query.customType === undefined || (entry.type === \"custom\" && entry.customType === query.customType)) &&\n\t\t(query.cursor === undefined ||\n\t\t\t(query.order === \"oldestFirst\" ? entry.seq > query.cursor.afterSeq : entry.seq < query.cursor.afterSeq))\n\t);\n}\n\nfunction assertUnusedId(db: SqliteDatabase, sessionId: string, id: string): void {\n\tif (idExistsInEntries(db, sessionId, id) || idExistsInRecords(db, sessionId, id)) {\n\t\tthrow new SessionError(\"already_exists\", `ID already exists: ${id}`);\n\t}\n}\n\nfunction requireSessionRow(db: SqliteDatabase, sessionId: string): SessionRow {\n\tconst row = readSessionRow(db, sessionId);\n\tif (!row) throw new SessionError(\"not_found\", `Session not found: ${sessionId}`);\n\treturn row;\n}\n\nclass SqliteSessionStorage implements SessionStorage<SqliteSessionMetadata> {\n\tprivate readonly db: SqliteDatabase;\n\tprivate readonly metadata: SqliteSessionMetadata;\n\tprivate readonly lease: WriterLease;\n\tprivate readonly leaseOptions: ResolvedWriterLeaseOptions;\n\tprivate readonly onRelease: () => void;\n\tprivate readonly operations = new SerialOperationQueue();\n\tprivate heartbeatTimer: ReturnType<typeof setTimeout> | undefined;\n\tprivate leaseError: SessionError | undefined;\n\tprivate closing = false;\n\tprivate releasePromise: Promise<void> | undefined;\n\n\tconstructor(\n\t\tdb: SqliteDatabase,\n\t\tmetadata: SqliteSessionMetadata,\n\t\tlease: WriterLease,\n\t\tleaseOptions: ResolvedWriterLeaseOptions,\n\t\tonRelease: () => void,\n\t) {\n\t\tthis.db = db;\n\t\tthis.metadata = metadata;\n\t\tthis.lease = lease;\n\t\tthis.leaseOptions = leaseOptions;\n\t\tthis.onRelease = onRelease;\n\t\tthis.scheduleHeartbeat();\n\t}\n\n\tasync release(): Promise<void> {\n\t\tthis.releasePromise ??= this.finishRelease();\n\t\tawait this.releasePromise;\n\t}\n\n\tprivate async finishRelease(): Promise<void> {\n\t\tthis.closing = true;\n\t\tif (this.heartbeatTimer !== undefined) clearTimeout(this.heartbeatTimer);\n\t\ttry {\n\t\t\tawait this.operations.enqueue(() =>\n\t\t\t\tthis.db.transaction(() => releaseWriterLease(this.db, this.metadata.id, this.lease)),\n\t\t\t);\n\t\t} finally {\n\t\t\tthis.onRelease();\n\t\t}\n\t}\n\n\tprivate enqueueWrite<T>(operation: () => T): Promise<T> {\n\t\tif (this.closing)\n\t\t\treturn Promise.reject(new SessionError(\"storage\", `SQLite session ${this.metadata.id} is closed`));\n\t\treturn this.operations.enqueue(() => {\n\t\t\tif (this.leaseError) throw this.leaseError;\n\t\t\treturn this.db.transaction(() => {\n\t\t\t\tconst now = Date.now();\n\t\t\t\tif (!renewWriterLease(this.db, this.metadata.id, this.lease, now, now + this.leaseOptions.ttlMs)) {\n\t\t\t\t\tthis.leaseError = lostWriterError(this.metadata.id);\n\t\t\t\t\tif (this.heartbeatTimer !== undefined) clearTimeout(this.heartbeatTimer);\n\t\t\t\t\tthrow this.leaseError;\n\t\t\t\t}\n\t\t\t\treturn operation();\n\t\t\t});\n\t\t});\n\t}\n\n\tprivate scheduleHeartbeat(): void {\n\t\tif (this.closing || this.leaseError) return;\n\t\tthis.heartbeatTimer = setTimeout(async () => {\n\t\t\tthis.heartbeatTimer = undefined;\n\t\t\ttry {\n\t\t\t\tawait this.operations.enqueue(() => {\n\t\t\t\t\tif (this.closing || this.leaseError) return;\n\t\t\t\t\tthis.db.transaction(() => {\n\t\t\t\t\t\tconst now = Date.now();\n\t\t\t\t\t\tif (!renewWriterLease(this.db, this.metadata.id, this.lease, now, now + this.leaseOptions.ttlMs)) {\n\t\t\t\t\t\t\tthis.leaseError = lostWriterError(this.metadata.id);\n\t\t\t\t\t\t}\n\t\t\t\t\t});\n\t\t\t\t});\n\t\t\t} catch {\n\t\t\t\t// A transient heartbeat failure is retried. Every write still verifies ownership transactionally.\n\t\t\t} finally {\n\t\t\t\tthis.scheduleHeartbeat();\n\t\t\t}\n\t\t}, this.leaseOptions.heartbeatIntervalMs);\n\t\tthis.heartbeatTimer.unref();\n\t}\n\n\tasync getMetadata(): Promise<SqliteSessionMetadata> {\n\t\treturn decodeSessionMetadata(requireSessionRow(this.db, this.metadata.id), this.metadata.path);\n\t}\n\n\tisForSession(sessionId: string): boolean {\n\t\treturn this.metadata.id === sessionId;\n\t}\n\n\tasync getLanes(): Promise<{ lane: string; leafId: string | null }[]> {\n\t\treturn readLanes(this.db, this.metadata.id).map((row) => ({ lane: row.lane, leafId: row.leaf_id }));\n\t}\n\n\tasync createLane(lane: string, at: string | null): Promise<void> {\n\t\treturn this.enqueueWrite(() => {\n\t\t\tif (readLane(this.db, this.metadata.id, lane)) {\n\t\t\t\tthrow new SessionError(\"already_exists\", `Lane already exists: ${lane}`);\n\t\t\t}\n\t\t\tif (at !== null && !readEntryRow(this.db, this.metadata.id, at)) {\n\t\t\t\tthrow new SessionError(\"not_found\", `Entry not found: ${at}`);\n\t\t\t}\n\t\t\tconst seq = getNextSequence(this.db, this.metadata.id);\n\t\t\tinsertLane(this.db, this.metadata.id, seq, lane, at);\n\t\t\tadvanceSequence(this.db, this.metadata.id, seq);\n\t\t});\n\t}\n\n\tasync moveLane(lane: string, to: string | null): Promise<void> {\n\t\treturn this.enqueueWrite(() => {\n\t\t\tif (!readLane(this.db, this.metadata.id, lane))\n\t\t\t\tthrow new SessionError(\"invalid_lane\", `Lane not found: ${lane}`);\n\t\t\tif (to !== null && !readEntryRow(this.db, this.metadata.id, to)) {\n\t\t\t\tthrow new SessionError(\"not_found\", `Entry not found: ${to}`);\n\t\t\t}\n\t\t\tconst seq = getNextSequence(this.db, this.metadata.id);\n\t\t\tupdateLane(this.db, this.metadata.id, seq, lane, to);\n\t\t\tadvanceSequence(this.db, this.metadata.id, seq);\n\t\t});\n\t}\n\n\tasync appendEntry<TEntry extends Entry>(entry: ProvisionedEntry<TEntry>, lane: string): Promise<TEntry> {\n\t\treturn this.enqueueWrite(() => {\n\t\t\tconst parentId = readLaneHead(this.db, this.metadata.id, lane).leafId;\n\t\t\tassertUnusedId(this.db, this.metadata.id, entry.id);\n\t\t\tconst seq = getNextSequence(this.db, this.metadata.id);\n\t\t\tconst committed = { ...entry, parentId, seq, timestamp: Date.now() } as Entry;\n\t\t\tinsertEntryRow(this.db, this.metadata.id, {\n\t\t\t\tseq,\n\t\t\t\tid: committed.id,\n\t\t\t\tparentId: committed.parentId,\n\t\t\t\ttype: committed.type,\n\t\t\t\ttimestamp: committed.timestamp,\n\t\t\t\tpayload: JSON.stringify(entryPayload(committed)),\n\t\t\t});\n\t\t\tsetLaneLeaf(this.db, this.metadata.id, lane, committed.id);\n\t\t\tappendEntryToBranchCache(\n\t\t\t\tthis.db,\n\t\t\t\tthis.metadata.id,\n\t\t\t\tcommitted.id,\n\t\t\t\tseq,\n\t\t\t\tcommitted.type,\n\t\t\t\tcommitted.type === \"custom\" ? committed.customType : null,\n\t\t\t\tcommitted.parentId,\n\t\t\t);\n\t\t\tif (committed.type === \"message\") incrementMessageCount(this.db, this.metadata.id);\n\t\t\tadvanceSequence(this.db, this.metadata.id, seq);\n\t\t\treturn structuredClone(committed as TEntry);\n\t\t});\n\t}\n\n\tasync appendRecord<TRecord extends LaneRecord>(record: NewRecord<TRecord>): Promise<TRecord>;\n\tasync appendRecord(record: NewRecord): Promise<LaneRecord> {\n\t\treturn this.enqueueWrite(() => {\n\t\t\tif (!readLane(this.db, this.metadata.id, record.lane)) {\n\t\t\t\tthrow new SessionError(\"invalid_lane\", `Lane not found: ${record.lane}`);\n\t\t\t}\n\t\t\tassertUnusedId(this.db, this.metadata.id, record.id);\n\t\t\tconst seq = getNextSequence(this.db, this.metadata.id);\n\t\t\tconst committed: LaneRecord = { ...record, seq, timestamp: Date.now() };\n\t\t\tif (record.type === \"operation_started\") {\n\t\t\t\tstartLaneOperation(this.db, this.metadata.id, record.lane, record.id);\n\t\t\t}\n\t\t\tappendRecordRow(this.db, this.metadata.id, {\n\t\t\t\tseq,\n\t\t\t\tid: record.id,\n\t\t\t\tlane: record.lane,\n\t\t\t\trunId: recordRunId(record),\n\t\t\t\ttype: record.type,\n\t\t\t\topKind: recordOpKind(record),\n\t\t\t\ttimestamp: committed.timestamp,\n\t\t\t\tpayload: JSON.stringify(record),\n\t\t\t});\n\t\t\tif (record.type === \"operation_finished\") {\n\t\t\t\tfinishLaneOperation(this.db, this.metadata.id, record.lane, record.runId);\n\t\t\t}\n\t\t\tif (record.type === \"usage\") addUsageToStats(this.db, this.metadata.id, record.usage);\n\t\t\tadvanceSequence(this.db, this.metadata.id, seq);\n\t\t\treturn structuredClone(committed);\n\t\t});\n\t}\n\n\tasync getEntry(id: string): Promise<Entry | undefined> {\n\t\tconst row = readEntryRow(this.db, this.metadata.id, id);\n\t\treturn row ? decodeEntry(row) : undefined;\n\t}\n\n\tasync findEntries(query: EntryQuery = {}): Promise<Entry[]> {\n\t\tconst sqlType = query.type ?? (query.customType === undefined ? undefined : \"custom\");\n\t\tconst sqlLimit = query.customType === undefined ? query.limit : undefined;\n\t\tconst rows = readEntryRows(this.db, this.metadata.id, {\n\t\t\tcursor: query.cursor,\n\t\t\tlimit: sqlLimit,\n\t\t\torder: query.order,\n\t\t\ttype: sqlType,\n\t\t});\n\t\tconst entries = rows.map(decodeEntry).filter((entry) => matchesEntryQuery(entry, query));\n\t\treturn query.limit === undefined ? entries : entries.slice(0, query.limit);\n\t}\n\n\tasync findEntriesOnBranch(query: EntryQuery & BranchBounds & { start: string }): Promise<Entry[]> {\n\t\tconst cached = readCachedBranch(this.db, this.metadata.id, query.start);\n\t\tif (!cached) {\n\t\t\tif (!readEntryRow(this.db, this.metadata.id, query.start))\n\t\t\t\tthrow new SessionError(\"not_found\", `Entry not found: ${query.start}`);\n\t\t\tthrow new SessionError(\"invalid_entry\", `Branch cache missing entry ${query.start}`);\n\t\t}\n\t\tconst rows = queryCachedBranchRows(this.db, this.metadata.id, cached, query);\n\t\tvalidateCachedBranchRows(rows, query);\n\t\tconst entries = rows\n\t\t\t.map(entryRowFromCached)\n\t\t\t.map(decodeEntry)\n\t\t\t.filter((entry) => matchesEntryQuery(entry, query));\n\t\treturn query.limit === undefined ? entries : entries.slice(0, query.limit);\n\t}\n\n\tasync findRecords(query: RecordQuery = {}): Promise<LaneRecord[]> {\n\t\tconst rows = readRecordRows(this.db, this.metadata.id, query);\n\t\treturn rows.map(decodeRecord);\n\t}\n\n\tasync findOpenOperations(lane: string, options?: { limit?: number }): Promise<OperationStartedRecord[]> {\n\t\tconst rows = readOpenOperationRows(this.db, this.metadata.id, lane, options);\n\n\t\treturn rows.map((row) => {\n\t\t\tconst record = decodeRecord(row);\n\t\t\tif (record.type !== \"operation_started\") {\n\t\t\t\tthrow new SessionError(\"storage\", \"Expected operation_started record\");\n\t\t\t}\n\t\t\treturn record;\n\t\t});\n\t}\n\n\tasync getLog(options: LogOptions = {}): Promise<LogItem[]> {\n\t\tconst afterSeq = options.afterSeq ?? 0;\n\t\tconst limit = options.limit;\n\t\tconst entryRows = readEntryRows(this.db, this.metadata.id, { afterSeq, order: \"oldestFirst\", limit });\n\t\tconst recordRows = readRecordRows(this.db, this.metadata.id, { afterSeq, order: \"oldestFirst\", limit });\n\t\tconst laneRows = readLaneMoveRows(this.db, this.metadata.id, { afterSeq, limit });\n\t\tconst factRows = readFactRows(this.db, this.metadata.id, { afterSeq, limit });\n\n\t\tconst logRows: { seq: number; decode: () => LogItem }[] = [\n\t\t\t...entryRows.map((row) => ({\n\t\t\t\tseq: row.seq,\n\t\t\t\tdecode: () => ({ kind: \"entry\" as const, seq: row.seq, entry: decodeEntry(row) }),\n\t\t\t})),\n\t\t\t...recordRows.map((row) => ({\n\t\t\t\tseq: row.seq,\n\t\t\t\tdecode: () => ({ kind: \"record\" as const, seq: row.seq, record: decodeRecord(row) }),\n\t\t\t})),\n\t\t\t...laneRows.map((row) => ({\n\t\t\t\tseq: row.seq,\n\t\t\t\tdecode: () => ({ kind: \"lane\" as const, seq: row.seq, lane: row.lane, leafId: row.leaf_id }),\n\t\t\t})),\n\t\t\t...factRows.map((row) => ({\n\t\t\t\tseq: row.seq,\n\t\t\t\tdecode: () => {\n\t\t\t\t\tif (row.kind === \"name\")\n\t\t\t\t\t\treturn {\n\t\t\t\t\t\t\tkind: \"fact\" as const,\n\t\t\t\t\t\t\tseq: row.seq,\n\t\t\t\t\t\t\tfact: \"name\" as const,\n\t\t\t\t\t\t\tname: row.value === null ? undefined : (JSON.parse(row.value) as string),\n\t\t\t\t\t\t};\n\t\t\t\t\treturn {\n\t\t\t\t\t\tkind: \"fact\" as const,\n\t\t\t\t\t\tseq: row.seq,\n\t\t\t\t\t\tfact: \"label\" as const,\n\t\t\t\t\t\ttargetId: row.key ?? \"\",\n\t\t\t\t\t\tlabel: row.value === null ? undefined : (JSON.parse(row.value) as string),\n\t\t\t\t\t};\n\t\t\t\t},\n\t\t\t})),\n\t\t].sort((left, right) => left.seq - right.seq);\n\t\tconst selectedRows = options.limit === undefined ? logRows : logRows.slice(0, options.limit);\n\t\treturn selectedRows.map((row) => row.decode());\n\t}\n\n\tasync getName(): Promise<string | undefined> {\n\t\tconst row = readLatestFact(this.db, this.metadata.id, \"name\", null);\n\t\treturn row?.value === undefined || row.value === null ? undefined : (JSON.parse(row.value) as string);\n\t}\n\n\tasync setName(name: string | undefined): Promise<void> {\n\t\treturn this.enqueueWrite(() => {\n\t\t\tconst seq = getNextSequence(this.db, this.metadata.id);\n\t\t\tappendFact(this.db, this.metadata.id, seq, \"name\", null, name === undefined ? null : JSON.stringify(name));\n\t\t\tadvanceSequence(this.db, this.metadata.id, seq);\n\t\t});\n\t}\n\n\tasync getLabel(id: string): Promise<string | undefined> {\n\t\tconst row = readLatestFact(this.db, this.metadata.id, \"label\", id);\n\t\treturn row?.value === undefined || row.value === null ? undefined : (JSON.parse(row.value) as string);\n\t}\n\n\tasync setLabel(id: string, label: string | undefined): Promise<void> {\n\t\treturn this.enqueueWrite(() => {\n\t\t\tif (!readEntryRow(this.db, this.metadata.id, id)) {\n\t\t\t\tthrow new SessionError(\"not_found\", `Entry not found: ${id}`);\n\t\t\t}\n\t\t\tconst seq = getNextSequence(this.db, this.metadata.id);\n\t\t\tappendFact(this.db, this.metadata.id, seq, \"label\", id, label === undefined ? null : JSON.stringify(label));\n\t\t\tadvanceSequence(this.db, this.metadata.id, seq);\n\t\t});\n\t}\n\n\tasync getStats(): Promise<SessionStats> {\n\t\treturn readStats(this.db, this.metadata.id);\n\t}\n}\n\nfunction claimStorage(\n\tdb: SqliteDatabase,\n\tmetadata: SqliteSessionMetadata,\n\tleaseOptions: ResolvedWriterLeaseOptions,\n\tonRelease: () => void,\n): SqliteSessionStorage {\n\trequireSessionRow(db, metadata.id);\n\tconst claimed = db.transaction(() => {\n\t\tconst lease = claimWriterLease(db, metadata.id, leaseOptions);\n\t\tconst row = requireSessionRow(db, metadata.id);\n\t\treadLanes(db, metadata.id);\n\t\treturn { lease, row };\n\t});\n\treturn new SqliteSessionStorage(\n\t\tdb,\n\t\tdecodeSessionMetadata(claimed.row, metadata.path),\n\t\tclaimed.lease,\n\t\tleaseOptions,\n\t\tonRelease,\n\t);\n}\n\nexport class SqliteSessionRepository\n\timplements\n\t\tSessionRepository<SqliteSessionMetadata, SqliteSessionCreateOptions, SqliteSessionListOptions>,\n\t\tAsyncDisposable\n{\n\tprivate databasePath: string | undefined;\n\tprivate database: SqliteDatabase | undefined;\n\tprivate databasePromise: Promise<SqliteDatabase> | undefined;\n\tprivate readonly operations = new SerialOperationQueue();\n\tprivate readonly activeStorages = new Set<SqliteSessionStorage>();\n\tprivate readonly options: SqliteSessionRepositoryOptions;\n\tprivate readonly leaseOptions: ResolvedWriterLeaseOptions;\n\n\tconstructor(options: SqliteSessionRepositoryOptions) {\n\t\tthis.options = options;\n\t\tthis.leaseOptions = resolveWriterLeaseOptions(options.writerLease);\n\t}\n\n\tprivate async releaseStoragesForSession(sessionId: string): Promise<void> {\n\t\tfor (const storage of [...this.activeStorages]) {\n\t\t\tif (storage.isForSession(sessionId)) await storage.release();\n\t\t}\n\t}\n\n\tprivate sessionFromLease(\n\t\tdb: SqliteDatabase,\n\t\tmetadata: SqliteSessionMetadata,\n\t\tlease: WriterLease,\n\t): Session<SqliteSessionMetadata> {\n\t\tlet storage: SqliteSessionStorage;\n\t\tstorage = new SqliteSessionStorage(db, metadata, lease, this.leaseOptions, () => {\n\t\t\tthis.activeStorages.delete(storage);\n\t\t});\n\t\tthis.activeStorages.add(storage);\n\t\treturn new Session(storage);\n\t}\n\n\tprivate claimSession(db: SqliteDatabase, metadata: SqliteSessionMetadata): Session<SqliteSessionMetadata> {\n\t\tconst active = [...this.activeStorages].find((storage) => storage.isForSession(metadata.id));\n\t\tif (active) {\n\t\t\treadLanes(db, metadata.id);\n\t\t\treturn new Session(active);\n\t\t}\n\t\tlet storage: SqliteSessionStorage;\n\t\tstorage = claimStorage(db, metadata, this.leaseOptions, () => {\n\t\t\tthis.activeStorages.delete(storage);\n\t\t});\n\t\tthis.activeStorages.add(storage);\n\t\treturn new Session(storage);\n\t}\n\n\tasync create(options: SqliteSessionCreateOptions): Promise<Session<SqliteSessionMetadata>> {\n\t\treturn this.operations.enqueue(async () => {\n\t\t\tconst db = await this.getDatabase();\n\t\t\tconst path = await this.getDatabasePath();\n\t\t\tconst id = options.id ?? uuidv7();\n\t\t\tif (sessionExists(db, id)) throw new SessionError(\"already_exists\", `Session already exists: ${id}`);\n\t\t\tconst createdAt = Date.now();\n\t\t\tconst lease = db.transaction(() => {\n\t\t\t\tinsertSessionRow(db, {\n\t\t\t\t\tid,\n\t\t\t\t\tcreatedAt,\n\t\t\t\t\tcwd: options.cwd,\n\t\t\t\t\tparentSessionId: options.parentSessionId,\n\t\t\t\t\tmetadata: options.metadata,\n\t\t\t\t});\n\t\t\t\tcreateSequence(db, id);\n\t\t\t\tcreateStats(db, id);\n\t\t\t\tcreateInitialLane(db, id);\n\t\t\t\treturn claimWriterLease(db, id, this.leaseOptions);\n\t\t\t});\n\t\t\tconst row = requireSessionRow(db, id);\n\t\t\treturn this.sessionFromLease(db, decodeSessionMetadata(row, path), lease);\n\t\t});\n\t}\n\n\tasync open(metadata: SqliteSessionMetadata): Promise<Session<SqliteSessionMetadata>> {\n\t\treturn this.operations.enqueue(async () => this.claimSession(await this.getDatabase(), metadata));\n\t}\n\n\t/** Rebuilds this session's private branch-read cache from canonical entry parent links. */\n\tasync repairBranchCache(metadata: SqliteSessionMetadata): Promise<void> {\n\t\treturn this.operations.enqueue(async () => {\n\t\t\tawait this.releaseStoragesForSession(metadata.id);\n\t\t\tconst db = await this.getDatabase();\n\t\t\tdb.transaction(() => {\n\t\t\t\tconst lease = claimWriterLease(db, metadata.id, this.leaseOptions);\n\t\t\t\trequireSessionRow(db, metadata.id);\n\t\t\t\trebuildBranchCache(db, metadata.id);\n\t\t\t\treleaseWriterLease(db, metadata.id, lease);\n\t\t\t});\n\t\t});\n\t}\n\n\t/** Reads the session catalog without acquiring or renewing per-session writer leases. */\n\tasync list(options: SqliteSessionListOptions = {}): Promise<SqliteSessionMetadata[]> {\n\t\treturn this.operations.enqueue(async () => {\n\t\t\tconst path = await this.getDatabasePath();\n\t\t\tif (!resultOrThrow(await this.options.env.exists(path), `Failed to check database ${path}`)) return [];\n\t\t\tconst db = await this.getDatabase();\n\t\t\tconst rows = readSessionRows(db, options);\n\t\t\treturn rows.map((row) => decodeSessionMetadata(row, path));\n\t\t});\n\t}\n\n\tasync delete(metadata: SqliteSessionMetadata): Promise<void> {\n\t\treturn this.operations.enqueue(async () => {\n\t\t\tawait this.releaseStoragesForSession(metadata.id);\n\t\t\tconst db = await this.getDatabase();\n\t\t\tdb.transaction(() => {\n\t\t\t\tif (!sessionExists(db, metadata.id)) {\n\t\t\t\t\tdeleteWriterLease(db, metadata.id);\n\t\t\t\t\treturn;\n\t\t\t\t}\n\t\t\t\tclaimWriterLease(db, metadata.id, this.leaseOptions);\n\t\t\t\tdeleteBranchCache(db, metadata.id);\n\t\t\t\tdeleteFactRows(db, metadata.id);\n\t\t\t\tdeleteLaneRows(db, metadata.id);\n\t\t\t\tdeleteRecordRows(db, metadata.id);\n\t\t\t\tdeleteEntryRows(db, metadata.id);\n\t\t\t\tdeleteWriterLease(db, metadata.id);\n\t\t\t\tdeleteStats(db, metadata.id);\n\t\t\t\tdeleteSequence(db, metadata.id);\n\t\t\t\tdeleteSessionRow(db, metadata.id);\n\t\t\t});\n\t\t});\n\t}\n\n\tasync fork(\n\t\tsource: SqliteSessionMetadata,\n\t\toptions: ForkOptions & SqliteSessionCreateOptions,\n\t): Promise<Session<SqliteSessionMetadata>> {\n\t\treturn this.operations.enqueue(async () => {\n\t\t\tconst db = await this.getDatabase();\n\t\t\tconst path = await this.getDatabasePath();\n\t\t\tconst sourceMetadata = decodeSessionMetadata(requireSessionRow(db, source.id), path);\n\t\t\tconst id = options.id ?? uuidv7();\n\t\t\tif (sessionExists(db, id)) throw new SessionError(\"already_exists\", `Session already exists: ${id}`);\n\n\t\t\tconst entries: EntryRow[] = [];\n\t\t\tconst lanes: { lane: string; leafId: string | null }[] = [];\n\t\t\tconst branchTips: string[] = [];\n\t\t\tlet branchForkTargetId: string | null = null;\n\n\t\t\tif (options.scope === \"tree\") {\n\t\t\t\tentries.push(...readEntryRows(db, source.id, { order: \"oldestFirst\" }));\n\t\t\t\tlanes.push(...readLanes(db, source.id).map((row) => ({ lane: row.lane, leafId: row.leaf_id })));\n\t\t\t\tbranchTips.push(...readBranchTipIds(db, source.id));\n\t\t\t} else {\n\t\t\t\tconst main = readLane(db, source.id, \"main\");\n\t\t\t\tif (!main) throw new SessionError(\"invalid_lane\", \"Lane not found: main\");\n\t\t\t\tconst selectedEntryId = options.entryId ?? main.leaf_id;\n\t\t\t\tif (selectedEntryId !== null) {\n\t\t\t\t\tconst target = readEntryRow(db, source.id, selectedEntryId);\n\t\t\t\t\tif (!target || target.type !== \"message\") {\n\t\t\t\t\t\tthrow new SessionError(\n\t\t\t\t\t\t\t\"invalid_fork_target\",\n\t\t\t\t\t\t\t`Fork target is not a message entry: ${selectedEntryId}`,\n\t\t\t\t\t\t);\n\t\t\t\t\t}\n\t\t\t\t\tconst position = options.position ?? (options.entryId === undefined ? \"at\" : \"before\");\n\t\t\t\t\tbranchForkTargetId = position === \"at\" ? target.id : target.parent_id;\n\t\t\t\t}\n\t\t\t\tlanes.push({ lane: \"main\", leafId: branchForkTargetId });\n\t\t\t\tif (branchForkTargetId !== null) {\n\t\t\t\t\tconst cached = readCachedBranch(db, source.id, branchForkTargetId);\n\t\t\t\t\tif (!cached) {\n\t\t\t\t\t\tthrow new SessionError(\n\t\t\t\t\t\t\t\"invalid_fork_target\",\n\t\t\t\t\t\t\t`Fork target is not on a cached branch: ${branchForkTargetId}`,\n\t\t\t\t\t\t);\n\t\t\t\t\t}\n\t\t\t\t\tconst rows = queryCachedBranchRows(db, source.id, cached, { order: \"oldestFirst\" });\n\t\t\t\t\tentries.push(...rows.map(entryRowFromCached));\n\t\t\t\t\tbranchTips.push(branchForkTargetId);\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tconst copiedIds = new Set(entries.map((entry) => entry.id));\n\t\t\tconst latestName = readLatestFact(db, source.id, \"name\", null);\n\t\t\tconst latestLabels = readLatestLabelFacts(db, source.id);\n\t\t\tconst labelsToCopy = latestLabels.filter(\n\t\t\t\t(row) => options.scope === \"tree\" || (row.key !== null && copiedIds.has(row.key)),\n\t\t\t);\n\t\t\tconst createdAt = Date.now();\n\t\t\tconst metadata = options.metadata ?? sourceMetadata.metadata;\n\t\t\tlet lease: WriterLease;\n\n\t\t\ttry {\n\t\t\t\tlease = db.transaction(() => {\n\t\t\t\t\tinsertSessionRow(db, {\n\t\t\t\t\t\tid,\n\t\t\t\t\t\tcreatedAt,\n\t\t\t\t\t\tcwd: options.cwd,\n\t\t\t\t\t\tparentSessionId: options.parentSessionId ?? source.id,\n\t\t\t\t\t\tmetadata,\n\t\t\t\t\t});\n\t\t\t\t\tcreateSequence(db, id);\n\t\t\t\t\tcreateStats(db, id, entries.filter((entry) => entry.type === \"message\").length);\n\n\t\t\t\t\tlet nextSeq = 1;\n\t\t\t\t\tconst allocateSeq = () => nextSeq++;\n\t\t\t\t\tfor (const entry of entries) {\n\t\t\t\t\t\tinsertEntryRow(db, id, {\n\t\t\t\t\t\t\tseq: allocateSeq(),\n\t\t\t\t\t\t\tid: entry.id,\n\t\t\t\t\t\t\tparentId: entry.parent_id,\n\t\t\t\t\t\t\ttype: entry.type,\n\t\t\t\t\t\t\ttimestamp: entry.timestamp,\n\t\t\t\t\t\t\tpayload: entry.payload,\n\t\t\t\t\t\t});\n\t\t\t\t\t}\n\n\t\t\t\t\tif (options.scope === \"tree\") {\n\t\t\t\t\t\tfor (const lane of lanes) insertLane(db, id, allocateSeq(), lane.lane, lane.leafId);\n\t\t\t\t\t} else {\n\t\t\t\t\t\tcreateInitialLane(db, id, \"main\", branchForkTargetId);\n\t\t\t\t\t}\n\n\t\t\t\t\tif (latestName?.value !== undefined && latestName.value !== null) {\n\t\t\t\t\t\tappendFact(db, id, allocateSeq(), \"name\", null, latestName.value);\n\t\t\t\t\t}\n\t\t\t\t\tfor (const label of labelsToCopy) appendFact(db, id, allocateSeq(), \"label\", label.key, label.value);\n\n\t\t\t\t\tsetNextSequence(db, id, nextSeq);\n\t\t\t\t\tfor (const tip of branchTips) buildCachedBranch(db, id, tip);\n\t\t\t\t\treturn claimWriterLease(db, id, this.leaseOptions);\n\t\t\t\t});\n\t\t\t} catch (error) {\n\t\t\t\tif (error instanceof SessionError) throw error;\n\t\t\t\tthrow new SessionError(\n\t\t\t\t\t\"storage\",\n\t\t\t\t\t`Failed to fork SQLite session ${id}`,\n\t\t\t\t\terror instanceof Error ? error : undefined,\n\t\t\t\t);\n\t\t\t}\n\n\t\t\tconst row = requireSessionRow(db, id);\n\t\t\treturn this.sessionFromLease(db, decodeSessionMetadata(row, path), lease);\n\t\t});\n\t}\n\n\tasync close(): Promise<void> {\n\t\tawait this.operations.drain();\n\t\tfor (const storage of [...this.activeStorages]) await storage.release();\n\t\tif (this.database) this.database.close();\n\t\tthis.database = undefined;\n\t\tthis.databasePromise = undefined;\n\t}\n\n\tasync [Symbol.asyncDispose](): Promise<void> {\n\t\tawait this.close();\n\t}\n\n\tprivate async getDatabasePath(): Promise<string> {\n\t\tthis.databasePath ??= resultOrThrow(\n\t\t\tawait this.options.env.absolutePath(this.options.databasePath),\n\t\t\t`Failed to resolve SQLite sessions database ${this.options.databasePath}`,\n\t\t);\n\t\treturn this.databasePath;\n\t}\n\n\tprivate async getDatabase(): Promise<SqliteDatabase> {\n\t\tif (!this.databasePromise) this.databasePromise = this.openDatabase();\n\t\tthis.database = await this.databasePromise;\n\t\treturn this.database;\n\t}\n\n\tprivate async openDatabase(): Promise<SqliteDatabase> {\n\t\tconst path = await this.getDatabasePath();\n\t\tresultOrThrow(\n\t\t\tawait this.options.env.createDir(getParentPath(path), { recursive: true }),\n\t\t\t`Failed to create SQLite sessions directory ${path}`,\n\t\t);\n\t\tconst db = await this.options.sqlite.open(path);\n\t\ttry {\n\t\t\tconfigureSqliteDatabase(db);\n\t\t\tawait applyMigrations(db);\n\t\t\treturn db;\n\t\t} catch (error) {\n\t\t\tdb.close();\n\t\t\tthrow error;\n\t\t}\n\t}\n}\n"]}
@@ -2,6 +2,7 @@ import { Session, SessionError, } from "@earendil-works/pi-agent-core";
2
2
  import { uuidv7 } from "@earendil-works/pi-ai";
3
3
  import { appendEntryToBranchCache, buildCachedBranch, deleteBranchCache, rebuildBranchCache } from "./branch-cache.js";
4
4
  import { applyMigrations } from "./migrations.js";
5
+ import { sql } from "./sql.js";
5
6
  import { queryCachedBranchRows, readCachedBranch } from "./storage/branch-entries.js";
6
7
  import { readBranchTipIds } from "./storage/branch-tips.js";
7
8
  import { deleteEntryRows, entryPayload, idExistsInEntries, insertEntryRow, readEntryRow, readEntryRows, } from "./storage/entries.js";
@@ -63,15 +64,9 @@ function getParentPath(path) {
63
64
  return normalized.slice(0, lastSlash);
64
65
  }
65
66
  function configureSqliteDatabase(db) {
66
- db.exec("PRAGMA journal_mode=WAL");
67
- db.exec("PRAGMA synchronous=FULL");
68
- db.exec("PRAGMA busy_timeout=5000");
69
- }
70
- function timestampToText(timestamp) {
71
- return new Date(timestamp).toISOString();
72
- }
73
- function timestampFromText(timestamp) {
74
- return Date.parse(timestamp);
67
+ sql `PRAGMA journal_mode=WAL`.exec(db);
68
+ sql `PRAGMA synchronous=FULL`.exec(db);
69
+ sql `PRAGMA busy_timeout=5000`.exec(db);
75
70
  }
76
71
  function entryRowFromCached(row) {
77
72
  return { ...row, seq: row.entry_seq, type: row.type };
@@ -86,10 +81,7 @@ function readObjectPayload(row) {
86
81
  function decodeEntry(row) {
87
82
  try {
88
83
  const payload = readObjectPayload(row);
89
- const timestamp = timestampFromText(row.timestamp);
90
- if (!Number.isFinite(timestamp))
91
- throw new Error(`Invalid timestamp ${row.timestamp}`);
92
- const base = { id: row.id, seq: row.seq, parentId: row.parent_id, timestamp };
84
+ const base = { id: row.id, seq: row.seq, parentId: row.parent_id, timestamp: row.timestamp };
93
85
  switch (row.type) {
94
86
  case "message":
95
87
  if (typeof payload.message !== "object" || payload.message === null)
@@ -170,13 +162,10 @@ function recordOpKind(record) {
170
162
  }
171
163
  function decodeRecord(row) {
172
164
  try {
173
- const timestamp = timestampFromText(row.timestamp);
174
- if (!Number.isFinite(timestamp))
175
- throw new Error(`Invalid timestamp ${row.timestamp}`);
176
165
  return {
177
166
  ...JSON.parse(row.payload),
178
167
  seq: row.seq,
179
- timestamp,
168
+ timestamp: row.timestamp,
180
169
  };
181
170
  }
182
171
  catch (error) {
@@ -184,10 +173,14 @@ function decodeRecord(row) {
184
173
  }
185
174
  }
186
175
  function validateCachedBranchRows(rows, query) {
187
- if (rows.length === 0)
176
+ if (rows.length === 0 || query.type !== undefined || query.customType !== undefined)
188
177
  return;
189
178
  const path = [...rows].sort((left, right) => left.entry_seq - right.entry_seq);
190
- if (query.stopAtId === undefined && query.stopAtType === undefined && path[0]?.parent_id !== null) {
179
+ const shouldIncludeRoot = query.stopAtId === undefined &&
180
+ query.stopAtType === undefined &&
181
+ query.cursor === undefined &&
182
+ (query.order === "oldestFirst" || query.limit === undefined);
183
+ if (shouldIncludeRoot && path[0]?.parent_id !== null) {
191
184
  throw new SessionError("invalid_entry", `Entry ${path[0]?.parent_id} not found`);
192
185
  }
193
186
  for (let index = 1; index < path.length; index++) {
@@ -338,7 +331,7 @@ class SqliteSessionStorage {
338
331
  id: committed.id,
339
332
  parentId: committed.parentId,
340
333
  type: committed.type,
341
- timestamp: timestampToText(committed.timestamp),
334
+ timestamp: committed.timestamp,
342
335
  payload: JSON.stringify(entryPayload(committed)),
343
336
  });
344
337
  setLaneLeaf(this.db, this.metadata.id, lane, committed.id);
@@ -367,7 +360,7 @@ class SqliteSessionStorage {
367
360
  runId: recordRunId(record),
368
361
  type: record.type,
369
362
  opKind: recordOpKind(record),
370
- timestamp: timestampToText(committed.timestamp),
363
+ timestamp: committed.timestamp,
371
364
  payload: JSON.stringify(record),
372
365
  });
373
366
  if (record.type === "operation_finished") {
@@ -384,7 +377,14 @@ class SqliteSessionStorage {
384
377
  return row ? decodeEntry(row) : undefined;
385
378
  }
386
379
  async findEntries(query = {}) {
387
- const rows = readEntryRows(this.db, this.metadata.id, { order: query.order });
380
+ const sqlType = query.type ?? (query.customType === undefined ? undefined : "custom");
381
+ const sqlLimit = query.customType === undefined ? query.limit : undefined;
382
+ const rows = readEntryRows(this.db, this.metadata.id, {
383
+ cursor: query.cursor,
384
+ limit: sqlLimit,
385
+ order: query.order,
386
+ type: sqlType,
387
+ });
388
388
  const entries = rows.map(decodeEntry).filter((entry) => matchesEntryQuery(entry, query));
389
389
  return query.limit === undefined ? entries : entries.slice(0, query.limit);
390
390
  }
@@ -419,32 +419,46 @@ class SqliteSessionStorage {
419
419
  }
420
420
  async getLog(options = {}) {
421
421
  const afterSeq = options.afterSeq ?? 0;
422
- const entryRows = readEntryRows(this.db, this.metadata.id, { afterSeq, order: "oldestFirst" });
423
- const recordRows = readRecordRows(this.db, this.metadata.id, { afterSeq });
424
- const laneRows = readLaneMoveRows(this.db, this.metadata.id, { afterSeq });
425
- const factRows = readFactRows(this.db, this.metadata.id, { afterSeq });
426
- const log = [
427
- ...entryRows.map((row) => ({ kind: "entry", seq: row.seq, entry: decodeEntry(row) })),
428
- ...recordRows.map((row) => ({ kind: "record", seq: row.seq, record: decodeRecord(row) })),
429
- ...laneRows.map((row) => ({ kind: "lane", seq: row.seq, lane: row.lane, leafId: row.leaf_id })),
430
- ...factRows.map((row) => {
431
- if (row.kind === "name")
422
+ const limit = options.limit;
423
+ const entryRows = readEntryRows(this.db, this.metadata.id, { afterSeq, order: "oldestFirst", limit });
424
+ const recordRows = readRecordRows(this.db, this.metadata.id, { afterSeq, order: "oldestFirst", limit });
425
+ const laneRows = readLaneMoveRows(this.db, this.metadata.id, { afterSeq, limit });
426
+ const factRows = readFactRows(this.db, this.metadata.id, { afterSeq, limit });
427
+ const logRows = [
428
+ ...entryRows.map((row) => ({
429
+ seq: row.seq,
430
+ decode: () => ({ kind: "entry", seq: row.seq, entry: decodeEntry(row) }),
431
+ })),
432
+ ...recordRows.map((row) => ({
433
+ seq: row.seq,
434
+ decode: () => ({ kind: "record", seq: row.seq, record: decodeRecord(row) }),
435
+ })),
436
+ ...laneRows.map((row) => ({
437
+ seq: row.seq,
438
+ decode: () => ({ kind: "lane", seq: row.seq, lane: row.lane, leafId: row.leaf_id }),
439
+ })),
440
+ ...factRows.map((row) => ({
441
+ seq: row.seq,
442
+ decode: () => {
443
+ if (row.kind === "name")
444
+ return {
445
+ kind: "fact",
446
+ seq: row.seq,
447
+ fact: "name",
448
+ name: row.value === null ? undefined : JSON.parse(row.value),
449
+ };
432
450
  return {
433
451
  kind: "fact",
434
452
  seq: row.seq,
435
- fact: "name",
436
- name: JSON.parse(row.value ?? "null"),
453
+ fact: "label",
454
+ targetId: row.key ?? "",
455
+ label: row.value === null ? undefined : JSON.parse(row.value),
437
456
  };
438
- return {
439
- kind: "fact",
440
- seq: row.seq,
441
- fact: "label",
442
- targetId: row.key ?? "",
443
- label: row.value === null ? undefined : JSON.parse(row.value),
444
- };
445
- }),
457
+ },
458
+ })),
446
459
  ].sort((left, right) => left.seq - right.seq);
447
- return options.limit === undefined ? log : log.slice(0, options.limit);
460
+ const selectedRows = options.limit === undefined ? logRows : logRows.slice(0, options.limit);
461
+ return selectedRows.map((row) => row.decode());
448
462
  }
449
463
  async getName() {
450
464
  const row = readLatestFact(this.db, this.metadata.id, "name", null);
@@ -453,7 +467,7 @@ class SqliteSessionStorage {
453
467
  async setName(name) {
454
468
  return this.enqueueWrite(() => {
455
469
  const seq = getNextSequence(this.db, this.metadata.id);
456
- appendFact(this.db, this.metadata.id, seq, "name", null, JSON.stringify(name));
470
+ appendFact(this.db, this.metadata.id, seq, "name", null, name === undefined ? null : JSON.stringify(name));
457
471
  advanceSequence(this.db, this.metadata.id, seq);
458
472
  });
459
473
  }
@@ -535,7 +549,7 @@ export class SqliteSessionRepository {
535
549
  const lease = db.transaction(() => {
536
550
  insertSessionRow(db, {
537
551
  id,
538
- createdAt: timestampToText(createdAt),
552
+ createdAt,
539
553
  cwd: options.cwd,
540
554
  parentSessionId: options.parentSessionId,
541
555
  metadata: options.metadata,
@@ -650,7 +664,7 @@ export class SqliteSessionRepository {
650
664
  lease = db.transaction(() => {
651
665
  insertSessionRow(db, {
652
666
  id,
653
- createdAt: timestampToText(createdAt),
667
+ createdAt,
654
668
  cwd: options.cwd,
655
669
  parentSessionId: options.parentSessionId ?? source.id,
656
670
  metadata,