opencode-token-tracker 1.4.0 → 1.5.1

package/dist/index.js

@@ -1,3 +1,4 @@
+import { BUILTIN_PRICING, DEFAULT_CONFIG, findModelConfigPricing, formatCost, formatTokens, getStartOfDay, getStartOfWeek, getStartOfMonth, validateConfig } from "./lib/shared.js";
 import { appendFileSync, existsSync, mkdirSync, readFileSync } from "fs";
 import { join } from "path";
 import { homedir } from "os";
@@ -5,97 +6,24 @@ const CONFIG_DIR = join(homedir(), ".config", "opencode");
 const CONFIG_FILE = join(CONFIG_DIR, "token-tracker.json");
 const LOG_DIR = join(CONFIG_DIR, "logs", "token-tracker");
 const LOG_FILE = join(LOG_DIR, "tokens.jsonl");
-const DEFAULT_CONFIG = {
-    providers: {},
-    models: {},
-    toast: {
-        enabled: true,
-        duration: 3000,
-        showOnIdle: true,
-    },
-    budget: {
-        warnAt: 0.8,
-    },
-};
-// Built-in pricing table (USD per 1M tokens) - Updated 2026-02-05
-// Sources:
-// - Anthropic: https://www.anthropic.com/pricing#api
-// - OpenAI: https://openai.com/api/pricing/
-// - DeepSeek: https://api-docs.deepseek.com/quick_start/pricing
-// - Google: https://cloud.google.com/vertex-ai/generative-ai/pricing
-const BUILTIN_PRICING = {
-    // Anthropic Claude (https://www.anthropic.com/pricing#api)
-    // Opus 4.5: $5 input, $25 output, cache write $6.25, cache read $0.50
-    "claude-opus-4.5": { input: 5, output: 25, cacheRead: 0.5, cacheWrite: 6.25 },
-    // Sonnet 4.5: $3 input, $15 output (≤200K), cache write $3.75, cache read $0.30
-    "claude-sonnet-4.5": { input: 3, output: 15, cacheRead: 0.3, cacheWrite: 3.75 },
-    // Sonnet 4: $3 input, $15 output
-    "claude-sonnet-4": { input: 3, output: 15, cacheRead: 0.3, cacheWrite: 3.75 },
-    // Haiku 4.5: $1 input, $5 output, cache write $1.25, cache read $0.10
-    "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 },
-    // Legacy models
-    "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: $1.75 input, $14 output (flagship)
-    "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 series (fine-tuning prices, base may differ)
-    "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 series (may be deprecated)
-    "gpt-4o": { input: 2.5, output: 10 },
-    "gpt-4o-mini": { input: 0.15, output: 0.6 },
-    // Reasoning models
-    "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-V3.2: unified pricing for both chat and reasoner
-    // $0.28 input (cache miss), $0.028 input (cache hit), $0.42 output
-    "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 Preview: $2 input, $12 output (≤200K)
-    "gemini-3-pro": { input: 2, output: 12, cacheRead: 0.2 },
-    "gemini-3-pro-preview": { input: 2, output: 12, cacheRead: 0.2 },
-    // Gemini 3 Flash Preview: $0.5 input
-    "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: $1.25 input, $10 output (≤200K)
-    "gemini-2.5-pro": { input: 1.25, output: 10, cacheRead: 0.125 },
-    // Gemini 2.5 Flash Lite: $0.1 input
-    "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: $0.15 input
-    "gemini-2.0-flash": { input: 0.15, output: 0.6, cacheRead: 0.015 },
-    // Fallback for unknown models
-    "_default": { input: 1, output: 4 },
-};
+// ============================================================================
+// Configuration
+// ============================================================================
 let config = DEFAULT_CONFIG;
+let configWarnings = [];
 function loadConfig() {
     try {
         if (existsSync(CONFIG_FILE)) {
             const content = readFileSync(CONFIG_FILE, "utf-8");
-            const userConfig = JSON.parse(content);
-            return {
-                providers: { ...DEFAULT_CONFIG.providers, ...userConfig.providers },
-                models: { ...DEFAULT_CONFIG.models, ...userConfig.models },
-                toast: { ...DEFAULT_CONFIG.toast, ...userConfig.toast },
-                budget: { ...DEFAULT_CONFIG.budget, ...userConfig.budget },
-            };
+            const raw = JSON.parse(content);
+            const result = validateConfig(raw);
+            configWarnings = result.warnings;
+            return result.config;
         }
     }
-    catch (e) {
-        // Config parse error - use defaults
+    catch {
+        // JSON parse error - use defaults
+        configWarnings = ["Config file is not valid JSON, using defaults"];
     }
     return DEFAULT_CONFIG;
 }
@@ -108,27 +36,22 @@ function getModelPricing(model, provider) {
         return config.providers[provider];
     }
     // 2. Check user-defined model pricing
-    if (config.models[model]) {
-        return config.models[model];
+    const configuredPricing = findModelConfigPricing(config.models, model, provider);
+    if (configuredPricing) {
+        return configuredPricing;
     }
     // 3. Check built-in exact match
     if (BUILTIN_PRICING[model]) {
         return BUILTIN_PRICING[model];
     }
-    // 4. Try partial match in user config
+    // 4. Try partial match in built-in pricing
     const modelLower = model.toLowerCase();
-    for (const [key, pricing] of Object.entries(config.models)) {
-        if (modelLower.includes(key.toLowerCase())) {
-            return pricing;
-        }
-    }
-    // 5. Try partial match in built-in pricing
     for (const [key, pricing] of Object.entries(BUILTIN_PRICING)) {
         if (key !== "_default" && modelLower.includes(key.toLowerCase())) {
             return pricing;
         }
     }
-    // 6. Fallback to default
+    // 5. Fallback to default
     return BUILTIN_PRICING["_default"];
 }
 function calculateCost(model, provider, input, output, cacheRead = 0, cacheWrite = 0) {
@@ -157,20 +80,6 @@ function getOrCreateSessionStats(sessionId) {
     }
     return sessionStats.get(sessionId);
 }
-function formatCost(cost) {
-    if (cost < 0.01)
-        return `$${cost.toFixed(4)}`;
-    if (cost < 1)
-        return `$${cost.toFixed(3)}`;
-    return `$${cost.toFixed(2)}`;
-}
-function formatTokens(tokens) {
-    if (tokens >= 1_000_000)
-        return `${(tokens / 1_000_000).toFixed(1)}M`;
-    if (tokens >= 1_000)
-        return `${(tokens / 1_000).toFixed(1)}K`;
-    return tokens.toString();
-}
 // ============================================================================
 // Deduplication
 // ============================================================================
@@ -199,65 +108,129 @@ function logJson(data) {
     const entry = JSON.stringify({ ...data, _ts: Date.now() }) + "\n";
     appendFileSync(LOG_FILE, entry);
 }
-function getStartOfDay(date = new Date()) {
-    const d = new Date(date);
-    d.setHours(0, 0, 0, 0);
-    return d.getTime();
-}
-function getStartOfWeek(date = new Date()) {
-    const d = new Date(date);
-    const day = d.getDay();
-    const diff = d.getDate() - day + (day === 0 ? -6 : 1); // Monday as first day
-    d.setDate(diff);
-    d.setHours(0, 0, 0, 0);
-    return d.getTime();
-}
-function getStartOfMonth(date = new Date()) {
-    const d = new Date(date);
-    d.setDate(1);
-    d.setHours(0, 0, 0, 0);
-    return d.getTime();
-}
-function loadEntriesSince(since) {
+const budgetTracker = {
+    dailySpent: 0,
+    weeklySpent: 0,
+    monthlySpent: 0,
+    dayStart: 0,
+    weekStart: 0,
+    monthStart: 0,
+    initialized: false,
+};
+/**
+ * Load cost entries from JSONL since a given timestamp.
+ * Used only during initialization and period rollovers.
+ */
+function loadCostsSince(since) {
     if (!existsSync(LOG_FILE))
-        return [];
+        return 0;
     try {
         const content = readFileSync(LOG_FILE, "utf-8");
         const lines = content.trim().split("\n").filter(Boolean);
-        const entries = [];
+        let total = 0;
         for (const line of lines) {
             try {
                 const entry = JSON.parse(line);
                 if (entry.type === "tokens" && entry._ts >= since && entry.cost) {
-                    entries.push(entry);
+                    total += entry.cost;
                 }
             }
             catch { }
         }
-        return entries;
+        return total;
     }
     catch {
-        return [];
+        return 0;
+    }
+}
+/**
+ * Initialize budgetTracker from JSONL file (called once at plugin init).
+ */
+function initBudgetTracker() {
+    const now = new Date();
+    budgetTracker.dayStart = getStartOfDay(now);
+    budgetTracker.weekStart = getStartOfWeek(now);
+    budgetTracker.monthStart = getStartOfMonth(now);
+    // Only load from file if budget is configured
+    const budget = config.budget;
+    if (!budget.daily && !budget.weekly && !budget.monthly) {
+        budgetTracker.initialized = true;
+        return;
     }
+    // Load once using the earliest period boundary
+    const earliest = Math.min(budgetTracker.dayStart, budgetTracker.weekStart, budgetTracker.monthStart);
+    if (!existsSync(LOG_FILE)) {
+        budgetTracker.initialized = true;
+        return;
+    }
+    try {
+        const content = readFileSync(LOG_FILE, "utf-8");
+        const lines = content.trim().split("\n").filter(Boolean);
+        let daily = 0;
+        let weekly = 0;
+        let monthly = 0;
+        for (const line of lines) {
+            try {
+                const entry = JSON.parse(line);
+                if (entry.type !== "tokens" || !entry.cost || entry._ts < earliest)
+                    continue;
+                if (entry._ts >= budgetTracker.dayStart)
+                    daily += entry.cost;
+                if (entry._ts >= budgetTracker.weekStart)
+                    weekly += entry.cost;
+                if (entry._ts >= budgetTracker.monthStart)
+                    monthly += entry.cost;
+            }
+            catch { }
+        }
+        budgetTracker.dailySpent = daily;
+        budgetTracker.weeklySpent = weekly;
+        budgetTracker.monthlySpent = monthly;
+    }
+    catch { }
+    budgetTracker.initialized = true;
 }
-function calculateSpentSince(since) {
-    const entries = loadEntriesSince(since);
-    return entries.reduce((sum, e) => sum + (e.cost ?? 0), 0);
+/**
+ * Accumulate cost into budgetTracker after a new token entry is logged.
+ */
+function accumulateBudget(cost) {
+    if (!budgetTracker.initialized)
+        return;
+    const now = new Date();
+    const currentDayStart = getStartOfDay(now);
+    const currentWeekStart = getStartOfWeek(now);
+    const currentMonthStart = getStartOfMonth(now);
+    // Period rollover detection — reset and reload from file for accuracy
+    if (currentDayStart !== budgetTracker.dayStart) {
+        budgetTracker.dayStart = currentDayStart;
+        budgetTracker.dailySpent = loadCostsSince(currentDayStart);
+    }
+    if (currentWeekStart !== budgetTracker.weekStart) {
+        budgetTracker.weekStart = currentWeekStart;
+        budgetTracker.weeklySpent = loadCostsSince(currentWeekStart);
+    }
+    if (currentMonthStart !== budgetTracker.monthStart) {
+        budgetTracker.monthStart = currentMonthStart;
+        budgetTracker.monthlySpent = loadCostsSince(currentMonthStart);
+    }
+    budgetTracker.dailySpent += cost;
+    budgetTracker.weeklySpent += cost;
+    budgetTracker.monthlySpent += cost;
 }
 function checkBudgetStatus() {
     const budget = config.budget;
     if (!budget.daily && !budget.weekly && !budget.monthly) {
         return null;
     }
+    if (!budgetTracker.initialized)
+        return null;
     const warnAt = budget.warnAt ?? 0.8;
-    const now = new Date();
     // Check in order: daily -> weekly -> monthly (most restrictive first)
     if (budget.daily) {
-        const spent = calculateSpentSince(getStartOfDay(now));
-        const percentage = spent / budget.daily;
+        const percentage = budgetTracker.dailySpent / budget.daily;
         return {
             period: "daily",
-            spent,
+            spent: budgetTracker.dailySpent,
             limit: budget.daily,
             percentage,
             exceeded: percentage >= 1,
@@ -265,11 +238,10 @@ function checkBudgetStatus() {
         };
     }
     if (budget.weekly) {
-        const spent = calculateSpentSince(getStartOfWeek(now));
-        const percentage = spent / budget.weekly;
+        const percentage = budgetTracker.weeklySpent / budget.weekly;
         return {
             period: "weekly",
-            spent,
+            spent: budgetTracker.weeklySpent,
             limit: budget.weekly,
             percentage,
             exceeded: percentage >= 1,
@@ -277,11 +249,10 @@ function checkBudgetStatus() {
         };
     }
     if (budget.monthly) {
-        const spent = calculateSpentSince(getStartOfMonth(now));
-        const percentage = spent / budget.monthly;
+        const percentage = budgetTracker.monthlySpent / budget.monthly;
         return {
             period: "monthly",
-            spent,
+            spent: budgetTracker.monthlySpent,
             limit: budget.monthly,
             percentage,
             exceeded: percentage >= 1,
@@ -296,9 +267,25 @@ function formatBudgetMessage(status) {
     return `${periodLabel}: ${formatCost(status.spent)}/${formatCost(status.limit)} (${pct}%)`;
 }
 export const TokenTrackerPlugin = async ({ directory, client }) => {
-    // Load config on plugin init
+    // Load config on plugin init (with validation)
     config = loadConfig();
+    // Initialize in-memory budget tracker (reads JSONL once)
+    initBudgetTracker();
     logJson({ type: "init", directory, configLoaded: existsSync(CONFIG_FILE) });
+    // Show config validation warnings via Toast
+    if (configWarnings.length > 0) {
+        try {
+            await client.tui.showToast({
+                body: {
+                    title: "Token Tracker: config warning",
+                    message: configWarnings.join("; "),
+                    variant: "warning",
+                    duration: 5000,
+                },
+            });
+        }
+        catch { }
+    }
     return {
         event: async ({ event }) => {
             try {
@@ -351,6 +338,8 @@ export const TokenTrackerPlugin = async ({ directory, client }) => {
                         cacheWrite,
                         cost,
                     });
+                    // Accumulate cost into in-memory budget tracker
+                    accumulateBudget(cost);
                     // Show toast for this message
                     if (config.toast.enabled) {
                         const totalTokens = input + output;

package/dist/lib/shared.d.ts

@@ -0,0 +1,48 @@
+export interface ModelPricing {
+    input: number;
+    output: number;
+    cacheRead?: number;
+    cacheWrite?: number;
+}
+export interface ProviderModelPricingMap {
+    [provider: string]: ModelPricing;
+}
+export declare const BUILTIN_PRICING: Record<string, ModelPricing>;
+export declare function formatCost(cost: number): string;
+/**
+ * Format token count for display.
+ * @param tokens - raw token count
+ * @param millionDecimals - decimal places for M-level values (default 1)
+ */
+export declare function formatTokens(tokens: number, millionDecimals?: number): string;
+export declare function getStartOfDay(date?: Date): number;
+export declare function getStartOfWeek(date?: Date): number;
+export declare function getStartOfMonth(date?: Date): number;
+export interface ToastConfig {
+    enabled: boolean;
+    duration: number;
+    showOnIdle: boolean;
+}
+export interface BudgetConfig {
+    daily?: number;
+    weekly?: number;
+    monthly?: number;
+    warnAt: number;
+}
+export interface TrackerConfig {
+    providers: Record<string, ModelPricing>;
+    models: Record<string, ModelPricing | ProviderModelPricingMap>;
+    toast: ToastConfig;
+    budget: BudgetConfig;
+}
+export declare const DEFAULT_CONFIG: TrackerConfig;
+export interface ConfigValidationResult {
+    config: TrackerConfig;
+    warnings: string[];
+}
+/**
+ * Validate and normalize raw config (from JSON.parse).
+ * Invalid fields are silently corrected to defaults with warnings.
+ */
+export declare function validateConfig(raw: unknown): ConfigValidationResult;
+export declare function findModelConfigPricing(models: TrackerConfig["models"], model: string, provider: string): ModelPricing | undefined;