init

马宇豪

2024-07-16 f591c27b57e2418c9495bc02ae8cfff84d35bc18

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
/*
    MIT License http://www.opensource.org/licenses/mit-license.php
    Author Tobias Koppers @sokra
*/
 
"use strict";
 
const RuntimeGlobals = require("../RuntimeGlobals");
const RuntimeModule = require("../RuntimeModule");
const Template = require("../Template");
 
/**
 * @typedef {Object} AsyncWasmLoadingRuntimeModuleOptions
 * @property {function(string): string} generateLoadBinaryCode
 * @property {boolean} supportsStreaming
 */
 
class AsyncWasmLoadingRuntimeModule extends RuntimeModule {
    /**
     * @param {AsyncWasmLoadingRuntimeModuleOptions} options options
     */
    constructor({ generateLoadBinaryCode, supportsStreaming }) {
        super("wasm loading", RuntimeModule.STAGE_NORMAL);
        this.generateLoadBinaryCode = generateLoadBinaryCode;
        this.supportsStreaming = supportsStreaming;
    }
 
    /**
     * @returns {string} runtime code
     */
    generate() {
        const { compilation, chunk } = this;
        const { outputOptions, runtimeTemplate } = compilation;
        const fn = RuntimeGlobals.instantiateWasm;
        const wasmModuleSrcPath = compilation.getPath(
            JSON.stringify(outputOptions.webassemblyModuleFilename),
            {
                hash: `" + ${RuntimeGlobals.getFullHash}() + "`,
                hashWithLength: length =>
                    `" + ${RuntimeGlobals.getFullHash}}().slice(0, ${length}) + "`,
                module: {
                    id: '" + wasmModuleId + "',
                    hash: `" + wasmModuleHash + "`,
                    hashWithLength(length) {
                        return `" + wasmModuleHash.slice(0, ${length}) + "`;
                    }
                },
                runtime: chunk.runtime
            }
        );
        return `${fn} = ${runtimeTemplate.basicFunction(
            "exports, wasmModuleId, wasmModuleHash, importsObj",
            [
                `var req = ${this.generateLoadBinaryCode(wasmModuleSrcPath)};`,
                this.supportsStreaming
                    ? Template.asString([
                            "if (typeof WebAssembly.instantiateStreaming === 'function') {",
                            Template.indent([
                                "return WebAssembly.instantiateStreaming(req, importsObj)",
                                Template.indent([
                                    `.then(${runtimeTemplate.returningFunction(
                                        "Object.assign(exports, res.instance.exports)",
                                        "res"
                                    )});`
                                ])
                            ]),
                            "}"
                      ])
                    : "// no support for streaming compilation",
                "return req",
                Template.indent([
                    `.then(${runtimeTemplate.returningFunction("x.arrayBuffer()", "x")})`,
                    `.then(${runtimeTemplate.returningFunction(
                        "WebAssembly.instantiate(bytes, importsObj)",
                        "bytes"
                    )})`,
                    `.then(${runtimeTemplate.returningFunction(
                        "Object.assign(exports, res.instance.exports)",
                        "res"
                    )});`
                ])
            ]
        )};`;
    }
}
 
module.exports = AsyncWasmLoadingRuntimeModule;