awarts 0.2.1 → 0.2.2

package/dist/index.js

@@ -5879,11 +5879,14 @@ 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");
+async function fileExists(filePath) {
+  try {
+    const stat = await fs8.stat(filePath);
+    return stat.isFile();
+  } catch {
+    return false;
+  }
 }
 async function dirExists(dir) {
   try {
@@ -5893,40 +5896,46 @@ async function dirExists(dir) {
     return false;
   }
 }
-async function readUsageDir(dir) {
+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 usage of Object.values(cache.modelUsage)) {
+      totalInputTokens += usage.inputTokens || 0;
+      totalOutputTokens += usage.outputTokens || 0;
+      totalCacheRead += usage.cacheReadInputTokens || 0;
+      totalCacheCreation += usage.cacheCreationInputTokens || 0;
+      totalCost += usage.costUSD || 0;
+    }
+  }
+  const dailyOutputSum = dailyTokens.reduce((sum, day) => {
+    return sum + Object.values(day.tokensByModel).reduce((s, t) => s + t, 0);
+  }, 0);
   const entries = [];
-  let files;
-  try {
-    files = await fs8.readdir(dir);
-  } catch {
-    return entries;
-  }
-  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 {}
+  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 +5943,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 +5968,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 +6010,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 +6043,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 +6085,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 +6117,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 +6159,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.2",
   "description": "Track your AI coding across Claude, Codex, Gemini & Antigravity",
   "type": "module",
   "bin": {