node8, const, async

.travis.yml

@@ -1,3 +1,3 @@
 language: node_js
 node_js:
-  - "6"
+  - "8"

dryRun.js

@@ -1,6 +1,6 @@
 'use strict';
 
-var {cleanupStacks} = require('./src');
+const {cleanupStacks} = require('./src');
 
 module.exports.test = (event, context, callback) => {
   cleanupStacks(true)

handler.js

@@ -1,8 +1,8 @@
 'use strict';
 
-var AWS = require('aws-sdk');
-var {cleanupStacks} = require('./src');
-var dynamo = require('./src/dynamo.js');
+const AWS = require('aws-sdk');
+const {cleanupStacks} = require('./src');
+const dynamo = require('./src/dynamo.js');
 
 module.exports.cleanupStacks = (event, context, callback) => {
   console.log('running some awesome logic');

serverless.yml

@@ -2,7 +2,7 @@ service: nib-batman
 
 provider:
   name: aws
-  runtime: nodejs6.10
+  runtime: nodejs8.10
   environment:
     GITHUB_SECRET: ${env:GITHUB_SECRET}
     BRANCH_KEY: slice

src/dynamo.js

@@ -1,13 +1,13 @@
 'use strict';
 
-var AWS = require("aws-sdk");
-var moment = require('moment');
-var dynamo = new AWS.DynamoDB();
-var tableName = "Batman";
+const AWS = require("aws-sdk");
+const moment = require('moment');
+const dynamo = new AWS.DynamoDB();
+const tableName = "Batman";
 
 function putItem(deleteDate, deltedStackName) {
   return new Promise((resolve, reject) => {
-    var params = {
+    const params = {
       TableName: tableName,
       Item: {
         "StackName": {S: deltedStackName},
@@ -30,10 +30,10 @@ function putItem(deleteDate, deltedStackName) {
   });
 }
 
-var writeRecords = function(deletedStackNames) {
+const writeRecords = function(deletedStackNames) {
   console.log('creating records...')
-  var nowUtcString = moment().utc().format();
-  var writeToDynamoPromises = deletedStackNames.map(stackName => putItem(nowUtcString, stackName));
+  const nowUtcString = moment().utc().format();
+  const writeToDynamoPromises = deletedStackNames.map(stackName => putItem(nowUtcString, stackName));
   return Promise.all(writeToDynamoPromises);
 }
 

src/index.js

@@ -1,72 +1,57 @@
 'use strict';
 
-var AWS = require('aws-sdk');
-var _ = require('lodash');
-var moment = require('moment');
+const AWS = require('aws-sdk');
+const _ = require('lodash');
+const moment = require('moment');
 
-var cloudformation = new AWS.CloudFormation();
+const cloudformation = new AWS.CloudFormation();
 
-var batmanTagKey = 'Batman';
+const batmanTagKey = 'Batman';
 
-function cleanupStacks(dryRun) {
+const cleanupStacks = async dryRun => {
   console.log("In Clean UP Stacks");
 
-  var batmanTaggedStacks = [];
-
-  return listAllStacksWrapper()
-    .then(allStacks => {
-      console.log("allStacks", allStacks.length);
-      var nonMasterStacks = findNonMasterStacks(allStacks);
-      var oldNonMasterStacks = getOldStacks(nonMasterStacks);
-      var withStatuses = filterStacksByStatus(oldNonMasterStacks);
-      var notBatmanTag = ignoreStacksWithBatmanTag(withStatuses);
-      batmanTaggedStacks = getBatmanStacks(withStatuses);
-      return notBatmanTag;
-    })
-    .then(filteredStacks => {
-      console.log('filteredStacks', filteredStacks);
-      console.log(filteredStacks);
-      return deleteStacks(filteredStacks, dryRun);
-    })
-    .then(deletedStacks => {
-      console.log('deletedStacks', deletedStacks);
-      var stackNames = deletedStacks.map(stack => stack.StackName).sort();
-      console.log('Stack names: ', stackNames);
-      return {
-        deleted: stackNames, 
-        mercyShown: batmanTaggedStacks.map(stack => stack.StackName).sort()
-      }
-    });
+  const allStacks = await listAllStacks(null, [])
+  console.log("allStacks", allStacks.length);
+  const nonMasterStacks = findNonMasterStacks(allStacks);
+  const oldNonMasterStacks = getOldStacks(nonMasterStacks);
+  const withStatuses = filterStacksByStatus(oldNonMasterStacks);
+  const filteredStacks = ignoreStacksWithBatmanTag(withStatuses);
+  const batmanTaggedStacks = getBatmanStacks(withStatuses);
+  console.log('filteredStacks', filteredStacks);
+
+  const deletedStacks = await deleteStacks(filteredStacks, dryRun);
+  console.log('deletedStacks', deletedStacks);
+  const stackNames = deletedStacks.map(stack => stack.StackName).sort();
+  console.log('Stack names: ', stackNames);
+  return {
+    deleted: stackNames, 
+    mercyShown: batmanTaggedStacks.map(stack => stack.StackName).sort()
+  }
 }
 
-function deleteFromGitBranch(branchDeleteInformation) {
+const deleteFromGitBranch = async (branchDeleteInformation) => {
   console.log(branchDeleteInformation);
-  return listAllStacksWrapper()
-    .then(allStacks => {
-      var nonMasterStacks = findNonMasterStacks(allStacks);
-      var foundStack = stacksWithRepoAndBranchName(nonMasterStacks, branchDeleteInformation.repository.name, branchDeleteInformation.ref);
-      return foundStack;
-    })
-    .then(filteredStacks => {
-      console.log('found stack', filteredStacks);
-      return deleteStacks(filteredStacks);
-    })
-    .then(deletedStacks => {
-      console.log('deletedStacks', deletedStacks);
-      var stackNames = deletedStacks.map(stack => stack.StackName).sort();
-      console.log('Stack names: ', stackNames);
-      return {
-        deleted: stackNames, 
-        mercyShown: []
-      }
-    });
+  const allStacks = await listAllStacks(null, [])
+  const nonMasterStacks = findNonMasterStacks(allStacks);
+  const foundStack = stacksWithRepoAndBranchName(nonMasterStacks, branchDeleteInformation.repository.name, branchDeleteInformation.ref);
+
+  console.log('found stack', foundStack);
+  const deletedStacks = await deleteStacks(foundStack);
+  console.log('deletedStacks', deletedStacks);
+  const stackNames = deletedStacks.map(stack => stack.StackName).sort();
+  console.log('Stack names: ', stackNames);
+  return {
+    deleted: stackNames, 
+    mercyShown: []
+  }
 }
 
-function deleteStacks(stacks, dryRun) {
+const deleteStacks = async (stacks, dryRun) => {
   if (dryRun)
-    return Promise.resolve(stacks);
+    return stacks;
 
-  var deleteAllTheThings = stacks.map(stack => {
+  const deleteAllTheThings = stacks.map(stack => {
     return cloudformation.deleteStack({StackName: stack.StackName}).promise();
   });
 
@@ -78,37 +63,26 @@ function deleteStacks(stacks, dryRun) {
   });
 }
 
-function listAllStacksWrapper() {
-  return new Promise((resolve, reject) => {
-    return listAllStacks(null, [], resolve, reject);
-  });
-}
-
-function listAllStacks(token, stackArray, resolve, reject) {
+const listAllStacks = (token, stackArray) => {
   console.log('getting stacks.....');
-  var params = {NextToken: token};
-  cloudformation.describeStacks(params, (err, data) => {
-    if(err) {
-      console.error("ERROR: ", err);
-      return reject(err);
-    }
-    var stacks = stackArray.concat(data.Stacks);
-    if(!data.NextToken)
-      return resolve(stacks);
-    return listAllStacks(data.NextToken, stacks, resolve, reject);
-  });
+  const params = { NextToken: token };
+  const data = await cloudformation.describeStacks(params).promise()
+  const stacks = stackArray.concat(data.Stacks);
+  if(!data.NextToken)
+    return stacks;
+  return listAllStacks(data.NextToken, stacks);
 }
 
 function filterStacksByStatus(stacksArray) {
   return _.filter(stacksArray, stack => {
-    var statuses = ["CREATE_COMPLETE", "UPDATE_COMPLETE", "DELETE_FAILED"];
+    const statuses = ["CREATE_COMPLETE", "UPDATE_COMPLETE", "DELETE_FAILED"];
     return statuses.indexOf(stack.StackStatus) >= 0;
   });
 }
 
 function findNonMasterStacks(stacksArray) {
-  var nonMasterStacks = _.filter(stacksArray, stack => {
-    var branchTag = _.filter(stack.Tags, (tag) => {return tag.Key.toLowerCase() == process.env.BRANCH_KEY.toLowerCase()});
+  const nonMasterStacks = _.filter(stacksArray, stack => {
+    const branchTag = _.filter(stack.Tags, (tag) => {return tag.Key.toLowerCase() == process.env.BRANCH_KEY.toLowerCase()});
     if(branchTag.length == 0) {
       // check if there are any tags at all - this might be an Elastic Beanstalk app!
       if (stack.Tags.length === 0) {
@@ -125,29 +99,29 @@ function isStackNonMasterElasticBeanstalk(stack) {
   if (stack.StackName.toLowerCase().indexOf('master') > -1) {
     return false;
   }
-  var containsBeanstalkOutput = _.filter(stack.Outputs, ['Description', 'Beanstalk Service Role'])
+  const containsBeanstalkOutput = _.filter(stack.Outputs, ['Description', 'Beanstalk Service Role'])
   return containsBeanstalkOutput.length > 0;
 }
 
 function getOldStacks(stacksArray) {
   return _.filter(stacksArray, stack => {
-    var mostRecentChange = stack.LastUpdatedTime ? stack.LastUpdatedTime : stack.CreationTime;
-    var daysOld = moment(new Date()).diff(mostRecentChange, 'days', false);
+    const mostRecentChange = stack.LastUpdatedTime ? stack.LastUpdatedTime : stack.CreationTime;
+    const daysOld = moment(new Date()).diff(mostRecentChange, 'days', false);
     return daysOld > 7;
   });
 }
 
 function ignoreStacksWithBatmanTag(stacksArray) {
   console.log(stacksArray)
-  var stacksWithoutBatmanTag = _.filter(stacksArray, stack => {
+  const stacksWithoutBatmanTag = _.filter(stacksArray, stack => {
     return batmanTagFilter(stack.Tags).length === 0;
   });
   console.log(stacksWithoutBatmanTag);
   return stacksWithoutBatmanTag
 }
 
-function getBatmanStacks(stacksArray) {
-  var stacksWithoutBatmanTag = _.filter(stacksArray, stack => {
+const getBatmanStacks = stacksArray => {
+  const stacksWithoutBatmanTag = _.filter(stacksArray, stack => {
     return batmanTagFilter(stack.Tags).length > 0;
   });
   return stacksWithoutBatmanTag
@@ -158,10 +132,10 @@ function batmanTagFilter(stackTags) {
 }
 
 function stacksWithRepoAndBranchName(stacksArray, repositoryName, branchName) {
-  var stacksWithRepoAndBranchName = _.filter(stacksArray, stack => {
-    var stackNameLower = stack.StackName.toLowerCase();
-    var repositoryNameLower = repositoryName.toLowerCase();
-    var branchNameLower = branchName.toLowerCase();
+  const stacksWithRepoAndBranchName = _.filter(stacksArray, stack => {
+    const stackNameLower = stack.StackName.toLowerCase();
+    const repositoryNameLower = repositoryName.toLowerCase();
+    const branchNameLower = branchName.toLowerCase();
 
     return stackNameLower.indexOf(repositoryNameLower) > -1 && stackNameLower.indexOf(branchNameLower) > -1;
   });