@marckrenn/pi-sub-bar 1.0.0

package/src/paths.ts

@@ -0,0 +1,14 @@
+/**
+ * Shared path helpers for sub-bar settings storage.
+ */
+
+import { dirname, join } from "node:path";
+import { fileURLToPath } from "node:url";
+
+export function getExtensionDir(): string {
+	return join(dirname(fileURLToPath(import.meta.url)), "..");
+}
+
+export function getSettingsPath(): string {
+	return join(getExtensionDir(), "settings.json");
+}

package/src/providers/extras.ts

@@ -0,0 +1,21 @@
+/**
+ * Provider-specific extra usage lines (non-window info).
+ */
+
+import type { UsageSnapshot } from "../types.js";
+import type { Settings } from "../settings-types.js";
+import { PROVIDER_METADATA, type UsageExtra } from "./metadata.js";
+
+export type { UsageExtra } from "./metadata.js";
+
+export function getUsageExtras(
+	usage: UsageSnapshot,
+	settings?: Settings,
+	modelId?: string
+): UsageExtra[] {
+	const handler = PROVIDER_METADATA[usage.provider]?.getExtras;
+	if (handler) {
+		return handler(usage, settings, modelId);
+	}
+	return [];
+}

package/src/providers/metadata.ts

@@ -0,0 +1,137 @@
+/**
+ * Provider metadata shared across the extension.
+ */
+
+import type { RateWindow, UsageSnapshot, ProviderName } from "../types.js";
+import type { Settings } from "../settings-types.js";
+import { getModelMultiplier } from "../utils.js";
+import { PROVIDER_METADATA as BASE_METADATA, type ProviderMetadata as BaseProviderMetadata } from "@marckrenn/pi-sub-shared";
+
+export { PROVIDERS, PROVIDER_DISPLAY_NAMES } from "@marckrenn/pi-sub-shared";
+export type { ProviderStatusConfig, ProviderDetectionConfig } from "@marckrenn/pi-sub-shared";
+
+export interface UsageExtra {
+	label: string;
+}
+
+export interface ProviderMetadata extends BaseProviderMetadata {
+	isWindowVisible?: (usage: UsageSnapshot, window: RateWindow, settings?: Settings) => boolean;
+	getExtras?: (usage: UsageSnapshot, settings?: Settings, modelId?: string) => UsageExtra[];
+}
+
+const anthropicWindowVisible: ProviderMetadata["isWindowVisible"] = (_usage, window, settings) => {
+	if (!settings) return true;
+	const ps = settings.providers.anthropic;
+	if (window.label === "5h") return ps.windows.show5h;
+	if (window.label === "7d") return ps.windows.show7d;
+	if (window.label.startsWith("Extra [")) return ps.windows.showExtra;
+	return true;
+};
+
+const copilotWindowVisible: ProviderMetadata["isWindowVisible"] = (_usage, window, settings) => {
+	if (!settings) return true;
+	const ps = settings.providers.copilot;
+	if (window.label === "Month") return ps.windows.showMonth;
+	return true;
+};
+
+const geminiWindowVisible: ProviderMetadata["isWindowVisible"] = (_usage, window, settings) => {
+	if (!settings) return true;
+	const ps = settings.providers.gemini;
+	if (window.label === "Pro") return ps.windows.showPro;
+	if (window.label === "Flash") return ps.windows.showFlash;
+	return true;
+};
+
+const antigravityWindowVisible: ProviderMetadata["isWindowVisible"] = (_usage, window, settings) => {
+	if (!settings) return true;
+	const ps = settings.providers.antigravity;
+	if (window.label === "Claude") return ps.windows.showClaude;
+	if (window.label === "Pro") return ps.windows.showPro;
+	if (window.label === "Flash") return ps.windows.showFlash;
+	return true;
+};
+
+const codexWindowVisible: ProviderMetadata["isWindowVisible"] = (_usage, window, settings) => {
+	if (!settings) return true;
+	const ps = settings.providers.codex;
+	if (window.label.match(/^\d+h$/)) return ps.windows.showPrimary;
+	if (window.label === "Day" || window.label === "Week") return ps.windows.showSecondary;
+	return true;
+};
+
+const kiroWindowVisible: ProviderMetadata["isWindowVisible"] = (_usage, window, settings) => {
+	if (!settings) return true;
+	const ps = settings.providers.kiro;
+	if (window.label === "Credits") return ps.windows.showCredits;
+	return true;
+};
+
+const zaiWindowVisible: ProviderMetadata["isWindowVisible"] = (_usage, window, settings) => {
+	if (!settings) return true;
+	const ps = settings.providers.zai;
+	if (window.label === "Tokens") return ps.windows.showTokens;
+	if (window.label === "Monthly") return ps.windows.showMonthly;
+	return true;
+};
+
+const anthropicExtras: ProviderMetadata["getExtras"] = (usage, settings) => {
+	const extras: UsageExtra[] = [];
+	const showExtraWindow = settings?.providers.anthropic.windows.showExtra ?? true;
+	if (showExtraWindow && usage.extraUsageEnabled === false) {
+		extras.push({ label: "Extra [off]" });
+	}
+	return extras;
+};
+
+const copilotExtras: ProviderMetadata["getExtras"] = (usage, settings, modelId) => {
+	const extras: UsageExtra[] = [];
+	const showMultiplier = settings?.providers.copilot.showMultiplier ?? true;
+	const showRequestsLeft = settings?.providers.copilot.showRequestsLeft ?? true;
+	if (!showMultiplier) return extras;
+
+	const multiplier = getModelMultiplier(modelId);
+	const remaining = usage.requestsRemaining;
+	if (multiplier !== undefined) {
+		let multiplierStr = `Model multiplier: ${multiplier}x`;
+		if (showRequestsLeft && remaining !== undefined) {
+			const leftCount = Math.floor(remaining / Math.max(multiplier, 0.0001));
+			multiplierStr += ` (${leftCount} req. left)`;
+		}
+		extras.push({ label: multiplierStr });
+	}
+	return extras;
+};
+
+export const PROVIDER_METADATA: Record<ProviderName, ProviderMetadata> = {
+	anthropic: {
+		...BASE_METADATA.anthropic,
+		isWindowVisible: anthropicWindowVisible,
+		getExtras: anthropicExtras,
+	},
+	copilot: {
+		...BASE_METADATA.copilot,
+		isWindowVisible: copilotWindowVisible,
+		getExtras: copilotExtras,
+	},
+	gemini: {
+		...BASE_METADATA.gemini,
+		isWindowVisible: geminiWindowVisible,
+	},
+	antigravity: {
+		...BASE_METADATA.antigravity,
+		isWindowVisible: antigravityWindowVisible,
+	},
+	codex: {
+		...BASE_METADATA.codex,
+		isWindowVisible: codexWindowVisible,
+	},
+	kiro: {
+		...BASE_METADATA.kiro,
+		isWindowVisible: kiroWindowVisible,
+	},
+	zai: {
+		...BASE_METADATA.zai,
+		isWindowVisible: zaiWindowVisible,
+	},
+};

package/src/providers/settings.ts

@@ -0,0 +1,328 @@
+/**
+ * Provider-specific settings helpers.
+ */
+
+import type { SettingItem } from "@mariozechner/pi-tui";
+import type { ProviderName } from "../types.js";
+import type {
+	Settings,
+	BaseProviderSettings,
+	AnthropicProviderSettings,
+	CopilotProviderSettings,
+	GeminiProviderSettings,
+	AntigravityProviderSettings,
+	CodexProviderSettings,
+	KiroProviderSettings,
+	ZaiProviderSettings,
+} from "../settings-types.js";
+
+function buildBaseProviderItems(ps: BaseProviderSettings): SettingItem[] {
+	return [
+		{
+			id: "showStatus",
+			label: "Show Status Indicator",
+			currentValue: ps.showStatus ? "on" : "off",
+			values: ["on", "off"],
+			description: "Show status indicator for this provider.",
+		},
+	];
+}
+
+function applyBaseProviderSetting(ps: BaseProviderSettings, id: string, value: string): boolean {
+	switch (id) {
+		case "showStatus":
+			ps.showStatus = value === "on";
+			return true;
+		default:
+			return false;
+	}
+}
+
+/**
+ * Build settings items for a specific provider.
+ */
+export function buildProviderSettingsItems(settings: Settings, provider: ProviderName): SettingItem[] {
+	const ps = settings.providers[provider];
+	const items: SettingItem[] = [...buildBaseProviderItems(ps)];
+
+	if (provider === "anthropic") {
+		const anthroSettings = ps as AnthropicProviderSettings;
+		items.push(
+			{
+				id: "show5h",
+				label: "Show 5h Window",
+				currentValue: anthroSettings.windows.show5h ? "on" : "off",
+				values: ["on", "off"],
+				description: "Show the 5-hour usage window.",
+			},
+			{
+				id: "show7d",
+				label: "Show 7d Window",
+				currentValue: anthroSettings.windows.show7d ? "on" : "off",
+				values: ["on", "off"],
+				description: "Show the 7-day usage window.",
+			},
+			{
+				id: "showExtra",
+				label: "Show Extra Window",
+				currentValue: anthroSettings.windows.showExtra ? "on" : "off",
+				values: ["on", "off"],
+				description: "Show the extra usage window.",
+			},
+		);
+	}
+
+	if (provider === "copilot") {
+		const copilotSettings = ps as CopilotProviderSettings;
+		items.push(
+			{
+				id: "showMultiplier",
+				label: "Show Model Multiplier",
+				currentValue: copilotSettings.showMultiplier ? "on" : "off",
+				values: ["on", "off"],
+				description: "Show request cost multiplier for the current model.",
+			},
+			{
+				id: "showRequestsLeft",
+				label: "Show Requests Remaining",
+				currentValue: copilotSettings.showRequestsLeft ? "on" : "off",
+				values: ["on", "off"],
+				description: "Estimate requests remaining based on the multiplier.",
+			},
+			{
+				id: "quotaDisplay",
+				label: "Show Quota in",
+				currentValue: copilotSettings.quotaDisplay,
+				values: ["percentage", "requests"],
+				description: "Display Copilot usage as percentage or requests.",
+			},
+			{
+				id: "showMonth",
+				label: "Show Month Window",
+				currentValue: copilotSettings.windows.showMonth ? "on" : "off",
+				values: ["on", "off"],
+				description: "Show the monthly usage window.",
+			},
+		);
+	}
+
+	if (provider === "gemini") {
+		const geminiSettings = ps as GeminiProviderSettings;
+		items.push(
+			{
+				id: "showPro",
+				label: "Show Pro Window",
+				currentValue: geminiSettings.windows.showPro ? "on" : "off",
+				values: ["on", "off"],
+				description: "Show the Pro quota window.",
+			},
+			{
+				id: "showFlash",
+				label: "Show Flash Window",
+				currentValue: geminiSettings.windows.showFlash ? "on" : "off",
+				values: ["on", "off"],
+				description: "Show the Flash quota window.",
+			},
+		);
+	}
+
+	if (provider === "antigravity") {
+		const antigravitySettings = ps as AntigravityProviderSettings;
+		items.push(
+			{
+				id: "showClaude",
+				label: "Show Claude Window",
+				currentValue: antigravitySettings.windows.showClaude ? "on" : "off",
+				values: ["on", "off"],
+				description: "Show the Claude quota window.",
+			},
+			{
+				id: "showPro",
+				label: "Show Pro Window",
+				currentValue: antigravitySettings.windows.showPro ? "on" : "off",
+				values: ["on", "off"],
+				description: "Show the Gemini Pro quota window.",
+			},
+			{
+				id: "showFlash",
+				label: "Show Flash Window",
+				currentValue: antigravitySettings.windows.showFlash ? "on" : "off",
+				values: ["on", "off"],
+				description: "Show the Gemini Flash quota window.",
+			},
+		);
+	}
+
+	if (provider === "codex") {
+		const codexSettings = ps as CodexProviderSettings;
+		items.push(
+			{
+				id: "invertUsage",
+				label: "Invert Usage",
+				currentValue: codexSettings.invertUsage ? "on" : "off",
+				values: ["on", "off"],
+				description: "Show remaining-style usage for Codex.",
+			},
+			{
+				id: "showPrimary",
+				label: "Show Primary Window",
+				currentValue: codexSettings.windows.showPrimary ? "on" : "off",
+				values: ["on", "off"],
+				description: "Show the primary usage window.",
+			},
+			{
+				id: "showSecondary",
+				label: "Show Secondary Window",
+				currentValue: codexSettings.windows.showSecondary ? "on" : "off",
+				values: ["on", "off"],
+				description: "Show secondary windows (day/week).",
+			},
+		);
+	}
+
+	if (provider === "kiro") {
+		const kiroSettings = ps as KiroProviderSettings;
+		items.push({
+			id: "showCredits",
+			label: "Show Credits Window",
+			currentValue: kiroSettings.windows.showCredits ? "on" : "off",
+			values: ["on", "off"],
+			description: "Show the credits usage window.",
+		});
+	}
+
+	if (provider === "zai") {
+		const zaiSettings = ps as ZaiProviderSettings;
+		items.push(
+			{
+				id: "showTokens",
+				label: "Show Tokens Window",
+				currentValue: zaiSettings.windows.showTokens ? "on" : "off",
+				values: ["on", "off"],
+				description: "Show the tokens usage window.",
+			},
+			{
+				id: "showMonthly",
+				label: "Show Monthly Window",
+				currentValue: zaiSettings.windows.showMonthly ? "on" : "off",
+				values: ["on", "off"],
+				description: "Show the monthly usage window.",
+			},
+		);
+	}
+
+	return items;
+}
+
+/**
+ * Apply a provider settings change in-place.
+ */
+export function applyProviderSettingsChange(
+	settings: Settings,
+	provider: ProviderName,
+	id: string,
+	value: string
+): Settings {
+	const ps = settings.providers[provider];
+	if (applyBaseProviderSetting(ps, id, value)) {
+		return settings;
+	}
+
+	if (provider === "anthropic") {
+		const anthroSettings = ps as AnthropicProviderSettings;
+		switch (id) {
+			case "show5h":
+				anthroSettings.windows.show5h = value === "on";
+				break;
+			case "show7d":
+				anthroSettings.windows.show7d = value === "on";
+				break;
+			case "showExtra":
+				anthroSettings.windows.showExtra = value === "on";
+				break;
+		}
+	}
+
+	if (provider === "copilot") {
+		const copilotSettings = ps as CopilotProviderSettings;
+		switch (id) {
+			case "showMultiplier":
+				copilotSettings.showMultiplier = value === "on";
+				break;
+			case "showRequestsLeft":
+				copilotSettings.showRequestsLeft = value === "on";
+				break;
+			case "quotaDisplay":
+				copilotSettings.quotaDisplay = value as "percentage" | "requests";
+				break;
+			case "showMonth":
+				copilotSettings.windows.showMonth = value === "on";
+				break;
+		}
+	}
+
+	if (provider === "gemini") {
+		const geminiSettings = ps as GeminiProviderSettings;
+		switch (id) {
+			case "showPro":
+				geminiSettings.windows.showPro = value === "on";
+				break;
+			case "showFlash":
+				geminiSettings.windows.showFlash = value === "on";
+				break;
+		}
+	}
+
+	if (provider === "antigravity") {
+		const antigravitySettings = ps as AntigravityProviderSettings;
+		switch (id) {
+			case "showClaude":
+				antigravitySettings.windows.showClaude = value === "on";
+				break;
+			case "showPro":
+				antigravitySettings.windows.showPro = value === "on";
+				break;
+			case "showFlash":
+				antigravitySettings.windows.showFlash = value === "on";
+				break;
+		}
+	}
+
+	if (provider === "codex") {
+		const codexSettings = ps as CodexProviderSettings;
+		switch (id) {
+			case "invertUsage":
+				codexSettings.invertUsage = value === "on";
+				break;
+			case "showPrimary":
+				codexSettings.windows.showPrimary = value === "on";
+				break;
+			case "showSecondary":
+				codexSettings.windows.showSecondary = value === "on";
+				break;
+		}
+	}
+
+	if (provider === "kiro") {
+		const kiroSettings = ps as KiroProviderSettings;
+		switch (id) {
+			case "showCredits":
+				kiroSettings.windows.showCredits = value === "on";
+				break;
+		}
+	}
+
+	if (provider === "zai") {
+		const zaiSettings = ps as ZaiProviderSettings;
+		switch (id) {
+			case "showTokens":
+				zaiSettings.windows.showTokens = value === "on";
+				break;
+			case "showMonthly":
+				zaiSettings.windows.showMonthly = value === "on";
+				break;
+		}
+	}
+
+	return settings;
+}

package/src/providers/windows.ts

@@ -0,0 +1,18 @@
+/**
+ * Provider-specific window visibility rules.
+ */
+
+import type { RateWindow, UsageSnapshot } from "../types.js";
+import type { Settings } from "../settings-types.js";
+import { PROVIDER_METADATA } from "./metadata.js";
+
+/**
+ * Check if a window should be shown based on settings.
+ */
+export function shouldShowWindow(usage: UsageSnapshot, window: RateWindow, settings?: Settings): boolean {
+	const handler = PROVIDER_METADATA[usage.provider]?.isWindowVisible;
+	if (handler) {
+		return handler(usage, window, settings);
+	}
+	return true;
+}