@m4ike1/ion-session-backend-sqlite-node 0.1.0

package/dist/sqlite/session/branch-entries.js

@@ -0,0 +1,213 @@
+import { joinSqlFragments, sql } from "../sql.js";
+import { decodeEntryRow } from "./entries.js";
+function readBranchMembership(db, sessionId, entryId) {
+    const row = sql `SELECT b.branch_id, b.entry_seq
+		FROM branch_entries b
+		JOIN branch_meta m ON m.session_id = b.session_id AND m.branch_id = b.branch_id
+		WHERE b.session_id = ${sessionId}
+			AND b.entry_id = ${entryId}
+			AND ((m.base_seq IS NULL AND b.entry_seq > 0) OR (m.base_seq IS NOT NULL AND b.entry_seq > m.base_seq))
+			AND b.entry_seq <= m.tip_seq
+		ORDER BY m.tip_seq DESC, b.branch_id
+		LIMIT 1`.get(db);
+    if (row === undefined)
+        throw new Error(`Branch cache missing entry ${entryId}`);
+    return row;
+}
+function readBranchMeta(db, sessionId, branchId) {
+    const row = sql `SELECT branch_id, tip_entry_id, tip_seq, base_branch_id, base_seq
+		FROM branch_meta
+		WHERE session_id = ${sessionId} AND branch_id = ${branchId}`.get(db);
+    if (row === undefined)
+        throw new Error(`Branch metadata missing for branch ${branchId}`);
+    return row;
+}
+function insertBranchEntry(db, sessionId, branchId, entry) {
+    sql `INSERT INTO branch_entries (session_id, branch_id, entry_id, entry_seq, entry_type)
+		VALUES (${sessionId}, ${branchId}, ${entry.id}, ${entry.seq}, ${entry.type})`.run(db);
+}
+function readBranchTipForParent(db, sessionId, parentId) {
+    return sql `SELECT branch_id
+		FROM branch_meta
+		WHERE session_id = ${sessionId} AND tip_entry_id = ${parentId}`.get(db);
+}
+function createRootBranchForEntry(db, sessionId, entry) {
+    sql `INSERT INTO branch_meta (session_id, branch_id, tip_entry_id, tip_seq, base_branch_id, base_seq)
+		VALUES (${sessionId}, ${entry.id}, ${entry.id}, ${entry.seq}, ${null}, ${null})`.run(db);
+    insertBranchEntry(db, sessionId, entry.id, entry);
+}
+function appendEntryToExistingBranch(db, sessionId, branchId, entry) {
+    insertBranchEntry(db, sessionId, branchId, entry);
+    const result = sql `UPDATE branch_meta
+		SET tip_entry_id = ${entry.id}, tip_seq = ${entry.seq}
+		WHERE session_id = ${sessionId} AND branch_id = ${branchId}`.run(db);
+    if (result.changes !== 1)
+        throw new Error(`Expected to update branch ${branchId}, updated ${result.changes}`);
+}
+function readBranchSegmentsNewestFirst(db, sessionId, start) {
+    let { branch_id: branchId, entry_seq: upperSeq } = readBranchMembership(db, sessionId, start);
+    const segments = [];
+    while (true) {
+        const meta = readBranchMeta(db, sessionId, branchId);
+        const lowerSeq = meta.base_seq ?? 0;
+        segments.push({ branchId, lowerSeq, upperSeq });
+        if (meta.base_branch_id === null)
+            break;
+        if (meta.base_seq === null)
+            throw new Error(`Branch ${branchId} has base branch without base_seq`);
+        branchId = meta.base_branch_id;
+        upperSeq = meta.base_seq;
+    }
+    return segments;
+}
+function readNewestCompactionBoundary(db, sessionId, segmentsNewestFirst) {
+    for (const segment of segmentsNewestFirst) {
+        const row = sql `SELECT MAX(entry_seq) AS entry_seq
+			FROM branch_entries
+			WHERE session_id = ${sessionId}
+				AND branch_id = ${segment.branchId}
+				AND entry_seq > ${segment.lowerSeq}
+				AND entry_seq <= ${segment.upperSeq}
+				AND entry_type = ${"compaction"}`.get(db);
+        if (row?.entry_seq !== null && row?.entry_seq !== undefined)
+            return { branchId: segment.branchId, seq: row.entry_seq };
+    }
+    return undefined;
+}
+function copyBranchEntriesAfterSeqThroughParent(db, sessionId, targetBranchId, segmentsNewestFirst, afterSeq) {
+    for (const segment of [...segmentsNewestFirst].reverse()) {
+        const lowerSeq = Math.max(segment.lowerSeq, afterSeq);
+        if (segment.upperSeq <= lowerSeq)
+            continue;
+        sql `INSERT INTO branch_entries (session_id, branch_id, entry_id, entry_seq, entry_type)
+			SELECT ${sessionId}, ${targetBranchId}, entry_id, entry_seq, entry_type
+			FROM branch_entries
+			WHERE session_id = ${sessionId}
+				AND branch_id = ${segment.branchId}
+				AND entry_seq > ${lowerSeq}
+				AND entry_seq <= ${segment.upperSeq}`.run(db);
+    }
+}
+function createDivergentBranchForEntry(db, sessionId, entry) {
+    if (entry.parentId === null)
+        throw new Error("Root entries do not create divergent branches");
+    const segmentsNewestFirst = readBranchSegmentsNewestFirst(db, sessionId, entry.parentId);
+    const compaction = readNewestCompactionBoundary(db, sessionId, segmentsNewestFirst);
+    const branchId = entry.id;
+    // A null base means this segment stores its own root-through-parent prefix.
+    sql `INSERT INTO branch_meta (session_id, branch_id, tip_entry_id, tip_seq, base_branch_id, base_seq)
+		VALUES (${sessionId}, ${branchId}, ${entry.id}, ${entry.seq}, ${compaction?.branchId ?? null}, ${compaction?.seq ?? null})`.run(db);
+    copyBranchEntriesAfterSeqThroughParent(db, sessionId, branchId, segmentsNewestFirst, compaction?.seq ?? 0);
+    insertBranchEntry(db, sessionId, branchId, entry);
+}
+export function appendEntryToBranchIndex(db, sessionId, entry) {
+    if (entry.parentId === null) {
+        createRootBranchForEntry(db, sessionId, entry);
+        return;
+    }
+    const branch = readBranchTipForParent(db, sessionId, entry.parentId);
+    if (branch === undefined) {
+        createDivergentBranchForEntry(db, sessionId, entry);
+        return;
+    }
+    appendEntryToExistingBranch(db, sessionId, branch.branch_id, entry);
+}
+function stopPredicates(query) {
+    const predicates = [];
+    if (query.stopAtType !== undefined)
+        predicates.push(sql `b.entry_type = ${query.stopAtType}`);
+    if (query.stopAtId !== undefined)
+        predicates.push(sql `b.entry_id = ${query.stopAtId}`);
+    return predicates;
+}
+function readStopSeq(db, sessionId, segment, query, oldestFirst) {
+    const stop = stopPredicates(query);
+    if (stop.length === 0)
+        return undefined;
+    const aggregate = oldestFirst ? sql `MIN(b.entry_seq)` : sql `MAX(b.entry_seq)`;
+    const row = sql `SELECT ${aggregate} AS stop_seq
+		FROM branch_entries b
+		WHERE b.session_id = ${sessionId}
+			AND b.branch_id = ${segment.branchId}
+			AND b.entry_seq > ${segment.lowerSeq}
+			AND b.entry_seq <= ${segment.upperSeq}
+			AND (${joinSqlFragments(stop, " OR ")})`.get(db);
+    return row?.stop_seq ?? undefined;
+}
+function branchScanPredicates(sessionId, segment, query, oldestFirst, stopSeq) {
+    const predicates = [
+        sql `b.session_id = ${sessionId}`,
+        sql `b.branch_id = ${segment.branchId}`,
+        sql `b.entry_seq > ${segment.lowerSeq}`,
+        sql `b.entry_seq <= ${segment.upperSeq}`,
+        sql `e.session_id = b.session_id`,
+    ];
+    if (stopSeq !== undefined)
+        predicates.push(oldestFirst ? sql `b.entry_seq <= ${stopSeq}` : sql `b.entry_seq >= ${stopSeq}`);
+    if (query.type !== undefined)
+        predicates.push(sql `b.entry_type = ${query.type}`);
+    if (query.customType !== undefined)
+        predicates.push(sql `e.custom_type = ${query.customType}`);
+    if (query.cursor !== undefined) {
+        predicates.push(oldestFirst ? sql `b.entry_seq > ${query.cursor.seq}` : sql `b.entry_seq < ${query.cursor.seq}`);
+    }
+    return predicates;
+}
+function limitSql(limit) {
+    return limit === undefined ? sql `` : sql `LIMIT ${Math.max(0, limit)}`;
+}
+function scanEntrySegmentRows(db, sessionId, segment, query, oldestFirst, stopSeq, limit) {
+    const predicates = branchScanPredicates(sessionId, segment, query, oldestFirst, stopSeq);
+    const order = oldestFirst ? sql `ASC` : sql `DESC`;
+    return sql `SELECT e.id, e.parent_id, e.seq, e.type, e.custom_type, e.timestamp, e.payload
+		FROM branch_entries b
+		CROSS JOIN entries e ON e.session_id = b.session_id AND e.id = b.entry_id
+		WHERE ${joinSqlFragments(predicates, " AND ")}
+		ORDER BY b.entry_seq ${order} ${limitSql(limit)}`.all(db);
+}
+function decodeEntryStructureRow(row) {
+    return {
+        id: row.id,
+        parentId: row.parent_id,
+        seq: row.seq,
+        timestamp: row.timestamp,
+        type: row.type,
+        ...(row.custom_type === null ? {} : { customType: row.custom_type }),
+    };
+}
+function scanStructureSegmentRows(db, sessionId, segment, query, oldestFirst, stopSeq, limit) {
+    const predicates = branchScanPredicates(sessionId, segment, query, oldestFirst, stopSeq);
+    const order = oldestFirst ? sql `ASC` : sql `DESC`;
+    const rows = sql `SELECT e.id, e.parent_id, e.seq, e.type, e.custom_type, e.timestamp
+		FROM branch_entries b
+		CROSS JOIN entries e ON e.session_id = b.session_id AND e.id = b.entry_id
+		WHERE ${joinSqlFragments(predicates, " AND ")}
+		ORDER BY b.entry_seq ${order} ${limitSql(limit)}`.all(db);
+    return rows.map(decodeEntryStructureRow);
+}
+function scanBranchSegments(db, sessionId, query, readSegment) {
+    const oldestFirst = query.order === "oldestFirst";
+    const segmentsNewestFirst = readBranchSegmentsNewestFirst(db, sessionId, query.start);
+    const segments = oldestFirst ? [...segmentsNewestFirst].reverse() : segmentsNewestFirst;
+    const limit = query.limit === undefined ? undefined : Math.max(0, query.limit);
+    if (limit === 0)
+        return [];
+    const rows = [];
+    for (const segment of segments) {
+        const remaining = limit === undefined ? undefined : limit - rows.length;
+        if (remaining !== undefined && remaining <= 0)
+            break;
+        const stopSeq = readStopSeq(db, sessionId, segment, query, oldestFirst);
+        rows.push(...readSegment(db, sessionId, segment, query, oldestFirst, stopSeq, remaining));
+        if (stopSeq !== undefined)
+            break;
+    }
+    return rows;
+}
+export function scanBranchEntries(db, sessionId, query) {
+    return scanBranchSegments(db, sessionId, query, scanEntrySegmentRows).map(decodeEntryRow);
+}
+export function scanBranchEntryStructures(db, sessionId, query) {
+    return scanBranchSegments(db, sessionId, query, scanStructureSegmentRows);
+}
+//# sourceMappingURL=branch-entries.js.map

package/dist/sqlite/session/branch-entries.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"branch-entries.js","sourceRoot":"","sources":["../../../src/sqlite/session/branch-entries.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,gBAAgB,EAAiB,GAAG,EAAE,MAAM,WAAW,CAAC;AAEjE,OAAO,EAAE,cAAc,EAAiB,MAAM,cAAc,CAAC;AA+C7D,SAAS,oBAAoB,CAAC,EAAkB,EAAE,SAAiB,EAAE,OAAe,EAAuB;IAC1G,MAAM,GAAG,GAAG,GAAG,CAAA;;;yBAGS,SAAS;sBACZ,OAAO;;;;UAInB,CAAC,GAAG,CAAsB,EAAE,CAAC,CAAC;IACvC,IAAI,GAAG,KAAK,SAAS;QAAE,MAAM,IAAI,KAAK,CAAC,8BAA8B,OAAO,EAAE,CAAC,CAAC;IAChF,OAAO,GAAG,CAAC;AAAA,CACX;AAED,SAAS,cAAc,CAAC,EAAkB,EAAE,SAAiB,EAAE,QAAgB,EAAiB;IAC/F,MAAM,GAAG,GAAG,GAAG,CAAA;;uBAEO,SAAS,oBAAoB,QAAQ,EAAE,CAAC,GAAG,CAAgB,EAAE,CAAC,CAAC;IACrF,IAAI,GAAG,KAAK,SAAS;QAAE,MAAM,IAAI,KAAK,CAAC,sCAAsC,QAAQ,EAAE,CAAC,CAAC;IACzF,OAAO,GAAG,CAAC;AAAA,CACX;AAED,SAAS,iBAAiB,CAAC,EAAkB,EAAE,SAAiB,EAAE,QAAgB,EAAE,KAAY,EAAQ;IACvG,GAAG,CAAA;YACQ,SAAS,KAAK,QAAQ,KAAK,KAAK,CAAC,EAAE,KAAK,KAAK,CAAC,GAAG,KAAK,KAAK,CAAC,IAAI,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;AAAA,CACvF;AAED,SAAS,sBAAsB,CAAC,EAAkB,EAAE,SAAiB,EAAE,QAAgB,EAA4B;IAClH,OAAO,GAAG,CAAA;;uBAEY,SAAS,uBAAuB,QAAQ,EAAE,CAAC,GAAG,CAAe,EAAE,CAAC,CAAC;AAAA,CACvF;AAED,SAAS,wBAAwB,CAAC,EAAkB,EAAE,SAAiB,EAAE,KAAY,EAAQ;IAC5F,GAAG,CAAA;YACQ,SAAS,KAAK,KAAK,CAAC,EAAE,KAAK,KAAK,CAAC,EAAE,KAAK,KAAK,CAAC,GAAG,KAAK,IAAI,KAAK,IAAI,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAC1F,iBAAiB,CAAC,EAAE,EAAE,SAAS,EAAE,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;AAAA,CAClD;AAED,SAAS,2BAA2B,CAAC,EAAkB,EAAE,SAAiB,EAAE,QAAgB,EAAE,KAAY,EAAQ;IACjH,iBAAiB,CAAC,EAAE,EAAE,SAAS,EAAE,QAAQ,EAAE,KAAK,CAAC,CAAC;IAClD,MAAM,MAAM,GAAG,GAAG,CAAA;uBACI,KAAK,CAAC,EAAE,eAAe,KAAK,CAAC,GAAG;uBAChC,SAAS,oBAAoB,QAAQ,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IACtE,IAAI,MAAM,CAAC,OAAO,KAAK,CAAC;QAAE,MAAM,IAAI,KAAK,CAAC,6BAA6B,QAAQ,aAAa,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC;AAAA,CAC9G;AAED,SAAS,6BAA6B,CAAC,EAAkB,EAAE,SAAiB,EAAE,KAAa,EAAmB;IAC7G,IAAI,EAAE,SAAS,EAAE,QAAQ,EAAE,SAAS,EAAE,QAAQ,EAAE,GAAG,oBAAoB,CAAC,EAAE,EAAE,SAAS,EAAE,KAAK,CAAC,CAAC;IAC9F,MAAM,QAAQ,GAAoB,EAAE,CAAC;IACrC,OAAO,IAAI,EAAE,CAAC;QACb,MAAM,IAAI,GAAG,cAAc,CAAC,EAAE,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC;QACrD,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,IAAI,CAAC,CAAC;QACpC,QAAQ,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC,CAAC;QAChD,IAAI,IAAI,CAAC,cAAc,KAAK,IAAI;YAAE,MAAM;QACxC,IAAI,IAAI,CAAC,QAAQ,KAAK,IAAI;YAAE,MAAM,IAAI,KAAK,CAAC,UAAU,QAAQ,mCAAmC,CAAC,CAAC;QACnG,QAAQ,GAAG,IAAI,CAAC,cAAc,CAAC;QAC/B,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC;IAC1B,CAAC;IACD,OAAO,QAAQ,CAAC;AAAA,CAChB;AAED,SAAS,4BAA4B,CACpC,EAAkB,EAClB,SAAiB,EACjB,mBAA6C,EACZ;IACjC,KAAK,MAAM,OAAO,IAAI,mBAAmB,EAAE,CAAC;QAC3C,MAAM,GAAG,GAAG,GAAG,CAAA;;wBAEO,SAAS;sBACX,OAAO,CAAC,QAAQ;sBAChB,OAAO,CAAC,QAAQ;uBACf,OAAO,CAAC,QAAQ;uBAChB,YAAY,EAAE,CAAC,GAAG,CAAwB,EAAE,CAAC,CAAC;QACnE,IAAI,GAAG,EAAE,SAAS,KAAK,IAAI,IAAI,GAAG,EAAE,SAAS,KAAK,SAAS;YAC1D,OAAO,EAAE,QAAQ,EAAE,OAAO,CAAC,QAAQ,EAAE,GAAG,EAAE,GAAG,CAAC,SAAS,EAAE,CAAC;IAC5D,CAAC;IACD,OAAO,SAAS,CAAC;AAAA,CACjB;AAED,SAAS,sCAAsC,CAC9C,EAAkB,EAClB,SAAiB,EACjB,cAAsB,EACtB,mBAA6C,EAC7C,QAAgB,EACT;IACP,KAAK,MAAM,OAAO,IAAI,CAAC,GAAG,mBAAmB,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC;QAC1D,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;QACtD,IAAI,OAAO,CAAC,QAAQ,IAAI,QAAQ;YAAE,SAAS;QAC3C,GAAG,CAAA;YACO,SAAS,KAAK,cAAc;;wBAEhB,SAAS;sBACX,OAAO,CAAC,QAAQ;sBAChB,QAAQ;uBACP,OAAO,CAAC,QAAQ,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IACjD,CAAC;AAAA,CACD;AAED,SAAS,6BAA6B,CAAC,EAAkB,EAAE,SAAiB,EAAE,KAAY,EAAQ;IACjG,IAAI,KAAK,CAAC,QAAQ,KAAK,IAAI;QAAE,MAAM,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAC;IAC9F,MAAM,mBAAmB,GAAG,6BAA6B,CAAC,EAAE,EAAE,SAAS,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;IACzF,MAAM,UAAU,GAAG,4BAA4B,CAAC,EAAE,EAAE,SAAS,EAAE,mBAAmB,CAAC,CAAC;IACpF,MAAM,QAAQ,GAAG,KAAK,CAAC,EAAE,CAAC;IAC1B,4EAA4E;IAC5E,GAAG,CAAA;YACQ,SAAS,KAAK,QAAQ,KAAK,KAAK,CAAC,EAAE,KAAK,KAAK,CAAC,GAAG,KAAK,UAAU,EAAE,QAAQ,IAAI,IAAI,KAAK,UAAU,EAAE,GAAG,IAAI,IAAI,GAAG,CAAC,GAAG,CAC/H,EAAE,CACF,CAAC;IACF,sCAAsC,CAAC,EAAE,EAAE,SAAS,EAAE,QAAQ,EAAE,mBAAmB,EAAE,UAAU,EAAE,GAAG,IAAI,CAAC,CAAC,CAAC;IAC3G,iBAAiB,CAAC,EAAE,EAAE,SAAS,EAAE,QAAQ,EAAE,KAAK,CAAC,CAAC;AAAA,CAClD;AAED,MAAM,UAAU,wBAAwB,CAAC,EAAkB,EAAE,SAAiB,EAAE,KAAY,EAAQ;IACnG,IAAI,KAAK,CAAC,QAAQ,KAAK,IAAI,EAAE,CAAC;QAC7B,wBAAwB,CAAC,EAAE,EAAE,SAAS,EAAE,KAAK,CAAC,CAAC;QAC/C,OAAO;IACR,CAAC;IAED,MAAM,MAAM,GAAG,sBAAsB,CAAC,EAAE,EAAE,SAAS,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;IACrE,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;QAC1B,6BAA6B,CAAC,EAAE,EAAE,SAAS,EAAE,KAAK,CAAC,CAAC;QACpD,OAAO;IACR,CAAC;IACD,2BAA2B,CAAC,EAAE,EAAE,SAAS,EAAE,MAAM,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;AAAA,CACpE;AAED,SAAS,cAAc,CAAC,KAAwB,EAAc;IAC7D,MAAM,UAAU,GAAe,EAAE,CAAC;IAClC,IAAI,KAAK,CAAC,UAAU,KAAK,SAAS;QAAE,UAAU,CAAC,IAAI,CAAC,GAAG,CAAA,kBAAkB,KAAK,CAAC,UAAU,EAAE,CAAC,CAAC;IAC7F,IAAI,KAAK,CAAC,QAAQ,KAAK,SAAS;QAAE,UAAU,CAAC,IAAI,CAAC,GAAG,CAAA,gBAAgB,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC;IACvF,OAAO,UAAU,CAAC;AAAA,CAClB;AAED,SAAS,WAAW,CACnB,EAAkB,EAClB,SAAiB,EACjB,OAAsB,EACtB,KAAwB,EACxB,WAAoB,EACC;IACrB,MAAM,IAAI,GAAG,cAAc,CAAC,KAAK,CAAC,CAAC;IACnC,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,SAAS,CAAC;IACxC,MAAM,SAAS,GAAG,WAAW,CAAC,CAAC,CAAC,GAAG,CAAA,kBAAkB,CAAC,CAAC,CAAC,GAAG,CAAA,kBAAkB,CAAC;IAC9E,MAAM,GAAG,GAAG,GAAG,CAAA,UAAU,SAAS;;yBAEV,SAAS;uBACX,OAAO,CAAC,QAAQ;uBAChB,OAAO,CAAC,QAAQ;wBACf,OAAO,CAAC,QAAQ;UAC9B,gBAAgB,CAAC,IAAI,EAAE,MAAM,CAAC,GAAG,CAAC,GAAG,CAAa,EAAE,CAAC,CAAC;IAC/D,OAAO,GAAG,EAAE,QAAQ,IAAI,SAAS,CAAC;AAAA,CAClC;AAED,SAAS,oBAAoB,CAC5B,SAAiB,EACjB,OAAsB,EACtB,KAAwB,EACxB,WAAoB,EACpB,OAA2B,EACd;IACb,MAAM,UAAU,GAAe;QAC9B,GAAG,CAAA,kBAAkB,SAAS,EAAE;QAChC,GAAG,CAAA,iBAAiB,OAAO,CAAC,QAAQ,EAAE;QACtC,GAAG,CAAA,iBAAiB,OAAO,CAAC,QAAQ,EAAE;QACtC,GAAG,CAAA,kBAAkB,OAAO,CAAC,QAAQ,EAAE;QACvC,GAAG,CAAA,6BAA6B;KAChC,CAAC;IACF,IAAI,OAAO,KAAK,SAAS;QACxB,UAAU,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,GAAG,CAAA,kBAAkB,OAAO,EAAE,CAAC,CAAC,CAAC,GAAG,CAAA,kBAAkB,OAAO,EAAE,CAAC,CAAC;IAChG,IAAI,KAAK,CAAC,IAAI,KAAK,SAAS;QAAE,UAAU,CAAC,IAAI,CAAC,GAAG,CAAA,kBAAkB,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;IACjF,IAAI,KAAK,CAAC,UAAU,KAAK,SAAS;QAAE,UAAU,CAAC,IAAI,CAAC,GAAG,CAAA,mBAAmB,KAAK,CAAC,UAAU,EAAE,CAAC,CAAC;IAC9F,IAAI,KAAK,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;QAChC,UAAU,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,GAAG,CAAA,iBAAiB,KAAK,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,GAAG,CAAA,iBAAiB,KAAK,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,CAAC;IAChH,CAAC;IACD,OAAO,UAAU,CAAC;AAAA,CAClB;AAED,SAAS,QAAQ,CAAC,KAAyB,EAAY;IACtD,OAAO,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,GAAG,CAAA,EAAE,CAAC,CAAC,CAAC,GAAG,CAAA,SAAS,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,EAAE,CAAC;AAAA,CACtE;AAED,SAAS,oBAAoB,CAC5B,EAAkB,EAClB,SAAiB,EACjB,OAAsB,EACtB,KAAwB,EACxB,WAAoB,EACpB,OAA2B,EAC3B,KAAyB,EACZ;IACb,MAAM,UAAU,GAAG,oBAAoB,CAAC,SAAS,EAAE,OAAO,EAAE,KAAK,EAAE,WAAW,EAAE,OAAO,CAAC,CAAC;IACzF,MAAM,KAAK,GAAG,WAAW,CAAC,CAAC,CAAC,GAAG,CAAA,KAAK,CAAC,CAAC,CAAC,GAAG,CAAA,MAAM,CAAC;IACjD,OAAO,GAAG,CAAA;;;UAGD,gBAAgB,CAAC,UAAU,EAAE,OAAO,CAAC;yBACtB,KAAK,IAAI,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC,GAAG,CAAW,EAAE,CAAC,CAAC;AAAA,CACrE;AAED,SAAS,uBAAuB,CAAC,GAAsB,EAAkB;IACxE,OAAO;QACN,EAAE,EAAE,GAAG,CAAC,EAAE;QACV,QAAQ,EAAE,GAAG,CAAC,SAAS;QACvB,GAAG,EAAE,GAAG,CAAC,GAAG;QACZ,SAAS,EAAE,GAAG,CAAC,SAAS;QACxB,IAAI,EAAE,GAAG,CAAC,IAAI;QACd,GAAG,CAAC,GAAG,CAAC,WAAW,KAAK,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,GAAG,CAAC,WAAW,EAAE,CAAC;KACpE,CAAC;AAAA,CACF;AAED,SAAS,wBAAwB,CAChC,EAAkB,EAClB,SAAiB,EACjB,OAAsB,EACtB,KAAwB,EACxB,WAAoB,EACpB,OAA2B,EAC3B,KAAyB,EACN;IACnB,MAAM,UAAU,GAAG,oBAAoB,CAAC,SAAS,EAAE,OAAO,EAAE,KAAK,EAAE,WAAW,EAAE,OAAO,CAAC,CAAC;IACzF,MAAM,KAAK,GAAG,WAAW,CAAC,CAAC,CAAC,GAAG,CAAA,KAAK,CAAC,CAAC,CAAC,GAAG,CAAA,MAAM,CAAC;IACjD,MAAM,IAAI,GAAG,GAAG,CAAA;;;UAGP,gBAAgB,CAAC,UAAU,EAAE,OAAO,CAAC;yBACtB,KAAK,IAAI,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC,GAAG,CAAoB,EAAE,CAAC,CAAC;IAC9E,OAAO,IAAI,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;AAAA,CACzC;AAED,SAAS,kBAAkB,CAC1B,EAAkB,EAClB,SAAiB,EACjB,KAAwB,EACxB,WAQQ,EACF;IACN,MAAM,WAAW,GAAG,KAAK,CAAC,KAAK,KAAK,aAAa,CAAC;IAClD,MAAM,mBAAmB,GAAG,6BAA6B,CAAC,EAAE,EAAE,SAAS,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC;IACtF,MAAM,QAAQ,GAAG,WAAW,CAAC,CAAC,CAAC,CAAC,GAAG,mBAAmB,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,mBAAmB,CAAC;IACxF,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC;IAC/E,IAAI,KAAK,KAAK,CAAC;QAAE,OAAO,EAAE,CAAC;IAE3B,MAAM,IAAI,GAAQ,EAAE,CAAC;IACrB,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAChC,MAAM,SAAS,GAAG,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC;QACxE,IAAI,SAAS,KAAK,SAAS,IAAI,SAAS,IAAI,CAAC;YAAE,MAAM;QACrD,MAAM,OAAO,GAAG,WAAW,CAAC,EAAE,EAAE,SAAS,EAAE,OAAO,EAAE,KAAK,EAAE,WAAW,CAAC,CAAC;QACxE,IAAI,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC,EAAE,EAAE,SAAS,EAAE,OAAO,EAAE,KAAK,EAAE,WAAW,EAAE,OAAO,EAAE,SAAS,CAAC,CAAC,CAAC;QAC1F,IAAI,OAAO,KAAK,SAAS;YAAE,MAAM;IAClC,CAAC;IACD,OAAO,IAAI,CAAC;AAAA,CACZ;AAED,MAAM,UAAU,iBAAiB,CAAC,EAAkB,EAAE,SAAiB,EAAE,KAAwB,EAAW;IAC3G,OAAO,kBAAkB,CAAC,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,oBAAoB,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;AAAA,CAC1F;AAED,MAAM,UAAU,yBAAyB,CACxC,EAAkB,EAClB,SAAiB,EACjB,KAAwB,EACL;IACnB,OAAO,kBAAkB,CAAC,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,wBAAwB,CAAC,CAAC;AAAA,CAC1E","sourcesContent":["import type { Entry, EntryStructure, StorageBranchScan } from \"@m4ike1/ion-agent-core\";\nimport { joinSqlFragments, type SqlQuery, sql } from \"../sql.ts\";\nimport type { SqliteDatabase } from \"../types.ts\";\nimport { decodeEntryRow, type EntryRow } from \"./entries.ts\";\n\ninterface BranchMembershipRow {\n\tbranch_id: string;\n\tentry_seq: number;\n}\n\ninterface BranchMetaRow {\n\tbranch_id: string;\n\ttip_entry_id: string;\n\ttip_seq: number;\n\tbase_branch_id: string | null;\n\tbase_seq: number | null;\n}\n\ninterface BranchSegment {\n\tbranchId: string;\n\tlowerSeq: number;\n\tupperSeq: number;\n}\n\ninterface StopSeqRow {\n\tstop_seq: number | null;\n}\n\ninterface EntryStructureRow {\n\tid: string;\n\tparent_id: string | null;\n\tseq: number;\n\ttype: Entry[\"type\"];\n\tcustom_type: string | null;\n\ttimestamp: number;\n}\n\ninterface BranchTipRow {\n\tbranch_id: string;\n}\n\ninterface CompactionBoundary {\n\tbranchId: string;\n\tseq: number;\n}\n\ninterface CompactionBoundaryRow {\n\tentry_seq: number | null;\n}\n\nfunction readBranchMembership(db: SqliteDatabase, sessionId: string, entryId: string): BranchMembershipRow {\n\tconst row = sql`SELECT b.branch_id, b.entry_seq\n\t\tFROM branch_entries b\n\t\tJOIN branch_meta m ON m.session_id = b.session_id AND m.branch_id = b.branch_id\n\t\tWHERE b.session_id = ${sessionId}\n\t\t\tAND b.entry_id = ${entryId}\n\t\t\tAND ((m.base_seq IS NULL AND b.entry_seq > 0) OR (m.base_seq IS NOT NULL AND b.entry_seq > m.base_seq))\n\t\t\tAND b.entry_seq <= m.tip_seq\n\t\tORDER BY m.tip_seq DESC, b.branch_id\n\t\tLIMIT 1`.get<BranchMembershipRow>(db);\n\tif (row === undefined) throw new Error(`Branch cache missing entry ${entryId}`);\n\treturn row;\n}\n\nfunction readBranchMeta(db: SqliteDatabase, sessionId: string, branchId: string): BranchMetaRow {\n\tconst row = sql`SELECT branch_id, tip_entry_id, tip_seq, base_branch_id, base_seq\n\t\tFROM branch_meta\n\t\tWHERE session_id = ${sessionId} AND branch_id = ${branchId}`.get<BranchMetaRow>(db);\n\tif (row === undefined) throw new Error(`Branch metadata missing for branch ${branchId}`);\n\treturn row;\n}\n\nfunction insertBranchEntry(db: SqliteDatabase, sessionId: string, branchId: string, entry: Entry): void {\n\tsql`INSERT INTO branch_entries (session_id, branch_id, entry_id, entry_seq, entry_type)\n\t\tVALUES (${sessionId}, ${branchId}, ${entry.id}, ${entry.seq}, ${entry.type})`.run(db);\n}\n\nfunction readBranchTipForParent(db: SqliteDatabase, sessionId: string, parentId: string): BranchTipRow | undefined {\n\treturn sql`SELECT branch_id\n\t\tFROM branch_meta\n\t\tWHERE session_id = ${sessionId} AND tip_entry_id = ${parentId}`.get<BranchTipRow>(db);\n}\n\nfunction createRootBranchForEntry(db: SqliteDatabase, sessionId: string, entry: Entry): void {\n\tsql`INSERT INTO branch_meta (session_id, branch_id, tip_entry_id, tip_seq, base_branch_id, base_seq)\n\t\tVALUES (${sessionId}, ${entry.id}, ${entry.id}, ${entry.seq}, ${null}, ${null})`.run(db);\n\tinsertBranchEntry(db, sessionId, entry.id, entry);\n}\n\nfunction appendEntryToExistingBranch(db: SqliteDatabase, sessionId: string, branchId: string, entry: Entry): void {\n\tinsertBranchEntry(db, sessionId, branchId, entry);\n\tconst result = sql`UPDATE branch_meta\n\t\tSET tip_entry_id = ${entry.id}, tip_seq = ${entry.seq}\n\t\tWHERE session_id = ${sessionId} AND branch_id = ${branchId}`.run(db);\n\tif (result.changes !== 1) throw new Error(`Expected to update branch ${branchId}, updated ${result.changes}`);\n}\n\nfunction readBranchSegmentsNewestFirst(db: SqliteDatabase, sessionId: string, start: string): BranchSegment[] {\n\tlet { branch_id: branchId, entry_seq: upperSeq } = readBranchMembership(db, sessionId, start);\n\tconst segments: BranchSegment[] = [];\n\twhile (true) {\n\t\tconst meta = readBranchMeta(db, sessionId, branchId);\n\t\tconst lowerSeq = meta.base_seq ?? 0;\n\t\tsegments.push({ branchId, lowerSeq, upperSeq });\n\t\tif (meta.base_branch_id === null) break;\n\t\tif (meta.base_seq === null) throw new Error(`Branch ${branchId} has base branch without base_seq`);\n\t\tbranchId = meta.base_branch_id;\n\t\tupperSeq = meta.base_seq;\n\t}\n\treturn segments;\n}\n\nfunction readNewestCompactionBoundary(\n\tdb: SqliteDatabase,\n\tsessionId: string,\n\tsegmentsNewestFirst: readonly BranchSegment[],\n): CompactionBoundary | undefined {\n\tfor (const segment of segmentsNewestFirst) {\n\t\tconst row = sql`SELECT MAX(entry_seq) AS entry_seq\n\t\t\tFROM branch_entries\n\t\t\tWHERE session_id = ${sessionId}\n\t\t\t\tAND branch_id = ${segment.branchId}\n\t\t\t\tAND entry_seq > ${segment.lowerSeq}\n\t\t\t\tAND entry_seq <= ${segment.upperSeq}\n\t\t\t\tAND entry_type = ${\"compaction\"}`.get<CompactionBoundaryRow>(db);\n\t\tif (row?.entry_seq !== null && row?.entry_seq !== undefined)\n\t\t\treturn { branchId: segment.branchId, seq: row.entry_seq };\n\t}\n\treturn undefined;\n}\n\nfunction copyBranchEntriesAfterSeqThroughParent(\n\tdb: SqliteDatabase,\n\tsessionId: string,\n\ttargetBranchId: string,\n\tsegmentsNewestFirst: readonly BranchSegment[],\n\tafterSeq: number,\n): void {\n\tfor (const segment of [...segmentsNewestFirst].reverse()) {\n\t\tconst lowerSeq = Math.max(segment.lowerSeq, afterSeq);\n\t\tif (segment.upperSeq <= lowerSeq) continue;\n\t\tsql`INSERT INTO branch_entries (session_id, branch_id, entry_id, entry_seq, entry_type)\n\t\t\tSELECT ${sessionId}, ${targetBranchId}, entry_id, entry_seq, entry_type\n\t\t\tFROM branch_entries\n\t\t\tWHERE session_id = ${sessionId}\n\t\t\t\tAND branch_id = ${segment.branchId}\n\t\t\t\tAND entry_seq > ${lowerSeq}\n\t\t\t\tAND entry_seq <= ${segment.upperSeq}`.run(db);\n\t}\n}\n\nfunction createDivergentBranchForEntry(db: SqliteDatabase, sessionId: string, entry: Entry): void {\n\tif (entry.parentId === null) throw new Error(\"Root entries do not create divergent branches\");\n\tconst segmentsNewestFirst = readBranchSegmentsNewestFirst(db, sessionId, entry.parentId);\n\tconst compaction = readNewestCompactionBoundary(db, sessionId, segmentsNewestFirst);\n\tconst branchId = entry.id;\n\t// A null base means this segment stores its own root-through-parent prefix.\n\tsql`INSERT INTO branch_meta (session_id, branch_id, tip_entry_id, tip_seq, base_branch_id, base_seq)\n\t\tVALUES (${sessionId}, ${branchId}, ${entry.id}, ${entry.seq}, ${compaction?.branchId ?? null}, ${compaction?.seq ?? null})`.run(\n\t\tdb,\n\t);\n\tcopyBranchEntriesAfterSeqThroughParent(db, sessionId, branchId, segmentsNewestFirst, compaction?.seq ?? 0);\n\tinsertBranchEntry(db, sessionId, branchId, entry);\n}\n\nexport function appendEntryToBranchIndex(db: SqliteDatabase, sessionId: string, entry: Entry): void {\n\tif (entry.parentId === null) {\n\t\tcreateRootBranchForEntry(db, sessionId, entry);\n\t\treturn;\n\t}\n\n\tconst branch = readBranchTipForParent(db, sessionId, entry.parentId);\n\tif (branch === undefined) {\n\t\tcreateDivergentBranchForEntry(db, sessionId, entry);\n\t\treturn;\n\t}\n\tappendEntryToExistingBranch(db, sessionId, branch.branch_id, entry);\n}\n\nfunction stopPredicates(query: StorageBranchScan): SqlQuery[] {\n\tconst predicates: SqlQuery[] = [];\n\tif (query.stopAtType !== undefined) predicates.push(sql`b.entry_type = ${query.stopAtType}`);\n\tif (query.stopAtId !== undefined) predicates.push(sql`b.entry_id = ${query.stopAtId}`);\n\treturn predicates;\n}\n\nfunction readStopSeq(\n\tdb: SqliteDatabase,\n\tsessionId: string,\n\tsegment: BranchSegment,\n\tquery: StorageBranchScan,\n\toldestFirst: boolean,\n): number | undefined {\n\tconst stop = stopPredicates(query);\n\tif (stop.length === 0) return undefined;\n\tconst aggregate = oldestFirst ? sql`MIN(b.entry_seq)` : sql`MAX(b.entry_seq)`;\n\tconst row = sql`SELECT ${aggregate} AS stop_seq\n\t\tFROM branch_entries b\n\t\tWHERE b.session_id = ${sessionId}\n\t\t\tAND b.branch_id = ${segment.branchId}\n\t\t\tAND b.entry_seq > ${segment.lowerSeq}\n\t\t\tAND b.entry_seq <= ${segment.upperSeq}\n\t\t\tAND (${joinSqlFragments(stop, \" OR \")})`.get<StopSeqRow>(db);\n\treturn row?.stop_seq ?? undefined;\n}\n\nfunction branchScanPredicates(\n\tsessionId: string,\n\tsegment: BranchSegment,\n\tquery: StorageBranchScan,\n\toldestFirst: boolean,\n\tstopSeq: number | undefined,\n): SqlQuery[] {\n\tconst predicates: SqlQuery[] = [\n\t\tsql`b.session_id = ${sessionId}`,\n\t\tsql`b.branch_id = ${segment.branchId}`,\n\t\tsql`b.entry_seq > ${segment.lowerSeq}`,\n\t\tsql`b.entry_seq <= ${segment.upperSeq}`,\n\t\tsql`e.session_id = b.session_id`,\n\t];\n\tif (stopSeq !== undefined)\n\t\tpredicates.push(oldestFirst ? sql`b.entry_seq <= ${stopSeq}` : sql`b.entry_seq >= ${stopSeq}`);\n\tif (query.type !== undefined) predicates.push(sql`b.entry_type = ${query.type}`);\n\tif (query.customType !== undefined) predicates.push(sql`e.custom_type = ${query.customType}`);\n\tif (query.cursor !== undefined) {\n\t\tpredicates.push(oldestFirst ? sql`b.entry_seq > ${query.cursor.seq}` : sql`b.entry_seq < ${query.cursor.seq}`);\n\t}\n\treturn predicates;\n}\n\nfunction limitSql(limit: number | undefined): SqlQuery {\n\treturn limit === undefined ? sql`` : sql`LIMIT ${Math.max(0, limit)}`;\n}\n\nfunction scanEntrySegmentRows(\n\tdb: SqliteDatabase,\n\tsessionId: string,\n\tsegment: BranchSegment,\n\tquery: StorageBranchScan,\n\toldestFirst: boolean,\n\tstopSeq: number | undefined,\n\tlimit: number | undefined,\n): EntryRow[] {\n\tconst predicates = branchScanPredicates(sessionId, segment, query, oldestFirst, stopSeq);\n\tconst order = oldestFirst ? sql`ASC` : sql`DESC`;\n\treturn sql`SELECT e.id, e.parent_id, e.seq, e.type, e.custom_type, e.timestamp, e.payload\n\t\tFROM branch_entries b\n\t\tCROSS JOIN entries e ON e.session_id = b.session_id AND e.id = b.entry_id\n\t\tWHERE ${joinSqlFragments(predicates, \" AND \")}\n\t\tORDER BY b.entry_seq ${order} ${limitSql(limit)}`.all<EntryRow>(db);\n}\n\nfunction decodeEntryStructureRow(row: EntryStructureRow): EntryStructure {\n\treturn {\n\t\tid: row.id,\n\t\tparentId: row.parent_id,\n\t\tseq: row.seq,\n\t\ttimestamp: row.timestamp,\n\t\ttype: row.type,\n\t\t...(row.custom_type === null ? {} : { customType: row.custom_type }),\n\t};\n}\n\nfunction scanStructureSegmentRows(\n\tdb: SqliteDatabase,\n\tsessionId: string,\n\tsegment: BranchSegment,\n\tquery: StorageBranchScan,\n\toldestFirst: boolean,\n\tstopSeq: number | undefined,\n\tlimit: number | undefined,\n): EntryStructure[] {\n\tconst predicates = branchScanPredicates(sessionId, segment, query, oldestFirst, stopSeq);\n\tconst order = oldestFirst ? sql`ASC` : sql`DESC`;\n\tconst rows = sql`SELECT e.id, e.parent_id, e.seq, e.type, e.custom_type, e.timestamp\n\t\tFROM branch_entries b\n\t\tCROSS JOIN entries e ON e.session_id = b.session_id AND e.id = b.entry_id\n\t\tWHERE ${joinSqlFragments(predicates, \" AND \")}\n\t\tORDER BY b.entry_seq ${order} ${limitSql(limit)}`.all<EntryStructureRow>(db);\n\treturn rows.map(decodeEntryStructureRow);\n}\n\nfunction scanBranchSegments<T>(\n\tdb: SqliteDatabase,\n\tsessionId: string,\n\tquery: StorageBranchScan,\n\treadSegment: (\n\t\tdb: SqliteDatabase,\n\t\tsessionId: string,\n\t\tsegment: BranchSegment,\n\t\tquery: StorageBranchScan,\n\t\toldestFirst: boolean,\n\t\tstopSeq: number | undefined,\n\t\tlimit: number | undefined,\n\t) => T[],\n): T[] {\n\tconst oldestFirst = query.order === \"oldestFirst\";\n\tconst segmentsNewestFirst = readBranchSegmentsNewestFirst(db, sessionId, query.start);\n\tconst segments = oldestFirst ? [...segmentsNewestFirst].reverse() : segmentsNewestFirst;\n\tconst limit = query.limit === undefined ? undefined : Math.max(0, query.limit);\n\tif (limit === 0) return [];\n\n\tconst rows: T[] = [];\n\tfor (const segment of segments) {\n\t\tconst remaining = limit === undefined ? undefined : limit - rows.length;\n\t\tif (remaining !== undefined && remaining <= 0) break;\n\t\tconst stopSeq = readStopSeq(db, sessionId, segment, query, oldestFirst);\n\t\trows.push(...readSegment(db, sessionId, segment, query, oldestFirst, stopSeq, remaining));\n\t\tif (stopSeq !== undefined) break;\n\t}\n\treturn rows;\n}\n\nexport function scanBranchEntries(db: SqliteDatabase, sessionId: string, query: StorageBranchScan): Entry[] {\n\treturn scanBranchSegments(db, sessionId, query, scanEntrySegmentRows).map(decodeEntryRow);\n}\n\nexport function scanBranchEntryStructures(\n\tdb: SqliteDatabase,\n\tsessionId: string,\n\tquery: StorageBranchScan,\n): EntryStructure[] {\n\treturn scanBranchSegments(db, sessionId, query, scanStructureSegmentRows);\n}\n"]}

package/dist/sqlite/session/entries.d.ts

@@ -0,0 +1,24 @@
+import type { Entry, EntryScan, EntryStructure } from "@m4ike1/ion-agent-core";
+import type { SqliteDatabase } from "../types.ts";
+export interface EntryRow {
+    id: string;
+    parent_id: string | null;
+    seq: number;
+    type: Entry["type"];
+    custom_type: string | null;
+    timestamp: number;
+    payload: string;
+}
+export declare class EntryRowWriter {
+    private readonly insertStatement;
+    private readonly sessionId;
+    constructor(db: SqliteDatabase, sessionId: string);
+    insert(entry: Entry): void;
+}
+export declare function insertEntryRow(db: SqliteDatabase, sessionId: string, entry: Entry): void;
+export declare function decodeEntryRow(row: EntryRow): Entry;
+export declare function entryStructureFromRow(row: EntryRow): EntryStructure;
+export declare function readEntryRows(db: SqliteDatabase, sessionId: string, ids: readonly string[]): EntryRow[];
+export declare function readAllEntryRows(db: SqliteDatabase, sessionId: string): EntryRow[];
+export declare function scanEntryRows(db: SqliteDatabase, sessionId: string, query: EntryScan): EntryRow[];
+//# sourceMappingURL=entries.d.ts.map

package/dist/sqlite/session/entries.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"entries.d.ts","sourceRoot":"","sources":["../../../src/sqlite/session/entries.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAIX,KAAK,EACL,SAAS,EACT,cAAc,EAEd,MAAM,wBAAwB,CAAC;AAEhC,OAAO,KAAK,EAAE,cAAc,EAAmB,MAAM,aAAa,CAAC;AAEnE,MAAM,WAAW,QAAQ;IACxB,EAAE,EAAE,MAAM,CAAC;IACX,SAAS,EAAE,MAAM,GAAG,IAAI,CAAC;IACzB,GAAG,EAAE,MAAM,CAAC;IACZ,IAAI,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;IACpB,WAAW,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;CAChB;AAgED,qBAAa,cAAc;IAC1B,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAkB;IAClD,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAS;IAEnC,YAAY,EAAE,EAAE,cAAc,EAAE,SAAS,EAAE,MAAM,EAGhD;IAED,MAAM,CAAC,KAAK,EAAE,KAAK,GAAG,IAAI,CAEzB;CACD;AAED,wBAAgB,cAAc,CAAC,EAAE,EAAE,cAAc,EAAE,SAAS,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,GAAG,IAAI,CAExF;AAED,wBAAgB,cAAc,CAAC,GAAG,EAAE,QAAQ,GAAG,KAAK,CAkBnD;AAED,wBAAgB,qBAAqB,CAAC,GAAG,EAAE,QAAQ,GAAG,cAAc,CASnE;AAED,wBAAgB,aAAa,CAAC,EAAE,EAAE,cAAc,EAAE,SAAS,EAAE,MAAM,EAAE,GAAG,EAAE,SAAS,MAAM,EAAE,GAAG,QAAQ,EAAE,CASvG;AAED,wBAAgB,gBAAgB,CAAC,EAAE,EAAE,cAAc,EAAE,SAAS,EAAE,MAAM,GAAG,QAAQ,EAAE,CAGlF;AAED,wBAAgB,aAAa,CAAC,EAAE,EAAE,cAAc,EAAE,SAAS,EAAE,MAAM,EAAE,KAAK,EAAE,SAAS,GAAG,QAAQ,EAAE,CAWjG","sourcesContent":["import type {\n\tBranchSummaryEntry,\n\tCompactionEntry,\n\tCustomEntry,\n\tEntry,\n\tEntryScan,\n\tEntryStructure,\n\tMessageEntry,\n} from \"@m4ike1/ion-agent-core\";\nimport { joinSqlFragments, type SqlQuery, sql } from \"../sql.ts\";\nimport type { SqliteDatabase, SqliteStatement } from \"../types.ts\";\n\nexport interface EntryRow {\n\tid: string;\n\tparent_id: string | null;\n\tseq: number;\n\ttype: Entry[\"type\"];\n\tcustom_type: string | null;\n\ttimestamp: number;\n\tpayload: string;\n}\n\ntype StoredEntryPayload<TEntry extends Entry> = Omit<\n\tTEntry,\n\t\"id\" | \"parentId\" | \"seq\" | \"timestamp\" | \"type\" | \"customType\"\n>;\n\nfunction entryPayload(entry: Entry): StoredEntryPayload<Entry> {\n\tswitch (entry.type) {\n\t\tcase \"message\": {\n\t\t\tconst payload: StoredEntryPayload<MessageEntry> = {\n\t\t\t\tmessage: entry.message,\n\t\t\t\t...(entry.terminate === undefined ? {} : { terminate: entry.terminate }),\n\t\t\t};\n\t\t\treturn payload;\n\t\t}\n\t\tcase \"compaction\": {\n\t\t\tconst payload: StoredEntryPayload<CompactionEntry> = {\n\t\t\t\tsummary: entry.summary,\n\t\t\t\tretainedTail: entry.retainedTail,\n\t\t\t\ttokensBefore: entry.tokensBefore,\n\t\t\t\t...(entry.details === undefined ? {} : { details: entry.details }),\n\t\t\t\t...(entry.usage === undefined ? {} : { usage: entry.usage }),\n\t\t\t\tfromHook: entry.fromHook,\n\t\t\t};\n\t\t\treturn payload;\n\t\t}\n\t\tcase \"branch_summary\": {\n\t\t\tconst payload: StoredEntryPayload<BranchSummaryEntry> = {\n\t\t\t\tfromId: entry.fromId,\n\t\t\t\tsummary: entry.summary,\n\t\t\t\t...(entry.details === undefined ? {} : { details: entry.details }),\n\t\t\t\t...(entry.usage === undefined ? {} : { usage: entry.usage }),\n\t\t\t\tfromHook: entry.fromHook,\n\t\t\t};\n\t\t\treturn payload;\n\t\t}\n\t\tcase \"custom\": {\n\t\t\tconst payload: StoredEntryPayload<CustomEntry> = entry.data === undefined ? {} : { data: entry.data };\n\t\t\treturn payload;\n\t\t}\n\t}\n}\n\nfunction parsePayload<TEntry extends Entry>(row: EntryRow): StoredEntryPayload<TEntry> {\n\treturn JSON.parse(row.payload) as StoredEntryPayload<TEntry>;\n}\n\nconst INSERT_ENTRY_SQL = `INSERT INTO entries (session_id, id, parent_id, seq, type, custom_type, timestamp, payload)\n\tVALUES (?, ?, ?, ?, ?, ?, ?, ?)`;\n\nfunction entryRowParams(sessionId: string, entry: Entry): unknown[] {\n\treturn [\n\t\tsessionId,\n\t\tentry.id,\n\t\tentry.parentId,\n\t\tentry.seq,\n\t\tentry.type,\n\t\tentry.type === \"custom\" ? entry.customType : null,\n\t\tentry.timestamp,\n\t\tJSON.stringify(entryPayload(entry)),\n\t];\n}\n\nexport class EntryRowWriter {\n\tprivate readonly insertStatement: SqliteStatement;\n\tprivate readonly sessionId: string;\n\n\tconstructor(db: SqliteDatabase, sessionId: string) {\n\t\tthis.insertStatement = db.prepare(INSERT_ENTRY_SQL);\n\t\tthis.sessionId = sessionId;\n\t}\n\n\tinsert(entry: Entry): void {\n\t\tthis.insertStatement.run(...entryRowParams(this.sessionId, entry));\n\t}\n}\n\nexport function insertEntryRow(db: SqliteDatabase, sessionId: string, entry: Entry): void {\n\tdb.prepare(INSERT_ENTRY_SQL).run(...entryRowParams(sessionId, entry));\n}\n\nexport function decodeEntryRow(row: EntryRow): Entry {\n\tconst base = {\n\t\tid: row.id,\n\t\tparentId: row.parent_id,\n\t\tseq: row.seq,\n\t\ttimestamp: row.timestamp,\n\t};\n\tswitch (row.type) {\n\t\tcase \"message\":\n\t\t\treturn { ...base, type: \"message\", ...parsePayload<MessageEntry>(row) };\n\t\tcase \"compaction\":\n\t\t\treturn { ...base, type: \"compaction\", ...parsePayload<CompactionEntry>(row) };\n\t\tcase \"branch_summary\":\n\t\t\treturn { ...base, type: \"branch_summary\", ...parsePayload<BranchSummaryEntry>(row) };\n\t\tcase \"custom\":\n\t\t\tif (row.custom_type === null) throw new Error(`Custom entry ${row.id} is missing custom_type`);\n\t\t\treturn { ...base, type: \"custom\", customType: row.custom_type, ...parsePayload<CustomEntry>(row) };\n\t}\n}\n\nexport function entryStructureFromRow(row: EntryRow): EntryStructure {\n\treturn {\n\t\tid: row.id,\n\t\tparentId: row.parent_id,\n\t\tseq: row.seq,\n\t\ttimestamp: row.timestamp,\n\t\ttype: row.type,\n\t\t...(row.custom_type === null ? {} : { customType: row.custom_type }),\n\t};\n}\n\nexport function readEntryRows(db: SqliteDatabase, sessionId: string, ids: readonly string[]): EntryRow[] {\n\tif (ids.length === 0) return [];\n\tconst placeholders = joinSqlFragments(\n\t\tids.map((id) => sql`${id}`),\n\t\t\", \",\n\t);\n\treturn sql`SELECT id, parent_id, seq, type, custom_type, timestamp, payload\n\t\tFROM entries\n\t\tWHERE session_id = ${sessionId} AND id IN (${placeholders})`.all<EntryRow>(db);\n}\n\nexport function readAllEntryRows(db: SqliteDatabase, sessionId: string): EntryRow[] {\n\treturn sql`SELECT id, parent_id, seq, type, custom_type, timestamp, payload\n\t\tFROM entries WHERE session_id = ${sessionId} ORDER BY seq ASC`.all<EntryRow>(db);\n}\n\nexport function scanEntryRows(db: SqliteDatabase, sessionId: string, query: EntryScan): EntryRow[] {\n\tconst filters: SqlQuery[] = [sql`session_id = ${sessionId}`];\n\tif (query.type !== undefined) filters.push(sql`type = ${query.type}`);\n\tif (query.customType !== undefined) filters.push(sql`custom_type = ${query.customType}`);\n\tif (query.fromSeq !== undefined) filters.push(sql`seq >= ${query.fromSeq}`);\n\tif (query.toSeq !== undefined) filters.push(sql`seq <= ${query.toSeq}`);\n\n\tconst order = query.order === \"desc\" ? sql`ORDER BY seq DESC` : sql`ORDER BY seq ASC`;\n\tconst limit = query.limit === undefined ? sql`` : sql`LIMIT ${Math.max(0, query.limit)}`;\n\treturn sql`SELECT id, parent_id, seq, type, custom_type, timestamp, payload\n\t\tFROM entries WHERE ${joinSqlFragments(filters, \" AND \")} ${order} ${limit}`.all<EntryRow>(db);\n}\n"]}

package/dist/sqlite/session/entries.js

@@ -0,0 +1,126 @@
+import { joinSqlFragments, sql } from "../sql.js";
+function entryPayload(entry) {
+    switch (entry.type) {
+        case "message": {
+            const payload = {
+                message: entry.message,
+                ...(entry.terminate === undefined ? {} : { terminate: entry.terminate }),
+            };
+            return payload;
+        }
+        case "compaction": {
+            const payload = {
+                summary: entry.summary,
+                retainedTail: entry.retainedTail,
+                tokensBefore: entry.tokensBefore,
+                ...(entry.details === undefined ? {} : { details: entry.details }),
+                ...(entry.usage === undefined ? {} : { usage: entry.usage }),
+                fromHook: entry.fromHook,
+            };
+            return payload;
+        }
+        case "branch_summary": {
+            const payload = {
+                fromId: entry.fromId,
+                summary: entry.summary,
+                ...(entry.details === undefined ? {} : { details: entry.details }),
+                ...(entry.usage === undefined ? {} : { usage: entry.usage }),
+                fromHook: entry.fromHook,
+            };
+            return payload;
+        }
+        case "custom": {
+            const payload = entry.data === undefined ? {} : { data: entry.data };
+            return payload;
+        }
+    }
+}
+function parsePayload(row) {
+    return JSON.parse(row.payload);
+}
+const INSERT_ENTRY_SQL = `INSERT INTO entries (session_id, id, parent_id, seq, type, custom_type, timestamp, payload)
+	VALUES (?, ?, ?, ?, ?, ?, ?, ?)`;
+function entryRowParams(sessionId, entry) {
+    return [
+        sessionId,
+        entry.id,
+        entry.parentId,
+        entry.seq,
+        entry.type,
+        entry.type === "custom" ? entry.customType : null,
+        entry.timestamp,
+        JSON.stringify(entryPayload(entry)),
+    ];
+}
+export class EntryRowWriter {
+    insertStatement;
+    sessionId;
+    constructor(db, sessionId) {
+        this.insertStatement = db.prepare(INSERT_ENTRY_SQL);
+        this.sessionId = sessionId;
+    }
+    insert(entry) {
+        this.insertStatement.run(...entryRowParams(this.sessionId, entry));
+    }
+}
+export function insertEntryRow(db, sessionId, entry) {
+    db.prepare(INSERT_ENTRY_SQL).run(...entryRowParams(sessionId, entry));
+}
+export function decodeEntryRow(row) {
+    const base = {
+        id: row.id,
+        parentId: row.parent_id,
+        seq: row.seq,
+        timestamp: row.timestamp,
+    };
+    switch (row.type) {
+        case "message":
+            return { ...base, type: "message", ...parsePayload(row) };
+        case "compaction":
+            return { ...base, type: "compaction", ...parsePayload(row) };
+        case "branch_summary":
+            return { ...base, type: "branch_summary", ...parsePayload(row) };
+        case "custom":
+            if (row.custom_type === null)
+                throw new Error(`Custom entry ${row.id} is missing custom_type`);
+            return { ...base, type: "custom", customType: row.custom_type, ...parsePayload(row) };
+    }
+}
+export function entryStructureFromRow(row) {
+    return {
+        id: row.id,
+        parentId: row.parent_id,
+        seq: row.seq,
+        timestamp: row.timestamp,
+        type: row.type,
+        ...(row.custom_type === null ? {} : { customType: row.custom_type }),
+    };
+}
+export function readEntryRows(db, sessionId, ids) {
+    if (ids.length === 0)
+        return [];
+    const placeholders = joinSqlFragments(ids.map((id) => sql `${id}`), ", ");
+    return sql `SELECT id, parent_id, seq, type, custom_type, timestamp, payload
+		FROM entries
+		WHERE session_id = ${sessionId} AND id IN (${placeholders})`.all(db);
+}
+export function readAllEntryRows(db, sessionId) {
+    return sql `SELECT id, parent_id, seq, type, custom_type, timestamp, payload
+		FROM entries WHERE session_id = ${sessionId} ORDER BY seq ASC`.all(db);
+}
+export function scanEntryRows(db, sessionId, query) {
+    const filters = [sql `session_id = ${sessionId}`];
+    if (query.type !== undefined)
+        filters.push(sql `type = ${query.type}`);
+    if (query.customType !== undefined)
+        filters.push(sql `custom_type = ${query.customType}`);
+    if (query.fromSeq !== undefined)
+        filters.push(sql `seq >= ${query.fromSeq}`);
+    if (query.toSeq !== undefined)
+        filters.push(sql `seq <= ${query.toSeq}`);
+    const order = query.order === "desc" ? sql `ORDER BY seq DESC` : sql `ORDER BY seq ASC`;
+    const limit = query.limit === undefined ? sql `` : sql `LIMIT ${Math.max(0, query.limit)}`;
+    return sql `SELECT id, parent_id, seq, type, custom_type, timestamp, payload
+		FROM entries WHERE ${joinSqlFragments(filters, " AND ")} ${order} ${limit}`.all(db);
+}
+//# sourceMappingURL=entries.js.map

package/dist/sqlite/session/entries.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"entries.js","sourceRoot":"","sources":["../../../src/sqlite/session/entries.ts"],"names":[],"mappings":"AASA,OAAO,EAAE,gBAAgB,EAAiB,GAAG,EAAE,MAAM,WAAW,CAAC;AAkBjE,SAAS,YAAY,CAAC,KAAY,EAA6B;IAC9D,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;QACpB,KAAK,SAAS,EAAE,CAAC;YAChB,MAAM,OAAO,GAAqC;gBACjD,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,GAAG,CAAC,KAAK,CAAC,SAAS,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,KAAK,CAAC,SAAS,EAAE,CAAC;aACxE,CAAC;YACF,OAAO,OAAO,CAAC;QAChB,CAAC;QACD,KAAK,YAAY,EAAE,CAAC;YACnB,MAAM,OAAO,GAAwC;gBACpD,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,YAAY,EAAE,KAAK,CAAC,YAAY;gBAChC,YAAY,EAAE,KAAK,CAAC,YAAY;gBAChC,GAAG,CAAC,KAAK,CAAC,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC;gBAClE,GAAG,CAAC,KAAK,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,CAAC;gBAC5D,QAAQ,EAAE,KAAK,CAAC,QAAQ;aACxB,CAAC;YACF,OAAO,OAAO,CAAC;QAChB,CAAC;QACD,KAAK,gBAAgB,EAAE,CAAC;YACvB,MAAM,OAAO,GAA2C;gBACvD,MAAM,EAAE,KAAK,CAAC,MAAM;gBACpB,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,GAAG,CAAC,KAAK,CAAC,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC;gBAClE,GAAG,CAAC,KAAK,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,CAAC;gBAC5D,QAAQ,EAAE,KAAK,CAAC,QAAQ;aACxB,CAAC;YACF,OAAO,OAAO,CAAC;QAChB,CAAC;QACD,KAAK,QAAQ,EAAE,CAAC;YACf,MAAM,OAAO,GAAoC,KAAK,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC;YACtG,OAAO,OAAO,CAAC;QAChB,CAAC;IACF,CAAC;AAAA,CACD;AAED,SAAS,YAAY,CAAuB,GAAa,EAA8B;IACtF,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAA+B,CAAC;AAAA,CAC7D;AAED,MAAM,gBAAgB,GAAG;iCACQ,CAAC;AAElC,SAAS,cAAc,CAAC,SAAiB,EAAE,KAAY,EAAa;IACnE,OAAO;QACN,SAAS;QACT,KAAK,CAAC,EAAE;QACR,KAAK,CAAC,QAAQ;QACd,KAAK,CAAC,GAAG;QACT,KAAK,CAAC,IAAI;QACV,KAAK,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI;QACjD,KAAK,CAAC,SAAS;QACf,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;KACnC,CAAC;AAAA,CACF;AAED,MAAM,OAAO,cAAc;IACT,eAAe,CAAkB;IACjC,SAAS,CAAS;IAEnC,YAAY,EAAkB,EAAE,SAAiB,EAAE;QAClD,IAAI,CAAC,eAAe,GAAG,EAAE,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;QACpD,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAAA,CAC3B;IAED,MAAM,CAAC,KAAY,EAAQ;QAC1B,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,GAAG,cAAc,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC,CAAC;IAAA,CACnE;CACD;AAED,MAAM,UAAU,cAAc,CAAC,EAAkB,EAAE,SAAiB,EAAE,KAAY,EAAQ;IACzF,EAAE,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC,GAAG,CAAC,GAAG,cAAc,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC,CAAC;AAAA,CACtE;AAED,MAAM,UAAU,cAAc,CAAC,GAAa,EAAS;IACpD,MAAM,IAAI,GAAG;QACZ,EAAE,EAAE,GAAG,CAAC,EAAE;QACV,QAAQ,EAAE,GAAG,CAAC,SAAS;QACvB,GAAG,EAAE,GAAG,CAAC,GAAG;QACZ,SAAS,EAAE,GAAG,CAAC,SAAS;KACxB,CAAC;IACF,QAAQ,GAAG,CAAC,IAAI,EAAE,CAAC;QAClB,KAAK,SAAS;YACb,OAAO,EAAE,GAAG,IAAI,EAAE,IAAI,EAAE,SAAS,EAAE,GAAG,YAAY,CAAe,GAAG,CAAC,EAAE,CAAC;QACzE,KAAK,YAAY;YAChB,OAAO,EAAE,GAAG,IAAI,EAAE,IAAI,EAAE,YAAY,EAAE,GAAG,YAAY,CAAkB,GAAG,CAAC,EAAE,CAAC;QAC/E,KAAK,gBAAgB;YACpB,OAAO,EAAE,GAAG,IAAI,EAAE,IAAI,EAAE,gBAAgB,EAAE,GAAG,YAAY,CAAqB,GAAG,CAAC,EAAE,CAAC;QACtF,KAAK,QAAQ;YACZ,IAAI,GAAG,CAAC,WAAW,KAAK,IAAI;gBAAE,MAAM,IAAI,KAAK,CAAC,gBAAgB,GAAG,CAAC,EAAE,yBAAyB,CAAC,CAAC;YAC/F,OAAO,EAAE,GAAG,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,GAAG,CAAC,WAAW,EAAE,GAAG,YAAY,CAAc,GAAG,CAAC,EAAE,CAAC;IACrG,CAAC;AAAA,CACD;AAED,MAAM,UAAU,qBAAqB,CAAC,GAAa,EAAkB;IACpE,OAAO;QACN,EAAE,EAAE,GAAG,CAAC,EAAE;QACV,QAAQ,EAAE,GAAG,CAAC,SAAS;QACvB,GAAG,EAAE,GAAG,CAAC,GAAG;QACZ,SAAS,EAAE,GAAG,CAAC,SAAS;QACxB,IAAI,EAAE,GAAG,CAAC,IAAI;QACd,GAAG,CAAC,GAAG,CAAC,WAAW,KAAK,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,GAAG,CAAC,WAAW,EAAE,CAAC;KACpE,CAAC;AAAA,CACF;AAED,MAAM,UAAU,aAAa,CAAC,EAAkB,EAAE,SAAiB,EAAE,GAAsB,EAAc;IACxG,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,EAAE,CAAC;IAChC,MAAM,YAAY,GAAG,gBAAgB,CACpC,GAAG,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,GAAG,CAAA,GAAG,EAAE,EAAE,CAAC,EAC3B,IAAI,CACJ,CAAC;IACF,OAAO,GAAG,CAAA;;uBAEY,SAAS,eAAe,YAAY,GAAG,CAAC,GAAG,CAAW,EAAE,CAAC,CAAC;AAAA,CAChF;AAED,MAAM,UAAU,gBAAgB,CAAC,EAAkB,EAAE,SAAiB,EAAc;IACnF,OAAO,GAAG,CAAA;oCACyB,SAAS,mBAAmB,CAAC,GAAG,CAAW,EAAE,CAAC,CAAC;AAAA,CAClF;AAED,MAAM,UAAU,aAAa,CAAC,EAAkB,EAAE,SAAiB,EAAE,KAAgB,EAAc;IAClG,MAAM,OAAO,GAAe,CAAC,GAAG,CAAA,gBAAgB,SAAS,EAAE,CAAC,CAAC;IAC7D,IAAI,KAAK,CAAC,IAAI,KAAK,SAAS;QAAE,OAAO,CAAC,IAAI,CAAC,GAAG,CAAA,UAAU,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;IACtE,IAAI,KAAK,CAAC,UAAU,KAAK,SAAS;QAAE,OAAO,CAAC,IAAI,CAAC,GAAG,CAAA,iBAAiB,KAAK,CAAC,UAAU,EAAE,CAAC,CAAC;IACzF,IAAI,KAAK,CAAC,OAAO,KAAK,SAAS;QAAE,OAAO,CAAC,IAAI,CAAC,GAAG,CAAA,UAAU,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;IAC5E,IAAI,KAAK,CAAC,KAAK,KAAK,SAAS;QAAE,OAAO,CAAC,IAAI,CAAC,GAAG,CAAA,UAAU,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC;IAExE,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,KAAK,MAAM,CAAC,CAAC,CAAC,GAAG,CAAA,mBAAmB,CAAC,CAAC,CAAC,GAAG,CAAA,kBAAkB,CAAC;IACtF,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,GAAG,CAAA,EAAE,CAAC,CAAC,CAAC,GAAG,CAAA,SAAS,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC;IACzF,OAAO,GAAG,CAAA;uBACY,gBAAgB,CAAC,OAAO,EAAE,OAAO,CAAC,IAAI,KAAK,IAAI,KAAK,EAAE,CAAC,GAAG,CAAW,EAAE,CAAC,CAAC;AAAA,CAC/F","sourcesContent":["import type {\n\tBranchSummaryEntry,\n\tCompactionEntry,\n\tCustomEntry,\n\tEntry,\n\tEntryScan,\n\tEntryStructure,\n\tMessageEntry,\n} from \"@m4ike1/ion-agent-core\";\nimport { joinSqlFragments, type SqlQuery, sql } from \"../sql.ts\";\nimport type { SqliteDatabase, SqliteStatement } from \"../types.ts\";\n\nexport interface EntryRow {\n\tid: string;\n\tparent_id: string | null;\n\tseq: number;\n\ttype: Entry[\"type\"];\n\tcustom_type: string | null;\n\ttimestamp: number;\n\tpayload: string;\n}\n\ntype StoredEntryPayload<TEntry extends Entry> = Omit<\n\tTEntry,\n\t\"id\" | \"parentId\" | \"seq\" | \"timestamp\" | \"type\" | \"customType\"\n>;\n\nfunction entryPayload(entry: Entry): StoredEntryPayload<Entry> {\n\tswitch (entry.type) {\n\t\tcase \"message\": {\n\t\t\tconst payload: StoredEntryPayload<MessageEntry> = {\n\t\t\t\tmessage: entry.message,\n\t\t\t\t...(entry.terminate === undefined ? {} : { terminate: entry.terminate }),\n\t\t\t};\n\t\t\treturn payload;\n\t\t}\n\t\tcase \"compaction\": {\n\t\t\tconst payload: StoredEntryPayload<CompactionEntry> = {\n\t\t\t\tsummary: entry.summary,\n\t\t\t\tretainedTail: entry.retainedTail,\n\t\t\t\ttokensBefore: entry.tokensBefore,\n\t\t\t\t...(entry.details === undefined ? {} : { details: entry.details }),\n\t\t\t\t...(entry.usage === undefined ? {} : { usage: entry.usage }),\n\t\t\t\tfromHook: entry.fromHook,\n\t\t\t};\n\t\t\treturn payload;\n\t\t}\n\t\tcase \"branch_summary\": {\n\t\t\tconst payload: StoredEntryPayload<BranchSummaryEntry> = {\n\t\t\t\tfromId: entry.fromId,\n\t\t\t\tsummary: entry.summary,\n\t\t\t\t...(entry.details === undefined ? {} : { details: entry.details }),\n\t\t\t\t...(entry.usage === undefined ? {} : { usage: entry.usage }),\n\t\t\t\tfromHook: entry.fromHook,\n\t\t\t};\n\t\t\treturn payload;\n\t\t}\n\t\tcase \"custom\": {\n\t\t\tconst payload: StoredEntryPayload<CustomEntry> = entry.data === undefined ? {} : { data: entry.data };\n\t\t\treturn payload;\n\t\t}\n\t}\n}\n\nfunction parsePayload<TEntry extends Entry>(row: EntryRow): StoredEntryPayload<TEntry> {\n\treturn JSON.parse(row.payload) as StoredEntryPayload<TEntry>;\n}\n\nconst INSERT_ENTRY_SQL = `INSERT INTO entries (session_id, id, parent_id, seq, type, custom_type, timestamp, payload)\n\tVALUES (?, ?, ?, ?, ?, ?, ?, ?)`;\n\nfunction entryRowParams(sessionId: string, entry: Entry): unknown[] {\n\treturn [\n\t\tsessionId,\n\t\tentry.id,\n\t\tentry.parentId,\n\t\tentry.seq,\n\t\tentry.type,\n\t\tentry.type === \"custom\" ? entry.customType : null,\n\t\tentry.timestamp,\n\t\tJSON.stringify(entryPayload(entry)),\n\t];\n}\n\nexport class EntryRowWriter {\n\tprivate readonly insertStatement: SqliteStatement;\n\tprivate readonly sessionId: string;\n\n\tconstructor(db: SqliteDatabase, sessionId: string) {\n\t\tthis.insertStatement = db.prepare(INSERT_ENTRY_SQL);\n\t\tthis.sessionId = sessionId;\n\t}\n\n\tinsert(entry: Entry): void {\n\t\tthis.insertStatement.run(...entryRowParams(this.sessionId, entry));\n\t}\n}\n\nexport function insertEntryRow(db: SqliteDatabase, sessionId: string, entry: Entry): void {\n\tdb.prepare(INSERT_ENTRY_SQL).run(...entryRowParams(sessionId, entry));\n}\n\nexport function decodeEntryRow(row: EntryRow): Entry {\n\tconst base = {\n\t\tid: row.id,\n\t\tparentId: row.parent_id,\n\t\tseq: row.seq,\n\t\ttimestamp: row.timestamp,\n\t};\n\tswitch (row.type) {\n\t\tcase \"message\":\n\t\t\treturn { ...base, type: \"message\", ...parsePayload<MessageEntry>(row) };\n\t\tcase \"compaction\":\n\t\t\treturn { ...base, type: \"compaction\", ...parsePayload<CompactionEntry>(row) };\n\t\tcase \"branch_summary\":\n\t\t\treturn { ...base, type: \"branch_summary\", ...parsePayload<BranchSummaryEntry>(row) };\n\t\tcase \"custom\":\n\t\t\tif (row.custom_type === null) throw new Error(`Custom entry ${row.id} is missing custom_type`);\n\t\t\treturn { ...base, type: \"custom\", customType: row.custom_type, ...parsePayload<CustomEntry>(row) };\n\t}\n}\n\nexport function entryStructureFromRow(row: EntryRow): EntryStructure {\n\treturn {\n\t\tid: row.id,\n\t\tparentId: row.parent_id,\n\t\tseq: row.seq,\n\t\ttimestamp: row.timestamp,\n\t\ttype: row.type,\n\t\t...(row.custom_type === null ? {} : { customType: row.custom_type }),\n\t};\n}\n\nexport function readEntryRows(db: SqliteDatabase, sessionId: string, ids: readonly string[]): EntryRow[] {\n\tif (ids.length === 0) return [];\n\tconst placeholders = joinSqlFragments(\n\t\tids.map((id) => sql`${id}`),\n\t\t\", \",\n\t);\n\treturn sql`SELECT id, parent_id, seq, type, custom_type, timestamp, payload\n\t\tFROM entries\n\t\tWHERE session_id = ${sessionId} AND id IN (${placeholders})`.all<EntryRow>(db);\n}\n\nexport function readAllEntryRows(db: SqliteDatabase, sessionId: string): EntryRow[] {\n\treturn sql`SELECT id, parent_id, seq, type, custom_type, timestamp, payload\n\t\tFROM entries WHERE session_id = ${sessionId} ORDER BY seq ASC`.all<EntryRow>(db);\n}\n\nexport function scanEntryRows(db: SqliteDatabase, sessionId: string, query: EntryScan): EntryRow[] {\n\tconst filters: SqlQuery[] = [sql`session_id = ${sessionId}`];\n\tif (query.type !== undefined) filters.push(sql`type = ${query.type}`);\n\tif (query.customType !== undefined) filters.push(sql`custom_type = ${query.customType}`);\n\tif (query.fromSeq !== undefined) filters.push(sql`seq >= ${query.fromSeq}`);\n\tif (query.toSeq !== undefined) filters.push(sql`seq <= ${query.toSeq}`);\n\n\tconst order = query.order === \"desc\" ? sql`ORDER BY seq DESC` : sql`ORDER BY seq ASC`;\n\tconst limit = query.limit === undefined ? sql`` : sql`LIMIT ${Math.max(0, query.limit)}`;\n\treturn sql`SELECT id, parent_id, seq, type, custom_type, timestamp, payload\n\t\tFROM entries WHERE ${joinSqlFragments(filters, \" AND \")} ${order} ${limit}`.all<EntryRow>(db);\n}\n"]}

package/dist/sqlite/session/session-row.d.ts

@@ -0,0 +1,23 @@
+import type { SessionMetadata } from "@m4ike1/ion-agent-core";
+import type { SqliteDatabase } from "../types.ts";
+export interface SessionRow {
+    id: string;
+    created_at: number;
+    parent_session_id: string | null;
+    storage_version: number;
+    metadata: string | null;
+    message_count: number;
+    usage_payload: string;
+    next_seq: number;
+}
+export interface SqliteSessionMetadata extends SessionMetadata {
+    /** SQLite container/shard path containing this session. */
+    path: string;
+}
+export declare function readSessionRow(db: SqliteDatabase, sessionId: string): SessionRow;
+export declare function readAllSessionRows(db: SqliteDatabase): SessionRow[];
+export declare function hasSessionRow(db: SqliteDatabase, sessionId: string): boolean;
+export declare function metadataFromSessionRow(path: string, row: SessionRow, currentStorageVersion: number): SqliteSessionMetadata;
+export declare function insertSessionRow(db: SqliteDatabase, metadata: SqliteSessionMetadata, storageVersion: number, nextSeq: number): void;
+export declare function deleteSessionRows(db: SqliteDatabase, sessionId: string): void;
+//# sourceMappingURL=session-row.d.ts.map

package/dist/sqlite/session/session-row.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"session-row.d.ts","sourceRoot":"","sources":["../../../src/sqlite/session/session-row.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AAG9D,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAElD,MAAM,WAAW,UAAU;IAC1B,EAAE,EAAE,MAAM,CAAC;IACX,UAAU,EAAE,MAAM,CAAC;IACnB,iBAAiB,EAAE,MAAM,GAAG,IAAI,CAAC;IACjC,eAAe,EAAE,MAAM,CAAC;IACxB,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;IACxB,aAAa,EAAE,MAAM,CAAC;IACtB,aAAa,EAAE,MAAM,CAAC;IACtB,QAAQ,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,qBAAsB,SAAQ,eAAe;IAC7D,2DAA2D;IAC3D,IAAI,EAAE,MAAM,CAAC;CACb;AAaD,wBAAgB,cAAc,CAAC,EAAE,EAAE,cAAc,EAAE,SAAS,EAAE,MAAM,GAAG,UAAU,CAOhF;AAED,wBAAgB,kBAAkB,CAAC,EAAE,EAAE,cAAc,GAAG,UAAU,EAAE,CAInE;AAED,wBAAgB,aAAa,CAAC,EAAE,EAAE,cAAc,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAE5E;AAED,wBAAgB,sBAAsB,CACrC,IAAI,EAAE,MAAM,EACZ,GAAG,EAAE,UAAU,EACf,qBAAqB,EAAE,MAAM,GAC3B,qBAAqB,CAcvB;AAED,wBAAgB,gBAAgB,CAC/B,EAAE,EAAE,cAAc,EAClB,QAAQ,EAAE,qBAAqB,EAC/B,cAAc,EAAE,MAAM,EACtB,OAAO,EAAE,MAAM,GACb,IAAI,CAaN;AAED,wBAAgB,iBAAiB,CAAC,EAAE,EAAE,cAAc,EAAE,SAAS,EAAE,MAAM,GAAG,IAAI,CAU7E","sourcesContent":["import type { SessionMetadata } from \"@m4ike1/ion-agent-core\";\nimport type { Usage } from \"@m4ike1/ion-ai\";\nimport { sql } from \"../sql.ts\";\nimport type { SqliteDatabase } from \"../types.ts\";\n\nexport interface SessionRow {\n\tid: string;\n\tcreated_at: number;\n\tparent_session_id: string | null;\n\tstorage_version: number;\n\tmetadata: string | null;\n\tmessage_count: number;\n\tusage_payload: string;\n\tnext_seq: number;\n}\n\nexport interface SqliteSessionMetadata extends SessionMetadata {\n\t/** SQLite container/shard path containing this session. */\n\tpath: string;\n}\n\nfunction zeroUsage(): Usage {\n\treturn {\n\t\tinput: 0,\n\t\toutput: 0,\n\t\tcacheRead: 0,\n\t\tcacheWrite: 0,\n\t\ttotalTokens: 0,\n\t\tcost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },\n\t};\n}\n\nexport function readSessionRow(db: SqliteDatabase, sessionId: string): SessionRow {\n\tconst row = sql`SELECT id, created_at, parent_session_id, storage_version, metadata,\n\t\t\tmessage_count, usage_payload, next_seq\n\t\tFROM sessions\n\t\tWHERE id = ${sessionId}`.get<SessionRow>(db);\n\tif (row === undefined) throw new Error(`Unknown SQLite session: ${sessionId}`);\n\treturn row;\n}\n\nexport function readAllSessionRows(db: SqliteDatabase): SessionRow[] {\n\treturn sql`SELECT id, created_at, parent_session_id, storage_version, metadata,\n\t\t\tmessage_count, usage_payload, next_seq\n\t\tFROM sessions`.all<SessionRow>(db);\n}\n\nexport function hasSessionRow(db: SqliteDatabase, sessionId: string): boolean {\n\treturn sql`SELECT id FROM sessions WHERE id = ${sessionId}`.get<{ id: string }>(db) !== undefined;\n}\n\nexport function metadataFromSessionRow(\n\tpath: string,\n\trow: SessionRow,\n\tcurrentStorageVersion: number,\n): SqliteSessionMetadata {\n\tif (row.storage_version > currentStorageVersion) {\n\t\tthrow new Error(`SQLite session storage version ${row.storage_version} is newer than ${currentStorageVersion}`);\n\t}\n\tif (row.storage_version < currentStorageVersion) {\n\t\tthrow new Error(`SQLite session storage version ${row.storage_version} requires migrations`);\n\t}\n\treturn {\n\t\tid: row.id,\n\t\tcreatedAt: row.created_at,\n\t\tstorageVersion: row.storage_version,\n\t\t...(row.parent_session_id === null ? {} : { parentSessionId: row.parent_session_id }),\n\t\tpath,\n\t};\n}\n\nexport function insertSessionRow(\n\tdb: SqliteDatabase,\n\tmetadata: SqliteSessionMetadata,\n\tstorageVersion: number,\n\tnextSeq: number,\n): void {\n\tsql`INSERT INTO sessions\n\t\t\t(id, created_at, parent_session_id, storage_version, metadata, message_count, usage_payload, next_seq)\n\t\tVALUES (\n\t\t\t${metadata.id},\n\t\t\t${metadata.createdAt},\n\t\t\t${metadata.parentSessionId ?? null},\n\t\t\t${storageVersion},\n\t\t\t${null},\n\t\t\t${0},\n\t\t\t${JSON.stringify(zeroUsage())},\n\t\t\t${nextSeq}\n\t\t)`.run(db);\n}\n\nexport function deleteSessionRows(db: SqliteDatabase, sessionId: string): void {\n\tsql`DELETE FROM entries WHERE session_id = ${sessionId}`.run(db);\n\tsql`DELETE FROM scalar_values WHERE session_id = ${sessionId}`.run(db);\n\tsql`DELETE FROM list_values WHERE session_id = ${sessionId}`.run(db);\n\tsql`DELETE FROM usage_ledger WHERE session_id = ${sessionId}`.run(db);\n\tsql`DELETE FROM branch_entries WHERE session_id = ${sessionId}`.run(db);\n\tsql`DELETE FROM branch_meta WHERE session_id = ${sessionId}`.run(db);\n\tconst result = sql`DELETE FROM sessions WHERE id = ${sessionId}`.run(db);\n\tif (result.changes !== 1)\n\t\tthrow new Error(`Expected to delete one SQLite session ${sessionId}, deleted ${result.changes}`);\n}\n"]}

package/dist/sqlite/session/session-row.js

@@ -0,0 +1,69 @@
+import { sql } from "../sql.js";
+function zeroUsage() {
+    return {
+        input: 0,
+        output: 0,
+        cacheRead: 0,
+        cacheWrite: 0,
+        totalTokens: 0,
+        cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },
+    };
+}
+export function readSessionRow(db, sessionId) {
+    const row = sql `SELECT id, created_at, parent_session_id, storage_version, metadata,
+			message_count, usage_payload, next_seq
+		FROM sessions
+		WHERE id = ${sessionId}`.get(db);
+    if (row === undefined)
+        throw new Error(`Unknown SQLite session: ${sessionId}`);
+    return row;
+}
+export function readAllSessionRows(db) {
+    return sql `SELECT id, created_at, parent_session_id, storage_version, metadata,
+			message_count, usage_payload, next_seq
+		FROM sessions`.all(db);
+}
+export function hasSessionRow(db, sessionId) {
+    return sql `SELECT id FROM sessions WHERE id = ${sessionId}`.get(db) !== undefined;
+}
+export function metadataFromSessionRow(path, row, currentStorageVersion) {
+    if (row.storage_version > currentStorageVersion) {
+        throw new Error(`SQLite session storage version ${row.storage_version} is newer than ${currentStorageVersion}`);
+    }
+    if (row.storage_version < currentStorageVersion) {
+        throw new Error(`SQLite session storage version ${row.storage_version} requires migrations`);
+    }
+    return {
+        id: row.id,
+        createdAt: row.created_at,
+        storageVersion: row.storage_version,
+        ...(row.parent_session_id === null ? {} : { parentSessionId: row.parent_session_id }),
+        path,
+    };
+}
+export function insertSessionRow(db, metadata, storageVersion, nextSeq) {
+    sql `INSERT INTO sessions
+			(id, created_at, parent_session_id, storage_version, metadata, message_count, usage_payload, next_seq)
+		VALUES (
+			${metadata.id},
+			${metadata.createdAt},
+			${metadata.parentSessionId ?? null},
+			${storageVersion},
+			${null},
+			${0},
+			${JSON.stringify(zeroUsage())},
+			${nextSeq}
+		)`.run(db);
+}
+export function deleteSessionRows(db, sessionId) {
+    sql `DELETE FROM entries WHERE session_id = ${sessionId}`.run(db);
+    sql `DELETE FROM scalar_values WHERE session_id = ${sessionId}`.run(db);
+    sql `DELETE FROM list_values WHERE session_id = ${sessionId}`.run(db);
+    sql `DELETE FROM usage_ledger WHERE session_id = ${sessionId}`.run(db);
+    sql `DELETE FROM branch_entries WHERE session_id = ${sessionId}`.run(db);
+    sql `DELETE FROM branch_meta WHERE session_id = ${sessionId}`.run(db);
+    const result = sql `DELETE FROM sessions WHERE id = ${sessionId}`.run(db);
+    if (result.changes !== 1)
+        throw new Error(`Expected to delete one SQLite session ${sessionId}, deleted ${result.changes}`);
+}
+//# sourceMappingURL=session-row.js.map

package/dist/sqlite/session/session-row.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"session-row.js","sourceRoot":"","sources":["../../../src/sqlite/session/session-row.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,GAAG,EAAE,MAAM,WAAW,CAAC;AAmBhC,SAAS,SAAS,GAAU;IAC3B,OAAO;QACN,KAAK,EAAE,CAAC;QACR,MAAM,EAAE,CAAC;QACT,SAAS,EAAE,CAAC;QACZ,UAAU,EAAE,CAAC;QACb,WAAW,EAAE,CAAC;QACd,IAAI,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE;KACpE,CAAC;AAAA,CACF;AAED,MAAM,UAAU,cAAc,CAAC,EAAkB,EAAE,SAAiB,EAAc;IACjF,MAAM,GAAG,GAAG,GAAG,CAAA;;;eAGD,SAAS,EAAE,CAAC,GAAG,CAAa,EAAE,CAAC,CAAC;IAC9C,IAAI,GAAG,KAAK,SAAS;QAAE,MAAM,IAAI,KAAK,CAAC,2BAA2B,SAAS,EAAE,CAAC,CAAC;IAC/E,OAAO,GAAG,CAAC;AAAA,CACX;AAED,MAAM,UAAU,kBAAkB,CAAC,EAAkB,EAAgB;IACpE,OAAO,GAAG,CAAA;;gBAEK,CAAC,GAAG,CAAa,EAAE,CAAC,CAAC;AAAA,CACpC;AAED,MAAM,UAAU,aAAa,CAAC,EAAkB,EAAE,SAAiB,EAAW;IAC7E,OAAO,GAAG,CAAA,sCAAsC,SAAS,EAAE,CAAC,GAAG,CAAiB,EAAE,CAAC,KAAK,SAAS,CAAC;AAAA,CAClG;AAED,MAAM,UAAU,sBAAsB,CACrC,IAAY,EACZ,GAAe,EACf,qBAA6B,EACL;IACxB,IAAI,GAAG,CAAC,eAAe,GAAG,qBAAqB,EAAE,CAAC;QACjD,MAAM,IAAI,KAAK,CAAC,kCAAkC,GAAG,CAAC,eAAe,kBAAkB,qBAAqB,EAAE,CAAC,CAAC;IACjH,CAAC;IACD,IAAI,GAAG,CAAC,eAAe,GAAG,qBAAqB,EAAE,CAAC;QACjD,MAAM,IAAI,KAAK,CAAC,kCAAkC,GAAG,CAAC,eAAe,sBAAsB,CAAC,CAAC;IAC9F,CAAC;IACD,OAAO;QACN,EAAE,EAAE,GAAG,CAAC,EAAE;QACV,SAAS,EAAE,GAAG,CAAC,UAAU;QACzB,cAAc,EAAE,GAAG,CAAC,eAAe;QACnC,GAAG,CAAC,GAAG,CAAC,iBAAiB,KAAK,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,eAAe,EAAE,GAAG,CAAC,iBAAiB,EAAE,CAAC;QACrF,IAAI;KACJ,CAAC;AAAA,CACF;AAED,MAAM,UAAU,gBAAgB,CAC/B,EAAkB,EAClB,QAA+B,EAC/B,cAAsB,EACtB,OAAe,EACR;IACP,GAAG,CAAA;;;KAGC,QAAQ,CAAC,EAAE;KACX,QAAQ,CAAC,SAAS;KAClB,QAAQ,CAAC,eAAe,IAAI,IAAI;KAChC,cAAc;KACd,IAAI;KACJ,CAAC;KACD,IAAI,CAAC,SAAS,CAAC,SAAS,EAAE,CAAC;KAC3B,OAAO;IACR,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;AAAA,CACZ;AAED,MAAM,UAAU,iBAAiB,CAAC,EAAkB,EAAE,SAAiB,EAAQ;IAC9E,GAAG,CAAA,0CAA0C,SAAS,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IACjE,GAAG,CAAA,gDAAgD,SAAS,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IACvE,GAAG,CAAA,8CAA8C,SAAS,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IACrE,GAAG,CAAA,+CAA+C,SAAS,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IACtE,GAAG,CAAA,iDAAiD,SAAS,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IACxE,GAAG,CAAA,8CAA8C,SAAS,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IACrE,MAAM,MAAM,GAAG,GAAG,CAAA,mCAAmC,SAAS,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IACzE,IAAI,MAAM,CAAC,OAAO,KAAK,CAAC;QACvB,MAAM,IAAI,KAAK,CAAC,yCAAyC,SAAS,aAAa,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC;AAAA,CAClG","sourcesContent":["import type { SessionMetadata } from \"@m4ike1/ion-agent-core\";\nimport type { Usage } from \"@m4ike1/ion-ai\";\nimport { sql } from \"../sql.ts\";\nimport type { SqliteDatabase } from \"../types.ts\";\n\nexport interface SessionRow {\n\tid: string;\n\tcreated_at: number;\n\tparent_session_id: string | null;\n\tstorage_version: number;\n\tmetadata: string | null;\n\tmessage_count: number;\n\tusage_payload: string;\n\tnext_seq: number;\n}\n\nexport interface SqliteSessionMetadata extends SessionMetadata {\n\t/** SQLite container/shard path containing this session. */\n\tpath: string;\n}\n\nfunction zeroUsage(): Usage {\n\treturn {\n\t\tinput: 0,\n\t\toutput: 0,\n\t\tcacheRead: 0,\n\t\tcacheWrite: 0,\n\t\ttotalTokens: 0,\n\t\tcost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },\n\t};\n}\n\nexport function readSessionRow(db: SqliteDatabase, sessionId: string): SessionRow {\n\tconst row = sql`SELECT id, created_at, parent_session_id, storage_version, metadata,\n\t\t\tmessage_count, usage_payload, next_seq\n\t\tFROM sessions\n\t\tWHERE id = ${sessionId}`.get<SessionRow>(db);\n\tif (row === undefined) throw new Error(`Unknown SQLite session: ${sessionId}`);\n\treturn row;\n}\n\nexport function readAllSessionRows(db: SqliteDatabase): SessionRow[] {\n\treturn sql`SELECT id, created_at, parent_session_id, storage_version, metadata,\n\t\t\tmessage_count, usage_payload, next_seq\n\t\tFROM sessions`.all<SessionRow>(db);\n}\n\nexport function hasSessionRow(db: SqliteDatabase, sessionId: string): boolean {\n\treturn sql`SELECT id FROM sessions WHERE id = ${sessionId}`.get<{ id: string }>(db) !== undefined;\n}\n\nexport function metadataFromSessionRow(\n\tpath: string,\n\trow: SessionRow,\n\tcurrentStorageVersion: number,\n): SqliteSessionMetadata {\n\tif (row.storage_version > currentStorageVersion) {\n\t\tthrow new Error(`SQLite session storage version ${row.storage_version} is newer than ${currentStorageVersion}`);\n\t}\n\tif (row.storage_version < currentStorageVersion) {\n\t\tthrow new Error(`SQLite session storage version ${row.storage_version} requires migrations`);\n\t}\n\treturn {\n\t\tid: row.id,\n\t\tcreatedAt: row.created_at,\n\t\tstorageVersion: row.storage_version,\n\t\t...(row.parent_session_id === null ? {} : { parentSessionId: row.parent_session_id }),\n\t\tpath,\n\t};\n}\n\nexport function insertSessionRow(\n\tdb: SqliteDatabase,\n\tmetadata: SqliteSessionMetadata,\n\tstorageVersion: number,\n\tnextSeq: number,\n): void {\n\tsql`INSERT INTO sessions\n\t\t\t(id, created_at, parent_session_id, storage_version, metadata, message_count, usage_payload, next_seq)\n\t\tVALUES (\n\t\t\t${metadata.id},\n\t\t\t${metadata.createdAt},\n\t\t\t${metadata.parentSessionId ?? null},\n\t\t\t${storageVersion},\n\t\t\t${null},\n\t\t\t${0},\n\t\t\t${JSON.stringify(zeroUsage())},\n\t\t\t${nextSeq}\n\t\t)`.run(db);\n}\n\nexport function deleteSessionRows(db: SqliteDatabase, sessionId: string): void {\n\tsql`DELETE FROM entries WHERE session_id = ${sessionId}`.run(db);\n\tsql`DELETE FROM scalar_values WHERE session_id = ${sessionId}`.run(db);\n\tsql`DELETE FROM list_values WHERE session_id = ${sessionId}`.run(db);\n\tsql`DELETE FROM usage_ledger WHERE session_id = ${sessionId}`.run(db);\n\tsql`DELETE FROM branch_entries WHERE session_id = ${sessionId}`.run(db);\n\tsql`DELETE FROM branch_meta WHERE session_id = ${sessionId}`.run(db);\n\tconst result = sql`DELETE FROM sessions WHERE id = ${sessionId}`.run(db);\n\tif (result.changes !== 1)\n\t\tthrow new Error(`Expected to delete one SQLite session ${sessionId}, deleted ${result.changes}`);\n}\n"]}

package/dist/sqlite/session/session-sequences.d.ts

@@ -0,0 +1,4 @@
+import type { SqliteDatabase } from "../types.ts";
+export declare function readNextSeq(db: SqliteDatabase, sessionId: string): number;
+export declare function advanceNextSeq(db: SqliteDatabase, sessionId: string, nextSeq: number): void;
+//# sourceMappingURL=session-sequences.d.ts.map

package/dist/sqlite/session/session-sequences.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"session-sequences.d.ts","sourceRoot":"","sources":["../../../src/sqlite/session/session-sequences.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAElD,wBAAgB,WAAW,CAAC,EAAE,EAAE,cAAc,EAAE,SAAS,EAAE,MAAM,GAAG,MAAM,CAIzE;AAED,wBAAgB,cAAc,CAAC,EAAE,EAAE,cAAc,EAAE,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI,CAI3F","sourcesContent":["import { sql } from \"../sql.ts\";\nimport type { SqliteDatabase } from \"../types.ts\";\n\nexport function readNextSeq(db: SqliteDatabase, sessionId: string): number {\n\tconst row = sql`SELECT next_seq FROM sessions WHERE id = ${sessionId}`.get<{ next_seq: number }>(db);\n\tif (row === undefined) throw new Error(`Unknown SQLite session: ${sessionId}`);\n\treturn row.next_seq;\n}\n\nexport function advanceNextSeq(db: SqliteDatabase, sessionId: string, nextSeq: number): void {\n\tconst result = sql`UPDATE sessions SET next_seq = ${nextSeq} WHERE id = ${sessionId}`.run(db);\n\tif (result.changes !== 1)\n\t\tthrow new Error(`Expected to update one SQLite session ${sessionId}, updated ${result.changes}`);\n}\n"]}

package/dist/sqlite/session/session-sequences.js

@@ -0,0 +1,13 @@
+import { sql } from "../sql.js";
+export function readNextSeq(db, sessionId) {
+    const row = sql `SELECT next_seq FROM sessions WHERE id = ${sessionId}`.get(db);
+    if (row === undefined)
+        throw new Error(`Unknown SQLite session: ${sessionId}`);
+    return row.next_seq;
+}
+export function advanceNextSeq(db, sessionId, nextSeq) {
+    const result = sql `UPDATE sessions SET next_seq = ${nextSeq} WHERE id = ${sessionId}`.run(db);
+    if (result.changes !== 1)
+        throw new Error(`Expected to update one SQLite session ${sessionId}, updated ${result.changes}`);
+}
+//# sourceMappingURL=session-sequences.js.map

package/dist/sqlite/session/session-sequences.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"session-sequences.js","sourceRoot":"","sources":["../../../src/sqlite/session/session-sequences.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,GAAG,EAAE,MAAM,WAAW,CAAC;AAGhC,MAAM,UAAU,WAAW,CAAC,EAAkB,EAAE,SAAiB,EAAU;IAC1E,MAAM,GAAG,GAAG,GAAG,CAAA,4CAA4C,SAAS,EAAE,CAAC,GAAG,CAAuB,EAAE,CAAC,CAAC;IACrG,IAAI,GAAG,KAAK,SAAS;QAAE,MAAM,IAAI,KAAK,CAAC,2BAA2B,SAAS,EAAE,CAAC,CAAC;IAC/E,OAAO,GAAG,CAAC,QAAQ,CAAC;AAAA,CACpB;AAED,MAAM,UAAU,cAAc,CAAC,EAAkB,EAAE,SAAiB,EAAE,OAAe,EAAQ;IAC5F,MAAM,MAAM,GAAG,GAAG,CAAA,kCAAkC,OAAO,eAAe,SAAS,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAC9F,IAAI,MAAM,CAAC,OAAO,KAAK,CAAC;QACvB,MAAM,IAAI,KAAK,CAAC,yCAAyC,SAAS,aAAa,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC;AAAA,CAClG","sourcesContent":["import { sql } from \"../sql.ts\";\nimport type { SqliteDatabase } from \"../types.ts\";\n\nexport function readNextSeq(db: SqliteDatabase, sessionId: string): number {\n\tconst row = sql`SELECT next_seq FROM sessions WHERE id = ${sessionId}`.get<{ next_seq: number }>(db);\n\tif (row === undefined) throw new Error(`Unknown SQLite session: ${sessionId}`);\n\treturn row.next_seq;\n}\n\nexport function advanceNextSeq(db: SqliteDatabase, sessionId: string, nextSeq: number): void {\n\tconst result = sql`UPDATE sessions SET next_seq = ${nextSeq} WHERE id = ${sessionId}`.run(db);\n\tif (result.changes !== 1)\n\t\tthrow new Error(`Expected to update one SQLite session ${sessionId}, updated ${result.changes}`);\n}\n"]}

package/dist/sqlite/session/session-stats.d.ts

@@ -0,0 +1,6 @@
+import type { SessionStats, UsageRow } from "@m4ike1/ion-agent-core";
+import type { SqliteDatabase } from "../types.ts";
+export declare function readSessionStats(db: SqliteDatabase, sessionId: string): SessionStats;
+export declare function incrementMessageCount(db: SqliteDatabase, sessionId: string): void;
+export declare function addUsageToSessionStats(db: SqliteDatabase, sessionId: string, usage: UsageRow["usage"]): void;
+//# sourceMappingURL=session-stats.d.ts.map

package/dist/sqlite/session/session-stats.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"session-stats.d.ts","sourceRoot":"","sources":["../../../src/sqlite/session/session-stats.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,QAAQ,EAAE,MAAM,wBAAwB,CAAC;AAGrE,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AA0BlD,wBAAgB,gBAAgB,CAAC,EAAE,EAAE,cAAc,EAAE,SAAS,EAAE,MAAM,GAAG,YAAY,CAMpF;AAED,wBAAgB,qBAAqB,CAAC,EAAE,EAAE,cAAc,EAAE,SAAS,EAAE,MAAM,GAAG,IAAI,CAEjF;AAED,wBAAgB,sBAAsB,CAAC,EAAE,EAAE,cAAc,EAAE,SAAS,EAAE,MAAM,EAAE,KAAK,EAAE,QAAQ,CAAC,OAAO,CAAC,GAAG,IAAI,CAG5G","sourcesContent":["import type { SessionStats, UsageRow } from \"@m4ike1/ion-agent-core\";\nimport type { Usage } from \"@m4ike1/ion-ai\";\nimport { sql } from \"../sql.ts\";\nimport type { SqliteDatabase } from \"../types.ts\";\nimport { readSessionRow } from \"./session-row.ts\";\n\nfunction addUsage(left: Usage, right: Usage): Usage {\n\treturn {\n\t\tinput: left.input + right.input,\n\t\toutput: left.output + right.output,\n\t\tcacheRead: left.cacheRead + right.cacheRead,\n\t\tcacheWrite: left.cacheWrite + right.cacheWrite,\n\t\t...(left.cacheWrite1h === undefined && right.cacheWrite1h === undefined\n\t\t\t? {}\n\t\t\t: { cacheWrite1h: (left.cacheWrite1h ?? 0) + (right.cacheWrite1h ?? 0) }),\n\t\t...(left.reasoning === undefined && right.reasoning === undefined\n\t\t\t? {}\n\t\t\t: { reasoning: (left.reasoning ?? 0) + (right.reasoning ?? 0) }),\n\t\ttotalTokens: left.totalTokens + right.totalTokens,\n\t\tcost: {\n\t\t\tinput: left.cost.input + right.cost.input,\n\t\t\toutput: left.cost.output + right.cost.output,\n\t\t\tcacheRead: left.cost.cacheRead + right.cost.cacheRead,\n\t\t\tcacheWrite: left.cost.cacheWrite + right.cost.cacheWrite,\n\t\t\ttotal: left.cost.total + right.cost.total,\n\t\t},\n\t};\n}\n\nexport function readSessionStats(db: SqliteDatabase, sessionId: string): SessionStats {\n\tconst row = readSessionRow(db, sessionId);\n\treturn {\n\t\tmessageCount: row.message_count,\n\t\tusage: JSON.parse(row.usage_payload) as Usage,\n\t};\n}\n\nexport function incrementMessageCount(db: SqliteDatabase, sessionId: string): void {\n\tsql`UPDATE sessions SET message_count = message_count + 1 WHERE id = ${sessionId}`.run(db);\n}\n\nexport function addUsageToSessionStats(db: SqliteDatabase, sessionId: string, usage: UsageRow[\"usage\"]): void {\n\tconst current = readSessionStats(db, sessionId).usage;\n\tsql`UPDATE sessions SET usage_payload = ${JSON.stringify(addUsage(current, usage))} WHERE id = ${sessionId}`.run(db);\n}\n"]}