mermaid/src/diagrams/classDiagram/classDb.js

var Logger = require('../../logger');
var log = Logger.Log;
var relations = [];

var classes;
var idCache;
classes = {
};

// Functions to be run after graph rendering
var funs = [];
/**
 * Function called by parser when a node definition has been found.
 * @param id
 * @param text
 * @param type
 * @param style
 */
exports.addClass = function (id) {
    if(typeof classes[id] === 'undefined'){
        classes[id] = {
            id:id,
            methods:[],
            members:[]
        };
    }
};

exports.clear = function () {
    relations = [];
    classes = {};
};

module.exports.getClass = function (id) {
    return classes[id];
};
module.exports.getClasses = function () {
    return classes;
};

module.exports.getRelations = function () {
    return relations;
};

exports.addRelation = function (relation) {
    log.warn('Adding relation: ' + JSON.stringify(relation));
    exports.addClass(relation.id1);
    exports.addClass(relation.id2);

    relations.push(relation);
};

exports.addMembers = function (className, MembersArr) {
    var theClass = classes[className];
    if(typeof MembersArr === 'string'){
        if(MembersArr.substr(-1) === ')'){
            theClass.methods.push(MembersArr);
        }
        else{
            theClass.members.push(MembersArr);
        }
    }
};

exports.cleanupLabel = function (label) {

    if(label.substring(0,1) === ':'){
        return label.substr(2).trim();
    }
    else{
        return label.trim();
    }
};

exports.lineType = {
    LINE:0,
    DOTTED_LINE:1
};

exports.relationType = {
    AGGREGATION:0,
    EXTENSION:1,
    COMPOSITION:2,
    DEPENDENCY:3
};