pi-free 2.0.1 → 2.0.4

package/lib/registry.ts

@@ -1,144 +1,193 @@
-/**
- * Global Provider Registry for pi-free.
- *
- * Decoupled from index.ts so providers can import toggle logic
- * without creating a circular dependency.
- */
-
-import type {
-	ExtensionAPI,
-	ProviderModelConfig,
-} from "@mariozechner/pi-coding-agent";
-import { getFreeOnly, saveConfig } from "../config.ts";
-import { createLogger } from "./logger.ts";
-
-const _logger = createLogger("pi-free");
-
-// =============================================================================
-// Types
-// =============================================================================
-
-interface ProviderEntry {
-	id: string;
-	stored: { free: ProviderModelConfig[]; all: ProviderModelConfig[] };
-	reRegister: (models: ProviderModelConfig[]) => void;
-	hasKey: boolean;
-}
-
-// =============================================================================
-// State
-// =============================================================================
-
-const providerRegistry = new Map<string, ProviderEntry>();
-let globalFreeOnly = getFreeOnly();
-
-// Providers that expose actual per-model pricing via API
-const PRICING_EXPOSED_PROVIDERS = new Set([
-	"openrouter",
-	"opencode",
-	"kilo",
-	"cline",
-]);
-
-// =============================================================================
-// Free-model detection
-// =============================================================================
-
-/**
- * Check if a model is free.
- *
- * For providers with pricing APIs: uses cost (input === 0 && output === 0)
- * For providers without pricing: ONLY uses name-based check (name includes "free")
- */
-export function isFreeModel(
-	model: ProviderModelConfig & { provider?: string },
-): boolean {
-	const provider = model.provider;
-	const hasPricing = provider && PRICING_EXPOSED_PROVIDERS.has(provider);
-
-	// For providers WITH pricing API: cost-based check
-	if (hasPricing) {
-		if ((model.cost?.input ?? 0) === 0 && (model.cost?.output ?? 0) === 0) {
-			return true;
-		}
-	}
-
-	// For providers WITHOUT pricing API: ONLY name-based check
-	if (model.name.toLowerCase().includes("free")) {
-		return true;
-	}
-
-	return false;
-}
-
-// =============================================================================
-// Registration
-// =============================================================================
-
-/** Register a provider with the global free/paid toggle system */
-export function registerWithGlobalToggle(
-	providerId: string,
-	stored: { free: ProviderModelConfig[]; all: ProviderModelConfig[] },
-	reRegister: (models: ProviderModelConfig[]) => void,
-	hasKey: boolean = false,
-): void {
-	providerRegistry.set(providerId, {
-		id: providerId,
-		stored,
-		reRegister,
-		hasKey,
-	});
-	_logger.info(
-		`[pi-free] Registered ${providerId} with global toggle (${stored.free.length} free, ${stored.all.length} total)`,
-	);
-}
-
-/** Get current global free-only state */
-export function getGlobalFreeOnly(): boolean {
-	return globalFreeOnly;
-}
-
-/** Access the raw registry (used by /free-providers command) */
-export function getProviderRegistry(): ReadonlyMap<string, ProviderEntry> {
-	return providerRegistry;
-}
-
-// =============================================================================
-// Global filter application
-// =============================================================================
-
-export function applyGlobalFilter(_pi: ExtensionAPI, freeOnly: boolean): void {
-	globalFreeOnly = freeOnly;
-	saveConfig({ free_only: freeOnly });
-
-	for (const [providerId, entry] of providerRegistry) {
-		try {
-			if (freeOnly) {
-				// Show only free models
-				if (entry.stored.free.length > 0) {
-					entry.reRegister(entry.stored.free);
-					_logger.info(
-						`[pi-free] ${providerId}: filtered to ${entry.stored.free.length} free models`,
-					);
-				} else {
-					_logger.warn(`[pi-free] ${providerId}: no free models available`);
-				}
-			} else {
-				// Show all models (paid + free)
-				const allModels =
-					entry.stored.all.length > 0 ? entry.stored.all : entry.stored.free;
-				if (allModels.length > 0) {
-					entry.reRegister(allModels);
-					_logger.info(
-						`[pi-free] ${providerId}: showing all ${allModels.length} models`,
-					);
-				}
-			}
-		} catch (err) {
-			_logger.error(
-				`[pi-free] Failed to apply filter to ${providerId}`,
-				err instanceof Error ? { error: err.message } : { error: String(err) },
-			);
-		}
-	}
-}
+/**
+ * Global Provider Registry for pi-free.
+ *
+ * Decoupled from index.ts so providers can import toggle logic
+ * without creating a circular dependency.
+ */
+
+import type {
+	ExtensionAPI,
+	ProviderModelConfig,
+} from "@mariozechner/pi-coding-agent";
+import { getFreeOnly, saveConfig } from "../config.ts";
+import { createLogger } from "./logger.ts";
+
+const _logger = createLogger("pi-free");
+
+// =============================================================================
+// Types
+// =============================================================================
+
+interface ProviderEntry {
+	id: string;
+	stored: { free: ProviderModelConfig[]; all: ProviderModelConfig[] };
+	reRegister: (models: ProviderModelConfig[]) => void;
+	hasKey: boolean;
+}
+
+// =============================================================================
+// State
+// =============================================================================
+
+const providerRegistry = new Map<string, ProviderEntry>();
+let globalFreeOnly = getFreeOnly();
+
+// =============================================================================
+// Free-model detection
+// =============================================================================
+
+/**
+ * Detect if a provider exposes actual per-model pricing.
+ *
+ * Heuristic: If ALL models have cost === 0, the provider likely doesn't expose
+ * real pricing (cost was defaulted to 0). If SOME models have cost > 0, the
+ * provider definitely exposes pricing.
+ *
+ * @param allModels - All models from the provider to check
+ * @returns true if pricing appears to be exposed (some costs > 0)
+ */
+function detectPricingExposed(allModels: ProviderModelConfig[]): boolean {
+	if (allModels.length === 0) return false;
+
+	// If ANY model has cost > 0, pricing is definitely exposed
+	return allModels.some(
+		(m) => (m.cost?.input ?? 0) > 0 || (m.cost?.output ?? 0) > 0,
+	);
+}
+
+/**
+ * Check if a model is free using adaptive Route A/B logic.
+ *
+ * **Automatic Detection:**
+ * The function detects whether the provider exposes pricing by checking if
+ * ALL models have cost === 0. If so, it assumes no pricing is exposed and
+ * falls back to name-based detection.
+ *
+ * **Route A (Pricing-Exposed Providers):** Uses ONLY cost-based detection.
+ *   - Detected when SOME models have cost > 0
+ *   - Free = cost.input === 0 && cost.output === 0
+ *   - No fallback to name-based detection
+ *
+ * **Route B (Non-Pricing-Exposed Providers):** Uses ONLY name-based detection.
+ *   - Detected when ALL models have cost === 0 (or no models)
+ *   - Free = model name contains "free" (case-insensitive)
+ *   - No cost-based detection (avoids marking freemium as free)
+ *
+ * This automatic detection handles providers without hardcoding - if a provider
+ * shows all models as zero cost, we assume pricing isn't exposed and check
+ * model names instead.
+ *
+ * @param model - The model config to check
+ * @param allModels - Optional: all models from the same provider for detection
+ * @returns true if the model is definitively free per the provider's API
+ */
+export function isFreeModel(
+	model: ProviderModelConfig & { provider?: string },
+	allModels?: ProviderModelConfig[],
+): boolean {
+	return isFreeModelInternal(model, allModels);
+}
+
+// Internal implementation to work around TypeScript filter callback issues
+function isFreeModelInternal(
+	model: ProviderModelConfig & { provider?: string },
+	allModels: ProviderModelConfig[] | undefined,
+): boolean {
+	// Determine if pricing is exposed
+	let pricingExposed: boolean;
+
+	if (allModels && allModels.length > 0) {
+		// Dynamic detection: check if ALL models have cost === 0
+		// If all costs are 0, assume pricing is NOT actually exposed
+		pricingExposed = detectPricingExposed(allModels);
+	} else {
+		// No allModels provided - default to cost-based detection
+		// This maintains backward compatibility
+		pricingExposed = true;
+	}
+
+	// Route A: Pricing-exposed providers - use OR logic
+	// Model is free if EITHER cost is zero OR name contains "free"
+	if (pricingExposed) {
+		const isZeroCost =
+			(model.cost?.input ?? 0) === 0 && (model.cost?.output ?? 0) === 0;
+		const hasFreeInName = model.name.toLowerCase().includes("free");
+		return isZeroCost || hasFreeInName;
+	}
+
+	// Route B: Non-pricing-exposed providers - use ONLY name-based detection
+	// This handles providers where all costs are defaulted to 0
+	return model.name.toLowerCase().includes("free");
+}
+
+// =============================================================================
+// Registration
+// =============================================================================
+
+/** Register a provider with the global free/paid toggle system */
+export function registerWithGlobalToggle(
+	providerId: string,
+	stored: { free: ProviderModelConfig[]; all: ProviderModelConfig[] },
+	reRegister: (models: ProviderModelConfig[]) => void,
+	hasKey: boolean = false,
+): void {
+	providerRegistry.set(providerId, {
+		id: providerId,
+		stored,
+		reRegister,
+		hasKey,
+	});
+	_logger.info(
+		`[pi-free] Registered ${providerId} with global toggle (${stored.free.length} free, ${stored.all.length} total)`,
+	);
+}
+
+/** Get current global free-only state */
+export function getGlobalFreeOnly(): boolean {
+	return globalFreeOnly;
+}
+
+/** Access the raw registry (used by /free-providers command) */
+export function getProviderRegistry(): ReadonlyMap<string, ProviderEntry> {
+	return providerRegistry;
+}
+
+// =============================================================================
+// Global filter application
+// =============================================================================
+
+export function applyGlobalFilter(_pi: ExtensionAPI, freeOnly: boolean): void {
+	globalFreeOnly = freeOnly;
+	saveConfig({ free_only: freeOnly });
+
+	for (const [providerId, entry] of providerRegistry) {
+		try {
+			if (freeOnly) {
+				// Show only free models
+				if (entry.stored.free.length > 0) {
+					entry.reRegister(entry.stored.free);
+					_logger.info(
+						`[pi-free] ${providerId}: filtered to ${entry.stored.free.length} free models`,
+					);
+				} else {
+					_logger.warn(`[pi-free] ${providerId}: no free models available`);
+				}
+			} else {
+				// Show all models (paid + free)
+				const allModels =
+					entry.stored.all.length > 0 ? entry.stored.all : entry.stored.free;
+				if (allModels.length > 0) {
+					entry.reRegister(allModels);
+					_logger.info(
+						`[pi-free] ${providerId}: showing all ${allModels.length} models`,
+					);
+				}
+			}
+		} catch (err) {
+			_logger.error(
+				`[pi-free] Failed to apply filter to ${providerId}`,
+				err instanceof Error ? { error: err.message } : { error: String(err) },
+			);
+		}
+	}
+}

package/lib/toggle-state.ts

@@ -0,0 +1,86 @@
+import { saveConfig } from "../config.ts";
+
+export type ToggleMode = "free" | "all";
+
+export interface ToggleModelStore<T> {
+	free: T[];
+	all: T[];
+}
+
+interface CreateToggleStateOptions<T> {
+	providerId: string;
+	initialShowPaid: boolean;
+	save?: typeof saveConfig;
+	initialModels?: ToggleModelStore<T>;
+}
+
+interface ToggleResult<T> {
+	mode: ToggleMode;
+	models: T[];
+}
+
+export function createToggleState<T>({
+	providerId,
+	initialShowPaid,
+	save = saveConfig,
+	initialModels,
+}: CreateToggleStateOptions<T>) {
+	let stored: ToggleModelStore<T> = initialModels ?? { free: [], all: [] };
+	let currentMode: ToggleMode = initialShowPaid ? "all" : "free";
+
+	function resolveMode(mode: ToggleMode): ToggleResult<T> {
+		if (mode === "all") {
+			if (stored.all.length > 0) {
+				return { mode: "all", models: stored.all };
+			}
+			return { mode: "free", models: stored.free };
+		}
+
+		if (stored.free.length > 0) {
+			return { mode: "free", models: stored.free };
+		}
+		return { mode: "all", models: stored.all };
+	}
+
+	function persist(mode: ToggleMode): void {
+		save({ [`${providerId}_show_paid`]: mode === "all" });
+	}
+
+	function applyMode(
+		mode: ToggleMode,
+		apply?: (models: T[]) => void,
+	): ToggleResult<T> {
+		const resolved = resolveMode(mode);
+		currentMode = resolved.mode;
+		if (apply) apply(resolved.models);
+		return resolved;
+	}
+
+	return {
+		setModels(next: ToggleModelStore<T>): ToggleModelStore<T> {
+			stored = next;
+			const resolved = resolveMode(currentMode);
+			currentMode = resolved.mode;
+			return stored;
+		},
+		getStored(): ToggleModelStore<T> {
+			return stored;
+		},
+		getCurrentMode(): ToggleMode {
+			return currentMode;
+		},
+		getCurrentModels(): T[] {
+			return resolveMode(currentMode).models;
+		},
+		applyCurrent(apply?: (models: T[]) => void): ToggleResult<T> {
+			return applyMode(currentMode, apply);
+		},
+		applyMode,
+		toggle(apply?: (models: T[]) => void): ToggleResult<T> {
+			const nextMode = currentMode === "all" ? "free" : "all";
+			const resolved = applyMode(nextMode, apply);
+			persist(resolved.mode);
+			return resolved;
+		},
+	};
+}

package/lib/types.ts

@@ -1,108 +1,101 @@
-/**
- * Shared types for pi-free-providers.
- * Interfaces duplicated across providers consolidated here.
- */
-
-// =============================================================================
-// Provider model configuration (matches Pi's ProviderModelConfig)
-// =============================================================================
-
-export interface CostConfig {
-	input: number;
-	output: number;
-	cacheRead: number;
-	cacheWrite: number;
-}
-
-export interface ProviderModelConfig {
-	id: string;
-	name: string;
-	reasoning: boolean;
-	input: ("text" | "image")[];
-	cost: CostConfig;
-	contextWindow: number;
-	maxTokens: number;
-}
-
-// =============================================================================
-// models.dev schema types
-// =============================================================================
-
-export interface ModelsDevCost {
-	input: number;
-	output: number;
-	cache_read?: number;
-	cache_write?: number;
-}
-
-export interface ModelsDevLimit {
-	context: number;
-	output: number;
-}
-
-export interface ModelsDevModalities {
-	input?: string[];
-	output?: string[];
-}
-
-export interface ModelsDevModel {
-	id: string;
-	name: string;
-	reasoning: boolean;
-	cost?: ModelsDevCost;
-	limit: ModelsDevLimit;
-	modalities?: ModelsDevModalities;
-}
-
-export interface ModelsDevProvider {
-	id: string;
-	api: string;
-	models: Record<string, ModelsDevModel>;
-}
-
-// =============================================================================
-// OpenRouter API response types
-// =============================================================================
-
-export interface OpenRouterPricing {
-	prompt?: string | null;
-	completion?: string | null;
-	input_cache_write?: string | null;
-	input_cache_read?: string | null;
-}
-
-export interface OpenRouterArchitecture {
-	input_modalities?: string[] | null;
-	output_modalities?: string[] | null;
-}
-
-export interface OpenRouterTopProvider {
-	max_completion_tokens?: number | null;
-}
-
-export interface OpenRouterModel {
-	id: string;
-	name: string;
-	context_length: number;
-	max_completion_tokens?: number | null;
-	pricing?: OpenRouterPricing;
-	architecture?: OpenRouterArchitecture;
-	top_provider?: OpenRouterTopProvider;
-	supported_parameters?: string[];
-}
-
-// =============================================================================
-// Zen gateway types
-// =============================================================================
-
-export interface ZenGatewayModel {
-	id: string;
-	object?: string;
-}
-
-// Test: LSP should handle new interface
-export interface LspTestInterface {
-	name: string;
-	value: number;
-	enabled: boolean;
-}
+/**
+ * Shared types for pi-free-providers.
+ * Interfaces duplicated across providers consolidated here.
+ */
+
+// =============================================================================
+// Provider model configuration (matches Pi's ProviderModelConfig)
+// =============================================================================
+
+export interface CostConfig {
+	input: number;
+	output: number;
+	cacheRead: number;
+	cacheWrite: number;
+}
+
+export interface ProviderModelConfig {
+	id: string;
+	name: string;
+	reasoning: boolean;
+	input: ("text" | "image")[];
+	cost: CostConfig;
+	contextWindow: number;
+	maxTokens: number;
+}
+
+// =============================================================================
+// models.dev schema types
+// =============================================================================
+
+export interface ModelsDevCost {
+	input: number;
+	output: number;
+	cache_read?: number;
+	cache_write?: number;
+}
+
+export interface ModelsDevLimit {
+	context: number;
+	output: number;
+}
+
+export interface ModelsDevModalities {
+	input?: string[];
+	output?: string[];
+}
+
+export interface ModelsDevModel {
+	id: string;
+	name: string;
+	reasoning: boolean;
+	cost?: ModelsDevCost;
+	limit: ModelsDevLimit;
+	modalities?: ModelsDevModalities;
+}
+
+export interface ModelsDevProvider {
+	id: string;
+	api: string;
+	models: Record<string, ModelsDevModel>;
+}
+
+// =============================================================================
+// OpenRouter API response types
+// =============================================================================
+
+export interface OpenRouterPricing {
+	prompt?: string | null;
+	completion?: string | null;
+	input_cache_write?: string | null;
+	input_cache_read?: string | null;
+}
+
+export interface OpenRouterArchitecture {
+	input_modalities?: string[] | null;
+	output_modalities?: string[] | null;
+}
+
+export interface OpenRouterTopProvider {
+	max_completion_tokens?: number | null;
+}
+
+export interface OpenRouterModel {
+	id: string;
+	name: string;
+	context_length: number;
+	max_completion_tokens?: number | null;
+	pricing?: OpenRouterPricing;
+	architecture?: OpenRouterArchitecture;
+	top_provider?: OpenRouterTopProvider;
+	supported_parameters?: string[];
+}
+
+// =============================================================================
+// Zen gateway types
+// =============================================================================
+
+export interface ZenGatewayModel {
+	id: string;
+	object?: string;
+}