Adding environment to env vars added in _setEnvironmentVars(). (#95)

* Adding environment to env vars added in _setEnvironmentVars().
motdotla · May 12, 2016 · 12c7eb1 · 12c7eb1
1 parent a59be77
commit 12c7eb1
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/lib/main.js b/lib/main.js
@@ -271,6 +271,10 @@ Lambda.prototype._setEnvironmentVars = function (program, codeDirectory) {
  var config = dotenv.parse(configValues);
  var contentStr = contents.toString();
 
+ if(program.environment){
+ prefix += 'process.env["environment"]=' + program.environment + ';\n';
+ }
+
  for (var k in config) {
  if (!config.hasOwnProperty(k)) {
  continue;