pi-tokenburn 0.3.0 → 0.3.1

package/README.md

@@ -86,9 +86,12 @@ Environment overrides for the session directory: `TOKENBURN_PI_SESSIONS`, `PI_CO
 ```bash
 bun install
 just check          # typecheck + tests + pack check + nushell tests (what CI runs)
-just test           # bun test only
+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.
 

package/extensions/tokenburn.ts

@@ -78,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");
@@ -172,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) => {
@@ -209,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) };
@@ -750,13 +761,14 @@ const TOP_LEVEL: CompletionItem[] = [
 /** `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+/, "");
-	const sub = (head: string, words: readonly string[], desc: (w: string) => string): CompletionItem[] =>
-		words.map((w) => ({ value: `${head} ${w}`, label: w, description: desc(w) }));
+	// `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)`);
+		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}`);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-tokenburn",
-  "version": "0.3.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",