pi-zentui 0.1.12 → 0.1.13

package/extensions/zentui/format.ts

@@ -11,9 +11,11 @@ export type UsageTotals = {
 };
 
 export function formatCount(value: number): string {
-	if (value < 1000) return `${value}`;
+	if (value < 1000) return value.toString();
 	if (value < 10_000) return `${(value / 1000).toFixed(1)}k`;
-	return `${Math.round(value / 1000)}k`;
+	if (value < 1_000_000) return `${Math.round(value / 1000)}k`;
+	if (value < 10_000_000) return `${(value / 1_000_000).toFixed(1)}M`;
+	return `${Math.round(value / 1_000_000)}M`;
 }
 
 export function formatProviderLabel(provider: string | undefined): string {
@@ -38,19 +40,23 @@ export function getUsageTotals(ctx: ExtensionContext): UsageTotals {
 	let output = 0;
 	let cost = 0;
 
-	for (const entry of ctx.sessionManager.getBranch()) {
+	const entries = ctx.sessionManager.getEntries?.() ?? ctx.sessionManager.getBranch();
+	for (const entry of entries) {
 		if (entry.type !== "message" || entry.message.role !== "assistant") continue;
-		const message = entry.message as AssistantMessage;
-		input += message.usage?.input ?? 0;
-		output += message.usage?.output ?? 0;
-		cost += message.usage?.cost?.total ?? 0;
+		const usage = (entry.message as AssistantMessage).usage;
+		input += usage?.input ?? 0;
+		output += usage?.output ?? 0;
+		cost += usage?.cost?.total ?? 0;
 	}
 
 	return { input, output, cost };
 }
 
 export function buildTokenLabel(totals: UsageTotals): string {
-	return `↑${formatCount(totals.input)} ↓${formatCount(totals.output)}`;
+	const parts: string[] = [];
+	if (totals.input) parts.push(`↑${formatCount(totals.input)}`);
+	if (totals.output) parts.push(`↓${formatCount(totals.output)}`);
+	return parts.length > 0 ? parts.join(" ") : "↑0 ↓0";
 }
 
 export function buildCostLabel(totals: UsageTotals): string {

package/extensions/zentui/index.ts

@@ -140,6 +140,13 @@ export default function (pi: ExtensionAPI) {
 		activeTheme = undefined;
 	};
 
+	const syncInteractiveState = (_event: unknown, ctx: ExtensionContext) => {
+		refreshInteractiveState(ctx);
+	};
+	const syncInteractiveAndProjectState = (_event: unknown, ctx: ExtensionContext) => {
+		refreshInteractiveState(ctx, true);
+	};
+
 	pi.on("session_start", async (_event, ctx) => {
 		installUi(ctx);
 	});
@@ -164,31 +171,12 @@ export default function (pi: ExtensionAPI) {
 		cleanupUi(ctx);
 	});
 
-	pi.on("agent_start", async (_event, ctx) => {
-		refreshInteractiveState(ctx);
-	});
-
-	pi.on("agent_end", async (_event, ctx) => {
-		refreshInteractiveState(ctx, true);
-	});
-
-	pi.on("model_select", async (_event, ctx) => {
-		refreshInteractiveState(ctx);
-	});
-
-	pi.on("thinking_level_select", async (_event, ctx) => {
-		refreshInteractiveState(ctx);
-	});
-
-	pi.on("message_end", async (_event, ctx) => {
-		refreshInteractiveState(ctx, true);
-	});
-
-	pi.on("tool_execution_end", async (_event, ctx) => {
-		refreshInteractiveState(ctx, true);
-	});
-
-	pi.on("session_compact", async (_event, ctx) => {
-		refreshInteractiveState(ctx, true);
-	});
+	pi.on("agent_start", syncInteractiveState);
+	pi.on("agent_end", syncInteractiveAndProjectState);
+	pi.on("model_select", syncInteractiveState);
+	pi.on("thinking_level_select", syncInteractiveState);
+	pi.on("message_end", syncInteractiveAndProjectState);
+	pi.on("tool_execution_end", syncInteractiveAndProjectState);
+	pi.on("session_compact", syncInteractiveAndProjectState);
+	pi.on("session_tree", syncInteractiveAndProjectState);
 }

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "pi-zentui",
-	"version": "0.1.12",
+	"version": "0.1.13",
 	"description": "A Starship-inspired statusline and Opencode-style TUI for Pi.",
 	"type": "module",
 	"license": "MIT",
@@ -45,6 +45,6 @@
 		"@earendil-works/pi-tui": "^0.74.0",
 		"@types/node": "^25.5.2",
 		"typescript": "^6.0.2",
-		"vitest": "^3.2.4"
+		"vitest": "^4.1.8"
 	}
 }