@shanepadgett/tau-agent 0.39.2 → 0.41.0

package/extensions/cost-report/index.ts

@@ -0,0 +1,181 @@
+import { access, mkdir, writeFile } from "node:fs/promises";
+import { homedir } from "node:os";
+import { join } from "node:path";
+import type { ExtensionAPI, ExtensionCommandContext } from "@earendil-works/pi-coding-agent";
+import { buildCostReport } from "./analyze.ts";
+import { renderCostReportHtml } from "./html.ts";
+import { CostReportStatusPanel } from "./panel.ts";
+import { LIVE_RANGE_CHOICES, liveRange, MONTH_CHOICES, specificMonthRange, yearChoices } from "./range.ts";
+import type { ReportRange, ReportScope } from "./types.ts";
+
+const COMMAND = "cost-report";
+
+export default function costReportExtension(pi: ExtensionAPI): void {
+	pi.registerCommand(COMMAND, {
+		description: "Generate an HTML cost report from local sessions",
+		handler: async (_args, ctx) => {
+			if (ctx.mode !== "tui") {
+				ctx.ui.notify("/cost-report requires TUI mode", "warning");
+				return;
+			}
+
+			await ctx.waitForIdle();
+
+			const range = await pickRange(ctx);
+			if (!range) return;
+
+			const scope = await pickScope(ctx);
+			if (!scope) return;
+
+			const secondary = `${range.label} · ${scope === "project" ? "Current project" : "All sessions"}`;
+
+			const path = outputPath(range, scope);
+			if (await fileExists(path)) {
+				try {
+					await openReport(pi, ctx, path);
+				} catch {
+					// Still notify the path if the OS open fails.
+				}
+				ctx.ui.notify(`Opening existing cost report · ${path}`, "info");
+				return;
+			}
+
+			const outcome = await ctx.ui.custom<
+				| { kind: "ok"; path: string }
+				| { kind: "empty" }
+				| { kind: "cancelled" }
+				| { kind: "error"; message: string }
+			>((tui, theme, _keys, done) => {
+				const panel = new CostReportStatusPanel(tui, theme, secondary);
+				void (async () => {
+					try {
+						const report = await buildCostReport({
+							cwd: ctx.cwd,
+							range,
+							scope,
+							signal: panel.signal,
+							onProgress: (status) => panel.update(status),
+						});
+						if (panel.signal.aborted) {
+							done({ kind: "cancelled" });
+							return;
+						}
+						if (!report) {
+							done({ kind: "empty" });
+							return;
+						}
+
+						panel.update("Writing file…");
+						await mkdir(join(homedir(), ".pi", "tau", "cost-reports"), { recursive: true });
+						await writeFile(path, renderCostReportHtml(report), "utf8");
+
+						if (panel.signal.aborted) {
+							done({ kind: "cancelled" });
+							return;
+						}
+
+						panel.update("Opening…");
+						try {
+							await openReport(pi, ctx, path);
+						} catch {
+							// File is written; notify still carries the path if open fails.
+						}
+						done({ kind: "ok", path });
+					} catch (error) {
+						if (panel.signal.aborted) {
+							done({ kind: "cancelled" });
+							return;
+						}
+						done({
+							kind: "error",
+							message: error instanceof Error ? error.message : String(error),
+						});
+					} finally {
+						panel.dispose();
+					}
+				})();
+				return panel;
+			});
+
+			if (outcome.kind === "ok") {
+				ctx.ui.notify(`Cost report written · ${outcome.path}`, "info");
+				return;
+			}
+			if (outcome.kind === "empty") {
+				ctx.ui.notify(`No session spend found for ${secondary}.`, "warning");
+				return;
+			}
+			if (outcome.kind === "cancelled") {
+				ctx.ui.notify("Cost report cancelled.", "info");
+				return;
+			}
+			ctx.ui.notify(`Unable to build cost report: ${outcome.message}`, "error");
+		},
+	});
+}
+
+async function pickRange(ctx: ExtensionCommandContext): Promise<ReportRange | undefined> {
+	const choice = await ctx.ui.select(
+		"Time frame",
+		LIVE_RANGE_CHOICES.map((item) => item.label),
+	);
+	if (!choice) return undefined;
+
+	const selected = LIVE_RANGE_CHOICES.find((item) => item.label === choice);
+	if (!selected) return undefined;
+	if (selected.id !== "specific-month") return liveRange(selected.id);
+
+	const monthLabel = await ctx.ui.select(
+		"Month",
+		MONTH_CHOICES.map((item) => item.label),
+	);
+	if (!monthLabel) return undefined;
+	const month = MONTH_CHOICES.find((item) => item.label === monthLabel);
+	if (!month) return undefined;
+
+	const years = yearChoices().map(String);
+	const yearLabel = await ctx.ui.select("Year", years);
+	if (!yearLabel) return undefined;
+	const year = Number(yearLabel);
+	if (!Number.isFinite(year)) return undefined;
+
+	return specificMonthRange(year, month.month);
+}
+
+async function pickScope(ctx: ExtensionCommandContext): Promise<ReportScope | undefined> {
+	const choice = await ctx.ui.select("Scope", ["Current project", "All sessions"]);
+	if (choice === "Current project") return "project";
+	if (choice === "All sessions") return "all";
+	return undefined;
+}
+
+function outputPath(range: ReportRange, scope: ReportScope): string {
+	// One file per window identity + scope. Re-running the same picks reopens this path.
+	return join(homedir(), ".pi", "tau", "cost-reports", `cost-report-${range.slug}-${scope}.html`);
+}
+
+async function fileExists(path: string): Promise<boolean> {
+	try {
+		await access(path);
+		return true;
+	} catch {
+		return false;
+	}
+}
+
+async function openReport(pi: ExtensionAPI, ctx: ExtensionCommandContext, path: string): Promise<void> {
+	const command = openCommand(path);
+	const result = await pi.exec(command.command, command.args, {
+		signal: ctx.signal,
+		timeout: 5_000,
+	});
+	if (result.code !== 0) {
+		throw new Error(result.stderr.trim() || `open command failed with exit ${result.code}`);
+	}
+}
+
+function openCommand(path: string): { command: string; args: string[] } {
+	if (process.platform === "darwin") return { command: "open", args: [path] };
+	if (process.platform === "win32") return { command: "cmd", args: ["/c", "start", "", path] };
+	return { command: "xdg-open", args: [path] };
+}

package/extensions/cost-report/panel.ts

@@ -0,0 +1,64 @@
+import type { Theme } from "@earendil-works/pi-coding-agent";
+import { getKeybindings, type Component, type TUI, truncateToWidth } from "@earendil-works/pi-tui";
+import { bindingHint, ToolPanel, type ToolPanelConfig } from "@shanepadgett/tau-tui";
+
+export class CostReportStatusPanel implements Component {
+	private readonly tui: TUI;
+	private readonly panel: ToolPanel;
+	private readonly config: ToolPanelConfig;
+	private readonly controller = new AbortController();
+	private readonly startedAt = Date.now();
+	private readonly timer: ReturnType<typeof setInterval>;
+	private status = "Starting…";
+	private readonly secondary: string;
+
+	constructor(tui: TUI, theme: Theme, secondary: string) {
+		this.tui = tui;
+		this.secondary = secondary;
+		this.config = {
+			title: "Cost report",
+			secondary,
+			body: {
+				render: (width) => [truncateToWidth(theme.fg("muted", this.status), width, "…")],
+				invalidate: () => {},
+			},
+			footer: { kind: "hints", hints: [bindingHint("tui.select.cancel", "cancel")] },
+			border: "box",
+		};
+		this.panel = new ToolPanel(theme, this.config);
+		this.timer = setInterval(() => this.tui.requestRender(), 1_000);
+		this.timer.unref();
+	}
+
+	get signal(): AbortSignal {
+		return this.controller.signal;
+	}
+
+	handleInput(data: string): void {
+		if (!getKeybindings().matches(data, "tui.select.cancel")) return;
+		if (this.controller.signal.aborted) return;
+		this.update("Cancelling…");
+		this.controller.abort();
+	}
+
+	update(status: string): void {
+		this.status = status.trim() || this.status;
+		this.tui.requestRender();
+	}
+
+	render(width: number): string[] {
+		const elapsed = Math.floor((Date.now() - this.startedAt) / 1_000);
+		const clock = elapsed < 60 ? `${elapsed}s` : `${Math.floor(elapsed / 60)}m ${elapsed % 60}s`;
+		this.config.title = `Cost report · ${clock}`;
+		this.config.secondary = this.secondary;
+		return this.panel.render(width);
+	}
+
+	invalidate(): void {
+		this.panel.invalidate();
+	}
+
+	dispose(): void {
+		clearInterval(this.timer);
+	}
+}

package/extensions/cost-report/range.ts

@@ -0,0 +1,145 @@
+import type { LiveRangeKind, ReportRange } from "./types.ts";
+
+const MONTH_LABELS = [
+	"January",
+	"February",
+	"March",
+	"April",
+	"May",
+	"June",
+	"July",
+	"August",
+	"September",
+	"October",
+	"November",
+	"December",
+] as const;
+
+export const LIVE_RANGE_CHOICES: ReadonlyArray<{ id: LiveRangeKind | "specific-month"; label: string }> = [
+	{ id: "past-7-days", label: "Past 7 days" },
+	{ id: "current-week", label: "Current week" },
+	{ id: "current-month", label: "Current month" },
+	{ id: "year-to-date", label: "Year to date" },
+	{ id: "specific-month", label: "Specific month…" },
+];
+
+export const MONTH_CHOICES: ReadonlyArray<{ month: number; label: string }> = MONTH_LABELS.map((label, month) => ({
+	month,
+	label,
+}));
+
+export function yearChoices(now = new Date()): number[] {
+	const current = now.getFullYear();
+	return Array.from({ length: 6 }, (_, index) => current - index);
+}
+
+export function liveRange(kind: LiveRangeKind, now = new Date()): ReportRange {
+	const endMs = now.getTime();
+	if (kind === "past-7-days") {
+		const start = startOfLocalDay(now);
+		start.setDate(start.getDate() - 6);
+		// Slides daily; slug pins the window end day so same-day reopens match.
+		return {
+			kind,
+			label: "Past 7 days",
+			slug: `7d-${localDateKey(endMs)}`,
+			startMs: start.getTime(),
+			endMs,
+			layout: "row",
+		};
+	}
+	if (kind === "current-week") {
+		const start = startOfLocalDay(now);
+		const day = start.getDay(); // 0 Sun … 6 Sat
+		const mondayOffset = day === 0 ? -6 : 1 - day;
+		start.setDate(start.getDate() + mondayOffset);
+		return {
+			kind,
+			label: "Current week",
+			slug: `week-${localDateKey(start.getTime())}`,
+			startMs: start.getTime(),
+			endMs,
+			layout: "row",
+		};
+	}
+	if (kind === "current-month") {
+		const start = new Date(now.getFullYear(), now.getMonth(), 1);
+		const ym = `${now.getFullYear()}-${String(now.getMonth() + 1).padStart(2, "0")}`;
+		return {
+			kind,
+			label: "Current month",
+			slug: `month-${ym}`,
+			startMs: start.getTime(),
+			endMs,
+			layout: "row",
+			year: now.getFullYear(),
+			month: now.getMonth(),
+		};
+	}
+	const start = new Date(now.getFullYear(), 0, 1);
+	return {
+		kind,
+		label: "Year to date",
+		slug: `ytd-${now.getFullYear()}`,
+		startMs: start.getTime(),
+		endMs,
+		layout: "ytd",
+		year: now.getFullYear(),
+	};
+}
+
+export function specificMonthRange(year: number, month: number, now = new Date()): ReportRange {
+	const start = new Date(year, month, 1);
+	const endExclusive = new Date(year, month + 1, 1);
+	const endMs = Math.min(now.getTime(), endExclusive.getTime() - 1);
+	const monthLabel = MONTH_LABELS[month] ?? `Month ${month + 1}`;
+	const ym = `${year}-${String(month + 1).padStart(2, "0")}`;
+	return {
+		kind: "specific-month",
+		label: `${monthLabel} ${year}`,
+		slug: ym,
+		startMs: start.getTime(),
+		endMs,
+		layout: "row",
+		year,
+		month,
+	};
+}
+
+export function startOfLocalDay(date: Date): Date {
+	return new Date(date.getFullYear(), date.getMonth(), date.getDate());
+}
+
+export function localDateKey(timestampMs: number): string {
+	const date = new Date(timestampMs);
+	const y = date.getFullYear();
+	const m = String(date.getMonth() + 1).padStart(2, "0");
+	const d = String(date.getDate()).padStart(2, "0");
+	return `${y}-${m}-${d}`;
+}
+
+export function shortDateLabel(timestampMs: number): string {
+	const date = new Date(timestampMs);
+	const months = ["Jan", "Feb", "Mar", "Apr", "May", "Jun", "Jul", "Aug", "Sep", "Oct", "Nov", "Dec"];
+	return `${date.getDate()} ${months[date.getMonth()]}`;
+}
+
+export function isWeekend(timestampMs: number): boolean {
+	const day = new Date(timestampMs).getDay();
+	return day === 0 || day === 6;
+}
+
+export function eachLocalDay(startMs: number, endMs: number): number[] {
+	const days: number[] = [];
+	const cursor = startOfLocalDay(new Date(startMs));
+	const end = startOfLocalDay(new Date(endMs));
+	while (cursor.getTime() <= end.getTime()) {
+		days.push(cursor.getTime());
+		cursor.setDate(cursor.getDate() + 1);
+	}
+	return days;
+}
+
+export function daysInMonth(year: number, month: number): number {
+	return new Date(year, month + 1, 0).getDate();
+}

package/extensions/cost-report/types.ts

@@ -0,0 +1,93 @@
+export type ReportScope = "project" | "all";
+
+export type LiveRangeKind = "past-7-days" | "current-week" | "current-month" | "year-to-date";
+
+export type RangeKind = LiveRangeKind | "specific-month";
+
+export interface ReportRange {
+	kind: RangeKind;
+	label: string;
+	slug: string;
+	startMs: number;
+	endMs: number;
+	/** Single full-width day row vs one row per month (YTD). */
+	layout: "row" | "ytd";
+	year?: number;
+	month?: number; // 0-11
+}
+
+export interface CostTotals {
+	input: number;
+	output: number;
+	cacheRead: number;
+	cacheWrite: number;
+	totalTokens: number;
+	cost: number;
+}
+
+export interface ModelCost {
+	key: string;
+	provider: string;
+	model: string;
+	cost: number;
+	tokens: number;
+	sessions: number;
+}
+
+export interface SubagentCost {
+	agent: string;
+	cost: number;
+	calls: number;
+}
+
+export interface ProjectCost {
+	key: string;
+	label: string;
+	cost: number;
+	sessions: number;
+	tokens: number;
+}
+
+export interface SessionModelCost {
+	key: string;
+	label: string;
+	cost: number;
+}
+
+export interface SessionCost {
+	id: string;
+	path: string;
+	name: string;
+	projectKey: string;
+	projectLabel: string;
+	startedAtMs: number;
+	cost: number;
+	tokens: number;
+	models: SessionModelCost[];
+}
+
+export interface DayCost {
+	dateKey: string; // YYYY-MM-DD local
+	label: string;
+	timestampMs: number;
+	cost: number;
+	weekend: boolean;
+}
+
+export interface CostReport {
+	generatedAtMs: number;
+	cwd: string;
+	scope: ReportScope;
+	range: ReportRange;
+	directCost: number;
+	subagentCost: number;
+	totalCost: number;
+	totalTokens: number;
+	sessionCount: number;
+	projectCount: number;
+	days: DayCost[];
+	models: ModelCost[];
+	subagents: SubagentCost[];
+	projects: ProjectCost[];
+	sessions: SessionCost[];
+}

package/extensions/tau-help/help.md

@@ -40,6 +40,10 @@ Adds `/clear-screen` to clear terminal output without changing the session.
 
 Adds `/commit` for semantic commit grouping, review, and committing selected repository changes.
 
+## cost-report
+
+Adds `/cost-report` to build an HTML spend report from local session usage. Pick a time frame (past 7 days, current week, current month, year to date, or a specific month) and scope (current project or all sessions). Tau scans sessions, writes under `~/.pi/tau/cost-reports/`, opens the file, and notifies with the path. Empty windows warn without writing a file.
+
 ## context
 
 Adds `/context` to inject reusable repository work scopes from `.pi/contexts`, and `/context-sync` or `/context-sync <nudge>` for human-driven catalog sync. Selecting entries injects them once into the conversation: `read` paths as complete files, `show` targets as current declaration slices, `outline` paths as Explore structures, and one hidden note listing `references` plus instructions to treat the injected material as current. Run `/context` again to inject more. Manual sync replaces the editor with a status panel; Escape or Ctrl+C cancels. When `sync.automation` is on, coding agent can also run `context-sync` after meaningful uncommitted work. Sync catalogs durable code and long-lived documentation; recurring scratch, planning, interview, and rough-idea paths belong in `validation.ignoreGlobs`. `sync.enabled` is master switch for command, automation, and validation auto-run. Context validation is off by default; when on (and sync enabled), Tau auto-runs context-sync on failure. Domain folders are `NN_slug` tabs (ordered by the two-digit prefix; UI shows the slug), TOML files are concepts, and TOML sections are selectable entries.

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@shanepadgett/tau-agent",
-	"version": "0.39.2",
+	"version": "0.41.0",
 	"description": "Tau is a custom agentic harness built with pi extensions",
 	"type": "module",
 	"main": "./src/index.ts",
@@ -35,7 +35,7 @@
 	],
 	"dependencies": {
 		"@ast-grep/wasm": "0.45.1",
-		"@shanepadgett/tau-tui": "0.39.2",
+		"@shanepadgett/tau-tui": "0.41.0",
 		"@vscode/tree-sitter-wasm": "0.3.1",
 		"image-size": "2.0.2",
 		"smol-toml": "1.8.0",