@cortexkit/opencode-magic-context 0.42.5 → 0.43.0

package/dist/{index-jkrqhc7y.js → index-ykg9sdzz.js}

@@ -9,15 +9,16 @@ import { isAbsolute, join } from "node:path";
 var cachedResolution = null;
 var lastReadFailure = null;
 var claimedDiagnostics = new Set;
-function openCodeDataDir() {
-  return join(process.env.XDG_DATA_HOME ?? join(homedir(), ".local", "share"), "opencode");
+function openCodeDataDir(env = process.env, dataHome) {
+  return join(dataHome ?? env.XDG_DATA_HOME ?? join(homedir(), ".local", "share"), "opencode");
 }
-function environmentKey(dataDir) {
+function environmentKey(dataDir, hostGeneration, channel, env) {
   return [
+    hostGeneration,
     dataDir,
-    process.env.OPENCODE_DB ?? "",
-    process.env.OPENCODE_DISABLE_CHANNEL_DB ?? "",
-    process.env.OPENCODE_CHANNEL ?? ""
+    env.OPENCODE_DB ?? "",
+    env.OPENCODE_DISABLE_CHANNEL_DB ?? "",
+    channel ?? env.OPENCODE_CHANNEL ?? ""
   ].join("\x00");
 }
 function channelPath(dataDir, channel) {
@@ -49,8 +50,8 @@ function discoverOpenCodeDb(dataDir) {
   const channel = name === "opencode.db" ? null : name.slice("opencode-".length, -".db".length) || null;
   return { path: existing.path, source: "discovered", channel };
 }
-function resolveFresh(dataDir) {
-  const explicit = process.env.OPENCODE_DB;
+function resolveV1Fresh(dataDir, env = process.env) {
+  const explicit = env.OPENCODE_DB;
   if (explicit !== undefined && explicit.length > 0) {
     if (explicit === ":memory:") {
       return { path: explicit, source: "OPENCODE_DB", channel: null };
@@ -61,24 +62,47 @@ function resolveFresh(dataDir) {
       channel: null
     };
   }
-  const disableChannelDb = process.env.OPENCODE_DISABLE_CHANNEL_DB;
+  const disableChannelDb = env.OPENCODE_DISABLE_CHANNEL_DB;
   if (disableChannelDb === "1" || disableChannelDb === "true") {
     return { path: join(dataDir, "opencode.db"), source: "default", channel: null };
   }
-  const channel = process.env.OPENCODE_CHANNEL;
+  const channel = env.OPENCODE_CHANNEL;
   if (channel !== undefined && channel.length > 0) {
     return { path: channelPath(dataDir, channel), source: "channel", channel };
   }
   return discoverOpenCodeDb(dataDir);
 }
-function resolveOpenCodeDbPath() {
-  const dataDir = openCodeDataDir();
-  const key = environmentKey(dataDir);
+function sourceOpenCodeDatabaseFilename(hostGeneration, channel, env = process.env) {
+  if (hostGeneration === "v1") {
+    const explicit = env.OPENCODE_DB;
+    if (explicit !== undefined && explicit.length > 0)
+      return explicit;
+    if (env.OPENCODE_DISABLE_CHANNEL_DB === "1" || env.OPENCODE_DISABLE_CHANNEL_DB === "true") {
+      return "opencode.db";
+    }
+    return ["latest", "beta", "prod"].includes(channel) ? "opencode.db" : `opencode-${channel}.db`;
+  }
+  return env.OPENCODE_DB ?? (["latest", "dev", "beta", "next", "prod"].includes(channel) || env.OPENCODE_DISABLE_CHANNEL_DB === "1" || env.OPENCODE_DISABLE_CHANNEL_DB === "true" ? "opencode.db" : `opencode-${channel.replace(/[^a-zA-Z0-9._-]/g, "")}.db`);
+}
+function resolveV2Fresh(dataDir, channel, env) {
+  const filename = sourceOpenCodeDatabaseFilename("v2", channel, env);
+  const explicit = env.OPENCODE_DB !== undefined;
+  return {
+    path: filename === ":memory:" ? filename : join(dataDir, filename),
+    source: explicit ? "OPENCODE_DB" : env.OPENCODE_CHANNEL ? "channel" : "default",
+    channel: explicit ? null : channel
+  };
+}
+function resolveOpenCodeDbPath(hostGeneration = "v1", options = {}) {
+  const env = options.env ?? process.env;
+  const dataDir = openCodeDataDir(env, options.dataHome);
+  const channel = options.channel ?? env.OPENCODE_CHANNEL;
+  const key = environmentKey(dataDir, hostGeneration, channel, env);
   if (cachedResolution?.key === key && (!cachedResolution.existed || existsSync(cachedResolution.resolution.path))) {
     if (cachedResolution.existed)
       return cachedResolution.resolution;
   }
-  const resolution = resolveFresh(dataDir);
+  const resolution = hostGeneration === "v2" ? resolveV2Fresh(dataDir, channel ?? "latest", env) : resolveV1Fresh(dataDir, env);
   cachedResolution = {
     key,
     resolution,
@@ -86,6 +110,41 @@ function resolveOpenCodeDbPath() {
   };
   return resolution;
 }
+function schemaTableNames(db, schema = "main") {
+  const rows = db.prepare(`SELECT name FROM ${schema}.sqlite_master WHERE type = 'table' AND name IN ('message', 'part', 'session', 'project', 'session_message', 'session_v2')`).all();
+  return new Set(rows.flatMap((row) => typeof row.name === "string" ? [row.name] : []));
+}
+function hasV1MessageTables(db, schema = "main") {
+  const tables = schemaTableNames(db, schema);
+  return tables.has("message") && tables.has("part");
+}
+function detectOpenCodeStoreGeneration(db, schema = "main") {
+  const tables = schemaTableNames(db, schema);
+  const hasV1Messages = tables.has("message") && tables.has("part");
+  if (hasV1Messages)
+    return "v1";
+  if (tables.has("session_message"))
+    return "v2";
+  if (tables.has("session") || tables.has("project"))
+    return "v1";
+  return "unknown";
+}
+function hasMigratedV2Schema(tables) {
+  return tables.has("session_message") && tables.has("session_v2");
+}
+function isOpenCodeV2Store(db, schema = "main") {
+  return detectOpenCodeStoreGeneration(db, schema) === "v2" || hasMigratedV2Schema(schemaTableNames(db, schema));
+}
+function assertOpenCodeStoreGeneration(db, expected, path, schema = "main") {
+  const actual = detectOpenCodeStoreGeneration(db, schema);
+  if (actual === expected)
+    return;
+  if (expected === "v2" && isOpenCodeV2Store(db, schema))
+    return;
+  if (actual === "unknown")
+    return;
+  throw new Error(`OpenCode store generation mismatch at ${path}: expected ${expected}, found ${actual}; refusing generation-specific database access`);
+}
 function openCodeDbPathExists(resolution = resolveOpenCodeDbPath()) {
   return resolution.path !== ":memory:" && existsSync(resolution.path);
 }
@@ -450,6 +509,12 @@ function getReadOnlySessionDb() {
   }
   closeCachedReadOnlyDb();
   const db = new Database(dbPath, { readonly: true });
+  try {
+    assertOpenCodeStoreGeneration(db, "v1", dbPath);
+  } catch (error) {
+    closeQuietly(db);
+    throw error;
+  }
   cachedReadOnlyDb = { path: dbPath, db };
   clearOpenCodeDbReadFailure();
   return db;
@@ -687,6 +752,32 @@ function assistantAwaitingToolsFromOpenCodeDb(db, sessionId) {
     }
   });
 }
+function latestPersistedMessageForRecovery(sessionId) {
+  return withReadOnlySessionDb((db) => {
+    const row = db.prepare(`SELECT id, data
+                   FROM message
+                  WHERE session_id = ?
+                  ORDER BY time_created DESC, id DESC
+                  LIMIT 1`).get(sessionId);
+    if (typeof row?.id !== "string" || typeof row.data !== "string")
+      return null;
+    try {
+      const data = JSON.parse(row.data);
+      if (typeof data.role !== "string")
+        return null;
+      const time = data.time && typeof data.time === "object" ? data.time : null;
+      return {
+        id: row.id,
+        role: data.role,
+        ...typeof data.parentID === "string" ? { parentID: data.parentID } : {},
+        ...typeof time?.completed === "number" ? { completedAt: time.completed } : {},
+        ...data.error === undefined ? {} : { error: data.error }
+      };
+    } catch {
+      return null;
+    }
+  });
+}
 function hasNewerRealUserMessage(db, sessionId, latestAssistantTimeCreated) {
   if (typeof latestAssistantTimeCreated !== "number")
     return false;
@@ -752,18 +843,44 @@ function getMessageTimesFromOpenCodeDb(sessionId, messageIds) {
   return result;
 }
 function findLastAssistantModelFromOpenCodeDb(sessionId) {
+  const resolution = resolveOpenCodeDbPath();
   try {
-    return withReadOnlySessionDb((db) => {
-      const row = db.prepare(`SELECT json_extract(data, '$.providerID') as providerID,
-                            json_extract(data, '$.modelID') as modelID,
-                            json_extract(data, '$.agent') as agent
-                     FROM message
-                     WHERE session_id = ?
-                       AND json_extract(data, '$.role') = 'assistant'
-                       AND json_extract(data, '$.providerID') IS NOT NULL
-                       AND json_extract(data, '$.modelID') IS NOT NULL
-                     ORDER BY time_created DESC
-                     LIMIT 1`).get(sessionId);
+    if (!openCodeDbPathExists(resolution)) {
+      throw new Error(`OpenCode session database is unavailable at ${resolution.path}`);
+    }
+    const db = new Database(resolution.path, { readonly: true });
+    try {
+      const queryV2 = () => db.prepare(`SELECT json_extract(data, '$.model.providerID') as providerID,
+                                json_extract(data, '$.model.id') as modelID,
+                                json_extract(data, '$.agent') as agent
+                         FROM session_message
+                         WHERE session_id = ?
+                           AND type = 'assistant'
+                           AND json_extract(data, '$.model.providerID') IS NOT NULL
+                           AND json_extract(data, '$.model.id') IS NOT NULL
+                         ORDER BY seq DESC
+                         LIMIT 1`).get(sessionId);
+      const queryV1 = () => db.prepare(`SELECT json_extract(data, '$.providerID') as providerID,
+                                json_extract(data, '$.modelID') as modelID,
+                                json_extract(data, '$.agent') as agent
+                         FROM message
+                         WHERE session_id = ?
+                           AND json_extract(data, '$.role') = 'assistant'
+                           AND json_extract(data, '$.providerID') IS NOT NULL
+                           AND json_extract(data, '$.modelID') IS NOT NULL
+                         ORDER BY time_created DESC
+                         LIMIT 1`).get(sessionId);
+      let row = null;
+      try {
+        assertOpenCodeStoreGeneration(db, "v2", resolution.path);
+        row = queryV2();
+      } catch {}
+      if (!row) {
+        try {
+          assertOpenCodeStoreGeneration(db, "v1", resolution.path);
+          row = queryV1();
+        } catch {}
+      }
       if (!row || typeof row.providerID !== "string" || typeof row.modelID !== "string") {
         return null;
       }
@@ -773,11 +890,13 @@ function findLastAssistantModelFromOpenCodeDb(sessionId) {
         modelID: row.modelID,
         ...agent ? { agent } : {}
       };
-    });
+    } finally {
+      closeQuietly(db);
+    }
   } catch (error) {
-    logProbeFailureOnce(resolveOpenCodeDbPath(), error);
+    logProbeFailureOnce(resolution, error);
     return null;
   }
 }
 
-export { resolveOpenCodeDbPath, openCodeDbPathExists, formatOpenCodeDbMissingBanner, formatOpenCodeDbMissingStatusLine, formatOpenCodeDbReadFailureStatusLine, recordOpenCodeDbReadFailure, clearOpenCodeDbReadFailure, getOpenCodeDbReadFailure, claimOpenCodeDbDiagnosticOnce, registerSlowWriteReporter, detectSqliteRuntime, Database, getSqliteMemoryStats, withPrivilegedWriter, closeQuietly, openCodeDbExists, withReadOnlySessionDb, getRawSessionMessageCountFromDb, observeOpenCodeTurnEvent, clearTrackedOpenCodeSession, assistantAwaitingTools, shouldHoldIgnoredNotification, getMessageTimesFromOpenCodeDb, findLastAssistantModelFromOpenCodeDb };
+export { resolveOpenCodeDbPath, hasV1MessageTables, assertOpenCodeStoreGeneration, openCodeDbPathExists, getOpenCodeDbProbeDescriptions, formatOpenCodeDbMissingBanner, formatOpenCodeDbMissingStatusLine, formatOpenCodeDbReadFailureStatusLine, recordOpenCodeDbReadFailure, clearOpenCodeDbReadFailure, getOpenCodeDbReadFailure, claimOpenCodeDbDiagnosticOnce, registerSlowWriteReporter, detectSqliteRuntime, Database, getSqliteMemoryStats, withPrivilegedWriter, closeQuietly, openCodeDbExists, withReadOnlySessionDb, getRawSessionMessageCountFromDb, observeOpenCodeTurnEvent, clearTrackedOpenCodeSession, assistantAwaitingTools, shouldHoldIgnoredNotification, latestPersistedMessageForRecovery, getMessageTimesFromOpenCodeDb, findLastAssistantModelFromOpenCodeDb };

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAiB,YAAY,EAAE,MAAM,qBAAqB,CAAC;AA4FvE,OAAO,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AAq6BpC,QAAA,MAAM,MAAM,EAAE,YAAY,GAAG;IAAE,KAAK,EAAE,OAAO,KAAK,CAAA;CAIjD,CAAC;AAEF,eAAe,MAAM,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAiB,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAgGvE,OAAO,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AA88BpC,QAAA,MAAM,MAAM,EAAE,YAAY,GAAG;IAAE,KAAK,EAAE,OAAO,KAAK,CAAA;CAIjD,CAAC;AAEF,eAAe,MAAM,CAAC"}