refactor changes to allow default values

neutrinojs · Jan 13, 2018 · 57fed5f · 57fed5f
1 parent 640c837
commit 57fed5f
Showing 1 changed file with 18 additions and 5 deletions.
diff --git a/packages/env/index.js b/packages/env/index.js
@@ -1,7 +1,20 @@
 const { EnvironmentPlugin } = require('webpack');
 
-module.exports = ({ config }, envs = []) => config
-  .plugin('env')
-  .use(EnvironmentPlugin, Array.isArray(envs) ?
-    ['NODE_ENV', ...envs] :
-    [{ NODE_ENV: 'development', ...envs }]);
+module.exports = ({ config }, envs = []) => {
+  let pluginOptions;
+
+  if (Array.isArray(envs)) {
+    pluginOptions = ['NODE_ENV', ...envs];
+  }
+  else {
+    pluginOptions = [
+      Object.assign({
+        NODE_ENV: 'development'
+      }, envs)
+    ];
+  }
+
+  config
+    .plugin('env')
+    .use(EnvironmentPlugin, pluginOptions);
+};