🎉 Extend "--env-from-file" support to update command

LinusU · Feb 14, 2018 · 8eb6a3c · 8eb6a3c
1 parent f8c0083
commit 8eb6a3c
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 0 deletions.
diff --git a/app.js b/app.js
@@ -50,6 +50,8 @@ async function main () {
 
   const updateList = new Listr([
     tasks.packageApp,
+    tasks.readEnvironmentFile,
+    tasks.updateLambdaEnvironment,
     tasks.updateLambdaFunction,
     tasks.loadSwaggerDefinition,
     tasks.generateSwaggerDefinition,

diff --git a/lib/amazon.js b/lib/amazon.js
@@ -64,6 +64,15 @@ exports.getFunctionEnvironment = function ({ functionName }) {
   return lambda.getFunctionConfiguration(params).promise().then(res => (res.Environment && res.Environment.Variables) || {})
 }
 
+exports.updateFunctionEnvironment = function ({ functionName, environment }) {
+  const params = {
+    FunctionName: functionName,
+    Environment: { Variables: environment }
+  }
+
+  return lambda.updateFunctionConfiguration(params).promise().then(res => res.FunctionArn)
+}
+
 exports.createApiGateway = async function ({ definition, lambdaArn }) {
   const params = {
     body: JSON.stringify(definition),

diff --git a/lib/tasks.js b/lib/tasks.js
@@ -65,6 +65,15 @@ exports.updateLambdaFunction = {
   }
 }
 
+exports.updateLambdaEnvironment = {
+  title: 'Update Lambda environment',
+  enabled: (ctx) => Boolean(ctx.args['--env-from-file']),
+  task: async (ctx, task) => {
+    await amazon.updateFunctionEnvironment({ functionName: ctx.args['<name>'], environment: ctx.environment })
+    task.title = `Updated existing Lambda environment`
+  }
+}
+
 exports.loadSwaggerDefinition = {
   title: 'Load Swagger definition',
   enabled: (ctx) => Boolean(ctx.args['--swagger']),