tokentracker-cli 0.56.0 → 0.57.0

package/src/lib/rollout.js

@@ -2520,6 +2520,67 @@ function readOpencodeDbMessages(dbPath, sqliteOptions = {}) {
   return out;
 }
 
+// Mimo (mimocode) reuses OpenCode's `message` table schema, but its DB ALSO
+// contains the user's imported Claude Code history: on install, mimocode reads
+// ~/.claude/projects/*.jsonl and copies those sessions into its own DB,
+// recording each in the `claude_import` table (source_path → session_id +
+// message_ids). Those imported messages are already counted by the Claude
+// parser, so counting them here would double-count Claude usage AND mislabel it
+// as "mimo".
+//
+// `message_ids` was added to claude_import via a migration, so rows written by
+// an older mimocode have it NULL. For those we fall back to excluding the whole
+// imported session by session_id — a claude_import.session_id is always an
+// import-created session, never a native mimo one, so this is safe. Rows that
+// DO carry message_ids use precise per-id exclusion, which preserves any native
+// turns the user later added to an imported session.
+function readMimoImportedKeys(dbPath, sqliteOptions = {}) {
+  const sql = `SELECT session_id, message_ids FROM claude_import`;
+  let rows;
+  try {
+    rows = readSqliteJsonRows(dbPath, sql, {
+      label: "Mimo import",
+      maxBuffer: 200 * 1024 * 1024,
+      timeout: 30_000,
+      ...sqliteOptions,
+    });
+  } catch (_e) {
+    // claude_import table absent (older/newer mimocode) → treat as no imports.
+    return { ids: new Set(), sessions: new Set() };
+  }
+  const ids = new Set();
+  const sessions = new Set();
+  for (const row of rows) {
+    if (!row) continue;
+    let arr = null;
+    if (typeof row.message_ids === "string") {
+      try {
+        arr = JSON.parse(row.message_ids);
+      } catch (_e) {
+        arr = null;
+      }
+    }
+    if (Array.isArray(arr) && arr.length > 0) {
+      for (const id of arr) if (typeof id === "string") ids.add(id);
+    } else if (typeof row.session_id === "string" && row.session_id) {
+      sessions.add(row.session_id);
+    }
+  }
+  return { ids, sessions };
+}
+
+// Read only NATIVE mimo assistant messages — token-bearing rows that are not
+// part of any claude_import row (by message id, or session id for legacy import
+// rows without message_ids). See readMimoImportedKeys for why.
+function readMimoDbMessages(dbPath, sqliteOptions = {}) {
+  if (!dbPath || !fssync.existsSync(dbPath)) return [];
+  const all = readOpencodeDbMessages(dbPath, sqliteOptions);
+  if (all.length === 0) return all;
+  const { ids, sessions } = readMimoImportedKeys(dbPath, sqliteOptions);
+  if (ids.size === 0 && sessions.size === 0) return all;
+  return all.filter((m) => !ids.has(m.id) && !sessions.has(m.sessionID));
+}
+
 async function parseOpencodeDbIncremental({
   dbMessages,
   cursors,
@@ -9033,6 +9094,7 @@ module.exports = {
   listGeminiSessionFiles,
   listOpencodeMessageFiles,
   readOpencodeDbMessages,
+  readMimoDbMessages,
   resolveKiroDbPath,
   resolveKiroJsonlPath,
   resolveHermesPath,