@aryee337/aery-ai 0.2.27 → 0.2.29

package/src/utils/oauth/index.ts

@@ -0,0 +1,484 @@
+// ============================================================================
+// High-level API
+// ============================================================================
+import type {
+	OAuthCredentials,
+	OAuthProvider,
+	OAuthProviderId,
+	OAuthProviderInfo,
+	OAuthProviderInterface,
+} from "./types";
+
+const builtInOAuthProviders: OAuthProviderInfo[] = [
+	// Most popular coding subscriptions / gateways.
+	{
+		id: "openai-codex",
+		name: "ChatGPT Plus/Pro (Codex Subscription)",
+		available: true,
+	},
+	{
+		id: "anthropic",
+		name: "Anthropic (Claude Pro/Max)",
+		available: true,
+	},
+	{
+		id: "zai",
+		name: "Z.AI (GLM Coding Plan)",
+		available: true,
+	},
+	{
+		id: "kimi-code",
+		name: "Kimi Code",
+		available: true,
+	},
+	{
+		id: "openrouter",
+		name: "OpenRouter",
+		available: true,
+	},
+	// Other coding subscriptions & first-party assistants.
+	{
+		id: "github-copilot",
+		name: "GitHub Copilot",
+		available: true,
+	},
+	{
+		id: "cursor",
+		name: "Cursor (Claude, GPT, etc.)",
+		available: true,
+	},
+	{
+		id: "google-antigravity",
+		name: "Antigravity (Gemini 3, Claude, GPT-OSS)",
+		available: true,
+	},
+	{
+		id: "google-gemini-cli",
+		name: "Google Cloud Code Assist (Gemini CLI)",
+		available: true,
+	},
+	{
+		id: "openai-codex-device",
+		name: "ChatGPT Plus/Pro (Codex, headless/device)",
+		available: true,
+	},
+	{
+		id: "xai-oauth",
+		name: "xAI Grok OAuth (SuperGrok Subscription)",
+		available: true,
+	},
+	{
+		id: "gitlab-duo",
+		name: "GitLab Duo",
+		available: true,
+	},
+	{
+		id: "alibaba-coding-plan",
+		name: "Alibaba Coding Plan",
+		available: true,
+	},
+	{
+		id: "zhipu-coding-plan",
+		name: "Zhipu Coding Plan (智谱)",
+		available: true,
+	},
+	{
+		id: "qwen-portal",
+		name: "Qwen Portal",
+		available: true,
+	},
+	{
+		id: "minimax-code",
+		name: "MiniMax Coding Plan (International)",
+		available: true,
+	},
+	{
+		id: "minimax-code-cn",
+		name: "MiniMax Coding Plan (China)",
+		available: true,
+	},
+	{
+		id: "xiaomi",
+		name: "Xiaomi MiMo",
+		available: true,
+	},
+	{
+		id: "firepass",
+		name: "Fire Pass (Fireworks Kimi K2.6 Turbo subscription)",
+		available: true,
+	},
+	{
+		id: "wafer-pass",
+		name: "Wafer Pass (flat-rate subscription)",
+		available: true,
+	},
+	// Direct model-provider APIs (pay-as-you-go inference).
+	{
+		id: "deepseek",
+		name: "DeepSeek",
+		available: true,
+	},
+	{
+		id: "moonshot",
+		name: "Moonshot (Kimi API)",
+		available: true,
+	},
+	{
+		id: "cerebras",
+		name: "Cerebras",
+		available: true,
+	},
+	{
+		id: "fireworks",
+		name: "Fireworks",
+		available: true,
+	},
+	{
+		id: "together",
+		name: "Together",
+		available: true,
+	},
+	{
+		id: "nvidia",
+		name: "NVIDIA",
+		available: true,
+	},
+	{
+		id: "huggingface",
+		name: "Hugging Face Inference",
+		available: true,
+	},
+	{
+		id: "perplexity",
+		name: "Perplexity (Pro/Max)",
+		available: true,
+	},
+	{
+		id: "qianfan",
+		name: "Qianfan",
+		available: true,
+	},
+	{
+		id: "venice",
+		name: "Venice",
+		available: true,
+	},
+	{
+		id: "synthetic",
+		name: "Synthetic",
+		available: true,
+	},
+	{
+		id: "nanogpt",
+		name: "NanoGPT",
+		available: true,
+	},
+	{
+		id: "wafer-serverless",
+		name: "Wafer Serverless (pay-as-you-go)",
+		available: true,
+	},
+	// Aggregator gateways / routers.
+	{
+		id: "vercel-ai-gateway",
+		name: "Vercel AI Gateway",
+		available: true,
+	},
+	{
+		id: "cloudflare-ai-gateway",
+		name: "Cloudflare AI Gateway",
+		available: true,
+	},
+	{
+		id: "litellm",
+		name: "LiteLLM",
+		available: true,
+	},
+	{
+		id: "kilo",
+		name: "Kilo Gateway",
+		available: true,
+	},
+	{
+		id: "zenmux",
+		name: "ZenMux",
+		available: true,
+	},
+	{
+		id: "opencode-zen",
+		name: "OpenCode Zen",
+		available: true,
+	},
+	{
+		id: "opencode-go",
+		name: "OpenCode Go",
+		available: true,
+	},
+	// Search & tool providers.
+	{
+		id: "tavily",
+		name: "Tavily",
+		available: true,
+	},
+	{
+		id: "kagi",
+		name: "Kagi",
+		available: true,
+	},
+	{
+		id: "parallel",
+		name: "Parallel",
+		available: true,
+	},
+	// Local runtimes.
+	{
+		id: "ollama",
+		name: "Ollama (Local OpenAI-compatible)",
+		available: true,
+	},
+	{
+		id: "ollama-cloud",
+		name: "Ollama Cloud",
+		available: true,
+	},
+	{
+		id: "lm-studio",
+		name: "LM Studio (Local OpenAI-compatible)",
+		available: true,
+	},
+	{
+		id: "vllm",
+		name: "vLLM (Local OpenAI-compatible)",
+		available: true,
+	},
+];
+
+const customOAuthProviders = new Map<string, OAuthProviderInterface>();
+
+/**
+ * Register a custom OAuth provider.
+ */
+export function registerOAuthProvider(provider: OAuthProviderInterface): void {
+	customOAuthProviders.set(provider.id, provider);
+}
+
+/**
+ * Get a custom OAuth provider by ID.
+ */
+export function getOAuthProvider(id: OAuthProviderId): OAuthProviderInterface | undefined {
+	return customOAuthProviders.get(id);
+}
+
+/**
+ * Remove all custom OAuth providers registered by a source.
+ */
+export function unregisterOAuthProviders(sourceId: string): void {
+	for (const [id, provider] of customOAuthProviders.entries()) {
+		if (provider.sourceId === sourceId) {
+			customOAuthProviders.delete(id);
+		}
+	}
+}
+
+/**
+ * Refresh token for any OAuth provider.
+ * Saves the new credentials and returns the new access token.
+ */
+export async function refreshOAuthToken(
+	provider: OAuthProvider,
+	credentials: OAuthCredentials,
+): Promise<OAuthCredentials> {
+	if (!credentials) {
+		throw new Error(`No OAuth credentials found for ${provider}`);
+	}
+
+	let newCredentials: OAuthCredentials;
+	switch (provider) {
+		case "anthropic": {
+			const { refreshAnthropicToken } = await import("./anthropic");
+			newCredentials = await refreshAnthropicToken(credentials.refresh);
+			break;
+		}
+		case "github-copilot": {
+			const { refreshGitHubCopilotToken } = await import("./github-copilot");
+			newCredentials = await refreshGitHubCopilotToken(credentials.refresh, credentials.enterpriseUrl);
+			break;
+		}
+		case "google-gemini-cli": {
+			const { refreshGoogleCloudToken } = await import("./google-gemini-cli");
+			if (!credentials.projectId) {
+				throw new Error("Google Cloud credentials missing projectId");
+			}
+			newCredentials = await refreshGoogleCloudToken(credentials.refresh, credentials.projectId);
+			break;
+		}
+		case "google-antigravity": {
+			const { refreshAntigravityToken } = await import("./google-antigravity");
+			if (!credentials.projectId) {
+				throw new Error("Antigravity credentials missing projectId");
+			}
+			newCredentials = await refreshAntigravityToken(credentials.refresh, credentials.projectId);
+			break;
+		}
+		case "openai-codex":
+		case "openai-codex-device": {
+			const { refreshOpenAICodexToken } = await import("./openai-codex");
+			newCredentials = await refreshOpenAICodexToken(credentials.refresh);
+			break;
+		}
+		case "kimi-code": {
+			const { refreshKimiToken } = await import("./kimi");
+			newCredentials = await refreshKimiToken(credentials.refresh);
+			break;
+		}
+		case "gitlab-duo": {
+			const { refreshGitLabDuoToken } = await import("./gitlab-duo");
+			newCredentials = await refreshGitLabDuoToken(credentials);
+			break;
+		}
+		case "cursor": {
+			const { refreshCursorToken } = await import("./cursor");
+			newCredentials = await refreshCursorToken(credentials.refresh);
+			break;
+		}
+		case "xai-oauth": {
+			const { refreshXAIOAuthToken } = await import("./xai-oauth");
+			newCredentials = await refreshXAIOAuthToken(credentials.refresh);
+			break;
+		}
+		case "kilo":
+		case "perplexity":
+		case "huggingface":
+		case "opencode-zen":
+		case "opencode-go":
+		case "openrouter":
+		case "cerebras":
+		case "fireworks":
+		case "firepass":
+		case "nvidia":
+		case "nanogpt":
+		case "synthetic":
+		case "together":
+		case "litellm":
+		case "lm-studio":
+		case "ollama":
+		case "ollama-cloud":
+		case "xiaomi":
+		case "zai":
+		case "zhipu-coding-plan":
+		case "qianfan":
+		case "venice":
+		case "minimax-code":
+		case "minimax-code-cn":
+		case "moonshot":
+		case "kagi":
+		case "cloudflare-ai-gateway":
+		case "vercel-ai-gateway":
+		case "qwen-portal":
+		case "wafer-pass":
+		case "wafer-serverless":
+		case "zenmux":
+		case "vllm":
+			// API keys / static bearer tokens don't expire, return as-is
+			newCredentials = credentials;
+			break;
+		default:
+			throw new Error(`Unknown OAuth provider: ${provider}`);
+	}
+	return newCredentials;
+}
+function getPerplexityJwtExpiryMs(token: string): number | undefined {
+	const parts = token.split(".");
+	if (parts.length !== 3) return undefined;
+	const payload = parts[1];
+	if (!payload) return undefined;
+	try {
+		const decoded = JSON.parse(Buffer.from(payload, "base64url").toString("utf8")) as { exp?: unknown };
+		if (typeof decoded.exp !== "number" || !Number.isFinite(decoded.exp)) return undefined;
+		return decoded.exp * 1000 - 5 * 60_000;
+	} catch {
+		return undefined;
+	}
+}
+
+/**
+ * Build API-key bytes for a provider from an already-fresh OAuth credential.
+ *
+ * Refresh is owned by AuthStorage. This helper deliberately refuses expired
+ * credentials so it cannot POST broker redaction sentinels to upstream token
+ * endpoints as a side channel.
+ *
+ * For providers that need credential metadata at request time, returns
+ * JSON-encoded credentials plus expiry metadata for diagnostics/edge guards.
+ * @returns API key string, or null if no credentials
+ * @throws Error if the credential is expired and must be refreshed upstream
+ */
+export async function getOAuthApiKey(
+	provider: OAuthProvider,
+	credentials: Record<string, OAuthCredentials>,
+): Promise<{ newCredentials: OAuthCredentials; apiKey: string } | null> {
+	let creds = credentials[provider];
+	if (!creds) {
+		return null;
+	}
+
+	if (provider === "perplexity") {
+		// Perplexity JWTs usually omit `exp` (server-side sessions). Trust the JWT
+		// claim when present; otherwise treat the credential as non-expiring rather
+		// than honoring a stale stored `expires` (older logins wrote loginTime+1h).
+		const NEVER_EXPIRES = 8.64e15;
+		const normalizedExpires =
+			creds.expires > 0 && creds.expires < 10_000_000_000 ? creds.expires * 1000 : creds.expires;
+		const jwtExpiry = getPerplexityJwtExpiryMs(creds.access);
+		const expires = jwtExpiry ?? Math.max(normalizedExpires, NEVER_EXPIRES);
+		if (expires !== creds.expires) {
+			creds = { ...creds, expires };
+		}
+	}
+	// Refresh is the sole responsibility of `AuthStorage` (which calls
+	// `refreshOAuthToken` directly with broker-aware single-flighting). If we
+	// reach here with an expired credential, the outer pipeline failed to
+	// refresh before this call OR the refresh slot is the broker sentinel —
+	// either way, posting the credential to a provider endpoint would only
+	// trigger a `__remote__`-against-real-provider failure that gets classified
+	// as `invalid_grant` and disables the row. Refuse loudly instead.
+	if (Date.now() >= creds.expires) {
+		if (provider === "perplexity") {
+			const jwtExpiry = getPerplexityJwtExpiryMs(creds.access);
+			if (jwtExpiry && Date.now() < jwtExpiry) {
+				const fallbackCredentials = { ...creds, expires: jwtExpiry };
+				return { newCredentials: fallbackCredentials, apiKey: fallbackCredentials.access };
+			}
+		}
+		throw new Error(
+			`OAuth credential for ${provider} is expired and must be refreshed via AuthStorage before getOAuthApiKey is called`,
+		);
+	}
+	// For providers that need request-time credential metadata, return JSON.
+	const needsStructuredApiKey =
+		provider === "github-copilot" || provider === "google-gemini-cli" || provider === "google-antigravity";
+	const apiKey = needsStructuredApiKey
+		? JSON.stringify({
+				token: creds.access,
+				enterpriseUrl: creds.enterpriseUrl,
+				projectId: creds.projectId,
+				refreshToken: creds.refresh,
+				expiresAt: creds.expires,
+				email: creds.email,
+				accountId: creds.accountId,
+			})
+		: creds.access;
+	return { newCredentials: creds, apiKey };
+}
+
+/**
+ * Get list of OAuth providers.
+ */
+export function getOAuthProviders(): OAuthProviderInfo[] {
+	const customProviders = Array.from(customOAuthProviders.values(), provider => ({
+		id: provider.id,
+		name: provider.name,
+		available: true,
+	}));
+	return [...builtInOAuthProviders, ...customProviders];
+}

package/src/utils/oauth/kagi.ts

@@ -0,0 +1,47 @@
+/**
+ * Kagi login flow.
+ *
+ * Kagi web search uses an API key from the account settings page.
+ * This is an API key flow:
+ * 1. Open browser to Kagi API settings
+ * 2. User copies API key
+ * 3. User pastes key into CLI
+ */
+
+import type { OAuthController } from "./types";
+
+const AUTH_URL = "https://kagi.com/settings/api";
+
+/**
+ * Login to Kagi.
+ *
+ * Opens browser to API settings and prompts user to paste their API key.
+ * Returns the API key directly (not OAuthCredentials - this isn't OAuth).
+ */
+export async function loginKagi(options: OAuthController): Promise<string> {
+	if (!options.onPrompt) {
+		throw new Error("Kagi login requires onPrompt callback");
+	}
+
+	options.onAuth?.({
+		url: AUTH_URL,
+		instructions:
+			"Copy your Kagi Search API key from Kagi API settings. Search API access is beta-only; if unavailable, email support@kagi.com.",
+	});
+
+	const apiKey = await options.onPrompt({
+		message: "Paste your Kagi API key",
+		placeholder: "KG_...",
+	});
+
+	if (options.signal?.aborted) {
+		throw new Error("Login cancelled");
+	}
+
+	const trimmed = apiKey.trim();
+	if (!trimmed) {
+		throw new Error("API key is required");
+	}
+
+	return trimmed;
+}

package/src/utils/oauth/kilo.ts

@@ -0,0 +1,87 @@
+import type { OAuthController, OAuthCredentials } from "./types";
+
+const KILO_DEVICE_AUTH_BASE_URL = "https://api.kilo.ai/api/device-auth";
+const POLL_INTERVAL_MS = 5000;
+const ONE_YEAR_MS = 365 * 24 * 60 * 60 * 1000;
+
+interface KiloDeviceAuthCodeResponse {
+	code?: string;
+	verificationUrl?: string;
+	expiresIn?: number;
+}
+
+interface KiloDeviceAuthPollResponse {
+	status?: string;
+	token?: string;
+}
+
+/**
+ * Login with Kilo Gateway OAuth (device code flow).
+ */
+export async function loginKilo(callbacks: OAuthController): Promise<OAuthCredentials> {
+	const initiateResponse = await fetch(`${KILO_DEVICE_AUTH_BASE_URL}/codes`, {
+		method: "POST",
+		headers: { "Content-Type": "application/json" },
+	});
+
+	if (!initiateResponse.ok) {
+		if (initiateResponse.status === 429) {
+			throw new Error("Too many pending authorization requests. Please try again later.");
+		}
+		throw new Error(`Failed to initiate device authorization: ${initiateResponse.status}`);
+	}
+
+	const initiateData = (await initiateResponse.json()) as KiloDeviceAuthCodeResponse;
+	const userCode = initiateData.code;
+	const verificationUrl = initiateData.verificationUrl;
+	const expiresInSeconds = initiateData.expiresIn;
+	if (!userCode || !verificationUrl || typeof expiresInSeconds !== "number" || expiresInSeconds <= 0) {
+		throw new Error("Kilo device authorization response missing required fields");
+	}
+
+	callbacks.onAuth?.({
+		url: verificationUrl,
+		instructions: `Enter code: ${userCode}`,
+	});
+
+	const deadline = Date.now() + expiresInSeconds * 1000;
+	while (Date.now() < deadline) {
+		if (callbacks.signal?.aborted) {
+			throw new Error("Login cancelled");
+		}
+
+		const pollResponse = await fetch(`${KILO_DEVICE_AUTH_BASE_URL}/codes/${encodeURIComponent(userCode)}`);
+		if (pollResponse.status === 202) {
+			await Bun.sleep(POLL_INTERVAL_MS);
+			continue;
+		}
+		if (pollResponse.status === 403) {
+			throw new Error("Authorization was denied");
+		}
+		if (pollResponse.status === 410) {
+			throw new Error("Authorization code expired. Please try again.");
+		}
+		if (!pollResponse.ok) {
+			throw new Error(`Failed to poll device authorization: ${pollResponse.status}`);
+		}
+
+		const pollData = (await pollResponse.json()) as KiloDeviceAuthPollResponse;
+		if (pollData.status === "approved" && pollData.token) {
+			return {
+				refresh: "",
+				access: pollData.token,
+				expires: Date.now() + ONE_YEAR_MS,
+			};
+		}
+		if (pollData.status === "denied") {
+			throw new Error("Authorization was denied");
+		}
+		if (pollData.status === "expired") {
+			throw new Error("Authorization code expired. Please try again.");
+		}
+
+		await Bun.sleep(POLL_INTERVAL_MS);
+	}
+
+	throw new Error("Authentication timed out. Please try again.");
+}