You cannot select more than 25 topics
			Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
		
		
		
		
		
			
		
			
	
	
		
			149 lines
		
	
	
		
			4.4 KiB
		
	
	
	
		
			JavaScript
		
	
		
		
			
		
	
	
			149 lines
		
	
	
		
			4.4 KiB
		
	
	
	
		
			JavaScript
		
	
| 
											9 months ago
										 | /* | ||
|  | 	MIT License http://www.opensource.org/licenses/mit-license.php
 | ||
|  | 	Author Ivan Kopeykin @vankop | ||
|  | */ | ||
|  | 
 | ||
|  | "use strict"; | ||
|  | 
 | ||
|  | const { pathToFileURL } = require("url"); | ||
|  | const { | ||
|  | 	JAVASCRIPT_MODULE_TYPE_AUTO, | ||
|  | 	JAVASCRIPT_MODULE_TYPE_ESM | ||
|  | } = require("../ModuleTypeConstants"); | ||
|  | const BasicEvaluatedExpression = require("../javascript/BasicEvaluatedExpression"); | ||
|  | const { approve } = require("../javascript/JavascriptParserHelpers"); | ||
|  | const InnerGraph = require("../optimize/InnerGraph"); | ||
|  | const URLDependency = require("./URLDependency"); | ||
|  | 
 | ||
|  | /** @typedef {import("estree").NewExpression} NewExpressionNode */ | ||
|  | /** @typedef {import("../../declarations/WebpackOptions").JavascriptParserOptions} JavascriptParserOptions */ | ||
|  | /** @typedef {import("../Compiler")} Compiler */ | ||
|  | /** @typedef {import("../Dependency").DependencyLocation} DependencyLocation */ | ||
|  | /** @typedef {import("../NormalModule")} NormalModule */ | ||
|  | /** @typedef {import("../javascript/JavascriptParser")} JavascriptParser */ | ||
|  | /** @typedef {import("../javascript/JavascriptParser")} Parser */ | ||
|  | /** @typedef {import("../javascript/JavascriptParser").Range} Range */ | ||
|  | 
 | ||
|  | const PLUGIN_NAME = "URLPlugin"; | ||
|  | 
 | ||
|  | class URLPlugin { | ||
|  | 	/** | ||
|  | 	 * @param {Compiler} compiler compiler | ||
|  | 	 */ | ||
|  | 	apply(compiler) { | ||
|  | 		compiler.hooks.compilation.tap( | ||
|  | 			PLUGIN_NAME, | ||
|  | 			(compilation, { normalModuleFactory }) => { | ||
|  | 				compilation.dependencyFactories.set(URLDependency, normalModuleFactory); | ||
|  | 				compilation.dependencyTemplates.set( | ||
|  | 					URLDependency, | ||
|  | 					new URLDependency.Template() | ||
|  | 				); | ||
|  | 
 | ||
|  | 				/** | ||
|  | 				 * @param {NormalModule} module module | ||
|  | 				 * @returns {URL} file url | ||
|  | 				 */ | ||
|  | 				const getUrl = module => { | ||
|  | 					return pathToFileURL(module.resource); | ||
|  | 				}; | ||
|  | 
 | ||
|  | 				/** | ||
|  | 				 * @param {Parser} parser parser parser | ||
|  | 				 * @param {JavascriptParserOptions} parserOptions parserOptions | ||
|  | 				 * @returns {void} | ||
|  | 				 */ | ||
|  | 				const parserCallback = (parser, parserOptions) => { | ||
|  | 					if (parserOptions.url === false) return; | ||
|  | 					const relative = parserOptions.url === "relative"; | ||
|  | 
 | ||
|  | 					/** | ||
|  | 					 * @param {NewExpressionNode} expr expression | ||
|  | 					 * @returns {undefined | string} request | ||
|  | 					 */ | ||
|  | 					const getUrlRequest = expr => { | ||
|  | 						if (expr.arguments.length !== 2) return; | ||
|  | 
 | ||
|  | 						const [arg1, arg2] = expr.arguments; | ||
|  | 
 | ||
|  | 						if ( | ||
|  | 							arg2.type !== "MemberExpression" || | ||
|  | 							arg1.type === "SpreadElement" | ||
|  | 						) | ||
|  | 							return; | ||
|  | 
 | ||
|  | 						const chain = parser.extractMemberExpressionChain(arg2); | ||
|  | 
 | ||
|  | 						if ( | ||
|  | 							chain.members.length !== 1 || | ||
|  | 							chain.object.type !== "MetaProperty" || | ||
|  | 							chain.object.meta.name !== "import" || | ||
|  | 							chain.object.property.name !== "meta" || | ||
|  | 							chain.members[0] !== "url" | ||
|  | 						) | ||
|  | 							return; | ||
|  | 
 | ||
|  | 						return parser.evaluateExpression(arg1).asString(); | ||
|  | 					}; | ||
|  | 
 | ||
|  | 					parser.hooks.canRename.for("URL").tap(PLUGIN_NAME, approve); | ||
|  | 					parser.hooks.evaluateNewExpression | ||
|  | 						.for("URL") | ||
|  | 						.tap(PLUGIN_NAME, expr => { | ||
|  | 							const request = getUrlRequest(expr); | ||
|  | 							if (!request) return; | ||
|  | 							const url = new URL(request, getUrl(parser.state.module)); | ||
|  | 
 | ||
|  | 							return new BasicEvaluatedExpression() | ||
|  | 								.setString(url.toString()) | ||
|  | 								.setRange(/** @type {Range} */ (expr.range)); | ||
|  | 						}); | ||
|  | 					parser.hooks.new.for("URL").tap(PLUGIN_NAME, _expr => { | ||
|  | 						const expr = /** @type {NewExpressionNode} */ (_expr); | ||
|  | 
 | ||
|  | 						const request = getUrlRequest(expr); | ||
|  | 
 | ||
|  | 						if (!request) return; | ||
|  | 
 | ||
|  | 						const [arg1, arg2] = expr.arguments; | ||
|  | 						const dep = new URLDependency( | ||
|  | 							request, | ||
|  | 							[ | ||
|  | 								/** @type {Range} */ (arg1.range)[0], | ||
|  | 								/** @type {Range} */ (arg2.range)[1] | ||
|  | 							], | ||
|  | 							/** @type {Range} */ (expr.range), | ||
|  | 							relative | ||
|  | 						); | ||
|  | 						dep.loc = /** @type {DependencyLocation} */ (expr.loc); | ||
|  | 						parser.state.current.addDependency(dep); | ||
|  | 						InnerGraph.onUsage(parser.state, e => (dep.usedByExports = e)); | ||
|  | 						return true; | ||
|  | 					}); | ||
|  | 					parser.hooks.isPure.for("NewExpression").tap(PLUGIN_NAME, _expr => { | ||
|  | 						const expr = /** @type {NewExpressionNode} */ (_expr); | ||
|  | 						const { callee } = expr; | ||
|  | 						if (callee.type !== "Identifier") return; | ||
|  | 						const calleeInfo = parser.getFreeInfoFromVariable(callee.name); | ||
|  | 						if (!calleeInfo || calleeInfo.name !== "URL") return; | ||
|  | 
 | ||
|  | 						const request = getUrlRequest(expr); | ||
|  | 
 | ||
|  | 						if (request) return true; | ||
|  | 					}); | ||
|  | 				}; | ||
|  | 
 | ||
|  | 				normalModuleFactory.hooks.parser | ||
|  | 					.for(JAVASCRIPT_MODULE_TYPE_AUTO) | ||
|  | 					.tap(PLUGIN_NAME, parserCallback); | ||
|  | 
 | ||
|  | 				normalModuleFactory.hooks.parser | ||
|  | 					.for(JAVASCRIPT_MODULE_TYPE_ESM) | ||
|  | 					.tap(PLUGIN_NAME, parserCallback); | ||
|  | 			} | ||
|  | 		); | ||
|  | 	} | ||
|  | } | ||
|  | 
 | ||
|  | module.exports = URLPlugin; |