@vibe-cafe/vibe-usage 0.9.8 → 0.9.10

package/README.md

@@ -66,7 +66,7 @@ npx @vibe-cafe/vibe-usage status       # Show config & detected tools
 | Kiro | Kiro CLI native event streams `~/.kiro/sessions/cli/*.jsonl` (estimated tokens from message text: input = prompt + tool results, output = reply + tool calls, reasoning = thinking, cacheRead = re-sent context; thinking-block signatures excluded). Falls back to `~/Library/Application Support/kiro-cli/data.sqlite3` / `~/.local/share/kiro-cli/data.sqlite3` + optional `~/.kiro_sessions/*.json` archives, then IDE `q-client.log` whole-credit deltas as `kiro-credits` (floored cumulative diff — the server stores token counts as bigint); legacy IDE `dev_data/devdata.sqlite` token telemetry is opt-in with `VIBE_USAGE_KIRO_LEGACY_TOKENS=1` |
 | Cline | `<host>/User/globalStorage/saoudrizwan.claude-dev/{state/taskHistory.json,tasks/<id>/ui_messages.json}` (walks all VSCode-fork hosts: Code, Cursor, Windsurf, VSCodium, Trae, ...) |
 | Roo Code | `<host>/User/globalStorage/rooveterinaryinc.roo-cline/{tasks/_index.json,tasks/<id>/{history_item,ui_messages}.json}` (walks all VSCode-fork hosts) |
-| Antigravity | `~/.gemini/antigravity/conversations/*.pb` to discover cascades, then reads token usage + sessions from the running language server via Connect RPC (process discovered with `ps`/`lsof` on macOS/Linux, PowerShell CIM with a `wmic` fallback on Windows) |
+| Antigravity | `~/.gemini/antigravity/conversations/*.pb` / `*.db` to discover cascades, then reads token usage + sessions from the running language server via Connect RPC (process discovered with `ps`/`lsof` on macOS/Linux, PowerShell CIM with a `wmic` fallback on Windows) |
 | ZCode | `~/.zcode/cli/db/db.sqlite` (SQLite; reads the `message` table for per-message tokens, model, and project `cwd`/`root`, joined to `session.directory`) |
 
 ## How It Works

package/package.json

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

package/src/parsers/antigravity.js

@@ -8,7 +8,7 @@ import { aggregateToBuckets, extractSessions } from './index.js';
 
 /**
  * Antigravity parser (file-based).
- * Scans .pb files in ~/.gemini/antigravity/conversations/ to discover cascade IDs.
+ * Scans .pb/.db files in ~/.gemini/antigravity/conversations/ to discover cascade IDs.
  * Calls GetCascadeTrajectory via a running language server to extract token usage
  * (from generatorMetadata) and session events (from trajectory steps).
  */
@@ -42,7 +42,7 @@ function findLanguageServer() {
 }
 
 function findLanguageServerUnix() {
-  const out = execSync("ps aux | grep 'antigravity/bin/language_server_'", { encoding: 'utf-8', timeout: 5000 });
+  const out = execSync("ps aux | grep -i 'antigravity.*language_server'", { encoding: 'utf-8', timeout: 5000 });
   for (const line of out.split('\n')) {
     if (!line.trim()) continue;
     if (line.includes('grep')) continue;
@@ -224,6 +224,8 @@ async function probeHttpPort(ports, csrfToken) {
 const MODEL_NORMALIZE_MAP = {
   'claude-opus-4-6-thinking': 'claude-opus-4-6',
   'claude-sonnet-4-6-thinking': 'claude-sonnet-4-6',
+  'gemini-3-flash-a': 'gemini-3-flash',
+  'gemini-3-flash-b': 'gemini-3-flash',
   'gemini-3-flash-c': 'gemini-3-flash',
   "gemini-3.1-pro-high": "gemini-3.1-pro",
   "gemini-3.1-pro-low": "gemini-3.1-pro",
@@ -242,6 +244,10 @@ const PLACEHOLDER_MODEL_MAP = {
   'MODEL_PLACEHOLDER_M35': 'claude-sonnet-4-6',
   'MODEL_PLACEHOLDER_M26': 'claude-opus-4-6',
   'MODEL_OPENAI_GPT_OSS_120B_MEDIUM': 'gpt-oss-120b',
+  // Antigravity 2.0+ standalone app placeholders (verified via responseModel)
+  'MODEL_PLACEHOLDER_M132': 'gemini-3-flash',
+  'MODEL_PLACEHOLDER_M133': 'gemini-3-flash',
+  'MODEL_PLACEHOLDER_M20': 'gemini-3-flash',
 };
 
 function normalizeModel(raw) {
@@ -274,17 +280,21 @@ function projectFromUri(uri) {
 }
 
 /**
- * List cascade IDs from .pb files in the conversations directory.
+ * List cascade IDs from .pb and .db files in the conversations directory.
+ * Antigravity 2.0+ stores new conversations as SQLite .db files instead of .pb.
  */
 function listCascades() {
   try {
     const files = readdirSync(CONVERSATIONS_DIR);
-    const results = [];
+    const seen = new Set();
     for (const f of files) {
-      if (!f.endsWith('.pb')) continue;
-      results.push(f.slice(0, -3)); // strip .pb → cascadeId
+      // SQLite auxiliaries (foo.db-wal / foo.db-shm) don't end with .db, so no
+      // extra filtering is needed here.
+      if (f.endsWith('.pb') || f.endsWith('.db')) {
+        seen.add(f.slice(0, -3));
+      }
     }
-    return results;
+    return [...seen];
   } catch {
     return [];
   }
@@ -293,7 +303,7 @@ function listCascades() {
 // ── Main parse ───────────────────────────────────────────────────────
 
 export async function parse() {
-  // Step 1: List cascade .pb files
+  // Step 1: List cascade conversation files (.pb / .db)
   const cascadeIds = listCascades();
   if (cascadeIds.length === 0) return { buckets: [], sessions: [] };
 

package/src/parsers/cursor.js

@@ -95,20 +95,34 @@ const FETCH_TIMEOUT_MS = 10_000;
 async function fetchUsageCsv(token) {
   const url = `${(process.env.CURSOR_WEB_BASE_URL?.trim() || 'https://cursor.com').replace(/\/+$/, '')}/api/dashboard/export-usage-events-csv?strategy=tokens`;
   const sub = decodeJwtSub(token);
-  const cookieValues = sub ? [token, `${sub}::${token}`] : [token];
+  // The dashboard API authenticates via the WorkosCursorSessionToken cookie in
+  // `{sub}%3A%3A{jwt}` form (what the browser sends). Bearer and bare-token
+  // cookies now return 401, so they're kept only as last-resort fallbacks.
+  const userId = sub?.includes('|') ? sub.split('|').pop() : null;
+  const cookieValues = [
+    ...(sub ? [`${sub}%3A%3A${token}`] : []),
+    ...(userId ? [`${userId}%3A%3A${token}`] : []),
+    token,
+  ];
 
-  const attempts = [{ Authorization: `Bearer ${token}` }];
-  for (const cv of cookieValues) {
-    attempts.push({ Cookie: `${SESSION_COOKIE}=${cv}` });
-    attempts.push({ Authorization: `Bearer ${token}`, Cookie: `${SESSION_COOKIE}=${cv}` });
-  }
+  // Browser-mimicking headers, matching what the dashboard sends (and what
+  // cursor-stats / cursor-price-tracking send) — Node's default UA is a
+  // common target for intermittent WAF blocks on cursor.com.
+  const baseHeaders = {
+    Accept: 'text/csv,*/*;q=0.8',
+    Origin: 'https://cursor.com',
+    Referer: 'https://cursor.com/dashboard?tab=usage',
+    'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36',
+  };
+  const attempts = cookieValues.map(cv => ({ Cookie: `${SESSION_COOKIE}=${cv}` }));
+  attempts.push({ Authorization: `Bearer ${token}` });
 
   const failures = [];
   for (const headers of attempts) {
     let resp;
     try {
       resp = await fetch(url, {
-        headers: { Accept: 'text/csv,*/*;q=0.8', ...headers },
+        headers: { ...baseHeaders, ...headers },
         signal: AbortSignal.timeout(FETCH_TIMEOUT_MS),
       });
     } catch (e) {
@@ -121,9 +135,18 @@ async function fetchUsageCsv(token) {
     }
     if (resp.ok) return await resp.text();
     failures.push(`${resp.status} ${resp.statusText}`);
+    // Only auth rejections are worth retrying with different credentials.
+    // 429/5xx are transient server-side states — soft-skip like network errors
+    // instead of surfacing them as auth failures every daemon cycle.
+    if (resp.status !== 401 && resp.status !== 403) {
+      const err = new Error(`Cursor usage export skipped (HTTP ${resp.status} ${resp.statusText})`);
+      err.skip = true;
+      throw err;
+    }
   }
-  // All auth combos rejected — token is likely expired. Surface to user.
-  throw new Error(`Cursor usage export auth failed (${failures.join('; ')})`);
+  // Every auth combo rejected — the stored token no longer works. Surface an
+  // actionable message: re-signing in inside Cursor rewrites the token.
+  throw new Error(`Cursor session rejected (${failures.join('; ')}). Open Cursor and sign in again (Cursor Settings → Account), then re-run sync.`);
 }
 
 function parseCsv(text) {