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
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
/*
    MIT License http://www.opensource.org/licenses/mit-license.php
    Author Tobias Koppers @sokra
*/
 
"use strict";
 
/** @typedef {import("../../declarations/WebpackOptions").LibraryOptions} LibraryOptions */
/** @typedef {import("../../declarations/WebpackOptions").WasmLoadingType} WasmLoadingType */
/** @typedef {import("../Compiler")} Compiler */
 
/** @type {WeakMap<Compiler, Set<WasmLoadingType>>} */
const enabledTypes = new WeakMap();
 
/**
 * @param {Compiler} compiler compiler instance
 * @returns {Set<WasmLoadingType>} enabled types
 */
const getEnabledTypes = compiler => {
    let set = enabledTypes.get(compiler);
    if (set === undefined) {
        set = new Set();
        enabledTypes.set(compiler, set);
    }
    return set;
};
 
class EnableWasmLoadingPlugin {
    /**
     * @param {WasmLoadingType} type library type that should be available
     */
    constructor(type) {
        this.type = type;
    }
 
    /**
     * @param {Compiler} compiler the compiler instance
     * @param {WasmLoadingType} type type of library
     * @returns {void}
     */
    static setEnabled(compiler, type) {
        getEnabledTypes(compiler).add(type);
    }
 
    /**
     * @param {Compiler} compiler the compiler instance
     * @param {WasmLoadingType} type type of library
     * @returns {void}
     */
    static checkEnabled(compiler, type) {
        if (!getEnabledTypes(compiler).has(type)) {
            throw new Error(
                `Library type "${type}" is not enabled. ` +
                    "EnableWasmLoadingPlugin need to be used to enable this type of wasm loading. " +
                    'This usually happens through the "output.enabledWasmLoadingTypes" option. ' +
                    'If you are using a function as entry which sets "wasmLoading", you need to add all potential library types to "output.enabledWasmLoadingTypes". ' +
                    "These types are enabled: " +
                    Array.from(getEnabledTypes(compiler)).join(", ")
            );
        }
    }
 
    /**
     * Apply the plugin
     * @param {Compiler} compiler the compiler instance
     * @returns {void}
     */
    apply(compiler) {
        const { type } = this;
 
        // Only enable once
        const enabled = getEnabledTypes(compiler);
        if (enabled.has(type)) return;
        enabled.add(type);
 
        if (typeof type === "string") {
            switch (type) {
                case "fetch": {
                    // TODO webpack 6 remove FetchCompileWasmPlugin
                    const FetchCompileWasmPlugin = require("../web/FetchCompileWasmPlugin");
                    const FetchCompileAsyncWasmPlugin = require("../web/FetchCompileAsyncWasmPlugin");
                    new FetchCompileWasmPlugin({
                        mangleImports: compiler.options.optimization.mangleWasmImports
                    }).apply(compiler);
                    new FetchCompileAsyncWasmPlugin().apply(compiler);
                    break;
                }
                case "async-node": {
                    // TODO webpack 6 remove ReadFileCompileWasmPlugin
                    const ReadFileCompileWasmPlugin = require("../node/ReadFileCompileWasmPlugin");
                    // @ts-expect-error typescript bug for duplicate require
                    const ReadFileCompileAsyncWasmPlugin = require("../node/ReadFileCompileAsyncWasmPlugin");
                    new ReadFileCompileWasmPlugin({
                        mangleImports: compiler.options.optimization.mangleWasmImports
                    }).apply(compiler);
                    new ReadFileCompileAsyncWasmPlugin({ type }).apply(compiler);
                    break;
                }
                case "async-node-module": {
                    // @ts-expect-error typescript bug for duplicate require
                    const ReadFileCompileAsyncWasmPlugin = require("../node/ReadFileCompileAsyncWasmPlugin");
                    new ReadFileCompileAsyncWasmPlugin({ type, import: true }).apply(
                        compiler
                    );
                    break;
                }
                case "universal":
                    throw new Error(
                        "Universal WebAssembly Loading is not implemented yet"
                    );
                default:
                    throw new Error(`Unsupported wasm loading type ${type}.
Plugins which provide custom wasm loading types must call EnableWasmLoadingPlugin.setEnabled(compiler, type) to disable this error.`);
            }
        } else {
            // TODO support plugin instances here
            // apply them to the compiler
        }
    }
}
 
module.exports = EnableWasmLoadingPlugin;