From 8ea7d5b80ee06e424af88b395c7e59d13b405110 Mon Sep 17 00:00:00 2001
From: Ashish Jain <ashishjain0512@gmail.com>
Date: Thu, 4 Jul 2024 21:44:06 +0200
Subject: [PATCH] fix for direction

---
 packages/mermaid/src/diagrams/state/stateDb.js | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/packages/mermaid/src/diagrams/state/stateDb.js b/packages/mermaid/src/diagrams/state/stateDb.js
index 1f12425e6a..799c356000 100644
--- a/packages/mermaid/src/diagrams/state/stateDb.js
+++ b/packages/mermaid/src/diagrams/state/stateDb.js
@@ -48,7 +48,6 @@ function newClassesList() {
 
 let nodes = [];
 let edges = [];
-
 let direction = DEFAULT_DIAGRAM_DIRECTION;
 let rootDoc = [];
 let classes = newClassesList(); // style classes defined by a classDef
@@ -222,6 +221,7 @@ const extract = (_doc) => {
   const look = config.look;
   resetDataFetching();
   dataFetcher(undefined, getRootDocV2(), diagramStates, nodes, edges, true, look, classes);
+
   nodes.forEach((node) => {
     if (Array.isArray(node.label)) {
       // add the rest as description
@@ -237,6 +237,8 @@ const extract = (_doc) => {
       node.label = node.label[0];
     }
   });
+
+  direction = getDir(getRootDocV2());
 };
 
 /**
@@ -589,7 +591,7 @@ const trimColon = (str) => (str && str[0] === ':' ? str.substr(1).trim() : str.t
 
 export const getData = () => {
   const config = getConfig();
-  return { nodes, edges, other: {}, config, direction: getDir(getRootDocV2()) };
+  return { nodes, edges, other: {}, config, direction };
 };
 
 export default {