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
/*
    MIT License http://www.opensource.org/licenses/mit-license.php
    Author Tobias Koppers @sokra
*/
 
"use strict";
 
const asyncLib = require("neo-async");
const NormalModule = require("./NormalModule");
const PrefetchDependency = require("./dependencies/PrefetchDependency");
 
/** @typedef {import("./Compiler")} Compiler */
 
class AutomaticPrefetchPlugin {
    /**
     * Apply the plugin
     * @param {Compiler} compiler the compiler instance
     * @returns {void}
     */
    apply(compiler) {
        compiler.hooks.compilation.tap(
            "AutomaticPrefetchPlugin",
            (compilation, { normalModuleFactory }) => {
                compilation.dependencyFactories.set(
                    PrefetchDependency,
                    normalModuleFactory
                );
            }
        );
        /** @type {{context: string, request: string}[] | null} */
        let lastModules = null;
        compiler.hooks.afterCompile.tap("AutomaticPrefetchPlugin", compilation => {
            lastModules = [];
 
            for (const m of compilation.modules) {
                if (m instanceof NormalModule) {
                    lastModules.push({
                        context: m.context,
                        request: m.request
                    });
                }
            }
        });
        compiler.hooks.make.tapAsync(
            "AutomaticPrefetchPlugin",
            (compilation, callback) => {
                if (!lastModules) return callback();
                asyncLib.forEach(
                    lastModules,
                    (m, callback) => {
                        compilation.addModuleChain(
                            m.context || compiler.context,
                            new PrefetchDependency(`!!${m.request}`),
                            callback
                        );
                    },
                    err => {
                        lastModules = null;
                        callback(err);
                    }
                );
            }
        );
    }
}
module.exports = AutomaticPrefetchPlugin;