@eiei114/pi-sub-bar 1.5.1

package/src/paths.ts

@@ -0,0 +1,21 @@
+/**
+ * Shared path helpers for sub-bar settings storage.
+ */
+
+import { getAgentDir } from "@mariozechner/pi-coding-agent";
+import { dirname, join } from "node:path";
+import { fileURLToPath } from "node:url";
+
+const SETTINGS_FILE_NAME = "pi-sub-bar-settings.json";
+
+export function getExtensionDir(): string {
+	return join(dirname(fileURLToPath(import.meta.url)), "..");
+}
+
+export function getSettingsPath(): string {
+	return join(getAgentDir(), SETTINGS_FILE_NAME);
+}
+
+export function getLegacySettingsPath(): 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,199 @@
+/**
+ * Provider metadata shared across the extension.
+ */
+
+import type { RateWindow, UsageSnapshot, ProviderName, ModelInfo } from "../types.js";
+import type { Settings } from "../settings-types.js";
+import { getModelMultiplier, normalizeTokens } from "../utils.js";
+import { PROVIDER_METADATA as BASE_METADATA, type ProviderMetadata as BaseProviderMetadata } from "@eiei114/pi-sub-shared";
+
+export { PROVIDERS, PROVIDER_DISPLAY_NAMES } from "@eiei114/pi-sub-shared";
+export type { ProviderStatusConfig, ProviderDetectionConfig } from "@eiei114/pi-sub-shared";
+
+export interface UsageExtra {
+	label: string;
+}
+
+export interface ProviderMetadata extends BaseProviderMetadata {
+	isWindowVisible?: (usage: UsageSnapshot, window: RateWindow, settings?: Settings, model?: ModelInfo) => boolean;
+	getExtras?: (usage: UsageSnapshot, settings?: Settings, modelId?: string) => UsageExtra[];
+}
+
+const anthropicWindowVisible: ProviderMetadata["isWindowVisible"] = (_usage, window, settings, _model) => {
+	if (!settings) return true;
+	const ps = settings.providers.anthropic;
+	if (window.label === "5h") return ps.windows.show5h;
+	if (window.label === "Week") return ps.windows.show7d;
+	if (window.label.startsWith("Extra [")) return ps.windows.showExtra;
+	return true;
+};
+
+const copilotWindowVisible: ProviderMetadata["isWindowVisible"] = (_usage, window, settings, _model) => {
+	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, _model) => {
+	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, model) => {
+	if (!settings) return true;
+	const ps = settings.providers.antigravity;
+	const label = window.label.trim();
+	const normalized = label.toLowerCase().replace(/\s+/g, "_");
+	if (normalized === "tab_flash_lite_preview") return false;
+
+	const labelTokens = normalizeTokens(label);
+
+	const modelProvider = model?.provider?.toLowerCase() ?? "";
+	const modelId = model?.id;
+	const providerMatches = modelProvider.includes("antigravity");
+	if (ps.showCurrentModel && providerMatches && modelId) {
+		const modelTokens = normalizeTokens(modelId);
+		const match = modelTokens.length > 0 && modelTokens.every((token) => labelTokens.includes(token));
+		if (match) return true;
+	}
+
+	if (ps.showScopedModels) {
+		const scopedPatterns = model?.scopedModelPatterns ?? [];
+		const matchesScoped = scopedPatterns.some((pattern) => {
+			if (!pattern) return false;
+			const [rawPattern] = pattern.split(":");
+			const trimmed = rawPattern?.trim();
+			if (!trimmed) return false;
+			const hasProvider = trimmed.includes("/");
+			if (!hasProvider) return false;
+			const providerPart = trimmed.slice(0, trimmed.indexOf("/")).trim().toLowerCase();
+			if (!providerPart.includes("antigravity")) return false;
+			const base = trimmed.slice(trimmed.lastIndexOf("/") + 1);
+			const tokens = normalizeTokens(base);
+			return tokens.length > 0 && tokens.every((token) => labelTokens.includes(token));
+		});
+		if (matchesScoped) return true;
+	}
+
+	const visibility = ps.modelVisibility?.[label];
+	return visibility === true;
+};
+
+const codexWindowVisible: ProviderMetadata["isWindowVisible"] = (_usage, window, settings, model) => {
+	if (!settings) return true;
+	const ps = settings.providers.codex;
+	const isSparkModel = isCodexSparkModel(model);
+	const isSparkWindow = isCodexSparkWindow(window);
+	if (isSparkWindow) {
+		if (!isSparkModel) return false;
+		return shouldShowCodexWindowBySetting(ps, window);
+	}
+	if (isSparkModel) {
+		return false;
+	}
+	return shouldShowCodexWindowBySetting(ps, window);
+};
+
+const isCodexSparkModel = (model?: ModelInfo): boolean => {
+	const tokens = normalizeTokens(model?.id ?? "");
+	return tokens.includes("codex") && tokens.includes("spark");
+};
+
+const isCodexSparkWindow = (window: RateWindow): boolean => {
+	const tokens = normalizeTokens(window.label ?? "");
+	return tokens.includes("codex") && tokens.includes("spark");
+};
+
+const shouldShowCodexWindowBySetting = (
+	ps: Settings["providers"]["codex"],
+	window: RateWindow
+): boolean => {
+	if (window.label === "") return true;
+	if (/\b\d+h$/.test(window.label.trim())) {
+		return ps.windows.showPrimary;
+	}
+	if (window.label === "Day" || window.label === "Week" || /\b(day|week)\b/.test(window.label.toLowerCase())) {
+		return ps.windows.showSecondary;
+	}
+	return true;
+};
+
+const kiroWindowVisible: ProviderMetadata["isWindowVisible"] = (_usage, window, settings, _model) => {
+	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, _model) => {
+	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,359 @@
+/**
+ * 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 Week Window",
+				currentValue: anthroSettings.windows.show7d ? "on" : "off",
+				values: ["on", "off"],
+				description: "Show the weekly 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: "showCurrentModel",
+				label: "Always Show Current Model",
+				currentValue: antigravitySettings.showCurrentModel ? "on" : "off",
+				values: ["on", "off"],
+				description: "Show the active Antigravity model even if hidden.",
+			},
+			{
+				id: "showScopedModels",
+				label: "Show Scoped Models",
+				currentValue: antigravitySettings.showScopedModels ? "on" : "off",
+				values: ["on", "off"],
+				description: "Show Antigravity models that are in the scoped model rotation.",
+			},
+		);
+
+		const modelVisibility = antigravitySettings.modelVisibility ?? {};
+		const modelOrder = antigravitySettings.modelOrder?.length
+			? antigravitySettings.modelOrder
+			: Object.keys(modelVisibility).sort((a, b) => a.localeCompare(b));
+		const seenModels = new Set<string>();
+
+		for (const model of modelOrder) {
+			if (!model || seenModels.has(model)) continue;
+			seenModels.add(model);
+			const normalized = model.toLowerCase().replace(/\s+/g, "_");
+			if (normalized === "tab_flash_lite_preview") continue;
+			const visible = modelVisibility[model] !== false;
+			items.push({
+				id: `model:${model}`,
+				label: model,
+				currentValue: visible ? "on" : "off",
+				values: ["on", "off"],
+				description: "Toggle this model 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 "showModels":
+				antigravitySettings.windows.showModels = value === "on";
+				break;
+			case "showCurrentModel":
+				antigravitySettings.showCurrentModel = value === "on";
+				break;
+			case "showScopedModels":
+				antigravitySettings.showScopedModels = value === "on";
+				break;
+			default:
+				if (id.startsWith("model:")) {
+					const model = id.slice("model:".length);
+					if (model) {
+						if (!antigravitySettings.modelVisibility) {
+							antigravitySettings.modelVisibility = {};
+						}
+						antigravitySettings.modelVisibility[model] = value === "on";
+						if (!antigravitySettings.modelOrder) {
+							antigravitySettings.modelOrder = [];
+						}
+						if (!antigravitySettings.modelOrder.includes(model)) {
+							antigravitySettings.modelOrder.push(model);
+						}
+					}
+				}
+				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,23 @@
+/**
+ * Provider-specific window visibility rules.
+ */
+
+import type { RateWindow, UsageSnapshot, ModelInfo } 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,
+	model?: ModelInfo
+): boolean {
+	const handler = PROVIDER_METADATA[usage.provider]?.isWindowVisible;
+	if (handler) {
+		return handler(usage, window, settings, model);
+	}
+	return true;
+}