import { ModuleFederationPlugin } from '@module-federation/enhanced/rspack';
import type { Configuration } from '@rspack/core';
import { DefinePlugin } from '@rspack/core';
import { SharedConfigContext } from '../../model';
import {
  ModuleFederationConfig,
  NxModuleFederationConfigOverride,
} from '../models';
import { getModuleFederationConfig } from './utils';

const isVarOrWindow = (libType?: string) =>
  libType === 'var' || libType === 'window';

/**
 * @param {ModuleFederationConfig} options
 * @param {NxModuleFederationConfigOverride} configOverride
 */
export async function withModuleFederation(
  options: ModuleFederationConfig,
  configOverride?: NxModuleFederationConfigOverride
) {
  if (global.NX_GRAPH_CREATION) {
    return function makeConfig(config: Configuration): Configuration {
      return config;
    };
  }

  const { sharedDependencies, sharedLibraries, mappedRemotes } =
    await getModuleFederationConfig(options);
  const isGlobal = isVarOrWindow(options.library?.type);

  return function makeConfig(
    config: Configuration,
    { context }: SharedConfigContext
  ): Configuration {
    config.output.uniqueName = options.name;
    config.output.publicPath = 'auto';

    if (isGlobal) {
      config.output.scriptType = 'text/javascript';
    }

    config.optimization = {
      ...(config.optimization ?? {}),
      runtimeChunk: false,
    };

    if (
      config.mode === 'development' &&
      Object.keys(mappedRemotes).length > 1 &&
      !options.exposes
    ) {
      config.optimization.runtimeChunk = 'single';
    }

    config.plugins.push(
      new ModuleFederationPlugin({
        name: options.name,
        filename: 'remoteEntry.js',
        exposes: options.exposes,
        remotes: mappedRemotes,
        shared: {
          ...sharedDependencies,
        },
        /**
         * remoteType: 'script' is required for the remote to be loaded as a script tag.
         * remotes will need to be defined as:
         *  { appX: 'appX@http://localhost:3001/remoteEntry.js' }
         *  { appY: 'appY@http://localhost:3002/remoteEntry.js' }
         */
        ...(isGlobal ? { remoteType: 'script' } : {}),
        /**
         * Apply user-defined config overrides
         */
        ...(configOverride ? configOverride : {}),
        runtimePlugins:
          process.env.NX_MF_DEV_REMOTES &&
          !options.disableNxRuntimeLibraryControlPlugin
            ? [
                ...(configOverride?.runtimePlugins ?? []),
                require.resolve(
                  '@nx/rspack/src/utils/module-federation/plugins/runtime-library-control.plugin.js'
                ),
              ]
            : configOverride?.runtimePlugins,
        virtualRuntimeEntry: true,
      }),
      sharedLibraries.getReplacementPlugin()
    );

    // The env var is only set from the module-federation-dev-server
    // Attach the runtime plugin
    config.plugins.push(
      new DefinePlugin({
        'process.env.NX_MF_DEV_REMOTES': process.env.NX_MF_DEV_REMOTES,
      })
    );

    return config;
  };
}