pi-tokenburn 0.6.0 → 0.7.0

package/extensions/core/aggregate.ts

@@ -1,6 +1,6 @@
 /** Aggregation of parsed turns into period totals and time series. */
 import { loadAll, type Turn } from "./parser.ts";
-import { selectTurns, type UsageOptions } from "./select.ts";
+import { cacheValue, selectTurns, type UsageOptions } from "./select.ts";
 import { bucketLabel, bucketStart, type Unit } from "./time.ts";
 import type { PeriodStats, TokenBurnReport } from "./types.ts";
 
@@ -56,10 +56,9 @@ export async function collectUsage(
 		p.cacheRead += t.cacheRead;
 		p.cacheWrite += t.cacheWrite;
 		p.cost += t.cost;
-		if (t.input > 0 && t.costInput > 0) {
-			p.cacheBase += (t.cacheRead * t.costInput) / t.input;
-			p.cacheActual += t.costCacheRead;
-		}
+		const cv = cacheValue(t);
+		p.cacheBase += cv.base;
+		p.cacheActual += cv.actual;
 	};
 
 	for (const turns of await loadSelected(sessionsRoot, opts)) {

package/extensions/core/config.ts

@@ -1,7 +1,7 @@
 /** Config file (tokenburn.json), path resolution, budgets. */
 import { homedir } from "node:os";
 import { join } from "node:path";
-import { readFile, writeFile, mkdir } from "node:fs/promises";
+import { mkdir, readFile, rename, writeFile } from "node:fs/promises";
 import { sanitizeFilter, sanitizePrices, type PriceMode, type PriceTable, type UsageFilter } from "./select.ts";
 import { STATUS_WINDOWS, type PeriodStats, type Window } from "./types.ts";
 
@@ -140,8 +140,12 @@ export function freshConfig(): TokenBurnConfig {
 }
 
 export async function saveConfig(cfg: TokenBurnConfig): Promise<void> {
-	await mkdir(getConfigDir(), { recursive: true });
-	await writeFile(join(getConfigDir(), "tokenburn.json"), JSON.stringify(cfg, null, 2) + "\n", "utf8");
+	const dir = getConfigDir();
+	await mkdir(dir, { recursive: true });
+	// Write beside, then rename: a crash or a second pi session never leaves half a file behind.
+	const tmp = join(dir, `tokenburn.json.${process.pid}.tmp`);
+	await writeFile(tmp, JSON.stringify(cfg, null, 2) + "\n", "utf8");
+	await rename(tmp, join(dir, "tokenburn.json"));
 }
 
 export function budgetsFor(cfg: TokenBurnConfig, w: Window): { tokens?: number; cost?: number } {

package/extensions/core/parser.ts

@@ -64,13 +64,13 @@ const GUARD_BYTES = 64;
 const HEAD_BYTES = 256;
 export const CHUNK_BYTES = 1 << 20;
 
-/** A Turn from one JSONL line, or undefined when it carries no usage. */
 /** A finite number, else 0 (logs can hold 1e999, strings, junk). */
 const finite = (v: unknown): number => {
 	const n = Number(v);
 	return Number.isFinite(n) ? n : 0;
 };
 
+/** A Turn from one JSONL line, or undefined when it carries no usage. */
 function turnFromLine(line: Buffer, meta: Meta): Turn | undefined {
 	if (meta.cwd === undefined && line.includes('"type":"session"')) {
 		try {

package/extensions/core/select.ts

@@ -103,3 +103,12 @@ export function selectTurns(turns: Turn[], opts: UsageOptions | undefined): Turn
 	}
 	return out;
 }
+
+/**
+ * What the turn's cached tokens would have cost as plain input (`base`) and what they did cost
+ * (`actual`). Both 0 when the log has no input price for the turn (then no savings are claimed).
+ */
+export function cacheValue(t: Turn): { base: number; actual: number } {
+	if (!(t.input > 0 && t.costInput > 0)) return { base: 0, actual: 0 };
+	return { base: (t.cacheRead * t.costInput) / t.input, actual: t.costCacheRead };
+}

package/extensions/features/breakdown/query.ts

@@ -1,6 +1,6 @@
 /** Grouped usage queries: per model, provider, project and session, over a period. */
 import { HOUR_MS, loadSelected, turnTokens } from "../../core/aggregate.ts";
-import type { UsageOptions } from "../../core/select.ts";
+import { cacheValue, type UsageOptions } from "../../core/select.ts";
 import { bucketStart } from "../../core/time.ts";
 
 export const PERIODS = ["hour", "day", "week", "month", "year", "all"] as const;
@@ -106,10 +106,9 @@ export async function collectGroups(
 			r.turns++;
 			if (t.ts < r.first) r.first = t.ts;
 			if (t.ts > r.last) r.last = t.ts;
-			if (t.input > 0 && t.costInput > 0) {
-				r.cacheBase += (t.cacheRead * t.costInput) / t.input;
-				r.cacheActual += t.costCacheRead;
-			}
+			const cv = cacheValue(t);
+			r.cacheBase += cv.base;
+			r.cacheActual += cv.actual;
 			if (!seen.has(key)) {
 				seen.add(key);
 				r.sessions++;

package/extensions/features/report/chart.ts

@@ -68,21 +68,28 @@ export function renderChart(
 			if (h >= row) cell = "█";
 			else if (h > row - 1 && series[i].tokens > 0) cell = SPARK[Math.min(7, Math.max(0, Math.floor((h - (row - 1)) * 8) - 1))];
 			else if (row === 1 && series[i].tokens > 0) cell = "▁";
+			else if (row === 1 && budgetRow !== 1) cell = "·"; // an empty bucket, not a missing one
 			else cell = row === budgetRow ? "┄" : " ";
-			const filled = cell !== " " && cell !== "┄";
+			const filled = cell !== " " && cell !== "┄" && cell !== "·";
 			const body = cell.repeat(barW);
-			line += filled ? paint(kindOf(i), body) : cell === "┄" ? paint("budget", body) : body;
+			line += filled ? paint(kindOf(i), body) : cell === "┄" ? paint("budget", body) : cell === "·" ? paint("axis", body) : body;
 			line += row === budgetRow ? paint("budget", "┄") : " ";
 		}
 		lines.push(line);
 	}
 
-	lines.push(paint("axis", `${"".padStart(yW)} └${"─".repeat(series.length * colW)}`));
+	lines.push(paint("axis", `${yLabels[2].padStart(yW)} └${"─".repeat(series.length * colW)}`));
 	const pad = " ".repeat(yW + 2);
 	const cols = series.map((p, i) => ({ l: columnLabels(p, unit), now: i === series.length - 1 }));
 	lines.push(pad + cols.map((c) => c.l[0].padEnd(colW)).join(""));
 	lines.push(paint("dim", pad + cols.map((c) => c.l[1].padEnd(colW)).join("")));
-	lines.push(pad + cols.map((c) => (c.now ? paint("now", "▲".padEnd(colW)) : " ".repeat(colW))).join(""));
+	// The marker under the current bucket, with its value so you need not read it off the axis.
+	const latest = series[series.length - 1];
+	lines.push(
+		pad +
+			cols.map((c) => (c.now ? paint("now", "▲".padEnd(colW)) : " ".repeat(colW))).join("") +
+			(latest ? paint("now", `now ${formatNum(latest.tokens)} (${formatCost(latest.cost)})`) : ""),
+	);
 
 	const avg = series.length ? total / series.length : 0;
 	lines.push(

package/extensions/features/status-line/status.ts

@@ -1,12 +1,11 @@
 /** Status line: window start, burn rate / cache hit / savings extras for the status window (pure). */
 import type { TokenBurnConfig } from "../../core/config.ts";
 import { formatCost, formatNum } from "../../core/format.ts";
+import { HOUR_MS as HOUR } from "../../core/aggregate.ts";
 import { bucketStart } from "../../core/time.ts";
 import type { PeriodStats, Rate, TokenBurnReport, Window } from "../../core/types.ts";
 import { cacheHitRatio } from "../breakdown/query.ts";
 
-const HOUR = 3_600_000;
-
 /** Start of a status window (`hour` = rolling last 60 minutes, `total` = 0). */
 export function windowStart(w: Window, now: number): number {
 	if (w === "hour") return now - HOUR;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-tokenburn",
-  "version": "0.6.0",
+  "version": "0.7.0",
   "description": "Pi extension for live token usage monitoring and budget tracking (Day / Week / Month / Total). 0 prompt tokens.",
   "keywords": ["pi-package", "pi-extension", "tokenburn", "tokens", "analytics", "budget", "stats"],
   "license": "MIT",
@@ -24,6 +24,9 @@
   "peerDependencies": {
     "@earendil-works/pi-coding-agent": "*"
   },
+  "overrides": {
+    "undici": "8.12.3"
+  },
   "devDependencies": {
     "@earendil-works/pi-coding-agent": "^1.4.0",
     "@types/bun": "^1.4.4",