GitHub/mermaid
1
0
Fork 0
mirror of https://github.com/mermaid-js/mermaid.git synced 2025-02-04 07:13:25 +08:00
Code Issues Projects Releases Wiki Activity

Updated as per Alois suggestion

Browse Source
This commit is contained in:
saurabhg772244 2025-01-22 15:49:51 +05:30
parent ef9bb53e67
commit bde653b1c2
1 changed files with 165 additions and 62 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -26,17 +26,55 @@ import {
} from './stateCommon.js'; } from './stateCommon.js';
export class StateDB { export class StateDB {
/**
* @private
* @type {string}
*/
#START_NODE = '[*]'; #START_NODE = '[*]';
/**
* @private
* @type {string}
*/
#START_TYPE = 'start'; #START_TYPE = 'start';
/**
* @private
* @type {string}
*/
#END_NODE = this.#START_NODE; #END_NODE = this.#START_NODE;
/**
* @private
* @type {string}
*/
#END_TYPE = 'end'; #END_TYPE = 'end';
/**
* @private
* @type {string}
*/
#COLOR_KEYWORD = 'color'; #COLOR_KEYWORD = 'color';
/**
* @private
* @type {string}
*/
#FILL_KEYWORD = 'fill'; #FILL_KEYWORD = 'fill';
/**
* @private
* @type {string}
*/
#BG_FILL = 'bgFill'; #BG_FILL = 'bgFill';
/**
* @private
* @type {string}
*/
#STYLECLASS_SEP = ','; #STYLECLASS_SEP = ',';
constructor() { constructor() {
this.clear(); this.clear();
// Needed for JISON since it only supports direct properties
this.setRootDoc = this.setRootDoc.bind(this);
this.getDividerId = this.getDividerId.bind(this);
this.setDirection = this.setDirection.bind(this);
this.trimColon = this.trimColon.bind(this);
} }
/** /**
@ -45,35 +83,76 @@ export class StateDB {
* ClassDef information = { id: id, styles: [], textStyles: [] } * ClassDef information = { id: id, styles: [], textStyles: [] }
* *
* @returns {Map<string, any>} * @returns {Map<string, any>}
* @private
*/ */
#newClassesList = () => { #newClassesList() {
return new Map(); return new Map();
}; }
/**
* @private
* @type {Array}
*/
#nodes = []; #nodes = [];
/**
* @private
* @type {Array}
*/
#edges = []; #edges = [];
/**
* @private
* @type {string}
*/
#direction = DEFAULT_DIAGRAM_DIRECTION; #direction = DEFAULT_DIAGRAM_DIRECTION;
/**
* @private
* @type {Array}
*/
#rootDoc = []; #rootDoc = [];
/**
* @private
* @type {Map<string, any>}
*/
#classes = this.#newClassesList(); // style classes defined by a classDef #classes = this.#newClassesList(); // style classes defined by a classDef
// -------------------------------------- // --------------------------------------
#newDoc = () => { /**
* @private
* @returns {Object}
*/
#newDoc() {
return { return {
/** @type {{ id1: string, id2: string, relationTitle: string }[]} */ /** @type {{ id1: string, id2: string, relationTitle: string }[]} */
relations: [], relations: [],
states: new Map(), states: new Map(),
documents: {}, documents: {},
}; };
}; }
/**
* @private
* @type {Object}
*/
#documents = { #documents = {
root: this.#newDoc(), root: this.#newDoc(),
}; };
/**
* @private
* @type {Object}
*/
#currentDocument = this.#documents.root; #currentDocument = this.#documents.root;
/**
* @private
* @type {number}
*/
#startEndCount = 0; #startEndCount = 0;
/**
* @private
* @type {number}
*/
#dividerCnt = 0; #dividerCnt = 0;
lineType = { lineType = {
@ -88,17 +167,31 @@ export class StateDB {
DEPENDENCY: 3, DEPENDENCY: 3,
}; };
#clone = (o) => JSON.parse(JSON.stringify(o)); /**
* @private
* @param {Object} o
*/
#clone(o) {
return JSON.parse(JSON.stringify(o));
}
setRootDoc = (o) => { setRootDoc(o) {
log.info('Setting root doc', o); log.info('Setting root doc', o);
// rootDoc = { id: 'root', doc: o }; // rootDoc = { id: 'root', doc: o };
this.#rootDoc = o; this.#rootDoc = o;
}; }
getRootDoc = () => this.#rootDoc; getRootDoc() {
return this.#rootDoc;
}
#docTranslator = (parent, node, first) => { /**
* @private
* @param {Object} parent
* @param {Object} node
* @param {boolean} first
*/
#docTranslator(parent, node, first) {
if (node.stmt === STMT_RELATION) { if (node.stmt === STMT_RELATION) {
this.#docTranslator(parent, node.state1, true); this.#docTranslator(parent, node.state1, true);
this.#docTranslator(parent, node.state2, false); this.#docTranslator(parent, node.state2, false);
@ -144,12 +237,12 @@ export class StateDB {
node.doc.forEach((docNode) => this.#docTranslator(node, docNode, true)); node.doc.forEach((docNode) => this.#docTranslator(node, docNode, true));
} }
} }
}; }
getRootDocV2 = () => { getRootDocV2() {
this.#docTranslator({ id: 'root' }, { id: 'root', doc: this.#rootDoc }, true); this.#docTranslator({ id: 'root' }, { id: 'root', doc: this.#rootDoc }, true);
return { id: 'root', doc: this.#rootDoc }; return { id: 'root', doc: this.#rootDoc };
// Here // Here
}; }
/** /**
* Convert all of the statements (stmts) that were parsed into states and relationships. * Convert all of the statements (stmts) that were parsed into states and relationships.
@ -162,7 +255,7 @@ export class StateDB {
* *
* @param _doc * @param _doc
*/ */
extract = (_doc) => { extract(_doc) {
// const res = { states: [], relations: [] }; // const res = { states: [], relations: [] };
let doc; let doc;
if (_doc.doc) { if (_doc.doc) {
@ -251,7 +344,7 @@ export class StateDB {
node.label = node.label[0]; node.label = node.label[0];
} }
}); });
}; }
/** /**
* Function called by parser when a node definition has been found. * Function called by parser when a node definition has been found.
@ -265,7 +358,7 @@ export class StateDB {
* @param {null | string | string[]} styles - styles to apply to this state. Can be a string (1 style) or an array of styles. If it's just 1 style, convert it to an array of that 1 style. * @param {null | string | string[]} styles - styles to apply to this state. Can be a string (1 style) or an array of styles. If it's just 1 style, convert it to an array of that 1 style.
* @param {null | string | string[]} textStyles - text styles to apply to this state. Can be a string (1 text test) or an array of text styles. If it's just 1 text style, convert it to an array of that 1 text style. * @param {null | string | string[]} textStyles - text styles to apply to this state. Can be a string (1 text test) or an array of text styles. If it's just 1 text style, convert it to an array of that 1 text style.
*/ */
addState = ( addState(
id, id,
type = DEFAULT_STATE_TYPE, type = DEFAULT_STATE_TYPE,
doc = null, doc = null,
@ -274,7 +367,7 @@ export class StateDB {
classes = null, classes = null,
styles = null, styles = null,
textStyles = null textStyles = null
) => { ) {
const trimmedId = id?.trim(); const trimmedId = id?.trim();
// add the state if needed // add the state if needed
if (!this.#currentDocument.states.has(trimmedId)) { if (!this.#currentDocument.states.has(trimmedId)) {
@ -332,9 +425,9 @@ export class StateDB {
const textStylesList = typeof textStyles === 'string' ? [textStyles] : textStyles; const textStylesList = typeof textStyles === 'string' ? [textStyles] : textStyles;
textStylesList.forEach((textStyle) => this.setTextStyle(trimmedId, textStyle.trim())); textStylesList.forEach((textStyle) => this.setTextStyle(trimmedId, textStyle.trim()));
} }
}; }
clear = (saveCommon) => { clear(saveCommon) {
this.#nodes = []; this.#nodes = [];
this.#edges = []; this.#edges = [];
this.#documents = { this.#documents = {
@ -348,20 +441,20 @@ export class StateDB {
if (!saveCommon) { if (!saveCommon) {
commonClear(); commonClear();
} }
}; }
getState = (id) => { getState(id) {
return this.#currentDocument.states.get(id); return this.#currentDocument.states.get(id);
}; }
getStates = () => { getStates() {
return this.#currentDocument.states; return this.#currentDocument.states;
}; }
logDocuments = () => { logDocuments() {
log.info('Documents = ', this.#documents); log.info('Documents = ', this.#documents);
}; }
getRelations = () => { getRelations() {
return this.#currentDocument.relations; return this.#currentDocument.relations;
}; }
/** /**
* If the id is a start node ( [*] ), then return a new id constructed from * If the id is a start node ( [*] ), then return a new id constructed from
@ -370,15 +463,16 @@ export class StateDB {
* *
* @param {string} id * @param {string} id
* @returns {string} - the id (original or constructed) * @returns {string} - the id (original or constructed)
* @private
*/ */
#startIdIfNeeded = (id = '') => { #startIdIfNeeded(id = '') {
let fixedId = id; let fixedId = id;
if (id === this.#START_NODE) { if (id === this.#START_NODE) {
this.#startEndCount++; this.#startEndCount++;
fixedId = `${this.#START_TYPE}${this.#startEndCount}`; fixedId = `${this.#START_TYPE}${this.#startEndCount}`;
} }
return fixedId; return fixedId;
}; }
/** /**
* If the id is a start node ( [*] ), then return the start type ('start') * If the id is a start node ( [*] ), then return the start type ('start')
@ -387,10 +481,11 @@ export class StateDB {
* @param {string} id * @param {string} id
* @param {string} type * @param {string} type
* @returns {string} - the type that should be used * @returns {string} - the type that should be used
* @private
*/ */
#startTypeIfNeeded = (id = '', type = DEFAULT_STATE_TYPE) => { #startTypeIfNeeded(id = '', type = DEFAULT_STATE_TYPE) {
return id === this.#START_NODE ? this.#START_TYPE : type; return id === this.#START_NODE ? this.#START_TYPE : type;
}; }
/** /**
* If the id is an end node ( [*] ), then return a new id constructed from * If the id is an end node ( [*] ), then return a new id constructed from
@ -399,15 +494,16 @@ export class StateDB {
* *
* @param {string} id * @param {string} id
* @returns {string} - the id (original or constructed) * @returns {string} - the id (original or constructed)
* @private
*/ */
#endIdIfNeeded = (id = '') => { #endIdIfNeeded(id = '') {
let fixedId = id; let fixedId = id;
if (id === this.#END_NODE) { if (id === this.#END_NODE) {
this.#startEndCount++; this.#startEndCount++;
fixedId = `${this.#END_TYPE}${this.#startEndCount}`; fixedId = `${this.#END_TYPE}${this.#startEndCount}`;
} }
return fixedId; return fixedId;
}; }
/** /**
* If the id is an end node ( [*] ), then return the end type * If the id is an end node ( [*] ), then return the end type
@ -416,10 +512,11 @@ export class StateDB {
* @param {string} id * @param {string} id
* @param {string} type * @param {string} type
* @returns {string} - the type that should be used * @returns {string} - the type that should be used
* @private
*/ */
#endTypeIfNeeded = (id = '', type = DEFAULT_STATE_TYPE) => { #endTypeIfNeeded(id = '', type = DEFAULT_STATE_TYPE) {
return id === this.#END_NODE ? this.#END_TYPE : type; return id === this.#END_NODE ? this.#END_TYPE : type;
}; }
/** /**
* *
@ -427,7 +524,7 @@ export class StateDB {
* @param item2 * @param item2
* @param relationTitle * @param relationTitle
*/ */
addRelationObjs = (item1, item2, relationTitle) => { addRelationObjs(item1, item2, relationTitle) {
let id1 = this.#startIdIfNeeded(item1.id.trim()); let id1 = this.#startIdIfNeeded(item1.id.trim());
let type1 = this.#startTypeIfNeeded(item1.id.trim(), item1.type); let type1 = this.#startTypeIfNeeded(item1.id.trim(), item1.type);
let id2 = this.#startIdIfNeeded(item2.id.trim()); let id2 = this.#startIdIfNeeded(item2.id.trim());
@ -459,7 +556,7 @@ export class StateDB {
id2, id2,
relationTitle: common.sanitizeText(relationTitle, getConfig()), relationTitle: common.sanitizeText(relationTitle, getConfig()),
}); });
}; }
/** /**
* Add a relation between two items. The items may be full objects or just the string id of a state. * Add a relation between two items. The items may be full objects or just the string id of a state.
@ -468,7 +565,7 @@ export class StateDB {
* @param {string | object} item2 * @param {string | object} item2
* @param {string} title * @param {string} title
*/ */
addRelation = (item1, item2, title) => { addRelation(item1, item2, title) {
if (typeof item1 === 'object') { if (typeof item1 === 'object') {
this.addRelationObjs(item1, item2, title); this.addRelationObjs(item1, item2, title);
} else { } else {
@ -485,26 +582,26 @@ export class StateDB {
title: common.sanitizeText(title, getConfig()), title: common.sanitizeText(title, getConfig()),
}); });
} }
}; }
addDescription = (id, descr) => { addDescription(id, descr) {
const theState = this.#currentDocument.states.get(id); const theState = this.#currentDocument.states.get(id);
const _descr = descr.startsWith(':') ? descr.replace(':', '').trim() : descr; const _descr = descr.startsWith(':') ? descr.replace(':', '').trim() : descr;
theState.descriptions.push(common.sanitizeText(_descr, getConfig())); theState.descriptions.push(common.sanitizeText(_descr, getConfig()));
}; }
cleanupLabel = (label) => { cleanupLabel(label) {
if (label.substring(0, 1) === ':') { if (label.substring(0, 1) === ':') {
return label.substr(2).trim(); return label.substr(2).trim();
} else { } else {
return label.trim(); return label.trim();
} }
}; }
getDividerId = () => { getDividerId() {
this.#dividerCnt++; this.#dividerCnt++;
return 'divider-id-' + this.#dividerCnt; return 'divider-id-' + this.#dividerCnt;
}; }
/** /**
* Called when the parser comes across a (style) class definition * Called when the parser comes across a (style) class definition
@ -513,7 +610,7 @@ export class StateDB {
* @param {string} id - the id of this (style) class * @param {string} id - the id of this (style) class
* @param {string | null} styleAttributes - the string with 1 or more style attributes (each separated by a comma) * @param {string | null} styleAttributes - the string with 1 or more style attributes (each separated by a comma)
*/ */
addStyleClass = (id, styleAttributes = '') => { addStyleClass(id, styleAttributes = '') {
// create a new style class object with this id // create a new style class object with this id
if (!this.#classes.has(id)) { if (!this.#classes.has(id)) {
this.#classes.set(id, { id: id, styles: [], textStyles: [] }); // This is a classDef this.#classes.set(id, { id: id, styles: [], textStyles: [] }); // This is a classDef
@ -533,15 +630,15 @@ export class StateDB {
foundClass.styles.push(fixedAttrib); foundClass.styles.push(fixedAttrib);
}); });
} }
}; }
/** /**
* Return all of the style classes * Return all of the style classes
* @returns {{} | any | classes} * @returns {{} | any | classes}
*/ */
getClasses = () => { getClasses() {
return this.#classes; return this.#classes;
}; }
/** /**
* Add a (style) class or css class to a state with the given id. * Add a (style) class or css class to a state with the given id.
@ -551,7 +648,7 @@ export class StateDB {
* @param {string | string[]} itemIds The id or a list of ids of the item(s) to apply the css class to * @param {string | string[]} itemIds The id or a list of ids of the item(s) to apply the css class to
* @param {string} cssClassName CSS class name * @param {string} cssClassName CSS class name
*/ */
setCssClass = (itemIds, cssClassName) => { setCssClass(itemIds, cssClassName) {
itemIds.split(',').forEach((id) => { itemIds.split(',').forEach((id) => {
let foundState = this.getState(id); let foundState = this.getState(id);
if (foundState === undefined) { if (foundState === undefined) {
@ -561,7 +658,7 @@ export class StateDB {
} }
foundState.classes.push(cssClassName); foundState.classes.push(cssClassName);
}); });
}; }
/** /**
* Add a style to a state with the given id. * Add a style to a state with the given id.
@ -573,12 +670,12 @@ export class StateDB {
* @param itemId The id of item to apply the style to * @param itemId The id of item to apply the style to
* @param styleText - the text of the attributes for the style * @param styleText - the text of the attributes for the style
*/ */
setStyle = (itemId, styleText) => { setStyle(itemId, styleText) {
const item = this.getState(itemId); const item = this.getState(itemId);
if (item !== undefined) { if (item !== undefined) {
item.styles.push(styleText); item.styles.push(styleText);
} }
}; }
/** /**
* Add a text style to a state with the given id * Add a text style to a state with the given id
@ -586,21 +683,25 @@ export class StateDB {
* @param itemId The id of item to apply the css class to * @param itemId The id of item to apply the css class to
* @param cssClassName CSS class name * @param cssClassName CSS class name
*/ */
setTextStyle = (itemId, cssClassName) => { setTextStyle(itemId, cssClassName) {
const item = this.getState(itemId); const item = this.getState(itemId);
if (item !== undefined) { if (item !== undefined) {
item.textStyles.push(cssClassName); item.textStyles.push(cssClassName);
} }
}; }
getDirection = () => this.#direction; getDirection() {
setDirection = (dir) => { return this.#direction;
}
setDirection(dir) {
this.#direction = dir; this.#direction = dir;
}; }
trimColon = (str) => (str && str[0] === ':' ? str.substr(1).trim() : str.trim()); trimColon(str) {
return str && str[0] === ':' ? str.substr(1).trim() : str.trim();
}
getData = () => { getData() {
const config = getConfig(); const config = getConfig();
return { return {
nodes: this.#nodes, nodes: this.#nodes,
@ -609,9 +710,11 @@ export class StateDB {
config, config,
direction: getDir(this.getRootDocV2()), direction: getDir(this.getRootDocV2()),
}; };
}; }
getConfig = () => getConfig().state; getConfig() {
return getConfig().state;
}
getAccTitle = getAccTitle; getAccTitle = getAccTitle;
setAccTitle = setAccTitle; setAccTitle = setAccTitle;
getAccDescription = getAccDescription; getAccDescription = getAccDescription;