/**
 * Xiaomi MiMo login flow.
 *
 * Xiaomi MiMo provides OpenAI-compatible models via
 * https://api.xiaomimimo.com/v1.
 *
 * Standard Xiaomi login opens the pay-as-you-go API key console. Token Plan
 * login opens plan management so users copy the regional `tp-...` key.
 */

import * as AIError from "../../error";
import type { FetchImpl } from "../../types";
import type { OAuthController } from "./types";

const PROVIDER_ID = "xiaomi";
const PROVIDER_NAME = "Xiaomi MiMo";
const STANDARD_AUTH_URL = "https://platform.xiaomimimo.com/#/console/api-keys";
const TOKEN_PLAN_AUTH_URL = "https://platform.xiaomimimo.com/console/plan-manage";
const STANDARD_API_BASE_URL = "https://api.xiaomimimo.com/v1";
const TOKEN_PLAN_KEY_PREFIX = "tp-";
const STANDARD_VALIDATION_MODEL = "mimo-v2-flash";
const TOKEN_PLAN_VALIDATION_MODEL = "mimo-v2.5";
const TOKEN_PLAN_SGP_API_BASE_URL = "https://token-plan-sgp.xiaomimimo.com/v1";
const TOKEN_PLAN_AMS_API_BASE_URL = "https://token-plan-ams.xiaomimimo.com/v1";
const TOKEN_PLAN_CN_API_BASE_URL = "https://token-plan-cn.xiaomimimo.com/v1";

/** Region codes accepted by the Xiaomi Token Plan login flow. */
export type XiaomiTokenPlanRegion = "sgp" | "ams" | "cn";

type XiaomiValidationEndpoint = {
	baseUrl: string;
	model: string;
};

const TOKEN_PLAN_VALIDATION_ENDPOINTS: Record<XiaomiTokenPlanRegion, XiaomiValidationEndpoint> = {
	sgp: { baseUrl: TOKEN_PLAN_SGP_API_BASE_URL, model: TOKEN_PLAN_VALIDATION_MODEL },
	ams: { baseUrl: TOKEN_PLAN_AMS_API_BASE_URL, model: TOKEN_PLAN_VALIDATION_MODEL },
	cn: { baseUrl: TOKEN_PLAN_CN_API_BASE_URL, model: TOKEN_PLAN_VALIDATION_MODEL },
};

const TOKEN_PLAN_REGION_NAMES: Record<XiaomiTokenPlanRegion, string> = {
	sgp: "Singapore",
	ams: "Europe",
	cn: "China",
};

function isTokenPlanKey(apiKey: string): boolean {
	return apiKey.startsWith(TOKEN_PLAN_KEY_PREFIX);
}

const VALIDATION_TIMEOUT_MS = 15_000;

async function validateXiaomiApiKey(
	apiKey: string,
	tokenPlanRegion: XiaomiTokenPlanRegion | undefined,
	signal?: AbortSignal,
	fetchOverride?: FetchImpl,
): Promise<void> {
	const fetchImpl = fetchOverride ?? fetch;
	// Region-specific Token Plan logins must validate against the selected
	// cluster. Generic Xiaomi login keeps the historical SGP → AMS → CN fallback.
	const endpoints = tokenPlanRegion
		? [TOKEN_PLAN_VALIDATION_ENDPOINTS[tokenPlanRegion]]
		: isTokenPlanKey(apiKey)
			? [
					TOKEN_PLAN_VALIDATION_ENDPOINTS.sgp,
					TOKEN_PLAN_VALIDATION_ENDPOINTS.ams,
					TOKEN_PLAN_VALIDATION_ENDPOINTS.cn,
				]
			: [{ baseUrl: STANDARD_API_BASE_URL, model: STANDARD_VALIDATION_MODEL }];

	let lastError: Error | null = null;

	for (const ep of endpoints) {
		// Fresh timeout per endpoint so SGP→AMS fallback works after a regional
		// timeout: a shared AbortSignal.timeout would stay aborted and instantly
		// abort the AMS fetch.
		const timeoutSignal = AbortSignal.timeout(VALIDATION_TIMEOUT_MS);
		const requestSignal = signal ? AbortSignal.any([signal, timeoutSignal]) : timeoutSignal;
		try {
			const response = await fetchImpl(`${ep.baseUrl}/chat/completions`, {
				method: "POST",
				headers: {
					"Content-Type": "application/json",
					Authorization: `Bearer ${apiKey}`,
				},
				body: JSON.stringify({
					model: ep.model,
					max_tokens: 1,
					messages: [{ role: "user", content: "ping" }],
				}),
				signal: requestSignal,
			});

			if (response.ok) {
				return;
			}

			// 401 means this endpoint didn't accept the key; try the next one
			if (response.status === 401) {
				let details = "";
				try {
					details = (await response.text()).trim();
				} catch {
					// ignore body parse errors, status is enough
				}
				lastError = new AIError.OAuthError(
					details
						? `${PROVIDER_NAME} API key validation failed (${response.status}): ${details}`
						: `${PROVIDER_NAME} API key validation failed (${response.status})`,
					{ kind: "validation", provider: PROVIDER_ID, status: response.status },
				);
				continue;
			}

			// Non-auth errors are real failures
			let details = "";
			try {
				details = (await response.text()).trim();
			} catch {
				// ignore body parse errors, status is enough
			}
			const message = details
				? `${PROVIDER_NAME} API key validation failed (${response.status}): ${details}`
				: `${PROVIDER_NAME} API key validation failed (${response.status})`;
			throw new AIError.OAuthError(message, {
				kind: "validation",
				provider: PROVIDER_ID,
				status: response.status,
			});
		} catch (e) {
			// Only re-throw AbortError when the caller explicitly cancelled.
			// Timeout aborts (from AbortSignal.timeout) should fall through to
			// the next endpoint so SGP→AMS fallback works during regional outages.
			if (e instanceof DOMException && e.name === "AbortError" && signal?.aborted) {
				throw e;
			}
			lastError = e instanceof Error ? e : new Error(String(e));
		}
	}
	throw (
		lastError ??
		new AIError.OAuthError(`${PROVIDER_NAME} API key validation failed`, {
			kind: "validation",
			provider: PROVIDER_ID,
		})
	);
}

/**
 * Login to Xiaomi MiMo.
 *
 * Opens browser to API keys page, prompts user to paste their API key.
 * Returns the API key directly (not OAuthCredentials - this isn't OAuth).
 */
export async function loginXiaomi(options: OAuthController): Promise<string> {
	const fetchImpl = options.fetch ?? fetch;
	if (!options.onPrompt) {
		throw new AIError.OnPromptRequiredError(PROVIDER_NAME);
	}
	options.onAuth?.({
		url: STANDARD_AUTH_URL,
		instructions: "Copy your API key from the Xiaomi MiMo console",
	});
	const apiKey = await options.onPrompt({
		message: "Paste your Xiaomi API key (sk-... or token-plan tp-...)",
		placeholder: "sk-... or tp-...",
	});
	if (options.signal?.aborted) {
		throw new AIError.LoginCancelledError();
	}
	const trimmed = apiKey.trim();
	if (!trimmed) {
		throw new AIError.ApiKeyRequiredError();
	}

	options.onProgress?.(`Validating ${PROVIDER_ID} API key...`);
	await validateXiaomiApiKey(trimmed, undefined, options.signal, fetchImpl);
	return trimmed;
}

/**
 * Login to a regional Xiaomi Token Plan endpoint.
 *
 * Prompts for a token-plan API key and validates it against the selected region.
 */
export async function loginXiaomiTokenPlan(options: OAuthController, region: XiaomiTokenPlanRegion): Promise<string> {
	const fetchImpl = options.fetch ?? fetch;
	if (!options.onPrompt) {
		throw new AIError.OnPromptRequiredError(`Xiaomi Token Plan (${TOKEN_PLAN_REGION_NAMES[region]})`);
	}
	options.onAuth?.({
		url: TOKEN_PLAN_AUTH_URL,
		instructions: `Copy your token-plan API key for the ${TOKEN_PLAN_REGION_NAMES[region]} region`,
	});
	const apiKey = await options.onPrompt({
		message: `Paste your Xiaomi Token Plan ${TOKEN_PLAN_REGION_NAMES[region]} API key (tp-...)`,
		placeholder: "tp-...",
	});
	if (options.signal?.aborted) {
		throw new AIError.LoginCancelledError();
	}
	const trimmed = apiKey.trim();
	if (!trimmed) {
		throw new AIError.ApiKeyRequiredError();
	}

	options.onProgress?.(`Validating Xiaomi Token Plan (${TOKEN_PLAN_REGION_NAMES[region]}) API key...`);
	await validateXiaomiApiKey(trimmed, region, options.signal, fetchImpl);
	return trimmed;
}