Added EnvPlugin.

Simen Brekken · Simen Brekken · commit b80edb6a4984 · 2014-12-23T11:06:48.000+01:00
diff --git a/lib/EnvPlugin.js b/lib/EnvPlugin.js
@@ -0,0 +1,32 @@
+/*
+	MIT License http://www.opensource.org/licenses/mit-license.php
+	Author Simen Brekken @simenbrekken
+*/
+var DefinePlugin = require("./DefinePlugin");
+
+function EnvPlugin(keys) {
+	this.keys = Array.isArray(keys) ? keys : Array.prototype.slice.call(arguments);
+}
+module.exports = EnvPlugin;
+
+EnvPlugin.prototype.apply = function(compiler) {
+	compiler.apply(new DefinePlugin(getDefinitions(this.keys)));
+
+	function getDefinitions(keys) {
+		return keys.reduce(function(definitions, key) {
+			var value = process.env[key];
+
+			if (value === undefined) {
+				compiler.plugin("compilation", function(compilation) {
+					var error = new Error(key + " environment variable is undefined.");
+					error.name = "EnvVariableNotDefinedError";
+					compilation.warnings.push(error);
+				});
+			}
+
+			definitions["process.env." + key] = value ? JSON.stringify(value) : "undefined";
+
+			return definitions;
+		}, {});
+	}
+};
diff --git a/lib/webpack.js b/lib/webpack.js
@@ -77,6 +77,7 @@ exportPlugins(exports, ".", [
 	"UmdMainTemplatePlugin",
 	"NoErrorsPlugin",
 	"NewWatchingPlugin",
+	"EnvPlugin"
 ]);
 exportPlugins(exports.optimize = {}, "./optimize", [
 	"AggressiveMergingPlugin",
