马宇豪
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
/*
    MIT License http://www.opensource.org/licenses/mit-license.php
    Author Tobias Koppers @sokra
*/
 
"use strict";
 
const { join, dirname } = require("./util/fs");
 
/** @typedef {import("./Compiler")} Compiler */
/** @typedef {function(import("./NormalModuleFactory").ResolveData): void} ModuleReplacer */
 
class NormalModuleReplacementPlugin {
    /**
     * Create an instance of the plugin
     * @param {RegExp} resourceRegExp the resource matcher
     * @param {string|ModuleReplacer} newResource the resource replacement
     */
    constructor(resourceRegExp, newResource) {
        this.resourceRegExp = resourceRegExp;
        this.newResource = newResource;
    }
 
    /**
     * Apply the plugin
     * @param {Compiler} compiler the compiler instance
     * @returns {void}
     */
    apply(compiler) {
        const resourceRegExp = this.resourceRegExp;
        const newResource = this.newResource;
        compiler.hooks.normalModuleFactory.tap(
            "NormalModuleReplacementPlugin",
            nmf => {
                nmf.hooks.beforeResolve.tap("NormalModuleReplacementPlugin", result => {
                    if (resourceRegExp.test(result.request)) {
                        if (typeof newResource === "function") {
                            newResource(result);
                        } else {
                            result.request = newResource;
                        }
                    }
                });
                nmf.hooks.afterResolve.tap("NormalModuleReplacementPlugin", result => {
                    const createData = result.createData;
                    if (
                        resourceRegExp.test(/** @type {string} */ (createData.resource))
                    ) {
                        if (typeof newResource === "function") {
                            newResource(result);
                        } else {
                            const fs = compiler.inputFileSystem;
                            if (
                                newResource.startsWith("/") ||
                                (newResource.length > 1 && newResource[1] === ":")
                            ) {
                                createData.resource = newResource;
                            } else {
                                createData.resource = join(
                                    fs,
                                    dirname(fs, /** @type {string} */ (createData.resource)),
                                    newResource
                                );
                            }
                        }
                    }
                });
            }
        );
    }
}
 
module.exports = NormalModuleReplacementPlugin;