marcel/mermaid
1
0
Fork 0
mirror of https://github.com/mermaid-js/mermaid.git synced 2025-09-25 18:30:10 +02:00
Code Issues Packages Projects Releases Wiki Activity

refatored code

Browse Source
This commit is contained in:
omkarht
2025-02-18 15:40:45 +05:30
parent 72eda9ce52
commit b78c061a8f
2 changed files with 10 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
packages/mermaid/src/diagrams/state/stateCommon.ts
View File

@@ -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

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

@@ -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 });
}
}