From b78c061a8f5e736cee0f5daebb6fa3017b7a35b2 Mon Sep 17 00:00:00 2001
From: omkarht <omkar@mermaidchart.com>
Date: Tue, 18 Feb 2025 15:40:45 +0530
Subject: [PATCH] refatored code

---
 .../mermaid/src/diagrams/state/stateCommon.ts    |  5 ++++-
 packages/mermaid/src/diagrams/state/stateDb.js   | 16 ++++++----------
 2 files changed, 10 insertions(+), 11 deletions(-)

diff --git a/packages/mermaid/src/diagrams/state/stateCommon.ts b/packages/mermaid/src/diagrams/state/stateCommon.ts
index 17a1bd24a..2902ce6b0 100644
--- a/packages/mermaid/src/diagrams/state/stateCommon.ts
+++ b/packages/mermaid/src/diagrams/state/stateCommon.ts
@@ -3,11 +3,14 @@
  */
 
 // default diagram direction
-export const DEFAULT_DIAGRAM_DIRECTION = 'LR';
+export const DEFAULT_DIAGRAM_DIRECTION = 'TB';
 
 // default direction for any nested documents (composites)
 export const DEFAULT_NESTED_DOC_DIR = 'TB';
 
+// parsed statement type for a direction
+export const STMT_DIRECTION = 'dir';
+
 // parsed statement type for a state
 export const STMT_STATE = 'state';
 // parsed statement type for a relation
diff --git a/packages/mermaid/src/diagrams/state/stateDb.js b/packages/mermaid/src/diagrams/state/stateDb.js
index 37cea295f..b6b5d11ff 100644
--- a/packages/mermaid/src/diagrams/state/stateDb.js
+++ b/packages/mermaid/src/diagrams/state/stateDb.js
@@ -20,6 +20,7 @@ import {
   DIVIDER_TYPE,
   STMT_APPLYCLASS,
   STMT_CLASSDEF,
+  STMT_DIRECTION,
   STMT_RELATION,
   STMT_STATE,
   STMT_STYLEDEF,
@@ -79,11 +80,6 @@ export class StateDB {
    */
   edges = [];
 
-  /**
-   * @private
-   * @type {string}
-   */
-  direction = DEFAULT_DIAGRAM_DIRECTION;
   /**
    * @private
    * @type {Array}
@@ -643,16 +639,16 @@ export class StateDB {
   }
 
   getDirection() {
-    const doc = this.rootDoc.find((doc) => doc.stmt === 'dir');
-    this.direction = doc ? doc.value : 'TB';
-    return this.direction;
+    const doc = this.rootDoc.find((doc) => doc.stmt === STMT_DIRECTION);
+    const direction = doc ? doc.value : DEFAULT_DIAGRAM_DIRECTION;
+    return direction;
   }
   setDirection(dir) {
-    let doc = this.rootDoc.find((doc) => doc.stmt === 'dir');
+    const doc = this.rootDoc.find((doc) => doc.stmt === STMT_DIRECTION);
     if (doc) {
       doc.value = dir;
     } else {
-      this.rootDoc.unshift({ stmt: 'dir', value: dir });
+      this.rootDoc.unshift({ stmt: STMT_DIRECTION, value: dir });
     }
   }