vibestats 1.0.6 → 1.0.8

package/dist/index.js

@@ -315,37 +315,48 @@ function parseCodexJsonl() {
   const entries = [];
   const codexDir = getCodexDir();
   const sessionsDir = join(codexDir, "sessions");
-  if (!existsSync(sessionsDir)) return entries;
-  const jsonlFiles = findJsonlFiles(sessionsDir);
+  const archivedDir = join(codexDir, "archived_sessions");
+  const jsonlFiles = [
+    ...findJsonlFiles(sessionsDir),
+    ...findJsonlFiles(archivedDir)
+  ];
+  if (jsonlFiles.length === 0) return entries;
   for (const filePath of jsonlFiles) {
     try {
       const content = readFileSync(filePath, "utf-8");
       const lines = content.split("\n");
+      let currentModel = "gpt-5";
       for (const line of lines) {
         if (!line.trim()) continue;
         try {
           const entry = JSON.parse(line);
-          if (entry.type !== "response" || !entry.response?.usage) continue;
-          const usage = entry.response.usage;
-          const model = entry.response.model || "unknown";
-          const timestamp = entry.timestamp;
-          if (!timestamp) continue;
-          const date = timestamp.split("T")[0];
-          const pricing = getCodexModelPricing(model);
-          const inputTokens = usage.input_tokens || 0;
-          const outputTokens = usage.output_tokens || 0;
-          const cachedInputTokens = usage.input_tokens_details?.cached_tokens || 0;
-          const cost = (inputTokens - cachedInputTokens) * pricing.input / 1e6 + outputTokens * pricing.output / 1e6 + cachedInputTokens * pricing.cachedInput / 1e6;
-          entries.push({
-            date,
-            model: getCodexModelDisplayName(model),
-            inputTokens,
-            outputTokens,
-            cacheWriteTokens: 0,
-            cacheReadTokens: cachedInputTokens,
-            cost,
-            source: "codex"
-          });
+          if (entry.type === "turn_context" && entry.payload?.model) {
+            currentModel = entry.payload.model;
+            continue;
+          }
+          if (entry.type === "event_msg" && entry.payload?.type === "token_count") {
+            const info = entry.payload.info;
+            if (!info?.last_token_usage) continue;
+            const usage = info.last_token_usage;
+            const timestamp = entry.timestamp;
+            if (!timestamp) continue;
+            const date = timestamp.split("T")[0];
+            const pricing = getCodexModelPricing(currentModel);
+            const inputTokens = usage.input_tokens || 0;
+            const outputTokens = usage.output_tokens || 0;
+            const cachedInputTokens = usage.cached_input_tokens || 0;
+            const cost = (inputTokens - cachedInputTokens) * pricing.input / 1e6 + outputTokens * pricing.output / 1e6 + cachedInputTokens * pricing.cachedInput / 1e6;
+            entries.push({
+              date,
+              model: getCodexModelDisplayName(currentModel),
+              inputTokens,
+              outputTokens,
+              cacheWriteTokens: 0,
+              cacheReadTokens: cachedInputTokens,
+              cost,
+              source: "codex"
+            });
+          }
         } catch {
         }
       }
@@ -1436,21 +1447,49 @@ function getModelAbbrevFromDisplayName(displayName) {
   };
   return map[displayName] || displayName.slice(0, 5).toLowerCase();
 }
+function aggregateRowsToMonthly(rows) {
+  const monthMap = /* @__PURE__ */ new Map();
+  for (const row of rows) {
+    const month = row.key.slice(0, 7);
+    const existing = monthMap.get(month);
+    if (existing) {
+      existing.inputTokens += row.inputTokens;
+      existing.outputTokens += row.outputTokens;
+      existing.cacheWriteTokens += row.cacheWriteTokens;
+      existing.cacheReadTokens += row.cacheReadTokens;
+      existing.totalTokens += row.totalTokens;
+      existing.cost += row.cost;
+    } else {
+      monthMap.set(month, { ...row, key: month });
+    }
+  }
+  return Array.from(monthMap.values()).sort((a, b) => a.key.localeCompare(b.key));
+}
 function encodeUsageToUrl(stats, baseUrl = "https://vibestats.wolfai.dev") {
   const params = new URLSearchParams();
-  const aggMap = { daily: "d", monthly: "m", model: "mo", total: "t" };
-  params.set("agg", aggMap[stats.aggregation] || "d");
   if (stats.source !== "claude") {
     params.set("src", stats.source);
   }
   const formatDateCompact = (d) => d.replace(/-/g, "");
-  const limitedRows = stats.rows.slice(-31);
-  const startDate = limitedRows[0]?.key || stats.dateRange.start;
-  const endDate = limitedRows[limitedRows.length - 1]?.key || stats.dateRange.end;
+  const startMs = new Date(stats.dateRange.start).getTime();
+  const endMs = new Date(stats.dateRange.end).getTime();
+  const daySpan = Math.ceil((endMs - startMs) / (1e3 * 60 * 60 * 24));
+  const useMonthly = stats.aggregation === "daily" && daySpan > 31;
+  const aggMap = { daily: "d", monthly: "m", model: "mo", total: "t" };
+  const effectiveAgg = useMonthly ? "monthly" : stats.aggregation;
+  params.set("agg", aggMap[effectiveAgg] || "d");
+  let rowsToEncode;
+  if (useMonthly) {
+    rowsToEncode = aggregateRowsToMonthly(stats.rows);
+  } else {
+    rowsToEncode = stats.rows.slice(-31);
+  }
+  const startDate = useMonthly ? stats.dateRange.start : rowsToEncode[0]?.key || stats.dateRange.start;
+  const endDate = useMonthly ? stats.dateRange.end : rowsToEncode[rowsToEncode.length - 1]?.key || stats.dateRange.end;
   params.set("dr", `${formatDateCompact(startDate)}-${formatDateCompact(endDate)}`);
-  const rows = limitedRows.map((row) => {
+  const rows = rowsToEncode.map((row) => {
     let key = row.key;
-    if (stats.aggregation === "daily" && row.key.length === 10) {
+    if (effectiveAgg === "daily" && row.key.length === 10) {
       key = row.key.slice(5).replace("-", "");
     }
     return [

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "vibestats",
-  "version": "1.0.6",
+  "version": "1.0.8",
   "description": "AI coding stats - usage tracking and annual wrapped for Claude Code & Codex",
   "type": "module",
   "main": "./dist/index.js",