[6a3a178] | 1 | /*
|
---|
| 2 | MIT License http://www.opensource.org/licenses/mit-license.php
|
---|
| 3 | Author Tobias Koppers @sokra
|
---|
| 4 | */
|
---|
| 5 |
|
---|
| 6 | "use strict";
|
---|
| 7 |
|
---|
| 8 | const RuntimeGlobals = require("../RuntimeGlobals");
|
---|
| 9 | const Template = require("../Template");
|
---|
| 10 | const AsyncWasmChunkLoadingRuntimeModule = require("../wasm-async/AsyncWasmChunkLoadingRuntimeModule");
|
---|
| 11 |
|
---|
| 12 | /** @typedef {import("../Compiler")} Compiler */
|
---|
| 13 |
|
---|
| 14 | class ReadFileCompileAsyncWasmPlugin {
|
---|
| 15 | constructor({ type = "async-node", import: useImport = false } = {}) {
|
---|
| 16 | this._type = type;
|
---|
| 17 | this._import = useImport;
|
---|
| 18 | }
|
---|
| 19 | /**
|
---|
| 20 | * Apply the plugin
|
---|
| 21 | * @param {Compiler} compiler the compiler instance
|
---|
| 22 | * @returns {void}
|
---|
| 23 | */
|
---|
| 24 | apply(compiler) {
|
---|
| 25 | compiler.hooks.thisCompilation.tap(
|
---|
| 26 | "ReadFileCompileAsyncWasmPlugin",
|
---|
| 27 | compilation => {
|
---|
| 28 | const globalWasmLoading = compilation.outputOptions.wasmLoading;
|
---|
| 29 | const isEnabledForChunk = chunk => {
|
---|
| 30 | const options = chunk.getEntryOptions();
|
---|
| 31 | const wasmLoading =
|
---|
| 32 | options && options.wasmLoading !== undefined
|
---|
| 33 | ? options.wasmLoading
|
---|
| 34 | : globalWasmLoading;
|
---|
| 35 | return wasmLoading === this._type;
|
---|
| 36 | };
|
---|
| 37 | const generateLoadBinaryCode = this._import
|
---|
| 38 | ? path =>
|
---|
| 39 | Template.asString([
|
---|
| 40 | "Promise.all([import('fs'), import('url')]).then(([{ readFile }, { URL }]) => new Promise((resolve, reject) => {",
|
---|
| 41 | Template.indent([
|
---|
| 42 | `readFile(new URL(${path}, import.meta.url), (err, buffer) => {`,
|
---|
| 43 | Template.indent([
|
---|
| 44 | "if (err) return reject(err);",
|
---|
| 45 | "",
|
---|
| 46 | "// Fake fetch response",
|
---|
| 47 | "resolve({",
|
---|
| 48 | Template.indent(["arrayBuffer() { return buffer; }"]),
|
---|
| 49 | "});"
|
---|
| 50 | ]),
|
---|
| 51 | "});"
|
---|
| 52 | ]),
|
---|
| 53 | "}))"
|
---|
| 54 | ])
|
---|
| 55 | : path =>
|
---|
| 56 | Template.asString([
|
---|
| 57 | "new Promise(function (resolve, reject) {",
|
---|
| 58 | Template.indent([
|
---|
| 59 | "try {",
|
---|
| 60 | Template.indent([
|
---|
| 61 | "var { readFile } = require('fs');",
|
---|
| 62 | "var { join } = require('path');",
|
---|
| 63 | "",
|
---|
| 64 | `readFile(join(__dirname, ${path}), function(err, buffer){`,
|
---|
| 65 | Template.indent([
|
---|
| 66 | "if (err) return reject(err);",
|
---|
| 67 | "",
|
---|
| 68 | "// Fake fetch response",
|
---|
| 69 | "resolve({",
|
---|
| 70 | Template.indent(["arrayBuffer() { return buffer; }"]),
|
---|
| 71 | "});"
|
---|
| 72 | ]),
|
---|
| 73 | "});"
|
---|
| 74 | ]),
|
---|
| 75 | "} catch (err) { reject(err); }"
|
---|
| 76 | ]),
|
---|
| 77 | "})"
|
---|
| 78 | ]);
|
---|
| 79 |
|
---|
| 80 | compilation.hooks.runtimeRequirementInTree
|
---|
| 81 | .for(RuntimeGlobals.instantiateWasm)
|
---|
| 82 | .tap("ReadFileCompileAsyncWasmPlugin", (chunk, set) => {
|
---|
| 83 | if (!isEnabledForChunk(chunk)) return;
|
---|
| 84 | const chunkGraph = compilation.chunkGraph;
|
---|
| 85 | if (
|
---|
| 86 | !chunkGraph.hasModuleInGraph(
|
---|
| 87 | chunk,
|
---|
| 88 | m => m.type === "webassembly/async"
|
---|
| 89 | )
|
---|
| 90 | ) {
|
---|
| 91 | return;
|
---|
| 92 | }
|
---|
| 93 | set.add(RuntimeGlobals.publicPath);
|
---|
| 94 | compilation.addRuntimeModule(
|
---|
| 95 | chunk,
|
---|
| 96 | new AsyncWasmChunkLoadingRuntimeModule({
|
---|
| 97 | generateLoadBinaryCode,
|
---|
| 98 | supportsStreaming: false
|
---|
| 99 | })
|
---|
| 100 | );
|
---|
| 101 | });
|
---|
| 102 | }
|
---|
| 103 | );
|
---|
| 104 | }
|
---|
| 105 | }
|
---|
| 106 |
|
---|
| 107 | module.exports = ReadFileCompileAsyncWasmPlugin;
|
---|