schema-to-erd/dist/cjs/plantuml_table.js

Generate ERD UML file from Schema DDL file

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
function generateEntity(tableName, columnNames, primaryKeys) {
    const columnNamesWithPk = columnNames.map((columnName) => (primaryKeys.includes(columnName) ? `*${columnName}` : columnName));
    const columnsStr = columnNamesWithPk.join('\n  ');
    return `entity ${tableName} {
  ${columnsStr}
}
`;
}
function generateRelation(tableName, columnNames, primaryKeys, allPKs) {
    const allPkNames = Object.keys(allPKs);
    return columnNames.reduce((acc, columnName) => {
        if (!primaryKeys.includes(columnName) && allPkNames.includes(columnName)) {
            const foreignKey = allPKs[columnName];
            return [...acc, `${tableName}::${columnName} --> ${foreignKey.tableName}::${foreignKey.columnName}`];
        }
        return acc;
    }, []);
}
function generatePlantUml(tableColumns) {
    const entities = Object.entries(tableColumns).reduce((acc, [tableName, { columnNames, primaryKeys }]) => (acc + generateEntity(tableName, columnNames, primaryKeys)), '');
    const allPKs = Object.entries(tableColumns).reduce((accTable, [tableName, { primaryKeys }]) => primaryKeys.reduce((accPk, primaryKey) => {
        const primaryKeyName = primaryKey.startsWith(tableName) ? primaryKey : `${tableName}_${primaryKey}`;
        const foreignKey = { tableName, columnName: primaryKey };
        return Object.assign(Object.assign({}, accPk), { [primaryKeyName]: foreignKey });
    }, accTable), {});
    const relations = Object.entries(tableColumns).reduce((acc, [tableName, { columnNames, primaryKeys }]) => [
        ...acc, ...generateRelation(tableName, columnNames, primaryKeys, allPKs),
    ], []);
    // https://plantuml.com/ko/ie-diagram
    return `@startuml

' hide the spot
hide circle
hide methods
hide stereotypes

' avoid problems with angled crows feet
skinparam linetype ortho

${entities}

${relations.join('\n')}

@enduml
`;
}
exports.default = generatePlantUml;
//# sourceMappingURL=plantuml_table.js.map