devrage 0.5.6 → 0.5.7

package/dist/cli.js

@@ -825,6 +825,97 @@ import { existsSync as existsSync2 } from "node:fs";
 import { readdir as readdir5 } from "node:fs/promises";
 import { homedir as homedir5 } from "node:os";
 import { join as join5 } from "node:path";
+
+// src/adapters/sqlite.ts
+async function openReadonlySqliteDatabase(dbPath) {
+  const requestedDriver = process.env["DEVRAGE_SQLITE_DRIVER"];
+  if (requestedDriver) {
+    const loader = driverLoader(requestedDriver);
+    if (!loader) {
+      return null;
+    }
+    try {
+      return await loader(dbPath);
+    } catch {
+      return null;
+    }
+  }
+  const loaders = isBunRuntime() ? [openWithBunSqlite, openWithNodeSqlite, openWithBetterSqlite3] : [openWithNodeSqlite, openWithBetterSqlite3];
+  for (const loader of loaders) {
+    try {
+      return await loader(dbPath);
+    } catch {
+      continue;
+    }
+  }
+  return null;
+}
+function driverLoader(driver) {
+  switch (driver) {
+    case "bun":
+    case "bun:sqlite":
+      return openWithBunSqlite;
+    case "node":
+    case "node:sqlite":
+      return openWithNodeSqlite;
+    case "better-sqlite3":
+      return openWithBetterSqlite3;
+    default:
+      return null;
+  }
+}
+function isBunRuntime() {
+  return Boolean(process.versions["bun"]);
+}
+async function openWithBunSqlite(dbPath) {
+  const specifier = "bun:sqlite";
+  const sqlite = await import(specifier);
+  const db = new sqlite.Database(dbPath, { readonly: true, create: false });
+  return {
+    prepare(sql) {
+      const prepare = db.prepare ?? db.query;
+      return prepare.call(db, sql);
+    },
+    close() {
+      db.close();
+    }
+  };
+}
+async function openWithNodeSqlite(dbPath) {
+  const sqlite = await importNodeSqlite();
+  return new sqlite.DatabaseSync(dbPath, { open: true, readOnly: true, timeout: 5e3 });
+}
+async function openWithBetterSqlite3(dbPath) {
+  const BetterSqlite3 = await import("better-sqlite3");
+  const Ctor = BetterSqlite3.default ?? BetterSqlite3;
+  return new Ctor(dbPath, {
+    readonly: true,
+    fileMustExist: true
+  });
+}
+async function importNodeSqlite() {
+  const originalEmitWarning = process.emitWarning;
+  process.emitWarning = ((warning, ...args) => {
+    const message = typeof warning === "string" ? warning : warning.message;
+    const type = typeof args[0] === "string" ? args[0] : void 0;
+    if (message === "SQLite is an experimental feature and might change at any time" && type === "ExperimentalWarning") {
+      return;
+    }
+    originalEmitWarning(warning, ...args);
+  });
+  try {
+    const specifier = "node:sqlite";
+    const sqlite = await import(specifier);
+    if (typeof sqlite.DatabaseSync !== "function") {
+      throw new Error("node:sqlite DatabaseSync is unavailable");
+    }
+    return sqlite;
+  } finally {
+    process.emitWarning = originalEmitWarning;
+  }
+}
+
+// src/adapters/cursor.ts
 var CANDIDATE_KEY_PREFIXES = [
   "bubbleId:",
   "composerData:",
@@ -978,16 +1069,7 @@ async function* parseCursorUsageStore(store, options) {
   }
 }
 async function openCursorDb(dbPath) {
-  try {
-    const BetterSqlite3 = await import("better-sqlite3");
-    const Ctor = BetterSqlite3.default ?? BetterSqlite3;
-    return new Ctor(
-      dbPath,
-      { readonly: true }
-    );
-  } catch {
-    return null;
-  }
+  return openReadonlySqliteDatabase(dbPath);
 }
 function readStateRows(db) {
   const rows = [];
@@ -1048,6 +1130,12 @@ function decodeStateValue(value) {
   if (Buffer.isBuffer(value)) {
     return value.toString("utf-8");
   }
+  if (value instanceof Uint8Array) {
+    return Buffer.from(value).toString("utf-8");
+  }
+  if (ArrayBuffer.isView(value)) {
+    return Buffer.from(value.buffer, value.byteOffset, value.byteLength).toString("utf-8");
+  }
   return null;
 }
 function extractCursorMessages(root, rowKey) {
@@ -1410,17 +1498,11 @@ async function openOpencodeDb() {
   if (!dbPath) {
     return null;
   }
-  try {
-    const BetterSqlite3 = await import("better-sqlite3");
-    const Ctor = BetterSqlite3.default ?? BetterSqlite3;
-    return new Ctor(
-      dbPath,
-      { readonly: true }
-    );
-  } catch {
-    console.warn("devrage: better-sqlite3 not available, skipping OpenCode sessions");
-    return null;
+  const db = await openReadonlySqliteDatabase(dbPath);
+  if (!db) {
+    console.warn("devrage: SQLite support not available, skipping OpenCode sessions");
   }
+  return db;
 }
 function* queryUserMessages(db, options) {
   let query = `
@@ -1763,16 +1845,7 @@ function resolveHomePath(value) {
   return isAbsolute(value) ? value : resolve(value);
 }
 async function openT3Db(dbPath) {
-  try {
-    const BetterSqlite3 = await import("better-sqlite3");
-    const Ctor = BetterSqlite3.default ?? BetterSqlite3;
-    return new Ctor(
-      dbPath,
-      { readonly: true, fileMustExist: true }
-    );
-  } catch {
-    return null;
-  }
+  return openReadonlySqliteDatabase(dbPath);
 }
 function* queryUserMessages2(db, location, options) {
   if (!hasColumns(db, "projection_thread_messages", ["thread_id", "role", "text", "created_at"])) {
@@ -2190,21 +2263,10 @@ async function* parseAgentThreads(dbDir, _options) {
   if (dbFiles.length === 0) {
     return;
   }
-  let Database;
-  try {
-    const mod = await import("better-sqlite3");
-    Database = mod.default ?? mod;
-  } catch {
-    return;
-  }
   for (const dbFile of dbFiles) {
     const dbPath = join9(dbDir, dbFile);
-    let db;
-    try {
-      db = new Database(dbPath, {
-        readonly: true
-      });
-    } catch {
+    const db = await openReadonlySqliteDatabase(dbPath);
+    if (!db) {
       continue;
     }
     try {
@@ -2214,14 +2276,12 @@ async function* parseAgentThreads(dbDir, _options) {
         (t) => t === "messages" || t === "thread_messages" || t.includes("message")
       );
       if (!msgTable) {
-        db.close();
         continue;
       }
       const columns = db.prepare(`PRAGMA table_info("${msgTable}")`).all();
       const colNames = columns.map((c2) => c2.name);
       const hasRole = colNames.includes("role");
       if (!hasRole) {
-        db.close();
         continue;
       }
       const contentCol = colNames.includes("content") ? "content" : colNames.includes("body") ? "body" : "text";
@@ -3667,7 +3727,7 @@ async function main() {
     process.exit(0);
   }
   if (command === "--version") {
-    console.log("0.5.6");
+    console.log("0.5.7");
     process.exit(0);
   }
   const parsed = parseCommand(args);