@tokscale/cli 1.0.0

package/dist/cursor.js

@@ -0,0 +1,432 @@
+/**
+ * Cursor IDE API Client
+ * Fetches usage data from Cursor's dashboard API via CSV export
+ *
+ * API Endpoint: https://cursor.com/api/dashboard/export-usage-events-csv?strategy=tokens
+ * Authentication: WorkosCursorSessionToken cookie
+ *
+ * CSV Format:
+ * Date,Model,Input (w/ Cache Write),Input (w/o Cache Write),Cache Read,Output Tokens,Total Tokens,Cost,Cost to you
+ */
+import * as fs from "node:fs";
+import * as path from "node:path";
+import * as os from "node:os";
+import { parse as parseCsv } from "csv-parse/sync";
+// ============================================================================
+// Credential Management
+// ============================================================================
+const OLD_CONFIG_DIR = path.join(os.homedir(), ".tokscale");
+const CONFIG_DIR = path.join(os.homedir(), ".config", "tokscale");
+const OLD_CURSOR_CREDENTIALS_FILE = path.join(OLD_CONFIG_DIR, "cursor-credentials.json");
+const CURSOR_CREDENTIALS_FILE = path.join(CONFIG_DIR, "cursor-credentials.json");
+function ensureConfigDir() {
+    if (!fs.existsSync(CONFIG_DIR)) {
+        fs.mkdirSync(CONFIG_DIR, { recursive: true, mode: 0o700 });
+    }
+}
+/**
+ * Migrate Cursor credentials and cache from old path to new XDG path
+ */
+function migrateCursorFromOldPath() {
+    try {
+        // Migrate cursor credentials
+        if (!fs.existsSync(CURSOR_CREDENTIALS_FILE) && fs.existsSync(OLD_CURSOR_CREDENTIALS_FILE)) {
+            ensureConfigDir();
+            fs.copyFileSync(OLD_CURSOR_CREDENTIALS_FILE, CURSOR_CREDENTIALS_FILE);
+            fs.chmodSync(CURSOR_CREDENTIALS_FILE, 0o600);
+            fs.unlinkSync(OLD_CURSOR_CREDENTIALS_FILE);
+        }
+        // Migrate cache directory (handled after CURSOR_CACHE_DIR is defined)
+        // Cache migration happens in migrateCursorCacheFromOldPath()
+        // Try to remove old config directory if empty
+        try {
+            fs.rmdirSync(OLD_CONFIG_DIR);
+        }
+        catch {
+            // Directory not empty - ignore
+        }
+    }
+    catch {
+        // Migration failed - continue with normal operation
+    }
+}
+export function saveCursorCredentials(credentials) {
+    ensureConfigDir();
+    fs.writeFileSync(CURSOR_CREDENTIALS_FILE, JSON.stringify(credentials, null, 2), {
+        encoding: "utf-8",
+        mode: 0o600,
+    });
+}
+export function loadCursorCredentials() {
+    migrateCursorFromOldPath();
+    try {
+        if (!fs.existsSync(CURSOR_CREDENTIALS_FILE)) {
+            return null;
+        }
+        const data = fs.readFileSync(CURSOR_CREDENTIALS_FILE, "utf-8");
+        const parsed = JSON.parse(data);
+        if (!parsed.sessionToken) {
+            return null;
+        }
+        return parsed;
+    }
+    catch {
+        return null;
+    }
+}
+export function clearCursorCredentials() {
+    try {
+        if (fs.existsSync(CURSOR_CREDENTIALS_FILE)) {
+            fs.unlinkSync(CURSOR_CREDENTIALS_FILE);
+            return true;
+        }
+        return false;
+    }
+    catch {
+        return false;
+    }
+}
+export function isCursorLoggedIn() {
+    return loadCursorCredentials() !== null;
+}
+// ============================================================================
+// API Client
+// ============================================================================
+const CURSOR_API_BASE = "https://cursor.com";
+const USAGE_CSV_ENDPOINT = `${CURSOR_API_BASE}/api/dashboard/export-usage-events-csv?strategy=tokens`;
+const USAGE_SUMMARY_ENDPOINT = `${CURSOR_API_BASE}/api/usage-summary`;
+/**
+ * Build HTTP headers for Cursor API requests
+ */
+function buildCursorHeaders(sessionToken) {
+    return {
+        Accept: "*/*",
+        "Accept-Language": "en-US,en;q=0.9",
+        Cookie: `WorkosCursorSessionToken=${sessionToken}`,
+        Referer: "https://www.cursor.com/settings",
+        "User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36",
+    };
+}
+/**
+ * Validate Cursor session token by hitting the usage-summary endpoint
+ */
+export async function validateCursorSession(sessionToken) {
+    try {
+        const response = await fetch(USAGE_SUMMARY_ENDPOINT, {
+            method: "GET",
+            headers: buildCursorHeaders(sessionToken),
+        });
+        if (response.status === 401 || response.status === 403) {
+            return { valid: false, error: "Session token expired or invalid" };
+        }
+        if (!response.ok) {
+            return { valid: false, error: `API returned status ${response.status}` };
+        }
+        const data = await response.json();
+        // Check for required fields that indicate valid auth
+        if (data.billingCycleStart && data.billingCycleEnd) {
+            return { valid: true, membershipType: data.membershipType };
+        }
+        return { valid: false, error: "Invalid response format" };
+    }
+    catch (error) {
+        return { valid: false, error: error.message };
+    }
+}
+/**
+ * Fetch usage CSV from Cursor API
+ */
+export async function fetchCursorUsageCsv(sessionToken) {
+    const response = await fetch(USAGE_CSV_ENDPOINT, {
+        method: "GET",
+        headers: buildCursorHeaders(sessionToken),
+    });
+    if (response.status === 401 || response.status === 403) {
+        throw new Error("Cursor session expired. Please run 'tokscale cursor login' to re-authenticate.");
+    }
+    if (!response.ok) {
+        throw new Error(`Cursor API returned status ${response.status}`);
+    }
+    const text = await response.text();
+    // Validate it's actually CSV (handle both old and new formats)
+    // Old: "Date,Model,..."
+    // New: "Date,Kind,Model,..."
+    if (!text.startsWith("Date,")) {
+        throw new Error("Invalid response from Cursor API - expected CSV format");
+    }
+    return text;
+}
+// ============================================================================
+// CSV Parsing
+// ============================================================================
+/**
+ * Parse cost string (e.g., "$0.50" or "0.50") to number
+ */
+function parseCost(costStr) {
+    if (!costStr)
+        return 0;
+    const cleaned = costStr.replace(/[$,]/g, "").trim();
+    const value = parseFloat(cleaned);
+    return isNaN(value) ? 0 : value;
+}
+/**
+ * Infer provider from model name
+ */
+function inferProvider(model) {
+    const lowerModel = model.toLowerCase();
+    if (lowerModel.includes("claude") || lowerModel.includes("sonnet") || lowerModel.includes("opus") || lowerModel.includes("haiku")) {
+        return "anthropic";
+    }
+    if (lowerModel.includes("gpt") || lowerModel.includes("o1") || lowerModel.includes("o3")) {
+        return "openai";
+    }
+    if (lowerModel.includes("gemini")) {
+        return "google";
+    }
+    if (lowerModel.includes("deepseek")) {
+        return "deepseek";
+    }
+    if (lowerModel.includes("llama") || lowerModel.includes("mixtral")) {
+        return "meta";
+    }
+    return "cursor"; // Default provider
+}
+/**
+ * Parse Cursor usage CSV into structured rows
+ */
+export function parseCursorCsv(csvText) {
+    try {
+        const records = parseCsv(csvText, {
+            columns: true,
+            skip_empty_lines: true,
+            trim: true,
+            relax_column_count: true,
+        });
+        return records
+            .filter((record) => record["Date"] && record["Model"])
+            .map((record) => {
+            const dateStr = record["Date"] || "";
+            const date = new Date(dateStr);
+            const isValidDate = !isNaN(date.getTime());
+            const dateOnly = isValidDate
+                ? date.toISOString().slice(0, 10)
+                : dateStr.length >= 10
+                    ? dateStr.slice(0, 10)
+                    : dateStr;
+            return {
+                date: dateOnly,
+                timestamp: isValidDate ? date.getTime() : 0,
+                model: (record["Model"] || "").trim(),
+                inputWithCacheWrite: parseInt(record["Input (w/ Cache Write)"] || "0", 10),
+                inputWithoutCacheWrite: parseInt(record["Input (w/o Cache Write)"] || "0", 10),
+                cacheRead: parseInt(record["Cache Read"] || "0", 10),
+                outputTokens: parseInt(record["Output Tokens"] || "0", 10),
+                totalTokens: parseInt(record["Total Tokens"] || "0", 10),
+                apiCost: parseCost(record["Cost"] || record["API Cost"] || "0"),
+                costToYou: parseCost(record["Cost to you"] || "0"),
+            };
+        });
+    }
+    catch (error) {
+        throw new Error(`Failed to parse Cursor CSV: ${error.message}`);
+    }
+}
+// ============================================================================
+// Data Aggregation (for table display)
+// ============================================================================
+/**
+ * Aggregate Cursor usage by model
+ */
+export function aggregateCursorByModel(rows) {
+    const modelMap = new Map();
+    for (const row of rows) {
+        const key = row.model;
+        const existing = modelMap.get(key);
+        // Cache write = inputWithCacheWrite - inputWithoutCacheWrite (tokens written to cache)
+        const cacheWrite = Math.max(0, row.inputWithCacheWrite - row.inputWithoutCacheWrite);
+        // Input tokens (without cache) = inputWithoutCacheWrite
+        const input = row.inputWithoutCacheWrite;
+        if (existing) {
+            existing.messageCount += 1;
+            existing.input += input;
+            existing.output += row.outputTokens;
+            existing.cacheRead += row.cacheRead;
+            existing.cacheWrite += cacheWrite;
+            existing.cost += row.costToYou || row.apiCost;
+        }
+        else {
+            modelMap.set(key, {
+                source: "cursor",
+                model: row.model,
+                providerId: inferProvider(row.model),
+                messageCount: 1,
+                input,
+                output: row.outputTokens,
+                cacheRead: row.cacheRead,
+                cacheWrite,
+                reasoning: 0, // Cursor doesn't expose reasoning tokens
+                cost: row.costToYou || row.apiCost,
+            });
+        }
+    }
+    return Array.from(modelMap.values()).sort((a, b) => b.cost - a.cost);
+}
+// ============================================================================
+// Data Conversion (for graph/native module)
+// ============================================================================
+/**
+ * Convert Cursor CSV rows to timestamped messages for graph generation
+ */
+export function cursorRowsToMessages(rows) {
+    return rows.map((row) => {
+        const cacheWrite = Math.max(0, row.inputWithCacheWrite - row.inputWithoutCacheWrite);
+        const input = row.inputWithoutCacheWrite;
+        return {
+            source: "cursor",
+            model: row.model,
+            providerId: inferProvider(row.model),
+            timestamp: row.timestamp,
+            input,
+            output: row.outputTokens,
+            cacheRead: row.cacheRead,
+            cacheWrite,
+            reasoning: 0,
+            cost: row.costToYou || row.apiCost,
+        };
+    });
+}
+// ============================================================================
+// High-Level API
+// ============================================================================
+/**
+ * Fetch and parse Cursor usage data
+ * Requires valid credentials to be stored
+ */
+export async function readCursorUsage() {
+    const credentials = loadCursorCredentials();
+    if (!credentials) {
+        throw new Error("Cursor not authenticated. Run 'tokscale cursor login' first.");
+    }
+    const csvText = await fetchCursorUsageCsv(credentials.sessionToken);
+    const rows = parseCursorCsv(csvText);
+    const byModel = aggregateCursorByModel(rows);
+    const messages = cursorRowsToMessages(rows);
+    return { rows, byModel, messages };
+}
+/**
+ * Get Cursor credentials file path (for debugging)
+ */
+export function getCursorCredentialsPath() {
+    return CURSOR_CREDENTIALS_FILE;
+}
+// ============================================================================
+// Cache Management (for Rust integration)
+// ============================================================================
+const OLD_CURSOR_CACHE_DIR = path.join(os.homedir(), ".tokscale", "cursor-cache");
+const CURSOR_CACHE_DIR = path.join(CONFIG_DIR, "cursor-cache");
+const CURSOR_CACHE_FILE = path.join(CURSOR_CACHE_DIR, "usage.csv");
+function ensureCacheDir() {
+    if (!fs.existsSync(CURSOR_CACHE_DIR)) {
+        fs.mkdirSync(CURSOR_CACHE_DIR, { recursive: true, mode: 0o700 });
+    }
+}
+/**
+ * Migrate cursor cache from old path to new XDG path
+ */
+function migrateCursorCacheFromOldPath() {
+    try {
+        if (!fs.existsSync(CURSOR_CACHE_DIR) && fs.existsSync(OLD_CURSOR_CACHE_DIR)) {
+            ensureCacheDir();
+            fs.cpSync(OLD_CURSOR_CACHE_DIR, CURSOR_CACHE_DIR, { recursive: true });
+            fs.rmSync(OLD_CURSOR_CACHE_DIR, { recursive: true });
+        }
+        // Try to remove old config directory if empty
+        try {
+            fs.rmdirSync(OLD_CONFIG_DIR);
+        }
+        catch {
+            // Directory not empty - ignore
+        }
+    }
+    catch {
+        // Migration failed - continue with normal operation
+    }
+}
+/**
+ * Sync Cursor usage data from API to local cache
+ * This downloads the CSV and saves it for the Rust module to parse
+ */
+export async function syncCursorCache() {
+    migrateCursorCacheFromOldPath();
+    const credentials = loadCursorCredentials();
+    if (!credentials) {
+        return { synced: false, rows: 0, error: "Not authenticated" };
+    }
+    try {
+        const csvText = await fetchCursorUsageCsv(credentials.sessionToken);
+        ensureCacheDir();
+        fs.writeFileSync(CURSOR_CACHE_FILE, csvText, { encoding: "utf-8", mode: 0o600 });
+        // Count rows for feedback
+        const rows = parseCursorCsv(csvText);
+        return { synced: true, rows: rows.length };
+    }
+    catch (error) {
+        return { synced: false, rows: 0, error: error.message };
+    }
+}
+/**
+ * Get the cache file path
+ */
+export function getCursorCachePath() {
+    return CURSOR_CACHE_FILE;
+}
+/**
+ * Check if cache exists and when it was last updated
+ */
+export function getCursorCacheStatus() {
+    const exists = fs.existsSync(CURSOR_CACHE_FILE);
+    let lastModified;
+    if (exists) {
+        try {
+            const stats = fs.statSync(CURSOR_CACHE_FILE);
+            lastModified = stats.mtime;
+        }
+        catch {
+            // Ignore stat errors
+        }
+    }
+    return { exists, lastModified, path: CURSOR_CACHE_FILE };
+}
+export function readCursorMessagesFromCache() {
+    if (!fs.existsSync(CURSOR_CACHE_FILE)) {
+        return [];
+    }
+    try {
+        const csvText = fs.readFileSync(CURSOR_CACHE_FILE, "utf-8");
+        const rows = parseCursorCsv(csvText);
+        return rows.map((row) => {
+            const cacheWrite = Math.max(0, row.inputWithCacheWrite - row.inputWithoutCacheWrite);
+            const input = row.inputWithoutCacheWrite;
+            return {
+                source: "cursor",
+                modelId: row.model,
+                providerId: inferProvider(row.model),
+                sessionId: `cursor-${row.date}-${row.model}`,
+                timestamp: row.timestamp,
+                date: row.date,
+                tokens: {
+                    input,
+                    output: row.outputTokens,
+                    cacheRead: row.cacheRead,
+                    cacheWrite,
+                    reasoning: 0,
+                },
+                cost: row.costToYou || row.apiCost,
+            };
+        });
+    }
+    catch {
+        return [];
+    }
+}
+//# sourceMappingURL=cursor.js.map

package/dist/cursor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cursor.js","sourceRoot":"","sources":["../src/cursor.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAEH,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAClC,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,EAAE,KAAK,IAAI,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAoDnD,+EAA+E;AAC/E,wBAAwB;AACxB,+EAA+E;AAE/E,MAAM,cAAc,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,WAAW,CAAC,CAAC;AAC5D,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,UAAU,CAAC,CAAC;AAClE,MAAM,2BAA2B,GAAG,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,yBAAyB,CAAC,CAAC;AACzF,MAAM,uBAAuB,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,yBAAyB,CAAC,CAAC;AAEjF,SAAS,eAAe;IACtB,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;QAC/B,EAAE,CAAC,SAAS,CAAC,UAAU,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;IAC7D,CAAC;AACH,CAAC;AAED;;GAEG;AACH,SAAS,wBAAwB;IAC/B,IAAI,CAAC;QACH,6BAA6B;QAC7B,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,uBAAuB,CAAC,IAAI,EAAE,CAAC,UAAU,CAAC,2BAA2B,CAAC,EAAE,CAAC;YAC1F,eAAe,EAAE,CAAC;YAClB,EAAE,CAAC,YAAY,CAAC,2BAA2B,EAAE,uBAAuB,CAAC,CAAC;YACtE,EAAE,CAAC,SAAS,CAAC,uBAAuB,EAAE,KAAK,CAAC,CAAC;YAC7C,EAAE,CAAC,UAAU,CAAC,2BAA2B,CAAC,CAAC;QAC7C,CAAC;QAED,sEAAsE;QACtE,6DAA6D;QAE7D,8CAA8C;QAC9C,IAAI,CAAC;YACH,EAAE,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;QAC/B,CAAC;QAAC,MAAM,CAAC;YACP,+BAA+B;QACjC,CAAC;IACH,CAAC;IAAC,MAAM,CAAC;QACP,oDAAoD;IACtD,CAAC;AACH,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,WAA8B;IAClE,eAAe,EAAE,CAAC;IAClB,EAAE,CAAC,aAAa,CAAC,uBAAuB,EAAE,IAAI,CAAC,SAAS,CAAC,WAAW,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE;QAC9E,QAAQ,EAAE,OAAO;QACjB,IAAI,EAAE,KAAK;KACZ,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,qBAAqB;IACnC,wBAAwB,EAAE,CAAC;IAC3B,IAAI,CAAC;QACH,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,uBAAuB,CAAC,EAAE,CAAC;YAC5C,OAAO,IAAI,CAAC;QACd,CAAC;QACD,MAAM,IAAI,GAAG,EAAE,CAAC,YAAY,CAAC,uBAAuB,EAAE,OAAO,CAAC,CAAC;QAC/D,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAEhC,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC;YACzB,OAAO,IAAI,CAAC;QACd,CAAC;QAED,OAAO,MAA2B,CAAC;IACrC,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAED,MAAM,UAAU,sBAAsB;IACpC,IAAI,CAAC;QACH,IAAI,EAAE,CAAC,UAAU,CAAC,uBAAuB,CAAC,EAAE,CAAC;YAC3C,EAAE,CAAC,UAAU,CAAC,uBAAuB,CAAC,CAAC;YACvC,OAAO,IAAI,CAAC;QACd,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC;AAED,MAAM,UAAU,gBAAgB;IAC9B,OAAO,qBAAqB,EAAE,KAAK,IAAI,CAAC;AAC1C,CAAC;AAED,+EAA+E;AAC/E,aAAa;AACb,+EAA+E;AAE/E,MAAM,eAAe,GAAG,oBAAoB,CAAC;AAC7C,MAAM,kBAAkB,GAAG,GAAG,eAAe,wDAAwD,CAAC;AACtG,MAAM,sBAAsB,GAAG,GAAG,eAAe,oBAAoB,CAAC;AAEtE;;GAEG;AACH,SAAS,kBAAkB,CAAC,YAAoB;IAC9C,OAAO;QACL,MAAM,EAAE,KAAK;QACb,iBAAiB,EAAE,gBAAgB;QACnC,MAAM,EAAE,4BAA4B,YAAY,EAAE;QAClD,OAAO,EAAE,iCAAiC;QAC1C,YAAY,EACV,uHAAuH;KAC1H,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,qBAAqB,CACzC,YAAoB;IAEpB,IAAI,CAAC;QACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,sBAAsB,EAAE;YACnD,MAAM,EAAE,KAAK;YACb,OAAO,EAAE,kBAAkB,CAAC,YAAY,CAAC;SAC1C,CAAC,CAAC;QAEH,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;YACvD,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,kCAAkC,EAAE,CAAC;QACrE,CAAC;QAED,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,uBAAuB,QAAQ,CAAC,MAAM,EAAE,EAAE,CAAC;QAC3E,CAAC;QAED,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QAEnC,qDAAqD;QACrD,IAAI,IAAI,CAAC,iBAAiB,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;YACnD,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,cAAc,EAAE,IAAI,CAAC,cAAc,EAAE,CAAC;QAC9D,CAAC;QAED,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,yBAAyB,EAAE,CAAC;IAC5D,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAG,KAAe,CAAC,OAAO,EAAE,CAAC;IAC3D,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,mBAAmB,CAAC,YAAoB;IAC5D,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,kBAAkB,EAAE;QAC/C,MAAM,EAAE,KAAK;QACb,OAAO,EAAE,kBAAkB,CAAC,YAAY,CAAC;KAC1C,CAAC,CAAC;IAEH,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;QACvD,MAAM,IAAI,KAAK,CAAC,gFAAgF,CAAC,CAAC;IACpG,CAAC;IAED,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;QACjB,MAAM,IAAI,KAAK,CAAC,8BAA8B,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;IACnE,CAAC;IAED,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;IAEnC,+DAA+D;IAC/D,wBAAwB;IACxB,6BAA6B;IAC7B,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;QAC9B,MAAM,IAAI,KAAK,CAAC,wDAAwD,CAAC,CAAC;IAC5E,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED,+EAA+E;AAC/E,cAAc;AACd,+EAA+E;AAE/E;;GAEG;AACH,SAAS,SAAS,CAAC,OAAe;IAChC,IAAI,CAAC,OAAO;QAAE,OAAO,CAAC,CAAC;IACvB,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;IACpD,MAAM,KAAK,GAAG,UAAU,CAAC,OAAO,CAAC,CAAC;IAClC,OAAO,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;AAClC,CAAC;AAED;;GAEG;AACH,SAAS,aAAa,CAAC,KAAa;IAClC,MAAM,UAAU,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC;IAEvC,IAAI,UAAU,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,UAAU,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,UAAU,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,UAAU,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;QAClI,OAAO,WAAW,CAAC;IACrB,CAAC;IACD,IAAI,UAAU,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,UAAU,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,UAAU,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;QACzF,OAAO,QAAQ,CAAC;IAClB,CAAC;IACD,IAAI,UAAU,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;QAClC,OAAO,QAAQ,CAAC;IAClB,CAAC;IACD,IAAI,UAAU,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;QACpC,OAAO,UAAU,CAAC;IACpB,CAAC;IACD,IAAI,UAAU,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,UAAU,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;QACnE,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,OAAO,QAAQ,CAAC,CAAC,mBAAmB;AACtC,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,cAAc,CAAC,OAAe;IAC5C,IAAI,CAAC;QACH,MAAM,OAAO,GAAkC,QAAQ,CAAC,OAAO,EAAE;YAC/D,OAAO,EAAE,IAAI;YACb,gBAAgB,EAAE,IAAI;YACtB,IAAI,EAAE,IAAI;YACV,kBAAkB,EAAE,IAAI;SACzB,CAAC,CAAC;QAEH,OAAO,OAAO;aACX,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,CAAC;aACrD,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE;YACd,MAAM,OAAO,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;YACrC,MAAM,IAAI,GAAG,IAAI,IAAI,CAAC,OAAO,CAAC,CAAC;YAC/B,MAAM,WAAW,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;YAC3C,MAAM,QAAQ,GAAG,WAAW;gBAC1B,CAAC,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC;gBACjC,CAAC,CAAC,OAAO,CAAC,MAAM,IAAI,EAAE;oBACpB,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC;oBACtB,CAAC,CAAC,OAAO,CAAC;YAEd,OAAO;gBACL,IAAI,EAAE,QAAQ;gBACd,SAAS,EAAE,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;gBAC3C,KAAK,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE;gBACrC,mBAAmB,EAAE,QAAQ,CAAC,MAAM,CAAC,wBAAwB,CAAC,IAAI,GAAG,EAAE,EAAE,CAAC;gBAC1E,sBAAsB,EAAE,QAAQ,CAAC,MAAM,CAAC,yBAAyB,CAAC,IAAI,GAAG,EAAE,EAAE,CAAC;gBAC9E,SAAS,EAAE,QAAQ,CAAC,MAAM,CAAC,YAAY,CAAC,IAAI,GAAG,EAAE,EAAE,CAAC;gBACpD,YAAY,EAAE,QAAQ,CAAC,MAAM,CAAC,eAAe,CAAC,IAAI,GAAG,EAAE,EAAE,CAAC;gBAC1D,WAAW,EAAE,QAAQ,CAAC,MAAM,CAAC,cAAc,CAAC,IAAI,GAAG,EAAE,EAAE,CAAC;gBACxD,OAAO,EAAE,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,MAAM,CAAC,UAAU,CAAC,IAAI,GAAG,CAAC;gBAC/D,SAAS,EAAE,SAAS,CAAC,MAAM,CAAC,aAAa,CAAC,IAAI,GAAG,CAAC;aACnD,CAAC;QACJ,CAAC,CAAC,CAAC;IACP,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,IAAI,KAAK,CAAC,+BAAgC,KAAe,CAAC,OAAO,EAAE,CAAC,CAAC;IAC7E,CAAC;AACH,CAAC;AAED,+EAA+E;AAC/E,uCAAuC;AACvC,+EAA+E;AAE/E;;GAEG;AACH,MAAM,UAAU,sBAAsB,CAAC,IAAsB;IAC3D,MAAM,QAAQ,GAAG,IAAI,GAAG,EAA2B,CAAC;IAEpD,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,MAAM,GAAG,GAAG,GAAG,CAAC,KAAK,CAAC;QACtB,MAAM,QAAQ,GAAG,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAEnC,uFAAuF;QACvF,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,GAAG,CAAC,mBAAmB,GAAG,GAAG,CAAC,sBAAsB,CAAC,CAAC;QACrF,wDAAwD;QACxD,MAAM,KAAK,GAAG,GAAG,CAAC,sBAAsB,CAAC;QAEzC,IAAI,QAAQ,EAAE,CAAC;YACb,QAAQ,CAAC,YAAY,IAAI,CAAC,CAAC;YAC3B,QAAQ,CAAC,KAAK,IAAI,KAAK,CAAC;YACxB,QAAQ,CAAC,MAAM,IAAI,GAAG,CAAC,YAAY,CAAC;YACpC,QAAQ,CAAC,SAAS,IAAI,GAAG,CAAC,SAAS,CAAC;YACpC,QAAQ,CAAC,UAAU,IAAI,UAAU,CAAC;YAClC,QAAQ,CAAC,IAAI,IAAI,GAAG,CAAC,SAAS,IAAI,GAAG,CAAC,OAAO,CAAC;QAChD,CAAC;aAAM,CAAC;YACN,QAAQ,CAAC,GAAG,CAAC,GAAG,EAAE;gBAChB,MAAM,EAAE,QAAQ;gBAChB,KAAK,EAAE,GAAG,CAAC,KAAK;gBAChB,UAAU,EAAE,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC;gBACpC,YAAY,EAAE,CAAC;gBACf,KAAK;gBACL,MAAM,EAAE,GAAG,CAAC,YAAY;gBACxB,SAAS,EAAE,GAAG,CAAC,SAAS;gBACxB,UAAU;gBACV,SAAS,EAAE,CAAC,EAAE,yCAAyC;gBACvD,IAAI,EAAE,GAAG,CAAC,SAAS,IAAI,GAAG,CAAC,OAAO;aACnC,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC;AACvE,CAAC;AAED,+EAA+E;AAC/E,4CAA4C;AAC5C,+EAA+E;AAE/E;;GAEG;AACH,MAAM,UAAU,oBAAoB,CAAC,IAAsB;IACzD,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;QACtB,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,GAAG,CAAC,mBAAmB,GAAG,GAAG,CAAC,sBAAsB,CAAC,CAAC;QACrF,MAAM,KAAK,GAAG,GAAG,CAAC,sBAAsB,CAAC;QAEzC,OAAO;YACL,MAAM,EAAE,QAAiB;YACzB,KAAK,EAAE,GAAG,CAAC,KAAK;YAChB,UAAU,EAAE,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC;YACpC,SAAS,EAAE,GAAG,CAAC,SAAS;YACxB,KAAK;YACL,MAAM,EAAE,GAAG,CAAC,YAAY;YACxB,SAAS,EAAE,GAAG,CAAC,SAAS;YACxB,UAAU;YACV,SAAS,EAAE,CAAC;YACZ,IAAI,EAAE,GAAG,CAAC,SAAS,IAAI,GAAG,CAAC,OAAO;SACnC,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC;AAED,+EAA+E;AAC/E,iBAAiB;AACjB,+EAA+E;AAE/E;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,eAAe;IAKnC,MAAM,WAAW,GAAG,qBAAqB,EAAE,CAAC;IAC5C,IAAI,CAAC,WAAW,EAAE,CAAC;QACjB,MAAM,IAAI,KAAK,CAAC,8DAA8D,CAAC,CAAC;IAClF,CAAC;IAED,MAAM,OAAO,GAAG,MAAM,mBAAmB,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC;IACpE,MAAM,IAAI,GAAG,cAAc,CAAC,OAAO,CAAC,CAAC;IACrC,MAAM,OAAO,GAAG,sBAAsB,CAAC,IAAI,CAAC,CAAC;IAC7C,MAAM,QAAQ,GAAG,oBAAoB,CAAC,IAAI,CAAC,CAAC;IAE5C,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC;AACrC,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,wBAAwB;IACtC,OAAO,uBAAuB,CAAC;AACjC,CAAC;AAED,+EAA+E;AAC/E,0CAA0C;AAC1C,+EAA+E;AAE/E,MAAM,oBAAoB,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,WAAW,EAAE,cAAc,CAAC,CAAC;AAClF,MAAM,gBAAgB,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,cAAc,CAAC,CAAC;AAC/D,MAAM,iBAAiB,GAAG,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,WAAW,CAAC,CAAC;AAEnE,SAAS,cAAc;IACrB,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,gBAAgB,CAAC,EAAE,CAAC;QACrC,EAAE,CAAC,SAAS,CAAC,gBAAgB,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;IACnE,CAAC;AACH,CAAC;AAED;;GAEG;AACH,SAAS,6BAA6B;IACpC,IAAI,CAAC;QACH,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,gBAAgB,CAAC,IAAI,EAAE,CAAC,UAAU,CAAC,oBAAoB,CAAC,EAAE,CAAC;YAC5E,cAAc,EAAE,CAAC;YACjB,EAAE,CAAC,MAAM,CAAC,oBAAoB,EAAE,gBAAgB,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YACvE,EAAE,CAAC,MAAM,CAAC,oBAAoB,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACvD,CAAC;QAED,8CAA8C;QAC9C,IAAI,CAAC;YACH,EAAE,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;QAC/B,CAAC;QAAC,MAAM,CAAC;YACP,+BAA+B;QACjC,CAAC;IACH,CAAC;IAAC,MAAM,CAAC;QACP,oDAAoD;IACtD,CAAC;AACH,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,eAAe;IACnC,6BAA6B,EAAE,CAAC;IAChC,MAAM,WAAW,GAAG,qBAAqB,EAAE,CAAC;IAC5C,IAAI,CAAC,WAAW,EAAE,CAAC;QACjB,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,mBAAmB,EAAE,CAAC;IAChE,CAAC;IAED,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,MAAM,mBAAmB,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC;QACpE,cAAc,EAAE,CAAC;QACjB,EAAE,CAAC,aAAa,CAAC,iBAAiB,EAAE,OAAO,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;QAEjF,0BAA0B;QAC1B,MAAM,IAAI,GAAG,cAAc,CAAC,OAAO,CAAC,CAAC;QACrC,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC;IAC7C,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,EAAE,KAAK,EAAG,KAAe,CAAC,OAAO,EAAE,CAAC;IACrE,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,kBAAkB;IAChC,OAAO,iBAAiB,CAAC;AAC3B,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,oBAAoB;IAClC,MAAM,MAAM,GAAG,EAAE,CAAC,UAAU,CAAC,iBAAiB,CAAC,CAAC;IAChD,IAAI,YAA8B,CAAC;IAEnC,IAAI,MAAM,EAAE,CAAC;QACX,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,EAAE,CAAC,QAAQ,CAAC,iBAAiB,CAAC,CAAC;YAC7C,YAAY,GAAG,KAAK,CAAC,KAAK,CAAC;QAC7B,CAAC;QAAC,MAAM,CAAC;YACP,qBAAqB;QACvB,CAAC;IACH,CAAC;IAED,OAAO,EAAE,MAAM,EAAE,YAAY,EAAE,IAAI,EAAE,iBAAiB,EAAE,CAAC;AAC3D,CAAC;AAmBD,MAAM,UAAU,2BAA2B;IACzC,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,iBAAiB,CAAC,EAAE,CAAC;QACtC,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,iBAAiB,EAAE,OAAO,CAAC,CAAC;QAC5D,MAAM,IAAI,GAAG,cAAc,CAAC,OAAO,CAAC,CAAC;QAErC,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;YACtB,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,GAAG,CAAC,mBAAmB,GAAG,GAAG,CAAC,sBAAsB,CAAC,CAAC;YACrF,MAAM,KAAK,GAAG,GAAG,CAAC,sBAAsB,CAAC;YAEzC,OAAO;gBACL,MAAM,EAAE,QAAiB;gBACzB,OAAO,EAAE,GAAG,CAAC,KAAK;gBAClB,UAAU,EAAE,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC;gBACpC,SAAS,EAAE,UAAU,GAAG,CAAC,IAAI,IAAI,GAAG,CAAC,KAAK,EAAE;gBAC5C,SAAS,EAAE,GAAG,CAAC,SAAS;gBACxB,IAAI,EAAE,GAAG,CAAC,IAAI;gBACd,MAAM,EAAE;oBACN,KAAK;oBACL,MAAM,EAAE,GAAG,CAAC,YAAY;oBACxB,SAAS,EAAE,GAAG,CAAC,SAAS;oBACxB,UAAU;oBACV,SAAS,EAAE,CAAC;iBACb;gBACD,IAAI,EAAE,GAAG,CAAC,SAAS,IAAI,GAAG,CAAC,OAAO;aACnC,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,CAAC;IACZ,CAAC;AACH,CAAC"}

package/dist/gemini.d.ts

@@ -0,0 +1,36 @@
+/**
+ * Gemini CLI data reader
+ * Reads from ~/.gemini/tmp/{projectHash}/chats/session-*.json
+ */
+export interface GeminiUsageData {
+    source: "gemini";
+    model: string;
+    messageCount: number;
+    input: number;
+    output: number;
+    cached: number;
+    thoughts: number;
+    tool: number;
+}
+/**
+ * Individual Gemini message with timestamp for graph generation
+ */
+export interface GeminiMessageWithTimestamp {
+    source: "gemini";
+    model: string;
+    timestamp: number;
+    tokens: {
+        input: number;
+        output: number;
+        cached: number;
+        thoughts: number;
+        tool: number;
+    };
+}
+export declare function getGeminiBasePath(): string;
+export declare function readGeminiSessions(): GeminiUsageData[];
+/**
+ * Read Gemini messages with timestamp for graph generation
+ */
+export declare function readGeminiMessagesWithTimestamp(): GeminiMessageWithTimestamp[];
+//# sourceMappingURL=gemini.d.ts.map

package/dist/gemini.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"gemini.d.ts","sourceRoot":"","sources":["../src/gemini.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAMH,MAAM,WAAW,eAAe;IAC9B,MAAM,EAAE,QAAQ,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,YAAY,EAAE,MAAM,CAAC;IACrB,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,MAAM,CAAC;CACd;AAED;;GAEG;AACH,MAAM,WAAW,0BAA0B;IACzC,MAAM,EAAE,QAAQ,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE;QACN,KAAK,EAAE,MAAM,CAAC;QACd,MAAM,EAAE,MAAM,CAAC;QACf,MAAM,EAAE,MAAM,CAAC;QACf,QAAQ,EAAE,MAAM,CAAC;QACjB,IAAI,EAAE,MAAM,CAAC;KACd,CAAC;CACH;AA0BD,wBAAgB,iBAAiB,IAAI,MAAM,CAE1C;AAED,wBAAgB,kBAAkB,IAAI,eAAe,EAAE,CAgEtD;AAED;;GAEG;AACH,wBAAgB,+BAA+B,IAAI,0BAA0B,EAAE,CA2D9E"}

package/dist/gemini.js

@@ -0,0 +1,125 @@
+/**
+ * Gemini CLI data reader
+ * Reads from ~/.gemini/tmp/{projectHash}/chats/session-*.json
+ */
+import * as fs from "node:fs";
+import * as path from "node:path";
+import * as os from "node:os";
+export function getGeminiBasePath() {
+    return path.join(os.homedir(), ".gemini");
+}
+export function readGeminiSessions() {
+    const basePath = getGeminiBasePath();
+    const tmpPath = path.join(basePath, "tmp");
+    if (!fs.existsSync(tmpPath)) {
+        return [];
+    }
+    const modelUsage = new Map();
+    // Find all project directories
+    const projectDirs = fs
+        .readdirSync(tmpPath, { withFileTypes: true })
+        .filter((d) => d.isDirectory())
+        .map((d) => path.join(tmpPath, d.name));
+    for (const projectDir of projectDirs) {
+        const chatsDir = path.join(projectDir, "chats");
+        if (!fs.existsSync(chatsDir))
+            continue;
+        // Find all session JSON files
+        const sessionFiles = fs
+            .readdirSync(chatsDir)
+            .filter((f) => f.startsWith("session-") && f.endsWith(".json"));
+        for (const sessionFile of sessionFiles) {
+            try {
+                const content = fs.readFileSync(path.join(chatsDir, sessionFile), "utf-8");
+                const session = JSON.parse(content);
+                for (const msg of session.messages) {
+                    // Only process gemini messages with token data
+                    if (msg.type !== "gemini" || !msg.tokens || !msg.model)
+                        continue;
+                    const model = msg.model;
+                    let usage = modelUsage.get(model);
+                    if (!usage) {
+                        usage = {
+                            source: "gemini",
+                            model,
+                            messageCount: 0,
+                            input: 0,
+                            output: 0,
+                            cached: 0,
+                            thoughts: 0,
+                            tool: 0,
+                        };
+                        modelUsage.set(model, usage);
+                    }
+                    usage.messageCount++;
+                    usage.input += msg.tokens.input || 0;
+                    usage.output += msg.tokens.output || 0;
+                    usage.cached += msg.tokens.cached || 0;
+                    usage.thoughts += msg.tokens.thoughts || 0;
+                    usage.tool += msg.tokens.tool || 0;
+                }
+            }
+            catch {
+                // Skip malformed files
+            }
+        }
+    }
+    return Array.from(modelUsage.values());
+}
+/**
+ * Read Gemini messages with timestamp for graph generation
+ */
+export function readGeminiMessagesWithTimestamp() {
+    const basePath = getGeminiBasePath();
+    const tmpPath = path.join(basePath, "tmp");
+    if (!fs.existsSync(tmpPath)) {
+        return [];
+    }
+    const messages = [];
+    // Find all project directories
+    const projectDirs = fs
+        .readdirSync(tmpPath, { withFileTypes: true })
+        .filter((d) => d.isDirectory())
+        .map((d) => path.join(tmpPath, d.name));
+    for (const projectDir of projectDirs) {
+        const chatsDir = path.join(projectDir, "chats");
+        if (!fs.existsSync(chatsDir))
+            continue;
+        // Find all session JSON files
+        const sessionFiles = fs
+            .readdirSync(chatsDir)
+            .filter((f) => f.startsWith("session-") && f.endsWith(".json"));
+        for (const sessionFile of sessionFiles) {
+            try {
+                const content = fs.readFileSync(path.join(chatsDir, sessionFile), "utf-8");
+                const session = JSON.parse(content);
+                for (const msg of session.messages) {
+                    // Only process gemini messages with token data and timestamp
+                    if (msg.type !== "gemini" || !msg.tokens || !msg.model || !msg.timestamp)
+                        continue;
+                    const timestamp = new Date(msg.timestamp).getTime();
+                    // Skip invalid timestamps
+                    if (isNaN(timestamp))
+                        continue;
+                    messages.push({
+                        source: "gemini",
+                        model: msg.model,
+                        timestamp,
+                        tokens: {
+                            input: msg.tokens.input || 0,
+                            output: msg.tokens.output || 0,
+                            cached: msg.tokens.cached || 0,
+                            thoughts: msg.tokens.thoughts || 0,
+                            tool: msg.tokens.tool || 0,
+                        },
+                    });
+                }
+            }
+            catch {
+                // Skip malformed files
+            }
+        }
+    }
+    return messages;
+}
+//# sourceMappingURL=gemini.js.map

package/dist/gemini.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"gemini.js","sourceRoot":"","sources":["../src/gemini.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAClC,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAqD9B,MAAM,UAAU,iBAAiB;IAC/B,OAAO,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,SAAS,CAAC,CAAC;AAC5C,CAAC;AAED,MAAM,UAAU,kBAAkB;IAChC,MAAM,QAAQ,GAAG,iBAAiB,EAAE,CAAC;IACrC,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;IAE3C,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;QAC5B,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,MAAM,UAAU,GAAG,IAAI,GAAG,EAA2B,CAAC;IAEtD,+BAA+B;IAC/B,MAAM,WAAW,GAAG,EAAE;SACnB,WAAW,CAAC,OAAO,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC;SAC7C,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;SAC9B,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;IAE1C,KAAK,MAAM,UAAU,IAAI,WAAW,EAAE,CAAC;QACrC,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QAChD,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC;YAAE,SAAS;QAEvC,8BAA8B;QAC9B,MAAM,YAAY,GAAG,EAAE;aACpB,WAAW,CAAC,QAAQ,CAAC;aACrB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC;QAElE,KAAK,MAAM,WAAW,IAAI,YAAY,EAAE,CAAC;YACvC,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,WAAW,CAAC,EAAE,OAAO,CAAC,CAAC;gBAC3E,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAkB,CAAC;gBAErD,KAAK,MAAM,GAAG,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;oBACnC,+CAA+C;oBAC/C,IAAI,GAAG,CAAC,IAAI,KAAK,QAAQ,IAAI,CAAC,GAAG,CAAC,MAAM,IAAI,CAAC,GAAG,CAAC,KAAK;wBAAE,SAAS;oBAEjE,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,CAAC;oBACxB,IAAI,KAAK,GAAG,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;oBAClC,IAAI,CAAC,KAAK,EAAE,CAAC;wBACX,KAAK,GAAG;4BACN,MAAM,EAAE,QAAQ;4BAChB,KAAK;4BACL,YAAY,EAAE,CAAC;4BACf,KAAK,EAAE,CAAC;4BACR,MAAM,EAAE,CAAC;4BACT,MAAM,EAAE,CAAC;4BACT,QAAQ,EAAE,CAAC;4BACX,IAAI,EAAE,CAAC;yBACR,CAAC;wBACF,UAAU,CAAC,GAAG,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;oBAC/B,CAAC;oBAED,KAAK,CAAC,YAAY,EAAE,CAAC;oBACrB,KAAK,CAAC,KAAK,IAAI,GAAG,CAAC,MAAM,CAAC,KAAK,IAAI,CAAC,CAAC;oBACrC,KAAK,CAAC,MAAM,IAAI,GAAG,CAAC,MAAM,CAAC,MAAM,IAAI,CAAC,CAAC;oBACvC,KAAK,CAAC,MAAM,IAAI,GAAG,CAAC,MAAM,CAAC,MAAM,IAAI,CAAC,CAAC;oBACvC,KAAK,CAAC,QAAQ,IAAI,GAAG,CAAC,MAAM,CAAC,QAAQ,IAAI,CAAC,CAAC;oBAC3C,KAAK,CAAC,IAAI,IAAI,GAAG,CAAC,MAAM,CAAC,IAAI,IAAI,CAAC,CAAC;gBACrC,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,uBAAuB;YACzB,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC,CAAC;AACzC,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,+BAA+B;IAC7C,MAAM,QAAQ,GAAG,iBAAiB,EAAE,CAAC;IACrC,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;IAE3C,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;QAC5B,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,MAAM,QAAQ,GAAiC,EAAE,CAAC;IAElD,+BAA+B;IAC/B,MAAM,WAAW,GAAG,EAAE;SACnB,WAAW,CAAC,OAAO,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC;SAC7C,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;SAC9B,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;IAE1C,KAAK,MAAM,UAAU,IAAI,WAAW,EAAE,CAAC;QACrC,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QAChD,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC;YAAE,SAAS;QAEvC,8BAA8B;QAC9B,MAAM,YAAY,GAAG,EAAE;aACpB,WAAW,CAAC,QAAQ,CAAC;aACrB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC;QAElE,KAAK,MAAM,WAAW,IAAI,YAAY,EAAE,CAAC;YACvC,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,WAAW,CAAC,EAAE,OAAO,CAAC,CAAC;gBAC3E,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAkB,CAAC;gBAErD,KAAK,MAAM,GAAG,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;oBACnC,6DAA6D;oBAC7D,IAAI,GAAG,CAAC,IAAI,KAAK,QAAQ,IAAI,CAAC,GAAG,CAAC,MAAM,IAAI,CAAC,GAAG,CAAC,KAAK,IAAI,CAAC,GAAG,CAAC,SAAS;wBAAE,SAAS;oBAEnF,MAAM,SAAS,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,CAAC;oBAEpD,0BAA0B;oBAC1B,IAAI,KAAK,CAAC,SAAS,CAAC;wBAAE,SAAS;oBAE/B,QAAQ,CAAC,IAAI,CAAC;wBACZ,MAAM,EAAE,QAAQ;wBAChB,KAAK,EAAE,GAAG,CAAC,KAAK;wBAChB,SAAS;wBACT,MAAM,EAAE;4BACN,KAAK,EAAE,GAAG,CAAC,MAAM,CAAC,KAAK,IAAI,CAAC;4BAC5B,MAAM,EAAE,GAAG,CAAC,MAAM,CAAC,MAAM,IAAI,CAAC;4BAC9B,MAAM,EAAE,GAAG,CAAC,MAAM,CAAC,MAAM,IAAI,CAAC;4BAC9B,QAAQ,EAAE,GAAG,CAAC,MAAM,CAAC,QAAQ,IAAI,CAAC;4BAClC,IAAI,EAAE,GAAG,CAAC,MAAM,CAAC,IAAI,IAAI,CAAC;yBAC3B;qBACF,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,uBAAuB;YACzB,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,QAAQ,CAAC;AAClB,CAAC"}