opencode-token-tracker 1.5.2 → 1.5.4

package/dist/bin/opencode-tokens.js

@@ -1,6 +1,6 @@
 #!/usr/bin/env node
 import { BUILTIN_PRICING, DEFAULT_CONFIG, findModelConfigPricing, formatCost, formatTokens, getStartOfDay, getStartOfWeek, getStartOfMonth, validateConfig } from "../lib/shared.js";
-import { readFileSync, existsSync, writeFileSync } from "fs";
+import { readFileSync, existsSync, writeFileSync, openSync, readSync, closeSync, statSync } from "fs";
 import { join } from "path";
 import { homedir } from "os";
 const CONFIG_DIR = join(homedir(), ".config", "opencode");
@@ -22,25 +22,62 @@ function loadEntries(since) {
     if (!existsSync(LOG_FILE)) {
         return [];
     }
-    const content = readFileSync(LOG_FILE, "utf-8");
-    const lines = content.trim().split("\n").filter(Boolean);
     const entries = [];
-    for (const line of lines) {
-        try {
-            const entry = JSON.parse(line);
-            if (entry.type !== "tokens")
-                continue;
-            if (since && entry._ts < since)
-                continue;
-            if (!entry.input && !entry.output)
+    const fd = openSync(LOG_FILE, "r");
+    const stat = statSync(LOG_FILE);
+    const fileSize = stat.size;
+    const CHUNK_SIZE = 64 * 1024; // 64KB chunks
+    const buffer = Buffer.alloc(CHUNK_SIZE);
+    let filePos = fileSize;
+    let leftover = "";
+    let shouldStop = false;
+    while (filePos > 0 && !shouldStop) {
+        const readLength = Math.min(CHUNK_SIZE, filePos);
+        filePos -= readLength;
+        readSync(fd, buffer, 0, readLength, filePos);
+        const chunkStr = buffer.toString("utf8", 0, readLength) + leftover;
+        const lines = chunkStr.split("\n");
+        // The leftmost line could be cut off, save it for the next chunk read to the left
+        leftover = lines[0];
+        // Iterate lines in reverse order (from end to start)
+        for (let i = lines.length - 1; i >= 1; i--) {
+            const line = lines[i].trim();
+            if (!line)
                 continue;
-            entries.push(entry);
+            try {
+                const entry = JSON.parse(line);
+                if (entry.type !== "tokens")
+                    continue;
+                // Early break pruning: once we hit a record older than the threshold,
+                // we can safely stop reading earlier history thanks to monotonic time progression in JSONL.
+                if (since && entry._ts < since) {
+                    shouldStop = true;
+                    break;
+                }
+                if (!entry.input && !entry.output)
+                    continue;
+                entries.push(entry);
+            }
+            catch {
+                // Skip malformed lines
+            }
+        }
+    }
+    // Include the very first line at the top
+    if (!shouldStop && leftover.trim()) {
+        try {
+            const entry = JSON.parse(leftover.trim());
+            if (entry.type === "tokens" && (!since || entry._ts >= since) && (entry.input || entry.output)) {
+                entries.push(entry);
+            }
         }
         catch {
-            // Skip malformed lines
+            // Ignore
         }
     }
-    return entries;
+    closeSync(fd);
+    // Re-establish original chronological order
+    return entries.reverse();
 }
 function loadConfig() {
     try {

package/dist/index.js

@@ -1,5 +1,5 @@
 import { BUILTIN_PRICING, DEFAULT_CONFIG, findModelConfigPricing, formatCost, formatTokens, getStartOfDay, getStartOfWeek, getStartOfMonth, validateConfig } from "./lib/shared.js";
-import { appendFileSync, existsSync, mkdirSync, readFileSync } from "fs";
+import { appendFileSync, existsSync, mkdirSync, readFileSync, statSync } from "fs";
 import { join } from "path";
 import { homedir } from "os";
 const CONFIG_DIR = join(homedir(), ".config", "opencode");
@@ -11,6 +11,8 @@ const LOG_FILE = join(LOG_DIR, "tokens.jsonl");
 // ============================================================================
 let config = DEFAULT_CONFIG;
 let configWarnings = [];
+let lastConfigLoadTime = 0;
+let lastConfigMtime = 0;
 function loadConfig() {
     try {
         if (existsSync(CONFIG_FILE)) {
@@ -27,6 +29,26 @@ function loadConfig() {
     }
     return DEFAULT_CONFIG;
 }
+function ensureLatestConfig() {
+    const now = Date.now();
+    if (now - lastConfigLoadTime < 2000) {
+        return;
+    }
+    lastConfigLoadTime = now;
+    try {
+        if (existsSync(CONFIG_FILE)) {
+            const stat = statSync(CONFIG_FILE);
+            const mtime = stat.mtimeMs;
+            if (mtime !== lastConfigMtime) {
+                config = loadConfig();
+                lastConfigMtime = mtime;
+            }
+        }
+    }
+    catch {
+        // Keep current config on error
+    }
+}
 // ============================================================================
 // Pricing
 // ============================================================================
@@ -270,6 +292,10 @@ export const TokenTrackerPlugin = async ({ directory, client }) => {
     try {
         // Load config on plugin init (with validation)
         config = loadConfig();
+        lastConfigLoadTime = Date.now();
+        if (existsSync(CONFIG_FILE)) {
+            lastConfigMtime = statSync(CONFIG_FILE).mtimeMs;
+        }
         // Initialize in-memory budget tracker (reads JSONL once)
         initBudgetTracker();
         logJson({ type: "init", directory, configLoaded: existsSync(CONFIG_FILE) });
@@ -292,6 +318,7 @@ export const TokenTrackerPlugin = async ({ directory, client }) => {
                 try {
                     // Handle message updates (token tracking)
                     if (event.type === "message.updated") {
+                        ensureLatestConfig();
                         const props = event.properties;
                         const info = props?.info;
                         if (!info?.tokens)
@@ -376,6 +403,7 @@ export const TokenTrackerPlugin = async ({ directory, client }) => {
                     }
                     // Handle session idle (show summary)
                     if (event.type === "session.idle") {
+                        ensureLatestConfig();
                         if (!config.toast.enabled || !config.toast.showOnIdle)
                             return;
                         const props = event.properties;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "opencode-token-tracker",
-  "version": "1.5.2",
+  "version": "1.5.4",
   "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",