Merge branch 'master' of github.com:evennode/serverless-step-functions-local

Author: ceecko

index.js

@@ -58,7 +58,7 @@ class ServerlessStepFunctionsLocal {
 
         const bootstrap = (async () => {
           await this.startStepFunctions();
-          await this.getStepFunctionsFromConfig();
+          await this.getStepFunctionsFromConfigurationInput();
           await this.createEndpoints();
         })()
 
@@ -103,24 +103,17 @@ class ServerlessStepFunctionsLocal {
     return this.stepfunctionsServer.stop();
   }
 
-  async getStepFunctionsFromConfig() {
-    const {servicePath} = this.serverless.config;
-
-    if (!servicePath) {
-      throw new Error('service path not found');
-    }
-
-    const configPath = path.join(servicePath, 'serverless.yml');
-
-    const preParsed = await this.serverless.yamlParser.parse(configPath);
-    const parsed = await this.serverless.variables.populateObject(preParsed);
-
-    this.stateMachines = this.stateMachineCFARNResolver(parsed.stepFunctions.stateMachines);
-
-    if (parsed.custom
-      && parsed.custom.stepFunctionsLocal
-      && parsed.custom.stepFunctionsLocal.TaskResourceMapping) {
-        this.replaceTaskResourceMappings(parsed.stepFunctions.stateMachines, parsed.custom.stepFunctionsLocal.TaskResourceMapping);
+  async getStepFunctionsFromConfigurationInput() {
+    this.stateMachines = this.stateMachineCFARNResolver(this.serverless.configurationInput.stepFunctions.stateMachines);
+
+    if (this.serverless.configurationInput.custom
+      && this.serverless.configurationInput.custom.stepFunctionsLocal
+      && this.serverless.configurationInput.custom.stepFunctionsLocal.TaskResourceMapping
+    ) {
+        this.replaceTaskResourceMappings(
+          this.serverless.configurationInput.stepFunctions.stateMachines,
+          this.serverless.configurationInput.custom.stepFunctionsLocal.TaskResourceMapping
+        );
     }
   }