From 72eda9ce5221e318f1e081b8992ecb6e71dbc26d Mon Sep 17 00:00:00 2001
From: omkarht <omkar@mermaidchart.com>
Date: Mon, 17 Feb 2025 20:53:31 +0530
Subject: [PATCH] Fix: state diagram default direction update

---
 packages/mermaid/src/diagrams/state/stateDb.js | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/packages/mermaid/src/diagrams/state/stateDb.js b/packages/mermaid/src/diagrams/state/stateDb.js
index cc44659eb..37cea295f 100644
--- a/packages/mermaid/src/diagrams/state/stateDb.js
+++ b/packages/mermaid/src/diagrams/state/stateDb.js
@@ -643,10 +643,17 @@ export class StateDB {
   }
 
   getDirection() {
+    const doc = this.rootDoc.find((doc) => doc.stmt === 'dir');
+    this.direction = doc ? doc.value : 'TB';
     return this.direction;
   }
   setDirection(dir) {
-    this.direction = dir;
+    let doc = this.rootDoc.find((doc) => doc.stmt === 'dir');
+    if (doc) {
+      doc.value = dir;
+    } else {
+      this.rootDoc.unshift({ stmt: 'dir', value: dir });
+    }
   }
 
   trimColon(str) {