@narumitw/pi-usage 0.51.0 → 0.52.1

package/src/providers/opencode-zen.ts

@@ -0,0 +1,83 @@
+import { sanitizeDisplayText } from "../core.js";
+import type { OpenCodeZenPayload, UsageBucket, UsageReport } from "../types.js";
+
+const ZEN_WINDOWS = [
+	{ key: "rolling", label: "Rolling" },
+	{ key: "weekly", label: "Weekly" },
+	{ key: "monthly", label: "Monthly" },
+] as const;
+
+export function normalizeOpenCodeZenPayload(
+	payload: OpenCodeZenPayload,
+	capturedAt: number,
+): UsageReport {
+	const usage = asObject(payload.usage);
+	if (!usage) throw new Error("OpenCode Zen usage response was not an object.");
+
+	const buckets: UsageBucket[] = [];
+	const notes: string[] = [];
+	for (const window of ZEN_WINDOWS) {
+		const raw = asObject(usage[window.key]);
+		if (!raw) continue;
+		const status = asString(raw.status);
+		if (status !== "ok" && status !== "rate-limited") {
+			notes.push(`${window.label} window unavailable (${status ?? "unknown status"}).`);
+			continue;
+		}
+		const used = asNonnegativeNumber(raw.percent);
+		if (used === undefined) continue;
+		const resetsAt = asEpochSeconds(raw.resetsAt);
+		buckets.push({
+			id: window.key,
+			label: `${window.label} window`,
+			used,
+			remaining: 100 - clampPercent(used),
+			limit: 100,
+			unit: "percent",
+			...(resetsAt !== undefined ? { resetsAt } : {}),
+		});
+	}
+	if (buckets.length === 0) {
+		throw new Error("OpenCode Zen usage endpoint returned no displayable usage data.");
+	}
+
+	return {
+		providerId: "opencode-go",
+		providerName: "OpenCode Go",
+		capturedAt,
+		source: "opencode-zen-usage",
+		semantics: {
+			kind: "consumer-subscription",
+			label: "OpenCode Zen plan usage",
+		},
+		buckets,
+		metrics: [],
+		...(notes.length > 0 ? { notes } : {}),
+	};
+}
+
+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 asString(value: unknown): string | undefined {
+	if (typeof value !== "string") return undefined;
+	return sanitizeDisplayText(value, 80) || undefined;
+}
+
+function asNonnegativeNumber(value: unknown): number | undefined {
+	if (typeof value !== "number" || !Number.isFinite(value) || value < 0) return undefined;
+	return value;
+}
+
+function asEpochSeconds(value: unknown): number | undefined {
+	if (typeof value !== "string" || !value.trim()) return undefined;
+	const parsed = Date.parse(value);
+	if (Number.isNaN(parsed)) return undefined;
+	return Math.floor(parsed / 1000);
+}
+
+function clampPercent(value: number): number {
+	return Math.min(100, Math.max(0, value));
+}

package/src/query.ts

@@ -3,10 +3,12 @@ import { type ExtensionContext, readStoredCredential } from "@earendil-works/pi-
 import { errorMessage, fingerprintResolvedAuth, redactUsageError } from "./core.js";
 import { normalizeCodexBackendPayload } from "./providers/codex.js";
 import { normalizeGitHubCopilotUsagePayload } from "./providers/github-copilot.js";
+import { normalizeOpenCodeZenPayload } from "./providers/opencode-zen.js";
 import { normalizeOpenRouterKeyPayload } from "./providers/openrouter.js";
 import type {
 	CodexBackendPayload,
 	GitHubCopilotUsagePayload,
+	OpenCodeZenPayload,
 	OpenRouterKeyPayload,
 	PiModel,
 	ResolvedUsageAuth,
@@ -74,6 +76,21 @@ export const SUPPORTED_ADAPTERS: readonly UsageProviderAdapter[] = [
 			return normalizeOpenRouterKeyPayload(payload as OpenRouterKeyPayload, Date.now());
 		},
 	},
+	{
+		id: "opencode-go",
+		displayName: "OpenCode Go",
+		semantics: { kind: "consumer-subscription", label: "OpenCode Zen plan usage" },
+		async query(auth, signal, timeoutMs) {
+			const payload = await fetchProviderJson(
+				opencodeUsageUrl(auth.model.baseUrl),
+				auth,
+				signal,
+				timeoutMs,
+				"OpenCode Zen usage endpoint",
+			);
+			return normalizeOpenCodeZenPayload(payload as OpenCodeZenPayload, Date.now());
+		},
+	},
 ];
 
 export function adapterForProvider(
@@ -393,6 +410,7 @@ function hasOfficialUrlOrigin(value: string, providerId: string): boolean {
 		const url = new URL(value);
 		if (providerId === "openai-codex") return url.origin === "https://chatgpt.com";
 		if (providerId === "openrouter") return url.origin === "https://openrouter.ai";
+		if (providerId === "opencode-go") return url.origin === "https://opencode.ai";
 		if (providerId === "github-copilot") {
 			return (
 				url.protocol === "https:" && /^api\.[a-z0-9-]+\.githubcopilot\.com$/u.test(url.hostname)
@@ -418,6 +436,12 @@ function hasHeader(headers: Record<string, string>, name: string): boolean {
 	return Object.keys(headers).some((key) => key.toLowerCase() === name.toLowerCase());
 }
 
+function opencodeUsageUrl(baseUrl: string | undefined): string {
+	const base = baseUrl?.trim().replace(/\/+$/u, "");
+	if (!base) throw new Error("OpenCode Go model base URL is unavailable.");
+	return `${base}/usage`;
+}
+
 function isAbortError(error: unknown): boolean {
 	return error instanceof Error && error.name === "AbortError";
 }

package/src/types.ts

@@ -93,6 +93,10 @@ export type OpenRouterKeyPayload = {
 	data?: unknown;
 };
 
+export type OpenCodeZenPayload = {
+	usage?: unknown;
+};
+
 export type CodexBackendPayload = {
 	plan_type?: unknown;
 	rate_limit?: unknown;