mirror of https://github.com/webpack/webpack.git
93 lines
3.4 KiB
JavaScript
93 lines
3.4 KiB
JavaScript
/*
|
|
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(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);
|
|
for(const wasmModuleId of Object.keys(chunkModuleMaps.hash)) {
|
|
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;
|