mirror of
https://github.com/mermaid-js/mermaid.git
synced 2025-02-04 07:13:25 +08:00
124 lines
2.8 KiB
JavaScript
124 lines
2.8 KiB
JavaScript
import { logger } from '../../logger';
|
|
|
|
let relations = [];
|
|
let classes = {};
|
|
|
|
/**
|
|
* Function called by parser when a node definition has been found.
|
|
* @param id
|
|
* @public
|
|
*/
|
|
export const addClass = function(id) {
|
|
// Only add class if not exists
|
|
if (typeof classes[id] !== 'undefined') return;
|
|
|
|
classes[id] = {
|
|
id: id,
|
|
methods: [],
|
|
members: [],
|
|
annotations: []
|
|
};
|
|
};
|
|
|
|
export const clear = function() {
|
|
relations = [];
|
|
classes = {};
|
|
};
|
|
|
|
export const getClass = function(id) {
|
|
return classes[id];
|
|
};
|
|
export const getClasses = function() {
|
|
return classes;
|
|
};
|
|
|
|
export const getRelations = function() {
|
|
return relations;
|
|
};
|
|
|
|
export const addRelation = function(relation) {
|
|
logger.debug('Adding relation: ' + JSON.stringify(relation));
|
|
addClass(relation.id1);
|
|
addClass(relation.id2);
|
|
relations.push(relation);
|
|
};
|
|
|
|
/**
|
|
* Adds an annotation to the specified class
|
|
* Annotations mark special properties of the given type (like 'interface' or 'service')
|
|
* @param className The class name
|
|
* @param annotation The name of the annotation without any brackets
|
|
* @public
|
|
*/
|
|
export const addAnnotation = function(className, annotation) {
|
|
classes[className].annotations.push(annotation);
|
|
};
|
|
|
|
/**
|
|
* Adds a member to the specified class
|
|
* @param className The class name
|
|
* @param member The full name of the member.
|
|
* If the member is enclosed in <<brackets>> it is treated as an annotation
|
|
* If the member is ending with a closing bracket ) it is treated as a method
|
|
* Otherwise the member will be treated as a normal property
|
|
* @public
|
|
*/
|
|
export const addMember = function(className, member) {
|
|
const theClass = classes[className];
|
|
if (typeof member === 'string') {
|
|
// Member can contain white spaces, we trim them out
|
|
const memberString = member.trim();
|
|
|
|
if (memberString.startsWith('<<') && memberString.endsWith('>>')) {
|
|
// Remove leading and trailing brackets
|
|
theClass.annotations.push(memberString.substring(2, memberString.length - 2));
|
|
} else if (memberString.endsWith(')')) {
|
|
theClass.methods.push(memberString);
|
|
} else if (memberString) {
|
|
theClass.members.push(memberString);
|
|
}
|
|
}
|
|
};
|
|
|
|
export const addMembers = function(className, members) {
|
|
if (Array.isArray(members)) {
|
|
members.reverse();
|
|
members.forEach(member => addMember(className, member));
|
|
}
|
|
};
|
|
|
|
export const cleanupLabel = function(label) {
|
|
if (label.substring(0, 1) === ':') {
|
|
return label.substr(2).trim();
|
|
} else {
|
|
return label.trim();
|
|
}
|
|
};
|
|
|
|
export const lineType = {
|
|
LINE: 0,
|
|
DOTTED_LINE: 1
|
|
};
|
|
|
|
export const relationType = {
|
|
AGGREGATION: 0,
|
|
EXTENSION: 1,
|
|
COMPOSITION: 2,
|
|
DEPENDENCY: 3
|
|
};
|
|
|
|
export default {
|
|
addClass,
|
|
clear,
|
|
getClass,
|
|
getClasses,
|
|
addAnnotation,
|
|
getRelations,
|
|
addRelation,
|
|
addMember,
|
|
addMembers,
|
|
cleanupLabel,
|
|
lineType,
|
|
relationType
|
|
};
|