opencode-token-tracker 1.1.0 → 1.2.0

package/README.md

@@ -135,12 +135,66 @@ Unknown models use a default pricing estimate.
 
 ## Configuration
 
-Currently the plugin works out of the box with no configuration needed.
+Create a config file at `~/.config/opencode/token-tracker.json`:
 
-Future versions may support:
-- Custom pricing overrides
-- Toast display options
-- Export formats
+```json
+{
+  "providers": {
+    "github-copilot": { "input": 0, "output": 0 }
+  },
+  "models": {
+    "my-custom-model": { "input": 1, "output": 2 }
+  },
+  "toast": {
+    "enabled": true,
+    "duration": 3000,
+    "showOnIdle": true
+  }
+}
+```
+
+### Pricing Override
+
+Pricing is resolved in this order (first match wins):
+
+1. **Provider-level** - Override all models for a provider
+2. **User model config** - Custom model pricing in config file
+3. **Built-in pricing** - Default pricing table
+4. **Fallback** - $1/M input, $4/M output
+
+#### Example: Free providers
+
+If you're using GitHub Copilot or other subscription-based services, set their cost to $0:
+
+```json
+{
+  "providers": {
+    "github-copilot": { "input": 0, "output": 0 },
+    "cursor": { "input": 0, "output": 0 }
+  }
+}
+```
+
+#### Example: Custom model pricing
+
+Override or add pricing for specific models:
+
+```json
+{
+  "models": {
+    "claude-opus-4.5": { "input": 12, "output": 60, "cacheRead": 1.2 },
+    "my-local-model": { "input": 0, "output": 0 }
+  }
+}
+```
+
+### Toast Settings
+
+| Option | Type | Default | Description |
+|--------|------|---------|-------------|
+| `enabled` | boolean | `true` | Show toast notifications |
+| `duration` | number | `3000` | Toast display duration (ms) |
+| `showOnIdle` | boolean | `true` | Show session summary on idle |
 
 ## Development
 

package/dist/index.js

@@ -1,10 +1,21 @@
-import { appendFileSync, existsSync, mkdirSync } from "fs";
+import { appendFileSync, existsSync, mkdirSync, readFileSync } from "fs";
 import { join } from "path";
 import { homedir } from "os";
-const LOG_DIR = join(homedir(), ".config", "opencode", "logs", "token-tracker");
+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");
-// Prices as of 2026-02 (update as needed)
-const PRICING = {
+const DEFAULT_CONFIG = {
+    providers: {},
+    models: {},
+    toast: {
+        enabled: true,
+        duration: 3000,
+        showOnIdle: true,
+    },
+};
+// Built-in pricing table (USD per 1M tokens) - as of 2026-02
+const BUILTIN_PRICING = {
     // Anthropic Claude
     "claude-opus-4.5": { input: 15, output: 75, cacheRead: 1.5, cacheWrite: 18.75 },
     "claude-sonnet-4.5": { input: 3, output: 15, cacheRead: 0.3, cacheWrite: 3.75 },
@@ -38,21 +49,58 @@ const PRICING = {
     // Fallback for unknown models
     "_default": { input: 1, output: 4 },
 };
-function getModelPricing(model) {
-    // Try exact match first
-    if (PRICING[model])
-        return PRICING[model];
-    // Try partial match (e.g., "claude-opus-4.5" matches "claude-opus-4.5-xxx")
+let config = DEFAULT_CONFIG;
+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 },
+            };
+        }
+    }
+    catch (e) {
+        // Config parse error - use defaults
+    }
+    return DEFAULT_CONFIG;
+}
+// ============================================================================
+// Pricing
+// ============================================================================
+function getModelPricing(model, provider) {
+    // 1. Check provider-level override first (highest priority)
+    if (config.providers[provider]) {
+        return config.providers[provider];
+    }
+    // 2. Check user-defined model pricing
+    if (config.models[model]) {
+        return config.models[model];
+    }
+    // 3. Check built-in exact match
+    if (BUILTIN_PRICING[model]) {
+        return BUILTIN_PRICING[model];
+    }
+    // 4. Try partial match in user config
     const modelLower = model.toLowerCase();
-    for (const [key, pricing] of Object.entries(PRICING)) {
+    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;
         }
     }
-    return PRICING["_default"];
+    // 6. Fallback to default
+    return BUILTIN_PRICING["_default"];
 }
-function calculateCost(model, input, output, cacheRead = 0, cacheWrite = 0) {
-    const pricing = getModelPricing(model);
+function calculateCost(model, provider, input, output, cacheRead = 0, cacheWrite = 0) {
+    const pricing = getModelPricing(model, provider);
     // Billable input = total input - cache read (cached tokens are charged at cache rate)
     const billableInput = Math.max(0, input - cacheRead);
     const inputCost = (billableInput / 1_000_000) * pricing.input;
@@ -120,7 +168,9 @@ function logJson(data) {
     appendFileSync(LOG_FILE, entry);
 }
 export const TokenTrackerPlugin = async ({ directory, client }) => {
-    logJson({ type: "init", directory });
+    // Load config on plugin init
+    config = loadConfig();
+    logJson({ type: "init", directory, configLoaded: existsSync(CONFIG_FILE) });
     return {
         event: async ({ event }) => {
             try {
@@ -147,7 +197,7 @@ export const TokenTrackerPlugin = async ({ directory, client }) => {
                         return;
                     const model = info.model?.modelID ?? info.modelID ?? "unknown";
                     const provider = info.model?.providerID ?? info.providerID ?? "unknown";
-                    const cost = calculateCost(model, input, output, cacheRead, cacheWrite);
+                    const cost = calculateCost(model, provider, input, output, cacheRead, cacheWrite);
                     // Update session stats
                     const stats = getOrCreateSessionStats(sessionId);
                     stats.totalInput += input;
@@ -174,21 +224,25 @@ export const TokenTrackerPlugin = async ({ directory, client }) => {
                         cost,
                     });
                     // Show toast for this message
-                    const totalTokens = input + output;
-                    try {
-                        await client.tui.showToast({
-                            body: {
-                                title: `${formatTokens(totalTokens)} tokens`,
-                                message: `${formatCost(cost)} | Session: ${formatCost(stats.totalCost)}`,
-                                variant: "info",
-                                duration: 3000,
-                            },
-                        });
+                    if (config.toast.enabled) {
+                        const totalTokens = input + output;
+                        try {
+                            await client.tui.showToast({
+                                body: {
+                                    title: `${formatTokens(totalTokens)} tokens`,
+                                    message: `${formatCost(cost)} | Session: ${formatCost(stats.totalCost)}`,
+                                    variant: "info",
+                                    duration: config.toast.duration,
+                                },
+                            });
+                        }
+                        catch { }
                     }
-                    catch { }
                 }
                 // Handle session idle (show summary)
                 if (event.type === "session.idle") {
+                    if (!config.toast.enabled || !config.toast.showOnIdle)
+                        return;
                     const props = event.properties;
                     const sessionId = props?.sessionID;
                     if (!sessionId)

package/package.json

@@ -1,12 +1,12 @@
 {
   "name": "opencode-token-tracker",
-  "version": "1.1.0",
+  "version": "1.2.0",
   "description": "Real-time token usage and cost tracking plugin for OpenCode with Toast notifications and CLI stats",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
   "type": "module",
   "bin": {
-    "opencode-tokens": "./dist/bin/opencode-tokens.js"
+    "opencode-tokens": "dist/bin/opencode-tokens.js"
   },
   "scripts": {
     "build": "tsc",