chore: Minor refactor

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

View File

@@ -2,4 +2,4 @@
'mermaid': patch '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

View File

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