205 lines
		
	
	
		
			6.8 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
			
		
		
	
	
			205 lines
		
	
	
		
			6.8 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
| /*
 | |
| 	MIT License http://www.opensource.org/licenses/mit-license.php
 | |
| 	Author Tobias Koppers @sokra
 | |
| */
 | |
| 
 | |
| "use strict";
 | |
| 
 | |
| const ContextElementDependency = require("./dependencies/ContextElementDependency");
 | |
| const { join } = require("./util/fs");
 | |
| 
 | |
| /** @typedef {import("./Compiler")} Compiler */
 | |
| /** @typedef {import("./ContextModule").ContextModuleOptions} ContextModuleOptions */
 | |
| /** @typedef {import("./util/fs").InputFileSystem} InputFileSystem */
 | |
| 
 | |
| /** @typedef {Record<string, string>} NewContentCreateContextMap */
 | |
| 
 | |
| const PLUGIN_NAME = "ContextReplacementPlugin";
 | |
| 
 | |
| class ContextReplacementPlugin {
 | |
| 	/**
 | |
| 	 * @param {RegExp} resourceRegExp A regular expression that determines which files will be selected
 | |
| 	 * @param {(string | ((context: TODO) => void) | RegExp | boolean)=} newContentResource A new resource to replace the match
 | |
| 	 * @param {(boolean | NewContentCreateContextMap | RegExp)=} newContentRecursive If true, all subdirectories are searched for matches
 | |
| 	 * @param {RegExp=} newContentRegExp A regular expression that determines which files will be selected
 | |
| 	 */
 | |
| 	constructor(
 | |
| 		resourceRegExp,
 | |
| 		newContentResource,
 | |
| 		newContentRecursive,
 | |
| 		newContentRegExp
 | |
| 	) {
 | |
| 		this.resourceRegExp = resourceRegExp;
 | |
| 
 | |
| 		// new webpack.ContextReplacementPlugin(/selector/, (context) => { /* Logic */ });
 | |
| 		if (typeof newContentResource === "function") {
 | |
| 			this.newContentCallback = newContentResource;
 | |
| 		}
 | |
| 		// new ContextReplacementPlugin(/selector/, './folder', { './request': './request' });
 | |
| 		else if (
 | |
| 			typeof newContentResource === "string" &&
 | |
| 			typeof newContentRecursive === "object"
 | |
| 		) {
 | |
| 			this.newContentResource = newContentResource;
 | |
| 			/**
 | |
| 			 * @param {InputFileSystem} fs input file system
 | |
| 			 * @param {(err: null | Error, newContentRecursive: NewContentCreateContextMap) => void} callback callback
 | |
| 			 */
 | |
| 			this.newContentCreateContextMap = (fs, callback) => {
 | |
| 				callback(
 | |
| 					null,
 | |
| 					/** @type {NewContentCreateContextMap} */ (newContentRecursive)
 | |
| 				);
 | |
| 			};
 | |
| 		}
 | |
| 		// new ContextReplacementPlugin(/selector/, './folder', (context) => { /* Logic */ });
 | |
| 		else if (
 | |
| 			typeof newContentResource === "string" &&
 | |
| 			typeof newContentRecursive === "function"
 | |
| 		) {
 | |
| 			this.newContentResource = newContentResource;
 | |
| 			this.newContentCreateContextMap = newContentRecursive;
 | |
| 		} else {
 | |
| 			// new webpack.ContextReplacementPlugin(/selector/, false, /reg-exp/);
 | |
| 			if (typeof newContentResource !== "string") {
 | |
| 				newContentRegExp = /** @type {RegExp} */ (newContentRecursive);
 | |
| 				newContentRecursive = /** @type {boolean} */ (newContentResource);
 | |
| 				newContentResource = undefined;
 | |
| 			}
 | |
| 			// new webpack.ContextReplacementPlugin(/selector/, /de|fr|hu/);
 | |
| 			if (typeof newContentRecursive !== "boolean") {
 | |
| 				newContentRegExp = /** @type {RegExp} */ (newContentRecursive);
 | |
| 				newContentRecursive = undefined;
 | |
| 			}
 | |
| 			// new webpack.ContextReplacementPlugin(/selector/, './folder', false, /selector/);
 | |
| 			this.newContentResource =
 | |
| 				/** @type {string | undefined} */
 | |
| 				(newContentResource);
 | |
| 			this.newContentRecursive =
 | |
| 				/** @type {boolean | undefined} */
 | |
| 				(newContentRecursive);
 | |
| 			this.newContentRegExp =
 | |
| 				/** @type {RegExp | undefined} */
 | |
| 				(newContentRegExp);
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	/**
 | |
| 	 * Apply the plugin
 | |
| 	 * @param {Compiler} compiler the compiler instance
 | |
| 	 * @returns {void}
 | |
| 	 */
 | |
| 	apply(compiler) {
 | |
| 		const resourceRegExp = this.resourceRegExp;
 | |
| 		const newContentCallback = this.newContentCallback;
 | |
| 		const newContentResource = this.newContentResource;
 | |
| 		const newContentRecursive = this.newContentRecursive;
 | |
| 		const newContentRegExp = this.newContentRegExp;
 | |
| 		const newContentCreateContextMap = this.newContentCreateContextMap;
 | |
| 
 | |
| 		compiler.hooks.contextModuleFactory.tap(PLUGIN_NAME, (cmf) => {
 | |
| 			cmf.hooks.beforeResolve.tap(PLUGIN_NAME, (result) => {
 | |
| 				if (!result) return;
 | |
| 				if (resourceRegExp.test(result.request)) {
 | |
| 					if (newContentResource !== undefined) {
 | |
| 						result.request = newContentResource;
 | |
| 					}
 | |
| 					if (newContentRecursive !== undefined) {
 | |
| 						result.recursive = newContentRecursive;
 | |
| 					}
 | |
| 					if (newContentRegExp !== undefined) {
 | |
| 						result.regExp = newContentRegExp;
 | |
| 					}
 | |
| 					if (typeof newContentCallback === "function") {
 | |
| 						newContentCallback(result);
 | |
| 					} else {
 | |
| 						for (const d of result.dependencies) {
 | |
| 							if (d.critical) d.critical = false;
 | |
| 						}
 | |
| 					}
 | |
| 				}
 | |
| 				return result;
 | |
| 			});
 | |
| 			cmf.hooks.afterResolve.tap(PLUGIN_NAME, (result) => {
 | |
| 				if (!result) return;
 | |
| 				if (resourceRegExp.test(result.resource)) {
 | |
| 					if (newContentResource !== undefined) {
 | |
| 						if (
 | |
| 							newContentResource.startsWith("/") ||
 | |
| 							(newContentResource.length > 1 && newContentResource[1] === ":")
 | |
| 						) {
 | |
| 							result.resource = newContentResource;
 | |
| 						} else {
 | |
| 							result.resource = join(
 | |
| 								/** @type {InputFileSystem} */
 | |
| 								(compiler.inputFileSystem),
 | |
| 								result.resource,
 | |
| 								newContentResource
 | |
| 							);
 | |
| 						}
 | |
| 					}
 | |
| 					if (newContentRecursive !== undefined) {
 | |
| 						result.recursive = newContentRecursive;
 | |
| 					}
 | |
| 					if (newContentRegExp !== undefined) {
 | |
| 						result.regExp = newContentRegExp;
 | |
| 					}
 | |
| 					if (typeof newContentCreateContextMap === "function") {
 | |
| 						result.resolveDependencies =
 | |
| 							createResolveDependenciesFromContextMap(
 | |
| 								newContentCreateContextMap
 | |
| 							);
 | |
| 					}
 | |
| 					if (typeof newContentCallback === "function") {
 | |
| 						const origResource = result.resource;
 | |
| 						newContentCallback(result);
 | |
| 						if (
 | |
| 							result.resource !== origResource &&
 | |
| 							!result.resource.startsWith("/") &&
 | |
| 							(result.resource.length <= 1 || result.resource[1] !== ":")
 | |
| 						) {
 | |
| 							// When the function changed it to an relative path
 | |
| 							result.resource = join(
 | |
| 								/** @type {InputFileSystem} */
 | |
| 								(compiler.inputFileSystem),
 | |
| 								origResource,
 | |
| 								result.resource
 | |
| 							);
 | |
| 						}
 | |
| 					} else {
 | |
| 						for (const d of result.dependencies) {
 | |
| 							if (d.critical) d.critical = false;
 | |
| 						}
 | |
| 					}
 | |
| 				}
 | |
| 				return result;
 | |
| 			});
 | |
| 		});
 | |
| 	}
 | |
| }
 | |
| 
 | |
| /**
 | |
|  * @param {(fs: InputFileSystem, callback: (err: null | Error, map: NewContentCreateContextMap) => void) => void} createContextMap create context map function
 | |
|  * @returns {(fs: InputFileSystem, options: ContextModuleOptions, callback: (err: null | Error, dependencies?: ContextElementDependency[]) => void) => void} resolve resolve dependencies from context map function
 | |
|  */
 | |
| const createResolveDependenciesFromContextMap =
 | |
| 	(createContextMap) => (fs, options, callback) => {
 | |
| 		createContextMap(fs, (err, map) => {
 | |
| 			if (err) return callback(err);
 | |
| 			const dependencies = Object.keys(map).map(
 | |
| 				(key) =>
 | |
| 					new ContextElementDependency(
 | |
| 						map[key] + options.resourceQuery + options.resourceFragment,
 | |
| 						key,
 | |
| 						options.typePrefix,
 | |
| 						/** @type {string} */
 | |
| 						(options.category),
 | |
| 						options.referencedExports
 | |
| 					)
 | |
| 			);
 | |
| 			callback(null, dependencies);
 | |
| 		});
 | |
| 	};
 | |
| 
 | |
| module.exports = ContextReplacementPlugin;
 |