import { DeploymentType } from "../constants";
import { svgAssets } from "./generated-svg";

/** Runtime registry to override default SVG assets with custom URLs. */
export interface PrismAssetConfig {
  /** Map of SVG logical names (e.g., 'phone_position' or 'phone_position.svg') to absolute URLs. */
  svg?: Record<string, string>;
}

const svgOverrides = new Map<string, string>();

/**
 * Apply asset overrides at runtime; safe to call multiple times.
 */
export function applyCustomAssets(config?: PrismAssetConfig) {
  if (!config?.svg) return;
  for (const [key, url] of Object.entries(config.svg)) {
    svgOverrides.set(normalizeKey(key), url);
  }
}

/**
 * Resolve an SVG asset by logical name or filename; falls back to public folder path.
 */
export function resolveSvg(name: string): string {
  const key = normalizeKey(name);
  const override = svgOverrides.get(key);
  if (override) return override;

  const rawDeployment = (process.env.DEPLOYMENT_TYPE as string) || "local";
  // Normalize: treat "module" as PACKAGE build
  const deployment: DeploymentType =
    (rawDeployment as any) === "module"
      ? DeploymentType.PACKAGE
      : (rawDeployment as any as DeploymentType);

  // For package/module build: return inlined data URL
  if (deployment === DeploymentType.PACKAGE) {
    return svgAssets[key] || `/images/svg/${key}.svg`;
  }

  // CDN build: prefix with CDN_URL
  if (deployment === DeploymentType.CDN) {
    const cdn = (process.env.CDN_URL as string) || "";
    if (cdn) return joinUrl(cdn, `images/svg/${key}.svg`);
  }

  // Local dev: serve from public folder via dev server
  if (deployment === DeploymentType.LOCAL) {
    return `/images/svg/${key}.svg`;
  }

  // Fallback for non-module environments: use public folder path
  return `/images/svg/${key}.svg`;
}

function normalizeKey(key: string): string {
  let k = key.trim();
  // Strip known folder prefix if provided
  if (k.startsWith("/")) k = k.slice(1);
  if (k.startsWith("images/svg/")) k = k.slice("images/svg/".length);
  // Remove extension
  if (k.toLowerCase().endsWith(".svg")) k = k.slice(0, -4);
  return k;
}

/** Clear overrides */
function _clearAssetOverrides() {
  svgOverrides.clear();
}

// Join base URL and path ensuring single slash
function joinUrl(base: string, path: string): string {
  const b = base.endsWith("/") ? base : base + "/";
  const p = path.startsWith("/") ? path.slice(1) : path;
  return b + p;
}