mirror of https://github.com/webpack/webpack.git
				
				
				
			
		
			
				
	
	
		
			143 lines
		
	
	
		
			4.1 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
			
		
		
	
	
			143 lines
		
	
	
		
			4.1 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
/*
 | 
						|
	MIT License http://www.opensource.org/licenses/mit-license.php
 | 
						|
	Author Tobias Koppers @sokra
 | 
						|
*/
 | 
						|
 | 
						|
"use strict";
 | 
						|
 | 
						|
const Generator = require("../Generator");
 | 
						|
const WebAssemblyExportImportedDependency = require("../dependencies/WebAssemblyExportImportedDependency");
 | 
						|
const WebAssemblyImportDependency = require("../dependencies/WebAssemblyImportDependency");
 | 
						|
const { compareModulesByIdentifier } = require("../util/comparators");
 | 
						|
const memoize = require("../util/memoize");
 | 
						|
const WebAssemblyInInitialChunkError = require("./WebAssemblyInInitialChunkError");
 | 
						|
 | 
						|
/** @typedef {import("webpack-sources").Source} Source */
 | 
						|
/** @typedef {import("../Compiler")} Compiler */
 | 
						|
/** @typedef {import("../Module")} Module */
 | 
						|
/** @typedef {import("../ModuleTemplate")} ModuleTemplate */
 | 
						|
/** @typedef {import("../javascript/JavascriptModulesPlugin").RenderContext} RenderContext */
 | 
						|
 | 
						|
const getWebAssemblyGenerator = memoize(() =>
 | 
						|
	require("./WebAssemblyGenerator")
 | 
						|
);
 | 
						|
const getWebAssemblyJavascriptGenerator = memoize(() =>
 | 
						|
	require("./WebAssemblyJavascriptGenerator")
 | 
						|
);
 | 
						|
const getWebAssemblyParser = memoize(() => require("./WebAssemblyParser"));
 | 
						|
 | 
						|
class WebAssemblyModulesPlugin {
 | 
						|
	constructor(options) {
 | 
						|
		this.options = options;
 | 
						|
	}
 | 
						|
 | 
						|
	/**
 | 
						|
	 * Apply the plugin
 | 
						|
	 * @param {Compiler} compiler the compiler instance
 | 
						|
	 * @returns {void}
 | 
						|
	 */
 | 
						|
	apply(compiler) {
 | 
						|
		compiler.hooks.compilation.tap(
 | 
						|
			"WebAssemblyModulesPlugin",
 | 
						|
			(compilation, { normalModuleFactory }) => {
 | 
						|
				compilation.dependencyFactories.set(
 | 
						|
					WebAssemblyImportDependency,
 | 
						|
					normalModuleFactory
 | 
						|
				);
 | 
						|
 | 
						|
				compilation.dependencyFactories.set(
 | 
						|
					WebAssemblyExportImportedDependency,
 | 
						|
					normalModuleFactory
 | 
						|
				);
 | 
						|
 | 
						|
				normalModuleFactory.hooks.createParser
 | 
						|
					.for("webassembly/sync")
 | 
						|
					.tap("WebAssemblyModulesPlugin", () => {
 | 
						|
						const WebAssemblyParser = getWebAssemblyParser();
 | 
						|
 | 
						|
						return new WebAssemblyParser();
 | 
						|
					});
 | 
						|
 | 
						|
				normalModuleFactory.hooks.createGenerator
 | 
						|
					.for("webassembly/sync")
 | 
						|
					.tap("WebAssemblyModulesPlugin", () => {
 | 
						|
						const WebAssemblyJavascriptGenerator =
 | 
						|
							getWebAssemblyJavascriptGenerator();
 | 
						|
						const WebAssemblyGenerator = getWebAssemblyGenerator();
 | 
						|
 | 
						|
						return Generator.byType({
 | 
						|
							javascript: new WebAssemblyJavascriptGenerator(),
 | 
						|
							webassembly: new WebAssemblyGenerator(this.options)
 | 
						|
						});
 | 
						|
					});
 | 
						|
 | 
						|
				compilation.hooks.renderManifest.tap(
 | 
						|
					"WebAssemblyModulesPlugin",
 | 
						|
					(result, options) => {
 | 
						|
						const { chunkGraph } = compilation;
 | 
						|
						const { chunk, outputOptions, codeGenerationResults } = options;
 | 
						|
 | 
						|
						for (const module of chunkGraph.getOrderedChunkModulesIterable(
 | 
						|
							chunk,
 | 
						|
							compareModulesByIdentifier
 | 
						|
						)) {
 | 
						|
							if (module.type === "webassembly/sync") {
 | 
						|
								const filenameTemplate =
 | 
						|
									outputOptions.webassemblyModuleFilename;
 | 
						|
 | 
						|
								result.push({
 | 
						|
									render: () =>
 | 
						|
										codeGenerationResults.getSource(
 | 
						|
											module,
 | 
						|
											chunk.runtime,
 | 
						|
											"webassembly"
 | 
						|
										),
 | 
						|
									filenameTemplate,
 | 
						|
									pathOptions: {
 | 
						|
										module,
 | 
						|
										runtime: chunk.runtime,
 | 
						|
										chunkGraph
 | 
						|
									},
 | 
						|
									auxiliary: true,
 | 
						|
									identifier: `webassemblyModule${chunkGraph.getModuleId(
 | 
						|
										module
 | 
						|
									)}`,
 | 
						|
									hash: chunkGraph.getModuleHash(module, chunk.runtime)
 | 
						|
								});
 | 
						|
							}
 | 
						|
						}
 | 
						|
 | 
						|
						return result;
 | 
						|
					}
 | 
						|
				);
 | 
						|
 | 
						|
				compilation.hooks.afterChunks.tap("WebAssemblyModulesPlugin", () => {
 | 
						|
					const chunkGraph = compilation.chunkGraph;
 | 
						|
					const initialWasmModules = new Set();
 | 
						|
					for (const chunk of compilation.chunks) {
 | 
						|
						if (chunk.canBeInitial()) {
 | 
						|
							for (const module of chunkGraph.getChunkModulesIterable(chunk)) {
 | 
						|
								if (module.type === "webassembly/sync") {
 | 
						|
									initialWasmModules.add(module);
 | 
						|
								}
 | 
						|
							}
 | 
						|
						}
 | 
						|
					}
 | 
						|
					for (const module of initialWasmModules) {
 | 
						|
						compilation.errors.push(
 | 
						|
							new WebAssemblyInInitialChunkError(
 | 
						|
								module,
 | 
						|
								compilation.moduleGraph,
 | 
						|
								compilation.chunkGraph,
 | 
						|
								compilation.requestShortener
 | 
						|
							)
 | 
						|
						);
 | 
						|
					}
 | 
						|
				});
 | 
						|
			}
 | 
						|
		);
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
module.exports = WebAssemblyModulesPlugin;
 |