@vibe-cafe/vibe-usage 0.7.0 → 0.7.1

package/package.json

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

package/src/parsers/antigravity.js

@@ -0,0 +1,346 @@
+import { execSync } from 'node:child_process';
+import { readdirSync, existsSync } from 'node:fs';
+import { join } from 'node:path';
+import { homedir } from 'node:os';
+import { aggregateToBuckets, extractSessions } from './index.js';
+
+
+
+/**
+ * Antigravity parser (file-based).
+ * Scans .pb 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).
+ */
+
+const SOURCE = 'antigravity';
+const CONVERSATIONS_DIR = join(homedir(), '.gemini', 'antigravity', 'conversations');
+
+// User sources → role 'user'; Model source → role 'assistant'; System sources → skip
+const USER_SOURCES = new Set([
+  'CORTEX_STEP_SOURCE_USER_EXPLICIT',
+  'CORTEX_STEP_SOURCE_USER_IMPLICIT',
+]);
+const ASSISTANT_SOURCES = new Set([
+  'CORTEX_STEP_SOURCE_MODEL',
+]);
+
+// ── Process discovery (single instance) ──────────────────────────────
+
+const IS_WIN = process.platform === 'win32';
+
+/**
+ * Find ONE running language server process with a CSRF token.
+ * Returns { pid, csrfToken } or null.
+ */
+function findLanguageServer() {
+  try {
+    return IS_WIN ? findLanguageServerWin() : findLanguageServerUnix();
+  } catch {
+    return null;
+  }
+}
+
+function findLanguageServerUnix() {
+  const out = execSync("ps aux | grep 'antigravity/bin/language_server_'", { encoding: 'utf-8', timeout: 5000 });
+  for (const line of out.split('\n')) {
+    if (!line.trim()) continue;
+    if (line.includes('grep')) continue;
+    const parts = line.trim().split(/\s+/);
+    if (parts.length < 2) continue;
+    const pid = parts[1];
+    const csrfMatch = line.match(/--csrf_token\s+([0-9a-f-]+)/);
+    const csrfToken = csrfMatch ? csrfMatch[1] : '';
+    if (csrfToken) return { pid, csrfToken };
+  }
+  return null;
+}
+
+function findLanguageServerWin() {
+  const out = execSync(
+    'wmic process where "CommandLine like \'%antigravity%language_server%\'" get ProcessId,CommandLine /format:list',
+    { encoding: 'utf-8', timeout: 5000, shell: 'cmd.exe' },
+  );
+  // wmic /format:list outputs lines like "CommandLine=..." and "ProcessId=..."
+  let cmdLine = '';
+  let pid = '';
+  for (const line of out.split('\n')) {
+    const trimmed = line.trim();
+    if (trimmed.startsWith('CommandLine=')) {
+      const val = trimmed.slice('CommandLine='.length);
+      if (/WMIC\.exe/i.test(val)) continue; // skip wmic's own process
+      cmdLine = val;
+    }
+    if (trimmed.startsWith('ProcessId=')) pid = trimmed.slice('ProcessId='.length);
+  }
+  if (!pid || !cmdLine) return null;
+  const csrfMatch = cmdLine.match(/--csrf_token\s+([0-9a-f-]+)/);
+  const csrfToken = csrfMatch ? csrfMatch[1] : '';
+  if (!csrfToken) return null;
+  return { pid, csrfToken };
+}
+
+function findListeningPorts(pid) {
+  try {
+    return IS_WIN ? findListeningPortsWin(pid) : findListeningPortsUnix(pid);
+  } catch {
+    return [];
+  }
+}
+
+function findListeningPortsUnix(pid) {
+  const out = execSync(`lsof -iTCP -sTCP:LISTEN -nP -a -p ${pid}`, {
+    encoding: 'utf-8',
+    timeout: 5000,
+  });
+  const ports = [];
+  for (const line of out.split('\n')) {
+    const match = line.match(/:(\d+)\s+\(LISTEN\)/);
+    if (match) ports.push(parseInt(match[1], 10));
+  }
+  return ports;
+}
+
+function findListeningPortsWin(pid) {
+  // netstat output: TCP  127.0.0.1:49327  0.0.0.0:0  LISTENING  12345
+  const out = execSync('netstat -ano', { encoding: 'utf-8', timeout: 5000 });
+  const ports = [];
+  for (const line of out.split('\n')) {
+    if (!line.includes('LISTENING')) continue;
+    const parts = line.trim().split(/\s+/);
+    // parts: [TCP, local_addr:port, foreign_addr, LISTENING, pid]
+    const linePid = parts[parts.length - 1];
+    if (linePid !== String(pid)) continue;
+    const addrMatch = parts[1]?.match(/:(\d+)$/);
+    if (addrMatch) ports.push(parseInt(addrMatch[1], 10));
+  }
+  return ports;
+}
+
+async function rpcPost(baseUrl, path, body, csrfToken, timeoutMs = 10000) {
+  const url = new URL(path, baseUrl);
+  const headers = {
+    'Content-Type': 'application/json',
+    'Connect-Protocol-Version': '1',
+  };
+  if (csrfToken) headers['X-Codeium-Csrf-Token'] = csrfToken;
+
+  const res = await fetch(url, {
+    method: 'POST',
+    headers,
+    body: JSON.stringify(body),
+    signal: AbortSignal.timeout(timeoutMs),
+  });
+  if (!res.ok) throw new Error(`HTTP ${res.status} from ${path}`);
+  return res.json();
+}
+
+async function probeHttpPort(ports, csrfToken) {
+  for (const port of ports) {
+    const baseUrl = `http://127.0.0.1:${port}`;
+    try {
+      await rpcPost(
+        baseUrl,
+        '/exa.language_server_pb.LanguageServerService/GetWorkspaceInfos',
+        {},
+        csrfToken,
+        3000,
+      );
+      return baseUrl;
+    } catch {
+      // Not the right port, try next
+    }
+  }
+  return null;
+}
+
+// ── Helpers ──────────────────────────────────────────────────────────
+
+/**
+ * Normalize model names to canonical forms.
+ */
+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-c': 'gemini-3-flash',
+  "gemini-3.1-pro-high": "gemini-3.1-pro",
+  "gemini-3.1-pro-low": "gemini-3.1-pro",
+  "gemini-3-pro-high": "gemini-3-pro",
+  "gemini-3-pro-low": "gemini-3-pro",
+};
+
+/**
+ * Map internal placeholder model IDs to canonical names.
+ * Used when responseModel is empty and only chatModel.model is available.
+ */
+const PLACEHOLDER_MODEL_MAP = {
+  'MODEL_PLACEHOLDER_M37': 'gemini-3.1-pro',
+  'MODEL_PLACEHOLDER_M36': 'gemini-3.1-pro',
+  'MODEL_PLACEHOLDER_M47': 'gemini-3-flash',
+  'MODEL_PLACEHOLDER_M35': 'claude-sonnet-4-6',
+  'MODEL_PLACEHOLDER_M26': 'claude-opus-4-6',
+  'MODEL_OPENAI_GPT_OSS_120B_MEDIUM': 'gpt-oss-120b',
+};
+
+function normalizeModel(raw) {
+  return MODEL_NORMALIZE_MAP[raw] || raw;
+}
+
+/**
+ * Resolve model name: prefer responseModel, fall back to placeholder map.
+ */
+function resolveModel(chatModel) {
+  if (chatModel.responseModel) return normalizeModel(chatModel.responseModel);
+  const placeholder = chatModel.model || '';
+  if (PLACEHOLDER_MODEL_MAP[placeholder]) return PLACEHOLDER_MODEL_MAP[placeholder];
+  return 'unknown';
+}
+
+function toSafeNumber(value) {
+  if (value == null) return 0;
+  const n = Number(value);
+  return Number.isFinite(n) ? n : 0;
+}
+
+/**
+ * Extract project name from a workspace URI (e.g. "file:///Users/x/myproject" → "myproject").
+ */
+function projectFromUri(uri) {
+  if (!uri) return null;
+  const parts = uri.replace(/\/$/, '').split('/');
+  return parts[parts.length - 1] || null;
+}
+
+/**
+ * List cascade IDs from .pb files in the conversations directory.
+ */
+function listCascades() {
+  try {
+    const files = readdirSync(CONVERSATIONS_DIR);
+    const results = [];
+    for (const f of files) {
+      if (!f.endsWith('.pb')) continue;
+      results.push(f.slice(0, -3)); // strip .pb → cascadeId
+    }
+    return results;
+  } catch {
+    return [];
+  }
+}
+
+// ── Main parse ───────────────────────────────────────────────────────
+
+export async function parse() {
+  // Step 1: List cascade .pb files
+  const cascadeIds = listCascades();
+  if (cascadeIds.length === 0) return { buckets: [], sessions: [] };
+
+  // Step 2: Find a running language server to make RPC calls
+  const server = findLanguageServer();
+  if (!server) return { buckets: [], sessions: [] };
+
+  const ports = findListeningPorts(server.pid);
+  if (ports.length === 0) return { buckets: [], sessions: [] };
+
+  const baseUrl = await probeHttpPort(ports, server.csrfToken);
+  if (!baseUrl) return { buckets: [], sessions: [] };
+
+  const rpc = (method, body) =>
+    rpcPost(
+      baseUrl,
+      `/exa.language_server_pb.LanguageServerService/${method}`,
+      body,
+      server.csrfToken,
+    );
+
+  // Step 3: Fetch trajectory for each changed cascade
+  const entries = [];
+  const sessionEvents = [];
+  const seenResponseIds = new Set();
+
+  for (const cascadeId of cascadeIds) {
+    let resp;
+    try {
+      resp = await rpc('GetCascadeTrajectory', { cascadeId });
+    } catch {
+      continue; // skip this cascade if RPC fails
+    }
+
+    const trajectory = resp?.trajectory;
+    if (!trajectory) continue;
+
+    const steps = trajectory.steps || [];
+    const metadataList = trajectory.generatorMetadata || [];
+
+
+    // Extract project from trajectory metadata workspaces
+    let project = 'unknown';
+    const workspaces = trajectory.metadata?.workspaces || [];
+    if (workspaces.length > 0) {
+      project = workspaces[0].repository?.computedName || projectFromUri(workspaces[0].workspaceFolderAbsoluteUri) || 'unknown';
+    }
+
+    // ── Token entries from generatorMetadata ──
+    for (const meta of metadataList) {
+      const chatModel = meta?.chatModel;
+      if (!chatModel) continue;
+
+      const responseModel = resolveModel(chatModel);
+      const createdAt = chatModel?.chatStartMetadata?.createdAt;
+      const ts = createdAt ? new Date(createdAt) : null;
+      if (!ts || isNaN(ts.getTime())) continue;
+
+      const retryInfos = chatModel.retryInfos || [];
+      for (const retry of retryInfos) {
+        const usage = retry.usage;
+        if (!usage) continue;
+
+        const responseId = usage.responseId || '';
+        if (responseId && seenResponseIds.has(responseId)) continue;
+        if (responseId) seenResponseIds.add(responseId);
+
+        entries.push({
+          source: SOURCE,
+          model: responseModel,
+          project,
+          timestamp: ts,
+          inputTokens: toSafeNumber(usage.inputTokens),
+          outputTokens: toSafeNumber(usage.outputTokens),
+          cachedInputTokens: toSafeNumber(usage.cacheReadTokens),
+          reasoningOutputTokens: toSafeNumber(usage.thinkingOutputTokens),
+        });
+      }
+    }
+
+    // ── Session events from trajectory steps ──
+    for (const step of steps) {
+      const stepSource = step?.metadata?.source || '';
+      let role;
+      if (USER_SOURCES.has(stepSource)) {
+        role = 'user';
+      } else if (ASSISTANT_SOURCES.has(stepSource)) {
+        role = 'assistant';
+      } else {
+        continue; // skip SYSTEM / SYSTEM_SDK / UNSPECIFIED
+      }
+
+      const createdAt = step?.metadata?.createdAt;
+      const ts = createdAt ? new Date(createdAt) : null;
+      if (!ts || isNaN(ts.getTime())) continue;
+
+      sessionEvents.push({
+        sessionId: cascadeId,
+        source: SOURCE,
+        project,
+        timestamp: ts,
+        role,
+      });
+    }
+
+  }
+
+  return {
+    buckets: aggregateToBuckets(entries),
+    sessions: extractSessions(sessionEvents),
+  };
+}

package/src/parsers/index.js

@@ -9,6 +9,7 @@ import { parse as parseQwenCode } from './qwen-code.js';
 import { parse as parseKimiCode } from './kimi-code.js';
 import { parse as parseAmp } from './amp.js';
 import { parse as parseDroid } from './droid.js';
+import { parse as parseAntigravity } from './antigravity.js';
 import { parse as parsePiCodingAgent } from './pi-coding-agent.js';
 
 export const parsers = {
@@ -22,6 +23,7 @@ export const parsers = {
   'kimi-code': parseKimiCode,
   'amp': parseAmp,
   'droid': parseDroid,
+  'antigravity': parseAntigravity,
   'pi-coding-agent': parsePiCodingAgent,
 };
 

package/src/tools.js

@@ -3,6 +3,11 @@ import { join } from 'node:path';
 import { homedir } from 'node:os';
 
 export const TOOLS = [
+  {
+    name: 'Antigravity',
+    id: 'antigravity',
+    dataDir: join(homedir(), '.gemini', 'antigravity'),
+  },
   {
     name: 'Claude Code',
     id: 'claude-code',