Merge pull request #2 from adambartholomew/task-resource-mapping

Performing resource replacements for localhost using TaskResourceMappings

Author: codetheweb

CHANGELOG.md

@@ -0,0 +1,3 @@
+# 0.1.3
+
+- Allow for resource replacements for local development using custom `TaskResourceMapping` configuration

README.md

@@ -51,6 +51,7 @@ It also adds an environment variable for each created state machine that contain
 - `region` (required) your AWS region
 - `lambdaEndpoint` (defaults to `http://localhost:4000`) the endpoint for the lambda service
 - `path` (defaults to `./.step-functions-local`) the path to store the downloaded step function executables
+- `TaskResourceMapping` allows for Resource ARNs to be configured differently for local development
 
 ### Full Config Example
 
@@ -72,6 +73,9 @@ custom:
   stepFunctionsLocal:
     accountId: 101010101010
     region: us-east-1
+    TaskResourceMapping:
+      FirstState: arn:aws:lambda:us-east-1:101010101010:function:hello
+      FinalState: arn:aws:lambda:us-east-1:101010101010:function:hello
 
 functions:
   hello:
@@ -86,14 +90,14 @@ stepFunctions:
         States:
           FirstState:
             Type: Task
-            Resource: arn:aws:lambda:us-east-1:101010101010:function:hello
+            Resource: Fn::GetAtt: [hello, Arn]
             Next: wait_using_seconds
           wait_using_seconds:
             Type: Wait
             Seconds: 10
             Next: FinalState
           FinalState:
             Type: Task
-            Resource: arn:aws:lambda:us-east-1:101010101010:function:hello
+            Resource: Fn::GetAtt: [hello, Arn]
             End: true
 ```

index.js

@@ -79,6 +79,28 @@ class ServerlessStepFunctionsLocal {
     const parsed = await this.serverless.yamlParser.parse(configPath);
 
     this.stateMachines = parsed.stepFunctions.stateMachines;
+
+    if (parsed.custom 
+      && parsed.custom.stepFunctionsLocal
+      && parsed.custom.stepFunctionsLocal.TaskResourceMapping) {
+        this.replaceTaskResourceMappings(parsed.stepFunctions.stateMachines, parsed.custom.stepFunctionsLocal.TaskResourceMapping);
+    }
+  }
+
+  /**
+   * Replaces Resource properties with values mapped in TaskResourceMapping
+   */
+  replaceTaskResourceMappings(input, replacements, parentKey) {
+    for(var key in input) {
+      var property = input[key];
+      if (['object', 'array'].indexOf(typeof property) > -1) {
+        if (input['Resource'] && replacements[parentKey]) {
+          input['Resource'] = replacements[parentKey];
+        }
+        // Recursive replacement of nested states
+        this.replaceTaskResourceMappings(property, replacements, key);
+      }
+    }
   }
 
   async createEndpoints() {

package-lock.json

@@ -1,6 +1,6 @@
 {
   "name": "serverless-step-functions-local",
-  "version": "0.1.2",
+  "version": "0.1.3",
   "description": "Run AWS step functions offline with Serverless",
   "main": "index.js",
   "scripts": {