add invoke command

Author: horike37

index.js

@@ -2,6 +2,7 @@
 const BbPromise = require('bluebird');
 const deploy = require('./lib/deploy');
 const remove = require('./lib/remove');
+const invoke = require('./lib/invoke');
 
 class ServerlessStepFunctions {
   constructor(serverless, options) {
@@ -12,7 +13,8 @@ class ServerlessStepFunctions {
     Object.assign(
       this,
       deploy,
-      remove
+      remove,
+      invoke
     );
 
     this.commands = {
@@ -24,9 +26,9 @@ class ServerlessStepFunctions {
               'deploy',
             ],
             options: {
-              statemachine: {
+              state: {
                 usage: 'Name of the State Machine',
-                shortcut: 'sm',
+                shortcut: 's',
                 required: true,
               },
             },
@@ -41,9 +43,26 @@ class ServerlessStepFunctions {
               'remove',
             ],
             options: {
-              statemachine: {
+              state: {
                 usage: 'Name of the State Machine',
-                shortcut: 'sm',
+                shortcut: 's',
+                required: true,
+              },
+            },
+          },
+        },
+      },
+      invoke: {
+        commands: {
+          stepf: {
+            usage: 'Remove Step functions',
+            lifecycleEvents: [
+              'invoke',
+            ],
+            options: {
+              state: {
+                usage: 'Name of the State Machine',
+                shortcut: 's',
                 required: true,
               },
             },
@@ -57,6 +76,8 @@ class ServerlessStepFunctions {
         .then(this.deploy),
       'remove:stepf:remove': () => BbPromise.bind(this)
         .then(this.remove),
+      'invoke:stepf:invoke': () => BbPromise.bind(this)
+        .then(this.invoke),
     };
   }
 }

lib/deploy.js

@@ -92,7 +92,7 @@ module.exports = {
     .then((result) => {
       const region = this.options.region || 'us-east-1';
       this.stateMachineArn =
-      `arn:aws:states:${region}:${result.Account}:stateMachine:${this.options.statemachine}`;
+      `arn:aws:states:${region}:${result.Account}:stateMachine:${this.options.state}`;
       return BbPromise.resolve();
     });
   },
@@ -123,22 +123,22 @@ module.exports = {
       return BbPromise.resolve();
     }
 
-    if (typeof this.stepFunctions[this.options.statemachine] === 'undefined') {
+    if (typeof this.stepFunctions[this.options.state] === 'undefined') {
       const errorMessage = [
-        `Step function "${this.options.statemachine}" is not exists`,
+        `Step function "${this.options.state}" is not exists`,
       ].join('');
       throw new this.serverless.classes.Error(errorMessage);
     }
 
-    _.forEach(this.stepFunctions[this.options.statemachine].States, (value, key) => {
+    _.forEach(this.stepFunctions[this.options.state].States, (value, key) => {
       if (value.Resource && !value.Resource.match(/arn:aws:lambda/)) {
-        this.stepFunctions[this.options.statemachine].States[key].Resource
+        this.stepFunctions[this.options.state].States[key].Resource
         = this.functionArns[value.Resource];
       }
     });
 
-    this.awsStateLanguage[this.options.statemachine] =
-      JSON.stringify(this.stepFunctions[this.options.statemachine]);
+    this.awsStateLanguage[this.options.state] =
+      JSON.stringify(this.stepFunctions[this.options.state]);
     return BbPromise.resolve();
   },
 
@@ -157,16 +157,19 @@ module.exports = {
     return this.provider.request('StepFunctions',
       'createStateMachine',
       {
-        definition: this.awsStateLanguage[this.options.statemachine],
-        name: this.options.statemachine,
+        definition: this.awsStateLanguage[this.options.state],
+        name: this.options.state,
         roleArn: this.iamRoleArn,
       },
       this.options.stage,
       this.options.region)
     .then(() => BbPromise.resolve())
     .catch((error) => {
+    console.log(error)
       if (error.message.match(/State Machine is being deleted/)) {
         setTimeout(this.createStateMachine.bind(this), 5000);
+      } else {
+        throw new this.serverless.classes.Error(error.message);
       }
     });
   },

lib/invoke.js

@@ -0,0 +1,50 @@
+'use strict';
+const BbPromise = require('bluebird');
+
+module.exports = {
+  invoke() {
+    BbPromise.bind(this)
+    .then(this.getStateMachineArn)
+    .then(this.startExecution)
+    .then(this.describeExecution);
+
+    return BbPromise.resolve();
+  },
+  
+  startExecution() {
+    return this.provider.request('StepFunctions',
+      'startExecution',
+      {
+        stateMachineArn: this.stateMachineArn,
+      },
+      this.options.stage,
+      this.options.region)
+    .then((result) => {
+      this.executionArn = result.executionArn
+      return BbPromise.resolve();
+    }).catch((error) => {
+      console.log(error);
+    });
+  },
+
+  describeExecution() {
+    return this.provider.request('StepFunctions',
+      'describeExecution',
+      {
+        executionArn: this.executionArn,
+      },
+      this.options.stage,
+      this.options.region)
+    .then((result) => {
+
+      if (result.status === 'RUNNING') {
+        setTimeout(this.describeExecution.bind(this), 5000);
+      } else {
+        
+      }
+      return BbPromise.resolve();
+    }).catch((error) => {
+      console.log(error);
+    });
+  }
+};