opencode-token-tracker 1.3.2 → 1.5.0

package/dist/bin/opencode-tokens.js

@@ -1,95 +1,20 @@
 #!/usr/bin/env node
+import { BUILTIN_PRICING, DEFAULT_CONFIG, formatCost, formatTokens, getStartOfDay, getStartOfWeek, getStartOfMonth, validateConfig } from "../lib/shared.js";
 import { readFileSync, existsSync, writeFileSync } from "fs";
 import { join } from "path";
 import { homedir } from "os";
 const CONFIG_DIR = join(homedir(), ".config", "opencode");
 const CONFIG_FILE = join(CONFIG_DIR, "token-tracker.json");
 const LOG_FILE = join(CONFIG_DIR, "logs", "token-tracker", "tokens.jsonl");
-// Built-in pricing (USD per 1M tokens) - Updated 2026-02-05
-// Keep in sync with index.ts BUILTIN_PRICING
-const BUILTIN_PRICING = {
-    // Anthropic Claude (https://www.anthropic.com/pricing#api)
-    "claude-opus-4.5": { input: 5, output: 25, cacheRead: 0.5, cacheWrite: 6.25 },
-    "claude-sonnet-4.5": { input: 3, output: 15, cacheRead: 0.3, cacheWrite: 3.75 },
-    "claude-sonnet-4": { input: 3, output: 15, cacheRead: 0.3, cacheWrite: 3.75 },
-    "claude-haiku-4.5": { input: 1, output: 5, cacheRead: 0.1, cacheWrite: 1.25 },
-    "claude-haiku-4": { input: 1, output: 5, cacheRead: 0.1, cacheWrite: 1.25 },
-    "claude-opus-4.1": { input: 15, output: 75, cacheRead: 1.5, cacheWrite: 18.75 },
-    "claude-opus-4": { input: 15, output: 75, cacheRead: 1.5, cacheWrite: 18.75 },
-    "claude-haiku-3": { input: 0.25, output: 1.25, cacheRead: 0.03, cacheWrite: 0.3 },
-    // OpenAI GPT (https://openai.com/api/pricing/)
-    "gpt-5.2": { input: 1.75, output: 14, cacheRead: 0.175 },
-    "gpt-5.2-pro": { input: 21, output: 168 },
-    "gpt-5-mini": { input: 0.25, output: 2, cacheRead: 0.025 },
-    "gpt-5.1": { input: 2, output: 8 },
-    "gpt-5": { input: 5, output: 15 },
-    "gpt-4.1": { input: 3, output: 12, cacheRead: 0.75 },
-    "gpt-4.1-mini": { input: 0.8, output: 3.2, cacheRead: 0.2 },
-    "gpt-4.1-nano": { input: 0.2, output: 0.8, cacheRead: 0.05 },
-    "gpt-4o": { input: 2.5, output: 10 },
-    "gpt-4o-mini": { input: 0.15, output: 0.6 },
-    "o3": { input: 10, output: 40 },
-    "o3-mini": { input: 1.1, output: 4.4 },
-    "o4-mini": { input: 4, output: 16, cacheRead: 1 },
-    "o1": { input: 15, output: 60 },
-    "o1-mini": { input: 1.1, output: 4.4 },
-    // DeepSeek (https://api-docs.deepseek.com/quick_start/pricing)
-    "deepseek-chat": { input: 0.28, output: 0.42, cacheRead: 0.028 },
-    "deepseek-reasoner": { input: 0.28, output: 0.42, cacheRead: 0.028 },
-    // Google Gemini (https://cloud.google.com/vertex-ai/generative-ai/pricing)
-    "gemini-3-pro": { input: 2, output: 12, cacheRead: 0.2 },
-    "gemini-3-pro-preview": { input: 2, output: 12, cacheRead: 0.2 },
-    "gemini-3-flash": { input: 0.5, output: 2, cacheRead: 0.05 },
-    "gemini-3-flash-preview": { input: 0.5, output: 2, cacheRead: 0.05 },
-    "gemini-2.5-pro": { input: 1.25, output: 10, cacheRead: 0.125 },
-    "gemini-2.5-flash": { input: 0.1, output: 0.4, cacheRead: 0.01 },
-    "gemini-2.5-flash-lite": { input: 0.1, output: 0.4, cacheRead: 0.01 },
-    "gemini-2.0-flash": { input: 0.15, output: 0.6, cacheRead: 0.015 },
-    // Fallback
-    "_default": { input: 1, output: 4 },
-};
 // ============================================================================
 // Helpers
 // ============================================================================
-function formatTokens(tokens) {
-    if (tokens >= 1_000_000)
-        return `${(tokens / 1_000_000).toFixed(2)}M`;
-    if (tokens >= 1_000)
-        return `${(tokens / 1_000).toFixed(1)}K`;
-    return tokens.toString();
-}
-function formatCost(cost) {
-    if (cost < 0.01)
-        return `$${cost.toFixed(4)}`;
-    if (cost < 1)
-        return `$${cost.toFixed(3)}`;
-    return `$${cost.toFixed(2)}`;
-}
 function padRight(str, len) {
     return str.length >= len ? str : str + " ".repeat(len - str.length);
 }
 function padLeft(str, len) {
     return str.length >= len ? str : " ".repeat(len - str.length) + str;
 }
-function getStartOfDay(date) {
-    const d = new Date(date);
-    d.setHours(0, 0, 0, 0);
-    return d.getTime();
-}
-function getStartOfWeek(date) {
-    const d = new Date(date);
-    const day = d.getDay();
-    const diff = d.getDate() - day + (day === 0 ? -6 : 1);
-    d.setDate(diff);
-    d.setHours(0, 0, 0, 0);
-    return d.getTime();
-}
-function getStartOfMonth(date) {
-    const d = new Date(date);
-    d.setDate(1);
-    d.setHours(0, 0, 0, 0);
-    return d.getTime();
-}
 // ============================================================================
 // Data Loading
 // ============================================================================
@@ -120,11 +45,20 @@ function loadEntries(since) {
 function loadConfig() {
     try {
         if (existsSync(CONFIG_FILE)) {
-            return JSON.parse(readFileSync(CONFIG_FILE, "utf-8"));
+            const raw = JSON.parse(readFileSync(CONFIG_FILE, "utf-8"));
+            const result = validateConfig(raw);
+            if (result.warnings.length > 0) {
+                for (const w of result.warnings) {
+                    console.error(`  [token-tracker] config warning: ${w}`);
+                }
+            }
+            return result.config;
         }
     }
-    catch { }
-    return {};
+    catch {
+        console.error("  [token-tracker] config warning: Config file is not valid JSON, using defaults");
+    }
+    return DEFAULT_CONFIG;
 }
 // ============================================================================
 // Stats Aggregation
@@ -171,13 +105,13 @@ function printSummary(title, stats) {
     console.log();
     console.log(`  ${title}`);
     console.log(`  ${"─".repeat(50)}`);
-    console.log(`  Total Tokens:    ${padLeft(formatTokens(totalTokens), 12)}`);
-    console.log(`    Input:         ${padLeft(formatTokens(stats.input), 12)}`);
-    console.log(`    Output:        ${padLeft(formatTokens(stats.output), 12)}`);
+    console.log(`  Total Tokens:    ${padLeft(formatTokens(totalTokens, 2), 12)}`);
+    console.log(`    Input:         ${padLeft(formatTokens(stats.input, 2), 12)}`);
+    console.log(`    Output:        ${padLeft(formatTokens(stats.output, 2), 12)}`);
     if (stats.reasoning > 0) {
-        console.log(`    Reasoning:     ${padLeft(formatTokens(stats.reasoning), 12)}`);
+        console.log(`    Reasoning:     ${padLeft(formatTokens(stats.reasoning, 2), 12)}`);
     }
-    console.log(`  Cache Read:      ${padLeft(formatTokens(stats.cacheRead), 12)}`);
+    console.log(`  Cache Read:      ${padLeft(formatTokens(stats.cacheRead, 2), 12)}`);
     console.log(`  Total Cost:      ${padLeft(formatCost(stats.cost), 12)}`);
     console.log(`  Messages:        ${padLeft(stats.count.toString(), 12)}`);
     console.log();
@@ -199,7 +133,7 @@ function printTable(title, groups, labelHeader) {
     console.log(`  ${"-".repeat(labelWidth)}  ${"-".repeat(tokensWidth)}  ${"-".repeat(costWidth)}  ${"-".repeat(countWidth)}`);
     for (const [label, stats] of sorted) {
         const totalTokens = stats.input + stats.output;
-        console.log(`  ${padRight(label, labelWidth)}  ${padLeft(formatTokens(totalTokens), tokensWidth)}  ${padLeft(formatCost(stats.cost), costWidth)}  ${padLeft(stats.count.toString(), countWidth)}`);
+        console.log(`  ${padRight(label, labelWidth)}  ${padLeft(formatTokens(totalTokens, 2), tokensWidth)}  ${padLeft(formatCost(stats.cost), costWidth)}  ${padLeft(stats.count.toString(), countWidth)}`);
     }
     console.log();
 }
@@ -224,7 +158,7 @@ function printDailyBreakdown(entries) {
     console.log(`  ${"-".repeat(dateWidth)}  ${"-".repeat(tokensWidth)}  ${"-".repeat(costWidth)}  ${"-".repeat(countWidth)}`);
     for (const [date, stats] of sorted) {
         const totalTokens = stats.input + stats.output;
-        console.log(`  ${padRight(date, dateWidth)}  ${padLeft(formatTokens(totalTokens), tokensWidth)}  ${padLeft(formatCost(stats.cost), costWidth)}  ${padLeft(stats.count.toString(), countWidth)}`);
+        console.log(`  ${padRight(date, dateWidth)}  ${padLeft(formatTokens(totalTokens, 2), tokensWidth)}  ${padLeft(formatCost(stats.cost), costWidth)}  ${padLeft(stats.count.toString(), countWidth)}`);
     }
     console.log();
 }
@@ -285,15 +219,15 @@ function cmdStats(period, breakdown) {
 function cmdPricing() {
     const config = loadConfig();
     console.log(`
-  Built-in Pricing Table (USD per 1M tokens) - Updated 2026-02-05
+  Built-in Pricing Table (USD per 1M tokens) - Updated 2026-02-11
   ══════════════════════════════════════════════════════════════════
 `);
     // Group by provider
     const groups = {
-        "Anthropic Claude": ["claude-opus-4.5", "claude-sonnet-4.5", "claude-sonnet-4", "claude-haiku-4.5", "claude-haiku-4", "claude-opus-4.1", "claude-opus-4", "claude-haiku-3"],
+        "Anthropic Claude": ["claude-opus-4.6", "claude-opus-4.5", "claude-sonnet-4.5", "claude-sonnet-4", "claude-haiku-4.5", "claude-haiku-4", "claude-opus-4.1", "claude-opus-4", "claude-haiku-3"],
         "OpenAI": ["gpt-5.2", "gpt-5.2-pro", "gpt-5-mini", "gpt-5.1", "gpt-5", "gpt-4.1", "gpt-4.1-mini", "gpt-4.1-nano", "gpt-4o", "gpt-4o-mini", "o3", "o3-mini", "o4-mini", "o1", "o1-mini"],
         "DeepSeek": ["deepseek-chat", "deepseek-reasoner"],
-        "Google Gemini": ["gemini-3-pro", "gemini-3-pro-preview", "gemini-3-flash", "gemini-3-flash-preview", "gemini-2.5-pro", "gemini-2.5-flash", "gemini-2.5-flash-lite", "gemini-2.0-flash"],
+        "Google Gemini": ["gemini-3-pro", "gemini-3-pro-preview", "gemini-3-flash", "gemini-3-flash-preview", "gemini-2.5-pro", "gemini-2.5-flash", "gemini-2.5-flash-lite", "gemini-2.0-flash", "gemini-2.0-flash-lite"],
     };
     const modelWidth = 20;
     const priceWidth = 10;
@@ -404,6 +338,12 @@ function cmdConfig(action) {
                 duration: 3000,
                 showOnIdle: true,
             },
+            budget: {
+                daily: 5,
+                weekly: 25,
+                monthly: 100,
+                warnAt: 0.8,
+            },
         };
         // Add providers as comments/examples
         for (const provider of providers) {
@@ -418,11 +358,11 @@ function cmdConfig(action) {
         }
         // Print explanation first
         console.log(`
-  Pricing Configuration Guide
+  Configuration Guide
   ══════════════════════════════════════════════════════════════════
 
-  All prices are in USD per 1 MILLION tokens.
-
+  PRICING (prices in USD per 1 MILLION tokens)
+  ────────────────────────────────────────────────────────────────
   Fields:
     input      Cost for input/prompt tokens sent to the model
     output     Cost for output/completion tokens from the model
@@ -434,7 +374,7 @@ function cmdConfig(action) {
     { "input": 0, "output": 0 }       = Free (subscription or local model)
 
   Common scenarios:
-    - GitHub Copilot, Cursor, etc.   → Set to 0 (subscription-based)
+    - GitHub Copilot, Cursor, etc.   → Set provider to { input: 0, output: 0 }
     - Local/self-hosted models       → Set to 0
     - Direct API usage               → Look up provider's pricing page
 
@@ -445,6 +385,17 @@ function cmdConfig(action) {
     - Google:    https://ai.google.dev/pricing
     - Or run:    opencode-tokens pricing
 
+  BUDGET CONTROL
+  ────────────────────────────────────────────────────────────────
+  Set spending limits to avoid unexpected costs:
+    daily      Maximum spend per day (USD)
+    weekly     Maximum spend per week (USD)
+    monthly    Maximum spend per month (USD)
+    warnAt     Warning threshold (0-1), default 0.8 = 80%
+
+  When budget is exceeded, you'll see a warning toast.
+  Check status anytime with: opencode-tokens budget
+
   ────────────────────────────────────────────────────────────────
   Example config based on your usage:
 `);
@@ -485,6 +436,94 @@ function cmdConfig(action) {
     console.log(`    opencode-tokens config generate  Create config file`);
     console.log();
 }
+function cmdBudget() {
+    const config = loadConfig();
+    const budget = config.budget;
+    if (!budget?.daily && !budget?.weekly && !budget?.monthly) {
+        console.log(`
+  Budget Status
+  ══════════════════════════════════════════════════════════════════
+
+  No budget configured.
+
+  To set a budget, add to your config file (${CONFIG_FILE}):
+
+  {
+    "budget": {
+      "daily": 5,       // $5 per day
+      "weekly": 25,     // $25 per week (optional)
+      "monthly": 100,   // $100 per month (optional)
+      "warnAt": 0.8     // Warn at 80% usage
+    }
+  }
+
+  Run: opencode-tokens config init  for more details.
+`);
+        return;
+    }
+    const now = new Date();
+    const warnAt = budget.warnAt ?? 0.8;
+    console.log(`
+  Budget Status
+  ══════════════════════════════════════════════════════════════════
+`);
+    // Helper to create progress bar
+    const progressBar = (pct, width = 20) => {
+        const filled = Math.min(Math.round(pct * width), width);
+        const empty = width - filled;
+        const bar = "█".repeat(filled) + "░".repeat(empty);
+        return bar;
+    };
+    // Helper to get color indicator
+    const statusIndicator = (pct) => {
+        if (pct >= 1)
+            return "🔴";
+        if (pct >= warnAt)
+            return "🟡";
+        return "🟢";
+    };
+    // Calculate the earliest period start to minimize data loaded
+    const dayStart = getStartOfDay(now);
+    const weekStart = getStartOfWeek(now);
+    const monthStart = getStartOfMonth(now);
+    const earliestSince = Math.min(budget.daily ? dayStart : Infinity, budget.weekly ? weekStart : Infinity, budget.monthly ? monthStart : Infinity);
+    const entries = loadEntries(earliestSince);
+    // Daily budget
+    if (budget.daily) {
+        const dayEntries = entries.filter(e => e._ts >= dayStart);
+        const spent = dayEntries.reduce((sum, e) => sum + (e.cost ?? 0), 0);
+        const pct = spent / budget.daily;
+        const pctDisplay = Math.round(pct * 100);
+        console.log(`  ${statusIndicator(pct)} Daily`);
+        console.log(`    ${formatCost(spent)} / ${formatCost(budget.daily)}  [${progressBar(pct)}] ${pctDisplay}%`);
+        console.log(`    Remaining: ${formatCost(Math.max(0, budget.daily - spent))}`);
+        console.log();
+    }
+    // Weekly budget
+    if (budget.weekly) {
+        const weekEntries = entries.filter(e => e._ts >= weekStart);
+        const spent = weekEntries.reduce((sum, e) => sum + (e.cost ?? 0), 0);
+        const pct = spent / budget.weekly;
+        const pctDisplay = Math.round(pct * 100);
+        console.log(`  ${statusIndicator(pct)} Weekly`);
+        console.log(`    ${formatCost(spent)} / ${formatCost(budget.weekly)}  [${progressBar(pct)}] ${pctDisplay}%`);
+        console.log(`    Remaining: ${formatCost(Math.max(0, budget.weekly - spent))}`);
+        console.log();
+    }
+    // Monthly budget
+    if (budget.monthly) {
+        const monthEntries = entries.filter(e => e._ts >= monthStart);
+        const spent = monthEntries.reduce((sum, e) => sum + (e.cost ?? 0), 0);
+        const pct = spent / budget.monthly;
+        const pctDisplay = Math.round(pct * 100);
+        console.log(`  ${statusIndicator(pct)} Monthly`);
+        console.log(`    ${formatCost(spent)} / ${formatCost(budget.monthly)}  [${progressBar(pct)}] ${pctDisplay}%`);
+        console.log(`    Remaining: ${formatCost(Math.max(0, budget.monthly - spent))}`);
+        console.log();
+    }
+    console.log(`  Legend: 🟢 OK  🟡 Warning (>${Math.round(warnAt * 100)}%)  🔴 Exceeded`);
+    console.log();
+}
 function cmdHelp() {
     console.log(`
   opencode-tokens - Token usage statistics CLI
@@ -494,6 +533,7 @@ function cmdHelp() {
 
   Commands:
     (default)     Show usage statistics
+    budget        Show budget status (daily/weekly/monthly)
     pricing       Show built-in pricing table
     models        Show your used models and their pricing status  
     config        Show/generate configuration
@@ -508,6 +548,7 @@ function cmdHelp() {
 
   Examples:
     opencode-tokens                  # All-time summary
+    opencode-tokens budget           # Check budget status
     opencode-tokens today            # Today's summary
     opencode-tokens week --by model  # This week, by model
     opencode-tokens pricing          # Show pricing table
@@ -526,6 +567,10 @@ function main() {
         return;
     }
     // Handle subcommands
+    if (command === "budget") {
+        cmdBudget();
+        return;
+    }
     if (command === "pricing") {
         cmdPricing();
         return;