refactor(core): finish TypeScript workspace migration

2026-07-12 00:37:32 +08:00
parent 83d5703f7b
commit 535d8edb4d
31 changed files with 326 additions and 199 deletions
@@ -0,0 +1,80 @@
+// node:sqlite lifecycle and migrations for the Core package.
+import { createRequire } from 'node:module';
+import { CLAUDE_DIR, CODEX_DIR, TEXT_LIMIT, trunc, truncJson, extractText, extractContentType, extractMessageIsMeta, filePath, isDir, readLines } from './parsing.ts';
+import { configureConnection } from './tx.ts';
+const require = createRequire(import.meta.url);
+const fs = require('node:fs');
+const path = require('node:path');
+const os = require('node:os');
+const { DatabaseSync } = require('node:sqlite');
+
+type SqliteDb = any;
+
+const OBELISK_DIR = path.join(os.homedir(), '.obelisk');
+const LEGACY_DB_PATH = path.join(CLAUDE_DIR, 'obelisk.sqlite');
+const DB_PATH = path.join(OBELISK_DIR, 'obelisk.sqlite');
+const SCHEMA = fs.readFileSync(new URL('./schema.sql', import.meta.url), 'utf8');
+
+function migrateLegacyDbIfNeeded() {
+  if (fs.existsSync(DB_PATH)) return;
+  if (!fs.existsSync(LEGACY_DB_PATH)) return;
+  fs.mkdirSync(path.dirname(DB_PATH), { recursive: true });
+  fs.copyFileSync(LEGACY_DB_PATH, DB_PATH);
+}
+
+function openDb() {
+  migrateLegacyDbIfNeeded();
+  fs.mkdirSync(path.dirname(DB_PATH), { recursive: true });
+  const db = new DatabaseSync(DB_PATH);
+  configureConnection(db, { busyTimeoutMs: 250 });
+  migrateExistingColumns(db);
+  db.exec(SCHEMA);
+  migrateDb(db);
+  return db;
+}
+
+// Queries and daemon-arbitration checks must never migrate/configure the index.
+// The caller is responsible for ensuring the database exists first.
+function openReadDb() {
+  const db = new DatabaseSync(DB_PATH, { readOnly: true });
+  db.exec('PRAGMA busy_timeout=250');
+  return db;
+}
+
+function openWriterLeaseDb(lockPath: string): SqliteDb {
+  return new DatabaseSync(lockPath);
+}
+
+function ensureColumn(db: SqliteDb, table: string, column: string, definition: string): void {
+  const columns = db.prepare(`PRAGMA table_info(${table})`).all().map((c: { name: string }) => c.name);
+  if (!columns.includes(column)) db.exec(`ALTER TABLE ${table} ADD COLUMN ${column} ${definition}`);
+}
+
+function tableExists(db: SqliteDb, table: string): boolean {
+  return Boolean(db.prepare("SELECT name FROM sqlite_master WHERE type='table' AND name=?").get(table));
+}
+
+function migrateExistingColumns(db: SqliteDb): void {
+  if (tableExists(db, 'sessions')) ensureColumn(db, 'sessions', 'source', "TEXT DEFAULT 'claude'");
+  if (tableExists(db, 'messages')) {
+    ensureColumn(db, 'messages', 'content_type', 'TEXT');
+    ensureColumn(db, 'messages', 'is_meta', 'INTEGER DEFAULT 0');
+    ensureColumn(db, 'messages', 'source', "TEXT DEFAULT 'claude'");
+  }
+  if (tableExists(db, 'memories')) {
+    ensureColumn(db, 'memories', 'anchors', 'TEXT');
+    ensureColumn(db, 'memories', 'deleted_at', 'TEXT');
+    ensureColumn(db, 'memories', 'deleted_reason', 'TEXT');
+  }
+}
+
+function migrateDb(db: SqliteDb): void {
+  migrateExistingColumns(db);
+}
+
+function rebuildMemoryFts(db: SqliteDb): void {
+  db.exec("INSERT INTO memories_fts(memories_fts) VALUES('rebuild')");
+}
+
+
+export { CLAUDE_DIR, CODEX_DIR, OBELISK_DIR, DB_PATH, TEXT_LIMIT, openDb, openReadDb, openWriterLeaseDb, rebuildMemoryFts, trunc, truncJson, extractText, extractContentType, extractMessageIsMeta, filePath, isDir, readLines, fs, path, os };