pi-tokenburn 0.2.0 → 0.3.1

package/README.md

@@ -10,7 +10,7 @@ Totals match the `tokenburn` CLI exactly.
 
 - **Status bar:** `🔥 1.20M ($52.79) [today]`, turns red with `OVER BUDGET` past your limit.
 - **Report chart:** `/tokenburn` shows the summary plus a vertical column chart (y-axis, budget line, current period marked, over-budget columns in red, peak/avg footer). Daily (14 days), weekly (8 weeks) or monthly (6 months).
-- **Switch periods fast:** `/tokenburn week`, `/tokenburn month`, `/tokenburn cycle`.
+- **Views:** `/tokenburn day|week|month|year|all`. Status bar period: `/tokenburn status <window>` or `/tokenburn cycle`.
 - **Window panel (optional):** Today / Week / Month / Total below the editor, with in/out tokens and cost.
 - **Budgets** per day, week and month, in tokens and/or dollars.
 - **Fast:** per-file cache keyed on mtime+size. First scan of ~400 MB of sessions takes under a second; refreshes after that take milliseconds. Refresh never blocks the agent loop.
@@ -27,15 +27,21 @@ pi install git:github.com/sorinirimies/pi-tokenburn
 
 | Command | What it does |
 |---|---|
-| `/tokenburn` | Report: Today / Week / Month / All-time + **daily** chart |
-| `/tokenburn report week` · `report month` | Same report with a **weekly** / **monthly** chart (also `chart month`) |
-| `/tokenburn status <today\|week\|month\|total>` | Set the status-bar period (shortcut: `/tokenburn total`) |
-| `/tokenburn cycle` | Rotate the status-bar period: today → week → month → total |
+| `/tokenburn` · `day` · `today` | Report + **daily** chart (last 14 days) |
+| `/tokenburn week` | Report + **weekly** chart (last 8 weeks) |
+| `/tokenburn month` | Report + **monthly** chart (last 6 months) |
+| `/tokenburn year` | Report + **yearly** chart (last 5 years) |
+| `/tokenburn all` | Report + whole history (monthly columns up to 18 months, yearly beyond) |
+| `/tokenburn report <view>` · `chart <view>` | Same as above, explicit form |
+| `/tokenburn status <today\|week\|month\|year\|total>` | Set the status-bar period |
+| `/tokenburn cycle` | Rotate the status-bar period: today → week → month → year → total |
 | `/tokenburn window` | Toggle the panel below the editor |
 | `/tokenburn budget <day\|week\|month> <tokens>` | Token budget, e.g. `budget day 5000000` |
 | `/tokenburn budget <day\|week\|month> $<cost>` | Dollar budget, e.g. `budget week $50` |
 | `/tokenburn cache` | Toggle counting cache read/write tokens |
 
+Tab-completion works for all of the above. Unknown input prints usage instead of failing.
+
 Chart example (`/tokenburn`, with a 600M day budget):
 
 ```
@@ -75,12 +81,36 @@ Weeks start on Monday, in local time.
 
 Environment overrides for the session directory: `TOKENBURN_PI_SESSIONS`, `PI_CODING_AGENT_SESSION_DIR`, `PI_CODING_AGENT_DIR`.
 
-## Develop
+## Development
 
 ```bash
-bun test
+bun install
+just check          # typecheck + tests + pack check + nushell tests (what CI runs)
+just test           # bun test only (with coverage)
+just coverage       # tests + a coverage table
 ```
 
+**Coverage:** every `bun test` collects coverage (`bunfig.toml`) and **fails below 95% lines / 95% functions**. CI shows the table on the run page and uploads `lcov.info`.
+
+CI scripts are [nushell](https://www.nushell.sh) (`scripts/`), the same ones locally and in GitHub / Gitea Actions.
+`just --list` shows every task.
+
+## Releases (automatic)
+
+| Workflow | When | What |
+|---|---|---|
+| **CI** | push / PR | quality gate on Linux; tests on macOS, Windows and extreme time zones |
+| **Nightly Dependency Update** | every night (GitHub 02:00 UTC, Gitea 02:30) | `bun update` within ranges, verify on all platforms, commit `chore(deps)`; then **publish a patch** when a *runtime* dependency was upgraded or `feat`/`fix`/`perf` commits are waiting since the last tag |
+| **Release** | tag `vX.Y.Z` | validates the tag against `package.json`, runs the gate, `npm publish` (idempotent), creates the release |
+
+Dev-tooling-only bumps are committed but never released on their own. A downgrade blocks the whole update.
+
+Manual release: `just bump patch` (or `minor` / `major` / `X.Y.Z`), then `just release-push`.
+
+**Secrets** (repo settings): `NPM_TOKEN` — npm *granular access token* with publish rights and "bypass 2FA" (required for CI publishing). Optional: `GH_PAT` (lets a tag push trigger the release itself), `GITEA_TOKEN` (Gitea).
+
+Commits follow [Conventional Commits](https://www.conventionalcommits.org); the changelog is generated by git-cliff.
+
 ## Scope
 
 Pi sessions only. For Zed, Claude Code, Codex, Gemini and more, use the Rust [tokenburn](https://github.com/sorinirimies/tokenburn).

package/extensions/tokenburn.ts

@@ -44,9 +44,13 @@ export interface TokenBurnReport {
 	today: PeriodStats;
 	week: PeriodStats;
 	month: PeriodStats;
+	year: PeriodStats;
 	total: PeriodStats;
 }
 
+export const STATUS_WINDOWS = ["today", "week", "month", "year", "total"] as const;
+export type Window = (typeof STATUS_WINDOWS)[number];
+
 export interface TokenBurnConfig {
 	/** Count cache read/write tokens in totals (matches the tokenburn CLI). */
 	includeCache: boolean;
@@ -54,7 +58,7 @@ export interface TokenBurnConfig {
 	chartColor: boolean;
 	showStatus: boolean;
 	showWidget: boolean;
-	statusWindow: "today" | "week" | "month" | "total";
+	statusWindow: Window;
 	budgets: {
 		dayTokens?: number;
 		weekTokens?: number;
@@ -74,18 +78,24 @@ const DEFAULT_CONFIG: TokenBurnConfig = {
 	budgets: {},
 };
 
-function getConfigDir(): string {
-	if (process.env.PI_CODING_AGENT_DIR) return process.env.PI_CODING_AGENT_DIR;
-	if (process.env.XDG_CONFIG_HOME) return join(process.env.XDG_CONFIG_HOME, "pi", "agent");
-	return join(homedir(), ".pi", "agent");
+/** Where pi keeps its config: PI_CODING_AGENT_DIR, then $XDG_CONFIG_HOME/pi/agent, then ~/.pi/agent. */
+export function resolveAgentDir(env: Record<string, string | undefined>, home: string): string {
+	if (env.PI_CODING_AGENT_DIR) return env.PI_CODING_AGENT_DIR;
+	if (env.XDG_CONFIG_HOME) return join(env.XDG_CONFIG_HOME, "pi", "agent");
+	return join(home, ".pi", "agent");
 }
 
-function getSessionsDir(): string {
-	if (process.env.TOKENBURN_PI_SESSIONS) return process.env.TOKENBURN_PI_SESSIONS;
-	if (process.env.PI_CODING_AGENT_SESSION_DIR) return process.env.PI_CODING_AGENT_SESSION_DIR;
-	return join(getConfigDir(), "sessions");
+/** Pi session logs: TOKENBURN_PI_SESSIONS, PI_CODING_AGENT_SESSION_DIR, then <agent dir>/sessions. */
+export function resolveSessionsDir(env: Record<string, string | undefined>, home: string): string {
+	if (env.TOKENBURN_PI_SESSIONS) return env.TOKENBURN_PI_SESSIONS;
+	if (env.PI_CODING_AGENT_SESSION_DIR) return env.PI_CODING_AGENT_SESSION_DIR;
+	return join(resolveAgentDir(env, home), "sessions");
 }
 
+const getConfigDir = () => resolveAgentDir(process.env, homedir());
+
+const getSessionsDir = () => resolveSessionsDir(process.env, homedir());
+
 export async function loadConfig(): Promise<TokenBurnConfig> {
 	try {
 		const raw = await readFile(join(getConfigDir(), "tokenburn.json"), "utf8");
@@ -95,16 +105,21 @@ export async function loadConfig(): Promise<TokenBurnConfig> {
 			chartColor: typeof parsed.chartColor === "boolean" ? parsed.chartColor : DEFAULT_CONFIG.chartColor,
 			showStatus: typeof parsed.showStatus === "boolean" ? parsed.showStatus : DEFAULT_CONFIG.showStatus,
 			showWidget: typeof parsed.showWidget === "boolean" ? parsed.showWidget : DEFAULT_CONFIG.showWidget,
-			statusWindow: ["today", "week", "month", "total"].includes(parsed.statusWindow)
+			statusWindow: (STATUS_WINDOWS as readonly string[]).includes(parsed.statusWindow)
 				? parsed.statusWindow
 				: DEFAULT_CONFIG.statusWindow,
-			budgets: typeof parsed.budgets === "object" && parsed.budgets !== null ? parsed.budgets : {},
+			budgets: typeof parsed.budgets === "object" && parsed.budgets !== null ? { ...parsed.budgets } : {},
 		};
 	} catch {
-		return { ...DEFAULT_CONFIG };
+		return freshConfig();
 	}
 }
 
+/** A config that shares no mutable state with DEFAULT_CONFIG. */
+export function freshConfig(): TokenBurnConfig {
+	return { ...DEFAULT_CONFIG, budgets: {} };
+}
+
 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");
@@ -163,6 +178,16 @@ async function parseFile(file: string): Promise<Turn[]> {
 	return turns;
 }
 
+/** Cache key for a file: mtime + size. `undefined` when it cannot be stat'ed. */
+export function fileSignature(file: string): string | undefined {
+	try {
+		const st = statSync(file);
+		return `${st.mtimeMs}:${st.size}`;
+	} catch {
+		return undefined;
+	}
+}
+
 function listSessionFiles(root: string): string[] {
 	const files: string[] = [];
 	const walk = (dir: string) => {
@@ -200,13 +225,8 @@ async function loadAll(sessionsRoot: string): Promise<Turn[][]> {
 
 	const out: Turn[][] = [];
 	for (const file of files) {
-		let sig: string;
-		try {
-			const st = statSync(file);
-			sig = `${st.mtimeMs}:${st.size}`;
-		} catch {
-			continue;
-		}
+		const sig = fileSignature(file);
+		if (sig === undefined) continue; // vanished between listing and stat
 		let entry = FILE_CACHE.get(file);
 		if (!entry || entry.sig !== sig) {
 			entry = { sig, turns: await parseFile(file) };
@@ -228,11 +248,13 @@ export async function collectUsage(
 	const startOfDay = bucketStart(now.getTime(), "day");
 	const startOfWeek = bucketStart(now.getTime(), "week");
 	const startOfMonth = bucketStart(now.getTime(), "month");
+	const startOfYear = bucketStart(now.getTime(), "year");
 
 	const report: TokenBurnReport = {
 		today: emptyStats(),
 		week: emptyStats(),
 		month: emptyStats(),
+		year: emptyStats(),
 		total: emptyStats(),
 	};
 
@@ -250,11 +272,13 @@ export async function collectUsage(
 		for (const t of turns) {
 			if (t.ts > maxTs) maxTs = t.ts;
 			add(report.total, t);
+			if (t.ts >= startOfYear) add(report.year, t);
 			if (t.ts >= startOfMonth) add(report.month, t);
 			if (t.ts >= startOfWeek) add(report.week, t);
 			if (t.ts >= startOfDay) add(report.today, t);
 		}
 		if (turns.length > 0) report.total.sessions++;
+		if (maxTs >= startOfYear) report.year.sessions++;
 		if (maxTs >= startOfMonth) report.month.sessions++;
 		if (maxTs >= startOfWeek) report.week.sessions++;
 		if (maxTs >= startOfDay) report.today.sessions++;
@@ -267,7 +291,7 @@ export async function collectUsage(
 // Time series + chart
 // ---------------------------------------------------------------------------
 
-export type Unit = "day" | "week" | "month";
+export type Unit = "day" | "week" | "month" | "year";
 
 /** Local-time start of the day / Monday-week / month containing `ts`. */
 export function bucketStart(ts: number, unit: Unit): number {
@@ -277,6 +301,7 @@ export function bucketStart(ts: number, unit: Unit): number {
 		const dow = (d.getDay() + 6) % 7; // Monday = 0
 		return new Date(d.getFullYear(), d.getMonth(), d.getDate() - dow).getTime();
 	}
+	if (unit === "year") return new Date(d.getFullYear(), 0, 1).getTime();
 	return new Date(d.getFullYear(), d.getMonth(), 1).getTime();
 }
 
@@ -284,6 +309,7 @@ const pad2 = (n: number) => String(n).padStart(2, "0");
 
 export function bucketLabel(start: number, unit: Unit): string {
 	const d = new Date(start);
+	if (unit === "year") return String(d.getFullYear());
 	if (unit === "month") return `${d.getFullYear()}-${pad2(d.getMonth() + 1)}`;
 	const base = `${pad2(d.getMonth() + 1)}-${pad2(d.getDate())}`;
 	return unit === "day" ? `${base} ${["Sun", "Mon", "Tue", "Wed", "Thu", "Fri", "Sat"][d.getDay()]}` : `${base} wk`;
@@ -296,7 +322,7 @@ export interface SeriesPoint {
 	cost: number;
 }
 
-export const DEFAULT_COUNT: Record<Unit, number> = { day: 14, week: 8, month: 6 };
+export const DEFAULT_COUNT: Record<Unit, number> = { day: 14, week: 8, month: 6, year: 5 };
 
 /** Last `count` buckets ending at `now`, oldest first, empty buckets zero-filled. */
 export async function collectSeries(
@@ -316,7 +342,9 @@ export async function collectSeries(
 				? new Date(y, m, day - i)
 				: unit === "week"
 					? new Date(y, m, day - i * 7)
-					: new Date(y, m - i, 1);
+					: unit === "year"
+						? new Date(y - i, 0, 1)
+						: new Date(y, m - i, 1);
 		starts.push(bucketStart(anchor.getTime(), unit));
 	}
 	const points = new Map<number, SeriesPoint>(
@@ -333,6 +361,27 @@ export async function collectSeries(
 	return starts.map((s) => points.get(s)!);
 }
 
+/**
+ * Whole history: monthly columns while it spans <= 18 months, yearly after that.
+ * Always at least 2 columns so the chart has a shape.
+ */
+export async function collectAllSeries(
+	sessionsRoot: string,
+	now: Date = new Date(),
+	includeCache = true,
+): Promise<{ unit: Unit; series: SeriesPoint[] }> {
+	let first = now.getTime();
+	for (const turns of await loadAll(sessionsRoot)) for (const t of turns) if (t.ts < first) first = t.ts;
+	const f = new Date(first);
+	const months = (now.getFullYear() - f.getFullYear()) * 12 + (now.getMonth() - f.getMonth()) + 1;
+	if (months <= 18) {
+		const unit: Unit = "month";
+		return { unit, series: await collectSeries(sessionsRoot, unit, Math.max(2, months), now, includeCache) };
+	}
+	const years = now.getFullYear() - f.getFullYear() + 1;
+	return { unit: "year", series: await collectSeries(sessionsRoot, "year", years, now, includeCache) };
+}
+
 const EIGHTHS = ["", "▏", "▎", "▍", "▌", "▋", "▊", "▉"];
 const SPARK = "▁▂▃▄▅▆▇█";
 
@@ -350,8 +399,8 @@ export function sparkline(values: number[]): string {
 	return values.map((v) => SPARK[Math.min(7, Math.round((v / max) * 7))]).join("");
 }
 
-const UNIT_TITLE: Record<Unit, string> = { day: "Daily", week: "Weekly", month: "Monthly" };
-const UNIT_NOUN: Record<Unit, string> = { day: "day", week: "week", month: "month" };
+const UNIT_TITLE: Record<Unit, string> = { day: "Daily", week: "Weekly", month: "Monthly", year: "Yearly" };
+const UNIT_NOUN: Record<Unit, string> = { day: "day", week: "week", month: "month", year: "year" };
 const MONTHS = ["Jan", "Feb", "Mar", "Apr", "May", "Jun", "Jul", "Aug", "Sep", "Oct", "Nov", "Dec"];
 
 export type Paint = (kind: "bar" | "now" | "over" | "axis" | "budget" | "dim" | "title", text: string) => string;
@@ -362,6 +411,7 @@ function columnLabels(p: SeriesPoint, unit: Unit): [string, string] {
 	const d = new Date(p.start);
 	if (unit === "day") return [pad2(d.getDate()), ["Su", "Mo", "Tu", "We", "Th", "Fr", "Sa"][d.getDay()]];
 	if (unit === "week") return [`${pad2(d.getMonth() + 1)}-${pad2(d.getDate())}`, "wk"];
+	if (unit === "year") return [String(d.getFullYear()), ""];
 	return [MONTHS[d.getMonth()], String(d.getFullYear())];
 }
 
@@ -372,7 +422,7 @@ function columnLabels(p: SeriesPoint, unit: Unit): [string, string] {
 export function renderChart(
 	series: SeriesPoint[],
 	unit: Unit,
-	opts: { height?: number; budget?: { tokens?: number; cost?: number }; paint?: Paint } = {},
+	opts: { height?: number; budget?: { tokens?: number; cost?: number }; paint?: Paint; title?: string } = {},
 ): string[] {
 	const paint = opts.paint ?? plain;
 	const height = opts.height ?? 8;
@@ -398,8 +448,8 @@ export function renderChart(
 
 	const lines: string[] = [];
 	lines.push(
-		paint("title", `${UNIT_TITLE[unit]} tokens`) +
-			paint("dim", ` · last ${series.length} ${UNIT_NOUN[unit]}s`) +
+		paint("title", opts.title ?? `${UNIT_TITLE[unit]} tokens`) +
+			paint("dim", ` · ${opts.title ? "" : "last "}${series.length} ${UNIT_NOUN[unit]}${series.length === 1 ? "" : "s"}`) +
 			(budgetTokens ? paint("budget", `  ┄ budget ${formatNum(budgetTokens)}`) : ""),
 	);
 
@@ -443,8 +493,6 @@ export function clearCache(): void {
 	FILE_CACHE.clear();
 }
 
-export type Window = "today" | "week" | "month" | "total";
-
 export function budgetsFor(cfg: TokenBurnConfig, w: Window): { tokens?: number; cost?: number } {
 	const b = cfg.budgets;
 	if (w === "today") return { tokens: b.dayTokens, cost: b.dayCost };
@@ -472,7 +520,7 @@ export function themePaint(theme: { fg: (c: any, t: string) => string }): Paint
 }
 
 export default function tokenBurnExtension(pi: ExtensionAPI) {
-	let config: TokenBurnConfig = { ...DEFAULT_CONFIG };
+	let config: TokenBurnConfig = freshConfig();
 
 	const updateUI = async (ctx: ExtensionContext) => {
 		if (!ctx.hasUI) return;
@@ -510,6 +558,7 @@ export default function tokenBurnExtension(pi: ExtensionAPI) {
 				line("Today", report.today, budgetsFor(config, "today")),
 				line("Week", report.week, budgetsFor(config, "week")),
 				line("Month", report.month, budgetsFor(config, "month")),
+				line("Year", report.year),
 				line("Total", report.total),
 				`└${"─".repeat(46)}┘`,
 			];
@@ -531,91 +580,200 @@ export default function tokenBurnExtension(pi: ExtensionAPI) {
 	});
 
 	pi.registerCommand("tokenburn", {
-		description: "TokenBurn stats: show report, toggle widget, or set budgets",
-		getArgumentCompletions: (prefix: string) => {
-			const opts = ["report", "day", "week", "month", "today", "total", "cycle", "window", "cache", "status <today|week|month|total>", "budget <day|week|month> <tokens|$cost>"];
-			return opts.filter((o) => o.startsWith(prefix)).map((value) => ({ value }));
-		},
+		description: "TokenBurn: /tokenburn [day|week|month|year|all] chart, status <window>, cycle, window, budget, cache",
+		getArgumentCompletions: (prefix: string) => completions(prefix),
 		handler: async (args, ctx) => {
-			const [cmd, p1, p2] = (args ?? "").trim().split(/\s+/).filter(Boolean);
-
-			if (cmd === "window" || cmd === "widget") {
-				config.showWidget = !config.showWidget;
+			const cmd = parseCommand(args);
+			const persist = async () => {
 				await saveConfig(config);
 				await updateUI(ctx);
-				ctx.ui.notify(`tokenburn widget: ${config.showWidget ? "enabled" : "hidden"}`, "info");
-				return;
-			}
-
-			const WINDOWS = ["today", "week", "month", "total"] as const;
-			const setWindow = async (w: (typeof WINDOWS)[number]) => {
-				config.statusWindow = w;
-				await saveConfig(config);
-				await updateUI(ctx);
-				ctx.ui.notify(`tokenburn status window: ${w}`, "info");
 			};
 
-			// /tokenburn today|week|month|total  — shortcut for the status bar period
-			if ((WINDOWS as readonly string[]).includes(cmd ?? "")) {
-				await setWindow(cmd as (typeof WINDOWS)[number]);
-				return;
-			}
-			if (cmd === "status" && (WINDOWS as readonly string[]).includes(p1 ?? "")) {
-				await setWindow(p1 as (typeof WINDOWS)[number]);
-				return;
-			}
-			// /tokenburn cycle  — today → week → month → total → today
-			if (cmd === "cycle") {
-				await setWindow(WINDOWS[(WINDOWS.indexOf(config.statusWindow) + 1) % WINDOWS.length]);
-				return;
-			}
+			switch (cmd.kind) {
+				case "window":
+					config.showWidget = !config.showWidget;
+					await persist();
+					ctx.ui.notify(`tokenburn widget: ${config.showWidget ? "enabled" : "hidden"}`, "info");
+					return;
 
-			if (cmd === "cache") {
-				config.includeCache = !config.includeCache;
-				await saveConfig(config);
-				await updateUI(ctx);
-				ctx.ui.notify(`tokenburn: cache tokens ${config.includeCache ? "included" : "excluded"}`, "info");
-				return;
-			}
+				case "status":
+					config.statusWindow = cmd.window;
+					await persist();
+					ctx.ui.notify(`tokenburn status window: ${cmd.window}`, "info");
+					return;
 
-			if (cmd === "budget") {
-				const isCost = (p2 ?? "").startsWith("$");
-				const val = Number((p2 ?? "").replace(/^\$/, ""));
-				if (!["day", "week", "month"].includes(p1 ?? "") || !Number.isFinite(val) || val <= 0) {
-					ctx.ui.notify("Usage: /tokenburn budget <day|week|month> <tokens | $cost>", "warning");
+				case "cycle": {
+					const next = STATUS_WINDOWS[(STATUS_WINDOWS.indexOf(config.statusWindow) + 1) % STATUS_WINDOWS.length];
+					config.statusWindow = next;
+					await persist();
+					ctx.ui.notify(`tokenburn status window: ${next}`, "info");
+					return;
+				}
+
+				case "cache":
+					config.includeCache = !config.includeCache;
+					await persist();
+					ctx.ui.notify(`tokenburn: cache tokens ${config.includeCache ? "included" : "excluded"}`, "info");
+					return;
+
+				case "budget": {
+					const key = `${cmd.period}${cmd.isCost ? "Cost" : "Tokens"}` as keyof TokenBurnConfig["budgets"];
+					config.budgets[key] = cmd.amount;
+					await persist();
+					ctx.ui.notify(
+						`tokenburn ${cmd.period} budget: ${cmd.isCost ? formatCost(cmd.amount) : formatNum(cmd.amount) + " tokens"}`,
+						"info",
+					);
 					return;
 				}
-				const key = `${p1}${isCost ? "Cost" : "Tokens"}` as keyof TokenBurnConfig["budgets"];
-				config.budgets[key] = val;
-				await saveConfig(config);
-				await updateUI(ctx);
-				ctx.ui.notify(`tokenburn ${p1} budget: ${isCost ? formatCost(val) : formatNum(val) + " tokens"}`, "info");
-				return;
-			}
 
-			// /tokenburn [report] [day|week|month]  — summary + chart
-			const unitArg = (cmd === "report" || cmd === "chart" ? p1 : cmd) as string | undefined;
-			const unit: Unit = unitArg === "week" || unitArg === "month" ? unitArg : "day";
-			const report = await collectUsage(getSessionsDir(), new Date(), config.includeCache);
-			const series = await collectSeries(getSessionsDir(), unit, DEFAULT_COUNT[unit], new Date(), config.includeCache);
-			const budgetKey = unit === "day" ? "today" : unit === "week" ? "week" : "month";
-			const formatRow = (name: string, p: PeriodStats) =>
-				`${name.padEnd(10)} ${formatNum(p.tokens).padStart(8)} tokens ${formatCost(p.cost).padStart(9)} · in:${formatNum(p.input)} out:${formatNum(p.output)} cache:${formatNum(p.cacheRead + p.cacheWrite)} · ${p.sessions} sess`;
-
-			const output = [
-				"🔥 TokenBurn Report",
-				"─".repeat(60),
-				formatRow("Today", report.today),
-				formatRow("This Week", report.week),
-				formatRow("This Month", report.month),
-				formatRow("All Time", report.total),
-				"─".repeat(60),
-				...renderChart(series, unit, { budget: budgetsFor(config, budgetKey as Window), paint: config.chartColor ? themePaint(ctx.ui.theme) : undefined }),
-				"─".repeat(60),
-				"/tokenburn [day|week|month] chart · today|week|month|total|cycle status bar · window · budget · cache",
-			].join("\n");
-
-			ctx.ui.notify(output, "info");
+				case "invalid":
+					ctx.ui.notify(cmd.message, "warning");
+					return;
+
+				case "report": {
+					const root = getSessionsDir();
+					const now = new Date();
+					const report = await collectUsage(root, now, config.includeCache);
+					const paint = config.chartColor ? themePaint(ctx.ui.theme) : undefined;
+
+					let chart: string[];
+					if (cmd.view === "all") {
+						const all = await collectAllSeries(root, now, config.includeCache);
+						chart = renderChart(all.series, all.unit, { paint, title: "All time" });
+					} else {
+						const series = await collectSeries(root, cmd.view, DEFAULT_COUNT[cmd.view], now, config.includeCache);
+						const budgetWindow: Window | undefined =
+							cmd.view === "day" ? "today" : cmd.view === "year" ? undefined : cmd.view;
+						chart = renderChart(series, cmd.view, {
+							paint,
+							budget: budgetWindow ? budgetsFor(config, budgetWindow) : undefined,
+						});
+					}
+
+					const row = (name: string, p: PeriodStats) =>
+						`${name.padEnd(10)} ${formatNum(p.tokens).padStart(8)} tokens ${formatCost(p.cost).padStart(9)} · in:${formatNum(p.input)} out:${formatNum(p.output)} cache:${formatNum(p.cacheRead + p.cacheWrite)} · ${p.sessions} sess`;
+					const rule = "─".repeat(60);
+					ctx.ui.notify(
+						[
+							"🔥 TokenBurn Report",
+							rule,
+							row("Today", report.today),
+							row("This Week", report.week),
+							row("This Month", report.month),
+							row("This Year", report.year),
+							row("All Time", report.total),
+							rule,
+							...chart,
+							rule,
+							"/tokenburn day|week|month|year|all · status <window> · cycle · window · budget · cache",
+						].join("\n"),
+						"info",
+					);
+					return;
+				}
+			}
 		},
 	});
 }
+
+// ---------------------------------------------------------------------------
+// Command parsing + completions (pure, unit-tested)
+// ---------------------------------------------------------------------------
+
+export const REPORT_VIEWS = ["day", "week", "month", "year", "all"] as const;
+export type ReportView = (typeof REPORT_VIEWS)[number];
+export const BUDGET_PERIODS = ["day", "week", "month"] as const;
+
+export type Command =
+	| { kind: "report"; view: ReportView }
+	| { kind: "status"; window: Window }
+	| { kind: "cycle" }
+	| { kind: "window" }
+	| { kind: "cache" }
+	| { kind: "budget"; period: (typeof BUDGET_PERIODS)[number]; amount: number; isCost: boolean }
+	| { kind: "invalid"; message: string };
+
+export const USAGE =
+	"Usage: /tokenburn [day|week|month|year|all] · status <today|week|month|year|total> · cycle · window · budget <day|week|month> <tokens|$cost> · cache";
+
+const asView = (w: string | undefined): ReportView | undefined =>
+	w === "today" ? "day" : (REPORT_VIEWS as readonly string[]).includes(w ?? "") ? (w as ReportView) : undefined;
+
+export function parseCommand(args: string | undefined): Command {
+	const [cmd, p1, p2] = (args ?? "").trim().toLowerCase().split(/\s+/).filter(Boolean);
+	if (!cmd) return { kind: "report", view: "day" };
+
+	const view = asView(cmd);
+	if (view) return { kind: "report", view };
+
+	switch (cmd) {
+		case "report":
+		case "chart": {
+			if (p1 === undefined) return { kind: "report", view: "day" };
+			const v = asView(p1);
+			return v ? { kind: "report", view: v } : { kind: "invalid", message: `Unknown view "${p1}". ${USAGE}` };
+		}
+		case "status":
+			return (STATUS_WINDOWS as readonly string[]).includes(p1 ?? "")
+				? { kind: "status", window: p1 as Window }
+				: { kind: "invalid", message: `Usage: /tokenburn status <${STATUS_WINDOWS.join("|")}>` };
+		case "cycle":
+			return { kind: "cycle" };
+		case "window":
+		case "widget":
+			return { kind: "window" };
+		case "cache":
+			return { kind: "cache" };
+		case "budget": {
+			const isCost = (p2 ?? "").startsWith("$");
+			const amount = Number((p2 ?? "").replace(/^\$/, ""));
+			if (!(BUDGET_PERIODS as readonly string[]).includes(p1 ?? "") || !Number.isFinite(amount) || amount <= 0) {
+				return { kind: "invalid", message: "Usage: /tokenburn budget <day|week|month> <tokens | $cost>" };
+			}
+			return { kind: "budget", period: p1 as (typeof BUDGET_PERIODS)[number], amount, isCost };
+		}
+		default:
+			return { kind: "invalid", message: `Unknown option "${cmd}". ${USAGE}` };
+	}
+}
+
+export interface CompletionItem {
+	value: string;
+	/** Required by pi: it calls `label.endsWith(...)` when applying a completion. */
+	label: string;
+	description?: string;
+}
+
+const TOP_LEVEL: CompletionItem[] = [
+	{ value: "day", label: "day", description: "Report + daily chart (last 14 days)" },
+	{ value: "week", label: "week", description: "Report + weekly chart (last 8 weeks)" },
+	{ value: "month", label: "month", description: "Report + monthly chart (last 6 months)" },
+	{ value: "year", label: "year", description: "Report + yearly chart (last 5 years)" },
+	{ value: "all", label: "all", description: "Report + whole history" },
+	{ value: "status ", label: "status", description: "Set the status-bar period" },
+	{ value: "cycle", label: "cycle", description: "Rotate the status-bar period" },
+	{ value: "window", label: "window", description: "Toggle the panel below the editor" },
+	{ value: "budget ", label: "budget", description: "Set a day/week/month budget" },
+	{ value: "cache", label: "cache", description: "Toggle counting cache tokens" },
+];
+
+/** `prefix` is the whole argument text typed so far (pi replaces it with `value`). */
+export function completions(prefix: string): CompletionItem[] | null {
+	const text = (prefix ?? "").toLowerCase().replace(/^\s+/, "");
+	// `tail` = " " when the completed text still needs another word (a budget amount).
+	const sub = (head: string, words: readonly string[], desc: (w: string) => string, tail = ""): CompletionItem[] =>
+		words.map((w) => ({ value: `${head} ${w}${tail}`, label: w, description: desc(w) }));
+
+	let pool: CompletionItem[];
+	if (text.startsWith("status ")) pool = sub("status", STATUS_WINDOWS, (w) => `Show ${w} in the status bar`);
+	else if (text.startsWith("budget ") && !/^budget \S+ /.test(text))
+		pool = sub("budget", BUDGET_PERIODS, (w) => `Set the ${w} budget (tokens or $cost)`, " ");
+	else if (/^(report|chart) /.test(text)) {
+		const head = text.split(" ")[0];
+		pool = sub(head, REPORT_VIEWS, (w) => `${head} for ${w}`);
+	} else pool = TOP_LEVEL;
+
+	const out = pool.filter((i) => i.value.startsWith(text));
+	return out.length > 0 ? out : null;
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-tokenburn",
-  "version": "0.2.0",
+  "version": "0.3.1",
   "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",
@@ -11,6 +11,20 @@
   },
   "homepage": "https://github.com/sorinirimies/pi-tokenburn#readme",
   "files": ["extensions", "README.md", "LICENSE"],
-  "pi": { "extensions": ["./extensions/tokenburn.ts"] },
-  "peerDependencies": { "@earendil-works/pi-coding-agent": "*" }
+  "pi": {
+    "extensions": ["./extensions/tokenburn.ts"]
+  },
+  "scripts": {
+    "test": "bun test",
+    "typecheck": "tsc --noEmit",
+    "check": "nu scripts/ci/quality_gate.nu"
+  },
+  "peerDependencies": {
+    "@earendil-works/pi-coding-agent": "*"
+  },
+  "devDependencies": {
+    "@earendil-works/pi-coding-agent": "^1.1.0",
+    "@types/bun": "^1.4.2",
+    "typescript": "^7.0.2"
+  }
 }