opencode-token-tracker 1.0.0

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 tongshuanglong
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -0,0 +1,119 @@
+# opencode-token-tracker
+
+Real-time token usage and cost tracking plugin for [OpenCode](https://opencode.ai).
+
+![Toast Screenshot](./docs/toast-demo.png)
+
+## Features
+
+- **Real-time Toast notifications** - See token usage and cost after each AI response
+- **Session statistics** - Track cumulative usage across your entire session
+- **Cost calculation** - Automatic cost estimation based on model pricing
+- **JSONL logging** - All usage data saved locally for analysis
+- **Multi-model support** - Claude, GPT, DeepSeek, Gemini, and more
+
+## Installation
+
+Add to your OpenCode config file (`~/.config/opencode/opencode.json`):
+
+```json
+{
+  "$schema": "https://opencode.ai/config.json",
+  "plugin": ["opencode-token-tracker"]
+}
+```
+
+Restart OpenCode and the plugin will be automatically installed.
+
+## Usage
+
+Once installed, you'll see Toast notifications after each AI response:
+
+```
+12.5K tokens
+$0.023 | Session: $0.156
+```
+
+When a session becomes idle, you'll see a summary:
+
+```
+Session: 45.2K tokens
+$0.156 | 8 msgs | 5min
+```
+
+## Log Files
+
+Token usage is logged to:
+
+```
+~/.config/opencode/logs/token-tracker/tokens.jsonl
+```
+
+Each line is a JSON object:
+
+```json
+{
+  "type": "tokens",
+  "sessionId": "ses_xxx",
+  "messageId": "msg_xxx",
+  "agent": "build",
+  "model": "claude-opus-4.5",
+  "provider": "github-copilot",
+  "input": 1500,
+  "output": 350,
+  "reasoning": 0,
+  "cacheRead": 5000,
+  "cacheWrite": 0,
+  "cost": 0.0234,
+  "_ts": 1234567890123
+}
+```
+
+## Supported Models
+
+| Provider | Models |
+|----------|--------|
+| Anthropic | Claude Opus 4.5, Sonnet 4/4.5, Haiku 4/4.5 |
+| OpenAI | GPT-5.x, GPT-4.x, o1, o3 |
+| DeepSeek | deepseek-chat, deepseek-reasoner |
+| Google | Gemini 2.x, 3.x |
+
+Unknown models use a default pricing estimate.
+
+## Configuration
+
+Currently the plugin works out of the box with no configuration needed.
+
+Future versions may support:
+- Custom pricing overrides
+- Toast display options
+- Export formats
+
+## Development
+
+```bash
+# Clone the repo
+git clone https://github.com/tongsh6/opencode-token-tracker.git
+cd opencode-token-tracker
+
+# Install dependencies
+npm install
+
+# Build
+npm run build
+
+# Link for local testing
+npm link
+cd ~/.config/opencode
+npm link opencode-token-tracker
+```
+
+## License
+
+MIT © [tongsh6](https://github.com/tongsh6)
+
+## Related
+
+- [OpenCode](https://opencode.ai) - The AI coding assistant
+- [OpenCode Plugins](https://opencode.ai/docs/plugins) - Plugin documentation
+- [oh-my-opencode](https://github.com/code-yeongyu/oh-my-opencode) - OpenCode enhancement plugin

package/dist/index.d.ts

@@ -0,0 +1,3 @@
+import type { Plugin } from "@opencode-ai/plugin";
+export declare const TokenTrackerPlugin: Plugin;
+export default TokenTrackerPlugin;

package/dist/index.js

@@ -0,0 +1,218 @@
+import { appendFileSync, existsSync, mkdirSync } from "fs";
+import { join } from "path";
+import { homedir } from "os";
+const LOG_DIR = join(homedir(), ".config", "opencode", "logs", "token-tracker");
+const LOG_FILE = join(LOG_DIR, "tokens.jsonl");
+// Prices as of 2026-02 (update as needed)
+const 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 },
+    "claude-sonnet-4": { input: 3, output: 15, cacheRead: 0.3, cacheWrite: 3.75 },
+    "claude-haiku-4.5": { input: 0.8, output: 4, cacheRead: 0.08, cacheWrite: 1 },
+    "claude-haiku-4": { input: 0.8, output: 4, cacheRead: 0.08, cacheWrite: 1 },
+    // OpenAI GPT
+    "gpt-5.2": { input: 2.5, output: 10 },
+    "gpt-5.2-codex": { input: 3, output: 12 },
+    "gpt-5.1": { input: 2, output: 8 },
+    "gpt-5": { input: 5, output: 15 },
+    "gpt-4.1": { input: 2, output: 8 },
+    "gpt-4.1-mini": { input: 0.4, output: 1.6 },
+    "gpt-4.1-nano": { input: 0.1, output: 0.4 },
+    "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 },
+    "o1": { input: 15, output: 60 },
+    "o1-mini": { input: 1.1, output: 4.4 },
+    // DeepSeek
+    "deepseek-chat": { input: 0.14, output: 0.28, cacheRead: 0.014 },
+    "deepseek-reasoner": { input: 0.55, output: 2.19, cacheRead: 0.055 },
+    // Google Gemini
+    "gemini-3-pro": { input: 1.25, output: 5 },
+    "gemini-3-pro-preview": { input: 1.25, output: 5 },
+    "gemini-3-flash": { input: 0.1, output: 0.4 },
+    "gemini-2.5-pro": { input: 1.25, output: 5 },
+    "gemini-2.5-flash": { input: 0.075, output: 0.3 },
+    "gemini-2.0-flash": { input: 0.1, output: 0.4 },
+    // 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")
+    const modelLower = model.toLowerCase();
+    for (const [key, pricing] of Object.entries(PRICING)) {
+        if (key !== "_default" && modelLower.includes(key.toLowerCase())) {
+            return pricing;
+        }
+    }
+    return PRICING["_default"];
+}
+function calculateCost(model, input, output, cacheRead = 0, cacheWrite = 0) {
+    const pricing = getModelPricing(model);
+    // 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;
+    const outputCost = (output / 1_000_000) * pricing.output;
+    const cacheReadCost = (cacheRead / 1_000_000) * (pricing.cacheRead ?? pricing.input * 0.1);
+    const cacheWriteCost = (cacheWrite / 1_000_000) * (pricing.cacheWrite ?? pricing.input * 1.25);
+    return inputCost + outputCost + cacheReadCost + cacheWriteCost;
+}
+const sessionStats = new Map();
+function getOrCreateSessionStats(sessionId) {
+    if (!sessionStats.has(sessionId)) {
+        sessionStats.set(sessionId, {
+            totalInput: 0,
+            totalOutput: 0,
+            totalReasoning: 0,
+            totalCacheRead: 0,
+            totalCacheWrite: 0,
+            totalCost: 0,
+            messageCount: 0,
+            startTime: Date.now(),
+        });
+    }
+    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
+// ============================================================================
+const seen = new Set();
+function isDuplicate(key) {
+    if (seen.has(key))
+        return true;
+    seen.add(key);
+    // Cleanup old entries to prevent memory leak
+    if (seen.size > 10000) {
+        const entries = Array.from(seen);
+        entries.slice(0, 5000).forEach(k => seen.delete(k));
+    }
+    return false;
+}
+// ============================================================================
+// Logging
+// ============================================================================
+function ensureLogDir() {
+    if (!existsSync(LOG_DIR)) {
+        mkdirSync(LOG_DIR, { recursive: true });
+    }
+}
+function logJson(data) {
+    ensureLogDir();
+    const entry = JSON.stringify({ ...data, _ts: Date.now() }) + "\n";
+    appendFileSync(LOG_FILE, entry);
+}
+export const TokenTrackerPlugin = async ({ directory, client }) => {
+    logJson({ type: "init", directory });
+    return {
+        event: async ({ event }) => {
+            try {
+                // Handle message updates (token tracking)
+                if (event.type === "message.updated") {
+                    const props = event.properties;
+                    const info = props?.info;
+                    if (!info?.tokens)
+                        return;
+                    const messageId = info.id;
+                    const sessionId = info.sessionID;
+                    if (!messageId || !sessionId)
+                        return;
+                    const input = info.tokens.input ?? 0;
+                    const output = info.tokens.output ?? 0;
+                    const reasoning = info.tokens.reasoning ?? 0;
+                    const cacheRead = info.tokens.cache?.read ?? 0;
+                    const cacheWrite = info.tokens.cache?.write ?? 0;
+                    const hasTokens = input > 0 || output > 0;
+                    if (!hasTokens)
+                        return;
+                    const dedupeKey = `${messageId}-${input}-${output}`;
+                    if (isDuplicate(dedupeKey))
+                        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);
+                    // Update session stats
+                    const stats = getOrCreateSessionStats(sessionId);
+                    stats.totalInput += input;
+                    stats.totalOutput += output;
+                    stats.totalReasoning += reasoning;
+                    stats.totalCacheRead += cacheRead;
+                    stats.totalCacheWrite += cacheWrite;
+                    stats.totalCost += cost;
+                    stats.messageCount += 1;
+                    // Log to file
+                    logJson({
+                        type: "tokens",
+                        sessionId,
+                        messageId,
+                        role: info.role,
+                        agent: info.agent,
+                        model,
+                        provider,
+                        input,
+                        output,
+                        reasoning,
+                        cacheRead,
+                        cacheWrite,
+                        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,
+                            },
+                        });
+                    }
+                    catch { }
+                }
+                // Handle session idle (show summary)
+                if (event.type === "session.idle") {
+                    const props = event.properties;
+                    const sessionId = props?.sessionID;
+                    if (!sessionId)
+                        return;
+                    const stats = sessionStats.get(sessionId);
+                    if (!stats || stats.messageCount === 0)
+                        return;
+                    const duration = Math.round((Date.now() - stats.startTime) / 1000 / 60);
+                    const totalTokens = stats.totalInput + stats.totalOutput;
+                    try {
+                        await client.tui.showToast({
+                            body: {
+                                title: `Session: ${formatTokens(totalTokens)} tokens`,
+                                message: `${formatCost(stats.totalCost)} | ${stats.messageCount} msgs | ${duration}min`,
+                                variant: "info",
+                                duration: 5000,
+                            },
+                        });
+                    }
+                    catch { }
+                }
+            }
+            catch { }
+        },
+    };
+};
+export default TokenTrackerPlugin;

package/package.json

@@ -0,0 +1,51 @@
+{
+  "name": "opencode-token-tracker",
+  "version": "1.0.0",
+  "description": "Real-time token usage and cost tracking plugin for OpenCode with Toast notifications",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "type": "module",
+  "scripts": {
+    "build": "tsc",
+    "prepublishOnly": "npm run build"
+  },
+  "keywords": [
+    "opencode",
+    "plugin",
+    "token",
+    "usage",
+    "cost",
+    "tracking",
+    "ai",
+    "llm",
+    "claude",
+    "gpt",
+    "anthropic",
+    "openai"
+  ],
+  "author": "tongsh6",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/tongsh6/opencode-token-tracker.git"
+  },
+  "bugs": {
+    "url": "https://github.com/tongsh6/opencode-token-tracker/issues"
+  },
+  "homepage": "https://github.com/tongsh6/opencode-token-tracker#readme",
+  "dependencies": {
+    "@opencode-ai/plugin": "latest"
+  },
+  "devDependencies": {
+    "@types/node": "^20.0.0",
+    "typescript": "^5.0.0"
+  },
+  "files": [
+    "dist",
+    "README.md",
+    "LICENSE"
+  ],
+  "engines": {
+    "node": ">=18.0.0"
+  }
+}