import {
	EMBEDDING_PROFILES,
	type EmbeddingProfileId,
} from "@giselles-ai/protocol";
import { ConfigurationError } from "../errors";
import type { EmbedderConfig } from "./ai-sdk-embedder";
import {
	createGatewayEmbedder,
	isGatewaySupportedEmbeddingProfile,
} from "./gateway";
import { createGoogleEmbedder } from "./google";
import { createOpenAIEmbedder } from "./openai";
import type { EmbedderFunction } from "./types";

export type CreateEmbedderFromProfileOptions = Pick<
	EmbedderConfig,
	"maxRetries" | "embeddingComplete" | "headers"
> & {
	transport?: "gateway" | "provider";
};

export function createEmbedderFromProfile(
	profileId: EmbeddingProfileId,
	apiKey: string,
	options?: CreateEmbedderFromProfileOptions,
): EmbedderFunction {
	const profile = EMBEDDING_PROFILES[profileId];
	if (!profile) {
		throw ConfigurationError.invalidValue(
			"profileId",
			profileId,
			`one of: ${Object.keys(EMBEDDING_PROFILES).join(", ")}`,
		);
	}

	const { transport = "provider", ...embedderOptions } = options ?? {};
	const normalizedTransport = transport === "gateway" ? "gateway" : "provider";

	if (transport === "gateway") {
		if (!isGatewaySupportedEmbeddingProfile(profile)) {
			throw new ConfigurationError(
				`Embedding profile '${profileId}' (${profile.provider}/${profile.model}) is not supported by AI Gateway`,
			);
		}

		return createGatewayEmbedder({
			apiKey,
			profile,
			transport: "gateway",
			...embedderOptions,
		});
	}

	switch (profile.provider) {
		case "openai":
			return createOpenAIEmbedder({
				apiKey,
				profile,
				transport: normalizedTransport,
				...embedderOptions,
			});
		case "google":
			return createGoogleEmbedder({
				apiKey,
				profile,
				transport: normalizedTransport,
				...embedderOptions,
			});
		default: {
			const _exhaustiveCheck: never = profile;
			throw new Error(`Unknown provider: ${_exhaustiveCheck}`);
		}
	}
}
