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
/*
    MIT License http://www.opensource.org/licenses/mit-license.php
    Authors Simen Brekken @simenbrekken, Einar Löve @einarlove
*/
 
"use strict";
 
const DefinePlugin = require("./DefinePlugin");
const WebpackError = require("./WebpackError");
 
/** @typedef {import("./Compiler")} Compiler */
/** @typedef {import("./DefinePlugin").CodeValue} CodeValue */
 
class EnvironmentPlugin {
    constructor(...keys) {
        if (keys.length === 1 && Array.isArray(keys[0])) {
            this.keys = keys[0];
            this.defaultValues = {};
        } else if (keys.length === 1 && keys[0] && typeof keys[0] === "object") {
            this.keys = Object.keys(keys[0]);
            this.defaultValues = keys[0];
        } else {
            this.keys = keys;
            this.defaultValues = {};
        }
    }
 
    /**
     * Apply the plugin
     * @param {Compiler} compiler the compiler instance
     * @returns {void}
     */
    apply(compiler) {
        /** @type {Record<string, CodeValue>} */
        const definitions = {};
        for (const key of this.keys) {
            const value =
                process.env[key] !== undefined
                    ? process.env[key]
                    : this.defaultValues[key];
 
            if (value === undefined) {
                compiler.hooks.thisCompilation.tap("EnvironmentPlugin", compilation => {
                    const error = new WebpackError(
                        `EnvironmentPlugin - ${key} environment variable is undefined.\n\n` +
                            "You can pass an object with default values to suppress this warning.\n" +
                            "See https://webpack.js.org/plugins/environment-plugin for example."
                    );
 
                    error.name = "EnvVariableNotDefinedError";
                    compilation.errors.push(error);
                });
            }
 
            definitions[`process.env.${key}`] =
                value === undefined ? "undefined" : JSON.stringify(value);
        }
 
        new DefinePlugin(definitions).apply(compiler);
    }
}
 
module.exports = EnvironmentPlugin;