feat(webpack): include AEGIR_ prefixed env variables

ipfs · Jul 8, 2017 · a4adb59 · a4adb59
1 parent f87d6d3
commit a4adb59
Show file tree

Hide file tree

Showing 5 changed files with 74 additions and 2 deletions.
diff --git a/src/config/webpack/index.js b/src/config/webpack/index.js
@@ -1,6 +1,7 @@
 'use strict'
 
 const merge = require('webpack-merge')
+const webpack = require('webpack')
 
 const utils = require('../../utils')
 const base = require('./base')
@@ -21,7 +22,10 @@ function webpackConfig () {
         filename: entry.split('/').pop(),
         library: libraryName,
         path: utils.getPathToDist()
-      }
+      },
+      plugins: [
+        new webpack.DefinePlugin(utils.getEnv('production').stringified)
+      ]
     }, userConfig)
   })
 }

diff --git a/src/utils.js b/src/utils.js
@@ -90,3 +90,34 @@ exports.getListrConfig = () => {
     renderer: isCI ? VerboseRenderer : UpdateRenderer
   }
 }
+
+/**
+ * Get current env variables for inclusion.
+ *
+ * @param {string} [env='development']
+ *
+ * @returns {Object}
+ */
+exports.getEnv = (env) => {
+  const PREFIX = /^AEGIR_/i
+  const raw = Object.keys(process.env)
+    .filter((key) => PREFIX.test(key))
+    .reduce((env, key) => {
+      env[key] = process.env[key]
+      return env
+    }, {
+      NODE_ENV: process.env.NODE_ENV || env || 'development'
+    })
+
+  const stringifed = {
+    'process.env': Object.keys(raw).reduce((env, key) => {
+      env[key] = JSON.stringify(raw[key])
+      return env
+    }, {})
+  }
+
+  return {
+    raw: raw,
+    stringified: stringifed
+  }
+}
diff --git a/test/config/__snapshots__/webpack.spec.js.snap b/test/config/__snapshots__/webpack.spec.js.snap
@@ -25,7 +25,15 @@ Object {
   "performance": Object {
     "hints": false,
   },
-  "plugins": Array [],
+  "plugins": Array [
+    DefinePlugin {
+      "definitions": Object {
+        "process.env": Object {
+          "NODE_ENV": "production",
+        },
+      },
+    },
+  ],
   "resolve": Object {
     "alias": Object {
       "http": "stream-http",

diff --git a/test/config/webpack.spec.js b/test/config/webpack.spec.js
@@ -29,6 +29,15 @@ describe('config - webpack', () => {
       },
       getPathToNodeModules () {
         return 'aegir/node_modules'
+      },
+      getEnv () {
+        return {
+          stringified: {
+            'process.env': {
+              NODE_ENV: 'production'
+            }
+          }
+        }
       }
     }))
 

diff --git a/test/utils.spec.js b/test/utils.spec.js
@@ -50,4 +50,24 @@ describe('utils', () => {
   it('getPathToNodeModules', () => {
     expect(utils.getPathToNodeModules()).toEqual(expect.stringMatching(/node_modules$/))
   })
+
+  it('getEnv', () => {
+    process.env.AEGIR_TEST = 'hello'
+
+    const env = utils.getEnv()
+    expect(env.raw).toEqual({
+      NODE_ENV: 'test',
+      AEGIR_TEST: 'hello'
+    })
+    expect(env.stringified).toEqual({
+      'process.env': {
+        NODE_ENV: '"test"',
+        AEGIR_TEST: '"hello"'
+      }
+    })
+
+    process.env.NODE_ENV = ''
+    expect(utils.getEnv('production').raw).toHaveProperty('NODE_ENV', 'production')
+    process.env.NODE_ENV = 'test'
+  })
 })