@vibe-cafe/vibe-usage 0.10.5 → 0.10.6

package/README.md

@@ -51,7 +51,7 @@ npx @vibe-cafe/vibe-usage status       # Show config & detected tools
 
 | Tool | Data Location |
 |------|---------------|
-| Claude Code | `~/.claude/projects/` (tokens + sessions), `~/.claude/transcripts/` (sessions only); also scans `$CLAUDE_CONFIG_DIR` and data-bearing `~/.claude-*` profiles, selects the most complete duplicate session, and streams logs so large active sessions are not silently omitted. Cache creation tokens are included in input usage. |
+| Claude Code + Claude Desktop Code/Cowork | Claude Code data in `~/.claude/projects/` (tokens + sessions) and `~/.claude/transcripts/` (sessions only), plus Claude Desktop Cowork's per-session `.claude/projects/` directories. Also scans `$CLAUDE_CONFIG_DIR` and data-bearing `~/.claude-*` profiles. All variants use the existing `claude-code` source; the parser selects the most complete copy of each session so shared/copied transcripts are not counted twice. Logs are streamed and cache creation tokens are included in input usage. |
 | Codex CLI | `$CODEX_HOME/sessions/` and `$CODEX_HOME/archived_sessions/` (default `~/.codex`), plus an optional temporary `--extra-codex-home` or manually persisted `codexExtraHome`; a versioned local index avoids re-reading unchanged rollouts and reads only safe append tails for ordinary sessions, while fork/sub-agent replay matching, duplicate suppression, and live/archive/cross-root deduplication retain their existing semantics |
 | Grok | `$GROK_HOME/sessions/<encoded-cwd>/<session-id>/` (default `~/.grok`); token usage from `updates.jsonl` `turn_completed.usage` (per-model `modelUsage`, cache reads, reasoning); project from `summary.json` cwd; honors `GROK_HOME` |
 | GitHub Copilot CLI | `~/.copilot/session-state/*/events.jsonl` |

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@vibe-cafe/vibe-usage",
-  "version": "0.10.5",
+  "version": "0.10.6",
   "description": "Track your AI coding tool token usage and sync to vibecafe.ai",
   "type": "module",
   "scripts": {

package/src/claude-roots.js

@@ -2,6 +2,9 @@ import { existsSync, readdirSync, realpathSync, statSync } from 'node:fs';
 import { delimiter, join } from 'node:path';
 import { homedir } from 'node:os';
 
+const MAX_DESKTOP_DISCOVERY_DEPTH = 8;
+const DESKTOP_NON_SESSION_DIRS = new Set(['rpm', 'skills']);
+
 function expandHome(value) {
   const trimmed = value.trim().replace(/[/\\]+$/, '');
   if (trimmed === '~') return homedir();
@@ -15,18 +18,95 @@ function hasClaudeData(root) {
   return existsSync(join(root, 'projects')) || existsSync(join(root, 'transcripts'));
 }
 
+function defaultClaudeDesktopDataDir() {
+  if (process.platform === 'darwin') {
+    return join(homedir(), 'Library', 'Application Support', 'Claude');
+  }
+  if (process.platform === 'win32') {
+    const appData = process.env.APPDATA?.trim();
+    return appData
+      ? join(expandHome(appData), 'Claude')
+      : join(homedir(), 'AppData', 'Roaming', 'Claude');
+  }
+  const configHome = process.env.XDG_CONFIG_HOME?.trim();
+  return join(configHome ? expandHome(configHome) : join(homedir(), '.config'), 'Claude');
+}
+
+function getClaudeDesktopDataDirs() {
+  const override = process.env.VIBE_USAGE_CLAUDE_DESKTOP_DIRS?.trim();
+  return override
+    ? override.split(delimiter).map(expandHome).filter(Boolean)
+    : [defaultClaudeDesktopDataDir()];
+}
+
+function discoverDesktopRoots(dir, depth, roots, onWarning) {
+  let entries;
+  try {
+    entries = readdirSync(dir, { withFileTypes: true });
+  } catch (err) {
+    if (err?.code !== 'ENOENT') {
+      onWarning(`Claude Desktop: cannot read directory ${dir}: ${err.message}`);
+    }
+    return;
+  }
+
+  // Once a session root is found, do not descend into the user's Cowork files.
+  // Those directories can be large and are unrelated to Claude's transcript.
+  const claudeEntry = entries.find(
+    (entry) => entry.name === '.claude' && entry.isDirectory(),
+  );
+  if (claudeEntry) {
+    roots.push(join(dir, claudeEntry.name));
+    return;
+  }
+
+  for (const entry of entries) {
+    if (!entry.isDirectory()) continue;
+    if (DESKTOP_NON_SESSION_DIRS.has(entry.name)) continue;
+    const candidate = join(dir, entry.name);
+    if (depth < MAX_DESKTOP_DISCOVERY_DEPTH) {
+      discoverDesktopRoots(candidate, depth + 1, roots, onWarning);
+    }
+  }
+}
+
+/**
+ * Find the private Claude Code state roots created for Claude Desktop Cowork.
+ * Desktop Code itself uses the normal ~/.claude root, while Cowork isolates
+ * each local-agent session below the Electron user-data directory.
+ */
+export function findClaudeDesktopRoots(
+  desktopDataDirs = getClaudeDesktopDataDirs(),
+  onWarning = () => {},
+) {
+  const roots = [];
+  for (const dataDir of desktopDataDirs) {
+    discoverDesktopRoots(
+      join(dataDir, 'local-agent-mode-sessions'),
+      0,
+      roots,
+      onWarning,
+    );
+  }
+  return roots;
+}
+
 /**
- * Return every Claude Code state root visible from this process.
+ * Return every Claude Code-compatible state root visible from this process.
  *
  * In addition to the default and CLAUDE_CONFIG_DIR, discover the documented
  * multi-profile convention (~/.claude-work, ~/.claude-personal, ...). This is
  * important for launchd/systemd and GUI processes, which commonly do not
  * inherit the shell environment used to launch Claude Code.
  *
- * VIBE_USAGE_CLAUDE_DIRS is a test/diagnostic override. It replaces discovery
- * with a path.delimiter-separated root list.
+ * Claude Desktop Code uses the default Claude Code root. Cowork creates a
+ * private .claude root per local-agent session, so those roots are discovered
+ * recursively under the app's user-data directory.
+ *
+ * VIBE_USAGE_CLAUDE_DIRS is a test/diagnostic override. It replaces all normal
+ * and Desktop discovery with a path.delimiter-separated root list.
  */
-export function getClaudeRoots() {
+export function getClaudeRoots({ onWarning = () => {} } = {}) {
   const override = process.env.VIBE_USAGE_CLAUDE_DIRS?.trim();
   const roots = override
     ? override.split(delimiter).map(expandHome).filter(Boolean)
@@ -47,6 +127,8 @@ export function getClaudeRoots() {
     } catch {
       // The default/configured roots remain usable if home discovery fails.
     }
+
+    roots.push(...findClaudeDesktopRoots(getClaudeDesktopDataDirs(), onWarning));
   }
 
   const seen = new Set();

package/src/parsers/claude-code.js

@@ -272,7 +272,9 @@ export async function parse() {
     warnings: [],
     incomplete: false,
   };
-  const roots = getClaudeRoots();
+  const roots = getClaudeRoots({
+    onWarning: (message) => addWarning(ctx, message),
+  });
   const projectGroups = collectCandidates(roots, 'projects', ctx);
   const projectSessionIds = new Set();