awarts 0.2.1 → 0.2.3

package/dist/index.js

@@ -5879,11 +5879,35 @@ import fs8 from "node:fs/promises";
 import path4 from "node:path";
 import os5 from "node:os";
 var CLAUDE_DIR = path4.join(os5.homedir(), ".claude");
-var USAGE_DIR = path4.join(CLAUDE_DIR, "usage");
-var ALT_USAGE_DIR = path4.join(CLAUDE_DIR, "projects");
-function dateFromFilename(filename) {
-  const match = filename.match(/^(\d{4}-\d{2}-\d{2})\.json$/);
-  return match ? match[1] : null;
+var STATS_CACHE = path4.join(CLAUDE_DIR, "stats-cache.json");
+var MODEL_PRICING = {
+  "claude-opus-4-6": { input: 15, output: 75, cacheRead: 1.875, cacheWrite: 18.75 },
+  "claude-opus-4": { input: 15, output: 75, cacheRead: 1.875, cacheWrite: 18.75 },
+  "claude-sonnet-4": { input: 3, output: 15, cacheRead: 0.3, cacheWrite: 3.75 },
+  "claude-sonnet-3-5": { input: 3, output: 15, cacheRead: 0.3, cacheWrite: 3.75 },
+  "claude-haiku-3-5": { input: 0.8, output: 4, cacheRead: 0.08, cacheWrite: 1 }
+};
+var DEFAULT_PRICING = { input: 3, output: 15, cacheRead: 0.3, cacheWrite: 3.75 };
+function getPricing(model) {
+  if (MODEL_PRICING[model])
+    return MODEL_PRICING[model];
+  for (const [key, pricing] of Object.entries(MODEL_PRICING)) {
+    if (model.startsWith(key))
+      return pricing;
+  }
+  return DEFAULT_PRICING;
+}
+function estimateCost(model, inputTokens, outputTokens, cacheReadTokens, cacheCreationTokens) {
+  const p = getPricing(model);
+  return inputTokens / 1e6 * p.input + outputTokens / 1e6 * p.output + cacheReadTokens / 1e6 * p.cacheRead + cacheCreationTokens / 1e6 * p.cacheWrite;
+}
+async function fileExists(filePath) {
+  try {
+    const stat = await fs8.stat(filePath);
+    return stat.isFile();
+  } catch {
+    return false;
+  }
 }
 async function dirExists(dir) {
   try {
@@ -5893,40 +5917,50 @@ async function dirExists(dir) {
     return false;
   }
 }
-async function readUsageDir(dir) {
-  const entries = [];
-  let files;
-  try {
-    files = await fs8.readdir(dir);
-  } catch {
-    return entries;
+async function readStatsCache() {
+  const raw = await fs8.readFile(STATS_CACHE, "utf-8");
+  const cache = JSON.parse(raw);
+  const dailyTokens = cache.dailyModelTokens;
+  if (!dailyTokens || dailyTokens.length === 0)
+    return [];
+  let totalInputTokens = 0;
+  let totalOutputTokens = 0;
+  let totalCacheRead = 0;
+  let totalCacheCreation = 0;
+  let totalCost = 0;
+  if (cache.modelUsage) {
+    for (const [model, usage] of Object.entries(cache.modelUsage)) {
+      totalInputTokens += usage.inputTokens || 0;
+      totalOutputTokens += usage.outputTokens || 0;
+      totalCacheRead += usage.cacheReadInputTokens || 0;
+      totalCacheCreation += usage.cacheCreationInputTokens || 0;
+      if (usage.costUSD > 0) {
+        totalCost += usage.costUSD;
+      } else {
+        totalCost += estimateCost(model, usage.inputTokens || 0, usage.outputTokens || 0, usage.cacheReadInputTokens || 0, usage.cacheCreationInputTokens || 0);
+      }
+    }
   }
-  const jsonFiles = files.filter((f) => f.endsWith(".json"));
-  for (const file of jsonFiles) {
-    try {
-      const filePath = path4.join(dir, file);
-      const raw = await fs8.readFile(filePath, "utf-8");
-      const data = JSON.parse(raw);
-      const date = data.date ?? dateFromFilename(file);
-      if (!date)
-        continue;
-      const costUsd = data.cost_usd ?? data.totalCost ?? 0;
-      const inputTokens = data.input_tokens ?? data.totalInputTokens ?? 0;
-      const outputTokens = data.output_tokens ?? data.totalOutputTokens ?? 0;
-      const cacheCreation = data.cache_creation_tokens ?? data.cacheCreationTokens ?? 0;
-      const cacheRead = data.cache_read_tokens ?? data.cacheReadTokens ?? 0;
-      const models = data.models ? data.models : data.model ? [data.model] : [];
-      entries.push({
-        date,
-        provider: "claude",
-        cost_usd: Number(costUsd) || 0,
-        input_tokens: Number(inputTokens) || 0,
-        output_tokens: Number(outputTokens) || 0,
-        cache_creation_tokens: Number(cacheCreation) || 0,
-        cache_read_tokens: Number(cacheRead) || 0,
-        models
-      });
-    } catch {}
+  const dailyOutputSum = dailyTokens.reduce((sum, day) => {
+    return sum + Object.values(day.tokensByModel).reduce((s, t) => s + t, 0);
+  }, 0);
+  const entries = [];
+  for (const day of dailyTokens) {
+    const dayOutputTokens = Object.values(day.tokensByModel).reduce((s, t) => s + t, 0);
+    if (dayOutputTokens === 0)
+      continue;
+    const share = dailyOutputSum > 0 ? dayOutputTokens / dailyOutputSum : 0;
+    const models = Object.keys(day.tokensByModel).filter((m) => day.tokensByModel[m] > 0);
+    entries.push({
+      date: day.date,
+      provider: "claude",
+      output_tokens: dayOutputTokens,
+      input_tokens: Math.round(totalInputTokens * share),
+      cache_read_tokens: Math.round(totalCacheRead * share),
+      cache_creation_tokens: Math.round(totalCacheCreation * share),
+      cost_usd: Number((totalCost * share).toFixed(4)),
+      models
+    });
   }
   return entries;
 }
@@ -5934,21 +5968,18 @@ var claudeAdapter = {
   name: "claude",
   displayName: "Claude",
   async detect() {
-    if (await dirExists(USAGE_DIR))
+    if (await fileExists(STATS_CACHE))
       return true;
     if (await dirExists(CLAUDE_DIR))
       return true;
-    if (await dirExists(ALT_USAGE_DIR))
-      return true;
     return false;
   },
   async read() {
-    const entries = [];
-    entries.push(...await readUsageDir(USAGE_DIR));
-    if (entries.length === 0) {
-      entries.push(...await readUsageDir(ALT_USAGE_DIR));
+    try {
+      return await readStatsCache();
+    } catch {
+      return [];
     }
-    return entries;
   }
 };
 
@@ -5962,7 +5993,7 @@ var CANDIDATE_DIRS = [
   path5.join(HOME, ".openai-codex", "usage"),
   path5.join(HOME, ".codex")
 ];
-function dateFromFilename2(filename) {
+function dateFromFilename(filename) {
   const match = filename.match(/^(\d{4}-\d{2}-\d{2})\.json$/);
   return match ? match[1] : null;
 }
@@ -6004,7 +6035,7 @@ var codexAdapter = {
         const filePath = path5.join(dir, file);
         const raw = await fs9.readFile(filePath, "utf-8");
         const data = JSON.parse(raw);
-        const date = data.date ?? dateFromFilename2(file);
+        const date = data.date ?? dateFromFilename(file);
         if (!date)
           continue;
         const costUsd = data.cost_usd ?? data.total_cost ?? 0;
@@ -6037,7 +6068,7 @@ var CANDIDATE_DIRS2 = [
   path6.join(HOME2, ".config", "gemini", "usage"),
   path6.join(HOME2, ".gemini")
 ];
-function dateFromFilename3(filename) {
+function dateFromFilename2(filename) {
   const match = filename.match(/^(\d{4}-\d{2}-\d{2})\.json$/);
   return match ? match[1] : null;
 }
@@ -6079,7 +6110,7 @@ var geminiAdapter = {
         const filePath = path6.join(dir, file);
         const raw = await fs10.readFile(filePath, "utf-8");
         const data = JSON.parse(raw);
-        const date = data.date ?? dateFromFilename3(file);
+        const date = data.date ?? dateFromFilename2(file);
         if (!date)
           continue;
         const costUsd = data.cost_usd ?? data.total_cost ?? 0;
@@ -6111,7 +6142,7 @@ var CANDIDATE_DIRS3 = [
   path7.join(HOME3, ".antigravity", "usage"),
   path7.join(HOME3, ".antigravity")
 ];
-function dateFromFilename4(filename) {
+function dateFromFilename3(filename) {
   const match = filename.match(/^(\d{4}-\d{2}-\d{2})\.json$/);
   return match ? match[1] : null;
 }
@@ -6153,7 +6184,7 @@ var antigravityAdapter = {
         const filePath = path7.join(dir, file);
         const raw = await fs11.readFile(filePath, "utf-8");
         const data = JSON.parse(raw);
-        const date = data.date ?? dateFromFilename4(file);
+        const date = data.date ?? dateFromFilename3(file);
         if (!date)
           continue;
         const costUsd = data.cost_usd ?? data.total_cost ?? 0;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "awarts",
-  "version": "0.2.1",
+  "version": "0.2.3",
   "description": "Track your AI coding across Claude, Codex, Gemini & Antigravity",
   "type": "module",
   "bin": {