/*
	MIT License http://www.opensource.org/licenses/mit-license.php
	Author Tobias Koppers @sokra
*/

"use strict";

const Template = require("../Template");
const AwaitDependenciesInitFragment = require("../async-modules/AwaitDependenciesInitFragment");
const makeSerializable = require("../util/makeSerializable");
const HarmonyImportDependency = require("./HarmonyImportDependency");
const { ImportPhaseUtils } = require("./ImportPhase");
const NullDependency = require("./NullDependency");

/** @import { ReplaceSource } from "webpack-sources" */
/** @import Dependency from "../Dependency" */
/** @import { DependencyTemplateContext } from "../DependencyTemplate" */
/** @import { Range } from "../javascript/JavascriptParser" */
/** @import HarmonyAcceptImportDependency from "./HarmonyAcceptImportDependency" */
/** @typedef {import("../serialization/ObjectMiddleware").ObjectDeserializerContext<[Range, HarmonyAcceptImportDependency[], boolean]>} ObjectDeserializerContext */
/** @typedef {import("../serialization/ObjectMiddleware").ObjectSerializerContext<[Range, HarmonyAcceptImportDependency[], boolean]>} ObjectSerializerContext */

class HarmonyAcceptDependency extends NullDependency {
	/**
	 * Creates an instance of HarmonyAcceptDependency.
	 * @param {Range} range expression range
	 * @param {HarmonyAcceptImportDependency[]} dependencies import dependencies
	 * @param {boolean} hasCallback true, if the range wraps an existing callback
	 */
	constructor(range, dependencies, hasCallback) {
		super();
		this.range = range;
		/** @type {HarmonyAcceptImportDependency[]} */
		this.dependencies = dependencies;
		/** @type {boolean} */
		this.hasCallback = hasCallback;
	}

	get type() {
		return "accepted harmony modules";
	}

	/**
	 * Serializes this instance into the provided serializer context.
	 * @param {ObjectSerializerContext} context context
	 */
	serialize(context) {
		context.write(this.range).write(this.dependencies).write(this.hasCallback);
		super.serialize(context);
	}

	/**
	 * Restores this instance from the provided deserializer context.
	 * @param {ObjectDeserializerContext} context context
	 */
	deserialize(context) {
		this.range = context.read();
		const c1 = context.rest;
		this.dependencies = c1.read();
		const c2 = c1.rest;
		this.hasCallback = c2.read();
		super.deserialize(c2.rest);
	}
}

makeSerializable(
	HarmonyAcceptDependency,
	"webpack/lib/dependencies/HarmonyAcceptDependency"
);

HarmonyAcceptDependency.Template = class HarmonyAcceptDependencyTemplate extends (
	NullDependency.Template
) {
	/**
	 * Applies the plugin by registering its hooks on the compiler.
	 * @param {Dependency} dependency the dependency for which the template should be applied
	 * @param {ReplaceSource} source the current replace source which can be modified
	 * @param {DependencyTemplateContext} templateContext the context object
	 * @returns {void}
	 */
	apply(dependency, source, templateContext) {
		const dep = /** @type {HarmonyAcceptDependency} */ (dependency);
		const {
			module,
			runtime,
			runtimeRequirements,
			runtimeTemplate,
			moduleGraph,
			chunkGraph
		} = templateContext;

		/**
		 * Checks whether this harmony accept dependency is related harmony import dependency.
		 * @param {Dependency} a the first dependency
		 * @param {Dependency} b the second dependency
		 * @returns {boolean} true if the dependencies are related
		 */
		const isRelatedHarmonyImportDependency = (a, b) => {
			if (a === b || !(b instanceof HarmonyImportDependency)) return false;
			// Compare modules by reference: an unresolved import (ignored/failed, or a
			// deferred lazy-barrel re-export) has no module, and a module not in any
			// chunk has a null id — so comparing ids would crash or miss real matches.
			const moduleA = moduleGraph.getModule(a);
			return moduleA !== null && moduleA === moduleGraph.getModule(b);
		};

		/**
		 * HarmonyAcceptImportDependency lacks a lot of information, such as the defer property.
		 * One HarmonyAcceptImportDependency may need to generate multiple ImportStatements.
		 * Therefore, we find its original HarmonyImportDependency for code generation.
		 * @param {HarmonyAcceptImportDependency} dependency the dependency to get harmony import dependencies for
		 * @returns {HarmonyImportDependency[]} array of related harmony import dependencies
		 */
		const getHarmonyImportDependencies = (dependency) => {
			/** @type {HarmonyImportDependency[]} */
			const result = [];
			/** @type {HarmonyImportDependency | null} */
			let deferDependency = null;
			/** @type {HarmonyImportDependency | null} */
			let noDeferredDependency = null;

			for (const d of module.dependencies) {
				if (deferDependency && noDeferredDependency) break;
				if (isRelatedHarmonyImportDependency(dependency, d)) {
					if (
						ImportPhaseUtils.isDefer(
							/** @type {HarmonyImportDependency} */ (d).phase
						)
					) {
						deferDependency = /** @type {HarmonyImportDependency} */ (d);
					} else {
						noDeferredDependency = /** @type {HarmonyImportDependency} */ (d);
					}
				}
			}
			if (deferDependency) result.push(deferDependency);
			if (noDeferredDependency) result.push(noDeferredDependency);
			if (result.length === 0) {
				// fallback to the original dependency
				result.push(dependency);
			}
			return result;
		};

		/** @type {HarmonyImportDependency[]} */
		const syncDeps = [];

		/** @type {HarmonyAcceptImportDependency[]} */
		const asyncDeps = [];

		for (const dependency of dep.dependencies) {
			const connection = moduleGraph.getConnection(dependency);

			if (connection && moduleGraph.isAsync(connection.module)) {
				asyncDeps.push(dependency);
			} else {
				syncDeps.push(...getHarmonyImportDependencies(dependency));
			}
		}

		let content = syncDeps
			.map((dependency) => {
				const referencedModule = moduleGraph.getModule(dependency);
				return {
					dependency,
					runtimeCondition: referencedModule
						? HarmonyImportDependency.Template.getImportEmittedRuntime(
								module,
								referencedModule
							)
						: false
				};
			})
			.filter(({ runtimeCondition }) => runtimeCondition !== false)
			.map(({ dependency, runtimeCondition }) => {
				const condition = runtimeTemplate.runtimeConditionExpression({
					chunkGraph,
					runtime,
					runtimeCondition,
					runtimeRequirements
				});
				const s = dependency.getImportStatement(true, templateContext);
				const code = s[0] + s[1];
				if (condition !== "true") {
					return `if (${condition}) {\n${Template.indent(code)}\n}\n`;
				}
				return code;
			})
			.join("");

		const promises = new Map(
			asyncDeps.map((dependency) => [
				dependency.getImportVar(moduleGraph),
				dependency.getModuleExports(templateContext)
			])
		);

		let optAsync = "";
		if (promises.size !== 0) {
			optAsync = "async ";
			content += new AwaitDependenciesInitFragment(promises).getContent({
				...templateContext,
				type: "javascript"
			});
		}

		if (dep.hasCallback) {
			if (runtimeTemplate.supportsArrowFunction()) {
				source.insert(
					dep.range[0],
					`${optAsync}__WEBPACK_OUTDATED_DEPENDENCIES__ => { ${content} return (`
				);
				source.insert(dep.range[1], ")(__WEBPACK_OUTDATED_DEPENDENCIES__); }");
			} else {
				source.insert(
					dep.range[0],
					`${optAsync}function(__WEBPACK_OUTDATED_DEPENDENCIES__) { ${content} return (`
				);
				source.insert(
					dep.range[1],
					")(__WEBPACK_OUTDATED_DEPENDENCIES__); }.bind(this)"
				);
			}
			return;
		}

		const arrow = runtimeTemplate.supportsArrowFunction();
		source.insert(
			dep.range[1] - 0.5,
			`, ${arrow ? `${optAsync}() =>` : `${optAsync}function()`} { ${content} }`
		);
	}
};

module.exports = HarmonyAcceptDependency;