marcel/mermaid
1
0
Fork 0
mirror of https://github.com/mermaid-js/mermaid.git synced 2025-08-15 06:19:24 +02:00
Code Issues Packages Projects Releases Wiki Activity

chore: Minor refactor

Browse Source
This commit is contained in:
Sidharth Vinod
2025-02-18 23:48:19 +05:30
parent c3ba23a4e9
commit d846a8c65a
2 changed files with 14 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
.changeset/weak-trees-perform.md
View File

@@ -2,4 +2,4 @@
'mermaid': patch
---
In State Diagram refactored getDirection to retrieve the direction from rootDoc, defaulting to DEFAULT_DIAGRAM_DIRECTION and Updated setDirection to modify the existing direction if found or prepend a new entry if not.
fix: `getDirection` and `setDirection` in `stateDb` refactored to return and set actual direction

18
packages/mermaid/src/diagrams/state/stateDb.js
View File

@@ -657,13 +657,21 @@ export class StateDB {
}
}
getDirection() {
const doc = this.rootDoc.find((doc) => doc.stmt === STMT_DIRECTION);
const direction = doc ? doc.value : DEFAULT_DIAGRAM_DIRECTION;
return direction;
/**
* Finds the direction statement in the root document.
* @private
* @returns {{ value: string } | undefined} - the direction statement if present
*/
getDirectionStatement() {
return this.rootDoc.find((doc) => doc.stmt === STMT_DIRECTION);
}
getDirection() {
return this.getDirectionStatement()?.value ?? DEFAULT_DIAGRAM_DIRECTION;
}
setDirection(dir) {
const doc = this.rootDoc.find((doc) => doc.stmt === STMT_DIRECTION);
const doc = this.getDirectionStatement();
if (doc) {
doc.value = dir;
} else {