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
/*
    MIT License http://www.opensource.org/licenses/mit-license.php
    Author Tobias Koppers @sokra
*/
 
"use strict";
 
const HarmonyImportDependency = require("../dependencies/HarmonyImportDependency");
 
/** @typedef {import("../Compiler")} Compiler */
/** @typedef {import("../Module")} Module */
 
class InferAsyncModulesPlugin {
    /**
     * Apply the plugin
     * @param {Compiler} compiler the compiler instance
     * @returns {void}
     */
    apply(compiler) {
        compiler.hooks.compilation.tap("InferAsyncModulesPlugin", compilation => {
            const { moduleGraph } = compilation;
            compilation.hooks.finishModules.tap(
                "InferAsyncModulesPlugin",
                modules => {
                    /** @type {Set<Module>} */
                    const queue = new Set();
                    for (const module of modules) {
                        if (module.buildMeta && module.buildMeta.async) {
                            queue.add(module);
                        }
                    }
                    for (const module of queue) {
                        moduleGraph.setAsync(module);
                        for (const [
                            originModule,
                            connections
                        ] of moduleGraph.getIncomingConnectionsByOriginModule(module)) {
                            if (
                                connections.some(
                                    c =>
                                        c.dependency instanceof HarmonyImportDependency &&
                                        c.isTargetActive(undefined)
                                )
                            ) {
                                queue.add(/** @type {Module} */ (originModule));
                            }
                        }
                    }
                }
            );
        });
    }
}
 
module.exports = InferAsyncModulesPlugin;