add deploy

Author: horike37

lib/index.js

@@ -1,13 +1,15 @@
 'use strict';
 const BbPromise = require('bluebird');
 const path = require('path');
+const _ = require('lodash');
 
 class AwsStepFunctionsDeploy {
   constructor(serverless, options) {
     this.serverless = serverless;
     this.options = options;
     this.provider = this.serverless.getProvider('aws');
     this.awsStateLanguage = {};
+    this.functionArns = {};
     this.commands = {
       deploy: {
         commands: {
@@ -52,7 +54,8 @@ class AwsStepFunctionsDeploy {
     this.serverless.cli.consoleLog('Start Deploy Step Functions');
     BbPromise.bind(this)
     .then(this.yamlParse)
-    .then(this.setStateMachineArn)
+    .then(this.getStateMachineArn)
+    .then(this.getFunctionArns)
     .then(this.compile)
     .then(this.getIamRole)
     .then(this.deleteStateMachine)
@@ -78,6 +81,22 @@ class AwsStepFunctionsDeploy {
     });
   }
 
+  getFunctionArns() {
+    return this.provider.request('STS',
+      'getCallerIdentity',
+      {},
+      this.options.stage,
+      this.options.region)
+    .then((result) => {
+      const region = this.options.region || 'us-east-1';
+      _.forEach(this.serverless.service.functions, (value, key) => {
+        this.functionArns[key]
+        = `arn:aws:lambda:${region}:${result.Account}:function:${value.name}`;
+      });
+      return BbPromise.resolve();
+    });
+  }
+
   createIamRole() {
     return this.provider.request('IAM',
       'createRole',
@@ -93,7 +112,7 @@ class AwsStepFunctionsDeploy {
     });
   }
 
-  setStateMachineArn() {
+  getStateMachineArn() {
     return this.provider.request('STS',
       'getCallerIdentity',
       {},
@@ -140,6 +159,13 @@ class AwsStepFunctionsDeploy {
       throw new this.serverless.classes.Error(errorMessage);
     }
 
+    _.forEach(this.stepFunctions[this.options.statemachine].States, (value, key) => {
+      if (value.Resource && !value.Resource.match(/arn:aws:lambda/)) {
+        this.stepFunctions[this.options.statemachine].States[key].Resource
+        = this.functionArns[value.Resource];
+      }
+    });
+
     this.awsStateLanguage[this.options.statemachine] =
       JSON.stringify(this.stepFunctions[this.options.statemachine]);
     return BbPromise.resolve();
@@ -173,12 +199,5 @@ class AwsStepFunctionsDeploy {
       }
     });
   }
-
-  deploy() {
-    return BbPromise.bind(this)
-    .then(this.deleteStateMachine)
-    .then(this.createStateMachine);
-  }
-
 }
 module.exports = AwsStepFunctionsDeploy;