ds4-context-core 0.1.0

package/dist/persistence/repositories/summary-repository.js

@@ -0,0 +1,242 @@
+import { parseDs4CompactionDetails, } from "../../compaction/compaction-record.js";
+const SELECT_SUMMARY = `
+  SELECT summary_id, session_id, summary_kind, content, source_hash, created_at,
+         validation_status, provider, model, first_kept_entry_id, tokens_before,
+         compaction_reason, lifecycle_status, pi_compaction_entry_id, metadata_json,
+         graph_level
+  FROM summaries
+`;
+function parseKind(value) {
+    if (!["segment", "aggregate", "task-state", "branch"].includes(value)) {
+        throw new Error(`Unsupported summary kind in database: ${value}`);
+    }
+    return value;
+}
+function rowToRecord(database, row) {
+    const parsed = JSON.parse(row.metadata_json);
+    const metadata = parseDs4CompactionDetails({
+        readFiles: [],
+        modifiedFiles: [],
+        ds4ContextEngine: parsed,
+    })?.ds4ContextEngine;
+    if (!metadata)
+        throw new Error(`Summary ${row.summary_id} contains invalid DS4 metadata`);
+    if (metadata.summaryId !== row.summary_id
+        || metadata.summaryKind !== row.summary_kind
+        || metadata.sourceHash !== row.source_hash
+        || metadata.graphLevel !== row.graph_level) {
+        throw new Error(`Summary ${row.summary_id} metadata does not match graph columns`);
+    }
+    const sourceRows = database.prepare(`
+    SELECT entry.entry_id
+    FROM summary_sources AS source
+    JOIN entries AS entry ON entry.entry_key = source.entry_key
+    WHERE source.summary_id = ?
+    ORDER BY entry.created_at, entry.entry_id
+  `).all(row.summary_id);
+    const childRows = database.prepare(`
+    SELECT child_summary_id
+    FROM summary_edges
+    WHERE parent_summary_id = ?
+    ORDER BY child_order, child_summary_id
+  `).all(row.summary_id);
+    const sourceEntryIds = sourceRows.map((source) => source.entry_id);
+    const childSummaryIds = childRows.map((child) => child.child_summary_id);
+    const sorted = (values) => [...values].sort((left, right) => left.localeCompare(right));
+    if (JSON.stringify(sorted(sourceEntryIds)) !== JSON.stringify(sorted(metadata.sourceEntryIds))) {
+        throw new Error(`Summary ${row.summary_id} source links do not match metadata`);
+    }
+    if (JSON.stringify(childSummaryIds) !== JSON.stringify(metadata.childSummaryIds)) {
+        throw new Error(`Summary ${row.summary_id} child order does not match metadata`);
+    }
+    return {
+        id: row.summary_id,
+        sessionId: row.session_id,
+        kind: parseKind(row.summary_kind),
+        content: row.content,
+        sourceHash: row.source_hash,
+        sourceEntryIds,
+        childSummaryIds,
+        graphLevel: row.graph_level,
+        createdAt: row.created_at,
+        validationStatus: metadata.validationStatus,
+        ...(row.provider ? { provider: row.provider } : {}),
+        ...(row.model ? { model: row.model } : {}),
+        firstKeptEntryId: row.first_kept_entry_id ?? metadata.firstKeptEntryId,
+        tokensBefore: row.tokens_before ?? metadata.tokensBefore,
+        reason: metadata.reason,
+        lifecycleStatus: row.lifecycle_status,
+        ...(row.pi_compaction_entry_id ? { piCompactionEntryId: row.pi_compaction_entry_id } : {}),
+        metadata,
+    };
+}
+function assertImmutable(database, record) {
+    const existing = database.prepare(`
+    SELECT session_id, summary_kind, content, source_hash, created_at, graph_level
+    FROM summaries WHERE summary_id = ?
+  `).get(record.id);
+    if (!existing)
+        return;
+    if (existing.session_id !== record.sessionId
+        || existing.summary_kind !== record.kind
+        || existing.content !== record.content
+        || existing.source_hash !== record.sourceHash
+        || existing.created_at !== record.createdAt
+        || existing.graph_level !== record.graphLevel) {
+        throw new Error(`Summary node is immutable and conflicts with existing node: ${record.id}`);
+    }
+}
+export class SummaryRepository {
+    database;
+    constructor(database) {
+        this.database = database;
+    }
+    save(record) {
+        this.saveGraph([record]);
+    }
+    saveGraph(records) {
+        if (records.length === 0)
+            return;
+        const ids = new Set();
+        for (const record of records) {
+            if (ids.has(record.id))
+                throw new Error(`Duplicate summary node in graph batch: ${record.id}`);
+            ids.add(record.id);
+        }
+        this.database.exec("BEGIN IMMEDIATE");
+        try {
+            for (const record of records) {
+                assertImmutable(this.database, record);
+                this.database.prepare(`
+          INSERT INTO summaries(
+            summary_id, session_id, summary_kind, content, source_hash, created_at,
+            validation_status, provider, model, first_kept_entry_id, tokens_before,
+            compaction_reason, lifecycle_status, pi_compaction_entry_id, metadata_json,
+            graph_level
+          ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+          ON CONFLICT(summary_id) DO UPDATE SET
+            validation_status = excluded.validation_status,
+            provider = excluded.provider,
+            model = excluded.model,
+            first_kept_entry_id = excluded.first_kept_entry_id,
+            tokens_before = excluded.tokens_before,
+            compaction_reason = excluded.compaction_reason,
+            lifecycle_status = CASE
+              WHEN excluded.lifecycle_status = 'committed' THEN 'committed'
+              WHEN summaries.lifecycle_status = 'committed' THEN 'committed'
+              ELSE excluded.lifecycle_status
+            END,
+            pi_compaction_entry_id = COALESCE(excluded.pi_compaction_entry_id, summaries.pi_compaction_entry_id),
+            metadata_json = excluded.metadata_json,
+            graph_level = excluded.graph_level
+        `).run(record.id, record.sessionId, record.kind, record.content, record.sourceHash, record.createdAt, record.validationStatus, record.provider ?? null, record.model ?? null, record.firstKeptEntryId, record.tokensBefore, record.reason, record.lifecycleStatus, record.piCompactionEntryId ?? null, JSON.stringify(record.metadata), record.graphLevel);
+            }
+            for (const record of records) {
+                this.database.prepare("DELETE FROM summary_sources WHERE summary_id = ?").run(record.id);
+                this.database.prepare("DELETE FROM summary_edges WHERE parent_summary_id = ?").run(record.id);
+            }
+            const insertSource = this.database.prepare(`
+        INSERT INTO summary_sources(summary_id, entry_key)
+        SELECT ?, entry_key FROM entries WHERE session_id = ? AND entry_id = ?
+      `);
+            const childLookup = this.database.prepare(`
+        SELECT session_id, graph_level FROM summaries WHERE summary_id = ?
+      `);
+            const insertEdge = this.database.prepare(`
+        INSERT INTO summary_edges(parent_summary_id, child_summary_id, child_order)
+        VALUES (?, ?, ?)
+      `);
+            for (const record of records) {
+                for (const entryId of new Set(record.sourceEntryIds)) {
+                    const result = insertSource.run(record.id, record.sessionId, entryId);
+                    if (result.changes !== 1) {
+                        throw new Error(`Summary source entry is not indexed: ${record.sessionId}:${entryId}`);
+                    }
+                }
+                const children = [...new Set(record.childSummaryIds)];
+                for (const [order, childId] of children.entries()) {
+                    if (childId === record.id)
+                        throw new Error(`Summary node cannot reference itself: ${record.id}`);
+                    const child = childLookup.get(childId);
+                    if (!child || child.session_id !== record.sessionId) {
+                        throw new Error(`Summary child is missing from the same session: ${record.sessionId}:${childId}`);
+                    }
+                    if (child.graph_level >= record.graphLevel) {
+                        throw new Error(`Summary child level must be lower than parent: ${childId} -> ${record.id}`);
+                    }
+                    const cycle = this.database.prepare(`
+            WITH RECURSIVE descendants(summary_id) AS (
+              SELECT child_summary_id FROM summary_edges WHERE parent_summary_id = ?
+              UNION
+              SELECT edge.child_summary_id
+              FROM summary_edges AS edge
+              JOIN descendants AS child ON edge.parent_summary_id = child.summary_id
+            )
+            SELECT 1 AS found FROM descendants WHERE summary_id = ? LIMIT 1
+          `).get(childId, record.id);
+                    if (cycle)
+                        throw new Error(`Summary edge would create a cycle: ${record.id} -> ${childId}`);
+                    insertEdge.run(record.id, childId, order);
+                }
+            }
+            this.database.exec("COMMIT");
+        }
+        catch (error) {
+            if (this.database.isTransaction)
+                this.database.exec("ROLLBACK");
+            throw error;
+        }
+    }
+    markCommitted(summaryId, piCompactionEntryId) {
+        const result = this.database.prepare(`
+      UPDATE summaries
+      SET lifecycle_status = 'committed',
+          pi_compaction_entry_id = COALESCE(?, pi_compaction_entry_id)
+      WHERE summary_id = ?
+    `).run(piCompactionEntryId ?? null, summaryId);
+        return result.changes === 1;
+    }
+    markFailed(summaryId) {
+        const result = this.database.prepare(`
+      UPDATE summaries SET lifecycle_status = 'failed'
+      WHERE summary_id = ? AND lifecycle_status = 'prepared'
+    `).run(summaryId);
+        return result.changes === 1;
+    }
+    markFailedMany(summaryIds) {
+        let changed = 0;
+        for (const summaryId of new Set(summaryIds))
+            changed += this.markFailed(summaryId) ? 1 : 0;
+        return changed;
+    }
+    failPreparedForSession(sessionId) {
+        const result = this.database.prepare(`
+      UPDATE summaries SET lifecycle_status = 'failed'
+      WHERE session_id = ? AND lifecycle_status = 'prepared'
+    `).run(sessionId);
+        return Number(result.changes);
+    }
+    getById(summaryId) {
+        const row = this.database.prepare(`${SELECT_SUMMARY} WHERE summary_id = ? AND metadata_json <> '{}'`)
+            .get(summaryId);
+        return row ? rowToRecord(this.database, row) : undefined;
+    }
+    getLatest(sessionId) {
+        const row = this.database.prepare(`
+      ${SELECT_SUMMARY}
+      WHERE session_id = ? AND metadata_json <> '{}'
+      ORDER BY created_at DESC, graph_level DESC, rowid DESC
+      LIMIT 1
+    `).get(sessionId);
+        return row ? rowToRecord(this.database, row) : undefined;
+    }
+    listBySession(sessionId) {
+        const rows = this.database.prepare(`
+      ${SELECT_SUMMARY}
+      WHERE session_id = ? AND metadata_json <> '{}'
+      ORDER BY created_at, graph_level, rowid
+    `).all(sessionId);
+        return rows.map((row) => rowToRecord(this.database, row));
+    }
+}
+//# sourceMappingURL=summary-repository.js.map

package/dist/persistence/repositories/summary-repository.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"summary-repository.js","sourceRoot":"","sources":["../../../src/persistence/repositories/summary-repository.ts"],"names":[],"mappings":"AACA,OAAO,EACL,yBAAyB,GAI1B,MAAM,uCAAuC,CAAC;AA8B/C,MAAM,cAAc,GAAG;;;;;;CAMtB,CAAC;AAEF,SAAS,SAAS,CAAC,KAAa;IAC9B,IAAI,CAAC,CAAC,SAAS,EAAE,WAAW,EAAE,YAAY,EAAE,QAAQ,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;QACtE,MAAM,IAAI,KAAK,CAAC,yCAAyC,KAAK,EAAE,CAAC,CAAC;IACpE,CAAC;IACD,OAAO,KAAoB,CAAC;AAC9B,CAAC;AAED,SAAS,WAAW,CAAC,QAAsB,EAAE,GAAe;IAC1D,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,aAAa,CAAY,CAAC;IACxD,MAAM,QAAQ,GAAG,yBAAyB,CAAC;QACzC,SAAS,EAAE,EAAE;QACb,aAAa,EAAE,EAAE;QACjB,gBAAgB,EAAE,MAAM;KACzB,CAAC,EAAE,gBAAgB,CAAC;IACrB,IAAI,CAAC,QAAQ;QAAE,MAAM,IAAI,KAAK,CAAC,WAAW,GAAG,CAAC,UAAU,gCAAgC,CAAC,CAAC;IAC1F,IACE,QAAQ,CAAC,SAAS,KAAK,GAAG,CAAC,UAAU;WAClC,QAAQ,CAAC,WAAW,KAAK,GAAG,CAAC,YAAY;WACzC,QAAQ,CAAC,UAAU,KAAK,GAAG,CAAC,WAAW;WACvC,QAAQ,CAAC,UAAU,KAAK,GAAG,CAAC,WAAW,EAC1C,CAAC;QACD,MAAM,IAAI,KAAK,CAAC,WAAW,GAAG,CAAC,UAAU,wCAAwC,CAAC,CAAC;IACrF,CAAC;IACD,MAAM,UAAU,GAAG,QAAQ,CAAC,OAAO,CAAC;;;;;;GAMnC,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,UAAU,CAA2C,CAAC;IACjE,MAAM,SAAS,GAAG,QAAQ,CAAC,OAAO,CAAC;;;;;GAKlC,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,UAAU,CAAmD,CAAC;IACzE,MAAM,cAAc,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IACnE,MAAM,eAAe,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC;IACzE,MAAM,MAAM,GAAG,CAAC,MAAyB,EAAY,EAAE,CAAC,CAAC,GAAG,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,CAAC;IACrH,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,KAAK,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC,EAAE,CAAC;QAC/F,MAAM,IAAI,KAAK,CAAC,WAAW,GAAG,CAAC,UAAU,qCAAqC,CAAC,CAAC;IAClF,CAAC;IACD,IAAI,IAAI,CAAC,SAAS,CAAC,eAAe,CAAC,KAAK,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,eAAe,CAAC,EAAE,CAAC;QACjF,MAAM,IAAI,KAAK,CAAC,WAAW,GAAG,CAAC,UAAU,sCAAsC,CAAC,CAAC;IACnF,CAAC;IAED,OAAO;QACL,EAAE,EAAE,GAAG,CAAC,UAAU;QAClB,SAAS,EAAE,GAAG,CAAC,UAAU;QACzB,IAAI,EAAE,SAAS,CAAC,GAAG,CAAC,YAAY,CAAC;QACjC,OAAO,EAAE,GAAG,CAAC,OAAO;QACpB,UAAU,EAAE,GAAG,CAAC,WAAW;QAC3B,cAAc;QACd,eAAe;QACf,UAAU,EAAE,GAAG,CAAC,WAAW;QAC3B,SAAS,EAAE,GAAG,CAAC,UAAU;QACzB,gBAAgB,EAAE,QAAQ,CAAC,gBAAgB;QAC3C,GAAG,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,GAAG,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACnD,GAAG,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,GAAG,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC1C,gBAAgB,EAAE,GAAG,CAAC,mBAAmB,IAAI,QAAQ,CAAC,gBAAgB;QACtE,YAAY,EAAE,GAAG,CAAC,aAAa,IAAI,QAAQ,CAAC,YAAY;QACxD,MAAM,EAAE,QAAQ,CAAC,MAAM;QACvB,eAAe,EAAE,GAAG,CAAC,gBAAgB;QACrC,GAAG,CAAC,GAAG,CAAC,sBAAsB,CAAC,CAAC,CAAC,EAAE,mBAAmB,EAAE,GAAG,CAAC,sBAAsB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC1F,QAAQ;KACT,CAAC;AACJ,CAAC;AAED,SAAS,eAAe,CAAC,QAAsB,EAAE,MAAqB;IACpE,MAAM,QAAQ,GAAG,QAAQ,CAAC,OAAO,CAAC;;;GAGjC,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAA+C,CAAC;IAChE,IAAI,CAAC,QAAQ;QAAE,OAAO;IACtB,IACE,QAAQ,CAAC,UAAU,KAAK,MAAM,CAAC,SAAS;WACrC,QAAQ,CAAC,YAAY,KAAK,MAAM,CAAC,IAAI;WACrC,QAAQ,CAAC,OAAO,KAAK,MAAM,CAAC,OAAO;WACnC,QAAQ,CAAC,WAAW,KAAK,MAAM,CAAC,UAAU;WAC1C,QAAQ,CAAC,UAAU,KAAK,MAAM,CAAC,SAAS;WACxC,QAAQ,CAAC,WAAW,KAAK,MAAM,CAAC,UAAU,EAC7C,CAAC;QACD,MAAM,IAAI,KAAK,CAAC,+DAA+D,MAAM,CAAC,EAAE,EAAE,CAAC,CAAC;IAC9F,CAAC;AACH,CAAC;AAED,MAAM,OAAO,iBAAiB;IACC;IAA7B,YAA6B,QAAsB;QAAtB,aAAQ,GAAR,QAAQ,CAAc;IAAG,CAAC;IAEvD,IAAI,CAAC,MAAqB;QACxB,IAAI,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;IAC3B,CAAC;IAED,SAAS,CAAC,OAAiC;QACzC,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QACjC,MAAM,GAAG,GAAG,IAAI,GAAG,EAAU,CAAC;QAC9B,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;YAC7B,IAAI,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;gBAAE,MAAM,IAAI,KAAK,CAAC,0CAA0C,MAAM,CAAC,EAAE,EAAE,CAAC,CAAC;YAC/F,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QACrB,CAAC;QAED,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QACtC,IAAI,CAAC;YACH,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;gBAC7B,eAAe,CAAC,IAAI,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;gBACvC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;;;;;;;;;;;;;;;;;;;;;;SAsBrB,CAAC,CAAC,GAAG,CACJ,MAAM,CAAC,EAAE,EACT,MAAM,CAAC,SAAS,EAChB,MAAM,CAAC,IAAI,EACX,MAAM,CAAC,OAAO,EACd,MAAM,CAAC,UAAU,EACjB,MAAM,CAAC,SAAS,EAChB,MAAM,CAAC,gBAAgB,EACvB,MAAM,CAAC,QAAQ,IAAI,IAAI,EACvB,MAAM,CAAC,KAAK,IAAI,IAAI,EACpB,MAAM,CAAC,gBAAgB,EACvB,MAAM,CAAC,YAAY,EACnB,MAAM,CAAC,MAAM,EACb,MAAM,CAAC,eAAe,EACtB,MAAM,CAAC,mBAAmB,IAAI,IAAI,EAClC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,EAC/B,MAAM,CAAC,UAAU,CAClB,CAAC;YACJ,CAAC;YAED,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;gBAC7B,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,kDAAkD,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;gBACzF,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,uDAAuD,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YAChG,CAAC;YAED,MAAM,YAAY,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;;;OAG1C,CAAC,CAAC;YACH,MAAM,WAAW,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;;OAEzC,CAAC,CAAC;YACH,MAAM,UAAU,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;;;OAGxC,CAAC,CAAC;YACH,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;gBAC7B,KAAK,MAAM,OAAO,IAAI,IAAI,GAAG,CAAC,MAAM,CAAC,cAAc,CAAC,EAAE,CAAC;oBACrD,MAAM,MAAM,GAAG,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,EAAE,MAAM,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;oBACtE,IAAI,MAAM,CAAC,OAAO,KAAK,CAAC,EAAE,CAAC;wBACzB,MAAM,IAAI,KAAK,CAAC,wCAAwC,MAAM,CAAC,SAAS,IAAI,OAAO,EAAE,CAAC,CAAC;oBACzF,CAAC;gBACH,CAAC;gBACD,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC,CAAC;gBACtD,KAAK,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC,IAAI,QAAQ,CAAC,OAAO,EAAE,EAAE,CAAC;oBAClD,IAAI,OAAO,KAAK,MAAM,CAAC,EAAE;wBAAE,MAAM,IAAI,KAAK,CAAC,yCAAyC,MAAM,CAAC,EAAE,EAAE,CAAC,CAAC;oBACjG,MAAM,KAAK,GAAG,WAAW,CAAC,GAAG,CAAC,OAAO,CAAuE,CAAC;oBAC7G,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,UAAU,KAAK,MAAM,CAAC,SAAS,EAAE,CAAC;wBACpD,MAAM,IAAI,KAAK,CAAC,mDAAmD,MAAM,CAAC,SAAS,IAAI,OAAO,EAAE,CAAC,CAAC;oBACpG,CAAC;oBACD,IAAI,KAAK,CAAC,WAAW,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;wBAC3C,MAAM,IAAI,KAAK,CAAC,kDAAkD,OAAO,OAAO,MAAM,CAAC,EAAE,EAAE,CAAC,CAAC;oBAC/F,CAAC;oBACD,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;;;;;;;;;WASnC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,MAAM,CAAC,EAAE,CAA6C,CAAC;oBACvE,IAAI,KAAK;wBAAE,MAAM,IAAI,KAAK,CAAC,sCAAsC,MAAM,CAAC,EAAE,OAAO,OAAO,EAAE,CAAC,CAAC;oBAC5F,UAAU,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,EAAE,OAAO,EAAE,KAAK,CAAC,CAAC;gBAC5C,CAAC;YACH,CAAC;YACD,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC/B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,IAAI,CAAC,QAAQ,CAAC,aAAa;gBAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAChE,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED,aAAa,CAAC,SAAiB,EAAE,mBAA4B;QAC3D,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;;;;;KAKpC,CAAC,CAAC,GAAG,CAAC,mBAAmB,IAAI,IAAI,EAAE,SAAS,CAAC,CAAC;QAC/C,OAAO,MAAM,CAAC,OAAO,KAAK,CAAC,CAAC;IAC9B,CAAC;IAED,UAAU,CAAC,SAAiB;QAC1B,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;;;KAGpC,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAClB,OAAO,MAAM,CAAC,OAAO,KAAK,CAAC,CAAC;IAC9B,CAAC;IAED,cAAc,CAAC,UAA6B;QAC1C,IAAI,OAAO,GAAG,CAAC,CAAC;QAChB,KAAK,MAAM,SAAS,IAAI,IAAI,GAAG,CAAC,UAAU,CAAC;YAAE,OAAO,IAAI,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC3F,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,sBAAsB,CAAC,SAAiB;QACtC,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;;;KAGpC,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAClB,OAAO,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAChC,CAAC;IAED,OAAO,CAAC,SAAiB;QACvB,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,cAAc,iDAAiD,CAAC;aAClG,GAAG,CAAC,SAAS,CAAsC,CAAC;QACvD,OAAO,GAAG,CAAC,CAAC,CAAC,WAAW,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAC3D,CAAC;IAED,SAAS,CAAC,SAAiB;QACzB,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;QAC9B,cAAc;;;;KAIjB,CAAC,CAAC,GAAG,CAAC,SAAS,CAAsC,CAAC;QACvD,OAAO,GAAG,CAAC,CAAC,CAAC,WAAW,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAC3D,CAAC;IAED,aAAa,CAAC,SAAiB;QAC7B,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;QAC/B,cAAc;;;KAGjB,CAAC,CAAC,GAAG,CAAC,SAAS,CAA4B,CAAC;QAC7C,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC,CAAC;IAC5D,CAAC;CACF"}

package/dist/persistence/sqlite.d.ts

@@ -0,0 +1,45 @@
+import type { Logger } from "../shared/logging.ts";
+import { type AppliedMigration } from "./migrations.ts";
+import { ArtifactRepository } from "./repositories/artifact-repository.ts";
+import { ContextManifestRepository } from "./repositories/context-manifest-repository.ts";
+import { MemoryRepository } from "./repositories/memory-repository.ts";
+import { ProjectKnowledgeRepository } from "./repositories/project-knowledge-repository.ts";
+import { SessionIndexRepository, type SessionIdentity, type SessionIndexStats } from "./repositories/session-index-repository.ts";
+import { SummaryRepository } from "./repositories/summary-repository.ts";
+export type SessionRecord = SessionIdentity;
+export type { SessionIndexStats } from "./repositories/session-index-repository.ts";
+export interface DatabaseHealth {
+    ok: boolean;
+    quickCheck: string;
+    journalMode: string;
+    foreignKeys: boolean;
+    schemaVersion: number;
+    appliedMigrations: number;
+}
+export interface OpenDatabaseOptions {
+    logger?: Logger;
+    now?: number;
+}
+export declare class ContextDatabase {
+    readonly path: string;
+    private readonly database;
+    private readonly logger;
+    readonly schemaVersion: number;
+    readonly migrations: readonly AppliedMigration[];
+    readonly sessionIndex: SessionIndexRepository;
+    readonly artifacts: ArtifactRepository;
+    readonly manifests: ContextManifestRepository;
+    readonly memory: MemoryRepository;
+    readonly summaries: SummaryRepository;
+    readonly projectKnowledge: ProjectKnowledgeRepository;
+    private closed;
+    private constructor();
+    static open(path: string, options?: OpenDatabaseOptions): ContextDatabase;
+    upsertSession(record: SessionIdentity): void;
+    getSessionStats(sessionId: string): SessionIndexStats;
+    listTables(): string[];
+    health(): DatabaseHealth;
+    close(): void;
+    private assertOpen;
+}
+//# sourceMappingURL=sqlite.d.ts.map

package/dist/persistence/sqlite.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"sqlite.d.ts","sourceRoot":"","sources":["../../src/persistence/sqlite.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,sBAAsB,CAAC;AAEnD,OAAO,EAA2C,KAAK,gBAAgB,EAAE,MAAM,iBAAiB,CAAC;AACjG,OAAO,EAAE,kBAAkB,EAAE,MAAM,uCAAuC,CAAC;AAC3E,OAAO,EAAE,yBAAyB,EAAE,MAAM,+CAA+C,CAAC;AAC1F,OAAO,EAAE,gBAAgB,EAAE,MAAM,qCAAqC,CAAC;AACvE,OAAO,EAAE,0BAA0B,EAAE,MAAM,gDAAgD,CAAC;AAC5F,OAAO,EACL,sBAAsB,EACtB,KAAK,eAAe,EACpB,KAAK,iBAAiB,EACvB,MAAM,4CAA4C,CAAC;AACpD,OAAO,EAAE,iBAAiB,EAAE,MAAM,sCAAsC,CAAC;AAEzE,MAAM,MAAM,aAAa,GAAG,eAAe,CAAC;AAC5C,YAAY,EAAE,iBAAiB,EAAE,MAAM,4CAA4C,CAAC;AAEpF,MAAM,WAAW,cAAc;IAC7B,EAAE,EAAE,OAAO,CAAC;IACZ,UAAU,EAAE,MAAM,CAAC;IACnB,WAAW,EAAE,MAAM,CAAC;IACpB,WAAW,EAAE,OAAO,CAAC;IACrB,aAAa,EAAE,MAAM,CAAC;IACtB,iBAAiB,EAAE,MAAM,CAAC;CAC3B;AAED,MAAM,WAAW,mBAAmB;IAClC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,GAAG,CAAC,EAAE,MAAM,CAAC;CACd;AAoBD,qBAAa,eAAe;IAYxB,QAAQ,CAAC,IAAI,EAAE,MAAM;IACrB,OAAO,CAAC,QAAQ,CAAC,QAAQ;IACzB,OAAO,CAAC,QAAQ,CAAC,MAAM;IAbzB,QAAQ,CAAC,aAAa,SAA0B;IAChD,QAAQ,CAAC,UAAU,EAAE,SAAS,gBAAgB,EAAE,CAAC;IACjD,QAAQ,CAAC,YAAY,EAAE,sBAAsB,CAAC;IAC9C,QAAQ,CAAC,SAAS,EAAE,kBAAkB,CAAC;IACvC,QAAQ,CAAC,SAAS,EAAE,yBAAyB,CAAC;IAC9C,QAAQ,CAAC,MAAM,EAAE,gBAAgB,CAAC;IAClC,QAAQ,CAAC,SAAS,EAAE,iBAAiB,CAAC;IACtC,QAAQ,CAAC,gBAAgB,EAAE,0BAA0B,CAAC;IACtD,OAAO,CAAC,MAAM,CAAS;IAEvB,OAAO;IAeP,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,GAAE,mBAAwB,GAAG,eAAe;IAwC7E,aAAa,CAAC,MAAM,EAAE,eAAe,GAAG,IAAI;IAK5C,eAAe,CAAC,SAAS,EAAE,MAAM,GAAG,iBAAiB;IAKrD,UAAU,IAAI,MAAM,EAAE;IAUtB,MAAM,IAAI,cAAc;IAiBxB,KAAK,IAAI,IAAI;IAOb,OAAO,CAAC,UAAU;CAGnB"}

package/dist/persistence/sqlite.js

@@ -0,0 +1,137 @@
+import { chmodSync, mkdirSync } from "node:fs";
+import { dirname } from "node:path";
+import { DatabaseSync } from "node:sqlite";
+import { silentLogger } from "../shared/logging.js";
+import { applyMigrations, CURRENT_SCHEMA_VERSION } from "./migrations.js";
+import { ArtifactRepository } from "./repositories/artifact-repository.js";
+import { ContextManifestRepository } from "./repositories/context-manifest-repository.js";
+import { MemoryRepository } from "./repositories/memory-repository.js";
+import { ProjectKnowledgeRepository } from "./repositories/project-knowledge-repository.js";
+import { SessionIndexRepository, } from "./repositories/session-index-repository.js";
+import { SummaryRepository } from "./repositories/summary-repository.js";
+function firstColumn(row) {
+    if (!row || typeof row !== "object")
+        throw new Error("SQLite PRAGMA returned no row");
+    const value = Object.values(row)[0];
+    if (typeof value !== "string" && typeof value !== "number") {
+        throw new Error("SQLite PRAGMA returned an unexpected value");
+    }
+    return value;
+}
+function bestEffortChmod(path, mode) {
+    try {
+        chmodSync(path, mode);
+    }
+    catch (error) {
+        const code = error && typeof error === "object" && "code" in error ? String(error.code) : "";
+        if (code !== "ENOSYS" && code !== "EPERM" && code !== "EINVAL")
+            throw error;
+    }
+}
+export class ContextDatabase {
+    path;
+    database;
+    logger;
+    schemaVersion = CURRENT_SCHEMA_VERSION;
+    migrations;
+    sessionIndex;
+    artifacts;
+    manifests;
+    memory;
+    summaries;
+    projectKnowledge;
+    closed = false;
+    constructor(path, database, logger, migrations) {
+        this.path = path;
+        this.database = database;
+        this.logger = logger;
+        this.migrations = migrations;
+        this.sessionIndex = new SessionIndexRepository(database);
+        this.artifacts = new ArtifactRepository(database);
+        this.manifests = new ContextManifestRepository(database);
+        this.memory = new MemoryRepository(database);
+        this.summaries = new SummaryRepository(database);
+        this.projectKnowledge = new ProjectKnowledgeRepository(database);
+    }
+    static open(path, options = {}) {
+        const logger = options.logger ?? silentLogger;
+        const memory = path === ":memory:";
+        if (!memory) {
+            const directory = dirname(path);
+            mkdirSync(directory, { recursive: true, mode: 0o700 });
+            bestEffortChmod(directory, 0o700);
+        }
+        const database = new DatabaseSync(path, {
+            enableForeignKeyConstraints: true,
+            enableDoubleQuotedStringLiterals: false,
+            allowExtension: false,
+            timeout: 5000,
+        });
+        try {
+            database.exec("PRAGMA foreign_keys = ON");
+            database.exec("PRAGMA busy_timeout = 5000");
+            database.exec("PRAGMA synchronous = NORMAL");
+            database.exec("PRAGMA trusted_schema = OFF");
+            database.exec("PRAGMA secure_delete = FAST");
+            if (!memory)
+                database.exec("PRAGMA journal_mode = WAL");
+            const migrations = applyMigrations(database, options.now);
+            if (!memory)
+                bestEffortChmod(path, 0o600);
+            logger.info("database.opened", {
+                databasePath: path,
+                schemaVersion: CURRENT_SCHEMA_VERSION,
+                migrations: migrations.length,
+            });
+            return new ContextDatabase(path, database, logger, migrations);
+        }
+        catch (error) {
+            database.close();
+            throw error;
+        }
+    }
+    upsertSession(record) {
+        this.assertOpen();
+        this.sessionIndex.upsertSession(record);
+    }
+    getSessionStats(sessionId) {
+        this.assertOpen();
+        return this.sessionIndex.getStats(sessionId);
+    }
+    listTables() {
+        this.assertOpen();
+        const rows = this.database.prepare(`
+      SELECT name FROM sqlite_schema
+      WHERE type IN ('table', 'view') AND name NOT LIKE 'sqlite_%'
+      ORDER BY name
+    `).all();
+        return rows.map((row) => row.name);
+    }
+    health() {
+        this.assertOpen();
+        const quickCheck = String(firstColumn(this.database.prepare("PRAGMA quick_check").get()));
+        const journalMode = String(firstColumn(this.database.prepare("PRAGMA journal_mode").get()));
+        const foreignKeys = Number(firstColumn(this.database.prepare("PRAGMA foreign_keys").get())) === 1;
+        const schemaVersion = Number(firstColumn(this.database.prepare("PRAGMA user_version").get()));
+        return {
+            ok: quickCheck === "ok" && foreignKeys && schemaVersion === CURRENT_SCHEMA_VERSION,
+            quickCheck,
+            journalMode,
+            foreignKeys,
+            schemaVersion,
+            appliedMigrations: this.migrations.length,
+        };
+    }
+    close() {
+        if (this.closed)
+            return;
+        this.closed = true;
+        this.database.close();
+        this.logger.info("database.closed", { databasePath: this.path });
+    }
+    assertOpen() {
+        if (this.closed)
+            throw new Error("Context database is closed");
+    }
+}
+//# sourceMappingURL=sqlite.js.map

package/dist/persistence/sqlite.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"sqlite.js","sourceRoot":"","sources":["../../src/persistence/sqlite.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AAC/C,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAE3C,OAAO,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAC;AACpD,OAAO,EAAE,eAAe,EAAE,sBAAsB,EAAyB,MAAM,iBAAiB,CAAC;AACjG,OAAO,EAAE,kBAAkB,EAAE,MAAM,uCAAuC,CAAC;AAC3E,OAAO,EAAE,yBAAyB,EAAE,MAAM,+CAA+C,CAAC;AAC1F,OAAO,EAAE,gBAAgB,EAAE,MAAM,qCAAqC,CAAC;AACvE,OAAO,EAAE,0BAA0B,EAAE,MAAM,gDAAgD,CAAC;AAC5F,OAAO,EACL,sBAAsB,GAGvB,MAAM,4CAA4C,CAAC;AACpD,OAAO,EAAE,iBAAiB,EAAE,MAAM,sCAAsC,CAAC;AAmBzE,SAAS,WAAW,CAAC,GAAY;IAC/B,IAAI,CAAC,GAAG,IAAI,OAAO,GAAG,KAAK,QAAQ;QAAE,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAC;IACtF,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;IACpC,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC3D,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAC;IAChE,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,eAAe,CAAC,IAAY,EAAE,IAAY;IACjD,IAAI,CAAC;QACH,SAAS,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;IACxB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,IAAI,GAAG,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,MAAM,IAAI,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAC7F,IAAI,IAAI,KAAK,QAAQ,IAAI,IAAI,KAAK,OAAO,IAAI,IAAI,KAAK,QAAQ;YAAE,MAAM,KAAK,CAAC;IAC9E,CAAC;AACH,CAAC;AAED,MAAM,OAAO,eAAe;IAYf;IACQ;IACA;IAbV,aAAa,GAAG,sBAAsB,CAAC;IACvC,UAAU,CAA8B;IACxC,YAAY,CAAyB;IACrC,SAAS,CAAqB;IAC9B,SAAS,CAA4B;IACrC,MAAM,CAAmB;IACzB,SAAS,CAAoB;IAC7B,gBAAgB,CAA6B;IAC9C,MAAM,GAAG,KAAK,CAAC;IAEvB,YACW,IAAY,EACJ,QAAsB,EACtB,MAAc,EAC/B,UAA8B;QAHrB,SAAI,GAAJ,IAAI,CAAQ;QACJ,aAAQ,GAAR,QAAQ,CAAc;QACtB,WAAM,GAAN,MAAM,CAAQ;QAG/B,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,YAAY,GAAG,IAAI,sBAAsB,CAAC,QAAQ,CAAC,CAAC;QACzD,IAAI,CAAC,SAAS,GAAG,IAAI,kBAAkB,CAAC,QAAQ,CAAC,CAAC;QAClD,IAAI,CAAC,SAAS,GAAG,IAAI,yBAAyB,CAAC,QAAQ,CAAC,CAAC;QACzD,IAAI,CAAC,MAAM,GAAG,IAAI,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QAC7C,IAAI,CAAC,SAAS,GAAG,IAAI,iBAAiB,CAAC,QAAQ,CAAC,CAAC;QACjD,IAAI,CAAC,gBAAgB,GAAG,IAAI,0BAA0B,CAAC,QAAQ,CAAC,CAAC;IACnE,CAAC;IAED,MAAM,CAAC,IAAI,CAAC,IAAY,EAAE,UAA+B,EAAE;QACzD,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,YAAY,CAAC;QAC9C,MAAM,MAAM,GAAG,IAAI,KAAK,UAAU,CAAC;QAEnC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;YAChC,SAAS,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;YACvD,eAAe,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;QACpC,CAAC;QAED,MAAM,QAAQ,GAAG,IAAI,YAAY,CAAC,IAAI,EAAE;YACtC,2BAA2B,EAAE,IAAI;YACjC,gCAAgC,EAAE,KAAK;YACvC,cAAc,EAAE,KAAK;YACrB,OAAO,EAAE,IAAI;SACd,CAAC,CAAC;QAEH,IAAI,CAAC;YACH,QAAQ,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;YAC1C,QAAQ,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;YAC5C,QAAQ,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;YAC7C,QAAQ,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;YAC7C,QAAQ,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;YAC7C,IAAI,CAAC,MAAM;gBAAE,QAAQ,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAC;YAExD,MAAM,UAAU,GAAG,eAAe,CAAC,QAAQ,EAAE,OAAO,CAAC,GAAG,CAAC,CAAC;YAC1D,IAAI,CAAC,MAAM;gBAAE,eAAe,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;YAE1C,MAAM,CAAC,IAAI,CAAC,iBAAiB,EAAE;gBAC7B,YAAY,EAAE,IAAI;gBAClB,aAAa,EAAE,sBAAsB;gBACrC,UAAU,EAAE,UAAU,CAAC,MAAM;aAC9B,CAAC,CAAC;YACH,OAAO,IAAI,eAAe,CAAC,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC;QACjE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,QAAQ,CAAC,KAAK,EAAE,CAAC;YACjB,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED,aAAa,CAAC,MAAuB;QACnC,IAAI,CAAC,UAAU,EAAE,CAAC;QAClB,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;IAC1C,CAAC;IAED,eAAe,CAAC,SAAiB;QAC/B,IAAI,CAAC,UAAU,EAAE,CAAC;QAClB,OAAO,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;IAC/C,CAAC;IAED,UAAU;QACR,IAAI,CAAC,UAAU,EAAE,CAAC;QAClB,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;;;;KAIlC,CAAC,CAAC,GAAG,EAAwC,CAAC;QAC/C,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACrC,CAAC;IAED,MAAM;QACJ,IAAI,CAAC,UAAU,EAAE,CAAC;QAClB,MAAM,UAAU,GAAG,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,oBAAoB,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;QAC1F,MAAM,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;QAC5F,MAAM,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC;QAClG,MAAM,aAAa,GAAG,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;QAE9F,OAAO;YACL,EAAE,EAAE,UAAU,KAAK,IAAI,IAAI,WAAW,IAAI,aAAa,KAAK,sBAAsB;YAClF,UAAU;YACV,WAAW;YACX,WAAW;YACX,aAAa;YACb,iBAAiB,EAAE,IAAI,CAAC,UAAU,CAAC,MAAM;SAC1C,CAAC;IACJ,CAAC;IAED,KAAK;QACH,IAAI,IAAI,CAAC,MAAM;YAAE,OAAO;QACxB,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACnB,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;QACtB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,iBAAiB,EAAE,EAAE,YAAY,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;IACnE,CAAC;IAEO,UAAU;QAChB,IAAI,IAAI,CAAC,MAAM;YAAE,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;IACjE,CAAC;CACF"}

package/dist/planner/atomic-groups.d.ts

@@ -0,0 +1,15 @@
+export type AtomicGroupKind = "turn" | "summary" | "prefix";
+export interface AtomicMessageGroup {
+    id: string;
+    kind: AtomicGroupKind;
+    messageIndices: number[];
+    startIndex: number;
+    endIndex: number;
+    estimatedTokens: number;
+    containsToolExchange: boolean;
+    reason: string;
+}
+export declare function messageRole(message: unknown): string | undefined;
+export declare function buildAtomicGroups(messages: readonly unknown[]): AtomicMessageGroup[];
+export declare function validateAtomicSelection(messages: readonly unknown[], selectedIndices: ReadonlySet<number>): string[];
+//# sourceMappingURL=atomic-groups.d.ts.map

package/dist/planner/atomic-groups.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"atomic-groups.d.ts","sourceRoot":"","sources":["../../src/planner/atomic-groups.ts"],"names":[],"mappings":"AAEA,MAAM,MAAM,eAAe,GAAG,MAAM,GAAG,SAAS,GAAG,QAAQ,CAAC;AAE5D,MAAM,WAAW,kBAAkB;IACjC,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,eAAe,CAAC;IACtB,cAAc,EAAE,MAAM,EAAE,CAAC;IACzB,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,eAAe,EAAE,MAAM,CAAC;IACxB,oBAAoB,EAAE,OAAO,CAAC;IAC9B,MAAM,EAAE,MAAM,CAAC;CAChB;AAgBD,wBAAgB,WAAW,CAAC,OAAO,EAAE,OAAO,GAAG,MAAM,GAAG,SAAS,CAGhE;AA0ED,wBAAgB,iBAAiB,CAAC,QAAQ,EAAE,SAAS,OAAO,EAAE,GAAG,kBAAkB,EAAE,CA+EpF;AAED,wBAAgB,uBAAuB,CACrC,QAAQ,EAAE,SAAS,OAAO,EAAE,EAC5B,eAAe,EAAE,WAAW,CAAC,MAAM,CAAC,GACnC,MAAM,EAAE,CAyBV"}

package/dist/planner/atomic-groups.js

@@ -0,0 +1,177 @@
+import { estimateMessageTokens } from "../core/token-estimator.js";
+function isRecord(value) {
+    return value !== null && typeof value === "object" && !Array.isArray(value);
+}
+export function messageRole(message) {
+    if (!isRecord(message))
+        return undefined;
+    return typeof message.role === "string" ? message.role : undefined;
+}
+function isSummaryMessage(message) {
+    const role = messageRole(message);
+    return role === "compactionSummary" || role === "branchSummary";
+}
+function toolRelations(messages) {
+    const calls = new Map();
+    const results = new Map();
+    for (let index = 0; index < messages.length; index++) {
+        const message = messages[index];
+        if (!isRecord(message))
+            continue;
+        if (message.role === "assistant" && Array.isArray(message.content)) {
+            for (const block of message.content) {
+                if (!isRecord(block) || block.type !== "toolCall" || typeof block.id !== "string")
+                    continue;
+                calls.set(block.id, index);
+            }
+        }
+        if (message.role === "toolResult" && typeof message.toolCallId === "string") {
+            const indices = results.get(message.toolCallId) ?? [];
+            indices.push(index);
+            results.set(message.toolCallId, indices);
+        }
+    }
+    return { calls, results };
+}
+function initialGroups(messages) {
+    const groups = [];
+    let activeTurn;
+    let activePrefix;
+    for (let index = 0; index < messages.length; index++) {
+        const message = messages[index];
+        if (isSummaryMessage(message)) {
+            groups.push({ kind: "summary", messageIndices: [index] });
+            activeTurn = undefined;
+            activePrefix = undefined;
+            continue;
+        }
+        if (messageRole(message) === "user") {
+            activeTurn = { kind: "turn", messageIndices: [index] };
+            groups.push(activeTurn);
+            activePrefix = undefined;
+            continue;
+        }
+        if (activeTurn) {
+            activeTurn.messageIndices.push(index);
+            continue;
+        }
+        if (!activePrefix) {
+            activePrefix = { kind: "prefix", messageIndices: [] };
+            groups.push(activePrefix);
+        }
+        activePrefix.messageIndices.push(index);
+    }
+    return groups;
+}
+function mergedKind(groups, members) {
+    if (members.some((index) => groups[index]?.kind === "turn"))
+        return "turn";
+    if (members.every((index) => groups[index]?.kind === "summary"))
+        return "summary";
+    return "prefix";
+}
+export function buildAtomicGroups(messages) {
+    const drafts = initialGroups(messages);
+    if (drafts.length === 0)
+        return [];
+    const parents = drafts.map((_, index) => index);
+    const messageGroups = new Map();
+    drafts.forEach((group, groupIndex) => {
+        for (const messageIndex of group.messageIndices)
+            messageGroups.set(messageIndex, groupIndex);
+    });
+    const find = (index) => {
+        let root = index;
+        while (parents[root] !== root)
+            root = parents[root] ?? root;
+        while (parents[index] !== index) {
+            const parent = parents[index] ?? index;
+            parents[index] = root;
+            index = parent;
+        }
+        return root;
+    };
+    const union = (left, right) => {
+        const leftRoot = find(left);
+        const rightRoot = find(right);
+        if (leftRoot !== rightRoot)
+            parents[Math.max(leftRoot, rightRoot)] = Math.min(leftRoot, rightRoot);
+    };
+    const relations = toolRelations(messages);
+    for (const [toolCallId, resultIndices] of relations.results) {
+        const callIndex = relations.calls.get(toolCallId);
+        if (callIndex === undefined)
+            continue;
+        const callGroup = messageGroups.get(callIndex);
+        if (callGroup === undefined)
+            continue;
+        for (const resultIndex of resultIndices) {
+            const resultGroup = messageGroups.get(resultIndex);
+            if (resultGroup !== undefined)
+                union(callGroup, resultGroup);
+        }
+    }
+    const merged = new Map();
+    drafts.forEach((group, groupIndex) => {
+        const root = find(groupIndex);
+        const aggregate = merged.get(root) ?? { draftIndices: [], messageIndices: [] };
+        aggregate.draftIndices.push(groupIndex);
+        aggregate.messageIndices.push(...group.messageIndices);
+        merged.set(root, aggregate);
+    });
+    return [...merged.values()]
+        .map((aggregate) => {
+        const messageIndices = [...aggregate.messageIndices].sort((left, right) => left - right);
+        const startIndex = messageIndices[0] ?? 0;
+        const endIndex = messageIndices.at(-1) ?? startIndex;
+        const selected = new Set(messageIndices);
+        const containsToolExchange = [...relations.results].some(([id, resultIndices]) => {
+            const callIndex = relations.calls.get(id);
+            return callIndex !== undefined && selected.has(callIndex) && resultIndices.some((index) => selected.has(index));
+        });
+        const kind = mergedKind(drafts, aggregate.draftIndices);
+        return {
+            id: `group:${startIndex}-${endIndex}`,
+            kind,
+            messageIndices,
+            startIndex,
+            endIndex,
+            estimatedTokens: messageIndices.reduce((total, index) => total + estimateMessageTokens(messages[index]), 0),
+            containsToolExchange,
+            reason: containsToolExchange
+                ? "Atomic conversation turn containing a complete tool call/result batch"
+                : kind === "turn"
+                    ? "Conversation turn boundary"
+                    : kind === "summary"
+                        ? "Active Pi summary message"
+                        : "Messages preceding the first user turn",
+        };
+    })
+        .sort((left, right) => left.startIndex - right.startIndex);
+}
+export function validateAtomicSelection(messages, selectedIndices) {
+    const relations = toolRelations(messages);
+    const errors = [];
+    for (const [toolCallId, callIndex] of relations.calls) {
+        if (!selectedIndices.has(callIndex))
+            continue;
+        const resultIndices = relations.results.get(toolCallId) ?? [];
+        if (resultIndices.length === 0) {
+            errors.push(`selected tool call ${toolCallId} has no result`);
+            continue;
+        }
+        if (resultIndices.some((index) => !selectedIndices.has(index))) {
+            errors.push(`selected tool call ${toolCallId} is missing one or more results`);
+        }
+    }
+    for (const [toolCallId, resultIndices] of relations.results) {
+        if (!resultIndices.some((index) => selectedIndices.has(index)))
+            continue;
+        const callIndex = relations.calls.get(toolCallId);
+        if (callIndex === undefined || !selectedIndices.has(callIndex)) {
+            errors.push(`selected tool result ${toolCallId} has no selected call`);
+        }
+    }
+    return [...new Set(errors)];
+}
+//# sourceMappingURL=atomic-groups.js.map