add invoke command

Author: horike37

lib/index.js

@@ -11,6 +11,7 @@ const httpIamRole = require('./deploy/events/apiGateway/iamRole');
 const httpDeployment = require('./deploy/events/apiGateway/deployment');
 const httpRestApi = require('./deploy/events/apiGateway/restApi');
 const httpInfo = require('./deploy/events/apiGateway/endpointInfo');
+const invoke = require('./invoke/invoke');
 const yamlParser = require('./yamlParser');
 const naming = require('./naming');
 const _ = require('lodash');
@@ -37,6 +38,7 @@ class ServerlessStepFunctions {
       httpMethods,
       httpIamRole,
       httpDeployment,
+      invoke,
       yamlParser,
       naming
     );
@@ -51,7 +53,7 @@ class ServerlessStepFunctions {
             ],
             options: {
               name: {
-                usage: 'State Machine name',
+                usage: 'The StateMachine name',
                 shortcut: 'n',
                 required: true,
               },
@@ -82,7 +84,7 @@ class ServerlessStepFunctions {
       'invoke:stepf:invoke': () => BbPromise.bind(this)
         .then(this.invoke),
       'deploy:initialize': () => BbPromise.bind(this)
-          .then(this.yamlParse),
+        .then(this.yamlParse),
       'deploy:compileFunctions': () => BbPromise.bind(this)
         .then(this.compileIamRole)
         .then(this.compileStateMachines)
@@ -108,7 +110,23 @@ class ServerlessStepFunctions {
   }
 
   invoke() {
-
+    return BbPromise.bind(this)
+    .then(this.yamlParse)
+    .then(this.getStateMachineArn)
+    .then(this.startExecution)
+    .then(this.describeExecution)
+    .then((result) => {
+      if (result.status === 'FAILED') {
+        return this.getExecutionHistory()
+        .then((error) => {
+          this.serverless.cli.consoleLog(_.merge(result, error.events[error.events.length - 1]
+          .executionFailedEventDetails));
+          process.exitCode = 1;
+        });
+      }
+      this.serverless.cli.consoleLog(result);
+      return BbPromise.resolve();
+    });
   }
 
   display() {

lib/invoke/invoke.js

@@ -0,0 +1,102 @@
+'use strict';
+
+const BbPromise = require('bluebird');
+const path = require('path');
+
+module.exports = {
+  getStateMachineArn() {
+    const stateMachineOutputKey =
+      this.getStateMachineOutputLogicalId(this.options.name);
+    const stackName = this.provider.naming.getStackName(this.options.stage);
+
+    return this.provider.request('CloudFormation',
+      'describeStacks',
+      { StackName: stackName },
+      this.options.stage,
+      this.options.region)
+    .then((result) => {
+      if (result) {
+        result.Stacks[0].Outputs.forEach((output) => {
+          if (output.OutputKey === stateMachineOutputKey) {
+            this.stateMachineArn = output.OutputValue;
+          }
+        });
+
+        if (!this.stateMachineArn) {
+          const errorMessage = [
+            `"${this.options.name}" stateMachine does not exists.`,
+          ].join('');
+          throw new this.serverless.classes.Error(errorMessage);
+        }
+      }
+
+      return BbPromise.resolve();
+    });
+  },
+
+  startExecution() {
+    let inputData;
+
+    if (this.options.data) {
+      inputData = this.options.data;
+    } else if (this.options.path) {
+      const absolutePath = path.isAbsolute(this.options.path) ?
+        this.options.path :
+      path.join(this.serverless.config.servicePath, this.options.path);
+      if (!this.serverless.utils.fileExistsSync(absolutePath)) {
+        throw new this.serverless.classes.Error('The file you provided does not exist.');
+      }
+      inputData = JSON.stringify(this.serverless.utils.readFileSync(absolutePath));
+    }
+
+    return this.provider.request('StepFunctions',
+      'startExecution',
+      {
+        stateMachineArn: this.stateMachineArn,
+        input: inputData,
+      },
+      this.options.stage,
+      this.options.region)
+    .then((result) => {
+      this.executionArn = result.executionArn;
+      return BbPromise.resolve();
+    }).catch((error) => {
+      throw new this.serverless.classes.Error(error.message);
+    });
+  },
+
+  describeExecution() {
+    return this.provider.request('StepFunctions',
+      'describeExecution',
+      {
+        executionArn: this.executionArn,
+      },
+      this.options.stage,
+      this.options.region)
+    .then((result) => {
+      if (result.status === 'RUNNING') {
+        this.serverless.cli.printDot();
+        return this.setTimeout()
+        .then(() => this.describeExecution());
+      }
+      return BbPromise.resolve(result);
+    });
+  },
+
+  getExecutionHistory() {
+    return this.provider.request('StepFunctions',
+      'getExecutionHistory',
+      {
+        executionArn: this.executionArn,
+      },
+      this.options.stage,
+      this.options.region)
+    .then((result) => BbPromise.resolve(result));
+  },
+
+  setTimeout() {
+    return new BbPromise((resolve) => {
+      setTimeout(resolve, 5000);
+    });
+  },
+};