whoburnedmore 0.2.0 → 0.4.0

package/dist/index.js

@@ -8,9 +8,9 @@ var __export = (target, all) => {
 // src/index.ts
 import { spawn } from "node:child_process";
 import { mkdirSync as mkdirSync3, writeFileSync as writeFileSync3 } from "node:fs";
-import { createRequire as createRequire2 } from "node:module";
-import { platform as platform2 } from "node:os";
-import { join as join4 } from "node:path";
+import { createRequire as createRequire4 } from "node:module";
+import { platform as platform3 } from "node:os";
+import { join as join6 } from "node:path";
 import { createInterface } from "node:readline/promises";
 import pc2 from "picocolors";
 
@@ -29,16 +29,34 @@ function parseBoard(args) {
 function apiBase() {
   return process.env.WHOBURNEDMORE_API ?? "https://api.whoburnedmore.com";
 }
+async function readJson(res) {
+  const text = await res.text();
+  if (!text) return {};
+  try {
+    return JSON.parse(text);
+  } catch {
+    return {
+      error: res.status >= 500 ? "the leaderboard server is temporarily unavailable \u2014 try again in a minute" : `unexpected response from the server (HTTP ${res.status})`
+    };
+  }
+}
 async function post(path, body, token) {
-  const res = await fetch(`${apiBase()}${path}`, {
-    method: "POST",
-    headers: {
-      "Content-Type": "application/json",
-      ...token ? { Authorization: `Bearer ${token}` } : {}
-    },
-    body: JSON.stringify(body)
-  });
-  return { status: res.status, body: await res.json() };
+  let res;
+  try {
+    res = await fetch(`${apiBase()}${path}`, {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json",
+        ...token ? { Authorization: `Bearer ${token}` } : {}
+      },
+      body: JSON.stringify(body)
+    });
+  } catch {
+    throw new Error(
+      "couldn't reach the leaderboard server \u2014 check your connection and try again"
+    );
+  }
+  return { status: res.status, body: await readJson(res) };
 }
 async function deviceStart() {
   const { status, body } = await post("/v1/auth/device", {});
@@ -266,9 +284,259 @@ function autoSyncInstalled() {
 }
 
 // src/collect.ts
+import { spawnSync as spawnSync4 } from "node:child_process";
+import { createRequire as createRequire3 } from "node:module";
+import { dirname as dirname2, join as join5 } from "node:path";
+
+// src/cursor.ts
+import { spawnSync as spawnSync3 } from "node:child_process";
+import { existsSync as existsSync3 } from "node:fs";
+import { createRequire as createRequire2 } from "node:module";
+import { homedir as homedir3, platform as platform2 } from "node:os";
+import { join as join4 } from "node:path";
+
+// src/tokscale.ts
 import { spawnSync as spawnSync2 } from "node:child_process";
 import { createRequire } from "node:module";
 import { dirname, join as join3 } from "node:path";
+var LOOKBACK_DAYS = 30;
+function num(n) {
+  const v = Math.round(Number(n));
+  return Number.isFinite(v) && v > 0 ? v : 0;
+}
+function numCost(n) {
+  const v = Number(n);
+  return Number.isFinite(v) && v > 0 ? v : 0;
+}
+function mapTokscaleDay(date, json) {
+  const entries = json?.entries;
+  if (!Array.isArray(entries)) return [];
+  const out = [];
+  for (const e of entries) {
+    const inputTokens = num(e.input);
+    const outputTokens = num(e.output) + num(e.reasoning);
+    const cacheCreationTokens = num(e.cacheWrite);
+    const cacheReadTokens = num(e.cacheRead);
+    const costUSD = numCost(e.cost);
+    const total = inputTokens + outputTokens + cacheCreationTokens + cacheReadTokens;
+    if (total === 0 && costUSD === 0) continue;
+    out.push({
+      date,
+      tool: "cursor",
+      model: typeof e.model === "string" && e.model ? e.model : "cursor",
+      inputTokens,
+      outputTokens,
+      cacheCreationTokens,
+      cacheReadTokens,
+      costUSD,
+      origin: "cli",
+      verified: false
+    });
+  }
+  return out;
+}
+function resolveTokscaleBin() {
+  try {
+    const require3 = createRequire(import.meta.url);
+    const pkgPath = require3.resolve("tokscale/package.json");
+    const pkg = require3("tokscale/package.json");
+    const rel = typeof pkg.bin === "string" ? pkg.bin : pkg.bin?.tokscale ?? "";
+    if (!rel) return null;
+    const binPath = join3(dirname(pkgPath), rel);
+    if (/\.(c|m)?js$/.test(binPath)) {
+      return { cmd: process.execPath, prefixArgs: [binPath] };
+    }
+    return { cmd: binPath, prefixArgs: [] };
+  } catch {
+    return null;
+  }
+}
+function runTokscaleDay(bin, day) {
+  const res = spawnSync2(
+    bin.cmd,
+    [
+      ...bin.prefixArgs,
+      "--client",
+      "cursor",
+      "--json",
+      "--since",
+      day,
+      "--until",
+      day,
+      "--group-by",
+      "model",
+      "--no-spinner"
+    ],
+    { encoding: "utf8", maxBuffer: 32 * 1024 * 1024, timeout: 6e4 }
+  );
+  if (res.status !== 0 || !res.stdout) return null;
+  try {
+    return JSON.parse(res.stdout);
+  } catch {
+    return null;
+  }
+}
+function collectCursorViaTokscale(lookbackDays = LOOKBACK_DAYS) {
+  const bin = resolveTokscaleBin();
+  if (!bin) return [];
+  const today = /* @__PURE__ */ new Date();
+  const day = (offset) => new Date(today.getTime() - offset * 864e5).toISOString().slice(0, 10);
+  if (mapTokscaleDay(day(0), runTokscaleDay(bin, day(0))).length === 0) {
+    if (mapTokscaleDay(day(1), runTokscaleDay(bin, day(1))).length === 0) {
+      return [];
+    }
+  }
+  const out = [];
+  for (let i = 0; i < lookbackDays; i++) {
+    const d = day(i);
+    out.push(...mapTokscaleDay(d, runTokscaleDay(bin, d)));
+  }
+  return out;
+}
+
+// src/cursor.ts
+var EVENTS_URL = "https://cursor.com/api/dashboard/get-filtered-usage-events";
+function cursorDbPath() {
+  const home = homedir3();
+  const os = platform2();
+  const p = os === "darwin" ? join4(home, "Library", "Application Support", "Cursor", "User", "globalStorage", "state.vscdb") : os === "win32" ? join4(process.env.APPDATA ?? join4(home, "AppData", "Roaming"), "Cursor", "User", "globalStorage", "state.vscdb") : join4(process.env.XDG_CONFIG_HOME ?? join4(home, ".config"), "Cursor", "User", "globalStorage", "state.vscdb");
+  return existsSync3(p) ? p : null;
+}
+function readCursorToken(db) {
+  const require3 = createRequire2(import.meta.url);
+  try {
+    const { DatabaseSync } = require3("node:sqlite");
+    const d = new DatabaseSync(db, { readOnly: true });
+    const row = d.prepare("SELECT value FROM ItemTable WHERE key = ?").get("cursorAuth/accessToken");
+    d.close();
+    if (row?.value) return String(row.value);
+  } catch {
+  }
+  try {
+    const res = spawnSync3(
+      "sqlite3",
+      [db, "SELECT value FROM ItemTable WHERE key='cursorAuth/accessToken';"],
+      { encoding: "utf8", timeout: 1e4 }
+    );
+    const out = res.stdout?.trim();
+    if (res.status === 0 && out) return out;
+  } catch {
+  }
+  return null;
+}
+function cursorCookie(token) {
+  try {
+    const part = token.split(".")[1];
+    if (!part) return null;
+    const json = JSON.parse(
+      Buffer.from(part, "base64url").toString("utf8")
+    );
+    if (!json.sub) return null;
+    return `WorkosCursorSessionToken=${json.sub}%3A%3A${token}`;
+  } catch {
+    return null;
+  }
+}
+function num2(n) {
+  const v = Math.round(Number(n));
+  return Number.isFinite(v) && v > 0 ? v : 0;
+}
+function mapCursorEvents(events) {
+  const byDay = /* @__PURE__ */ new Map();
+  const byHour = /* @__PURE__ */ new Map();
+  for (const e of events) {
+    const tu = e.tokenUsage;
+    const ms = Number(e.timestamp);
+    if (!tu || !Number.isFinite(ms)) continue;
+    const d = new Date(ms);
+    const date = d.toISOString().slice(0, 10);
+    const model = e.model || "cursor";
+    const input = num2(tu.inputTokens);
+    const output = num2(tu.outputTokens);
+    const cacheWrite = num2(tu.cacheWriteTokens);
+    const cacheRead = num2(tu.cacheReadTokens);
+    const cost = Math.max(0, (Number(tu.totalCents) || 0) / 100);
+    const total = input + output + cacheWrite + cacheRead;
+    if (total === 0 && cost === 0) continue;
+    const key = `${date}|${model}`;
+    const day = byDay.get(key) ?? {
+      date,
+      tool: "cursor",
+      model,
+      inputTokens: 0,
+      outputTokens: 0,
+      cacheCreationTokens: 0,
+      cacheReadTokens: 0,
+      costUSD: 0,
+      origin: "cli",
+      verified: false
+    };
+    day.inputTokens += input;
+    day.outputTokens += output;
+    day.cacheCreationTokens += cacheWrite;
+    day.cacheReadTokens += cacheRead;
+    day.costUSD += cost;
+    byDay.set(key, day);
+    const hour = new Date(
+      Date.UTC(d.getUTCFullYear(), d.getUTCMonth(), d.getUTCDate(), d.getUTCHours())
+    ).toISOString();
+    const blk = byHour.get(hour) ?? { startTime: hour, totalTokens: 0, costUSD: 0 };
+    blk.totalTokens += total;
+    blk.costUSD += cost;
+    byHour.set(hour, blk);
+  }
+  const entries = [...byDay.values()].map((e) => ({
+    ...e,
+    costUSD: Number(e.costUSD.toFixed(4))
+  }));
+  const blocks = [...byHour.values()].map((b) => ({
+    ...b,
+    costUSD: Number(b.costUSD.toFixed(4))
+  }));
+  return { entries, blocks };
+}
+async function fetchCursorEvents(cookie, maxPages = 30, pageSize = 500) {
+  const all = [];
+  for (let page = 1; page <= maxPages; page++) {
+    const res = await fetch(EVENTS_URL, {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json",
+        Origin: "https://cursor.com",
+        Cookie: cookie
+      },
+      body: JSON.stringify({ page, pageSize }),
+      signal: AbortSignal.timeout(2e4)
+    });
+    if (!res.ok) break;
+    const body = await res.json();
+    const batch = body.usageEventsDisplay ?? [];
+    all.push(...batch);
+    if (batch.length < pageSize) break;
+  }
+  return all;
+}
+async function collectCursor() {
+  try {
+    const db = cursorDbPath();
+    const token = db ? readCursorToken(db) : null;
+    const cookie = token ? cursorCookie(token) : null;
+    if (cookie) {
+      const events = await fetchCursorEvents(cookie);
+      const { entries, blocks } = mapCursorEvents(events);
+      if (entries.length > 0) return { entries, blocks, found: true };
+    }
+  } catch {
+  }
+  try {
+    const entries = collectCursorViaTokscale();
+    if (entries.length > 0) return { entries, blocks: [], found: true };
+  } catch {
+  }
+  return { entries: [], blocks: [], found: false };
+}
+
+// src/collect.ts
 var SOURCES = [
   "claude",
   "codex",
@@ -400,18 +668,58 @@ function mapCcusageBlocks(json) {
   return out;
 }
 function resolveCcusageBin() {
-  const require3 = createRequire(import.meta.url);
+  const require3 = createRequire3(import.meta.url);
   const pkgPath = require3.resolve("ccusage/package.json");
   const pkg = require3("ccusage/package.json");
   const rel = typeof pkg.bin === "string" ? pkg.bin : pkg.bin?.ccusage ?? "ccusage";
-  const binPath = join3(dirname(pkgPath), rel);
+  const binPath = join5(dirname2(pkgPath), rel);
   if (/\.(c|m)?js$/.test(binPath)) {
     return { cmd: process.execPath, prefixArgs: [binPath] };
   }
   return { cmd: binPath, prefixArgs: [] };
 }
+function dedupeDaily(entries) {
+  const byKey = /* @__PURE__ */ new Map();
+  for (const e of entries) {
+    const key = `${e.date}|${e.tool}|${e.model}|${e.origin}`;
+    const prev = byKey.get(key);
+    if (!prev) {
+      byKey.set(key, { ...e });
+      continue;
+    }
+    prev.inputTokens += e.inputTokens;
+    prev.outputTokens += e.outputTokens;
+    prev.cacheCreationTokens += e.cacheCreationTokens;
+    prev.cacheReadTokens += e.cacheReadTokens;
+    prev.costUSD = Number((prev.costUSD + e.costUSD).toFixed(6));
+    prev.verified = prev.verified && e.verified;
+  }
+  return [...byKey.values()];
+}
+function dedupeSessions(sessions) {
+  const byId = /* @__PURE__ */ new Map();
+  const total = (s) => s.inputTokens + s.outputTokens + s.cacheCreationTokens + s.cacheReadTokens;
+  for (const s of sessions) {
+    const prev = byId.get(s.sessionId);
+    if (!prev || total(s) >= total(prev)) byId.set(s.sessionId, s);
+  }
+  return [...byId.values()];
+}
+function dedupeBlocks(blocks) {
+  const byStart = /* @__PURE__ */ new Map();
+  for (const b of blocks) {
+    const prev = byStart.get(b.startTime);
+    if (!prev) {
+      byStart.set(b.startTime, { ...b });
+      continue;
+    }
+    prev.totalTokens += b.totalTokens;
+    prev.costUSD = Number((prev.costUSD + b.costUSD).toFixed(6));
+  }
+  return [...byStart.values()];
+}
 function runCcusage(cmd, args) {
-  const res = spawnSync2(cmd, args, {
+  const res = spawnSync4(cmd, args, {
     encoding: "utf8",
     maxBuffer: 64 * 1024 * 1024,
     timeout: 12e4
@@ -423,7 +731,7 @@ function runCcusage(cmd, args) {
     return null;
   }
 }
-function collectAll() {
+async function collectAll() {
   const { cmd, prefixArgs } = resolveCcusageBin();
   const entries = [];
   const toolsFound = [];
@@ -446,7 +754,18 @@ function collectAll() {
   const blockJson = runCcusage(cmd, [...prefixArgs, "blocks", "--json", "--offline"]);
   const sessions = sessionJson ? mapCcusageSessions(sessionJson) : [];
   const blocks = blockJson ? mapCcusageBlocks(blockJson) : [];
-  return { entries, sessions, blocks, toolsFound };
+  const cursor = await collectCursor();
+  if (cursor.found) {
+    entries.push(...cursor.entries);
+    blocks.push(...cursor.blocks);
+    toolsFound.push("cursor");
+  }
+  return {
+    entries: dedupeDaily(entries),
+    sessions: dedupeSessions(sessions),
+    blocks: dedupeBlocks(blocks),
+    toolsFound
+  };
 }
 
 // ../../node_modules/.pnpm/zod@3.25.76/node_modules/zod/v3/external.js
@@ -4778,10 +5097,29 @@ async function publishLocal(payload, deps) {
 }
 
 // src/index.ts
-var require2 = createRequire2(import.meta.url);
+var require2 = createRequire4(import.meta.url);
 var VERSION = require2("../package.json").version;
+function startSpinner(label) {
+  if (!process.stdout.isTTY) {
+    console.log(pc2.dim(`  ${label}`));
+    return () => {
+    };
+  }
+  const frames = ["\u280B", "\u2819", "\u2839", "\u2838", "\u283C", "\u2834", "\u2826", "\u2827", "\u2807", "\u280F"];
+  let i = 0;
+  process.stdout.write("\x1B[?25l");
+  const timer = setInterval(() => {
+    i = (i + 1) % frames.length;
+    process.stdout.write(`\r  ${pc2.yellow(frames[i])} ${pc2.dim(label)}`);
+  }, 80);
+  return () => {
+    clearInterval(timer);
+    process.stdout.write("\r\x1B[2K");
+    process.stdout.write("\x1B[?25h");
+  };
+}
 function openBrowser(url) {
-  const os = platform2();
+  const os = platform3();
   const [cmd, args] = os === "darwin" ? ["open", [url]] : os === "win32" ? ["cmd", ["/c", "start", "", url]] : ["xdg-open", [url]];
   spawn(cmd, args, { stdio: "ignore", detached: true }).unref();
 }
@@ -4816,7 +5154,7 @@ async function confirm(question) {
 function showLocalDashboard(entries) {
   const dir = defaultConfigDir();
   mkdirSync3(dir, { recursive: true });
-  const file = join4(dir, "dashboard.html");
+  const file = join6(dir, "dashboard.html");
   writeFileSync3(file, renderDashboardHtml(entries));
   console.log();
   console.log(`  Local dashboard: ${pc2.cyan(`file://${file}`)}`);
@@ -4827,7 +5165,15 @@ async function run(flags) {
   if (!flags.quiet) {
     console.log(pc2.dim(`whoburnedmore v${VERSION} \xB7 ${flags.local ? "local mode" : apiBase()}`));
   }
-  const { entries, sessions, blocks, toolsFound } = collectAll();
+  const stop = flags.quiet ? () => {
+  } : startSpinner("Calculating your burn from local usage\u2026");
+  let collected;
+  try {
+    collected = await collectAll();
+  } finally {
+    stop();
+  }
+  const { entries, sessions, blocks, toolsFound } = collected;
   if (entries.length === 0) {
     console.log();
     console.log("  Nothing to burn yet \u2014 no local usage found from any coding agent.");
@@ -4864,6 +5210,10 @@ async function run(flags) {
   const config = loadConfig();
   if (config?.token) {
     const result = await submitUsage(config.token, payload);
+    if (!flags.quiet) {
+      console.log(pc2.dim("  Opening your dashboard in your browser\u2026"));
+      openBrowser(result.boardUrl ?? result.profileUrl);
+    }
     console.log(
       `  Submitted ${pc2.bold(String(result.upserted))} day-entries from ${toolsFound.join(", ")}.`
     );
@@ -4876,10 +5226,14 @@ async function run(flags) {
     if (result.boardUrl) {
       console.log(`  \u{1F91D} You're on the friends board: ${pc2.cyan(result.boardUrl)}`);
     }
-    if (!flags.quiet) openBrowser(result.boardUrl ?? result.profileUrl);
   } else {
     const anonKey = ensureAnonKey();
     const result = await anonSubmit(anonKey, payload);
+    const target = result.boardUrl ?? claimUrl(result.dashboardUrl, anonKey);
+    if (!flags.quiet) {
+      console.log(pc2.dim("  Opening your dashboard in your browser\u2026"));
+      openBrowser(target);
+    }
     console.log(
       `  Submitted ${pc2.bold(String(result.upserted))} day-entries from ${toolsFound.join(", ")}.`
     );
@@ -4889,19 +5243,14 @@ async function run(flags) {
       );
       console.log(`  ${pc2.cyan(result.boardUrl)}`);
       console.log(pc2.dim(`  Your dashboard: ${result.dashboardUrl}`));
-      if (!flags.quiet) {
-        openBrowser(result.boardUrl);
-        console.log(pc2.dim("  Opened the board \u2014 see how you stack up. Re-run anytime to update."));
-      }
     } else {
       console.log(
         `  You burned ${pc2.bold(formatTokens(result.totalTokens))} tokens \u2014 you're on the public leaderboard:`
       );
       console.log(`  ${pc2.cyan(result.dashboardUrl)}`);
       if (!flags.quiet) {
-        openBrowser(claimUrl(result.dashboardUrl, anonKey));
         console.log(
-          pc2.dim("  Opened your dashboard \u2014 claim it (name + X) to own your rank, or make it private / remove it.")
+          pc2.dim("  Claim it (name + X) to own your rank, or make it private / remove it.")
         );
         console.log(
           pc2.dim("  Manage anytime: `npx whoburnedmore private` \xB7 `npx whoburnedmore public` \xB7 `npx whoburnedmore remove`.")
@@ -4910,15 +5259,10 @@ async function run(flags) {
     }
   }
   console.log();
-  if (!flags.quiet && !autoSyncInstalled()) {
-    if (await confirm("  Keep your stats live? Install background sync (every 3h)?")) {
-      console.log(`  ${installAutoSync()}`);
-      console.log(pc2.dim("  Your page now updates in the background. Stop anytime with `npx whoburnedmore uninstall-sync`."));
-    } else {
-      console.log(pc2.dim("  OK \u2014 re-run `npx whoburnedmore` anytime, or `npx whoburnedmore install-sync` later."));
-    }
-  } else if (!flags.quiet && autoSyncInstalled()) {
-    console.log(pc2.dim("  Background sync is on \u2014 your page keeps updating automatically. Stop with `npx whoburnedmore uninstall-sync`."));
+  if (!flags.quiet) {
+    console.log(
+      autoSyncInstalled() ? pc2.dim("  Background sync is on \u2014 your page keeps updating automatically (`npx whoburnedmore uninstall-sync` to stop).") : pc2.dim("  Re-run anytime to update \xB7 `npx whoburnedmore install-sync` to keep it live in the background.")
+    );
   }
 }
 async function main() {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "whoburnedmore",
-  "version": "0.2.0",
+  "version": "0.4.0",
   "description": "Find out who burned more — submit your AI coding-agent token usage to the public leaderboard at whoburnedmore.com",
   "type": "module",
   "bin": {
@@ -10,7 +10,7 @@
     "dist/index.js"
   ],
   "scripts": {
-    "build": "rm -rf dist && esbuild src/index.ts --bundle --platform=node --format=esm --target=node20 --outfile=dist/index.js --external:ccusage --external:picocolors",
+    "build": "rm -rf dist && esbuild src/index.ts --bundle --platform=node --format=esm --target=node20 --outfile=dist/index.js --external:ccusage --external:picocolors --external:tokscale",
     "test": "vitest run",
     "lint": "tsc -p tsconfig.json --noEmit",
     "prepublishOnly": "pnpm run build",
@@ -20,7 +20,8 @@
   },
   "dependencies": {
     "ccusage": "20.0.9",
-    "picocolors": "^1.1.1"
+    "picocolors": "^1.1.1",
+    "tokscale": "^1.2.7"
   },
   "devDependencies": {
     "@types/node": "^22.10.0",