ai-usage-analyzer 0.4.0 → 0.5.1

package/src/render.js

@@ -72,9 +72,10 @@ export function fmtCompact(n) {
   return String(n);
 }
 
-export function fmtCost(n) {
+export function fmtCost(n, status) {
+  if (status === "unavailable") return "n/a";
   if (!n) return "—";
-  return "$" + Number(n).toFixed(2);
+  return (status === "estimated" || status === "partial" ? "~" : "") + "$" + Number(n).toFixed(2) + (status === "partial" ? "*" : "");
 }
 function fmtDuration(ms) {
   if (!ms || ms <= 0) return chalk.dim("—");
@@ -175,6 +176,7 @@ export function renderHeader({
   totalSessions,
   totalTokens,
   totalCost,
+  costStatus,
   dateRange,
 }) {
   const title = gradient.pastel.multiline("◆  AI TOKEN ANALYZER  ◆");
@@ -183,7 +185,7 @@ export function renderHeader({
     "  •  " +
     chalk.bold.cyan(`${fmtCompact(totalTokens)} total tokens`) +
     (totalCost
-      ? "  •  " + chalk.bold.yellow(`$${totalCost.toFixed(2)} USD`)
+      ? "  •  " + chalk.bold.yellow(`${fmtCost(totalCost, costStatus)} USD`)
       : "");
   const range =
     dateRange[0] && dateRange[1]
@@ -338,7 +340,7 @@ export function renderOverview(records, detections) {
       fmtInt(p.n),
       fmtCompact(p.tokensTotal),
       fmtCompact(p.avg),
-      fmtCost(p.cost),
+      fmtCost(p.cost, p.costStatus),
     ]);
   }
   pt.push([
@@ -346,15 +348,18 @@ export function renderOverview(records, detections) {
     chalk.bold(fmtInt(records.length)),
     chalk.bold(fmtCompact(t.tokensTotal)),
     chalk.bold(fmtCompact(t.avg)),
-    chalk.bold(fmtCost(t.cost)),
+    chalk.bold(fmtCost(t.cost, t.costStatus)),
   ]);
 
   const body =
     pt.toString() +
+    "\n\n" + `Average daily tokens: ${fmtCompact(t.avgDailyTokens)}` +
     "\n\n" +
     chalk.bold.underline("Token Breakdown") +
     "\n" +
-    bt.toString();
+    bt.toString() +
+    "\n\n" + chalk.dim("~ estimated API cost; * partial; n/a unavailable") +
+    (t.unknownModels.length ? "\n" + chalk.dim(`Unpriced models: ${t.unknownModels.map(m => String(m).replace(/[\x00-\x1f\x7f-\x9f]/g, "")).join(", ")}`) : "");
   return boxen(body, {
     title: chalk.bold("Overview"),
     borderStyle: "round",
@@ -526,7 +531,7 @@ export function renderProjectSessions(records, projectName) {
         colorize(r.tool, toolColor(r.tool)),
         dur,
         fmtCompact(r.tokensTotal),
-        r.cost ? "$" + r.cost.toFixed(4) : "—",
+        fmtCost(r.cost, r.costStatus),
         titleStr,
       ]);
     } else {
@@ -539,7 +544,7 @@ export function renderProjectSessions(records, projectName) {
         fmtCompact(r.tokensInput),
         fmtCompact(r.tokensOutput),
         fmtCompact(cache),
-        r.cost ? "$" + r.cost.toFixed(4) : "—",
+        fmtCost(r.cost, r.costStatus),
         titleStr,
       ]);
     }
@@ -600,7 +605,7 @@ export function renderPerTool(records) {
     }
     row.push(
       fmtCompact(p.tokensTotal),
-      fmtCost(p.cost),
+      fmtCost(p.cost, p.costStatus),
       fmtCompact(p.avg),
       bar(p.tokensTotal, max, barW),
     );
@@ -796,7 +801,7 @@ export function renderTopSessions(records, n = 5) {
         chalk.bold(String(i + 1)),
         chalk.bold.yellow(fmtCompact(r.tokensTotal)),
         `${fmtCompact(r.tokensInput)}/${fmtCompact(r.tokensOutput)}`,
-        r.cost ? "$" + r.cost.toFixed(4) : "—",
+        fmtCost(r.cost, r.costStatus),
         truncEnd(r.title, 30) || "—",
       ]);
     });
@@ -830,7 +835,7 @@ export function renderTopSessions(records, n = 5) {
       chalk.bold.yellow(fmtCompact(r.tokensTotal)),
       fmtCompact(r.tokensInput),
       fmtCompact(r.tokensOutput),
-      r.cost ? "$" + r.cost.toFixed(4) : "—",
+      fmtCost(r.cost, r.costStatus),
       shortModel(r.model),
       truncEnd(r.project, 30),
       truncEnd(r.title, 40) || "—",
@@ -850,6 +855,7 @@ export function renderTopSessions(records, n = 5) {
 
 export function renderNotes(detections, errors) {
   const lines = [];
+  lines.push(chalk.dim("Costs: ~ estimated API rates; * partial, excludes unpriced models; n/a unavailable."));
   lines.push(chalk.bold.underline("Token definitions"));
   lines.push(
     `  • ${chalk.cyan("Input")}        — prompt tokens (what was sent to the model)`,
@@ -861,7 +867,7 @@ export function renderNotes(detections, errors) {
     `  • ${chalk.cyan("Cache Read")}   — prompt tokens served from provider cache (cheap)`,
   );
   lines.push(
-    `  • ${chalk.cyan("Cache Write")}  — prompt tokens cached for future use (OpenCode only)`,
+    `  • ${chalk.cyan("Cache Write")}  — prompt tokens cached for future use`,
   );
   lines.push(
     `  • ${chalk.yellow("Reasoning")}    — extended thinking / chain-of-thought tokens`,
@@ -902,9 +908,9 @@ export function renderNotes(detections, errors) {
   lines.push("");
   lines.push(chalk.bold.underline("Env overrides"));
   lines.push(
-    "  $CLAUDE_HOME    $CODEX_HOME    $OPENCODE_HOME    $MIMOCODE_HOME",
+    "  $CLAUDE_HOME    $CODEX_HOME    $OPENCODE_HOME",
   );
-  lines.push("  $COPILOT_HOME   $ANTIGRAVITY_HOME    $GEMINI_HOME");
+  lines.push("  $ANTIGRAVITY_HOME    $GEMINI_HOME");
   lines.push(
     "  $AI_USAGE_PATHS_JSON = " + chalk.italic('\'{"codex":"/custom/path"}\''),
   );

package/src/server.js

@@ -0,0 +1,68 @@
+import { createServer } from 'node:http';
+import { readFile } from 'node:fs/promises';
+import { spawn } from 'node:child_process';
+import { detectAll } from './detectors.js';
+import { loadAll } from './loaders.js';
+
+const assets = new Map([
+  ['/', ['../web/index.html', 'text/html; charset=utf-8']],
+  ['/app.js', ['../web/app.js', 'text/javascript; charset=utf-8']],
+  ['/dates.js', ['../web/dates.js', 'text/javascript; charset=utf-8']],
+  ['/aggregate.js', ['./aggregate.js', 'text/javascript; charset=utf-8']],
+  ['/style.css', ['../web/style.css', 'text/css; charset=utf-8']],
+  ['/favicon.svg', ['../web/favicon.svg', 'image/svg+xml']],
+  ['/gsap.js', [import.meta.resolve('gsap/dist/gsap.min.js'), 'text/javascript; charset=utf-8']],
+  ['/fonts/manrope.woff2', ['../web/fonts/manrope.woff2', 'font/woff2']],
+  ['/icons/claude.svg', ['../web/icons/claude.svg', 'image/svg+xml']],
+  ['/icons/openai.svg', ['../web/icons/openai.svg', 'image/svg+xml']],
+  ['/icons/opencode.svg', ['../web/icons/opencode.svg', 'image/svg+xml']],
+]);
+
+export function createDashboardServer() {
+  return createServer(async (req, res) => {
+    const host = req.headers.host || '';
+    if (!/^127\.0\.0\.1:\d+$/.test(host) || (req.headers.origin && req.headers.origin !== `http://${host}`) || req.headers['sec-fetch-site'] === 'cross-site') {
+      res.writeHead(403).end('Forbidden');
+      return;
+    }
+    res.setHeader('X-Content-Type-Options', 'nosniff');
+    res.setHeader('Cache-Control', 'no-store');
+    res.setHeader('Content-Security-Policy', "default-src 'self'; style-src 'self'; script-src 'self'; connect-src 'self'; base-uri 'none'; frame-ancestors 'none'");
+    if (req.method !== 'GET') {
+      res.writeHead(405, { Allow: 'GET' }).end();
+      return;
+    }
+    try {
+      const path = new URL(req.url, `http://${host}`).pathname;
+      if (path === '/api/usage') {
+        const detections = detectAll();
+        const { records, errors } = await loadAll(detections);
+        res.writeHead(200, { 'Content-Type': 'application/json; charset=utf-8' }).end(JSON.stringify({ detections, records, errors, generatedAt: new Date().toISOString() }));
+      } else if (assets.has(path)) {
+        const [file, type] = assets.get(path);
+        const content = await readFile(new URL(file, import.meta.url));
+        res.writeHead(200, { 'Content-Type': type }).end(content);
+      } else {
+        res.writeHead(404).end('Not found');
+      }
+    } catch {
+      res.writeHead(500, { 'Content-Type': 'application/json' }).end(JSON.stringify({ error: 'Could not load local usage data.' }));
+    }
+  });
+}
+
+export async function startDashboard() {
+  const server = createDashboardServer();
+  await new Promise((resolve, reject) => {
+    server.once('error', reject);
+    server.listen(0, '127.0.0.1', resolve);
+  });
+  const url = `http://127.0.0.1:${server.address().port}/`;
+  console.log(`AI Usage Analyzer is running at ${url}\nPress Ctrl+C to stop.`);
+  const command = process.platform === 'darwin' ? 'open' : process.platform === 'win32' ? 'cmd' : 'xdg-open';
+  const args = process.platform === 'win32' ? ['/c', 'start', '', url] : [url];
+  const child = spawn(command, args, { detached: true, stdio: 'ignore' });
+  child.on('error', () => {});
+  child.unref();
+  return server;
+}

package/src/tools.js

@@ -8,7 +8,7 @@
 // data, also wire a loader branch in src/loaders.js.
 
 import { statSync, readdirSync, existsSync } from 'node:fs';
-import { join } from 'node:path';
+import { basename, dirname, join } from 'node:path';
 import { homedir, platform } from 'node:os';
 import { env } from 'node:process';
 import { createRequire } from 'node:module';
@@ -112,10 +112,10 @@ export const TOOLS = [
     barChar: '▓',
     kind: 'jsonl-rollout',
     envVar: 'CODEX_HOME',
-    candidatePaths: () => [
-      env.CODEX_HOME || join(HOME, '.codex', 'sessions'),
-    ],
-    count: countJsonlRecursive,
+    candidatePaths: () => env.CODEX_HOME
+      ? [env.CODEX_HOME]
+      : [join(HOME, '.codex', 'sessions'), join(HOME, '.codex', 'archived_sessions')],
+    count: path => countJsonlRecursive(path) + (basename(path) === 'sessions' ? countJsonlRecursive(join(dirname(path), 'archived_sessions')) : 0),
     hasTokens: true,
     description: '~/.codex/sessions/YYYY/MM/DD/rollout-*.jsonl',
   },
@@ -141,50 +141,6 @@ export const TOOLS = [
     description: '~/.local/share/opencode/opencode.db  (tokens + cost)',
   },
 
-  // -----------------------------------------------------------------------
-  // MimoCode (same schema as OpenCode)
-  // -----------------------------------------------------------------------
-  {
-    key: 'mimocode',
-    name: 'MimoCode',
-    label: 'mimocode',
-    color: '#f1fa8c',     // yellow
-    barChar: '░',
-    kind: 'sqlite',
-    envVar: 'MIMOCODE_HOME',
-    candidatePaths: () => [
-      env.MIMOCODE_HOME
-        ? join(env.MIMOCODE_HOME, 'mimocode.db')
-        : join(HOME, '.local', 'share', 'mimocode', 'mimocode.db'),
-    ],
-    count: countSqliteRows,
-    hasTokens: true,
-    description: '~/.local/share/mimocode/mimocode.db  (tokens + cost)',
-  },
-
-  // -----------------------------------------------------------------------
-  // GitHub Copilot CLI (presence only)
-  // -----------------------------------------------------------------------
-  {
-    key: 'copilot',
-    name: 'GitHub Copilot',
-    label: 'copilot',
-    color: '#3b82f6',     // deep blue
-    barChar: '·',
-    kind: 'jsonl-events',
-    envVar: 'COPILOT_HOME',
-    candidatePaths: () => {
-      const base = env.COPILOT_HOME || join(HOME, '.copilot');
-      return [
-        join(base, 'session-state'),
-        base,
-      ];
-    },
-    count: countJsonlRecursive,
-    hasTokens: false,
-    description: '~/.copilot/session-state/*/events.jsonl  (no token data)',
-  },
-
   // -----------------------------------------------------------------------
   // Antigravity (VS Code variant) - mostly cache; no token data
   // -----------------------------------------------------------------------

package/src/tui.js

@@ -5,6 +5,7 @@ import chalk from "chalk";
 import process from "node:process";
 import readline from "node:readline";
 import { perProject } from "./aggregate.js";
+import { renderQuota } from "./quota.js";
 import {
     CURSOR_MARK,
     renderDetections,
@@ -25,11 +26,14 @@ function visLen(s) {
   return s.replace(ANSI_RE, "").length;
 }
 
-export async function runInteractive({ records, detections, errors, topN }) {
+export async function runInteractive({ records, detections, errors, topN, quotaPromise, quotaController, liveTailer }) {
   if (!process.stdout.isTTY || !process.stdin.isTTY) return false;
 
   const PROJECTS_VIEW = 1; // index into VIEWS
 
+  let liveRate = liveTailer ? liveTailer.tick() : null;
+  let quotas = null;
+  let active = true;
   const VIEWS = [
     {
       key: "1",
@@ -53,6 +57,7 @@ export async function runInteractive({ records, detections, errors, topN }) {
     },
     { key: "8", label: "Detections", fn: () => renderDetections(detections) },
     { key: "9", label: "Notes", fn: () => renderNotes(detections, errors) },
+    { key: "0", label: "Quota", fn: () => renderQuota(quotas) },
   ];
 
   // Project drill-down state (only active when current === PROJECTS_VIEW).
@@ -128,6 +133,7 @@ export async function runInteractive({ records, detections, errors, topN }) {
         "  ←/→ h/l switch  ↑/↓ j/k scroll  PgUp/PgDn  g/G top/bot  q quit",
       );
     }
+    if (liveRate !== null) hint = chalk.dim(` ${Math.round(liveRate)} tokens/min (10m) `) + hint;
     const gap = Math.max(1, w - visLen(hint) - visLen(pct));
     return chalk.bgBlack(hint + " ".repeat(gap) + pct);
   }
@@ -195,20 +201,29 @@ export async function runInteractive({ records, detections, errors, topN }) {
   process.stdout.write("\x1b[?1049h\x1b[?25l");
   draw(true);
 
-  process.stdout.on("resize", () => {
-    draw(true);
+  const liveTimer = liveTailer ? setInterval(() => { liveRate = liveTailer.tick(); if (active) draw(false); }, 5000) : null;
+  const onResize = () => draw(true);
+  process.stdout.on("resize", onResize);
+  if (quotaPromise) quotaPromise.then((result) => {
+    quotas = result;
+    if (active && current === 9) { loadView(); draw(false); }
   });
 
   return new Promise((resolve) => {
     function quit() {
       // Leave alternate screen — original terminal content is restored automatically.
+      active = false;
+      if (liveTimer) clearInterval(liveTimer);
+      quotaController?.abort();
+      process.stdout.removeListener("resize", onResize);
+      process.stdin.removeListener("keypress", onKeypress);
       process.stdout.write("\x1b[?25h\x1b[?1049l");
       process.stdin.setRawMode(false);
       process.stdin.pause();
-      resolve();
+      resolve(true);
     }
 
-    process.stdin.on("keypress", (_str, key) => {
+    function onKeypress(_str, key) {
       if (!key) return;
       const usable = usableRows();
       const maxScroll = Math.max(0, lines.length - usable);
@@ -270,8 +285,8 @@ export async function runInteractive({ records, detections, errors, topN }) {
           draw(false);
           return;
         }
-        if (_str && /^[1-9]$/.test(_str)) {
-          const idx = parseInt(_str, 10) - 1;
+        if (_str && /^[0-9]$/.test(_str)) {
+          const idx = _str === "0" ? 9 : parseInt(_str, 10) - 1;
           if (idx < VIEWS.length) {
             current = idx;
             if (current === PROJECTS_VIEW) proj.mode = "list";
@@ -305,6 +320,7 @@ export async function runInteractive({ records, detections, errors, topN }) {
       }
 
       if (moved) draw(false);
-    });
+    }
+    process.stdin.on("keypress", onKeypress);
   });
 }