@vibe-cafe/vibe-usage 0.1.8 → 0.1.10

package/package.json

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

package/src/init.js

@@ -62,7 +62,7 @@ export async function runInit() {
   const config = {
     apiKey,
     apiUrl,
-    lastSync: null,
+    lastSync: existing?.lastSync || null,
   };
   saveConfig(config);
 

package/src/parsers/codex.js

@@ -1,23 +1,39 @@
 import { readdirSync, readFileSync, statSync, existsSync } from 'node:fs';
-import { join, basename } from 'node:path';
+import { join } from 'node:path';
 import { homedir } from 'node:os';
 import { aggregateToBuckets } from './index.js';
 
 const SESSIONS_DIR = join(homedir(), '.codex', 'sessions');
 
-export async function parse(lastSync) {
-  if (!existsSync(SESSIONS_DIR)) return [];
-
-  const entries = [];
-  let files;
+/**
+ * Recursively find all .jsonl files under a directory.
+ * Codex CLI stores sessions as: ~/.codex/sessions/YYYY/MM/DD/rollout-*.jsonl
+ */
+function findJsonlFiles(dir) {
+  const results = [];
+  if (!existsSync(dir)) return results;
   try {
-    files = readdirSync(SESSIONS_DIR).filter(f => f.endsWith('.jsonl'));
+    for (const entry of readdirSync(dir, { withFileTypes: true })) {
+      const fullPath = join(dir, entry.name);
+      if (entry.isDirectory()) {
+        results.push(...findJsonlFiles(fullPath));
+      } else if (entry.name.endsWith('.jsonl')) {
+        results.push(fullPath);
+      }
+    }
   } catch {
-    return [];
+    // ignore unreadable directories
   }
+  return results;
+}
 
-  for (const file of files) {
-    const filePath = join(SESSIONS_DIR, file);
+export async function parse(lastSync) {
+  if (!existsSync(SESSIONS_DIR)) return [];
+
+  const entries = [];
+  const files = findJsonlFiles(SESSIONS_DIR);
+  if (files.length === 0) return [];
+  for (const filePath of files) {
     if (lastSync) {
       try {
         const stat = statSync(filePath);
@@ -27,8 +43,6 @@ export async function parse(lastSync) {
       }
     }
 
-    const project = basename(file, '.jsonl');
-
     let content;
     try {
       content = readFileSync(filePath, 'utf-8');
@@ -36,9 +50,30 @@ export async function parse(lastSync) {
       continue;
     }
 
+    // Extract project name and model from session_meta line
+    let sessionProject = 'unknown';
+    let sessionModel = 'unknown';
+    for (const line of content.split('\n')) {
+      if (!line.trim()) continue;
+      try {
+        const obj = JSON.parse(line);
+        if (obj.type === 'session_meta' && obj.payload) {
+          const meta = obj.payload;
+          if (meta.cwd) {
+            sessionProject = meta.cwd.split('/').pop() || 'unknown';
+          }
+          if (meta.git?.repository_url) {
+            // e.g. https://github.com/org/repo.git → org/repo
+            const match = meta.git.repository_url.match(/([^/]+\/[^/]+?)(?:\.git)?$/);
+            if (match) sessionProject = match[1];
+          }
+          break;
+        }
+      } catch { break; }
+    }
+
     // Track previous cumulative totals per model to compute deltas when only total_token_usage is available
     const prevTotal = new Map();
-
     for (const line of content.split('\n')) {
       if (!line.trim()) continue;
       try {
@@ -78,12 +113,12 @@ export async function parse(lastSync) {
         }
         if (!usage) continue;
 
-        const model = info.model || payload.model || 'unknown';
+        const model = info.model || payload.model || sessionModel;
 
         entries.push({
           source: 'codex',
           model,
-          project,
+          project: sessionProject,
           timestamp,
           inputTokens: usage.input_tokens || 0,
           outputTokens: usage.output_tokens || 0,

package/src/parsers/gemini-cli.js

@@ -55,8 +55,11 @@ export async function parse(lastSync) {
 
     const messages = data.messages || data.history || [];
     for (const msg of messages) {
+      // New format: tokens on type=gemini messages (ChatRecordingService)
+      // Old format: usage/usageMetadata on any message
+      const tokens = msg.tokens;
       const usage = msg.usage || msg.usageMetadata || msg.token_count;
-      if (!usage) continue;
+      if (!tokens && !usage) continue;
 
       const timestamp = msg.timestamp || msg.createTime || data.createTime;
       if (!timestamp) continue;
@@ -64,16 +67,31 @@ export async function parse(lastSync) {
       if (isNaN(ts.getTime())) continue;
       if (lastSync && ts <= new Date(lastSync)) continue;
 
-      entries.push({
-        source: 'gemini-cli',
-        model: msg.model || data.model || 'unknown',
-        project: 'unknown',
-        timestamp: ts,
-        inputTokens: usage.promptTokenCount || usage.input_tokens || 0,
-        outputTokens: usage.candidatesTokenCount || usage.output_tokens || 0,
-        cachedInputTokens: usage.cachedContentTokenCount || 0,
-        reasoningOutputTokens: usage.thoughtsTokenCount || 0,
-      });
+      if (tokens) {
+        // New format: { input, output, cached, thoughts, tool, total }
+        entries.push({
+          source: 'gemini-cli',
+          model: msg.model || data.model || 'unknown',
+          project: 'unknown',
+          timestamp: ts,
+          inputTokens: tokens.input || 0,
+          outputTokens: tokens.output || 0,
+          cachedInputTokens: tokens.cached || 0,
+          reasoningOutputTokens: tokens.thoughts || 0,
+        });
+      } else {
+        // Old format: { promptTokenCount, candidatesTokenCount, ... }
+        entries.push({
+          source: 'gemini-cli',
+          model: msg.model || data.model || 'unknown',
+          project: 'unknown',
+          timestamp: ts,
+          inputTokens: usage.promptTokenCount || usage.input_tokens || 0,
+          outputTokens: usage.candidatesTokenCount || usage.output_tokens || 0,
+          cachedInputTokens: usage.cachedContentTokenCount || 0,
+          reasoningOutputTokens: usage.thoughtsTokenCount || 0,
+        });
+      }
     }
   }
 

package/src/sync.js

@@ -1,10 +1,14 @@
 import { loadConfig, saveConfig } from './config.js';
 import { ingest } from './api.js';
 import { parsers } from './parsers/index.js';
+import { TOOLS } from './hooks.js';
 
 const BATCH_SIZE = 500;
 
 export async function runSync() {
+  // Self-heal: re-inject any missing hooks before syncing
+  ensureHooks();
+
   const config = loadConfig();
   if (!config?.apiKey) {
     console.error('Not configured. Run `npx @vibe-cafe/vibe-usage init` first.');
@@ -73,3 +77,21 @@ export async function runSync() {
     process.exit(1);
   }
 }
+
+/**
+ * Re-inject hooks for any installed tool whose hook is missing.
+ * Runs silently — meant as a self-healing side effect of sync.
+ */
+function ensureHooks() {
+  for (const tool of TOOLS) {
+    if (!tool.inject) continue;
+    try {
+      const result = tool.inject();
+      if (result.injected) {
+        process.stderr.write(`hook: re-installed ${tool.name} hook\n`);
+      }
+    } catch {
+      // ignore — best effort
+    }
+  }
+}