mirror of https://github.com/webpack/webpack.git
120 lines
3.6 KiB
JavaScript
120 lines
3.6 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 WebAssembly.Modules",
|
|
`${mainTemplate.requireFn}.w = {};`
|
|
]);
|
|
}
|
|
);
|
|
mainTemplate.hooks.hash.tap("FetchCompileWasmMainTemplatePlugin", hash => {
|
|
hash.update("FetchCompileWasmMainTemplatePlugin");
|
|
hash.update("1");
|
|
hash.update(`${mainTemplate.outputOptions.webassemblyModuleFilename}`);
|
|
});
|
|
}
|
|
}
|
|
module.exports = FetchCompileWasmMainTemplatePlugin;
|