mermaid/src/diagrams/class/classDb.js

import { logger } from '../../logger';

let relations = [];
let classes = {};

/**
 * Function called by parser when a node definition has been found.
 * @param id
 */
export const addClass = function(id) {
  if (typeof classes[id] === 'undefined') {
    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);
};

export const addAnnotation = function(className, annotation) {
  classes[className].annotations.push(annotation);
};

export const addMember = function(className, member) {
  const theClass = classes[className];
  if (typeof member === 'string') {
    const memberString = member.trim();
    if (memberString.startsWith('<<') && memberString.endsWith('>>')) {
      theClass.annotations.push(memberString.substring(2, memberString.length - 2));
    } else if (memberString.endsWith(')')) {
      theClass.methods.push(memberString);
    } else {
      theClass.members.push(memberString);
    }
  }
};

export const addMembers = function(className, MembersArr) {
  if (Array.isArray(MembersArr)) {
    MembersArr.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
};