@cortexkit/pi-magic-context 0.49.16 → 0.49.17
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/auto-search-worker.js +76 -63
- package/dist/chunk-embedding-write-worker.js +10 -10
- package/dist/{compaction-marker-candidates-k4tqg4jb.js → compaction-marker-candidates-wjzqbspm.js} +2 -2
- package/dist/compaction-marker-candidates-worker.js +2 -2
- package/dist/embedding-backfill-vwag34t0.js +23 -0
- package/dist/embedding-worker-process.js +1 -1
- package/dist/embedding-worker.js +4 -4
- package/dist/historian-calibration-extension.js +4 -1
- package/dist/{index-g4jf772p.js → index-1kcfwk9c.js} +22 -2
- package/dist/{index-657xh9w4.js → index-2mmy81qt.js} +2 -2
- package/dist/{index-v8d4vgam.js → index-469ttgzv.js} +6 -6
- package/dist/index-4knyh1kt.js +1387 -0
- package/dist/{index-44h64qza.js → index-5k2mshvq.js} +9 -9
- package/dist/{index-fhyy4ave.js → index-8r2t8fv2.js} +842 -73
- package/dist/{index-xmzhrht0.js → index-9j809q6f.js} +29 -18
- package/dist/{index-6dz2ed1v.js → index-d8gc0112.js} +2 -2
- package/dist/{index-x3k1530z.js → index-dvbjsh98.js} +3 -3
- package/dist/{index-hy44m6jc.js → index-e1rza8gp.js} +1 -1
- package/dist/{index-dne4ja7a.js → index-ej4eak9d.js} +2 -2
- package/dist/{index-4178cbfw.js → index-es03azap.js} +121 -2038
- package/dist/{index-ajfpyk0x.js → index-fa485mqs.js} +59 -1
- package/dist/{index-y5c73caa.js → index-g7p75667.js} +32 -3
- package/dist/{index-tympym62.js → index-gdjv1nnd.js} +2 -2
- package/dist/{index-46q4e104.js → index-qpq6w67c.js} +1 -1
- package/dist/{index-asrxgdt4.js → index-rrwv9y2k.js} +8 -0
- package/dist/{index-mx59cve4.js → index-svgkv4g2.js} +1 -1
- package/dist/{index-qqxqwqhp.js → index-vktahpdj.js} +1 -1
- package/dist/{index-n3r2d2pr.js → index-vmcgew7x.js} +10 -7
- package/dist/{index-afy3dzf9.js → index-xnch62s6.js} +12 -6
- package/dist/{index-s1xn8qfw.js → index-yygsh681.js} +6 -3
- package/dist/{index-p3evwech.js → index-z2rkv0mb.js} +3 -3
- package/dist/{index-ja1c33qa.js → index-z7kbck29.js} +2 -2
- package/dist/index.js +523 -227
- package/dist/{message-fts-rowid-map-mbanfkt1.js → message-fts-rowid-map-cvpy2dmb.js} +5 -3
- package/dist/{message-index-easvmz22.js → message-index-2g23b7fe.js} +11 -11
- package/dist/{message-index-bounded-xzerjxmp.js → message-index-bounded-d6zj2tn5.js} +15 -15
- package/dist/{message-time-backfill-3q4kemfc.js → message-time-backfill-ssshbdy8.js} +1 -1
- package/dist/migration-worker.js +7 -7
- package/dist/{project-embedding-registry-xwnmqmx8.js → project-embedding-registry-vf1ncptn.js} +10 -10
- package/dist/raw-ordinal-warmup-worker.js +10 -10
- package/dist/{read-session-chunk-1sb5n7zs.js → read-session-chunk-2ky7mbqp.js} +5 -5
- package/dist/{read-session-raw-jz5e9jz0.js → read-session-raw-9s3xn7fc.js} +2 -2
- package/dist/{search-archive-3ma5n0wr.js → search-archive-9akqcr90.js} +6 -6
- package/dist/{search-x3fa0561.js → search-zjyc65g1.js} +17 -17
- package/dist/{sqlite-adjpy399.js → sqlite-fsp0sf3r.js} +1 -1
- package/dist/{storage-memory-axnsk5t1.js → storage-memory-yxxc5v4n.js} +2 -2
- package/dist/{storage-tags-jtessrec.js → storage-tags-cx6ccsv1.js} +3 -3
- package/dist/subagent-entry.js +22 -21
- package/dist/tag-snapshot-worker-pi.js +9 -9
- package/package.json +1 -1
- package/dist/embedding-backfill-c4bdw8pp.js +0 -23
|
@@ -9,23 +9,32 @@ import {
|
|
|
9
9
|
readProcessStartTime
|
|
10
10
|
} from "./index-apgwdby5.js";
|
|
11
11
|
import {
|
|
12
|
-
|
|
13
|
-
|
|
14
|
-
|
|
15
|
-
} from "./index-g4jf772p.js";
|
|
12
|
+
resolveOpenCodeDbPath,
|
|
13
|
+
assertOpenCodeStoreGeneration
|
|
14
|
+
} from "./index-8976mezv.js";
|
|
16
15
|
import {
|
|
16
|
+
Database2,
|
|
17
|
+
getSqliteDatabasePath2,
|
|
18
|
+
prepareCachedStatement2,
|
|
17
19
|
withSqliteBackgroundWriter2,
|
|
18
20
|
withoutSqliteTransformPass2
|
|
19
|
-
} from "./index-
|
|
21
|
+
} from "./index-rrwv9y2k.js";
|
|
22
|
+
import {
|
|
23
|
+
removeSystemReminders,
|
|
24
|
+
removeNestedBlocks,
|
|
25
|
+
hasMeaningfulUserText2,
|
|
26
|
+
extractTexts2
|
|
27
|
+
} from "./index-1kcfwk9c.js";
|
|
20
28
|
import {
|
|
21
29
|
recordMessageFtsRowid2,
|
|
22
30
|
recordIndexedMessageTime2,
|
|
23
31
|
messageFtsOrdinalRangeIsMapped2,
|
|
32
|
+
getMessageFtsRowidMapBackfillProgress2,
|
|
24
33
|
deleteUnmappedMessageFtsRows2
|
|
25
|
-
} from "./index-
|
|
34
|
+
} from "./index-yygsh681.js";
|
|
26
35
|
import {
|
|
27
36
|
deleteChunkEmbedBackoffForSession
|
|
28
|
-
} from "./index-
|
|
37
|
+
} from "./index-xnch62s6.js";
|
|
29
38
|
import {
|
|
30
39
|
closeQuietly,
|
|
31
40
|
TAG_TOMBSTONE_OWNER_PREFIX,
|
|
@@ -37,7 +46,10 @@ import {
|
|
|
37
46
|
} from "./index-jrap43gc.js";
|
|
38
47
|
import {
|
|
39
48
|
cleanUserText2
|
|
40
|
-
} from "./index-
|
|
49
|
+
} from "./index-5k2mshvq.js";
|
|
50
|
+
import {
|
|
51
|
+
sameProjectDirectory
|
|
52
|
+
} from "./index-wrr9rp9a.js";
|
|
41
53
|
import {
|
|
42
54
|
logSlowWriteTransaction
|
|
43
55
|
} from "./index-d4dt19md.js";
|
|
@@ -51,7 +63,7 @@ import {
|
|
|
51
63
|
|
|
52
64
|
// ../plugin/src/hooks/magic-context/lkg-recovery-fence.ts
|
|
53
65
|
import { createHash as createHash2, randomUUID as randomUUID2 } from "node:crypto";
|
|
54
|
-
import { lstatSync, readdirSync as readdirSync2, readFileSync as readFileSync2, rmSync as rmSync2 } from "node:fs";
|
|
66
|
+
import { lstatSync, readdirSync as readdirSync2, readFileSync as readFileSync2, rmdirSync, rmSync as rmSync2 } from "node:fs";
|
|
55
67
|
import { basename, dirname, join as join2 } from "node:path";
|
|
56
68
|
|
|
57
69
|
// ../plugin/src/shared/bounded-session-map.ts
|
|
@@ -141,7 +153,16 @@ function withStorageFileMutex(directory, work, timeoutMs = 1000) {
|
|
|
141
153
|
const birth = processBirth(process.pid);
|
|
142
154
|
const ticket = { choosing: true, number: 0, birth, checksum: "" };
|
|
143
155
|
const encodedTicket = () => JSON.stringify({ ...ticket, checksum: ticketChecksum(name, ticket) });
|
|
144
|
-
|
|
156
|
+
for (let attempt = 0;; attempt++) {
|
|
157
|
+
try {
|
|
158
|
+
writeStorageFileSync(ownPath, encodedTicket(), { flag: "wx", forcePrivate: true });
|
|
159
|
+
break;
|
|
160
|
+
} catch (error) {
|
|
161
|
+
if (error.code !== "ENOENT" || attempt >= 2)
|
|
162
|
+
throw error;
|
|
163
|
+
ensurePrivateStorageDirectorySync(directory);
|
|
164
|
+
}
|
|
165
|
+
}
|
|
145
166
|
const deadline = performance.now() + timeoutMs;
|
|
146
167
|
const peers = () => {
|
|
147
168
|
const found = [];
|
|
@@ -438,15 +459,610 @@ function commitLkgWithRecoveryFence(args) {
|
|
|
438
459
|
function clearLkgRecoveryFence(sessionId, remove, shouldClear) {
|
|
439
460
|
if (!durableFencesSupported())
|
|
440
461
|
return remove?.();
|
|
441
|
-
|
|
442
|
-
|
|
443
|
-
|
|
444
|
-
|
|
445
|
-
|
|
446
|
-
|
|
462
|
+
const path = lkgRecoveryFencePath(sessionId);
|
|
463
|
+
const missing = (file) => {
|
|
464
|
+
try {
|
|
465
|
+
lstatSync(file);
|
|
466
|
+
return false;
|
|
467
|
+
} catch (error) {
|
|
468
|
+
return error.code === "ENOENT";
|
|
469
|
+
}
|
|
470
|
+
};
|
|
471
|
+
if (!remove && !shouldClear && missing(path) && missing(`${path}.locks`)) {
|
|
447
472
|
clearLkgFenceProcessState(sessionId);
|
|
473
|
+
return;
|
|
474
|
+
}
|
|
475
|
+
try {
|
|
476
|
+
return locked(sessionId, () => {
|
|
477
|
+
const result = remove?.();
|
|
478
|
+
if (shouldClear && !shouldClear(result))
|
|
479
|
+
return result;
|
|
480
|
+
rmSync2(lkgRecoveryFencePath(sessionId), { force: true });
|
|
481
|
+
syncStorageDirectorySync(dirname(lkgRecoveryFencePath(sessionId)));
|
|
482
|
+
clearLkgFenceProcessState(sessionId);
|
|
483
|
+
return result;
|
|
484
|
+
});
|
|
485
|
+
} finally {
|
|
486
|
+
try {
|
|
487
|
+
rmdirSync(`${path}.locks`);
|
|
488
|
+
} catch {}
|
|
489
|
+
}
|
|
490
|
+
}
|
|
491
|
+
|
|
492
|
+
// ../plugin/src/v2/store-reader.ts
|
|
493
|
+
import { createHash as createHash3 } from "node:crypto";
|
|
494
|
+
import { statSync } from "node:fs";
|
|
495
|
+
import { hostname } from "node:os";
|
|
496
|
+
import { resolve } from "node:path";
|
|
497
|
+
function gaDatabasePath(dataHome, channel = "latest", env = process.env) {
|
|
498
|
+
return resolveOpenCodeDbPath("v2", { dataHome, channel, env }).path;
|
|
499
|
+
}
|
|
500
|
+
var RAW_MESSAGE_TYPES = [
|
|
501
|
+
"user",
|
|
502
|
+
"synthetic",
|
|
503
|
+
"assistant",
|
|
504
|
+
"skill",
|
|
505
|
+
"shell",
|
|
506
|
+
"system"
|
|
507
|
+
];
|
|
508
|
+
var RAW_MESSAGE_TYPE_PARAMETERS = RAW_MESSAGE_TYPES.map(() => "?").join(", ");
|
|
509
|
+
var V2_MESSAGE_PAGE_SQL = `WITH bounds AS (
|
|
510
|
+
SELECT
|
|
511
|
+
CASE WHEN ? = 0 THEN -1 ELSE COALESCE((
|
|
512
|
+
SELECT seq FROM session_message
|
|
513
|
+
WHERE session_id = ? AND type IN (${RAW_MESSAGE_TYPE_PARAMETERS})
|
|
514
|
+
ORDER BY seq ASC LIMIT 1 OFFSET ?
|
|
515
|
+
), (SELECT MAX(seq) FROM session_message WHERE session_id = ?), -1) END AS after_seq,
|
|
516
|
+
COALESCE((
|
|
517
|
+
SELECT seq FROM session_message
|
|
518
|
+
WHERE session_id = ? AND type IN (${RAW_MESSAGE_TYPE_PARAMETERS})
|
|
519
|
+
ORDER BY seq ASC LIMIT 1 OFFSET ?
|
|
520
|
+
), ?) AS watermark_seq
|
|
521
|
+
)
|
|
522
|
+
SELECT id, session_id, type, seq, time_created, data FROM session_message, bounds
|
|
523
|
+
WHERE session_id = ?
|
|
524
|
+
AND type IN (${RAW_MESSAGE_TYPE_PARAMETERS})
|
|
525
|
+
AND seq > bounds.after_seq
|
|
526
|
+
AND seq <= bounds.watermark_seq
|
|
527
|
+
ORDER BY seq ASC LIMIT ?`;
|
|
528
|
+
var V2_STORE_READER_DEBUG_COUNTER_KEY = "magic-context.v2.store-reader-debug";
|
|
529
|
+
var debugSymbol = Symbol.for(V2_STORE_READER_DEBUG_COUNTER_KEY);
|
|
530
|
+
var debugGlobal = globalThis;
|
|
531
|
+
var QUERY_STATEMENTS = {
|
|
532
|
+
messagePage: V2_MESSAGE_PAGE_SQL,
|
|
533
|
+
messageCount: "SELECT COUNT(*) AS count FROM session_message WHERE session_id = ? AND type IN (?)",
|
|
534
|
+
storedMessageCount: "SELECT COUNT(*) AS count FROM session_message WHERE session_id = ?",
|
|
535
|
+
messageById: "SELECT id, session_id, type, seq, time_created, data FROM session_message WHERE session_id = ? AND id = ? AND type IN (?) LIMIT 1",
|
|
536
|
+
messageExistsById: "SELECT 1 FROM session_message WHERE id = ? AND session_id = ? AND type IN (?) LIMIT 1",
|
|
537
|
+
messageOrdinalById: `WITH target AS (
|
|
538
|
+
SELECT seq FROM session_message WHERE session_id = ? AND id = ? AND type IN (?) LIMIT 1
|
|
539
|
+
)
|
|
540
|
+
SELECT (SELECT COUNT(*) FROM session_message WHERE session_id = ? AND type IN (?) AND seq <= target.seq) AS ordinal FROM target`,
|
|
541
|
+
messageIdOrdinals: "SELECT id FROM session_message WHERE session_id = ? AND type IN (?) ORDER BY seq ASC LIMIT ? OFFSET ?",
|
|
542
|
+
messageOrdinalPage: "SELECT id, type, seq, json_valid(data) AS valid FROM session_message WHERE session_id = ? AND seq > ? ORDER BY seq ASC, id ASC LIMIT ?",
|
|
543
|
+
rawRowsThrough: "SELECT id, session_id, type, seq, time_created, data FROM session_message WHERE session_id = ? AND type IN (?) AND seq <= ? ORDER BY seq DESC LIMIT ?"
|
|
544
|
+
};
|
|
545
|
+
function debugCounters() {
|
|
546
|
+
const counters = debugGlobal[debugSymbol] ??= {
|
|
547
|
+
decodedRows: 0,
|
|
548
|
+
operations: {},
|
|
549
|
+
openReaders: 0,
|
|
550
|
+
maxOpenReaders: 0,
|
|
551
|
+
readersOpened: 0,
|
|
552
|
+
readersClosed: 0
|
|
553
|
+
};
|
|
554
|
+
counters.openReaders ??= 0;
|
|
555
|
+
counters.maxOpenReaders ??= 0;
|
|
556
|
+
counters.readersOpened ??= 0;
|
|
557
|
+
counters.readersClosed ??= 0;
|
|
558
|
+
return counters;
|
|
559
|
+
}
|
|
560
|
+
function returnedRowCount(value) {
|
|
561
|
+
if (Array.isArray(value))
|
|
562
|
+
return value.length;
|
|
563
|
+
if (value === null || value === undefined || value === false)
|
|
564
|
+
return 0;
|
|
565
|
+
return 1;
|
|
566
|
+
}
|
|
567
|
+
function trackDecodeOperation(name, operation) {
|
|
568
|
+
const counters = debugCounters();
|
|
569
|
+
const before = counters.decodedRows;
|
|
570
|
+
const started = performance.now();
|
|
571
|
+
let value;
|
|
572
|
+
try {
|
|
573
|
+
value = operation();
|
|
574
|
+
return value;
|
|
575
|
+
} finally {
|
|
576
|
+
const decodedRows = counters.decodedRows - before;
|
|
577
|
+
const current = counters.operations[name] ?? {
|
|
578
|
+
calls: 0,
|
|
579
|
+
decodedRows: 0,
|
|
580
|
+
maxDecodedRows: 0
|
|
581
|
+
};
|
|
582
|
+
current.calls += 1;
|
|
583
|
+
current.decodedRows += decodedRows;
|
|
584
|
+
current.maxDecodedRows = Math.max(current.maxDecodedRows, decodedRows);
|
|
585
|
+
counters.operations[name] = current;
|
|
586
|
+
const statement = QUERY_STATEMENTS[name];
|
|
587
|
+
if (counters.captureQueries && statement) {
|
|
588
|
+
(counters.queries ??= []).push({
|
|
589
|
+
operation: name,
|
|
590
|
+
statement,
|
|
591
|
+
rows: returnedRowCount(value),
|
|
592
|
+
elapsedMs: performance.now() - started
|
|
593
|
+
});
|
|
594
|
+
}
|
|
595
|
+
}
|
|
596
|
+
}
|
|
597
|
+
function decode(row) {
|
|
598
|
+
debugCounters().decodedRows += 1;
|
|
599
|
+
const data = JSON.parse(row.data);
|
|
600
|
+
if (!data || typeof data !== "object" || Array.isArray(data)) {
|
|
601
|
+
throw new Error(`Invalid session_message data at seq ${row.seq}`);
|
|
602
|
+
}
|
|
603
|
+
if (row.type === "idle" && !["succeeded", "failed", "interrupted"].includes(String(data.outcome))) {
|
|
604
|
+
throw new Error(`Invalid idle outcome at seq ${row.seq}`);
|
|
605
|
+
}
|
|
606
|
+
return { ...row, data };
|
|
607
|
+
}
|
|
608
|
+
function connect(path, identity = "") {
|
|
609
|
+
const stat = statSync(path);
|
|
610
|
+
const acquiredIdentity = identity || `${stat.dev}:${stat.ino}:${stat.birthtimeMs}`;
|
|
611
|
+
const db = new Database2(path, { readonly: true, fileMustExist: true });
|
|
612
|
+
try {
|
|
613
|
+
assertOpenCodeStoreGeneration(db, "v2", path);
|
|
614
|
+
} catch (error) {
|
|
615
|
+
db.close();
|
|
616
|
+
throw error;
|
|
617
|
+
}
|
|
618
|
+
const counters = debugCounters();
|
|
619
|
+
counters.openReaders += 1;
|
|
620
|
+
counters.readersOpened += 1;
|
|
621
|
+
counters.maxOpenReaders = Math.max(counters.maxOpenReaders, counters.openReaders);
|
|
622
|
+
return { db, statements: new Map, identity: acquiredIdentity, leases: 0, retired: false };
|
|
623
|
+
}
|
|
624
|
+
function closeConnection(connection) {
|
|
625
|
+
connection.statements.clear();
|
|
626
|
+
try {
|
|
627
|
+
connection.db.close();
|
|
628
|
+
} finally {
|
|
629
|
+
const counters = debugCounters();
|
|
630
|
+
counters.openReaders = Math.max(0, counters.openReaders - 1);
|
|
631
|
+
counters.readersClosed += 1;
|
|
632
|
+
}
|
|
633
|
+
}
|
|
634
|
+
class V2StoreReader {
|
|
635
|
+
path;
|
|
636
|
+
pool;
|
|
637
|
+
db;
|
|
638
|
+
connection;
|
|
639
|
+
closed = false;
|
|
640
|
+
constructor(path, pool) {
|
|
641
|
+
this.path = path;
|
|
642
|
+
this.pool = pool;
|
|
643
|
+
this.connection = pool ? pool.acquire(path) : connect(path);
|
|
644
|
+
this.db = this.connection.db;
|
|
645
|
+
}
|
|
646
|
+
hostIdentity() {
|
|
647
|
+
return createHash3("sha256").update(JSON.stringify([hostname(), resolve(this.path), this.connection.identity])).digest("hex");
|
|
648
|
+
}
|
|
649
|
+
prepare(sql) {
|
|
650
|
+
if (this.closed)
|
|
651
|
+
throw new Error("Store reader is closed");
|
|
652
|
+
const statements = this.connection.statements;
|
|
653
|
+
let statement = statements.get(sql);
|
|
654
|
+
if (!statement) {
|
|
655
|
+
statement = this.db.prepare(sql);
|
|
656
|
+
if (statements.size >= 128)
|
|
657
|
+
statements.clear();
|
|
658
|
+
statements.set(sql, statement);
|
|
659
|
+
}
|
|
660
|
+
return statement;
|
|
661
|
+
}
|
|
662
|
+
close() {
|
|
663
|
+
if (this.closed)
|
|
664
|
+
return;
|
|
665
|
+
this.closed = true;
|
|
666
|
+
if (this.pool)
|
|
667
|
+
this.pool.release(this.connection);
|
|
668
|
+
else
|
|
669
|
+
closeConnection(this.connection);
|
|
670
|
+
}
|
|
671
|
+
page(sessionID, options = {}) {
|
|
672
|
+
return trackDecodeOperation("page", () => {
|
|
673
|
+
const limit = options.limit ?? 100;
|
|
674
|
+
if (!Number.isSafeInteger(limit) || limit < 1 || limit > 1e4)
|
|
675
|
+
throw new Error("Invalid page limit");
|
|
676
|
+
const after = options.after ?? -1;
|
|
677
|
+
if (!Number.isSafeInteger(after))
|
|
678
|
+
throw new Error("Invalid seq cursor");
|
|
679
|
+
if (options.through !== undefined && !Number.isSafeInteger(options.through))
|
|
680
|
+
throw new Error("Invalid seq upper bound");
|
|
681
|
+
const predicates = ["session_id = ?", "seq > ?"];
|
|
682
|
+
const parameters = [sessionID, after];
|
|
683
|
+
if (options.through !== undefined) {
|
|
684
|
+
predicates.push("seq <= ?");
|
|
685
|
+
parameters.push(options.through);
|
|
686
|
+
}
|
|
687
|
+
if (options.type) {
|
|
688
|
+
predicates.push("type = ?");
|
|
689
|
+
parameters.push(options.type);
|
|
690
|
+
}
|
|
691
|
+
const rows = this.prepare(`SELECT id, session_id, type, seq, time_created, data FROM session_message
|
|
692
|
+
WHERE ${predicates.join(" AND ")}
|
|
693
|
+
ORDER BY seq ASC LIMIT ?`).all(...parameters, limit).map(decode);
|
|
694
|
+
return { rows, cursor: rows.at(-1)?.seq };
|
|
695
|
+
});
|
|
696
|
+
}
|
|
697
|
+
messagePage(sessionID, afterOrdinal, limit, finalWatermark, after) {
|
|
698
|
+
return trackDecodeOperation("messagePage", () => {
|
|
699
|
+
if (!Number.isSafeInteger(afterOrdinal) || afterOrdinal < 0)
|
|
700
|
+
throw new Error("Invalid raw-message ordinal cursor");
|
|
701
|
+
if (!Number.isSafeInteger(limit) || limit < 1 || limit > 1e4)
|
|
702
|
+
throw new Error("Invalid page limit");
|
|
703
|
+
if (!Number.isSafeInteger(finalWatermark) || finalWatermark < 0)
|
|
704
|
+
throw new Error("Invalid raw-message watermark");
|
|
705
|
+
const pageSize = Math.min(limit, finalWatermark - afterOrdinal);
|
|
706
|
+
if (pageSize <= 0)
|
|
707
|
+
return [];
|
|
708
|
+
if (after) {
|
|
709
|
+
const anchor = this.prepare("SELECT seq FROM session_message WHERE session_id = ? AND id = ?").get(sessionID, after.id);
|
|
710
|
+
if (anchor) {
|
|
711
|
+
return this.prepare(`SELECT id, session_id, type, seq, time_created, data
|
|
712
|
+
FROM session_message WHERE session_id = ? AND seq > ?
|
|
713
|
+
AND type IN (${RAW_MESSAGE_TYPE_PARAMETERS}) ORDER BY seq ASC LIMIT ?`).all(sessionID, anchor.seq, ...RAW_MESSAGE_TYPES, pageSize).map(decode);
|
|
714
|
+
}
|
|
715
|
+
}
|
|
716
|
+
const maximumSeq = Number.MAX_SAFE_INTEGER;
|
|
717
|
+
const rows = this.prepare(V2_MESSAGE_PAGE_SQL).all(afterOrdinal, sessionID, ...RAW_MESSAGE_TYPES, Math.max(0, afterOrdinal - 1), sessionID, sessionID, ...RAW_MESSAGE_TYPES, finalWatermark - 1, maximumSeq, sessionID, ...RAW_MESSAGE_TYPES, pageSize);
|
|
718
|
+
return rows.map(decode);
|
|
719
|
+
});
|
|
720
|
+
}
|
|
721
|
+
messageCount(sessionID) {
|
|
722
|
+
return trackDecodeOperation("messageCount", () => {
|
|
723
|
+
const rawTypes = RAW_MESSAGE_TYPES.map(() => "?").join(", ");
|
|
724
|
+
const row = this.prepare(`SELECT COUNT(*) AS count FROM session_message
|
|
725
|
+
WHERE session_id = ? AND type IN (${rawTypes})`).get(sessionID, ...RAW_MESSAGE_TYPES);
|
|
726
|
+
return typeof row?.count === "number" ? row.count : 0;
|
|
727
|
+
});
|
|
728
|
+
}
|
|
729
|
+
sessionDirectoryPage(afterSessionID, limit) {
|
|
730
|
+
const rows = this.prepare(`SELECT id, COALESCE(directory, '') AS directory FROM session_v2
|
|
731
|
+
WHERE id > ? ORDER BY id ASC LIMIT ?`).all(afterSessionID ?? "", limit);
|
|
732
|
+
return rows.map((row) => ({ sessionId: row.id, directory: row.directory }));
|
|
733
|
+
}
|
|
734
|
+
latestMessageTime(sessionID) {
|
|
735
|
+
const row = this.prepare("SELECT MAX(time_created) AS time FROM session_message WHERE session_id = ?").get(sessionID);
|
|
736
|
+
return row?.time ?? undefined;
|
|
737
|
+
}
|
|
738
|
+
isSubagent(sessionID) {
|
|
739
|
+
const row = this.prepare("SELECT parent_id FROM session_v2 WHERE id = ?").get(sessionID);
|
|
740
|
+
if (!row)
|
|
741
|
+
return null;
|
|
742
|
+
return typeof row.parent_id === "string" && row.parent_id.length > 0;
|
|
743
|
+
}
|
|
744
|
+
sessionIds() {
|
|
745
|
+
return this.prepare("SELECT id FROM session_v2").all().map((row) => row.id);
|
|
746
|
+
}
|
|
747
|
+
latestRootSessionInDirectory(directory) {
|
|
748
|
+
const columns = new Set(this.prepare("PRAGMA table_info(session_v2)").all().map((column) => column.name));
|
|
749
|
+
const archived = columns.has("time_archived") ? " AND time_archived IS NULL" : "";
|
|
750
|
+
const order = columns.has("time_updated") ? "time_updated DESC, id DESC" : "id DESC";
|
|
751
|
+
const page = this.prepare(`SELECT id, COALESCE(directory, '') AS directory FROM session_v2
|
|
752
|
+
WHERE parent_id IS NULL${archived}
|
|
753
|
+
AND COALESCE(json_extract(metadata, '$.magic_context'), '') <> 'hidden-run'
|
|
754
|
+
ORDER BY ${order} LIMIT ? OFFSET ?`);
|
|
755
|
+
const pageSize = 500;
|
|
756
|
+
for (let offset = 0;; offset += pageSize) {
|
|
757
|
+
const rows = page.all(pageSize, offset);
|
|
758
|
+
const match = rows.find((row) => sameProjectDirectory(row.directory, directory));
|
|
759
|
+
if (match)
|
|
760
|
+
return match.id;
|
|
761
|
+
if (rows.length < pageSize)
|
|
762
|
+
return;
|
|
763
|
+
}
|
|
764
|
+
}
|
|
765
|
+
rootSessionActivity(sessionIDs) {
|
|
766
|
+
const roots = new Map;
|
|
767
|
+
for (let offset = 0;offset < sessionIDs.length; offset += 500) {
|
|
768
|
+
const chunk = sessionIDs.slice(offset, offset + 500);
|
|
769
|
+
const rows = this.prepare(`SELECT id, (
|
|
770
|
+
SELECT MAX(time_created) FROM session_message
|
|
771
|
+
WHERE session_id = session_v2.id AND type = 'user'
|
|
772
|
+
) AS time FROM session_v2
|
|
773
|
+
WHERE parent_id IS NULL
|
|
774
|
+
AND COALESCE(json_extract(metadata, '$.magic_context'), '') <> 'hidden-run'
|
|
775
|
+
AND id IN (${chunk.map(() => "?").join(",")})`).all(...chunk);
|
|
776
|
+
for (const row of rows)
|
|
777
|
+
if (row.time !== null)
|
|
778
|
+
roots.set(row.id, row.time);
|
|
779
|
+
}
|
|
780
|
+
return roots;
|
|
781
|
+
}
|
|
782
|
+
oldestUserMessageTimesSince(sessionIDs, sinceMs) {
|
|
783
|
+
const result = new Map;
|
|
784
|
+
for (let offset = 0;offset < sessionIDs.length; offset += 500) {
|
|
785
|
+
const chunk = sessionIDs.slice(offset, offset + 500);
|
|
786
|
+
const rows = this.prepare(`SELECT session_id, MIN(time_created) AS time FROM session_message
|
|
787
|
+
WHERE type = 'user' AND time_created > ? AND session_id IN (${chunk.map(() => "?").join(",")})
|
|
788
|
+
GROUP BY session_id`).all(sinceMs, ...chunk);
|
|
789
|
+
for (const row of rows)
|
|
790
|
+
result.set(row.session_id, row.time);
|
|
791
|
+
}
|
|
448
792
|
return result;
|
|
449
|
-
}
|
|
793
|
+
}
|
|
794
|
+
retrospectiveUserPage(sessionID, options) {
|
|
795
|
+
const { boundaryMs, before, limit, maxChars, truncationMarker } = options;
|
|
796
|
+
if (!Number.isSafeInteger(limit) || limit < 1 || limit > 1e4)
|
|
797
|
+
throw new Error("Invalid page limit");
|
|
798
|
+
if (!Number.isSafeInteger(maxChars) || maxChars <= truncationMarker.length)
|
|
799
|
+
throw new Error("Invalid text limit");
|
|
800
|
+
const direction = before ? "DESC" : "ASC";
|
|
801
|
+
const head = Math.floor((maxChars - truncationMarker.length) / 2);
|
|
802
|
+
const tail = maxChars - truncationMarker.length - head;
|
|
803
|
+
return this.prepare(`WITH users AS (
|
|
804
|
+
SELECT session_id, seq, time_created, data FROM session_message
|
|
805
|
+
WHERE session_id = ? AND type = 'user' AND time_created ${before ? "<=" : ">"}
|
|
806
|
+
?
|
|
807
|
+
ORDER BY time_created ${direction}, seq ${direction} LIMIT ?
|
|
808
|
+
), texts AS (
|
|
809
|
+
SELECT session_id, seq, time_created, CASE
|
|
810
|
+
WHEN json_extract(data, '$.synthetic') = 1 OR json_extract(data, '$.ignored') = 1 THEN ''
|
|
811
|
+
WHEN json_type(data, '$.text') = 'text' THEN json_extract(data, '$.text')
|
|
812
|
+
ELSE COALESCE((SELECT group_concat(json_extract(value, '$.text'), char(10))
|
|
813
|
+
FROM json_each(users.data, '$.content')
|
|
814
|
+
WHERE json_extract(value, '$.type') IN ('text', 'input_text')
|
|
815
|
+
AND COALESCE(json_extract(value, '$.synthetic'), 0) <> 1
|
|
816
|
+
AND COALESCE(json_extract(value, '$.ignored'), 0) <> 1), '') END AS text
|
|
817
|
+
FROM users WHERE json_valid(data)
|
|
818
|
+
) SELECT seq, time_created, CASE WHEN length(text) > ?
|
|
819
|
+
THEN substr(text, 1, ?) || ? || substr(text, -?) ELSE text END AS text,
|
|
820
|
+
EXISTS (
|
|
821
|
+
SELECT 1 FROM session_message assistant_reply
|
|
822
|
+
WHERE assistant_reply.session_id = texts.session_id
|
|
823
|
+
AND assistant_reply.type = 'assistant'
|
|
824
|
+
AND assistant_reply.seq < texts.seq
|
|
825
|
+
AND EXISTS (
|
|
826
|
+
SELECT 1 FROM session_message prior_user
|
|
827
|
+
WHERE prior_user.session_id = texts.session_id
|
|
828
|
+
AND prior_user.type = 'user'
|
|
829
|
+
AND prior_user.seq < assistant_reply.seq
|
|
830
|
+
)
|
|
831
|
+
) AS follows_assistant_reply
|
|
832
|
+
FROM texts ORDER BY time_created ${direction}, seq ${direction}`).all(sessionID, boundaryMs, limit, maxChars, head, truncationMarker, tail);
|
|
833
|
+
}
|
|
834
|
+
storedRowsAfter(sessionID, after) {
|
|
835
|
+
const row = this.prepare("SELECT COUNT(*) AS count FROM session_message WHERE session_id = ? AND seq > ?").get(sessionID, after);
|
|
836
|
+
return row.count;
|
|
837
|
+
}
|
|
838
|
+
storedMessageCount(sessionID) {
|
|
839
|
+
return trackDecodeOperation("storedMessageCount", () => {
|
|
840
|
+
const row = this.prepare("SELECT COUNT(*) AS count FROM session_message WHERE session_id = ?").get(sessionID);
|
|
841
|
+
return typeof row?.count === "number" ? row.count : 0;
|
|
842
|
+
});
|
|
843
|
+
}
|
|
844
|
+
messageById(sessionID, id) {
|
|
845
|
+
return trackDecodeOperation("messageById", () => {
|
|
846
|
+
const rawTypes = RAW_MESSAGE_TYPES.map(() => "?").join(", ");
|
|
847
|
+
const row = this.prepare(`SELECT id, session_id, type, seq, time_created, data FROM session_message
|
|
848
|
+
WHERE session_id = ? AND id = ? AND type IN (${rawTypes}) LIMIT 1`).get(sessionID, id, ...RAW_MESSAGE_TYPES);
|
|
849
|
+
return row ? decode(row) : null;
|
|
850
|
+
});
|
|
851
|
+
}
|
|
852
|
+
messageTimes(sessionID, ids) {
|
|
853
|
+
const times = new Map;
|
|
854
|
+
for (let offset = 0;offset < ids.length; offset += 500) {
|
|
855
|
+
const batch = ids.slice(offset, offset + 500);
|
|
856
|
+
const rows = this.prepare(`SELECT id, COALESCE(time_created, json_extract(data, '$.time.created')) AS created,
|
|
857
|
+
json_extract(data, '$.time.completed') AS completed
|
|
858
|
+
FROM session_message WHERE session_id = ? AND id IN (${batch.map(() => "?").join(",")})`).all(sessionID, ...batch);
|
|
859
|
+
for (const row of rows)
|
|
860
|
+
times.set(row.id, {
|
|
861
|
+
created: typeof row.created === "number" ? row.created : undefined,
|
|
862
|
+
completed: typeof row.completed === "number" ? row.completed : undefined
|
|
863
|
+
});
|
|
864
|
+
}
|
|
865
|
+
return times;
|
|
866
|
+
}
|
|
867
|
+
messageExistsById(sessionID, id) {
|
|
868
|
+
return trackDecodeOperation("messageExistsById", () => {
|
|
869
|
+
const rawTypes = RAW_MESSAGE_TYPES.map(() => "?").join(", ");
|
|
870
|
+
const row = this.prepare(`SELECT 1 FROM session_message
|
|
871
|
+
WHERE id = ? AND session_id = ? AND type IN (${rawTypes}) LIMIT 1`).get(id, sessionID, ...RAW_MESSAGE_TYPES);
|
|
872
|
+
return row != null;
|
|
873
|
+
});
|
|
874
|
+
}
|
|
875
|
+
messageOrdinalById(sessionID, id) {
|
|
876
|
+
return trackDecodeOperation("messageOrdinalById", () => {
|
|
877
|
+
const rawTypes = RAW_MESSAGE_TYPES.map(() => "?").join(", ");
|
|
878
|
+
const row = this.prepare(`WITH target AS (
|
|
879
|
+
SELECT seq FROM session_message
|
|
880
|
+
WHERE session_id = ? AND id = ? AND type IN (${rawTypes})
|
|
881
|
+
LIMIT 1
|
|
882
|
+
)
|
|
883
|
+
SELECT (
|
|
884
|
+
SELECT COUNT(*) FROM session_message
|
|
885
|
+
WHERE session_id = ?
|
|
886
|
+
AND type IN (${rawTypes})
|
|
887
|
+
AND seq <= target.seq
|
|
888
|
+
) AS ordinal
|
|
889
|
+
FROM target`).get(sessionID, id, ...RAW_MESSAGE_TYPES, sessionID, ...RAW_MESSAGE_TYPES);
|
|
890
|
+
return typeof row?.ordinal === "number" ? row.ordinal : null;
|
|
891
|
+
});
|
|
892
|
+
}
|
|
893
|
+
messageIdOrdinals(sessionID, fromOrdinal, toOrdinal) {
|
|
894
|
+
return trackDecodeOperation("messageIdOrdinals", () => {
|
|
895
|
+
if (!Number.isSafeInteger(fromOrdinal) || fromOrdinal < 1)
|
|
896
|
+
throw new Error("Invalid raw-message range start");
|
|
897
|
+
if (!Number.isSafeInteger(toOrdinal) || toOrdinal < fromOrdinal)
|
|
898
|
+
throw new Error("Invalid raw-message range end");
|
|
899
|
+
const rawTypes = RAW_MESSAGE_TYPES.map(() => "?").join(", ");
|
|
900
|
+
const rows = this.prepare(`SELECT id FROM session_message
|
|
901
|
+
WHERE session_id = ? AND type IN (${rawTypes})
|
|
902
|
+
ORDER BY seq ASC LIMIT ? OFFSET ?`).all(sessionID, ...RAW_MESSAGE_TYPES, toOrdinal - fromOrdinal + 1, fromOrdinal - 1);
|
|
903
|
+
return new Map(rows.map((row, index) => [row.id, fromOrdinal + index]));
|
|
904
|
+
});
|
|
905
|
+
}
|
|
906
|
+
messageOrdinalPage(sessionID, after, limit) {
|
|
907
|
+
return trackDecodeOperation("messageOrdinalPage", () => {
|
|
908
|
+
if (!Number.isSafeInteger(limit) || limit < 1 || limit > 1e4)
|
|
909
|
+
throw new Error("Invalid page limit");
|
|
910
|
+
const rows = after ? this.prepare(`SELECT id, type, seq, json_valid(data) AS valid
|
|
911
|
+
FROM session_message
|
|
912
|
+
WHERE session_id = ? AND (seq > ? OR (seq = ? AND id > ?))
|
|
913
|
+
ORDER BY seq ASC, id ASC LIMIT ?`).all(sessionID, after.timeCreated, after.timeCreated, after.id, limit) : this.prepare(`SELECT id, type, seq, json_valid(data) AS valid
|
|
914
|
+
FROM session_message
|
|
915
|
+
WHERE session_id = ?
|
|
916
|
+
ORDER BY seq ASC, id ASC LIMIT ?`).all(sessionID, limit);
|
|
917
|
+
const rawTypes = new Set(RAW_MESSAGE_TYPES);
|
|
918
|
+
return rows.map((row) => ({
|
|
919
|
+
id: row.id,
|
|
920
|
+
timeCreated: row.seq,
|
|
921
|
+
contributesOrdinal: rawTypes.has(row.type),
|
|
922
|
+
hasValidInfo: row.valid === 1
|
|
923
|
+
}));
|
|
924
|
+
});
|
|
925
|
+
}
|
|
926
|
+
range(sessionID, after, through) {
|
|
927
|
+
return trackDecodeOperation("range", () => through <= after ? [] : this.all(sessionID, after, undefined, through));
|
|
928
|
+
}
|
|
929
|
+
spanRowStamps(sessionID, after, through) {
|
|
930
|
+
if (!Number.isSafeInteger(after) || !Number.isSafeInteger(through))
|
|
931
|
+
throw new Error("Invalid seq span");
|
|
932
|
+
const rows = this.prepare(`SELECT seq, id, time_updated AS updated, length(CAST(data AS BLOB)) AS bytes
|
|
933
|
+
FROM session_message WHERE session_id = ? AND seq > ? AND seq <= ?
|
|
934
|
+
ORDER BY seq ASC`).all(sessionID, after, through);
|
|
935
|
+
const stamps = new Map;
|
|
936
|
+
for (const row of rows)
|
|
937
|
+
stamps.set(row.seq, `${row.id}\x00${row.updated ?? ""}\x00${row.bytes ?? ""}`);
|
|
938
|
+
return stamps;
|
|
939
|
+
}
|
|
940
|
+
rawRowsThrough(sessionID, throughSeq, limit) {
|
|
941
|
+
return trackDecodeOperation("rawRowsThrough", () => {
|
|
942
|
+
if (!Number.isSafeInteger(throughSeq))
|
|
943
|
+
throw new Error("Invalid seq upper bound");
|
|
944
|
+
if (!Number.isSafeInteger(limit) || limit < 1 || limit > 1e4)
|
|
945
|
+
throw new Error("Invalid page limit");
|
|
946
|
+
const rawTypes = RAW_MESSAGE_TYPES.map(() => "?").join(", ");
|
|
947
|
+
return this.prepare(`SELECT id, session_id, type, seq, time_created, data FROM session_message
|
|
948
|
+
WHERE session_id = ? AND type IN (${rawTypes}) AND seq <= ?
|
|
949
|
+
ORDER BY seq DESC LIMIT ?`).all(sessionID, ...RAW_MESSAGE_TYPES, throughSeq, limit).map(decode);
|
|
950
|
+
});
|
|
951
|
+
}
|
|
952
|
+
sequenceForId(sessionID, id) {
|
|
953
|
+
if (!id)
|
|
954
|
+
return;
|
|
955
|
+
const row = this.prepare("SELECT seq FROM session_message WHERE session_id = ? AND id = ? LIMIT 1").get(sessionID, id);
|
|
956
|
+
return typeof row?.seq === "number" ? row.seq : undefined;
|
|
957
|
+
}
|
|
958
|
+
syntheticMessageIDs(sessionID) {
|
|
959
|
+
const rows = this.prepare("SELECT id FROM session_message WHERE session_id = ? AND type = 'synthetic'").all(sessionID);
|
|
960
|
+
return new Set(rows.map((row) => row.id));
|
|
961
|
+
}
|
|
962
|
+
forkOrigin(sessionID) {
|
|
963
|
+
const columns = new Set(this.prepare("PRAGMA table_info(session_v2)").all().map((column) => column.name));
|
|
964
|
+
if (!columns.has("fork_session_id") || !columns.has("fork_boundary"))
|
|
965
|
+
return null;
|
|
966
|
+
const row = this.prepare("SELECT fork_session_id, fork_boundary FROM session_v2 WHERE id = ?").get(sessionID);
|
|
967
|
+
if (!row?.fork_session_id || !row.fork_boundary)
|
|
968
|
+
return null;
|
|
969
|
+
try {
|
|
970
|
+
const boundary = JSON.parse(row.fork_boundary);
|
|
971
|
+
if (boundary.type !== "before" && boundary.type !== "through" || typeof boundary.messageID !== "string" || boundary.messageID.length === 0)
|
|
972
|
+
return null;
|
|
973
|
+
return {
|
|
974
|
+
parentSessionID: row.fork_session_id,
|
|
975
|
+
boundary: { type: boundary.type, messageID: boundary.messageID }
|
|
976
|
+
};
|
|
977
|
+
} catch {
|
|
978
|
+
return null;
|
|
979
|
+
}
|
|
980
|
+
}
|
|
981
|
+
sessionExists(sessionID) {
|
|
982
|
+
return this.prepare("SELECT 1 FROM session_v2 WHERE id = ?").get(sessionID) != null;
|
|
983
|
+
}
|
|
984
|
+
rowStampsThrough(sessionID, throughSeq) {
|
|
985
|
+
return this.prepare(`SELECT id, type, seq, time_created FROM session_message
|
|
986
|
+
WHERE session_id = ? AND seq <= ? ORDER BY seq ASC`).all(sessionID, throughSeq);
|
|
987
|
+
}
|
|
988
|
+
latestSequenceForIds(sessionID, ids) {
|
|
989
|
+
let latest = -1;
|
|
990
|
+
for (let offset = 0;offset < ids.length; offset += 500) {
|
|
991
|
+
const chunk = ids.slice(offset, offset + 500);
|
|
992
|
+
const row = this.prepare(`SELECT MAX(seq) AS seq FROM session_message
|
|
993
|
+
WHERE session_id = ? AND id IN (${chunk.map(() => "?").join(", ")})`).get(sessionID, ...chunk);
|
|
994
|
+
if (typeof row?.seq === "number")
|
|
995
|
+
latest = Math.max(latest, row.seq);
|
|
996
|
+
}
|
|
997
|
+
return latest;
|
|
998
|
+
}
|
|
999
|
+
compactionByStatus(sessionID, status) {
|
|
1000
|
+
const row = this.prepare(`SELECT id, session_id, type, seq, time_created, data FROM session_message
|
|
1001
|
+
WHERE session_id = ? AND type = 'compaction'
|
|
1002
|
+
AND json_extract(data, '$.status') = ?
|
|
1003
|
+
ORDER BY seq DESC LIMIT 1`).get(sessionID, status);
|
|
1004
|
+
return row ? decode(row) : undefined;
|
|
1005
|
+
}
|
|
1006
|
+
latestCompaction(sessionID) {
|
|
1007
|
+
return trackDecodeOperation("latestCompaction", () => this.compactionByStatus(sessionID, "completed"));
|
|
1008
|
+
}
|
|
1009
|
+
latestRunningCompaction(sessionID) {
|
|
1010
|
+
return trackDecodeOperation("latestRunningCompaction", () => this.compactionByStatus(sessionID, "running"));
|
|
1011
|
+
}
|
|
1012
|
+
idleRows(sessionID, after = -1) {
|
|
1013
|
+
return trackDecodeOperation("idleRows", () => this.all(sessionID, after, "idle"));
|
|
1014
|
+
}
|
|
1015
|
+
earliestSequence(sessionID) {
|
|
1016
|
+
const row = this.prepare("SELECT MIN(seq) AS seq FROM session_message WHERE session_id = ?").get(sessionID);
|
|
1017
|
+
return typeof row?.seq === "number" ? row.seq : undefined;
|
|
1018
|
+
}
|
|
1019
|
+
latestSequence(sessionID) {
|
|
1020
|
+
const row = this.prepare("SELECT MAX(seq) AS seq FROM session_message WHERE session_id = ?").get(sessionID);
|
|
1021
|
+
return typeof row?.seq === "number" ? row.seq : -1;
|
|
1022
|
+
}
|
|
1023
|
+
assistantSince(sessionID, afterSeq) {
|
|
1024
|
+
return trackDecodeOperation("assistantSince", () => {
|
|
1025
|
+
const rows = this.prepare(`SELECT id, session_id, type, seq, time_created, data FROM session_message
|
|
1026
|
+
WHERE session_id = ? AND type = 'assistant' AND seq > ?
|
|
1027
|
+
ORDER BY seq ASC`).all(sessionID, afterSeq);
|
|
1028
|
+
return rows.map((row) => decode(row));
|
|
1029
|
+
});
|
|
1030
|
+
}
|
|
1031
|
+
latestAssistant(sessionID) {
|
|
1032
|
+
return trackDecodeOperation("latestAssistant", () => {
|
|
1033
|
+
const row = this.prepare(`SELECT id, session_id, type, seq, time_created, data FROM session_message
|
|
1034
|
+
WHERE session_id = ? AND type = 'assistant'
|
|
1035
|
+
ORDER BY seq DESC LIMIT 1`).get(sessionID);
|
|
1036
|
+
return row ? decode(row) : undefined;
|
|
1037
|
+
});
|
|
1038
|
+
}
|
|
1039
|
+
latestIdle(sessionID) {
|
|
1040
|
+
return trackDecodeOperation("latestIdle", () => {
|
|
1041
|
+
const row = this.prepare(`SELECT id, session_id, type, seq, time_created, data FROM session_message
|
|
1042
|
+
WHERE session_id = ? AND type = 'idle'
|
|
1043
|
+
ORDER BY seq DESC LIMIT 1`).get(sessionID);
|
|
1044
|
+
return row ? decode(row) : undefined;
|
|
1045
|
+
});
|
|
1046
|
+
}
|
|
1047
|
+
window(sessionID) {
|
|
1048
|
+
return trackDecodeOperation("window", () => this.db.transaction(() => {
|
|
1049
|
+
const cut = this.latestCompaction(sessionID);
|
|
1050
|
+
return this.all(sessionID, cut ? cut.seq - 1 : -1);
|
|
1051
|
+
}).deferred());
|
|
1052
|
+
}
|
|
1053
|
+
history(sessionID) {
|
|
1054
|
+
return trackDecodeOperation("history", () => this.all(sessionID, -1));
|
|
1055
|
+
}
|
|
1056
|
+
all(sessionID, after, type, through) {
|
|
1057
|
+
const rows = [];
|
|
1058
|
+
for (;; ) {
|
|
1059
|
+
const page = this.page(sessionID, { after, through, type });
|
|
1060
|
+
rows.push(...page.rows);
|
|
1061
|
+
if (page.rows.length < 100 || page.cursor === undefined)
|
|
1062
|
+
return rows;
|
|
1063
|
+
after = page.cursor;
|
|
1064
|
+
}
|
|
1065
|
+
}
|
|
450
1066
|
}
|
|
451
1067
|
|
|
452
1068
|
// ../plugin/src/features/magic-context/session-activity.ts
|
|
@@ -578,6 +1194,22 @@ var SESSION_SCOPED_TABLES = [
|
|
|
578
1194
|
{ table: "lkg_slots" },
|
|
579
1195
|
{ table: "lkg_slot_chunks" }
|
|
580
1196
|
];
|
|
1197
|
+
var cleanupStatements = new WeakMap;
|
|
1198
|
+
function cleanupStatement(db, sql) {
|
|
1199
|
+
let cache = cleanupStatements.get(db);
|
|
1200
|
+
if (!cache) {
|
|
1201
|
+
cache = new Map;
|
|
1202
|
+
cleanupStatements.set(db, cache);
|
|
1203
|
+
}
|
|
1204
|
+
let statement = cache.get(sql);
|
|
1205
|
+
if (!statement) {
|
|
1206
|
+
if (cache.size >= 256)
|
|
1207
|
+
cache.clear();
|
|
1208
|
+
statement = prepareCachedStatement2(db, sql);
|
|
1209
|
+
cache.set(sql, statement);
|
|
1210
|
+
}
|
|
1211
|
+
return statement;
|
|
1212
|
+
}
|
|
581
1213
|
function deleteSessionScopedRows(db, sessionIds, harness, options = {}) {
|
|
582
1214
|
if (sessionIds.length === 0)
|
|
583
1215
|
return 0;
|
|
@@ -585,7 +1217,7 @@ function deleteSessionScopedRows(db, sessionIds, harness, options = {}) {
|
|
|
585
1217
|
if (options.rustModuleCleanupAcknowledged !== true) {
|
|
586
1218
|
const placeholders = sessionIds.map(() => "?").join(", ");
|
|
587
1219
|
const harnessPredicate = harness === undefined ? "" : " AND harness = ?";
|
|
588
|
-
const protectedRows = db
|
|
1220
|
+
const protectedRows = cleanupStatement(db, `SELECT session_id
|
|
589
1221
|
FROM pending_session_cleanup
|
|
590
1222
|
WHERE session_id IN (${placeholders})
|
|
591
1223
|
AND harness LIKE '%:rust'${harnessPredicate}`).all(...sessionIds, ...harness === undefined ? [] : [`${harness}:rust`]);
|
|
@@ -597,11 +1229,11 @@ function deleteSessionScopedRows(db, sessionIds, harness, options = {}) {
|
|
|
597
1229
|
const placeholders = deletableSessionIds.map(() => "?").join(", ");
|
|
598
1230
|
const harnessTables = SESSION_SCOPED_TABLES.filter((definition) => definition.harnessScoped);
|
|
599
1231
|
if (options.preserveTagRevisionTombstones) {
|
|
600
|
-
const owners = db
|
|
601
|
-
const record = db
|
|
1232
|
+
const owners = cleanupStatement(db, harnessTables.map((definition) => `SELECT session_id, harness FROM ${definition.table} WHERE session_id IN (${placeholders})${definition.extraPredicate ? ` AND ${definition.extraPredicate}` : ""}`).join(" UNION ")).all(...harnessTables.flatMap(() => deletableSessionIds));
|
|
1233
|
+
const record = cleanupStatement(db, "INSERT OR IGNORE INTO schema_migrations_meta (key, value) VALUES (?, ?)");
|
|
602
1234
|
for (const owner of owners)
|
|
603
1235
|
record.run(tagTombstoneOwnerKey(owner.session_id, owner.harness), owner.harness);
|
|
604
|
-
const fallback = db
|
|
1236
|
+
const fallback = cleanupStatement(db, `INSERT INTO schema_migrations_meta (key, value)
|
|
605
1237
|
SELECT ?, ? WHERE NOT EXISTS (SELECT 1 FROM schema_migrations_meta WHERE key >= ? AND key < ?)`);
|
|
606
1238
|
for (const id of deletableSessionIds)
|
|
607
1239
|
fallback.run(tagTombstoneOwnerKey(id, getHarness2()), getHarness2(), ...tagTombstoneOwnerRange(id));
|
|
@@ -610,7 +1242,7 @@ function deleteSessionScopedRows(db, sessionIds, harness, options = {}) {
|
|
|
610
1242
|
deleteChunkEmbedBackoffForSession(db, sessionId);
|
|
611
1243
|
for (const definition of SESSION_SCOPED_TABLES) {
|
|
612
1244
|
if (definition.table === "message_history_fts") {
|
|
613
|
-
db
|
|
1245
|
+
cleanupStatement(db, `DELETE FROM message_history_fts
|
|
614
1246
|
WHERE rowid IN (
|
|
615
1247
|
SELECT fts_rowid
|
|
616
1248
|
FROM message_fts_rowid_map
|
|
@@ -625,7 +1257,7 @@ function deleteSessionScopedRows(db, sessionIds, harness, options = {}) {
|
|
|
625
1257
|
const bindHarness = harness !== undefined && definition.harnessScoped === true;
|
|
626
1258
|
if (bindHarness)
|
|
627
1259
|
predicates.push("harness = ?");
|
|
628
|
-
const statement = db
|
|
1260
|
+
const statement = cleanupStatement(db, `DELETE FROM ${definition.table} WHERE ${predicates.join(" AND ")}`);
|
|
629
1261
|
if (bindHarness)
|
|
630
1262
|
statement.run(...deletableSessionIds, harness);
|
|
631
1263
|
else
|
|
@@ -633,9 +1265,9 @@ function deleteSessionScopedRows(db, sessionIds, harness, options = {}) {
|
|
|
633
1265
|
}
|
|
634
1266
|
deleteSessionActivity(db, deletableSessionIds);
|
|
635
1267
|
if (!options.preserveTagRevisionTombstones) {
|
|
636
|
-
const retireOwner = db
|
|
1268
|
+
const retireOwner = cleanupStatement(db, harness === undefined ? "DELETE FROM schema_migrations_meta WHERE key >= ? AND key < ?" : "DELETE FROM schema_migrations_meta WHERE key = ?");
|
|
637
1269
|
const survivingRows = harnessTables.map((definition) => `EXISTS (SELECT 1 FROM ${definition.table} WHERE session_id = ?${definition.extraPredicate ? ` AND ${definition.extraPredicate}` : ""})`).join(" OR ");
|
|
638
|
-
const retire = db
|
|
1270
|
+
const retire = cleanupStatement(db, `DELETE FROM schema_migrations_meta WHERE key IN (?, ?)
|
|
639
1271
|
AND NOT (${survivingRows})
|
|
640
1272
|
AND NOT EXISTS (SELECT 1 FROM schema_migrations_meta AS owners WHERE owners.key >= ? AND owners.key < ?)`);
|
|
641
1273
|
for (const id of deletableSessionIds) {
|
|
@@ -646,10 +1278,59 @@ function deleteSessionScopedRows(db, sessionIds, harness, options = {}) {
|
|
|
646
1278
|
retire.run(tagIdentityRevisionKey(id), tagRevisionKey(id), ...harnessTables.map(() => id), ...tagTombstoneOwnerRange(id));
|
|
647
1279
|
}
|
|
648
1280
|
}
|
|
649
|
-
|
|
650
|
-
|
|
1281
|
+
if (!options.deferRecoveryFenceCleanup)
|
|
1282
|
+
for (const sessionId of deletableSessionIds)
|
|
1283
|
+
clearLkgRecoveryFence(sessionId);
|
|
651
1284
|
return deletableSessionIds.length;
|
|
652
1285
|
}
|
|
1286
|
+
var SESSION_CLEANUP_SLICE_ROWS = 64;
|
|
1287
|
+
function deleteSessionScopedRowsSlice(db, sessionId, harness) {
|
|
1288
|
+
if (cleanupStatement(db, "SELECT 1 FROM pending_session_cleanup WHERE session_id = ? AND harness = ?").get(sessionId, `${harness}:rust`))
|
|
1289
|
+
return { complete: true, deleted: 0 };
|
|
1290
|
+
cleanupStatement(db, "INSERT OR IGNORE INTO schema_migrations_meta(key,value) VALUES (?,?)").run(tagTombstoneOwnerKey(sessionId, harness), harness);
|
|
1291
|
+
let remaining = SESSION_CLEANUP_SLICE_ROWS;
|
|
1292
|
+
const started = performance.now();
|
|
1293
|
+
remaining -= deleteChunkEmbedBackoffForSession(db, sessionId, remaining);
|
|
1294
|
+
if (remaining <= 0)
|
|
1295
|
+
return { complete: false, deleted: 0 };
|
|
1296
|
+
for (const definition of SESSION_SCOPED_TABLES) {
|
|
1297
|
+
if (definition.table === "message_history_fts") {
|
|
1298
|
+
const rows = cleanupStatement(db, "SELECT fts_rowid AS id FROM message_fts_rowid_map WHERE session_id = ? LIMIT ?").all(sessionId, remaining);
|
|
1299
|
+
const removeFts = cleanupStatement(db, "DELETE FROM message_history_fts WHERE rowid = ?");
|
|
1300
|
+
const removeMap = cleanupStatement(db, "DELETE FROM message_fts_rowid_map WHERE session_id = ? AND fts_rowid = ?");
|
|
1301
|
+
for (const row of rows) {
|
|
1302
|
+
removeFts.run(row.id);
|
|
1303
|
+
removeMap.run(sessionId, row.id);
|
|
1304
|
+
}
|
|
1305
|
+
remaining -= rows.length;
|
|
1306
|
+
if (remaining > 0 && !getMessageFtsRowidMapBackfillProgress2(db).completed)
|
|
1307
|
+
remaining -= cleanupStatement(db, `DELETE FROM message_history_fts WHERE rowid IN (
|
|
1308
|
+
SELECT rowid FROM message_history_fts WHERE session_id = ? LIMIT ?
|
|
1309
|
+
)`).run(sessionId, remaining).changes;
|
|
1310
|
+
} else {
|
|
1311
|
+
const predicates = ["session_id = ?"];
|
|
1312
|
+
if (definition.extraPredicate)
|
|
1313
|
+
predicates.push(definition.extraPredicate);
|
|
1314
|
+
if (definition.harnessScoped)
|
|
1315
|
+
predicates.push("harness = ?");
|
|
1316
|
+
const args = [sessionId, ...definition.harnessScoped ? [harness] : [], remaining];
|
|
1317
|
+
const withoutRowid = definition.table === "temporal_decisions" || definition.table === "session_replay_decisions";
|
|
1318
|
+
const key = withoutRowid ? "message_id" : "rowid";
|
|
1319
|
+
remaining -= cleanupStatement(db, `DELETE FROM ${definition.table}
|
|
1320
|
+
WHERE ${withoutRowid ? "session_id = ? AND " : ""}${key} IN (
|
|
1321
|
+
SELECT ${key} FROM ${definition.table} WHERE ${predicates.join(" AND ")} LIMIT ?
|
|
1322
|
+
)`).run(...withoutRowid ? [sessionId] : [], ...args).changes;
|
|
1323
|
+
}
|
|
1324
|
+
if (remaining <= 0 || performance.now() - started >= 10)
|
|
1325
|
+
return { complete: false, deleted: 0 };
|
|
1326
|
+
}
|
|
1327
|
+
return {
|
|
1328
|
+
complete: true,
|
|
1329
|
+
deleted: deleteSessionScopedRows(db, [sessionId], harness, {
|
|
1330
|
+
deferRecoveryFenceCleanup: true
|
|
1331
|
+
})
|
|
1332
|
+
};
|
|
1333
|
+
}
|
|
653
1334
|
|
|
654
1335
|
// ../plugin/src/features/magic-context/compression-depth-storage.ts
|
|
655
1336
|
var incrementDepthStatements = new WeakMap;
|
|
@@ -675,7 +1356,54 @@ function clearCompressionDepthRange(db, sessionId, startOrdinal, endOrdinal) {
|
|
|
675
1356
|
}
|
|
676
1357
|
|
|
677
1358
|
// ../plugin/src/features/magic-context/message-index.ts
|
|
678
|
-
import { createHash as
|
|
1359
|
+
import { createHash as createHash4 } from "node:crypto";
|
|
1360
|
+
|
|
1361
|
+
// ../plugin/src/shared/search-text.ts
|
|
1362
|
+
function removeTaskNotifications(text) {
|
|
1363
|
+
return removeNestedBlocks(text, ["task", "task_cohort"]);
|
|
1364
|
+
}
|
|
1365
|
+
function cleanSearchText(text) {
|
|
1366
|
+
return removeNestedBlocks(text, ["system-reminder", "task", "task_cohort"]).replace(/\s+/g, " ").trim();
|
|
1367
|
+
}
|
|
1368
|
+
var QUERY_STOP_WORDS = new Set("a an and are as at be been by can could did do does for from how i in is it me my of on or our please should that the their them there these they this to us was we were what when where which who why will with would you your".split(" "));
|
|
1369
|
+
function terms(text) {
|
|
1370
|
+
return [
|
|
1371
|
+
...new Set((text.toLowerCase().match(/[\p{L}\p{N}_]+/gu) ?? []).filter((term) => !QUERY_STOP_WORDS.has(term)))
|
|
1372
|
+
];
|
|
1373
|
+
}
|
|
1374
|
+
function keywordRelevance(query, content) {
|
|
1375
|
+
const needles = terms(query);
|
|
1376
|
+
if (needles.length === 0)
|
|
1377
|
+
return 0;
|
|
1378
|
+
const words = new Set(terms(cleanSearchText(content)));
|
|
1379
|
+
return 0.8 * needles.filter((term) => words.has(term)).length / needles.length;
|
|
1380
|
+
}
|
|
1381
|
+
function matchedSearchSnippet(content, query, cap = 220) {
|
|
1382
|
+
const text = cleanSearchText(content);
|
|
1383
|
+
const needles = new Set(terms(query));
|
|
1384
|
+
const matches = [...text.matchAll(/[\p{L}\p{N}_]+/gu)].filter((match) => needles.has(match[0].toLowerCase()));
|
|
1385
|
+
if (matches.length === 0)
|
|
1386
|
+
return "";
|
|
1387
|
+
let bestStart = matches[0].index;
|
|
1388
|
+
let bestCoverage = -1;
|
|
1389
|
+
for (const match of matches) {
|
|
1390
|
+
const start = Math.max(0, match.index - Math.floor(cap / 5));
|
|
1391
|
+
const coverage = keywordRelevance(query, text.slice(start, start + cap));
|
|
1392
|
+
if (coverage > bestCoverage) {
|
|
1393
|
+
bestStart = start;
|
|
1394
|
+
bestCoverage = coverage;
|
|
1395
|
+
}
|
|
1396
|
+
}
|
|
1397
|
+
if (bestStart > 0) {
|
|
1398
|
+
const boundary = text.indexOf(" ", bestStart);
|
|
1399
|
+
if (boundary >= 0 && boundary < matches[matches.length - 1].index)
|
|
1400
|
+
bestStart = boundary + 1;
|
|
1401
|
+
}
|
|
1402
|
+
const passage = text.slice(bestStart, bestStart + cap).trim();
|
|
1403
|
+
return `${bestStart > 0 ? "…" : ""}${passage}${bestStart + cap < text.length ? "…" : ""}`;
|
|
1404
|
+
}
|
|
1405
|
+
|
|
1406
|
+
// ../plugin/src/features/magic-context/message-index.ts
|
|
679
1407
|
var MESSAGE_HISTORY_ORPHAN_SWEEP_BATCH_SIZE = 200;
|
|
680
1408
|
var MESSAGE_HISTORY_ORPHAN_SAFETY_AGE_MS = 24 * 60 * 60 * 1000;
|
|
681
1409
|
var MESSAGE_HISTORY_ORPHAN_SWEEP_COOLDOWN_MS = 10 * 60 * 1000;
|
|
@@ -904,7 +1632,7 @@ function toMessageIndexSource2(message) {
|
|
|
904
1632
|
role: message.role,
|
|
905
1633
|
createdAt: message.createdAt,
|
|
906
1634
|
sourceVersion: normalizeSourceVersion(message.version),
|
|
907
|
-
contentHash:
|
|
1635
|
+
contentHash: createHash4("sha256").update(content).digest("hex"),
|
|
908
1636
|
content
|
|
909
1637
|
};
|
|
910
1638
|
}
|
|
@@ -1006,10 +1734,10 @@ function getIndexableContent2(role, parts) {
|
|
|
1006
1734
|
if (!hasMeaningfulUserText2(parts)) {
|
|
1007
1735
|
return "";
|
|
1008
1736
|
}
|
|
1009
|
-
return extractTexts2(parts).map(cleanUserText2).map(normalizeIndexText).filter((text) => text.length > 0).join(" / ");
|
|
1737
|
+
return extractTexts2(parts).map(removeTaskNotifications).map(cleanUserText2).map(normalizeIndexText).filter((text) => text.length > 0).join(" / ");
|
|
1010
1738
|
}
|
|
1011
1739
|
if (role === "assistant") {
|
|
1012
|
-
return extractTexts2(parts).map(removeSystemReminders).map(normalizeIndexText).filter((text) => text.length > 0).join(" / ");
|
|
1740
|
+
return extractTexts2(parts).map(removeTaskNotifications).map(removeSystemReminders).map(normalizeIndexText).filter((text) => text.length > 0).join(" / ");
|
|
1013
1741
|
}
|
|
1014
1742
|
return "";
|
|
1015
1743
|
}
|
|
@@ -1144,7 +1872,7 @@ function indexItemsAfterOrdinal(db, sessionId, items, finalWatermark, toSource)
|
|
|
1144
1872
|
return inserted;
|
|
1145
1873
|
}
|
|
1146
1874
|
function harnessSupportsOpenCodeOrphanSweep(harness) {
|
|
1147
|
-
return harness === "opencode";
|
|
1875
|
+
return harness === "opencode" || harness === "opencode2";
|
|
1148
1876
|
}
|
|
1149
1877
|
function getMessageHistoryOrphanSweepState(db, harness) {
|
|
1150
1878
|
return db.prepare("SELECT cursor_session_id, last_swept_at FROM message_history_orphan_sweep WHERE harness = ?").get(harness) ?? {};
|
|
@@ -1173,7 +1901,7 @@ function getOpenCodeSessionScopedCandidateSourceSql(harness) {
|
|
|
1173
1901
|
UNION
|
|
1174
1902
|
`);
|
|
1175
1903
|
}
|
|
1176
|
-
function sweepOrphanedOpenCodeMessageIndexesInBackground(db, openReadableOpenCodeDb, options = {}) {
|
|
1904
|
+
function* sweepOrphanedOpenCodeMessageIndexesInBackground(db, openReadableOpenCodeDb, options = {}) {
|
|
1177
1905
|
const now = options.now ?? Date.now();
|
|
1178
1906
|
const batchSize = Math.max(1, Math.floor(options.batchSize ?? MESSAGE_HISTORY_ORPHAN_SWEEP_BATCH_SIZE));
|
|
1179
1907
|
const safetyAgeMs = Math.max(0, options.safetyAgeMs ?? MESSAGE_HISTORY_ORPHAN_SAFETY_AGE_MS);
|
|
@@ -1187,15 +1915,6 @@ function sweepOrphanedOpenCodeMessageIndexesInBackground(db, openReadableOpenCod
|
|
|
1187
1915
|
}
|
|
1188
1916
|
if (!harnessSupportsOpenCodeOrphanSweep(harness)) {
|
|
1189
1917
|
persistMessageHistoryOrphanSweepState(db, cursor, now + unavailableReprobeMs - cooldownMs, harness);
|
|
1190
|
-
if (harness === "opencode2") {
|
|
1191
|
-
return {
|
|
1192
|
-
status: "unavailable",
|
|
1193
|
-
scanned: 0,
|
|
1194
|
-
deleted: 0,
|
|
1195
|
-
cursor,
|
|
1196
|
-
reason: "OpenCode 2 keeps its sessions in session_v2, which this sweep does not read"
|
|
1197
|
-
};
|
|
1198
|
-
}
|
|
1199
1918
|
return { status: "unavailable", scanned: 0, deleted: 0, cursor };
|
|
1200
1919
|
}
|
|
1201
1920
|
let openCodeDb = null;
|
|
@@ -1208,7 +1927,27 @@ function sweepOrphanedOpenCodeMessageIndexesInBackground(db, openReadableOpenCod
|
|
|
1208
1927
|
persistMessageHistoryOrphanSweepState(db, cursor, now + unavailableReprobeMs - cooldownMs, harness);
|
|
1209
1928
|
return { status: "source_unavailable", scanned: 0, deleted: 0, cursor };
|
|
1210
1929
|
}
|
|
1930
|
+
let v2Reader;
|
|
1211
1931
|
try {
|
|
1932
|
+
let sessionExists;
|
|
1933
|
+
if (harness === "opencode2") {
|
|
1934
|
+
try {
|
|
1935
|
+
const path = getSqliteDatabasePath2(openCodeDb);
|
|
1936
|
+
if (!path)
|
|
1937
|
+
throw new Error("OpenCode 2 source has no durable path");
|
|
1938
|
+
v2Reader = new V2StoreReader(path);
|
|
1939
|
+
const liveIds = new Set(v2Reader.sessionIds());
|
|
1940
|
+
if (liveIds.size === 0)
|
|
1941
|
+
throw new Error("OpenCode 2 source has no sessions");
|
|
1942
|
+
sessionExists = (id) => liveIds.has(id);
|
|
1943
|
+
} catch {
|
|
1944
|
+
persistMessageHistoryOrphanSweepState(db, cursor, now + unavailableReprobeMs - cooldownMs, harness);
|
|
1945
|
+
return { status: "source_unavailable", scanned: 0, deleted: 0, cursor };
|
|
1946
|
+
}
|
|
1947
|
+
} else {
|
|
1948
|
+
const query = openCodeDb.prepare("SELECT 1 FROM session WHERE id = ? LIMIT 1");
|
|
1949
|
+
sessionExists = (id) => !!query.get(id);
|
|
1950
|
+
}
|
|
1212
1951
|
const cutoff = now - safetyAgeMs;
|
|
1213
1952
|
const candidateSourceSql = getOpenCodeSessionScopedCandidateSourceSql(harness);
|
|
1214
1953
|
const candidates = db.prepare(`SELECT session_id
|
|
@@ -1223,39 +1962,56 @@ function sweepOrphanedOpenCodeMessageIndexesInBackground(db, openReadableOpenCod
|
|
|
1223
1962
|
)
|
|
1224
1963
|
ORDER BY session_id ASC
|
|
1225
1964
|
LIMIT ?`).all(cursor, cutoff, batchSize);
|
|
1226
|
-
const
|
|
1227
|
-
const missingSessionIds = candidates.filter((candidate) => !sessionExists.get(candidate.session_id)).map((candidate) => candidate.session_id);
|
|
1965
|
+
const missingSessionIds = candidates.filter((candidate) => !sessionExists(candidate.session_id)).map((candidate) => candidate.session_id);
|
|
1228
1966
|
const nextCursor = candidates.length < batchSize ? "" : candidates[candidates.length - 1]?.session_id ?? cursor;
|
|
1229
1967
|
const completedAt = candidates.length < batchSize ? now : null;
|
|
1230
|
-
db.exec("BEGIN IMMEDIATE");
|
|
1231
|
-
const transactionStartedAt = performance.now();
|
|
1232
|
-
let committed = false;
|
|
1233
1968
|
let deleted = 0;
|
|
1234
|
-
|
|
1235
|
-
|
|
1236
|
-
|
|
1237
|
-
|
|
1238
|
-
|
|
1239
|
-
|
|
1240
|
-
|
|
1241
|
-
|
|
1242
|
-
|
|
1243
|
-
|
|
1244
|
-
|
|
1245
|
-
|
|
1246
|
-
const eligibleSessionIds = missingSessionIds.filter((sessionId) => !sessionExists.get(sessionId) && stillEligible.get(sessionId, cutoff));
|
|
1247
|
-
deleted = deleteSessionScopedRows(db, eligibleSessionIds, harness);
|
|
1248
|
-
persistMessageHistoryOrphanSweepState(db, nextCursor, completedAt, harness);
|
|
1249
|
-
db.exec("COMMIT");
|
|
1250
|
-
committed = true;
|
|
1251
|
-
logSlowWriteTransaction("message_index_orphan_sweep", transactionStartedAt);
|
|
1252
|
-
} finally {
|
|
1253
|
-
if (!committed) {
|
|
1969
|
+
let slices = 0;
|
|
1970
|
+
const stillEligible = db.prepare(`SELECT 1 WHERE NOT EXISTS (
|
|
1971
|
+
SELECT 1 FROM message_history_index
|
|
1972
|
+
WHERE session_id = ? AND harness = '${harness}' AND updated_at > ?
|
|
1973
|
+
)`);
|
|
1974
|
+
for (const sessionId of missingSessionIds) {
|
|
1975
|
+
let complete = false;
|
|
1976
|
+
do {
|
|
1977
|
+
let removed = 0;
|
|
1978
|
+
db.exec("BEGIN IMMEDIATE");
|
|
1979
|
+
const transactionStartedAt = performance.now();
|
|
1980
|
+
let committed = false;
|
|
1254
1981
|
try {
|
|
1255
|
-
|
|
1256
|
-
|
|
1257
|
-
|
|
1982
|
+
const hostStillMissing = v2Reader ? !v2Reader.sessionExists(sessionId) : !sessionExists(sessionId);
|
|
1983
|
+
if (hostStillMissing && stillEligible.get(sessionId, cutoff)) {
|
|
1984
|
+
const slice = deleteSessionScopedRowsSlice(db, sessionId, harness);
|
|
1985
|
+
removed = slice.deleted;
|
|
1986
|
+
complete = slice.complete;
|
|
1987
|
+
} else
|
|
1988
|
+
complete = true;
|
|
1989
|
+
db.exec("COMMIT");
|
|
1990
|
+
committed = true;
|
|
1991
|
+
logSlowWriteTransaction("message_index_orphan_sweep", transactionStartedAt);
|
|
1992
|
+
} finally {
|
|
1993
|
+
if (!committed) {
|
|
1994
|
+
try {
|
|
1995
|
+
db.exec("ROLLBACK");
|
|
1996
|
+
} catch {}
|
|
1997
|
+
}
|
|
1998
|
+
}
|
|
1999
|
+
deleted += removed;
|
|
2000
|
+
if (removed)
|
|
2001
|
+
clearLkgRecoveryFence(sessionId);
|
|
2002
|
+
yield;
|
|
2003
|
+
if (++slices >= Math.max(1, options.maxSlices ?? 512)) {
|
|
2004
|
+
return {
|
|
2005
|
+
status: "swept",
|
|
2006
|
+
scanned: candidates.length,
|
|
2007
|
+
deleted,
|
|
2008
|
+
cursor,
|
|
2009
|
+
partial: true
|
|
2010
|
+
};
|
|
2011
|
+
}
|
|
2012
|
+
} while (!complete);
|
|
1258
2013
|
}
|
|
2014
|
+
persistMessageHistoryOrphanSweepState(db, nextCursor, completedAt, harness);
|
|
1259
2015
|
return {
|
|
1260
2016
|
status: "swept",
|
|
1261
2017
|
scanned: candidates.length,
|
|
@@ -1263,11 +2019,24 @@ function sweepOrphanedOpenCodeMessageIndexesInBackground(db, openReadableOpenCod
|
|
|
1263
2019
|
cursor: nextCursor
|
|
1264
2020
|
};
|
|
1265
2021
|
} finally {
|
|
2022
|
+
v2Reader?.close();
|
|
1266
2023
|
closeQuietly(openCodeDb);
|
|
1267
2024
|
}
|
|
1268
2025
|
}
|
|
1269
|
-
function
|
|
1270
|
-
return withoutSqliteTransformPass2(() => withSqliteBackgroundWriter2(() =>
|
|
2026
|
+
async function sweepOrphanedOpenCodeMessageIndexesCooperatively2(db, openSource, options = {}) {
|
|
2027
|
+
return withoutSqliteTransformPass2(() => withSqliteBackgroundWriter2(async () => {
|
|
2028
|
+
const sweep = sweepOrphanedOpenCodeMessageIndexesInBackground(db, openSource, options);
|
|
2029
|
+
try {
|
|
2030
|
+
for (;; ) {
|
|
2031
|
+
const step = sweep.next();
|
|
2032
|
+
if (step.done)
|
|
2033
|
+
return step.value;
|
|
2034
|
+
await new Promise((resolve) => setTimeout(resolve, 0));
|
|
2035
|
+
}
|
|
2036
|
+
} finally {
|
|
2037
|
+
sweep.return({ status: "source_unavailable", scanned: 0, deleted: 0, cursor: "" });
|
|
2038
|
+
}
|
|
2039
|
+
}));
|
|
1271
2040
|
}
|
|
1272
2041
|
|
|
1273
|
-
export { BoundedSessionMap, LkgRecoveryFenceError, clearLkgFenceProcessState, lkgFenceRequiresRebuild, lkgRecoveryFenceUnclean, ownsLkgDirtyFence, invalidateCleanLkgRecoveryFence, markLkgRecoveryDirty, commitLkgWithRecoveryFence, clearLkgRecoveryFence, readSessionActivity, observeSessionActivity, backfillSessionActivity, deleteSessionScopedRows, clearCompressionDepth, clearCompressionDepthRange, MESSAGE_INDEX_MAX_FTS_BYTES2, messageIndexSkipKey2, toMessageIndexSource2, getMessageIndexSourceIdentity2, isMessageIndexSourceCurrent2, getLastIndexedOrdinal2, getIndexedMessageCorpusSize2, markMessageIndexDirty2, getMessageIndexReconciliationStartOrdinal2, isMessageIndexReconciledThrough2, deleteIndexedMessage2, clearIndexedMessagesInTransaction2, clearIndexedMessages2, getIndexableContent2, indexSingleMessage2, indexMessagesAfterOrdinal2, indexSourcesAfterOrdinal2,
|
|
2042
|
+
export { BoundedSessionMap, LkgRecoveryFenceError, clearLkgFenceProcessState, lkgFenceRequiresRebuild, lkgRecoveryFenceUnclean, ownsLkgDirtyFence, invalidateCleanLkgRecoveryFence, markLkgRecoveryDirty, commitLkgWithRecoveryFence, clearLkgRecoveryFence, gaDatabasePath, V2StoreReader, readSessionActivity, observeSessionActivity, backfillSessionActivity, deleteSessionScopedRows, clearCompressionDepth, clearCompressionDepthRange, cleanSearchText, keywordRelevance, matchedSearchSnippet, MESSAGE_INDEX_MAX_FTS_BYTES2, messageIndexSkipKey2, toMessageIndexSource2, getMessageIndexSourceIdentity2, isMessageIndexSourceCurrent2, getLastIndexedOrdinal2, getIndexedMessageCorpusSize2, markMessageIndexDirty2, getMessageIndexReconciliationStartOrdinal2, isMessageIndexReconciledThrough2, deleteIndexedMessage2, clearIndexedMessagesInTransaction2, clearIndexedMessages2, getIndexableContent2, indexSingleMessage2, indexMessagesAfterOrdinal2, indexSourcesAfterOrdinal2, sweepOrphanedOpenCodeMessageIndexesCooperatively2 };
|