opencode-token-tracker 1.6.4 → 1.7.0

package/dist/lib/shared.d.ts

@@ -7,6 +7,12 @@ export interface ModelPricing {
 export interface ProviderModelPricingMap {
     [provider: string]: ModelPricing;
 }
+export declare const BUILTIN_PRICING_META: {
+    readonly pricingLastUpdated: "2026-02-11";
+    readonly metadataLastUpdated: "2026-05-24";
+    readonly source: "Provider official pricing pages";
+    readonly notes: "Manually maintained. Report stale prices: https://github.com/tongsh6/opencode-token-tracker/issues/new";
+};
 export declare const BUILTIN_PRICING: Record<string, ModelPricing>;
 export declare function formatCost(cost: number): string;
 /**
@@ -46,3 +52,22 @@ export interface ConfigValidationResult {
  */
 export declare function validateConfig(raw: unknown): ConfigValidationResult;
 export declare function findModelConfigPricing(models: TrackerConfig["models"], model: string, provider: string, partial?: boolean): ModelPricing | undefined;
+export interface BudgetStatus {
+    period: "daily" | "weekly" | "monthly";
+    spent: number;
+    limit: number;
+    percentage: number;
+    exceeded: boolean;
+    warning: boolean;
+}
+export interface BudgetSpentSnapshot {
+    dailySpent: number;
+    weeklySpent: number;
+    monthlySpent: number;
+}
+export declare function evaluateBudgetStatus(budget: BudgetConfig, spent: BudgetSpentSnapshot, initialized: boolean): BudgetStatus | null;
+export declare function resolvePricingStatus(config: TrackerConfig, model: string, provider: string): "provider cfg" | "model cfg" | "built-in" | "default";
+export declare function round2(val: number): number;
+export type ProviderFamily = "anthropic" | "openai" | "deepseek" | "google" | "other";
+export declare function getProviderFamily(model: string, provider: string): ProviderFamily;
+export declare function calculateCost(model: string, provider: string, input: number, output: number, cacheRead?: number, cacheWrite?: number, config?: TrackerConfig): number;

package/dist/lib/shared.js

@@ -9,6 +9,12 @@
 // - DeepSeek: https://api-docs.deepseek.com/quick_start/pricing
 // - Google: https://cloud.google.com/vertex-ai/generative-ai/pricing
 // ============================================================================
+export const BUILTIN_PRICING_META = {
+    pricingLastUpdated: "2026-02-11",
+    metadataLastUpdated: "2026-05-24",
+    source: "Provider official pricing pages",
+    notes: "Manually maintained. Report stale prices: https://github.com/tongsh6/opencode-token-tracker/issues/new",
+};
 export const BUILTIN_PRICING = {
     // Anthropic Claude (https://www.anthropic.com/pricing#api)
     // Opus 4.6: $5 input, $25 output (≤200K), cache write $6.25, cache read $0.50
@@ -47,10 +53,10 @@ export const BUILTIN_PRICING = {
     "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 },
+    // DeepSeek-V4 Flash compatibility pricing for deepseek-chat / deepseek-reasoner.
+    // $0.14 input (cache miss), $0.0028 input (cache hit), $0.28 output
+    "deepseek-chat": { input: 0.14, output: 0.28, cacheRead: 0.0028 },
+    "deepseek-reasoner": { input: 0.14, output: 0.28, cacheRead: 0.0028 },
     // 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 },
@@ -361,3 +367,149 @@ function validateBudget(raw, warnings) {
     }
     return result;
 }
+export function evaluateBudgetStatus(budget, spent, initialized) {
+    if (!initialized)
+        return null;
+    if (!budget.daily && !budget.weekly && !budget.monthly)
+        return null;
+    const warnAt = budget.warnAt ?? 0.8;
+    const candidates = [];
+    const evaluate = (period, limit, spentAmount) => {
+        if (!limit)
+            return;
+        const percentage = spentAmount / limit;
+        candidates.push({
+            period,
+            spent: spentAmount,
+            limit,
+            percentage,
+            exceeded: percentage >= 1,
+            warning: percentage >= warnAt && percentage < 1,
+        });
+    };
+    evaluate("daily", budget.daily, spent.dailySpent);
+    evaluate("weekly", budget.weekly, spent.weeklySpent);
+    evaluate("monthly", budget.monthly, spent.monthlySpent);
+    // Severity sort priority: exceeded (2) > warning (1) > ok (0); tie-breaker: percentage desc
+    const severity = (s) => s.exceeded ? 2 : s.warning ? 1 : 0;
+    candidates.sort((a, b) => severity(b) - severity(a) || b.percentage - a.percentage);
+    return candidates[0] ?? null;
+}
+export function resolvePricingStatus(config, model, provider) {
+    // Step 1: config.providers[provider] -> "provider cfg"
+    if (config.providers && config.providers[provider]) {
+        return "provider cfg";
+    }
+    // Step 2: Exact-match in config.models (partial=false) -> "model cfg"
+    if (findModelConfigPricing(config.models, model, provider, false)) {
+        return "model cfg";
+    }
+    // Step 3: Exact-match in BUILTIN_PRICING -> "built-in"
+    if (BUILTIN_PRICING[model]) {
+        return "built-in";
+    }
+    // Step 4: Partial-match in BUILTIN_PRICING (sorted by key length desc) -> "built-in"
+    const modelLower = model.toLowerCase();
+    const sortedBuiltin = Object.keys(BUILTIN_PRICING)
+        .filter(k => k !== "_default")
+        .sort((a, b) => b.length - a.length);
+    for (const key of sortedBuiltin) {
+        if (modelLower.includes(key.toLowerCase())) {
+            return "built-in";
+        }
+    }
+    // Step 5: Partial-match in config.models (sorted by key length desc) -> "model cfg"
+    if (findModelConfigPricing(config.models, model, provider, true)) {
+        return "model cfg";
+    }
+    // Step 6: Default fallback -> "default"
+    return "default";
+}
+export function round2(val) {
+    return Math.round(val * 100) / 100;
+}
+export function getProviderFamily(model, provider) {
+    const p = provider.toLowerCase();
+    const m = model.toLowerCase();
+    if (p.includes("anthropic") || m.startsWith("claude-")) {
+        return "anthropic";
+    }
+    if (p.includes("openai") ||
+        m.startsWith("gpt-") ||
+        m.startsWith("o1-") ||
+        m.startsWith("o3-") ||
+        m.startsWith("o4-") ||
+        m.startsWith("chatgpt-") ||
+        m === "o3" ||
+        m === "o1") {
+        return "openai";
+    }
+    if (p.includes("deepseek") || m.includes("deepseek")) {
+        return "deepseek";
+    }
+    if (p.includes("google") || p.includes("vertex") || m.startsWith("gemini-")) {
+        return "google";
+    }
+    return "other";
+}
+function getModelPricing(model, provider, config) {
+    if (config && config.providers && config.providers[provider]) {
+        return config.providers[provider];
+    }
+    if (config && config.models) {
+        const configuredPricing = findModelConfigPricing(config.models, model, provider, false);
+        if (configuredPricing) {
+            return configuredPricing;
+        }
+    }
+    if (BUILTIN_PRICING[model]) {
+        return BUILTIN_PRICING[model];
+    }
+    const modelLower = model.toLowerCase();
+    const sortedKeys = Object.keys(BUILTIN_PRICING)
+        .filter(k => k !== "_default")
+        .sort((a, b) => b.length - a.length);
+    for (const key of sortedKeys) {
+        if (modelLower.includes(key.toLowerCase())) {
+            return BUILTIN_PRICING[key];
+        }
+    }
+    if (config && config.models) {
+        const partialUserPricing = findModelConfigPricing(config.models, model, provider, true);
+        if (partialUserPricing) {
+            return partialUserPricing;
+        }
+    }
+    return BUILTIN_PRICING["_default"];
+}
+export function calculateCost(model, provider, input, output, cacheRead = 0, cacheWrite = 0, config) {
+    const pricing = getModelPricing(model, provider, config);
+    const family = getProviderFamily(model, provider);
+    let defaultCacheReadRate = 0.5; // Default 50% discount (OpenAI style)
+    let defaultCacheWriteRate = 0; // Default free cache writing
+    if (family === "anthropic") {
+        defaultCacheReadRate = 0.1;
+        defaultCacheWriteRate = 1.25;
+    }
+    else if (family === "deepseek" || family === "google") {
+        defaultCacheReadRate = 0.1;
+        defaultCacheWriteRate = 0;
+    }
+    else if (family === "openai") {
+        defaultCacheReadRate = 0.5;
+        defaultCacheWriteRate = 0;
+    }
+    else {
+        defaultCacheReadRate = 0.5;
+        defaultCacheWriteRate = 0;
+    }
+    const finalCacheReadPrice = pricing.cacheRead ?? (pricing.input * defaultCacheReadRate);
+    const finalCacheWritePrice = pricing.cacheWrite ?? (pricing.input * defaultCacheWriteRate);
+    // OpenCode exposes net-new input tokens separately from cacheRead tokens.
+    // Charge both fields independently instead of subtracting cacheRead from input.
+    const inputCost = (input / 1_000_000) * pricing.input;
+    const outputCost = (output / 1_000_000) * pricing.output;
+    const cacheReadCost = (cacheRead / 1_000_000) * finalCacheReadPrice;
+    const cacheWriteCost = (cacheWrite / 1_000_000) * finalCacheWritePrice;
+    return inputCost + outputCost + cacheReadCost + cacheWriteCost;
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "opencode-token-tracker",
-  "version": "1.6.4",
+  "version": "1.7.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",
@@ -10,7 +10,7 @@
   },
   "scripts": {
     "build": "tsc",
-    "test": "tsc && node --test dist/test/shared.test.js dist/test/cli.test.js",
+    "test": "tsc && node --test dist/test/*.test.js",
     "prepublishOnly": "npm run build"
   },
   "keywords": [
@@ -45,8 +45,12 @@
     "typescript": "^5.0.0"
   },
   "files": [
-    "dist",
+    "dist/index.js",
+    "dist/index.d.ts",
+    "dist/lib",
+    "dist/bin",
     "README.md",
+    "README.zh-CN.md",
     "LICENSE"
   ],
   "engines": {

package/dist/test/cli.test.d.ts

@@ -1 +0,0 @@
-export {};

package/dist/test/cli.test.js

@@ -1,55 +0,0 @@
-import { describe, it } from "node:test";
-import { strict as assert } from "node:assert";
-import { execSync } from "child_process";
-import { fileURLToPath } from "url";
-import { join, dirname } from "path";
-const __dirname = dirname(fileURLToPath(import.meta.url));
-const CLI = join(__dirname, "..", "bin", "opencode-tokens.js");
-function run(args) {
-    return execSync(`node ${CLI} ${args}`, { encoding: "utf-8", timeout: 10000 });
-}
-describe("CLI help and stats", () => {
-    it("should show help", () => {
-        const out = run("--help");
-        assert.ok(out.includes("opencode-tokens - Token usage statistics CLI"));
-        assert.ok(out.includes("trend"));
-        assert.ok(out.includes("export"));
-        assert.ok(out.includes("config set"));
-    });
-    it("should show stats by session", () => {
-        const out = run("--by session");
-        assert.ok(out.includes("By Session") || out.includes("No data"));
-    });
-    it("should show budget", () => {
-        const out = run("budget");
-        assert.ok(out.includes("Budget Status"));
-    });
-});
-describe("CLI config", () => {
-    it("should set, get, and unset config value", () => {
-        run("config set toast.duration 5000");
-        const out = run("config get toast.duration");
-        assert.ok(out.includes("5000"));
-        run("config unset toast.duration");
-    });
-    it("should reject unknown config key", () => {
-        const out = run("config get nonexistent.key");
-        assert.ok(out.includes("Unknown key"));
-    });
-});
-describe("CLI export", () => {
-    it("should export as CSV", () => {
-        const out = run("export --format csv --period today");
-        assert.ok(out.includes("timestamp") || out.includes("No data"));
-    });
-    it("should export as JSON", () => {
-        const out = run("export --format json --period today");
-        assert.ok(out.startsWith("[") || out.includes("No data"));
-    });
-});
-describe("CLI trend", () => {
-    it("should show trend chart", () => {
-        const out = run("trend --days 7");
-        assert.ok(out.includes("┤") || out.includes("(no data"));
-    });
-});

package/dist/test/shared.test.d.ts

@@ -1 +0,0 @@
-export {};