Blame view

node_modules/webpack/lib/EnvironmentPlugin.js 1.76 KB
aaac7fed   liuqimichale   add
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
  /*
  	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 needsEnvVarFix = ["8", "9"].indexOf(process.versions.node.split(".")[0]) >= 0 &&
  	process.platform === "win32";
  
  class EnvironmentPlugin {
  	constructor(keys) {
  		if(Array.isArray(keys)) {
  			this.keys = keys;
  			this.defaultValues = {};
  		} else if(keys && typeof keys === "object") {
  			this.keys = Object.keys(keys);
  			this.defaultValues = keys;
  		} else {
  			this.keys = Array.prototype.slice.call(arguments);
  			this.defaultValues = {};
  		}
  	}
  
  	apply(compiler) {
  		const definitions = this.keys.reduce((defs, key) => {
  			// TODO remove once the fix has made its way into Node 8.
  			// Work around https://github.com/nodejs/node/pull/18463,
  			// affecting Node 8 & 9 by performing an OS-level
  			// operation that always succeeds before reading
  			// environment variables:
  			if(needsEnvVarFix) require("os").cpus();
  
  			const value = process.env[key] !== undefined ? process.env[key] : this.defaultValues[key];
  
  			if(value === undefined) {
  				compiler.plugin("this-compilation", compilation => {
  					const error = new Error(
  						`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.warnings.push(error);
  				});
  			}
  
  			defs[`process.env.${key}`] = typeof value === "undefined" ? "undefined" : JSON.stringify(value);
  
  			return defs;
  		}, {});
  
  		compiler.apply(new DefinePlugin(definitions));
  	}
  }
  
  module.exports = EnvironmentPlugin;