@burn0/burn0 0.1.0 → 0.2.1

package/dist/index.js

@@ -465,13 +465,16 @@ function createDispatcher(mode2, deps) {
         break;
       case "dev-cloud":
         deps.logEvent?.(event);
+        deps.writeLedger?.(event);
         deps.addToBatch?.(event);
         break;
       case "prod-cloud":
+        deps.logEvent?.(event);
+        deps.writeLedger?.(event);
         deps.addToBatch?.(event);
         break;
       case "prod-local":
-        deps.accumulate?.(event);
+        deps.logEvent?.(event);
         break;
       case "test-enabled":
         deps.logEvent?.(event);
@@ -702,114 +705,114 @@ function estimateLocalCost(event) {
 }
 
 // src/transport/logger.ts
-var DIM = "\x1B[2m";
 var RESET = "\x1B[0m";
-var CYAN = "\x1B[36m";
 var GREEN = "\x1B[32m";
-var YELLOW = "\x1B[33m";
-var WHITE = "\x1B[37m";
 var BOLD = "\x1B[1m";
 var ORANGE = "\x1B[38;2;250;93;25m";
 var GRAY = "\x1B[90m";
-var headerPrinted = false;
-var sessionTotal = 0;
-var eventCount = 0;
-function formatTokens(count) {
-  if (count >= 1e6) return `${(count / 1e6).toFixed(1)}M`;
-  if (count >= 1e3) return `${(count / 1e3).toFixed(1)}K`;
-  return count.toString();
-}
 function formatCost(cost) {
   if (cost >= 1) return `$${cost.toFixed(2)}`;
   if (cost >= 0.01) return `$${cost.toFixed(4)}`;
   return `$${cost.toFixed(6)}`;
 }
-function formatCostEstimate(estimate) {
-  switch (estimate.type) {
-    case "priced":
-      return `${GREEN}${formatCost(estimate.cost)}${RESET}`;
-    case "free":
-      return `${GRAY}free${RESET}`;
-    case "no-tokens":
-      return `${YELLOW}no usage${RESET}`;
-    case "fixed-tier":
-      return `${YELLOW}plan?${RESET}`;
-    case "unknown":
-      return `${GRAY}untracked${RESET}`;
-    case "loading":
-      return `${GRAY}...${RESET}`;
-  }
-}
-function printHeader() {
-  if (headerPrinted) return;
-  headerPrinted = true;
-  process.stdout.write(`
-`);
-  process.stdout.write(`  ${ORANGE}${BOLD} burn0 ${RESET} ${DIM}live cost tracking${RESET}
-`);
-  process.stdout.write(`
-`);
-  process.stdout.write(`  ${GRAY}SERVICE          ENDPOINT / MODEL              USAGE          COST${RESET}
-`);
-  process.stdout.write(`  ${GRAY}${"\u2500".repeat(68)}${RESET}
-`);
-}
-function printSessionTotal() {
-  process.stdout.write(`  ${GRAY}${"\u2500".repeat(68)}${RESET}
-`);
-  if (sessionTotal > 0) {
-    process.stdout.write(`  ${GRAY}${eventCount} calls${RESET}                                              ${ORANGE}${BOLD}${formatCost(sessionTotal)}${RESET}
-`);
-  } else {
-    process.stdout.write(`  ${GRAY}${eventCount} calls${RESET}                                              ${GRAY}$0${RESET}
-`);
-  }
-  process.stdout.write(`  ${GRAY}${"\u2500".repeat(68)}${RESET}
-`);
-}
-function formatEventLine(event) {
-  const service = event.service.length > 15 ? event.service.substring(0, 14) + "." : event.service;
-  const modelOrEndpoint = event.model ? event.model.length > 29 ? event.model.substring(0, 28) + "." : event.model : event.endpoint.length > 29 ? event.endpoint.substring(0, 28) + "." : event.endpoint;
-  let usage = "";
-  if (event.tokens_in !== void 0 && event.tokens_out !== void 0) {
-    usage = `${formatTokens(event.tokens_in)} \u2192 ${formatTokens(event.tokens_out)}`;
-  }
-  const estimate = estimateLocalCost(event);
-  const costStr = formatCostEstimate(estimate);
-  return `  ${CYAN}${service.padEnd(16)}${RESET} ${WHITE}${modelOrEndpoint.padEnd(30)}${RESET}${GRAY}${usage.padEnd(15)}${RESET}${costStr}`;
-}
-function formatProcessSummary(events, uptimeSeconds) {
-  const services = {};
-  for (const event of events) {
-    if (!services[event.service]) services[event.service] = { calls: 0 };
-    services[event.service].calls++;
-    if (event.tokens_in !== void 0) services[event.service].tokens_in = (services[event.service].tokens_in ?? 0) + event.tokens_in;
-    if (event.tokens_out !== void 0) services[event.service].tokens_out = (services[event.service].tokens_out ?? 0) + event.tokens_out;
-  }
-  for (const svc of Object.values(services)) {
-    if (svc.tokens_in === void 0) delete svc.tokens_in;
-    if (svc.tokens_out === void 0) delete svc.tokens_out;
-  }
-  return JSON.stringify({
-    burn0: "process-summary",
-    uptime_hours: +(uptimeSeconds / 3600).toFixed(1),
-    total_calls: events.length,
-    services,
-    message: "Add BURN0_API_KEY to see cost breakdowns \u2192 burn0.dev"
-  });
-}
-function logEvent(event) {
-  printHeader();
-  const estimate = estimateLocalCost(event);
-  if (estimate.type === "priced" && estimate.cost > 0) {
-    sessionTotal += estimate.cost;
-  }
-  eventCount++;
-  process.stdout.write(`${formatEventLine(event)}
-`);
-  if (eventCount % 5 === 0) {
-    printSessionTotal();
+function formatDuration(ms) {
+  const seconds = Math.floor(ms / 1e3);
+  if (seconds < 60) return `${seconds}s`;
+  const minutes = Math.floor(seconds / 60);
+  const remainingSeconds = seconds % 60;
+  if (minutes < 60) return `${minutes}m ${remainingSeconds}s`;
+  const hours = Math.floor(minutes / 60);
+  const remainingMinutes = minutes % 60;
+  return `${hours}h ${remainingMinutes}m`;
+}
+function formatServiceBreakdown(perServiceCosts2, maxWidth) {
+  const sorted = Object.entries(perServiceCosts2).filter(([, cost]) => cost > 0).sort((a, b) => b[1] - a[1]);
+  if (sorted.length === 0) return "";
+  const parts = [];
+  let currentWidth = 0;
+  let shown = 0;
+  for (let i = 0; i < sorted.length && shown < 3; i++) {
+    const [name, cost] = sorted[i];
+    const part = `${name}: ${formatCost(cost)}`;
+    if (currentWidth + part.length + 3 > maxWidth && shown > 0) {
+      break;
+    }
+    parts.push(part);
+    currentWidth += part.length + 3;
+    shown++;
+  }
+  const remaining = sorted.length - shown;
+  if (remaining > 0) {
+    parts.push(`+${remaining} more`);
+  }
+  return parts.join(" \xB7 ");
+}
+function createTicker(init) {
+  let sessionCost = 0;
+  let sessionCalls = 0;
+  const sessionStartTime = Date.now();
+  let todayCost2 = init.todayCost;
+  let todayCalls2 = init.todayCalls;
+  const perServiceCosts2 = { ...init.perServiceCosts };
+  let exitPrinted = false;
+  let pricedCalls = 0;
+  let lastLineLen = 0;
+  function render() {
+    if (!process.stderr.isTTY) return;
+    if (todayCalls2 === 0) return;
+    let content;
+    if (pricedCalls === 0 && todayCost2 === 0) {
+      content = `  burn0 \u25B8 ${todayCalls2} calls today`;
+    } else {
+      const breakdown = formatServiceBreakdown(perServiceCosts2, 40);
+      const breakdownPart = breakdown ? ` \u2500\u2500 ${breakdown}` : "";
+      content = `  burn0 \u25B8 ${formatCost(todayCost2)} today (${todayCalls2} calls)${breakdownPart}`;
+    }
+    const pad = lastLineLen > content.length ? " ".repeat(lastLineLen - content.length) : "";
+    lastLineLen = content.length;
+    let colored;
+    if (pricedCalls === 0 && todayCost2 === 0) {
+      colored = `  ${ORANGE}${BOLD}burn0 \u25B8${RESET} ${GRAY}${todayCalls2} calls today${RESET}`;
+    } else {
+      const breakdown = formatServiceBreakdown(perServiceCosts2, 40);
+      const breakdownPart = breakdown ? ` ${GRAY}\u2500\u2500${RESET} ${breakdown}` : "";
+      colored = `  ${ORANGE}${BOLD}burn0 \u25B8${RESET} ${GREEN}${formatCost(todayCost2)}${RESET} ${GRAY}today (${todayCalls2} calls)${RESET}${breakdownPart}`;
+    }
+    process.stderr.write(`\r${colored}${pad}`);
+  }
+  function tick(event) {
+    const estimate = estimateLocalCost(event);
+    todayCalls2++;
+    sessionCalls++;
+    if (estimate.type === "priced" && estimate.cost > 0) {
+      todayCost2 += estimate.cost;
+      sessionCost += estimate.cost;
+      pricedCalls++;
+      perServiceCosts2[event.service] = (perServiceCosts2[event.service] ?? 0) + estimate.cost;
+    }
+    render();
+  }
+  function printExitSummary() {
+    if (!process.stderr.isTTY) return;
+    if (sessionCalls === 0) return;
+    if (exitPrinted) return;
+    exitPrinted = true;
+    const duration = formatDuration(Date.now() - sessionStartTime);
+    let line;
+    if (pricedCalls === 0 && sessionCost === 0) {
+      line = `
+  ${ORANGE}${BOLD}burn0 \u25B8${RESET} ${GRAY}session: ${sessionCalls} calls (${duration})${RESET} ${GRAY}\u2500\u2500${RESET} ${GRAY}today: ${todayCalls2} calls${RESET}
+`;
+    } else {
+      const monthlyEst = todayCost2 > 0 ? formatCost(todayCost2 * 30) : null;
+      const projPart = monthlyEst ? ` ${GRAY}\u2500\u2500${RESET} ${GRAY}~${GREEN}${monthlyEst}${RESET}${GRAY}/mo${RESET}` : "";
+      line = `
+  ${ORANGE}${BOLD}burn0 \u25B8${RESET} ${GRAY}session:${RESET} ${GREEN}${formatCost(sessionCost)}${RESET} ${GRAY}(${sessionCalls} calls, ${duration})${RESET} ${GRAY}\u2500\u2500${RESET} ${GRAY}today:${RESET} ${GREEN}${formatCost(todayCost2)}${RESET}${projPart}
+`;
+    }
+    process.stderr.write(line);
   }
+  return { tick, printExitSummary };
 }
 
 // src/index.ts
@@ -818,12 +821,36 @@ var apiKey = getApiKey();
 var mode = detectMode({ isTTY: isTTY(), apiKey });
 var { track, startSpan, enrichEvent } = createTracker();
 var originalFetch2 = globalThis.fetch;
-if (mode !== "test-disabled") {
+if (mode !== "test-disabled" && mode !== "prod-local") {
   fetchPricing(BURN0_API_URL, originalFetch2).catch(() => {
   });
 }
-var accumulatedEvents = [];
-var ledger = mode === "dev-local" || mode === "test-enabled" ? new LocalLedger(process.cwd()) : null;
+var ledger = new LocalLedger(process.cwd());
+function getTodayDateStr() {
+  const d = /* @__PURE__ */ new Date();
+  return `${d.getFullYear()}-${String(d.getMonth() + 1).padStart(2, "0")}-${String(d.getDate()).padStart(2, "0")}`;
+}
+var todayCost = 0;
+var todayCalls = 0;
+var perServiceCosts = {};
+try {
+  const todayStr = getTodayDateStr();
+  const allEvents = ledger.read();
+  for (const event of allEvents) {
+    const eventDate = new Date(event.timestamp);
+    const eventDateStr = `${eventDate.getFullYear()}-${String(eventDate.getMonth() + 1).padStart(2, "0")}-${String(eventDate.getDate()).padStart(2, "0")}`;
+    if (eventDateStr === todayStr) {
+      todayCalls++;
+      const estimate = estimateLocalCost(event);
+      if (estimate.type === "priced" && estimate.cost > 0) {
+        todayCost += estimate.cost;
+        perServiceCosts[event.service] = (perServiceCosts[event.service] ?? 0) + estimate.cost;
+      }
+    }
+  }
+} catch {
+}
+var ticker = createTicker({ todayCost, todayCalls, perServiceCosts });
 var batch = null;
 if ((mode === "dev-cloud" || mode === "prod-cloud") && apiKey) {
   batch = new BatchBuffer({
@@ -836,17 +863,20 @@ if ((mode === "dev-cloud" || mode === "prod-cloud") && apiKey) {
     }
   });
 }
+var shouldWriteLedger = mode !== "test-disabled" && mode !== "prod-local";
 var dispatch = createDispatcher(mode, {
-  logEvent,
-  writeLedger: ledger ? (e) => ledger.write(e) : void 0,
-  addToBatch: batch ? (e) => batch.add(e) : void 0,
-  accumulate: (e) => accumulatedEvents.push(e)
+  logEvent: (e) => ticker.tick(e),
+  writeLedger: shouldWriteLedger ? (e) => ledger.write(e) : void 0,
+  addToBatch: batch ? (e) => batch.add(e) : void 0
 });
 var preloaded = checkImportOrder();
 if (preloaded.length > 0) {
-  console.warn(`[burn0] Warning: These SDKs were imported before burn0 and may not be tracked: ${preloaded.join(", ")}. Move \`import 'burn0'\` to the top of your entry file.`);
+  console.warn(`[burn0] Warning: These SDKs were imported before burn0 and may not be tracked: ${preloaded.join(", ")}. Move \`import '@burn0/burn0'\` to the top of your entry file.`);
 }
-if (canPatch() && mode !== "test-disabled") {
+if (mode === "prod-local") {
+  console.warn("[burn0] No API key \u2014 costs not tracked. Get one free at burn0.dev/api");
+}
+if (canPatch() && mode !== "test-disabled" && mode !== "prod-local") {
   const onEvent = (event) => {
     const enriched = enrichEvent(event);
     dispatch(enriched);
@@ -855,25 +885,16 @@ if (canPatch() && mode !== "test-disabled") {
   patchHttp(onEvent);
   markPatched();
 }
-if (mode === "prod-local") {
-  const startTime = Date.now();
-  process.on("beforeExit", () => {
-    if (accumulatedEvents.length > 0) {
-      const uptimeSeconds = (Date.now() - startTime) / 1e3;
-      console.log(formatProcessSummary(accumulatedEvents, uptimeSeconds));
-    }
-  });
-}
-if (batch) {
-  const exitFlush = () => {
+var exitHandled = false;
+process.on("exit", () => {
+  if (exitHandled) return;
+  exitHandled = true;
+  if (batch) {
     batch.flush();
     batch.destroy();
-  };
-  process.on("beforeExit", exitFlush);
-  process.on("SIGTERM", exitFlush);
-  process.on("SIGINT", exitFlush);
-  process.on("SIGHUP", exitFlush);
-}
+  }
+  ticker.printExitSummary();
+});
 var restore = createRestorer({ unpatchFetch, unpatchHttp, resetGuard });
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {

package/dist/index.mjs

@@ -2,7 +2,7 @@ import {
   restore,
   startSpan,
   track
-} from "./chunk-ZHAS7BCI.mjs";
+} from "./chunk-KKYHE4ZV.mjs";
 import "./chunk-DJ72YN4C.mjs";
 export {
   restore,