Add create statemachine and update

Author: horike37

lib/index.js

@@ -1,7 +1,6 @@
 'use strict';
 const BbPromise = require('bluebird');
 const path = require('path');
-const _ = require('lodash');
 
 class AwsStepFunctionsDeploy {
   constructor(serverless, options) {
@@ -15,31 +14,47 @@ class AwsStepFunctionsDeploy {
           stepf: {
             usage: 'Deploy Step functions',
             lifecycleEvents: [
-              'deploy'
+              'deploy',
             ],
             options: {
               statemachine: {
                 usage: 'Name of the State Machine',
                 shortcut: 'sm',
                 required: true,
-              }
-            }
-          }
-        }
-      }
+              },
+            },
+          },
+        },
+      },
     };
 
     this.hooks = {
       'deploy:stepf:deploy': this.action.bind(this),
     };
   }
 
+  setStateMachineArn() {
+    return this.provider.request('STS',
+      'getCallerIdentity',
+      {},
+      this.options.stage,
+      this.options.region)
+    .then((result) => {
+      const region = this.options.region || 'us-east-1';
+      this.stateMachineArn =
+      `arn:aws:states:${region}:${result.Account}:stateMachine:${this.options.statemachine}`;
+      return BbPromise.resolve();
+    });
+  }
+
   action() {
     this.serverless.cli.consoleLog('Start Deploy Step Functions');
     BbPromise.bind(this)
     .then(this.yamlParse)
+    .then(this.setStateMachineArn)
     .then(this.compile)
-    .then(this.deploy);
+    .then(this.deleteStateMachine)
+    .then(this.createStateMachine);
   }
 
   yamlParse() {
@@ -70,60 +85,51 @@ class AwsStepFunctionsDeploy {
 
     if (typeof this.stepFunctions[this.options.statemachine] === 'undefined') {
       const errorMessage = [
-         `Step function "${this.options.statemachine}" is not exists`,
-       ].join('');
-       throw new this.serverless.classes
-       .Error(errorMessage);
+        `Step function "${this.options.statemachine}" is not exists`,
+      ].join('');
+      throw new this.serverless.classes.Error(errorMessage);
     }
 
-    //@todo get lambda arn from functionname
-    this.awsStateLanguage[this.options.statemachine] = JSON.stringify(this.stepFunctions[this.options.statemachine]);
+    // @todo get lambda arn from functionname
+    this.awsStateLanguage[this.options.statemachine] =
+      JSON.stringify(this.stepFunctions[this.options.statemachine]);
     return BbPromise.resolve();
   }
 
-  deploy() {
-    let stateMachineArn;
-
-    return this.provider.request('STS',
-      'getCallerIdentity',
-      {},
+  deleteStateMachine() {
+    return this.provider.request('StepFunctions',
+      'deleteStateMachine',
+      {
+        stateMachineArn: this.stateMachineArn,
+      },
       this.options.stage,
       this.options.region)
-    .then((result) => {
-      const region = this.options.region || 'us-east-1';
-      stateMachineArn = `arn:aws:states:${region}:${result.Account}:stateMachine:${this.options.statemachine}`;
-    }).then((result) => {
-      return this.provider.request('StepFunctions',
-        'describeStateMachine',
-        {
-          stateMachineArn
-        },
-        this.options.stage,
-        this.options.region);
-    }).then((result) => {
-    console.log(result)
-      return this.provider.request('StepFunctions',
-        'deleteStateMachine',
-        {
-          stateMachineArn
-        },
-        this.options.stage,
-        this.options.region);
+    .then(() => BbPromise.resolve());
+  }
 
-    }).then((result) => {
-      return this.provider.request('StepFunctions',
-        'createStateMachine',
-        {
-          definition: this.awsStateLanguage[this.options.statemachine],
-          name: this.options.statemachine,
-          roleArn: ''
-        },
-        this.options.stage,
-        this.options.region)
-    }).catch((error) => {
-      console.log(error);
+  createStateMachine() {
+    return this.provider.request('StepFunctions',
+      'createStateMachine',
+      {
+        definition: this.awsStateLanguage[this.options.statemachine],
+        name: this.options.statemachine,
+        roleArn: '',
+      },
+      this.options.stage,
+      this.options.region)
+    .then(() => BbPromise.resolve())
+    .catch((error) => {
+      if (error.message.match(/State Machine is being deleted/)) {
+        setTimeout(this.createStateMachine.bind(this), 5000);
+      }
     });
   }
 
+  deploy() {
+    return BbPromise.bind(this)
+    .then(this.deleteStateMachine)
+    .then(this.createStateMachine);
+  }
+
 }
 module.exports = AwsStepFunctionsDeploy;