@narumitw/pi-usage 0.53.0 → 0.57.0

package/package.json

@@ -1,7 +1,7 @@
 {
 	"name": "@narumitw/pi-usage",
-	"version": "0.53.0",
-	"description": "Pi extension that shows current-account usage for Codex, GitHub Copilot, OpenRouter, OpenCode Zen, and Z.AI.",
+	"version": "0.57.0",
+	"description": "Pi extension that shows current-account usage and DeepSeek API balance for supported providers.",
 	"type": "module",
 	"license": "MIT",
 	"private": false,
@@ -11,13 +11,19 @@
 		"pi",
 		"usage",
 		"quota",
+		"balance",
+		"deepseek",
 		"codex",
+		"kimi",
+		"kimi-coding",
 		"copilot",
 		"openrouter",
 		"opencode",
 		"zen",
 		"zai",
-		"glm"
+		"glm",
+		"xai",
+		"grok"
 	],
 	"files": [
 		"src",
@@ -30,9 +36,6 @@
 			"./dist/index.ts"
 		]
 	},
-	"piExtension": {
-		"lifecycle": "stable"
-	},
 	"scripts": {
 		"build": "node scripts/build-runtime.mjs",
 		"check": "npm run build && biome check --vcs-use-ignore-file=false src test scripts package.json tsconfig.json README.md && npm run typecheck",
@@ -42,13 +45,15 @@
 	},
 	"peerDependencies": {
 		"@earendil-works/pi-ai": "*",
-		"@earendil-works/pi-coding-agent": "*"
+		"@earendil-works/pi-coding-agent": "*",
+		"@earendil-works/pi-tui": "*"
 	},
 	"devDependencies": {
-		"@biomejs/biome": "2.5.10",
-		"@earendil-works/pi-ai": "0.84.3",
-		"@earendil-works/pi-coding-agent": "0.84.3",
-		"@types/node": "26.2.0",
+		"@biomejs/biome": "2.5.11",
+		"@earendil-works/pi-ai": "0.84.4",
+		"@earendil-works/pi-coding-agent": "0.84.4",
+		"@earendil-works/pi-tui": "0.84.4",
+		"@types/node": "26.4.0",
 		"esbuild": "0.28.2",
 		"typescript": "7.0.2"
 	},

package/src/format.ts

@@ -11,14 +11,19 @@ const VALUE_COLUMN = 29;
 
 export function formatUsageReport(report: UsageReport, displayState: UsageDisplayState): string {
 	const stateLabel = displayState === "current" ? "Current" : "Configured";
-	const lines = [`${report.providerName} Usage · ${stateLabel}`];
+	const title =
+		report.providerId === "deepseek" ? "DeepSeek API Balance" : `${report.providerName} Usage`;
+	const lines = [`${title} · ${stateLabel}`];
 	if (report.accountLabel) lines.push(`Account: ${report.accountLabel}`);
 	lines.push(`Semantics: ${report.semantics.label}`, "");
 
 	if (report.providerId === "openai-codex") formatCodexReport(lines, report);
+	else if (report.providerId === "deepseek") formatDeepSeekReport(lines, report);
 	else if (report.providerId === "github-copilot") formatGitHubCopilotReport(lines, report);
 	else if (report.providerId === "openrouter") formatOpenRouterReport(lines, report);
 	else if (report.providerId === "opencode-go") formatOpenCodeZenReport(lines, report);
+	else if (report.providerId === "kimi-coding") formatKimiCodingReport(lines, report);
+	else if (report.providerId === "xai") formatXaiReport(lines, report);
 	else if (report.providerId === "zai" || report.providerId === "zai-coding-cn") {
 		formatZaiReport(lines, report);
 	} else formatGenericReport(lines, report);
@@ -29,8 +34,16 @@ export function formatUsageReport(report: UsageReport, displayState: UsageDispla
 	return lines.join("\n").trimEnd();
 }
 
-export function formatUsageStatusline(report: UsageReport, model?: UsageModel): string | undefined {
-	if (report.providerId === "openai-codex") return formatCodexStatusline(report, model);
+export function formatUsageStatusline(
+	report: UsageReport,
+	model?: UsageModel,
+	now = Date.now(),
+	showCodexResetCountdown = true,
+): string | undefined {
+	if (report.providerId === "openai-codex") {
+		return formatCodexStatusline(report, model, now, showCodexResetCountdown);
+	}
+	if (report.providerId === "deepseek") return formatDeepSeekStatusline(report);
 	if (report.providerId === "github-copilot") return formatGitHubCopilotStatusline(report);
 	if (report.providerId === "openrouter") {
 		const limit = report.buckets.find((bucket) => bucket.id === "key-limit");
@@ -39,6 +52,10 @@ export function formatUsageStatusline(report: UsageReport, model?: UsageModel):
 		if (typeof total?.value === "number") return `openrouter ${formatUsd(total.value)} used`;
 	}
 	if (report.providerId === "opencode-go") return formatOpenCodeZenStatusline(report);
+	if (report.providerId === "kimi-coding") return formatKimiCodingStatusline(report);
+	if (report.providerId === "zai" || report.providerId === "zai-coding-cn") {
+		return formatZaiStatusline(report);
+	}
 	return undefined;
 }
 
@@ -81,6 +98,33 @@ function formatCodexReport(lines: string[], report: UsageReport): void {
 	}
 }
 
+function formatDeepSeekReport(lines: string[], report: UsageReport): void {
+	const availability = report.metrics.find((metric) => metric.id === "api-availability");
+	lines.push(
+		`${"API calls:".padEnd(VALUE_COLUMN)}${availability?.value === "available" ? "Available" : "Unavailable"}`,
+	);
+	for (const currency of ["CNY", "USD"]) {
+		const metrics = report.metrics.filter((metric) => metric.currency === currency);
+		if (metrics.length === 0) continue;
+		lines.push("", `${currency} balance:`);
+		for (const metric of metrics) {
+			lines.push(`${`${metric.label}:`.padEnd(VALUE_COLUMN)}${currency} ${metric.value}`);
+		}
+	}
+}
+
+function formatDeepSeekStatusline(report: UsageReport): string {
+	const availability = report.metrics.find((metric) => metric.id === "api-availability");
+	if (availability?.value !== "available") return "deepseek API unavailable";
+	const totals = ["CNY", "USD"].flatMap((currency) => {
+		const metric = report.metrics.find(
+			(candidate) => candidate.id === `${currency.toLowerCase()}-total`,
+		);
+		return metric ? [`${currency} ${metric.value}`] : [];
+	});
+	return totals.length > 0 ? `deepseek ${totals.join(" · ")}` : "deepseek balance unavailable";
+}
+
 function formatGitHubCopilotReport(lines: string[], report: UsageReport): void {
 	const quota = findGitHubCopilotQuota(report);
 	if (!quota || quota.limit === undefined || quota.remaining === undefined) {
@@ -162,6 +206,109 @@ function formatOpenCodeZenStatusline(report: UsageReport): string | undefined {
 	return parts.length > 1 ? parts.join(" ") : undefined;
 }
 
+function formatKimiCodingReport(lines: string[], report: UsageReport): void {
+	for (const bucket of report.buckets) {
+		const reset = bucket.resetsAt ? ` (resets ${formatReset(bucket.resetsAt)})` : "";
+		if (bucket.used === undefined || bucket.limit === undefined) {
+			lines.push(`${`${bucket.label}:`.padEnd(VALUE_COLUMN)}unavailable${reset}`);
+			continue;
+		}
+		lines.push(
+			`${`${bucket.label}:`.padEnd(VALUE_COLUMN)}${bucket.used} of ${bucket.limit} used · ${percentRemaining(bucket)}% left${reset}`,
+		);
+	}
+	const balance = report.metrics.find((metric) => metric.id === "booster-balance");
+	const total = report.metrics.find((metric) => metric.id === "booster-total");
+	const monthlyUsed = report.metrics.find((metric) => metric.id === "booster-monthly-used");
+	const monthlyLimit = report.metrics.find((metric) => metric.id === "booster-monthly-limit");
+	if (!balance && !monthlyUsed && !monthlyLimit) return;
+	lines.push("", "Extra usage wallet:");
+	if (balance) {
+		const totalSuffix = total ? ` of ${formatCurrencyMetric(total)}` : "";
+		lines.push(`${"Balance:".padEnd(VALUE_COLUMN)}${formatCurrencyMetric(balance)}${totalSuffix}`);
+	}
+	if (monthlyUsed) {
+		lines.push(`${"Used this month:".padEnd(VALUE_COLUMN)}${formatCurrencyMetric(monthlyUsed)}`);
+	}
+	if (monthlyLimit) {
+		lines.push(`${"Monthly limit:".padEnd(VALUE_COLUMN)}${formatCurrencyMetric(monthlyLimit)}`);
+	}
+}
+
+function formatKimiCodingStatusline(report: UsageReport): string | undefined {
+	const fiveHour = report.buckets.find((bucket) => bucket.id === "five-hour");
+	const weekly = report.buckets.find((bucket) => bucket.id === "weekly");
+	const subWindow = fiveHour ?? report.buckets.find((bucket) => bucket.id !== "weekly");
+	const selected = [subWindow, weekly].filter(
+		(bucket, index, buckets): bucket is UsageBucket =>
+			bucket !== undefined && buckets.indexOf(bucket) === index,
+	);
+	const parts = ["kimi"];
+	for (const bucket of selected) {
+		if (!bucket.limit || bucket.remaining === undefined) continue;
+		const fallback = bucket.id === "weekly" ? "weekly" : "5h";
+		parts.push(
+			`${percentRemaining(bucket)}% ${formatWindowLabel(bucket.windowMinutes, fallback, true)}`,
+		);
+	}
+	return parts.length > 1 ? parts.join(" ") : undefined;
+}
+
+function formatZaiStatusline(report: UsageReport): string | undefined {
+	const selected = [
+		report.buckets.find((bucket) => bucket.id === "five-hour"),
+		report.buckets.find((bucket) => bucket.id === "weekly"),
+	];
+	const parts = ["zai"];
+	for (const bucket of selected) {
+		if (!bucket?.limit || bucket.remaining === undefined) continue;
+		const fallback = bucket.id === "weekly" ? "weekly" : "5h";
+		parts.push(
+			`${percentRemaining(bucket)}% ${formatWindowLabel(bucket.windowMinutes, fallback, true)}`,
+		);
+	}
+	return parts.length > 1 ? parts.join(" ") : undefined;
+}
+
+function formatCurrencyMetric(metric: UsageReport["metrics"][number]): string {
+	if (typeof metric.value !== "number") return String(metric.value);
+	if (!metric.currency) return "unavailable";
+	if (metric.currency === "USD") return `$${metric.value.toFixed(2)}`;
+	if (metric.currency === "CNY") return `¥${metric.value.toFixed(2)}`;
+	return `${metric.value.toFixed(2)} ${metric.currency}`;
+}
+
+function formatXaiReport(lines: string[], report: UsageReport): void {
+	const included = report.buckets.find((bucket) => bucket.id === "included-allowance");
+	if (included) {
+		let value = "unavailable";
+		if (included.unit === "percent" && included.used !== undefined) {
+			value = `${included.used}% used`;
+			if (included.remaining !== undefined) value += ` · ${included.remaining}% left`;
+		} else if (included.used !== undefined) {
+			value = `${formatUsd(included.used)} used`;
+			if (included.limit !== undefined) value += ` of ${formatUsd(included.limit)}`;
+		} else if (included.limit !== undefined) {
+			value = `usage unavailable · ${formatUsd(included.limit)} limit`;
+		}
+		const period = included.period ? ` · ${included.period}` : "";
+		const reset = included.resetsAt ? ` (resets ${formatReset(included.resetsAt)})` : "";
+		lines.push(`${"Included allowance:".padEnd(VALUE_COLUMN)}${value}${period}${reset}`);
+	}
+	const onDemand = report.buckets.find((bucket) => bucket.id === "on-demand");
+	if (onDemand) {
+		let value =
+			onDemand.used === undefined ? "usage unavailable" : `${formatUsd(onDemand.used)} used`;
+		if (onDemand.limit !== undefined) value += ` of ${formatUsd(onDemand.limit)} cap`;
+		lines.push(`${"On-demand usage:".padEnd(VALUE_COLUMN)}${value}`);
+	}
+	for (const metric of report.metrics) {
+		lines.push(
+			`${`${metric.label}:`.padEnd(VALUE_COLUMN)}${formatMetricValue(metric.value, metric.unit)}`,
+		);
+	}
+}
+
 function formatZaiReport(lines: string[], report: UsageReport): void {
 	for (const bucket of report.buckets) {
 		const reset = bucket.resetsAt ? ` (resets ${formatReset(bucket.resetsAt)})` : "";
@@ -199,7 +346,12 @@ function formatGenericReport(lines: string[], report: UsageReport): void {
 	}
 }
 
-function formatCodexStatusline(report: UsageReport, model?: UsageModel): string | undefined {
+function formatCodexStatusline(
+	report: UsageReport,
+	model?: UsageModel,
+	now = Date.now(),
+	showResetCountdown = true,
+): string | undefined {
 	const group = selectCodexGroup(report, model);
 	if (!group) return formatCodexCreditsStatus(report);
 	const buckets = report.buckets.filter((bucket) => (bucket.groupId ?? bucket.id) === group);
@@ -209,10 +361,15 @@ function formatCodexStatusline(report: UsageReport, model?: UsageModel): string
 	];
 	for (const bucket of buckets) {
 		if (bucket.remaining === undefined) continue;
+		const percent = `${clampPercent(bucket.remaining).toFixed(0)}%`;
 		const fallback = bucket.id.endsWith(":secondary") ? "weekly" : "5h";
-		parts.push(
-			`${clampPercent(bucket.remaining).toFixed(0)}% ${formatWindowLabel(bucket.windowMinutes, fallback, true)}`,
-		);
+		const window = formatWindowLabel(bucket.windowMinutes, fallback, true);
+		if (!showResetCountdown) {
+			parts.push(`${percent} ${window}`);
+			continue;
+		}
+		const reset = formatResetCountdown(bucket.resetsAt, now);
+		parts.push(`${percent} ${reset ? `↻ ${reset}` : window}`);
 	}
 	return parts.length > 1 ? parts.join(" ") : formatCodexCreditsStatus(report);
 }
@@ -315,6 +472,26 @@ function formatWindowLabel(
 	return `${minutes}m`;
 }
 
+function formatResetCountdown(resetsAt: number | undefined, now: number): string | undefined {
+	if (resetsAt === undefined || !Number.isFinite(resetsAt) || !Number.isFinite(now))
+		return undefined;
+	const totalMinutes = Math.max(0, Math.ceil((resetsAt * 1_000 - now) / 60_000));
+	const days = Math.floor(totalMinutes / 1_440);
+	const hours = Math.floor((totalMinutes % 1_440) / 60);
+	const minutes = totalMinutes % 60;
+	if (days > 0) {
+		return [
+			`${String(days)}d`,
+			hours > 0 ? `${String(hours)}h` : minutes > 0 ? `${String(minutes)}m` : "",
+		]
+			.filter(Boolean)
+			.join("");
+	}
+	if (hours > 0)
+		return [`${String(hours)}h`, minutes > 0 ? `${String(minutes)}m` : ""].filter(Boolean).join("");
+	return `${String(minutes)}m`;
+}
+
 function formatMetricValue(value: number | string, unit: UsageBucket["unit"] | undefined): string {
 	if (unit === "usd" && typeof value === "number") return formatUsd(value);
 	return String(value);

package/src/index.ts

@@ -33,9 +33,12 @@ export {
 } from "./core.js";
 export { formatProviderStates, formatUsageReport, formatUsageStatusline } from "./format.js";
 export { normalizeCodexBackendPayload } from "./providers/codex.js";
+export { normalizeDeepSeekBalancePayload } from "./providers/deepseek.js";
 export { normalizeGitHubCopilotUsagePayload } from "./providers/github-copilot.js";
+export { normalizeKimiCodingUsagePayload } from "./providers/kimi-coding.js";
 export { normalizeOpenCodeZenPayload } from "./providers/opencode-zen.js";
 export { normalizeOpenRouterKeyPayload } from "./providers/openrouter.js";
+export { normalizeXaiBillingPayload } from "./providers/xai.js";
 export { normalizeZaiQuotaPayload } from "./providers/zai.js";
 export {
 	adapterForProvider,
@@ -44,6 +47,8 @@ export {
 	queryProviderUsage,
 	resolveUsageAuth,
 	SUPPORTED_ADAPTERS,
+	usageAdapters,
+	XAI_ADAPTER,
 } from "./query.js";
 export type {
 	UsageSettings,
@@ -58,6 +63,8 @@ export {
 	usageSettingsPath,
 } from "./settings.js";
 export type {
+	DeepSeekBalancePayload,
+	KimiCodingUsagePayload,
 	ProviderUsageState,
 	ResolvedUsageAuth,
 	UsageBucket,
@@ -69,5 +76,7 @@ export type {
 	UsageSemantics,
 	UsageSemanticsKind,
 	UsageUnit,
+	XaiBillingPayload,
+	XaiUserPayload,
 } from "./types.js";
 export { default } from "./usage.js";

package/src/providers/deepseek.ts

@@ -0,0 +1,85 @@
+import type { DeepSeekBalancePayload, UsageMetric, UsageReport } from "../types.js";
+
+const CURRENCIES = ["CNY", "USD"] as const;
+type DeepSeekCurrency = (typeof CURRENCIES)[number];
+
+const BALANCE_FIELDS = [
+	["total", "Total balance", "total_balance"],
+	["granted", "Granted balance", "granted_balance"],
+	["topped-up", "Topped-up balance", "topped_up_balance"],
+] as const;
+
+export function normalizeDeepSeekBalancePayload(
+	payload: DeepSeekBalancePayload,
+	capturedAt: number,
+): UsageReport {
+	if (typeof payload.is_available !== "boolean") {
+		throw new Error("DeepSeek API balance response availability was not a boolean.");
+	}
+	if (!Array.isArray(payload.balance_infos) || payload.balance_infos.length === 0) {
+		throw new Error("DeepSeek API balance response returned no balance information.");
+	}
+
+	const balances = new Map<DeepSeekCurrency, Record<string, unknown>>();
+	for (const raw of payload.balance_infos) {
+		const balance = asObject(raw);
+		if (!balance) throw new Error("DeepSeek API balance row was not an object.");
+		const currency = deepSeekCurrency(balance.currency);
+		if (!currency) throw new Error("DeepSeek API balance row returned an unsupported currency.");
+		if (balances.has(currency)) {
+			throw new Error(`DeepSeek API balance response repeated ${currency}.`);
+		}
+		for (const [, label, field] of BALANCE_FIELDS) {
+			if (!decimalAmount(balance[field])) {
+				throw new Error(`DeepSeek API balance ${label.toLowerCase()} was not a valid amount.`);
+			}
+		}
+		balances.set(currency, balance);
+	}
+
+	const metrics: UsageMetric[] = [
+		{
+			id: "api-availability",
+			label: "API calls",
+			value: payload.is_available ? "available" : "unavailable",
+		},
+	];
+	for (const currency of CURRENCIES) {
+		const balance = balances.get(currency);
+		if (!balance) continue;
+		for (const [id, label, field] of BALANCE_FIELDS) {
+			metrics.push({
+				id: `${currency.toLowerCase()}-${id}`,
+				label,
+				value: balance[field] as string,
+				unit: "currency",
+				currency,
+			});
+		}
+	}
+
+	return {
+		providerId: "deepseek",
+		providerName: "DeepSeek",
+		capturedAt,
+		source: "deepseek-balance",
+		semantics: { kind: "api-key", label: "DeepSeek API balance" },
+		buckets: [],
+		metrics,
+	};
+}
+
+function asObject(value: unknown): Record<string, unknown> | undefined {
+	if (!value || typeof value !== "object" || Array.isArray(value)) return undefined;
+	return value as Record<string, unknown>;
+}
+
+function deepSeekCurrency(value: unknown): DeepSeekCurrency | undefined {
+	return CURRENCIES.find((currency) => currency === value);
+}
+
+function decimalAmount(value: unknown): value is string {
+	return (
+		typeof value === "string" && value.length <= 64 && /^(?:0|[1-9]\d*)(?:\.\d+)?$/u.test(value)
+	);
+}