import { createRequire } from 'node:module';
import type { Configuration as RspackConfig } from '@rspack/core';
import type { ImportmapDeclaration, RoutesDeclaration } from './importmap';
import { setEnvVariables } from './variables';

export interface BuildOptions {
  backend?: string;
  defaultLocale?: string;
  importmap?: ImportmapDeclaration;
  routes?: RoutesDeclaration;
  apiUrl?: string;
  spaPath?: string;
  pageTitle?: string;
  supportOffline?: boolean;
  configUrls?: Array<string>;
  env?: string;
  coreAppsDir?: string;
  addCookie?: string;
  fresh?: boolean;
  assets?: Array<string>;
}

/**
 * Maps {@link BuildOptions} to the `OMRS_*` environment variables that the
 * rspack config reads at evaluation time.
 */
export function setBundlerEnv(options: BuildOptions = {}) {
  const variables: Record<string, unknown> = {};

  if (typeof options.backend === 'string') {
    variables.OMRS_PROXY_TARGET = options.backend;
  }

  if (typeof options.spaPath === 'string') {
    variables.OMRS_PUBLIC_PATH = options.spaPath;
  }

  if (typeof options.apiUrl === 'string') {
    variables.OMRS_API_URL = options.apiUrl;
  }

  if (typeof options.pageTitle === 'string') {
    variables.OMRS_PAGE_TITLE = options.pageTitle;
  }

  if (typeof options.addCookie === 'string') {
    variables.OMRS_ADD_COOKIE = options.addCookie;
  }

  if (typeof options.supportOffline === 'boolean') {
    variables.OMRS_OFFLINE = options.supportOffline ? 'enable' : 'disable';
  }

  if (Array.isArray(options.configUrls)) {
    variables.OMRS_CONFIG_URLS = options.configUrls.join(';');
  }

  if (typeof options.env === 'string') {
    variables.OMRS_ENV = options.env;
    variables.NODE_ENV = options.env;
  }

  if (typeof options.defaultLocale === 'string') {
    variables.OMRS_ESM_DEFAULT_LOCALE = options.defaultLocale;
  }

  if (typeof options.importmap === 'object') {
    switch (options.importmap.type) {
      case 'inline':
        variables.OMRS_ESM_IMPORTMAP = options.importmap.value;
        break;
      case 'url':
        variables.OMRS_ESM_IMPORTMAP_URL = options.importmap.value;
        break;
    }
  }

  if (typeof options.routes === 'object') {
    switch (options.routes.type) {
      case 'inline':
        variables.OMRS_ROUTES = options.routes.value;
        break;
      case 'url':
        variables.OMRS_ROUTES_URL = options.routes.value;
        break;
    }
  }

  if (typeof options.coreAppsDir === 'string') {
    variables.OMRS_ESM_CORE_APPS_DIR = options.coreAppsDir;
  }

  if (typeof options.fresh === 'boolean') {
    variables.OMRS_CLEAN_BEFORE_BUILD = options.fresh;
  }

  if (Array.isArray(options.assets)) {
    variables.OMRS_JS_CSS_ASSETS = options.assets.join(';');
  }

  setEnvVariables(variables);
}

export function loadBundlerConfig(options: BuildOptions = {}) {
  setBundlerEnv(options);

  const require = createRequire(import.meta.url);
  const config:
    | ((env: Record<string | number | symbol, unknown>) => RspackConfig)
    | RspackConfig = require('@openmrs/esm-app-shell/rspack.config.js');

  if (typeof config === 'function') {
    return config({});
  }

  return config;
}