Merge pull request #14 from TaniaPash/master

fix: add cleanup-states-script

Author: piercus

package.json

@@ -36,6 +36,7 @@
   "homepage": "https://github.com/piercus/step-function-worker#readme",
   "devDependencies": {
     "ava": "^0.25.0",
+    "bluebird": "^3.5.3",
     "semantic-release": "^15.1.7",
     "winston": "^2.4.1",
     "xo": "^0.23.0"
@@ -44,4 +45,4 @@
     "aws-arn-parser": "^1.0.0",
     "aws-sdk": "^2.82.0"
   }
-}
+}

test/scripts/cleanup-states.js

@@ -0,0 +1,56 @@
+const AWS = require('aws-sdk');
+const PromiseBlue = require('bluebird');
+const winston = require('winston');
+
+const logger = new winston.Logger({
+	transports: [new winston.transports.Console({
+		level: 'info'
+	})]
+});
+
+if (typeof (process.env.AWS_REGION) !== 'string') {
+	throw (new TypeError('$AWS_REGION must be defined'));
+}
+
+const stepfunctions = new AWS.StepFunctions({
+	region: process.env.AWS_REGION
+});
+
+const reg = new RegExp('stateMachine:test-state-machine');
+
+const removeStateMachines = function (reg) {
+	const params = {};
+	return stepfunctions.listStateMachines(params).promise().then(data => {
+		const stateMachinesArn = [];
+		data.stateMachines.forEach(stm => {
+			if (reg.test(stm.stateMachineArn)) {
+				stateMachinesArn.push(stm.stateMachineArn);
+			}
+		});
+		return stateMachinesArn;
+	}).then(stateMachinesArn => {
+		return PromiseBlue.map(stateMachinesArn, stateMachineArn => {
+			const params = {
+				stateMachineArn
+			};
+			return stepfunctions.describeStateMachine(params).promise().then(data => {
+				const definition = JSON.parse(data.definition);
+				const activityArn = definition.States.FirstState.Resource;
+				const params = {
+					activityArn
+				};
+				return stepfunctions.deleteActivity(params).promise().then(() => {
+					logger.info('Activity ', params.activityArn, ' was deleted');
+				});
+			}).then(() => {
+				return stepfunctions.deleteStateMachine(params).promise().then(() => {
+					logger.info('StateMachine ', params.stateMachineArn, ' was deleted');
+				});
+			});
+		}, {concurrency: 1});
+	});
+};
+
+removeStateMachines(reg).catch(err => {
+	logger.error(err);
+});