webpack/lib/web/FetchCompileWasmMainTemplatePlugin.js

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

const Template = require("../Template");

class FetchCompileWasmMainTemplatePlugin {

	apply(mainTemplate) {
		mainTemplate.hooks.localVars.tap("FetchCompileWasmMainTemplatePlugin", (source, chunk) => {
			if(!chunk.hasModuleInGraph(m => m.type.startsWith("webassembly")))
				return source;
			return Template.asString([
				source,
				"",
				"// object to store loaded and loading wasm modules",
				"var installedWasmModules = {};",
			]);
		});
		mainTemplate.hooks.requireEnsure.tap("FetchCompileWasmMainTemplatePlugin", (source, chunk, hash) => {
			const webassemblyModuleFilename = mainTemplate.outputOptions.webassemblyModuleFilename;
			const chunkModuleMaps = chunk.getChunkModuleMaps(false, m => m.type.startsWith("webassembly"));
			if(Object.keys(chunkModuleMaps.id).length === 0) return source;
			const wasmModuleSrcPath = mainTemplate.getAssetPath(JSON.stringify(webassemblyModuleFilename), {
				hash: `" + ${mainTemplate.renderCurrentHashCode(hash)} + "`,
				hashWithLength: length => `" + ${mainTemplate.renderCurrentHashCode(hash, length)} + "`,
				module: {
					id: "\" + wasmModuleId + \"",
					hash: `" + ${JSON.stringify(chunkModuleMaps.hash)}[wasmModuleId] + "`,
					hashWithLength(length) {
						const shortChunkHashMap = Object.create(null);
						Object.keys(chunkModuleMaps.hash).forEach(wasmModuleId => {
							if(typeof chunkModuleMaps.hash[wasmModuleId] === "string")
								shortChunkHashMap[wasmModuleId] = chunkModuleMaps.hash[wasmModuleId].substr(0, length);
						});
						return `" + ${JSON.stringify(shortChunkHashMap)}[wasmModuleId] + "`;
					}
				}
			});
			return Template.asString([
				source,
				"",
				"// Fetch + compile chunk loading for webassembly",
				"",
				`var wasmModules = ${JSON.stringify(chunkModuleMaps.id)}[chunkId] || [];`,
				"",
				"wasmModules.forEach(function(wasmModuleId) {",
				Template.indent([
					"var installedWasmModuleData = installedWasmModules[wasmModuleId];",
					"",
					"// a Promise means \"currently loading\" or \"already loaded\".",
					"promises.push(installedWasmModuleData ||",
					Template.indent([
						`(installedWasmModules[wasmModuleId] = fetch(${mainTemplate.requireFn}.p + ${wasmModuleSrcPath}).then(function(response) {`,
						Template.indent([
							"if(WebAssembly.compileStreaming) {",
							Template.indent([
								"return WebAssembly.compileStreaming(response);"
							]),
							"} else {",
							Template.indent([
								"return response.arrayBuffer().then(function(bytes) { return WebAssembly.compile(bytes); });",
							]),
							"}"
						]),
						`}).then(function(module) { ${mainTemplate.requireFn}.w[wasmModuleId] = module; }))`
					]),
					");",
				]),
				"});",
			]);
		});
		mainTemplate.hooks.requireExtensions.tap("FetchCompileWasmMainTemplatePlugin", (source, chunk) => {
			if(!chunk.hasModuleInGraph(m => m.type.startsWith("webassembly")))
				return source;
			return Template.asString([
				source,
				"",
				"// object with all compiled WebAssmbly.Modules",
				`${mainTemplate.requireFn}.w = {};`
			]);
		});
		mainTemplate.hooks.hash.tap("FetchCompileWasmMainTemplatePlugin", hash => {
			hash.update("FetchCompileWasmMainTemplatePlugin");
			hash.update("1");
			hash.update(`${mainTemplate.outputOptions.webassemblyModuleFilename}`);
		});
	}
}
module.exports = FetchCompileWasmMainTemplatePlugin;