jeopi-ai 16.2.13

package/src/registry/oauth/callback-server.ts

@@ -0,0 +1,315 @@
+/**
+ * Abstract base class for OAuth flows with local callback servers.
+ *
+ * Handles:
+ * - Port allocation (tries expected port, falls back to random)
+ * - Callback server setup and request handling
+ * - Common OAuth flow logic
+ *
+ * Providers extend this and implement:
+ * - generateAuthUrl(): Build provider-specific authorization URL
+ * - exchangeToken(): Exchange authorization code for tokens
+ */
+import * as AIError from "../../error";
+import templateHtml from "./oauth.html" with { type: "text" };
+import type { OAuthController, OAuthCredentials } from "./types";
+
+const DEFAULT_TIMEOUT = 300_000;
+const DEFAULT_HOSTNAME = "localhost";
+const CALLBACK_PATH = "/callback";
+
+export type CallbackResult = { code: string; state: string };
+
+export interface OAuthCallbackFlowOptions {
+	preferredPort: number;
+	callbackPath?: string;
+	callbackHostname?: string;
+	/** Exact redirect URI advertised to the provider; disables port fallback. */
+	redirectUri?: string;
+	/**
+	 * Whether the flow may bind to a random port when {@link preferredPort} is
+	 * unavailable. Defaults to `true` so historical AI-provider flows (which
+	 * pick uncommon ports and tolerate any loopback callback) keep working.
+	 *
+	 * Set to `false` for providers that validate the redirect URI against a
+	 * registered callback — silently advertising a random-port URI would be
+	 * rejected by the authorization server, leaving the browser on an opaque
+	 * 500 page and the local callback waiting until the 5-minute timeout fires.
+	 * With fallback disabled, {@link OAuthCallbackFlow.login} throws a
+	 * {@link AIError.ConfigurationError} immediately so the caller can surface
+	 * an actionable message before opening the browser.
+	 */
+	allowPortFallback?: boolean;
+}
+
+/**
+ * Abstract base class for OAuth flows with local callback servers.
+ */
+export abstract class OAuthCallbackFlow {
+	ctrl: OAuthController;
+	preferredPort: number;
+	callbackPath: string;
+	callbackHostname: string;
+	redirectUri?: string;
+	allowPortFallback: boolean;
+	#callbackResolve?: (result: CallbackResult) => void;
+	#callbackReject?: (error: string) => void;
+
+	constructor(
+		ctrl: OAuthController,
+		preferredPortOrOptions: number | OAuthCallbackFlowOptions,
+		callbackPath: string = CALLBACK_PATH,
+	) {
+		this.ctrl = ctrl;
+		if (typeof preferredPortOrOptions === "number") {
+			this.preferredPort = preferredPortOrOptions;
+			this.callbackPath = callbackPath;
+			this.callbackHostname = DEFAULT_HOSTNAME;
+			this.allowPortFallback = true;
+			return;
+		}
+
+		this.preferredPort = preferredPortOrOptions.preferredPort;
+		this.callbackPath = preferredPortOrOptions.callbackPath ?? CALLBACK_PATH;
+		this.callbackHostname = preferredPortOrOptions.callbackHostname ?? DEFAULT_HOSTNAME;
+		this.redirectUri = preferredPortOrOptions.redirectUri;
+		this.allowPortFallback = preferredPortOrOptions.allowPortFallback ?? true;
+	}
+
+	/**
+	 * Generate provider-specific authorization URL.
+	 * @param state - CSRF state token
+	 * @param redirectUri - The actual redirect URI to use (may differ from expected if port fallback occurred)
+	 * @returns Authorization URL and optional instructions
+	 */
+	abstract generateAuthUrl(state: string, redirectUri: string): Promise<{ url: string; instructions?: string }>;
+
+	/**
+	 * Exchange authorization code for OAuth tokens.
+	 * @param code - Authorization code from callback
+	 * @param state - CSRF state token
+	 * @param redirectUri - The actual redirect URI used (must match authorization request)
+	 * @returns OAuth credentials
+	 */
+	abstract exchangeToken(code: string, state: string, redirectUri: string): Promise<OAuthCredentials>;
+
+	/**
+	 * Generate CSRF state token. Override if provider needs custom state generation.
+	 */
+	generateState(): string {
+		const bytes = new Uint8Array(16);
+		crypto.getRandomValues(bytes);
+		return Array.from(bytes)
+			.map(value => value.toString(16).padStart(2, "0"))
+			.join("");
+	}
+
+	#loginCancelledError(): AIError.LoginCancelledError {
+		return new AIError.LoginCancelledError(`OAuth callback cancelled: ${this.ctrl.signal?.reason}`);
+	}
+
+	#throwIfCancelled(): void {
+		if (this.ctrl.signal?.aborted) throw this.#loginCancelledError();
+	}
+
+	/**
+	 * Execute the OAuth login flow.
+	 */
+	async login(): Promise<OAuthCredentials> {
+		const state = this.generateState();
+		this.#throwIfCancelled();
+
+		// Start callback server first to get actual redirect URI
+		const { server, redirectUri } = await this.#startCallbackServer(state);
+
+		try {
+			this.#throwIfCancelled();
+			// Generate auth URL with the ACTUAL redirect URI (may differ from expected if port was busy)
+			const { url: authUrl, instructions } = await this.generateAuthUrl(state, redirectUri);
+			this.#throwIfCancelled();
+
+			// Notify controller that auth is ready
+			this.ctrl.onAuth?.({ url: authUrl, instructions });
+			this.ctrl.onProgress?.("Waiting for browser authentication...");
+
+			// Wait for callback or manual input
+			const { code } = await this.#waitForCallback(state);
+			this.#throwIfCancelled();
+
+			this.ctrl.onProgress?.("Exchanging authorization code for tokens...");
+
+			return await this.exchangeToken(code, state, redirectUri);
+		} finally {
+			server.stop();
+		}
+	}
+
+	/**
+	 * Start callback server, trying preferred port first, falling back to random.
+	 */
+	async #startCallbackServer(expectedState: string): Promise<{ server: Bun.Server<unknown>; redirectUri: string }> {
+		try {
+			const server = this.#createServer(this.preferredPort, expectedState);
+			if (this.redirectUri) {
+				return { server, redirectUri: this.redirectUri };
+			}
+			const redirectUri = `http://${this.callbackHostname}:${this.preferredPort}${this.callbackPath}`;
+			return { server, redirectUri };
+		} catch (cause) {
+			if (this.redirectUri) {
+				throw new AIError.ConfigurationError(
+					`OAuth callback port ${this.preferredPort} is in use, but oauth.redirectUri (${this.redirectUri}) requires this exact port. Free port ${this.preferredPort} (e.g. stop the process bound to it) and retry, or change oauth.redirectUri to point at an available port.`,
+					{ cause },
+				);
+			}
+			if (!this.allowPortFallback) {
+				throw new AIError.ConfigurationError(
+					`OAuth callback port ${this.preferredPort} is in use. The OAuth provider validates redirect URIs against its registered callback, so falling back to a random port would be rejected. Free port ${this.preferredPort} (e.g. stop the process bound to it) and retry, or set oauth.callbackPort/oauth.redirectUri to a port the provider has registered.`,
+					{ cause },
+				);
+			}
+			const server = this.#createServer(0, expectedState);
+			const actualPort = server.port;
+			const redirectUri = `http://${this.callbackHostname}:${actualPort}${this.callbackPath}`;
+			this.ctrl.onProgress?.(`Preferred port ${this.preferredPort} unavailable, using port ${actualPort}`);
+			return { server, redirectUri };
+		}
+	}
+
+	/**
+	 * Create HTTP server for OAuth callback.
+	 */
+	#createServer(port: number, expectedState: string): Bun.Server<unknown> {
+		const hostname = this.callbackHostname === DEFAULT_HOSTNAME ? undefined : this.callbackHostname;
+		return Bun.serve({
+			...(hostname === undefined ? {} : { hostname }),
+			port,
+			reusePort: false,
+			fetch: req => this.#handleCallback(req, expectedState),
+		});
+	}
+
+	/**
+	 * Handle OAuth callback HTTP request.
+	 */
+	#handleCallback(req: Request, expectedState: string): Response {
+		const url = new URL(req.url);
+
+		if (url.pathname !== this.callbackPath) {
+			return new Response("Not Found", { status: 404 });
+		}
+
+		const code = url.searchParams.get("code");
+		const state = url.searchParams.get("state") || "";
+		const error = url.searchParams.get("error") || "";
+		const errorDescription = url.searchParams.get("error_description") || error;
+
+		type OkState = { ok: true; code: string; state: string };
+		type ErrorState = { ok?: false; error?: string };
+		let resultState: OkState | ErrorState;
+
+		if (error) {
+			resultState = { ok: false, error: `Authorization failed: ${errorDescription}` };
+		} else if (!code) {
+			resultState = { ok: false, error: "Missing authorization code" };
+		} else if (expectedState && state !== expectedState) {
+			resultState = { ok: false, error: "State mismatch - possible CSRF attack" };
+		} else {
+			resultState = { ok: true, code, state };
+		}
+
+		// Signal to waitForCallback - capture refs before they could be cleared
+		const resolve = this.#callbackResolve;
+		const reject = this.#callbackReject;
+		queueMicrotask(() => {
+			if (resultState.ok) {
+				resolve?.({ code: resultState.code, state: resultState.state });
+			} else {
+				reject?.(resultState.error ?? "Unknown error");
+			}
+		});
+
+		return new Response(
+			(templateHtml as unknown as string).replaceAll("__OAUTH_STATE__", JSON.stringify(resultState)),
+			{
+				status: resultState.ok ? 200 : 500,
+				headers: { "Content-Type": "text/html" },
+			},
+		);
+	}
+
+	/**
+	 * Wait for OAuth callback or manual input (whichever comes first).
+	 */
+	#waitForCallback(expectedState: string): Promise<CallbackResult> {
+		const timeoutSignal = AbortSignal.timeout(DEFAULT_TIMEOUT);
+		const signal = this.ctrl.signal ? AbortSignal.any([this.ctrl.signal, timeoutSignal]) : timeoutSignal;
+		if (signal.aborted) return Promise.reject(this.#loginCancelledError());
+
+		const callback = Promise.withResolvers<CallbackResult>();
+		this.#callbackResolve = callback.resolve;
+		this.#callbackReject = callback.reject;
+
+		signal.addEventListener("abort", () => {
+			this.#callbackResolve = undefined;
+			this.#callbackReject = undefined;
+			callback.reject(new AIError.LoginCancelledError(`OAuth callback cancelled: ${signal.reason}`));
+		});
+		const callbackPromise = callback.promise;
+
+		// Manual input race (if supported)
+		if (this.ctrl.onManualCodeInput) {
+			const requestManualInput = this.ctrl.onManualCodeInput;
+			const manualPromise = (async (): Promise<CallbackResult> => {
+				while (true) {
+					const result = await Promise.race([
+						callbackPromise,
+						requestManualInput()
+							.then((input): CallbackResult | null => {
+								const parsed = parseCallbackInput(input);
+								if (!parsed.code) return null;
+								if (expectedState && parsed.state && parsed.state !== expectedState) return null;
+								return { code: parsed.code, state: parsed.state ?? "" };
+							})
+							.catch((): CallbackResult | null => null),
+					]);
+					if (result) return result;
+				}
+			})();
+
+			return Promise.race([callbackPromise, manualPromise]);
+		}
+
+		return callbackPromise;
+	}
+}
+
+/**
+ * Parse a redirect URL or code string to extract code and state.
+ */
+export function parseCallbackInput(input: string): { code?: string; state?: string } {
+	const value = input.trim();
+	if (!value) return {};
+
+	try {
+		const url = new URL(value);
+		return {
+			code: url.searchParams.get("code") ?? undefined,
+			state: url.searchParams.get("state") ?? undefined,
+		};
+	} catch {
+		// Not a URL - check for query string format
+	}
+
+	if (value.includes("code=")) {
+		const params = new URLSearchParams(value.replace(/^[?#]/, ""));
+		return {
+			code: params.get("code") ?? undefined,
+			state: params.get("state") ?? undefined,
+		};
+	}
+
+	// Assume raw code, possibly with state after #
+	const [code, state] = value.split("#", 2);
+	return { code, state };
+}

package/src/registry/oauth/cursor.ts

@@ -0,0 +1,171 @@
+import * as AIError from "../../error";
+import { generatePKCE } from "./pkce";
+import type { OAuthCredentials } from "./types";
+
+const CURSOR_LOGIN_URL = "https://cursor.com/loginDeepControl";
+const CURSOR_POLL_URL = "https://api2.cursor.sh/auth/poll";
+const CURSOR_REFRESH_URL = "https://api2.cursor.sh/auth/exchange_user_api_key";
+
+const POLL_MAX_ATTEMPTS = 150;
+const POLL_BASE_DELAY = 1000;
+const POLL_MAX_DELAY = 10000;
+const POLL_BACKOFF_MULTIPLIER = 1.2;
+
+export interface CursorAuthParams {
+	verifier: string;
+	challenge: string;
+	uuid: string;
+	loginUrl: string;
+}
+
+export async function generateCursorAuthParams(): Promise<CursorAuthParams> {
+	const { verifier, challenge } = await generatePKCE();
+	const uuid = crypto.randomUUID();
+
+	const params = new URLSearchParams({
+		challenge,
+		uuid,
+		mode: "login",
+		redirectTarget: "cli",
+	});
+
+	const loginUrl = `${CURSOR_LOGIN_URL}?${params.toString()}`;
+
+	return { verifier, challenge, uuid, loginUrl };
+}
+
+export async function pollCursorAuth(
+	uuid: string,
+	verifier: string,
+): Promise<{ accessToken: string; refreshToken: string }> {
+	let delay = POLL_BASE_DELAY;
+	let consecutiveErrors = 0;
+
+	for (let attempt = 0; attempt < POLL_MAX_ATTEMPTS; attempt++) {
+		await Bun.sleep(delay);
+
+		try {
+			const response = await fetch(`${CURSOR_POLL_URL}?uuid=${uuid}&verifier=${verifier}`);
+
+			if (response.status === 404) {
+				consecutiveErrors = 0;
+				delay = Math.min(delay * POLL_BACKOFF_MULTIPLIER, POLL_MAX_DELAY);
+				continue;
+			}
+
+			if (response.ok) {
+				const data = (await response.json()) as {
+					accessToken: string;
+					refreshToken: string;
+				};
+				return {
+					accessToken: data.accessToken,
+					refreshToken: data.refreshToken,
+				};
+			}
+
+			throw new AIError.OAuthError(`Poll failed: ${response.status}`, {
+				kind: "polling",
+				provider: "cursor",
+				status: response.status,
+			});
+		} catch {
+			consecutiveErrors++;
+			if (consecutiveErrors >= 3) {
+				throw new AIError.OAuthError("Too many consecutive errors during Cursor auth polling", {
+					kind: "polling",
+					provider: "cursor",
+				});
+			}
+		}
+	}
+
+	throw new AIError.OAuthError("Cursor authentication polling timeout", {
+		kind: "timeout",
+		provider: "cursor",
+	});
+}
+
+export async function loginCursor(
+	onAuthUrl: (url: string) => void,
+	onPollStart?: () => void,
+): Promise<OAuthCredentials> {
+	const { verifier, uuid, loginUrl } = await generateCursorAuthParams();
+
+	onAuthUrl(loginUrl);
+	onPollStart?.();
+
+	const { accessToken, refreshToken } = await pollCursorAuth(uuid, verifier);
+
+	const expiresAt = getTokenExpiry(accessToken);
+
+	return {
+		access: accessToken,
+		refresh: refreshToken,
+		expires: expiresAt,
+	};
+}
+
+export async function refreshCursorToken(apiKeyOrRefreshToken: string): Promise<OAuthCredentials> {
+	const response = await fetch(CURSOR_REFRESH_URL, {
+		method: "POST",
+		headers: {
+			Authorization: `Bearer ${apiKeyOrRefreshToken}`,
+			"Content-Type": "application/json",
+		},
+		body: "{}",
+	});
+
+	if (!response.ok) {
+		const error = await response.text();
+		throw new AIError.OAuthError(`Cursor token refresh failed: ${error}`, {
+			kind: "token-refresh",
+			provider: "cursor",
+		});
+	}
+
+	const data = (await response.json()) as {
+		accessToken: string;
+		refreshToken: string;
+	};
+
+	const expiresAt = getTokenExpiry(data.accessToken);
+
+	return {
+		access: data.accessToken,
+		refresh: data.refreshToken || apiKeyOrRefreshToken,
+		expires: expiresAt,
+	};
+}
+
+function getTokenExpiry(token: string): number {
+	try {
+		const parts = token.split(".");
+		if (parts.length !== 3) {
+			return Date.now() + 3600 * 1000;
+		}
+		const payload = parts[1];
+		if (!payload) {
+			return Date.now() + 3600 * 1000;
+		}
+		const decoded = JSON.parse(atob(payload.replace(/-/g, "+").replace(/_/g, "/")));
+		if (decoded && typeof decoded === "object" && typeof decoded.exp === "number") {
+			return decoded.exp * 1000 - 5 * 60 * 1000;
+		}
+	} catch {
+		// Ignore parsing errors
+	}
+	return Date.now() + 3600 * 1000;
+}
+
+export function isCursorTokenExpiringSoon(token: string, thresholdSeconds = 300): boolean {
+	try {
+		const [, payload] = token.split(".");
+		if (!payload) return true;
+		const decoded = JSON.parse(atob(payload.replace(/-/g, "+").replace(/_/g, "/")));
+		const currentTime = Math.floor(Date.now() / 1000);
+		return decoded.exp - currentTime < thresholdSeconds;
+	} catch {
+		return true;
+	}
+}

package/src/registry/oauth/devin.ts

@@ -0,0 +1,124 @@
+import * as AIError from "../../error";
+import { OAuthCallbackFlow } from "./callback-server";
+import { generatePKCE } from "./pkce";
+import type { OAuthController, OAuthCredentials } from "./types";
+
+type FetchFunction = NonNullable<OAuthController["fetch"]>;
+
+const DEVIN_WEBAPP_URL = "https://app.devin.ai";
+const DEVIN_API_URL = "https://api.devin.ai";
+const CALLBACK_PORT = 59653;
+const CALLBACK_PATH = "/callback";
+const TOKEN_PATH = "/auth/cli/token";
+const FALLBACK_EXPIRES_MS = 365 * 24 * 60 * 60 * 1000;
+
+interface DevinPKCEParams {
+	verifier: string;
+	challenge: string;
+}
+
+export async function loginDevin(ctrl: OAuthController): Promise<OAuthCredentials> {
+	const flow = new DevinOAuthFlow(ctrl);
+	return flow.login();
+}
+
+class DevinOAuthFlow extends OAuthCallbackFlow {
+	#pkce?: DevinPKCEParams;
+
+	constructor(ctrl: OAuthController) {
+		super(ctrl, {
+			preferredPort: CALLBACK_PORT,
+			callbackPath: CALLBACK_PATH,
+			callbackHostname: "127.0.0.1",
+		});
+	}
+
+	generateState(): string {
+		return crypto.randomUUID();
+	}
+
+	async generateAuthUrl(state: string, redirectUri: string): Promise<{ url: string; instructions?: string }> {
+		this.#pkce = await generatePKCE();
+		const params = new URLSearchParams({
+			redirect_uri: redirectUri,
+			state,
+			prompt: "select_account",
+			code_challenge: this.#pkce.challenge,
+			code_challenge_method: "S256",
+		});
+
+		return {
+			url: `${DEVIN_WEBAPP_URL}/auth/cli/continue?${params.toString()}`,
+			instructions: "Sign in to Devin in your browser.",
+		};
+	}
+
+	async exchangeToken(code: string): Promise<OAuthCredentials> {
+		if (!this.#pkce) {
+			throw new AIError.OAuthError("Devin PKCE verifier was not initialized", {
+				kind: "configuration",
+				provider: "devin",
+			});
+		}
+		const token = await exchangeDevinCliToken(code, this.#pkce.verifier, this.ctrl.fetch);
+
+		return {
+			access: token,
+			refresh: token,
+			expires: getTokenExpiry(token),
+			apiEndpoint: DEVIN_API_URL,
+			enterpriseUrl: DEVIN_WEBAPP_URL,
+		};
+	}
+}
+
+export async function exchangeDevinCliToken(
+	authorizationCode: string,
+	codeVerifier: string,
+	fetchImpl: FetchFunction = fetch,
+): Promise<string> {
+	const response = await fetchImpl(`${DEVIN_API_URL}${TOKEN_PATH}`, {
+		method: "POST",
+		headers: {
+			Accept: "application/json",
+			"Content-Type": "application/json",
+		},
+		body: JSON.stringify({
+			code: authorizationCode,
+			code_verifier: codeVerifier,
+		}),
+	});
+
+	if (!response.ok) {
+		const error = await response.text();
+		throw new AIError.OAuthError(`Devin CLI token exchange failed: ${response.status} ${error}`.trim(), {
+			kind: "token-exchange",
+			provider: "devin",
+			status: response.status,
+		});
+	}
+
+	const data = (await response.json()) as { token?: unknown };
+	if (typeof data.token !== "string" || data.token.length === 0) {
+		throw new AIError.OAuthError("Devin CLI token exchange returned an empty token", {
+			kind: "validation",
+			provider: "devin",
+		});
+	}
+	return data.token;
+}
+
+function getTokenExpiry(token: string): number {
+	try {
+		const [, payload] = token.split(".");
+		if (payload) {
+			const decoded = JSON.parse(Buffer.from(payload, "base64url").toString("utf8")) as { exp?: unknown };
+			if (typeof decoded.exp === "number" && Number.isFinite(decoded.exp)) {
+				return decoded.exp * 1000 - 5 * 60 * 1000;
+			}
+		}
+	} catch {
+		// Ignore malformed non-JWT tokens and use a conservative long-lived fallback.
+	}
+	return Date.now() + FALLBACK_EXPIRES_MS;
+}