@cortexkit/pi-magic-context 0.47.0 → 0.48.0
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 +27 -22
- package/dist/embedding-activity-ew6dshfd.js +10 -0
- package/dist/embedding-backfill-bnnytehy.js +20 -0
- package/dist/embedding-cache-7rp4gqzr.js +16 -0
- package/dist/embedding-worker.js +11 -8
- package/dist/harness-9872e0ym.js +10 -0
- package/dist/historian-calibration-extension.js +3 -2
- package/dist/index-0am8a7k1.js +152 -0
- package/dist/index-0h2mkab2.js +17 -0
- package/dist/index-135e8hvg.js +43 -0
- package/dist/{index-0qejsd7f.js → index-3cgbv47m.js} +289 -45
- package/dist/index-419bndtm.js +2309 -0
- package/dist/index-544c0e0x.js +3026 -0
- package/dist/index-5hd7h4rr.js +3098 -0
- package/dist/index-5jp1h0bz.js +750 -0
- package/dist/index-603mt6nm.js +171 -0
- package/dist/index-674f6qs2.js +16 -0
- package/dist/{index-vgcemg0n.js → index-6pgq5ns2.js} +151 -2452
- package/dist/{index-xpbh1ypk.js → index-72bdc1bd.js} +38 -741
- package/dist/index-8976mezv.js +209 -0
- package/dist/{index-td885aah.js → index-8y5knskr.js} +98 -231
- package/dist/{index-k3ezsjng.js → index-98z1c1r0.js} +14574 -13942
- package/dist/index-9p40dgkm.js +22 -0
- package/dist/index-9x1f96r8.js +11 -0
- package/dist/{index-266qrgss.js → index-9xb0axhc.js} +301 -27
- package/dist/{index-spk5vrhy.js → index-bgmv2kaf.js} +1 -16
- package/dist/index-dkx0b1kt.js +514 -0
- package/dist/{index-7n8gb6j3.js → index-f956htf3.js} +28 -31
- package/dist/index-fapenpch.js +13 -0
- package/dist/index-fm3ksh5e.js +1060 -0
- package/dist/{index-2p1e8706.js → index-fp37he9a.js} +3 -3
- package/dist/index-hrf13mn3.js +2388 -0
- package/dist/index-pvm4pd5d.js +3069 -0
- package/dist/{index-kztsbkqq.js → index-r7xg4hvw.js} +171 -168
- package/dist/index-rt5jgzp1.js +1127 -0
- package/dist/{index-n3g5mwa8.js → index-wrm4nbj7.js} +3 -3
- package/dist/index-xjqheg7a.js +931 -0
- package/dist/index-yhzhdp3m.js +143 -0
- package/dist/{index-w76x3af7.js → index-zrr5t5a7.js} +1 -3
- package/dist/index.js +4203 -2183
- package/dist/logger-v3dsxdnm.js +19 -0
- package/dist/message-index-bounded-t7vvty76.js +120 -0
- package/dist/{message-time-backfill-d3db2etr.js → message-time-backfill-4ndf7hrq.js} +7 -6
- package/dist/migration-worker.js +20 -14
- package/dist/project-embedding-registry-jy2hmq28.js +85 -0
- package/dist/raw-ordinal-warmup-worker.js +60 -0
- package/dist/{read-session-chunk-atxsyfqv.js → read-session-chunk-47pjn7fz.js} +12 -6
- package/dist/read-session-raw-yn8nawav.js +55 -0
- package/dist/{safe-notification-target-6cn3z99t.js → safe-notification-target-3c1ns931.js} +3 -2
- package/dist/search-42szfhwp.js +40 -0
- package/dist/sqlite-3yye76ht.js +44 -0
- package/dist/sqlite-context-pragmas-gcgch6qz.js +6 -0
- package/dist/storage-memory-1z20y55x.js +71 -0
- package/dist/storage-tags-jym5z6yk.js +127 -0
- package/dist/subagent-entry.js +34 -17
- package/dist/transformers-node-wasm.js +18415 -18415
- package/package.json +3 -3
- package/dist/index-7s8vw2y0.js +0 -5207
- package/dist/index-qaa5j277.js +0 -8544
- /package/dist/{index-en1k8h8m.js → index-5prbx026.js} +0 -0
|
@@ -0,0 +1,750 @@
|
|
|
1
|
+
import {
|
|
2
|
+
removeSystemReminders,
|
|
3
|
+
hasMeaningfulUserText2,
|
|
4
|
+
extractTexts2
|
|
5
|
+
} from "./index-9xb0axhc.js";
|
|
6
|
+
import {
|
|
7
|
+
getHarness2
|
|
8
|
+
} from "./index-0h2mkab2.js";
|
|
9
|
+
import {
|
|
10
|
+
withSqliteBackgroundWriter2,
|
|
11
|
+
withoutSqliteTransformPass2
|
|
12
|
+
} from "./index-3cgbv47m.js";
|
|
13
|
+
import {
|
|
14
|
+
recordMessageFtsRowid,
|
|
15
|
+
recordIndexedMessageTime,
|
|
16
|
+
messageFtsOrdinalRangeIsMapped,
|
|
17
|
+
deleteUnmappedMessageFtsRows
|
|
18
|
+
} from "./index-603mt6nm.js";
|
|
19
|
+
import {
|
|
20
|
+
deleteChunkEmbedBackoffForSession
|
|
21
|
+
} from "./index-fm3ksh5e.js";
|
|
22
|
+
import {
|
|
23
|
+
closeQuietly,
|
|
24
|
+
deleteTagIdentityRevisions
|
|
25
|
+
} from "./index-8y5knskr.js";
|
|
26
|
+
import {
|
|
27
|
+
cleanUserText2
|
|
28
|
+
} from "./index-hrf13mn3.js";
|
|
29
|
+
import {
|
|
30
|
+
logSlowWriteTransaction
|
|
31
|
+
} from "./index-fp37he9a.js";
|
|
32
|
+
|
|
33
|
+
// ../plugin/src/features/magic-context/session-activity.ts
|
|
34
|
+
var PREFIX = "retrospective_activity:";
|
|
35
|
+
var INTERVAL_MS = 15000;
|
|
36
|
+
var pending = new WeakMap;
|
|
37
|
+
function sessionActivityKey(sessionId) {
|
|
38
|
+
return `${PREFIX}${sessionId}`;
|
|
39
|
+
}
|
|
40
|
+
function deleteSessionActivity(db, sessionIds) {
|
|
41
|
+
if (sessionIds.length === 0)
|
|
42
|
+
return;
|
|
43
|
+
const sessions = pending.get(db);
|
|
44
|
+
for (const sessionId of sessionIds) {
|
|
45
|
+
const state = sessions?.get(sessionId);
|
|
46
|
+
if (state?.timer)
|
|
47
|
+
clearTimeout(state.timer);
|
|
48
|
+
sessions?.delete(sessionId);
|
|
49
|
+
}
|
|
50
|
+
const placeholders = sessionIds.map(() => "?").join(", ");
|
|
51
|
+
db.prepare(`DELETE FROM schema_migrations_meta WHERE key IN (${placeholders})`).run(...sessionIds.map(sessionActivityKey));
|
|
52
|
+
deleteTagIdentityRevisions(db, sessionIds);
|
|
53
|
+
}
|
|
54
|
+
function readSessionActivity(db, sessionId) {
|
|
55
|
+
const row = db.prepare("SELECT value FROM schema_migrations_meta WHERE key = ?").get(sessionActivityKey(sessionId));
|
|
56
|
+
const time = Number(row?.value);
|
|
57
|
+
return row && Number.isFinite(time) ? time : undefined;
|
|
58
|
+
}
|
|
59
|
+
function advanceSessionActivity(db, sessionId, time) {
|
|
60
|
+
if (!sessionId || !Number.isFinite(time) || time <= 0)
|
|
61
|
+
return;
|
|
62
|
+
db.prepare(`INSERT INTO schema_migrations_meta (key, value) VALUES (?, ?)
|
|
63
|
+
ON CONFLICT(key) DO UPDATE SET value = CAST(MAX(CAST(value AS INTEGER), CAST(excluded.value AS INTEGER)) AS TEXT)`).run(sessionActivityKey(sessionId), String(Math.floor(time)));
|
|
64
|
+
}
|
|
65
|
+
function observeSessionActivity(db, sessionId, time = Date.now()) {
|
|
66
|
+
if (!sessionId || !Number.isFinite(time) || time <= 0)
|
|
67
|
+
return;
|
|
68
|
+
let sessions = pending.get(db);
|
|
69
|
+
if (!sessions) {
|
|
70
|
+
sessions = new Map;
|
|
71
|
+
pending.set(db, sessions);
|
|
72
|
+
}
|
|
73
|
+
const current = sessions.get(sessionId);
|
|
74
|
+
if (!current || time - current.lastWrite >= INTERVAL_MS) {
|
|
75
|
+
if (current?.timer)
|
|
76
|
+
clearTimeout(current.timer);
|
|
77
|
+
advanceSessionActivity(db, sessionId, Math.max(time, current?.latest ?? 0));
|
|
78
|
+
sessions.set(sessionId, { lastWrite: time, latest: time });
|
|
79
|
+
return;
|
|
80
|
+
}
|
|
81
|
+
current.latest = Math.max(current.latest, time);
|
|
82
|
+
if (!current.timer) {
|
|
83
|
+
current.timer = setTimeout(() => {
|
|
84
|
+
current.timer = undefined;
|
|
85
|
+
try {
|
|
86
|
+
advanceSessionActivity(db, sessionId, current.latest);
|
|
87
|
+
current.lastWrite = Date.now();
|
|
88
|
+
} catch {
|
|
89
|
+
sessions.delete(sessionId);
|
|
90
|
+
}
|
|
91
|
+
}, Math.max(1, INTERVAL_MS - (time - current.lastWrite)));
|
|
92
|
+
current.timer.unref?.();
|
|
93
|
+
}
|
|
94
|
+
}
|
|
95
|
+
async function backfillSessionActivity(db, harness, latestTime) {
|
|
96
|
+
const completedKey = `retrospective_activity_backfill:${harness}:v1`;
|
|
97
|
+
if (db.prepare("SELECT 1 FROM schema_migrations_meta WHERE key = ?").get(completedKey))
|
|
98
|
+
return;
|
|
99
|
+
let cursor = "";
|
|
100
|
+
while (true) {
|
|
101
|
+
const rows = db.prepare(`SELECT session_id FROM session_projects
|
|
102
|
+
WHERE harness = ? AND session_id > ? ORDER BY session_id LIMIT 100`).all(harness, cursor);
|
|
103
|
+
if (rows.length === 0)
|
|
104
|
+
break;
|
|
105
|
+
for (const row of rows) {
|
|
106
|
+
const time = await latestTime(row.session_id);
|
|
107
|
+
if (time !== undefined)
|
|
108
|
+
advanceSessionActivity(db, row.session_id, time);
|
|
109
|
+
}
|
|
110
|
+
cursor = rows[rows.length - 1].session_id;
|
|
111
|
+
await new Promise((resolve) => setTimeout(resolve, 0));
|
|
112
|
+
}
|
|
113
|
+
db.prepare("INSERT OR IGNORE INTO schema_migrations_meta (key, value) VALUES (?, 'completed')").run(completedKey);
|
|
114
|
+
}
|
|
115
|
+
|
|
116
|
+
// ../plugin/src/features/magic-context/storage-session-tables.ts
|
|
117
|
+
var SESSION_SCOPED_TABLES = [
|
|
118
|
+
{ table: "pending_ops", harnessScoped: true },
|
|
119
|
+
{ table: "source_contents", harnessScoped: true },
|
|
120
|
+
{ table: "tool_owner_backfill_state" },
|
|
121
|
+
{ table: "tags", harnessScoped: true },
|
|
122
|
+
{ table: "session_meta", harnessScoped: true },
|
|
123
|
+
{
|
|
124
|
+
table: "temporal_decisions",
|
|
125
|
+
extraPredicate: "NOT EXISTS (SELECT 1 FROM session_meta AS remaining WHERE remaining.session_id = temporal_decisions.session_id)"
|
|
126
|
+
},
|
|
127
|
+
{
|
|
128
|
+
table: "session_replay_decisions",
|
|
129
|
+
extraPredicate: "NOT EXISTS (SELECT 1 FROM session_meta AS remaining WHERE remaining.session_id = session_replay_decisions.session_id)"
|
|
130
|
+
},
|
|
131
|
+
{ table: "session_projects", harnessScoped: true },
|
|
132
|
+
{ table: "compartment_chunk_embeddings", harnessScoped: true },
|
|
133
|
+
{ table: "compartments", harnessScoped: true },
|
|
134
|
+
{
|
|
135
|
+
table: "compartment_history_versions",
|
|
136
|
+
extraPredicate: "NOT EXISTS (SELECT 1 FROM compartments AS remaining WHERE remaining.session_id = compartment_history_versions.session_id)"
|
|
137
|
+
},
|
|
138
|
+
{ table: "compression_depth", harnessScoped: true },
|
|
139
|
+
{ table: "session_facts", harnessScoped: true },
|
|
140
|
+
{ table: "compartment_state_lease" },
|
|
141
|
+
{ table: "notes", harnessScoped: true, extraPredicate: "type = 'session'" },
|
|
142
|
+
{ table: "recomp_compartments", harnessScoped: true },
|
|
143
|
+
{ table: "recomp_facts", harnessScoped: true },
|
|
144
|
+
{ table: "user_memory_candidates" },
|
|
145
|
+
{ table: "primer_candidates", harnessScoped: true },
|
|
146
|
+
{ table: "m0_mutation_log" },
|
|
147
|
+
{ table: "compartment_events", harnessScoped: true },
|
|
148
|
+
{ table: "subagent_invocations", harnessScoped: true },
|
|
149
|
+
{ table: "historian_runs", harnessScoped: true },
|
|
150
|
+
{ table: "plugin_messages" },
|
|
151
|
+
{ table: "transform_decisions", harnessScoped: true },
|
|
152
|
+
{ table: "synapse_batch_ledger" },
|
|
153
|
+
{ table: "embedding_measurement_corpus" },
|
|
154
|
+
{ table: "pending_session_cleanup", harnessScoped: true },
|
|
155
|
+
{ table: "message_history_fts" },
|
|
156
|
+
{ table: "message_fts_rowid_map" },
|
|
157
|
+
{ table: "message_history_source", harnessScoped: true },
|
|
158
|
+
{ table: "message_history_index", harnessScoped: true },
|
|
159
|
+
{ table: "lkg_slots" },
|
|
160
|
+
{ table: "lkg_slot_chunks" }
|
|
161
|
+
];
|
|
162
|
+
function deleteSessionScopedRows(db, sessionIds, harness, options = {}) {
|
|
163
|
+
if (sessionIds.length === 0)
|
|
164
|
+
return 0;
|
|
165
|
+
let deletableSessionIds = [...sessionIds];
|
|
166
|
+
if (options.rustModuleCleanupAcknowledged !== true) {
|
|
167
|
+
const placeholders = sessionIds.map(() => "?").join(", ");
|
|
168
|
+
const harnessPredicate = harness === undefined ? "" : " AND harness = ?";
|
|
169
|
+
const protectedRows = db.prepare(`SELECT session_id
|
|
170
|
+
FROM pending_session_cleanup
|
|
171
|
+
WHERE session_id IN (${placeholders})
|
|
172
|
+
AND harness LIKE '%:rust'${harnessPredicate}`).all(...sessionIds, ...harness === undefined ? [] : [`${harness}:rust`]);
|
|
173
|
+
const protectedSessionIds = new Set(protectedRows.map((row) => row.session_id));
|
|
174
|
+
deletableSessionIds = sessionIds.filter((sessionId) => !protectedSessionIds.has(sessionId));
|
|
175
|
+
}
|
|
176
|
+
if (deletableSessionIds.length === 0)
|
|
177
|
+
return 0;
|
|
178
|
+
const placeholders = deletableSessionIds.map(() => "?").join(", ");
|
|
179
|
+
for (const sessionId of deletableSessionIds)
|
|
180
|
+
deleteChunkEmbedBackoffForSession(db, sessionId);
|
|
181
|
+
for (const definition of SESSION_SCOPED_TABLES) {
|
|
182
|
+
if (definition.table === "message_history_fts") {
|
|
183
|
+
db.prepare(`DELETE FROM message_history_fts
|
|
184
|
+
WHERE rowid IN (
|
|
185
|
+
SELECT fts_rowid
|
|
186
|
+
FROM message_fts_rowid_map
|
|
187
|
+
WHERE session_id IN (${placeholders})
|
|
188
|
+
)`).run(...deletableSessionIds);
|
|
189
|
+
deleteUnmappedMessageFtsRows(db, deletableSessionIds);
|
|
190
|
+
continue;
|
|
191
|
+
}
|
|
192
|
+
const predicates = [`session_id IN (${placeholders})`];
|
|
193
|
+
if (definition.extraPredicate)
|
|
194
|
+
predicates.push(definition.extraPredicate);
|
|
195
|
+
const bindHarness = harness !== undefined && definition.harnessScoped === true;
|
|
196
|
+
if (bindHarness)
|
|
197
|
+
predicates.push("harness = ?");
|
|
198
|
+
const statement = db.prepare(`DELETE FROM ${definition.table} WHERE ${predicates.join(" AND ")}`);
|
|
199
|
+
if (bindHarness)
|
|
200
|
+
statement.run(...deletableSessionIds, harness);
|
|
201
|
+
else
|
|
202
|
+
statement.run(...deletableSessionIds);
|
|
203
|
+
}
|
|
204
|
+
deleteSessionActivity(db, deletableSessionIds);
|
|
205
|
+
return deletableSessionIds.length;
|
|
206
|
+
}
|
|
207
|
+
|
|
208
|
+
// ../plugin/src/features/magic-context/compression-depth-storage.ts
|
|
209
|
+
var incrementDepthStatements = new WeakMap;
|
|
210
|
+
var totalDepthStatements = new WeakMap;
|
|
211
|
+
var maxDepthStatements = new WeakMap;
|
|
212
|
+
var clearDepthStatements = new WeakMap;
|
|
213
|
+
function getClearDepthStatement(db) {
|
|
214
|
+
let stmt = clearDepthStatements.get(db);
|
|
215
|
+
if (!stmt) {
|
|
216
|
+
stmt = db.prepare("DELETE FROM compression_depth WHERE session_id = ?");
|
|
217
|
+
clearDepthStatements.set(db, stmt);
|
|
218
|
+
}
|
|
219
|
+
return stmt;
|
|
220
|
+
}
|
|
221
|
+
function clearCompressionDepth(db, sessionId) {
|
|
222
|
+
getClearDepthStatement(db).run(sessionId);
|
|
223
|
+
}
|
|
224
|
+
function clearCompressionDepthRange(db, sessionId, startOrdinal, endOrdinal) {
|
|
225
|
+
if (endOrdinal < startOrdinal) {
|
|
226
|
+
return;
|
|
227
|
+
}
|
|
228
|
+
db.prepare("DELETE FROM compression_depth WHERE session_id = ? AND message_ordinal BETWEEN ? AND ?").run(sessionId, startOrdinal, endOrdinal);
|
|
229
|
+
}
|
|
230
|
+
|
|
231
|
+
// ../plugin/src/features/magic-context/message-index.ts
|
|
232
|
+
import { createHash } from "node:crypto";
|
|
233
|
+
var MESSAGE_HISTORY_ORPHAN_SWEEP_BATCH_SIZE = 200;
|
|
234
|
+
var MESSAGE_HISTORY_ORPHAN_SAFETY_AGE_MS = 24 * 60 * 60 * 1000;
|
|
235
|
+
var MESSAGE_HISTORY_ORPHAN_SWEEP_COOLDOWN_MS = 10 * 60 * 1000;
|
|
236
|
+
var MESSAGE_HISTORY_ORPHAN_UNAVAILABLE_REPROBE_MS = 24 * 60 * 60 * 1000;
|
|
237
|
+
var lastIndexedStatements = new WeakMap;
|
|
238
|
+
var insertMessageStatements = new WeakMap;
|
|
239
|
+
var upsertProgressStatements = new WeakMap;
|
|
240
|
+
var upsertDirtyFloorStatements = new WeakMap;
|
|
241
|
+
var deleteFtsStatements = new WeakMap;
|
|
242
|
+
var deleteFtsRangeStatements = new WeakMap;
|
|
243
|
+
var deleteIndexStatements = new WeakMap;
|
|
244
|
+
var countIndexedMessageStatements = new WeakMap;
|
|
245
|
+
var getMessageSourceStatements = new WeakMap;
|
|
246
|
+
var upsertMessageSourceStatements = new WeakMap;
|
|
247
|
+
var deleteMessageSourceStatements = new WeakMap;
|
|
248
|
+
var deleteMessageSourceRangeStatements = new WeakMap;
|
|
249
|
+
var deleteMessageFtsStatements = new WeakMap;
|
|
250
|
+
var deleteFtsMapStatements = new WeakMap;
|
|
251
|
+
var deleteFtsMapRangeStatements = new WeakMap;
|
|
252
|
+
var deleteMessageFtsMapStatements = new WeakMap;
|
|
253
|
+
var MESSAGE_INDEX_MAX_FTS_BYTES = 64 * 1024;
|
|
254
|
+
var skipStatements = new WeakMap;
|
|
255
|
+
function skipStatement(db, sql) {
|
|
256
|
+
const statements = skipStatements.get(db) ?? new Map;
|
|
257
|
+
skipStatements.set(db, statements);
|
|
258
|
+
let statement = statements.get(sql);
|
|
259
|
+
if (!statement) {
|
|
260
|
+
statement = db.prepare(sql);
|
|
261
|
+
statements.set(sql, statement);
|
|
262
|
+
}
|
|
263
|
+
return statement;
|
|
264
|
+
}
|
|
265
|
+
var messageIndexSkipKey = (sessionId, messageId) => `message_index_skip:${JSON.stringify([sessionId, messageId])}`;
|
|
266
|
+
function normalizeIndexText(text) {
|
|
267
|
+
return text.replace(/\s+/g, " ").trim();
|
|
268
|
+
}
|
|
269
|
+
function getLastIndexedStatement(db) {
|
|
270
|
+
let stmt = lastIndexedStatements.get(db);
|
|
271
|
+
if (!stmt) {
|
|
272
|
+
stmt = db.prepare("SELECT last_indexed_ordinal, dirty_floor_ordinal FROM message_history_index WHERE session_id = ?");
|
|
273
|
+
lastIndexedStatements.set(db, stmt);
|
|
274
|
+
}
|
|
275
|
+
return stmt;
|
|
276
|
+
}
|
|
277
|
+
function getInsertMessageStatement(db) {
|
|
278
|
+
let stmt = insertMessageStatements.get(db);
|
|
279
|
+
if (!stmt) {
|
|
280
|
+
stmt = db.prepare("INSERT INTO message_history_fts (session_id, message_ordinal, message_id, role, content) VALUES (?, ?, ?, ?, ?)");
|
|
281
|
+
insertMessageStatements.set(db, stmt);
|
|
282
|
+
}
|
|
283
|
+
return stmt;
|
|
284
|
+
}
|
|
285
|
+
function getUpsertProgressStatement(db) {
|
|
286
|
+
let stmt = upsertProgressStatements.get(db);
|
|
287
|
+
if (!stmt) {
|
|
288
|
+
stmt = db.prepare("INSERT INTO message_history_index (session_id, last_indexed_ordinal, dirty_floor_ordinal, updated_at, harness) VALUES (?, ?, ?, ?, ?) ON CONFLICT(session_id) DO UPDATE SET last_indexed_ordinal = excluded.last_indexed_ordinal, dirty_floor_ordinal = excluded.dirty_floor_ordinal, updated_at = excluded.updated_at");
|
|
289
|
+
upsertProgressStatements.set(db, stmt);
|
|
290
|
+
}
|
|
291
|
+
return stmt;
|
|
292
|
+
}
|
|
293
|
+
function getUpsertDirtyFloorStatement(db) {
|
|
294
|
+
let stmt = upsertDirtyFloorStatements.get(db);
|
|
295
|
+
if (!stmt) {
|
|
296
|
+
stmt = db.prepare("INSERT INTO message_history_index (session_id, last_indexed_ordinal, dirty_floor_ordinal, updated_at, harness) VALUES (?, ?, ?, ?, ?) ON CONFLICT(session_id) DO UPDATE SET last_indexed_ordinal = MAX(message_history_index.last_indexed_ordinal, excluded.last_indexed_ordinal), dirty_floor_ordinal = CASE WHEN message_history_index.dirty_floor_ordinal <= 0 THEN excluded.dirty_floor_ordinal WHEN excluded.dirty_floor_ordinal <= 0 THEN message_history_index.dirty_floor_ordinal ELSE MIN(message_history_index.dirty_floor_ordinal, excluded.dirty_floor_ordinal) END, updated_at = excluded.updated_at");
|
|
297
|
+
upsertDirtyFloorStatements.set(db, stmt);
|
|
298
|
+
}
|
|
299
|
+
return stmt;
|
|
300
|
+
}
|
|
301
|
+
function getDeleteFtsRangeStatement(db) {
|
|
302
|
+
let stmt = deleteFtsRangeStatements.get(db);
|
|
303
|
+
if (!stmt) {
|
|
304
|
+
stmt = db.prepare(`DELETE FROM message_history_fts
|
|
305
|
+
WHERE rowid IN (
|
|
306
|
+
SELECT fts_rowid
|
|
307
|
+
FROM message_fts_rowid_map
|
|
308
|
+
WHERE session_id = ? AND message_ordinal BETWEEN ? AND ?
|
|
309
|
+
)`);
|
|
310
|
+
deleteFtsRangeStatements.set(db, stmt);
|
|
311
|
+
}
|
|
312
|
+
return stmt;
|
|
313
|
+
}
|
|
314
|
+
function getCountIndexedMessageStatement(db) {
|
|
315
|
+
let stmt = countIndexedMessageStatements.get(db);
|
|
316
|
+
if (!stmt) {
|
|
317
|
+
stmt = db.prepare(`SELECT COUNT(*) AS count
|
|
318
|
+
FROM message_history_source AS source
|
|
319
|
+
JOIN message_fts_rowid_map AS map
|
|
320
|
+
ON map.session_id = source.session_id
|
|
321
|
+
AND map.message_ordinal = source.message_ordinal
|
|
322
|
+
WHERE source.session_id = ? AND source.message_id = ?`);
|
|
323
|
+
countIndexedMessageStatements.set(db, stmt);
|
|
324
|
+
}
|
|
325
|
+
return stmt;
|
|
326
|
+
}
|
|
327
|
+
function getMessageSourceStatement(db) {
|
|
328
|
+
let stmt = getMessageSourceStatements.get(db);
|
|
329
|
+
if (!stmt) {
|
|
330
|
+
stmt = db.prepare("SELECT message_ordinal, source_version, normalized_content_hash, role FROM message_history_source WHERE session_id = ? AND message_id = ?");
|
|
331
|
+
getMessageSourceStatements.set(db, stmt);
|
|
332
|
+
}
|
|
333
|
+
return stmt;
|
|
334
|
+
}
|
|
335
|
+
function getUpsertMessageSourceStatement(db) {
|
|
336
|
+
let stmt = upsertMessageSourceStatements.get(db);
|
|
337
|
+
if (!stmt) {
|
|
338
|
+
stmt = db.prepare(`INSERT INTO message_history_source (
|
|
339
|
+
session_id, message_id, message_ordinal, source_version,
|
|
340
|
+
normalized_content_hash, role, harness, updated_at
|
|
341
|
+
) VALUES (?, ?, ?, ?, ?, ?, ?, ?)
|
|
342
|
+
ON CONFLICT(session_id, message_id) DO UPDATE SET
|
|
343
|
+
message_ordinal = excluded.message_ordinal,
|
|
344
|
+
source_version = excluded.source_version,
|
|
345
|
+
normalized_content_hash = excluded.normalized_content_hash,
|
|
346
|
+
role = excluded.role,
|
|
347
|
+
harness = excluded.harness,
|
|
348
|
+
updated_at = excluded.updated_at`);
|
|
349
|
+
upsertMessageSourceStatements.set(db, stmt);
|
|
350
|
+
}
|
|
351
|
+
return stmt;
|
|
352
|
+
}
|
|
353
|
+
function getDeleteMessageSourceRangeStatement(db) {
|
|
354
|
+
let stmt = deleteMessageSourceRangeStatements.get(db);
|
|
355
|
+
if (!stmt) {
|
|
356
|
+
stmt = db.prepare("DELETE FROM message_history_source WHERE session_id = ? AND message_ordinal BETWEEN ? AND ?");
|
|
357
|
+
deleteMessageSourceRangeStatements.set(db, stmt);
|
|
358
|
+
}
|
|
359
|
+
return stmt;
|
|
360
|
+
}
|
|
361
|
+
function getDeleteMessageFtsStatement(db) {
|
|
362
|
+
let stmt = deleteMessageFtsStatements.get(db);
|
|
363
|
+
if (!stmt) {
|
|
364
|
+
stmt = db.prepare(`DELETE FROM message_history_fts
|
|
365
|
+
WHERE rowid IN (
|
|
366
|
+
SELECT map.fts_rowid
|
|
367
|
+
FROM message_history_source AS source
|
|
368
|
+
JOIN message_fts_rowid_map AS map
|
|
369
|
+
ON map.session_id = source.session_id
|
|
370
|
+
AND map.message_ordinal = source.message_ordinal
|
|
371
|
+
WHERE source.session_id = ? AND source.message_id = ?
|
|
372
|
+
)`);
|
|
373
|
+
deleteMessageFtsStatements.set(db, stmt);
|
|
374
|
+
}
|
|
375
|
+
return stmt;
|
|
376
|
+
}
|
|
377
|
+
function getDeleteFtsMapRangeStatement(db) {
|
|
378
|
+
let stmt = deleteFtsMapRangeStatements.get(db);
|
|
379
|
+
if (!stmt) {
|
|
380
|
+
stmt = db.prepare("DELETE FROM message_fts_rowid_map WHERE session_id = ? AND message_ordinal BETWEEN ? AND ?");
|
|
381
|
+
deleteFtsMapRangeStatements.set(db, stmt);
|
|
382
|
+
}
|
|
383
|
+
return stmt;
|
|
384
|
+
}
|
|
385
|
+
function getDeleteMessageFtsMapStatement(db) {
|
|
386
|
+
let stmt = deleteMessageFtsMapStatements.get(db);
|
|
387
|
+
if (!stmt) {
|
|
388
|
+
stmt = db.prepare(`DELETE FROM message_fts_rowid_map
|
|
389
|
+
WHERE session_id = ?
|
|
390
|
+
AND message_ordinal = (
|
|
391
|
+
SELECT message_ordinal
|
|
392
|
+
FROM message_history_source
|
|
393
|
+
WHERE session_id = ? AND message_id = ?
|
|
394
|
+
)`);
|
|
395
|
+
deleteMessageFtsMapStatements.set(db, stmt);
|
|
396
|
+
}
|
|
397
|
+
return stmt;
|
|
398
|
+
}
|
|
399
|
+
function insertMessageFtsRow(db, sessionId, messageOrdinal, messageId, role, content, messageTimeMs) {
|
|
400
|
+
const result = getInsertMessageStatement(db).run(sessionId, messageOrdinal, messageId, role, content);
|
|
401
|
+
recordMessageFtsRowid(db, sessionId, messageOrdinal, result.lastInsertRowid, messageTimeMs ?? null);
|
|
402
|
+
}
|
|
403
|
+
function normalizeSourceVersion(version) {
|
|
404
|
+
if (typeof version === "number")
|
|
405
|
+
return `number:${version}`;
|
|
406
|
+
if (typeof version === "string")
|
|
407
|
+
return `string:${version}`;
|
|
408
|
+
return "null";
|
|
409
|
+
}
|
|
410
|
+
function toMessageIndexSource(message) {
|
|
411
|
+
const content = getIndexableContent(message.role, message.parts);
|
|
412
|
+
return {
|
|
413
|
+
id: message.id,
|
|
414
|
+
ordinal: message.ordinal,
|
|
415
|
+
role: message.role,
|
|
416
|
+
createdAt: message.createdAt,
|
|
417
|
+
sourceVersion: normalizeSourceVersion(message.version),
|
|
418
|
+
contentHash: createHash("sha256").update(content).digest("hex"),
|
|
419
|
+
content
|
|
420
|
+
};
|
|
421
|
+
}
|
|
422
|
+
function getMessageSourceSnapshot(message) {
|
|
423
|
+
return toMessageIndexSource(message);
|
|
424
|
+
}
|
|
425
|
+
function getMessageIndexSourceIdentity(message) {
|
|
426
|
+
const source = getMessageSourceSnapshot(message);
|
|
427
|
+
return JSON.stringify([
|
|
428
|
+
source.ordinal,
|
|
429
|
+
source.sourceVersion,
|
|
430
|
+
source.contentHash,
|
|
431
|
+
source.role,
|
|
432
|
+
message.createdAt ?? null
|
|
433
|
+
]);
|
|
434
|
+
}
|
|
435
|
+
function isMessageIndexSourceCurrent(db, sessionId, message) {
|
|
436
|
+
const source = getMessageSourceSnapshot(message);
|
|
437
|
+
return isIndexSourceCurrent(db, sessionId, source);
|
|
438
|
+
}
|
|
439
|
+
function isIndexSourceCurrent(db, sessionId, source) {
|
|
440
|
+
const row = getMessageSourceStatement(db).get(sessionId, source.id);
|
|
441
|
+
return row?.message_ordinal === source.ordinal && row.source_version === source.sourceVersion && row.normalized_content_hash === source.contentHash && row.role === source.role;
|
|
442
|
+
}
|
|
443
|
+
function writeMessageSource(db, sessionId, source, now) {
|
|
444
|
+
getUpsertMessageSourceStatement(db).run(sessionId, source.id, source.ordinal, source.sourceVersion, source.contentHash, source.role, getHarness2(), now);
|
|
445
|
+
const bytes = Buffer.byteLength(source.content);
|
|
446
|
+
if (source.ftsSkippedBytes !== undefined || bytes > MESSAGE_INDEX_MAX_FTS_BYTES) {
|
|
447
|
+
skipStatement(db, "INSERT INTO schema_migrations_meta(key,value) VALUES (?,?) ON CONFLICT(key) DO UPDATE SET value=excluded.value").run(messageIndexSkipKey(sessionId, source.id), JSON.stringify({
|
|
448
|
+
reason: "oversized",
|
|
449
|
+
bytes: source.ftsSkippedBytes ?? bytes,
|
|
450
|
+
contentHash: source.contentHash,
|
|
451
|
+
ordinal: source.ordinal,
|
|
452
|
+
complete: true
|
|
453
|
+
}));
|
|
454
|
+
return "";
|
|
455
|
+
}
|
|
456
|
+
skipStatement(db, "DELETE FROM schema_migrations_meta WHERE key=?").run(messageIndexSkipKey(sessionId, source.id));
|
|
457
|
+
return source.content;
|
|
458
|
+
}
|
|
459
|
+
function getLastIndexedOrdinal(db, sessionId) {
|
|
460
|
+
const row = getLastIndexedStatement(db).get(sessionId);
|
|
461
|
+
return typeof row?.last_indexed_ordinal === "number" ? row.last_indexed_ordinal : 0;
|
|
462
|
+
}
|
|
463
|
+
function getIndexedMessageCorpusSize(db, sessionId, maxOrdinal) {
|
|
464
|
+
const watermark = getLastIndexedOrdinal(db, sessionId);
|
|
465
|
+
return maxOrdinal === null ? watermark : Math.min(watermark, Math.max(0, maxOrdinal));
|
|
466
|
+
}
|
|
467
|
+
function getDirtyIndexFloor(db, sessionId) {
|
|
468
|
+
const row = getLastIndexedStatement(db).get(sessionId);
|
|
469
|
+
return typeof row?.dirty_floor_ordinal === "number" && row.dirty_floor_ordinal > 0 ? row.dirty_floor_ordinal : null;
|
|
470
|
+
}
|
|
471
|
+
function markMessageIndexDirty(db, sessionId, floorOrdinal) {
|
|
472
|
+
const dirtyFloor = Math.max(1, Math.floor(floorOrdinal));
|
|
473
|
+
getUpsertDirtyFloorStatement(db).run(sessionId, getLastIndexedOrdinal(db, sessionId), dirtyFloor, Date.now(), getHarness2());
|
|
474
|
+
}
|
|
475
|
+
function isMessageAlreadyIndexed(db, sessionId, messageId) {
|
|
476
|
+
const row = getCountIndexedMessageStatement(db).get(sessionId, messageId);
|
|
477
|
+
return (typeof row?.count === "number" ? row.count : 0) > 0;
|
|
478
|
+
}
|
|
479
|
+
function setIndexProgress(db, sessionId, watermark, dirtyFloor, now) {
|
|
480
|
+
getUpsertProgressStatement(db).run(sessionId, Math.max(0, Math.floor(watermark)), dirtyFloor ?? 0, now, getHarness2());
|
|
481
|
+
}
|
|
482
|
+
function getMessageIndexReconciliationStartOrdinal(db, sessionId) {
|
|
483
|
+
const watermark = getLastIndexedOrdinal(db, sessionId);
|
|
484
|
+
const dirtyFloor = getDirtyIndexFloor(db, sessionId);
|
|
485
|
+
return dirtyFloor === null ? watermark : Math.min(watermark, dirtyFloor - 1);
|
|
486
|
+
}
|
|
487
|
+
function isMessageIndexReconciledThrough(db, sessionId, finalWatermark) {
|
|
488
|
+
const dirtyFloor = getDirtyIndexFloor(db, sessionId);
|
|
489
|
+
return getLastIndexedOrdinal(db, sessionId) >= finalWatermark && dirtyFloor === null;
|
|
490
|
+
}
|
|
491
|
+
function getIndexableContent(role, parts) {
|
|
492
|
+
if (role === "user") {
|
|
493
|
+
if (!hasMeaningfulUserText2(parts)) {
|
|
494
|
+
return "";
|
|
495
|
+
}
|
|
496
|
+
return extractTexts2(parts).map(cleanUserText2).map(normalizeIndexText).filter((text) => text.length > 0).join(" / ");
|
|
497
|
+
}
|
|
498
|
+
if (role === "assistant") {
|
|
499
|
+
return extractTexts2(parts).map(removeSystemReminders).map(normalizeIndexText).filter((text) => text.length > 0).join(" / ");
|
|
500
|
+
}
|
|
501
|
+
return "";
|
|
502
|
+
}
|
|
503
|
+
function indexSingleMessageInTransaction(db, sessionId, message, now, dirtyFloorBeforeAttempt, source) {
|
|
504
|
+
const currentWatermark = getLastIndexedOrdinal(db, sessionId);
|
|
505
|
+
const dirtyFloor = getDirtyIndexFloor(db, sessionId);
|
|
506
|
+
if (message.ordinal <= currentWatermark) {
|
|
507
|
+
if (isIndexSourceCurrent(db, sessionId, source)) {
|
|
508
|
+
recordIndexedMessageTime(db, sessionId, message.ordinal, message.createdAt);
|
|
509
|
+
return false;
|
|
510
|
+
}
|
|
511
|
+
if (!messageFtsOrdinalRangeIsMapped(db, sessionId, message.ordinal, message.ordinal)) {
|
|
512
|
+
return false;
|
|
513
|
+
}
|
|
514
|
+
getDeleteMessageFtsStatement(db).run(sessionId, message.id);
|
|
515
|
+
getDeleteMessageFtsMapStatement(db).run(sessionId, sessionId, message.id);
|
|
516
|
+
const content = writeMessageSource(db, sessionId, source, now);
|
|
517
|
+
if (content.length > 0 && (message.role === "user" || message.role === "assistant")) {
|
|
518
|
+
insertMessageFtsRow(db, sessionId, message.ordinal, message.id, message.role, content, message.createdAt);
|
|
519
|
+
}
|
|
520
|
+
setIndexProgress(db, sessionId, currentWatermark, dirtyFloorBeforeAttempt === message.ordinal ? null : dirtyFloorBeforeAttempt, now);
|
|
521
|
+
return true;
|
|
522
|
+
}
|
|
523
|
+
if (message.ordinal !== currentWatermark + 1 || dirtyFloor !== null && dirtyFloor !== message.ordinal) {
|
|
524
|
+
return false;
|
|
525
|
+
}
|
|
526
|
+
const content = writeMessageSource(db, sessionId, source, now);
|
|
527
|
+
let inserted = false;
|
|
528
|
+
if (content.length > 0 && (message.role === "user" || message.role === "assistant") && !isMessageAlreadyIndexed(db, sessionId, message.id)) {
|
|
529
|
+
insertMessageFtsRow(db, sessionId, message.ordinal, message.id, message.role, content, message.createdAt);
|
|
530
|
+
inserted = true;
|
|
531
|
+
}
|
|
532
|
+
setIndexProgress(db, sessionId, message.ordinal, dirtyFloorBeforeAttempt === message.ordinal ? null : dirtyFloorBeforeAttempt, now);
|
|
533
|
+
return inserted;
|
|
534
|
+
}
|
|
535
|
+
function indexSingleMessage(db, sessionId, message, shouldCommit = () => true, preparedSource) {
|
|
536
|
+
if (!shouldCommit())
|
|
537
|
+
return false;
|
|
538
|
+
const source = preparedSource ?? getMessageSourceSnapshot(message);
|
|
539
|
+
const currentWatermark = getLastIndexedOrdinal(db, sessionId);
|
|
540
|
+
if (message.ordinal <= currentWatermark && isIndexSourceCurrent(db, sessionId, source)) {
|
|
541
|
+
recordIndexedMessageTime(db, sessionId, message.ordinal, message.createdAt);
|
|
542
|
+
return false;
|
|
543
|
+
}
|
|
544
|
+
const dirtyFloorBeforeAttempt = getDirtyIndexFloor(db, sessionId);
|
|
545
|
+
markMessageIndexDirty(db, sessionId, Math.min(message.ordinal, currentWatermark + 1));
|
|
546
|
+
db.exec("BEGIN IMMEDIATE");
|
|
547
|
+
const transactionStartedAt = performance.now();
|
|
548
|
+
let committed = false;
|
|
549
|
+
try {
|
|
550
|
+
const result = indexSingleMessageInTransaction(db, sessionId, message, Date.now(), dirtyFloorBeforeAttempt, source);
|
|
551
|
+
if (!shouldCommit())
|
|
552
|
+
return false;
|
|
553
|
+
db.exec("COMMIT");
|
|
554
|
+
committed = true;
|
|
555
|
+
logSlowWriteTransaction("message_index_incremental", transactionStartedAt);
|
|
556
|
+
return result;
|
|
557
|
+
} finally {
|
|
558
|
+
if (!committed) {
|
|
559
|
+
try {
|
|
560
|
+
db.exec("ROLLBACK");
|
|
561
|
+
} catch {}
|
|
562
|
+
}
|
|
563
|
+
}
|
|
564
|
+
}
|
|
565
|
+
function indexMessagesAfterOrdinal(db, sessionId, messages, _lastIndexedOrdinal, finalWatermark = messages.length) {
|
|
566
|
+
return indexItemsAfterOrdinal(db, sessionId, messages, finalWatermark, toMessageIndexSource);
|
|
567
|
+
}
|
|
568
|
+
function indexItemsAfterOrdinal(db, sessionId, items, finalWatermark, toSource) {
|
|
569
|
+
const now = Date.now();
|
|
570
|
+
let inserted = 0;
|
|
571
|
+
const sources = new Map(items.map((item) => [item.ordinal, toSource(item)]));
|
|
572
|
+
db.exec("BEGIN IMMEDIATE");
|
|
573
|
+
const transactionStartedAt = performance.now();
|
|
574
|
+
let committed = false;
|
|
575
|
+
try {
|
|
576
|
+
const currentWatermark = getLastIndexedOrdinal(db, sessionId);
|
|
577
|
+
const dirtyFloor = getDirtyIndexFloor(db, sessionId);
|
|
578
|
+
const effectiveWatermark = dirtyFloor === null ? currentWatermark : Math.min(currentWatermark, Math.max(0, dirtyFloor - 1));
|
|
579
|
+
if (dirtyFloor !== null && dirtyFloor <= currentWatermark && !messageFtsOrdinalRangeIsMapped(db, sessionId, dirtyFloor, Math.min(currentWatermark, finalWatermark))) {
|
|
580
|
+
db.exec("COMMIT");
|
|
581
|
+
committed = true;
|
|
582
|
+
logSlowWriteTransaction("message_index_reconcile", transactionStartedAt);
|
|
583
|
+
return 0;
|
|
584
|
+
}
|
|
585
|
+
if (dirtyFloor !== null && dirtyFloor <= finalWatermark) {
|
|
586
|
+
getDeleteFtsRangeStatement(db).run(sessionId, dirtyFloor, finalWatermark);
|
|
587
|
+
getDeleteFtsMapRangeStatement(db).run(sessionId, dirtyFloor, finalWatermark);
|
|
588
|
+
getDeleteMessageSourceRangeStatement(db).run(sessionId, dirtyFloor, finalWatermark);
|
|
589
|
+
}
|
|
590
|
+
const itemsByOrdinal = new Map;
|
|
591
|
+
for (const item of items) {
|
|
592
|
+
if (item.ordinal > effectiveWatermark && item.ordinal <= finalWatermark) {
|
|
593
|
+
itemsByOrdinal.set(item.ordinal, item);
|
|
594
|
+
}
|
|
595
|
+
}
|
|
596
|
+
let coveredWatermark = effectiveWatermark;
|
|
597
|
+
while (coveredWatermark < finalWatermark && itemsByOrdinal.has(coveredWatermark + 1)) {
|
|
598
|
+
coveredWatermark += 1;
|
|
599
|
+
}
|
|
600
|
+
for (let ordinal = effectiveWatermark + 1;ordinal <= coveredWatermark; ordinal++) {
|
|
601
|
+
const item = itemsByOrdinal.get(ordinal);
|
|
602
|
+
if (!item)
|
|
603
|
+
continue;
|
|
604
|
+
const message = sources.get(ordinal);
|
|
605
|
+
if (!message)
|
|
606
|
+
continue;
|
|
607
|
+
const content = writeMessageSource(db, sessionId, message, now);
|
|
608
|
+
if (content.length === 0 || message.role !== "user" && message.role !== "assistant" || isMessageAlreadyIndexed(db, sessionId, message.id)) {
|
|
609
|
+
continue;
|
|
610
|
+
}
|
|
611
|
+
insertMessageFtsRow(db, sessionId, message.ordinal, message.id, message.role, content, message.createdAt);
|
|
612
|
+
inserted += 1;
|
|
613
|
+
}
|
|
614
|
+
const missingFloor = coveredWatermark < finalWatermark ? coveredWatermark + 1 : null;
|
|
615
|
+
const preservedFloor = dirtyFloor !== null && dirtyFloor > finalWatermark ? dirtyFloor : null;
|
|
616
|
+
const nextDirtyFloor = missingFloor === null ? preservedFloor : preservedFloor === null ? missingFloor : Math.min(missingFloor, preservedFloor);
|
|
617
|
+
setIndexProgress(db, sessionId, coveredWatermark, nextDirtyFloor, now);
|
|
618
|
+
db.exec("COMMIT");
|
|
619
|
+
committed = true;
|
|
620
|
+
logSlowWriteTransaction("message_index_reconcile", transactionStartedAt);
|
|
621
|
+
} finally {
|
|
622
|
+
if (!committed) {
|
|
623
|
+
try {
|
|
624
|
+
db.exec("ROLLBACK");
|
|
625
|
+
} catch {}
|
|
626
|
+
}
|
|
627
|
+
}
|
|
628
|
+
return inserted;
|
|
629
|
+
}
|
|
630
|
+
function harnessSupportsOpenCodeOrphanSweep(harness) {
|
|
631
|
+
return harness === "opencode";
|
|
632
|
+
}
|
|
633
|
+
function getMessageHistoryOrphanSweepState(db, harness) {
|
|
634
|
+
return db.prepare("SELECT cursor_session_id, last_swept_at FROM message_history_orphan_sweep WHERE harness = ?").get(harness) ?? {};
|
|
635
|
+
}
|
|
636
|
+
function persistMessageHistoryOrphanSweepState(db, cursor, lastSweptAt, harness) {
|
|
637
|
+
db.prepare(`INSERT INTO message_history_orphan_sweep (harness, cursor_session_id, last_swept_at)
|
|
638
|
+
VALUES (?, ?, ?)
|
|
639
|
+
ON CONFLICT(harness) DO UPDATE SET
|
|
640
|
+
cursor_session_id = excluded.cursor_session_id,
|
|
641
|
+
last_swept_at = excluded.last_swept_at`).run(harness, cursor, lastSweptAt);
|
|
642
|
+
}
|
|
643
|
+
function getOpenCodeSessionScopedCandidateSourceSql(harness) {
|
|
644
|
+
return SESSION_SCOPED_TABLES.filter((definition) => definition.harnessScoped === true).map((definition) => {
|
|
645
|
+
const predicates = ["session_id IS NOT NULL", `harness = '${harness}'`];
|
|
646
|
+
if (definition.extraPredicate)
|
|
647
|
+
predicates.push(definition.extraPredicate);
|
|
648
|
+
return `SELECT session_id FROM ${definition.table} WHERE ${predicates.join(" AND ")}`;
|
|
649
|
+
}).join(`
|
|
650
|
+
UNION
|
|
651
|
+
`);
|
|
652
|
+
}
|
|
653
|
+
function sweepOrphanedOpenCodeMessageIndexesInBackground(db, openReadableOpenCodeDb, options = {}) {
|
|
654
|
+
const now = options.now ?? Date.now();
|
|
655
|
+
const batchSize = Math.max(1, Math.floor(options.batchSize ?? MESSAGE_HISTORY_ORPHAN_SWEEP_BATCH_SIZE));
|
|
656
|
+
const safetyAgeMs = Math.max(0, options.safetyAgeMs ?? MESSAGE_HISTORY_ORPHAN_SAFETY_AGE_MS);
|
|
657
|
+
const cooldownMs = Math.max(0, options.cooldownMs ?? MESSAGE_HISTORY_ORPHAN_SWEEP_COOLDOWN_MS);
|
|
658
|
+
const unavailableReprobeMs = Math.max(cooldownMs, options.unavailableReprobeMs ?? MESSAGE_HISTORY_ORPHAN_UNAVAILABLE_REPROBE_MS);
|
|
659
|
+
const harness = getHarness2();
|
|
660
|
+
const state = getMessageHistoryOrphanSweepState(db, harness);
|
|
661
|
+
const cursor = typeof state.cursor_session_id === "string" ? state.cursor_session_id : "";
|
|
662
|
+
if (typeof state.last_swept_at === "number" && state.last_swept_at + cooldownMs > now) {
|
|
663
|
+
return { status: "cooldown", scanned: 0, deleted: 0, cursor };
|
|
664
|
+
}
|
|
665
|
+
if (!harnessSupportsOpenCodeOrphanSweep(harness)) {
|
|
666
|
+
persistMessageHistoryOrphanSweepState(db, cursor, now + unavailableReprobeMs - cooldownMs, harness);
|
|
667
|
+
if (harness === "opencode2") {
|
|
668
|
+
return {
|
|
669
|
+
status: "unavailable",
|
|
670
|
+
scanned: 0,
|
|
671
|
+
deleted: 0,
|
|
672
|
+
cursor,
|
|
673
|
+
reason: "OpenCode 2 keeps its sessions in session_v2, which this sweep does not read"
|
|
674
|
+
};
|
|
675
|
+
}
|
|
676
|
+
return { status: "unavailable", scanned: 0, deleted: 0, cursor };
|
|
677
|
+
}
|
|
678
|
+
let openCodeDb = null;
|
|
679
|
+
try {
|
|
680
|
+
openCodeDb = openReadableOpenCodeDb();
|
|
681
|
+
} catch {
|
|
682
|
+
openCodeDb = null;
|
|
683
|
+
}
|
|
684
|
+
if (!openCodeDb) {
|
|
685
|
+
persistMessageHistoryOrphanSweepState(db, cursor, now + unavailableReprobeMs - cooldownMs, harness);
|
|
686
|
+
return { status: "source_unavailable", scanned: 0, deleted: 0, cursor };
|
|
687
|
+
}
|
|
688
|
+
try {
|
|
689
|
+
const cutoff = now - safetyAgeMs;
|
|
690
|
+
const candidateSourceSql = getOpenCodeSessionScopedCandidateSourceSql(harness);
|
|
691
|
+
const candidates = db.prepare(`SELECT session_id
|
|
692
|
+
FROM (${candidateSourceSql}) AS session_candidates
|
|
693
|
+
WHERE session_id > ?
|
|
694
|
+
AND NOT EXISTS (
|
|
695
|
+
SELECT 1
|
|
696
|
+
FROM message_history_index
|
|
697
|
+
WHERE message_history_index.session_id = session_candidates.session_id
|
|
698
|
+
AND message_history_index.harness = '${harness}'
|
|
699
|
+
AND message_history_index.updated_at > ?
|
|
700
|
+
)
|
|
701
|
+
ORDER BY session_id ASC
|
|
702
|
+
LIMIT ?`).all(cursor, cutoff, batchSize);
|
|
703
|
+
const sessionExists = openCodeDb.prepare("SELECT 1 FROM session WHERE id = ? LIMIT 1");
|
|
704
|
+
const missingSessionIds = candidates.filter((candidate) => !sessionExists.get(candidate.session_id)).map((candidate) => candidate.session_id);
|
|
705
|
+
const nextCursor = candidates.length < batchSize ? "" : candidates[candidates.length - 1]?.session_id ?? cursor;
|
|
706
|
+
const completedAt = candidates.length < batchSize ? now : null;
|
|
707
|
+
db.exec("BEGIN IMMEDIATE");
|
|
708
|
+
const transactionStartedAt = performance.now();
|
|
709
|
+
let committed = false;
|
|
710
|
+
let deleted = 0;
|
|
711
|
+
try {
|
|
712
|
+
const stillEligible = db.prepare(`SELECT 1
|
|
713
|
+
FROM (${candidateSourceSql}) AS session_candidates
|
|
714
|
+
WHERE session_id = ?
|
|
715
|
+
AND NOT EXISTS (
|
|
716
|
+
SELECT 1
|
|
717
|
+
FROM message_history_index
|
|
718
|
+
WHERE message_history_index.session_id = session_candidates.session_id
|
|
719
|
+
AND message_history_index.harness = '${harness}'
|
|
720
|
+
AND message_history_index.updated_at > ?
|
|
721
|
+
)
|
|
722
|
+
LIMIT 1`);
|
|
723
|
+
const eligibleSessionIds = missingSessionIds.filter((sessionId) => stillEligible.get(sessionId, cutoff));
|
|
724
|
+
deleted = deleteSessionScopedRows(db, eligibleSessionIds, harness);
|
|
725
|
+
persistMessageHistoryOrphanSweepState(db, nextCursor, completedAt, harness);
|
|
726
|
+
db.exec("COMMIT");
|
|
727
|
+
committed = true;
|
|
728
|
+
logSlowWriteTransaction("message_index_orphan_sweep", transactionStartedAt);
|
|
729
|
+
} finally {
|
|
730
|
+
if (!committed) {
|
|
731
|
+
try {
|
|
732
|
+
db.exec("ROLLBACK");
|
|
733
|
+
} catch {}
|
|
734
|
+
}
|
|
735
|
+
}
|
|
736
|
+
return {
|
|
737
|
+
status: "swept",
|
|
738
|
+
scanned: candidates.length,
|
|
739
|
+
deleted,
|
|
740
|
+
cursor: nextCursor
|
|
741
|
+
};
|
|
742
|
+
} finally {
|
|
743
|
+
closeQuietly(openCodeDb);
|
|
744
|
+
}
|
|
745
|
+
}
|
|
746
|
+
function sweepOrphanedOpenCodeMessageIndexes(db, openReadableOpenCodeDb, options = {}) {
|
|
747
|
+
return withoutSqliteTransformPass2(() => withSqliteBackgroundWriter2(() => sweepOrphanedOpenCodeMessageIndexesInBackground(db, openReadableOpenCodeDb, options)));
|
|
748
|
+
}
|
|
749
|
+
|
|
750
|
+
export { readSessionActivity, observeSessionActivity, backfillSessionActivity, deleteSessionScopedRows, clearCompressionDepth, clearCompressionDepthRange, MESSAGE_INDEX_MAX_FTS_BYTES, messageIndexSkipKey, getMessageIndexSourceIdentity, isMessageIndexSourceCurrent, getLastIndexedOrdinal, getIndexedMessageCorpusSize, markMessageIndexDirty, getMessageIndexReconciliationStartOrdinal, isMessageIndexReconciledThrough, getIndexableContent, indexSingleMessage, indexMessagesAfterOrdinal, sweepOrphanedOpenCodeMessageIndexes };
|