Add environment variables for ARNs

Author: codetheweb

README.md

@@ -39,6 +39,10 @@ custom:
 
 Although not neccessary, it's strongly recomended to add the folder with the downloaded step function executables to `.gitignore`.  By default, this path is `./.step-functions-local`.
 
+The plugin binds to port 8083, this cannot be changed.
+
+It also adds an environment variable for each created state machine that contains the ARN for it.  These variables are prefixed by `OFFLINE_STEP_FUNCTIONS_ARN_`, so the ARN of a state machine named 'WaitMachine', for example could be fetched by reading `OFFLINE_STEP_FUNCTIONS_ARN_WaitMachine`.
+
 ## Options
 
 (These go under `custom.stepFunctionsLocal`.)

index.js

@@ -81,13 +81,18 @@ class ServerlessStepFunctionsLocal {
     this.stateMachines = parsed.stepFunctions.stateMachines;
   }
 
-  createEndpoints() {
-    return Promise.all(Object.keys(this.stateMachines).map(stateMachineName => this.stepfunctionsAPI.createStateMachine({
+  async createEndpoints() {
+    const endpoints = await Promise.all(Object.keys(this.stateMachines).map(stateMachineName => this.stepfunctionsAPI.createStateMachine({
       definition: JSON.stringify(this.stateMachines[stateMachineName].definition),
       name: stateMachineName,
       roleArn: `arn:aws:iam::${this.config.accountId}:role/DummyRole`
     }).promise()
     ));
+
+    // Set environment variables with references to ARNs
+    endpoints.forEach(endpoint => {
+      process.env[`OFFLINE_STEP_FUNCTIONS_ARN_${endpoint.stateMachineArn.split(':')[6]}`] = endpoint.stateMachineArn;
+    });
   }
 }