@vibe-cafe/vibe-usage 0.10.19 → 0.10.21

package/src/pi-roots.js

@@ -1,6 +1,7 @@
-import { existsSync, readdirSync } from 'node:fs';
-import { delimiter, join } from 'node:path';
+import { existsSync, readdirSync, readFileSync } from 'node:fs';
+import { delimiter, isAbsolute, join } from 'node:path';
 import { homedir } from 'node:os';
+import { piSessionsDir } from './extra-roots.js';
 
 function expandHome(value) {
   const trimmed = value.trim();
@@ -41,18 +42,49 @@ export function looksLikeOmpAgentDir(agentDir) {
     || existsSync(join(agentDir, 'agent.db'));
 }
 
-export function getPiSessionDirs() {
+// Pi resolves a session directory from `--session-dir`, then
+// PI_CODING_AGENT_SESSION_DIR, then `sessionDir` in settings.json. Only the
+// last two are discoverable after the fact, and both name the sessions
+// directory itself (no `sessions` segment is appended).
+function settingsSessionDir(agentDir) {
+  try {
+    const raw = readFileSync(join(agentDir, 'settings.json'), 'utf-8');
+    const value = JSON.parse(raw)?.sessionDir;
+    if (typeof value !== 'string') return null;
+    const trimmed = value.trim();
+    // Pi also accepts project-relative values. Those resolve against a cwd we
+    // do not have here, so only absolute and ~-anchored paths are scanned.
+    if (!trimmed || !(isAbsolute(trimmed) || trimmed.startsWith('~'))) return null;
+    return expandHome(trimmed);
+  } catch {
+    return null;
+  }
+}
+
+export function getPiSessionDirs(extraRoots = []) {
+  // Explicitly configured roots are user intent, not a fixture: they are always
+  // scanned, and they never replace the default store. A root whose shape no
+  // longer resolves drops out here; the parser reports that as skipped.
+  const extraDirs = extraRoots.map(piSessionsDir).filter(dir => dir !== null);
+
   const override = process.env.VIBE_USAGE_PI_SESSION_DIRS?.trim();
-  if (override) return uniqueExistingDirs(override.split(delimiter));
+  if (override) return uniqueExistingDirs([...override.split(delimiter), ...extraDirs]);
 
-  const agentDir = process.env.PI_CODING_AGENT_DIR?.trim();
-  if (agentDir) {
-    const expanded = expandHome(agentDir);
-    // OMP inherits PI_CODING_AGENT_DIR from Pi. Do not parse an identifiable
-    // OMP store again as source=pi-coding-agent.
-    return looksLikeOmpAgentDir(expanded) ? [] : uniqueExistingDirs([join(expanded, 'sessions')]);
+  const envAgentDir = process.env.PI_CODING_AGENT_DIR?.trim();
+  const agentDir = envAgentDir ? expandHome(envAgentDir) : join(homedir(), '.pi', 'agent');
+  // OMP inherits PI_CODING_AGENT_DIR from Pi. Do not parse an identifiable
+  // OMP store again as source=pi-coding-agent.
+  const isOmpStore = Boolean(envAgentDir) && looksLikeOmpAgentDir(agentDir);
+
+  const dirs = [];
+  if (!isOmpStore) {
+    dirs.push(join(agentDir, 'sessions'));
+    const envSessionDir = process.env.PI_CODING_AGENT_SESSION_DIR?.trim();
+    if (envSessionDir) dirs.push(expandHome(envSessionDir));
+    const configured = settingsSessionDir(agentDir);
+    if (configured) dirs.push(configured);
   }
-  return uniqueExistingDirs([join(homedir(), '.pi', 'agent', 'sessions')]);
+  return uniqueExistingDirs([...dirs, ...extraDirs]);
 }
 
 export function getOmpSessionDirs() {

package/src/sync.js

@@ -9,6 +9,7 @@ import { createSyncClient, forBatch } from './client-meta.js';
 import { parsers } from './parsers/index.js';
 import { aggregateToBuckets } from './parsers/aggregate.js';
 import { normalizeParserResult } from './parsers/contract.js';
+import { extraRootList } from './extra-roots.js';
 import { success, failure, warn, arrow, link, dim } from './output.js';
 
 const BATCH_SIZE = 100;
@@ -20,6 +21,12 @@ function formatBytes(bytes) {
   return `${(bytes / (1024 * 1024)).toFixed(1)}MB`;
 }
 
+/** Hide only Cursor's intentional transient fetch soft-skip in quiet (daemon) syncs. */
+export function shouldSuppressParserWarning(source, message, quiet) {
+  if (!quiet) return false;
+  return source === 'cursor' && message.startsWith('cursor: Cursor usage export skipped (');
+}
+
 export function resolveUploadProjectSetting(settings) {
   if (typeof settings?.uploadProject !== 'boolean') {
     const error = new Error('SETTINGS_UNAVAILABLE');
@@ -147,9 +154,12 @@ export async function runSync({
     PARSER_CONCURRENCY,
     async ([source, parse]) => {
       try {
-        const result = source === 'codex'
-          ? await parse({ codexExtraHome: resolveCodexExtraHome(config.codexExtraHome, codexExtraHome) })
-          : await parse();
+        const result = await parse({
+          extraRoots: extraRootList(config.extraRoots?.[source]),
+          ...(source === 'codex' ? {
+            codexExtraHome: resolveCodexExtraHome(config.codexExtraHome, codexExtraHome),
+          } : {}),
+        });
         return { source, result };
       } catch (err) {
         return { source, error: err };
@@ -175,6 +185,7 @@ export async function runSync({
       parserProgress.push({ source, ...indexing });
     }
     for (const message of warnings) {
+      if (shouldSuppressParserWarning(source, message, quiet)) continue;
       process.stderr.write(`${dim(`  ${message}`)}\n`);
     }
     // A parser may deliberately suppress a transient error (Cursor network

package/src/tools.js

@@ -1,9 +1,15 @@
 import { existsSync, readdirSync, statSync } from 'node:fs';
-import { isAbsolute, join, posix, resolve, win32 } from 'node:path';
+import { dirname, isAbsolute, join, posix, resolve, win32 } from 'node:path';
 import { homedir } from 'node:os';
 import { findClaudeCodeDataDirs } from './claude-roots.js';
 import { findCindyDataDirs, getCindyDataRoots } from './cindy-roots.js';
 import { codexSessionDirs, resolveCodexHomes } from './codex-roots.js';
+import {
+  antigravityConversationDirs,
+  discoverCodexHomes,
+  extraRootList,
+  grokSessionsDir,
+} from './extra-roots.js';
 import { findClineDataDirs } from './cline-roots.js';
 import { findCraftDataDirs } from './craft-roots.js';
 import { findOmpDataDirs, findPiDataDirs } from './pi-roots.js';
@@ -106,8 +112,9 @@ function findOpenclawDataDirs() {
 // Codex keeps live sessions in ~/.codex/sessions and moves completed ones to
 // ~/.codex/archived_sessions. Detect Codex if either dir exists, so a user
 // whose sessions have all been archived is still recognized.
-export function findCodexDataDirs(codexExtraHome) {
-  return resolveCodexHomes(codexExtraHome)
+export function findCodexDataDirs(codexExtraHome, extraRoots = []) {
+  const configuredHomes = extraRoots.flatMap(root => discoverCodexHomes(root).homes);
+  return [...new Set([...resolveCodexHomes(codexExtraHome), ...configuredHomes])]
     .flatMap(codexSessionDirs)
     .filter(existsSync);
 }
@@ -143,6 +150,17 @@ export function findDshDataDirs() {
   return [getDshSessionsDir()].filter(existsSync);
 }
 
+/** mcode runtime database: VIBE_USAGE_MCODE_DB wins, then MCODE_HOME, then ~/.minimax. */
+export function getMcodeDbPath(env = process.env, home = homedir()) {
+  const override = env.VIBE_USAGE_MCODE_DB?.trim();
+  if (override) return isAbsolute(override) ? override : resolve(override);
+  if (env.MCODE_HOME && !isAbsolute(env.MCODE_HOME)) {
+    throw new Error(`MCODE_HOME must be an absolute path, got: ${JSON.stringify(env.MCODE_HOME)}`);
+  }
+  const root = env.MCODE_HOME || join(home, '.minimax');
+  return join(root, 'v2', 'sqlite', 'runtime-state.sqlite');
+}
+
 export function getMimocodeDbPath(env = process.env) {
   if (env.MIMOCODE_HOME && !isAbsolute(env.MIMOCODE_HOME)) {
     throw new Error(`MIMOCODE_HOME must be an absolute path, got: ${JSON.stringify(env.MIMOCODE_HOME)}`);
@@ -154,11 +172,12 @@ export function getMimocodeDbPath(env = process.env) {
   return isAbsolute(env.MIMOCODE_DB) ? env.MIMOCODE_DB : join(dataDir, env.MIMOCODE_DB);
 }
 
-function findAntigravityDataDirs() {
-  return [
+export function findAntigravityDataDirs(extraRoots = []) {
+  return [...new Set([
     join(homedir(), '.gemini', 'antigravity'),
     join(homedir(), '.gemini', 'antigravity-cli'),
-  ].filter(existsSync);
+    ...extraRoots.flatMap(root => antigravityConversationDirs(root).map(dirname)),
+  ])].filter(existsSync);
 }
 
 export function findTraeCliDataDirs() {
@@ -193,10 +212,13 @@ export function getGrokSessionsDir() {
 }
 
 // Detect Grok when sessions/ exists under GROK_HOME (or the test override).
-export function findGrokDataDirs() {
+export function findGrokDataDirs(extraRoots = []) {
   const testDir = process.env.VIBE_USAGE_GROK_SESSIONS?.trim();
   if (testDir) return [testDir].filter(existsSync);
-  return [join(getGrokHome(), 'sessions')].filter(existsSync);
+  return [...new Set([
+    join(getGrokHome(), 'sessions'),
+    ...extraRoots.map(grokSessionsDir),
+  ])].filter(existsSync);
 }
 
 export function getDimAgentDbPath() {
@@ -240,13 +262,15 @@ export const TOOLS = [
     name: 'Codex CLI',
     id: 'codex',
     dataDir: join(homedir(), '.codex', 'sessions'),
-    detectDataDirs: ({ codexExtraHome } = {}) => findCodexDataDirs(codexExtraHome),
+    detectDataDirs: ({ codexExtraHome, extraRoots } = {}) => (
+      findCodexDataDirs(codexExtraHome, extraRootList(extraRoots?.codex))
+    ),
   },
   {
     name: 'Grok',
     id: 'grok',
     dataDir: join(homedir(), '.grok', 'sessions'),
-    detectDataDirs: findGrokDataDirs,
+    detectDataDirs: ({ extraRoots } = {}) => findGrokDataDirs(extraRootList(extraRoots?.grok)),
   },
   {
     name: 'GitHub Copilot CLI',
@@ -296,7 +320,9 @@ export const TOOLS = [
     name: 'pi',
     id: 'pi-coding-agent',
     dataDir: join(homedir(), '.pi', 'agent', 'sessions'),
-    detectDataDirs: findPiDataDirs,
+    detectDataDirs: ({ extraRoots } = {}) => (
+      findPiDataDirs(extraRootList(extraRoots?.['pi-coding-agent']))
+    ),
   },
   {
     name: 'Qwen Code',
@@ -311,6 +337,12 @@ export const TOOLS = [
     dataDir: join(homedir(), '.kimi-code', 'sessions'),
     detectDataDirs: findKimiCodeDataDirs,
   },
+  {
+    name: 'MiniMax Code',
+    id: 'mcode',
+    dataDir: join(homedir(), '.minimax', 'v2', 'sqlite', 'runtime-state.sqlite'),
+    detectDataDirs: () => [getMcodeDbPath()].filter(existsSync),
+  },
   {
     name: 'MiMoCode',
     id: 'mimocode',
@@ -337,7 +369,9 @@ export const TOOLS = [
     name: 'Antigravity',
     id: 'antigravity',
     dataDir: join(homedir(), '.gemini', 'antigravity'),
-    detectDataDirs: findAntigravityDataDirs,
+    detectDataDirs: ({ extraRoots } = {}) => (
+      findAntigravityDataDirs(extraRootList(extraRoots?.antigravity))
+    ),
   },
   {
     name: 'Trae CLI',