WXL
2025-12-27 05e6b08007a86b5b10c680babc9c3bcc3a1a201b
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
/*
    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 */
 
const PLUGIN_NAME = "EnvironmentPlugin";
 
class EnvironmentPlugin {
    /**
     * @param {(string | string[] | Record<string, EXPECTED_ANY>)[]} keys keys
     */
    constructor(...keys) {
        if (keys.length === 1 && Array.isArray(keys[0])) {
            /** @type {string[]} */
            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 =
                /** @type {Record<string, EXPECTED_ANY>} */
                (keys[0]);
        } else {
            this.keys = /** @type {string[]} */ (keys);
            this.defaultValues = {};
        }
    }
 
    /**
     * Apply the plugin
     * @param {Compiler} compiler the compiler instance
     * @returns {void}
     */
    apply(compiler) {
        const definePlugin = new DefinePlugin({});
 
        compiler.hooks.thisCompilation.tap(PLUGIN_NAME, (compilation) => {
            /** @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) {
                    const error = new WebpackError(
                        `${PLUGIN_NAME} - ${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);
                }
                const defValue =
                    value === undefined ? "undefined" : JSON.stringify(value);
                definitions[`process.env.${key}`] = defValue;
                definitions[`import.meta.env.${key}`] = defValue;
            }
            definePlugin.definitions = definitions;
        });
        definePlugin.apply(compiler);
    }
}
 
module.exports = EnvironmentPlugin;