@danypops/pi-jittor 0.2.0 → 0.3.0

package/extension/src/{tui.ts → observability/status.ts}

@@ -1,5 +1,3 @@
-import type { ExtensionCommandContext } from "@earendil-works/pi-coding-agent";
-import { matchesKey, truncateToWidth } from "@earendil-works/pi-tui";
 import {
 	HUMAN_STATUS_MAX_SOURCES,
 	HUMAN_TEXT_FIELD_MAX_CHARACTERS,
@@ -8,9 +6,12 @@ import {
 	type Route,
 	type RouterStatus,
 	type StoredMetricObservation,
+	TELEMETRY_STALE_AFTER_MS,
 } from "@danypops/jittor";
+import type { ExtensionCommandContext } from "@earendil-works/pi-coding-agent";
+import { matchesKey, truncateToWidth } from "@earendil-works/pi-tui";
+import { sessionSecretField } from "../session-identity.ts";
 import type { ProviderBudget } from "./footer.ts";
-import { sessionSecretField } from "./session-identity.ts";
 
 export interface JittorPanelClient {
 	call(operation: string, input: unknown): Promise<any>;
@@ -18,22 +19,34 @@ export interface JittorPanelClient {
 
 export function providerBudgetMetricQuery(status: RouterStatus): MetricQuery | null {
 	switch (status.currentRoute?.provider) {
-		case "openai-codex": return { source: "codex-subscription", metric: "used-fraction", order: "desc", limit: 100 };
-		case "openrouter": return { source: "openrouter", order: "desc", limit: 20 };
-		case "anthropic": return { source: "anthropic", metric: "used-fraction", order: "desc", limit: 20 };
-		case "anthropic-vertex": return { source: "anthropic-vertex", metric: "used-fraction", order: "desc", limit: 20 };
-		default: return null;
+		case "openai-codex":
+			return { source: "codex-subscription", metric: "used-fraction", order: "desc", limit: 100 };
+		case "openrouter":
+			return { source: "openrouter", order: "desc", limit: 20 };
+		case "anthropic":
+			return { source: "anthropic", metric: "used-fraction", order: "desc", limit: 20 };
+		case "anthropic-vertex":
+			return { source: "anthropic-vertex", metric: "used-fraction", order: "desc", limit: 20 };
+		default:
+			return null;
 	}
 }
 
 type PanelAction = "pause" | "resume" | "refresh" | "override" | "clear-override" | "close";
 
-function latest(rows: StoredMetricObservation[], predicate: (row: StoredMetricObservation) => boolean): StoredMetricObservation | undefined {
+function latest(
+	rows: StoredMetricObservation[],
+	predicate: (row: StoredMetricObservation) => boolean,
+): StoredMetricObservation | undefined {
 	return rows.filter(predicate).sort((left, right) => right.observedAt - left.observedAt || right.id - left.id)[0];
 }
 
 function sanitizedText(value: string): string {
-	return value.replace(/[\r\n\t]/g, " ").replace(/ +/g, " ").trim().slice(0, HUMAN_TEXT_FIELD_MAX_CHARACTERS);
+	return value
+		.replace(/[\r\n\t]/g, " ")
+		.replace(/ +/g, " ")
+		.trim()
+		.slice(0, HUMAN_TEXT_FIELD_MAX_CHARACTERS);
 }
 
 function routeText(route: Route): string {
@@ -45,26 +58,31 @@ function normalizedIdentity(value: unknown): string {
 }
 
 function longestWindow(rows: StoredMetricObservation[]): StoredMetricObservation | undefined {
-	return [...rows].sort((left, right) =>
-		Number(right.attributes["windowSeconds"] ?? 0) - Number(left.attributes["windowSeconds"] ?? 0)
-		|| right.observedAt - left.observedAt
-		|| right.id - left.id,
+	return [...rows].sort(
+		(left, right) =>
+			Number(right.attributes.windowSeconds ?? 0) - Number(left.attributes.windowSeconds ?? 0) ||
+			right.observedAt - left.observedAt ||
+			right.id - left.id,
 	)[0];
 }
 
 function codexWindowForModel(rows: StoredMetricObservation[], model: string): StoredMetricObservation | undefined {
-	const codexRows = rows.filter((row) => row.source === "codex-subscription" && row.metric === "used-fraction" && typeof row.value === "number");
+	const codexRows = rows.filter(
+		(row) => row.source === "codex-subscription" && row.metric === "used-fraction" && typeof row.value === "number",
+	);
 	const modelIdentity = normalizedIdentity(model);
 	const matchingAdditional = codexRows.filter((row) => {
-		const limitId = normalizedIdentity(row.attributes["limitId"]);
-		const limitName = normalizedIdentity(row.attributes["limitName"]);
+		const limitId = normalizedIdentity(row.attributes.limitId);
+		const limitName = normalizedIdentity(row.attributes.limitName);
 		return limitId !== "codex" && limitName.length > 0 && limitName === modelIdentity;
 	});
 	if (matchingAdditional.length > 0) return longestWindow(matchingAdditional);
-	return longestWindow(codexRows.filter((row) => {
-		const limitId = normalizedIdentity(row.attributes["limitId"]);
-		return limitId === "codex" || (limitId.length === 0 && row.scope.startsWith("codex:"));
-	}));
+	return longestWindow(
+		codexRows.filter((row) => {
+			const limitId = normalizedIdentity(row.attributes.limitId);
+			return limitId === "codex" || (limitId.length === 0 && row.scope.startsWith("codex:"));
+		}),
+	);
 }
 
 function compactWindowName(seconds: number): string {
@@ -87,25 +105,68 @@ function windowName(seconds: number): string {
  * could ever read -- see google-vertex-contracts.ts); the footer omits the segment entirely rather
  * than showing a `?` that can never resolve.
  */
-export function buildFooterBudget(status: RouterStatus, metrics: StoredMetricObservation[]): ProviderBudget | null | undefined {
-	if (!status.ready || !status.currentRoute) return null;
+type CodexTelemetryState = "available" | "missing" | "failed" | "stale";
+
+function codexTelemetryState(status: RouterStatus, now: number): CodexTelemetryState {
+	const source = status.sources.find((candidate) => candidate.id === "codex-subscription");
+	if (!source) return "missing";
+	if (!source.ok) return "failed";
+	if (source.observedAt !== undefined && now - source.observedAt > TELEMETRY_STALE_AFTER_MS) return "stale";
+	return "available";
+}
+
+function unavailableCodexBudget(
+	label: string,
+	reason: "reset pending" | "telemetry unavailable" | "telemetry failed" | "telemetry stale",
+): ProviderBudget {
+	return { kind: "unavailable", label, valueText: reason };
+}
+
+export function buildFooterBudget(
+	status: RouterStatus,
+	metrics: StoredMetricObservation[],
+	now = Date.now(),
+): ProviderBudget | null | undefined {
+	if (!status.currentRoute) return null;
 	if (status.currentRoute.provider === "openai-codex") {
 		const codex = codexWindowForModel(metrics, status.currentRoute.model);
-		if (!codex || typeof codex.value !== "number") return null;
-		const resetsAtSeconds = Number(codex.attributes["resetsAt"]);
+		const sourceState = codexTelemetryState(status, now);
+		if (!codex || typeof codex.value !== "number") {
+			if (sourceState === "missing") return unavailableCodexBudget("Codex", "telemetry unavailable");
+			if (sourceState === "failed") return unavailableCodexBudget("Codex", "telemetry failed");
+			if (sourceState === "stale") return unavailableCodexBudget("Codex", "telemetry stale");
+			return null;
+		}
+		const label = compactWindowName(Number(codex.attributes.windowSeconds ?? 0));
+		const resetsAtSeconds = Number(codex.attributes.resetsAt);
+		const resetsAt = Number.isFinite(resetsAtSeconds) && resetsAtSeconds > 0 ? resetsAtSeconds * 1_000 : undefined;
+		if (resetsAt !== undefined && resetsAt <= now) return unavailableCodexBudget(label, "reset pending");
+		if (now - codex.observedAt > TELEMETRY_STALE_AFTER_MS) {
+			if (sourceState === "missing") return unavailableCodexBudget(label, "telemetry unavailable");
+			if (sourceState === "failed") return unavailableCodexBudget(label, "telemetry failed");
+			return unavailableCodexBudget(label, "telemetry stale");
+		}
 		return {
 			kind: "bounded",
-			label: compactWindowName(Number(codex.attributes["windowSeconds"] ?? 0)),
+			label,
 			remainingFraction: 1 - codex.value,
 			observedAt: codex.observedAt,
-			...(Number.isFinite(resetsAtSeconds) && resetsAtSeconds > 0 ? { resetsAt: resetsAtSeconds * 1_000 } : {}),
+			...(resetsAt !== undefined ? { resetsAt } : {}),
 		};
 	}
+	if (!status.ready) return null;
 	if (status.currentRoute.provider === "anthropic") {
-		const anthropic = latest(metrics, (row) => row.source === "anthropic" && row.metric === "used-fraction" && row.scope === "tokens" && typeof row.value === "number")
-			?? latest(metrics, (row) => row.source === "anthropic" && row.metric === "used-fraction" && row.scope === "requests" && typeof row.value === "number");
+		const anthropic =
+			latest(
+				metrics,
+				(row) => row.source === "anthropic" && row.metric === "used-fraction" && row.scope === "tokens" && typeof row.value === "number",
+			) ??
+			latest(
+				metrics,
+				(row) => row.source === "anthropic" && row.metric === "used-fraction" && row.scope === "requests" && typeof row.value === "number",
+			);
 		if (!anthropic || typeof anthropic.value !== "number") return null;
-		const resetsAt = Number(anthropic.attributes["resetsAt"]);
+		const resetsAt = Number(anthropic.attributes.resetsAt);
 		return {
 			kind: "bounded",
 			label: anthropic.scope === "tokens" ? "tok" : "req",
@@ -121,10 +182,19 @@ export function buildFooterBudget(status: RouterStatus, metrics: StoredMetricObs
 		// pool even if the header shape is identical. If nothing was ever observed, this stays null
 		// (may still resolve later), not undefined (never possible) -- unlike google-vertex, this
 		// provider's transport has not been shown to structurally lack the signal.
-		const anthropicVertex = latest(metrics, (row) => row.source === "anthropic-vertex" && row.metric === "used-fraction" && row.scope === "tokens" && typeof row.value === "number")
-			?? latest(metrics, (row) => row.source === "anthropic-vertex" && row.metric === "used-fraction" && row.scope === "requests" && typeof row.value === "number");
+		const anthropicVertex =
+			latest(
+				metrics,
+				(row) =>
+					row.source === "anthropic-vertex" && row.metric === "used-fraction" && row.scope === "tokens" && typeof row.value === "number",
+			) ??
+			latest(
+				metrics,
+				(row) =>
+					row.source === "anthropic-vertex" && row.metric === "used-fraction" && row.scope === "requests" && typeof row.value === "number",
+			);
 		if (!anthropicVertex || typeof anthropicVertex.value !== "number") return null;
-		const resetsAt = Number(anthropicVertex.attributes["resetsAt"]);
+		const resetsAt = Number(anthropicVertex.attributes.resetsAt);
 		return {
 			kind: "bounded",
 			label: anthropicVertex.scope === "tokens" ? "vtok" : "vreq",
@@ -135,9 +205,12 @@ export function buildFooterBudget(status: RouterStatus, metrics: StoredMetricObs
 	}
 	if (status.currentRoute.provider === "openrouter") {
 		const openRouter = latest(metrics, (row) => row.source === "openrouter" && row.metric === "usage" && typeof row.value === "number");
-		const remaining = latest(metrics, (row) => row.source === "openrouter" && row.metric === "remaining-fraction" && typeof row.value === "number");
+		const remaining = latest(
+			metrics,
+			(row) => row.source === "openrouter" && row.metric === "remaining-fraction" && typeof row.value === "number",
+		);
 		if (remaining && typeof remaining.value === "number" && (!openRouter || remaining.observedAt >= openRouter.observedAt)) {
-			const reset = typeof remaining.attributes["reset"] === "string" ? sanitizedText(remaining.attributes["reset"]) : undefined;
+			const reset = typeof remaining.attributes.reset === "string" ? sanitizedText(remaining.attributes.reset) : undefined;
 			return {
 				kind: "bounded",
 				label: "OR",
@@ -152,79 +225,122 @@ export function buildFooterBudget(status: RouterStatus, metrics: StoredMetricObs
 	return undefined;
 }
 
-export function formatFooterStatus(status: RouterStatus, metrics: StoredMetricObservation[]): string {
-	const budget = buildFooterBudget(status, metrics);
+export function formatFooterStatus(status: RouterStatus, metrics: StoredMetricObservation[], now = Date.now()): string {
+	const budget = buildFooterBudget(status, metrics, now);
 	if (!budget) return "";
-	return budget.kind === "unbounded" ? budget.valueText : `${budget.label} ${(budget.remainingFraction * 100).toFixed(1)}% left`;
+	if (budget.kind === "unbounded") return budget.valueText;
+	if (budget.kind === "unavailable") return `${budget.label} ${budget.valueText}`;
+	return `${budget.label} ${(budget.remainingFraction * 100).toFixed(1)}% left`;
 }
 
 function nextAction(action: PolicyAction | undefined): string {
 	switch (action) {
-		case "continue": return "throttle";
-		case "throttle": return "lower thinking";
-		case "lower-thinking": return "switch model";
-		case "switch-model": return "switch provider";
-		case "switch-provider": return "halt";
-		case "halt": return "halted";
-		default: return "waiting for decision";
+		case "continue":
+			return "throttle";
+		case "throttle":
+			return "lower thinking";
+		case "lower-thinking":
+			return "switch model";
+		case "switch-model":
+			return "switch provider";
+		case "switch-provider":
+			return "halt";
+		case "halt":
+			return "halted";
+		default:
+			return "waiting for decision";
 	}
 }
 
 function burnLine(rows: StoredMetricObservation[], current: StoredMetricObservation, now: number): string {
 	const previous = rows
-		.filter((row) => row.source === current.source && row.scope === current.scope && row.metric === current.metric && row.id !== current.id && row.observedAt < current.observedAt)
+		.filter(
+			(row) =>
+				row.source === current.source &&
+				row.scope === current.scope &&
+				row.metric === current.metric &&
+				row.id !== current.id &&
+				row.observedAt < current.observedAt,
+		)
 		.sort((left, right) => right.observedAt - left.observedAt)[0];
-	const resetsAt = Number(current.attributes["resetsAt"] ?? 0) * 1_000;
+	const resetsAt = Number(current.attributes.resetsAt ?? 0) * 1_000;
 	const remainingSeconds = (resetsAt - now) / 1_000;
 	const sustainable = typeof current.value === "number" && remainingSeconds > 0 ? (1 - current.value) / remainingSeconds : null;
-	const observed = previous && typeof previous.value === "number" && typeof current.value === "number" && current.observedAt > previous.observedAt
-		? (current.value - previous.value) / ((current.observedAt - previous.observedAt) / 1_000)
-		: null;
-	const perHour = (rate: number | null) => rate === null ? "n/a" : `${(rate * 3_600 * 100).toFixed(2)}%/h`;
+	const observed =
+		previous && typeof previous.value === "number" && typeof current.value === "number" && current.observedAt > previous.observedAt
+			? (current.value - previous.value) / ((current.observedAt - previous.observedAt) / 1_000)
+			: null;
+	const perHour = (rate: number | null) => (rate === null ? "n/a" : `${(rate * 3_600 * 100).toFixed(2)}%/h`);
 	return `Burn: observed ${perHour(observed)} · sustainable ${perHour(sustainable)}`;
 }
 
 export function buildStatusView(status: RouterStatus, metrics: StoredMetricObservation[], now = Date.now()): string[] {
 	const lines = [status.ready ? "Ready" : "Not ready"];
 	const codex = status.currentRoute?.provider === "openai-codex" ? codexWindowForModel(metrics, status.currentRoute.model) : undefined;
-	if (codex && typeof codex.value === "number") {
-		const seconds = Number(codex.attributes["windowSeconds"] ?? 0);
+	const budget = buildFooterBudget(status, metrics, now);
+	if (codex && typeof codex.value === "number" && budget?.kind === "bounded") {
+		const seconds = Number(codex.attributes.windowSeconds ?? 0);
 		lines.push(`Codex ${windowName(seconds)}: ${((1 - codex.value) * 100).toFixed(1)}% left`);
 		lines.push(burnLine(metrics, codex, now));
+	} else if (status.currentRoute?.provider === "openai-codex" && budget?.kind === "unavailable") {
+		const seconds = Number(codex?.attributes.windowSeconds ?? 0);
+		lines.push(`Codex ${codex ? windowName(seconds) : "subscription"}: ${budget.valueText}`);
 	}
-	const openRouter = status.currentRoute?.provider === "openrouter"
-		? latest(metrics, (row) => row.source === "openrouter" && row.metric === "usage" && typeof row.value === "number")
-		: undefined;
+	const openRouter =
+		status.currentRoute?.provider === "openrouter"
+			? latest(metrics, (row) => row.source === "openrouter" && row.metric === "usage" && typeof row.value === "number")
+			: undefined;
 	if (openRouter && typeof openRouter.value === "number") lines.push(`OpenRouter spend: $${openRouter.value.toFixed(3)}`);
-	const anthropic = status.currentRoute?.provider === "anthropic"
-		? latest(metrics, (row) => row.source === "anthropic" && row.metric === "used-fraction" && row.scope === "tokens" && typeof row.value === "number")
-			?? latest(metrics, (row) => row.source === "anthropic" && row.metric === "used-fraction" && row.scope === "requests" && typeof row.value === "number")
-		: undefined;
-	if (anthropic && typeof anthropic.value === "number") lines.push(`Anthropic ${anthropic.scope}: ${((1 - anthropic.value) * 100).toFixed(1)}% left`);
+	const anthropic =
+		status.currentRoute?.provider === "anthropic"
+			? (latest(
+					metrics,
+					(row) => row.source === "anthropic" && row.metric === "used-fraction" && row.scope === "tokens" && typeof row.value === "number",
+				) ??
+				latest(
+					metrics,
+					(row) =>
+						row.source === "anthropic" && row.metric === "used-fraction" && row.scope === "requests" && typeof row.value === "number",
+				))
+			: undefined;
+	if (anthropic && typeof anthropic.value === "number")
+		lines.push(`Anthropic ${anthropic.scope}: ${((1 - anthropic.value) * 100).toFixed(1)}% left`);
 	if (status.currentRoute) lines.push(`Route: ${routeText(status.currentRoute)}`);
-	if (status.lastDecision) lines.push(`Pressure: ${Number.isFinite(status.lastDecision.pressure) ? status.lastDecision.pressure.toFixed(3) : "∞"} · ${status.lastDecision.action}`);
+	if (status.lastDecision)
+		lines.push(
+			`Pressure: ${Number.isFinite(status.lastDecision.pressure) ? status.lastDecision.pressure.toFixed(3) : "∞"} · ${status.lastDecision.action}`,
+		);
 	lines.push(`Next: ${nextAction(status.lastDecision?.action)}`);
 	lines.push("Telemetry:");
 	const providerSources = status.sources.filter((source) => source.provider === status.currentRoute?.provider);
+	if (status.currentRoute?.provider === "openai-codex" && providerSources.length === 0)
+		lines.push("  codex-subscription: unavailable · not configured by active daemon");
 	for (const source of providerSources.slice(0, HUMAN_STATUS_MAX_SOURCES)) {
 		const freshness = !source.ok ? "failed" : source.observedAt !== undefined && now - source.observedAt > 120_000 ? "stale" : "fresh";
 		lines.push(`  ${sanitizedText(source.id)}: ${freshness} · ${source.metrics} metrics`);
 	}
-	if (providerSources.length > HUMAN_STATUS_MAX_SOURCES) lines.push(`  … ${providerSources.length - HUMAN_STATUS_MAX_SOURCES} more telemetry sources omitted`);
+	if (providerSources.length > HUMAN_STATUS_MAX_SOURCES)
+		lines.push(`  … ${providerSources.length - HUMAN_STATUS_MAX_SOURCES} more telemetry sources omitted`);
 	if (status.override) lines.push(`Override: ${routeText(status.override.route)}`);
 	if (status.paused) lines.push("Emergency halt is active");
 	return lines;
 }
 
-async function snapshot(client: JittorPanelClient, sessionId: string): Promise<{ status: RouterStatus; metrics: StoredMetricObservation[] }> {
-	const status = await client.call("router.status", { session_id: sessionId }) as RouterStatus;
+async function snapshot(
+	client: JittorPanelClient,
+	sessionId: string,
+): Promise<{ status: RouterStatus; metrics: StoredMetricObservation[] }> {
+	const status = (await client.call("router.status", { session_id: sessionId })) as RouterStatus;
 	const query = providerBudgetMetricQuery(status);
-	const metrics = query ? await client.call("metrics.query", query) as StoredMetricObservation[] : [];
+	const metrics = query ? ((await client.call("metrics.query", query)) as StoredMetricObservation[]) : [];
 	return { status, metrics };
 }
 
 async function chooseOverride(ctx: ExtensionCommandContext, routes: Route[]): Promise<Route | undefined> {
-	if (routes.length === 0) { ctx.ui.notify("Pi reports no authenticated routes for the current provider.", "warning"); return undefined; }
+	if (routes.length === 0) {
+		ctx.ui.notify("Pi reports no authenticated routes for the current provider.", "warning");
+		return undefined;
+	}
 	const labels = routes.map(routeText);
 	const selected = await ctx.ui.select("Override route", labels);
 	const index = selected ? labels.indexOf(selected) : -1;
@@ -265,20 +381,34 @@ export async function showJittorPanel(ctx: ExtensionCommandContext, client: Jitt
 			},
 		}));
 		if (!action || action === "close") return;
-		if (action === "refresh") { await client.call("telemetry.poll", {}); continue; }
+		if (action === "refresh") {
+			await client.call("telemetry.poll", {});
+			continue;
+		}
 		if (action === "pause" || action === "resume") {
-			if (await ctx.ui.confirm(action === "pause" ? "Emergency-halt provider requests?" : "Release emergency halt?", "This changes provider-request enforcement. Use /jittor off to disable blocking entirely.")) {
+			if (
+				await ctx.ui.confirm(
+					action === "pause" ? "Emergency-halt provider requests?" : "Release emergency halt?",
+					"This changes provider-request enforcement. Use /jittor off to disable blocking entirely.",
+				)
+			) {
 				await client.call(action === "pause" ? "router.pause" : "router.resume", { session_id, ...sessionSecretField(session_id) });
 			}
 			continue;
 		}
 		if (action === "clear-override") {
-			if (await ctx.ui.confirm("Clear route override?", "Policy-controlled routing will resume.")) await client.call("router.clear_override", { session_id, ...sessionSecretField(session_id) });
+			if (await ctx.ui.confirm("Clear route override?", "Policy-controlled routing will resume."))
+				await client.call("router.clear_override", { session_id, ...sessionSecretField(session_id) });
 			continue;
 		}
 		const route = await chooseOverride(ctx, current.status.availableRoutes);
-		if (route && await ctx.ui.confirm("Apply route override?", `${routeText(route)} for one hour`)) {
-			await client.call("router.override", { route, expiresAt: Date.now() + 60 * 60 * 1_000, session_id, ...sessionSecretField(session_id) });
+		if (route && (await ctx.ui.confirm("Apply route override?", `${routeText(route)} for one hour`))) {
+			await client.call("router.override", {
+				route,
+				expiresAt: Date.now() + 60 * 60 * 1_000,
+				session_id,
+				...sessionSecretField(session_id),
+			});
 		}
 	}
 }