Strip whitespaces from column types

[Helveg]

Currently the following entity (among many other types):

import { CreateDateColumn, Entity, PrimaryColumn } from 'typeorm';

@Entity()
export SomeEntity {
  @PrimaryGeneratedColumn('uuid')
  id!: string;

  @CreateDateColumn()
  createdAt!: Date;
}

renders the following invalid mermaid code:

erDiagram
  some_entity {
    character varying id PK
    timestamp without time zone createdAt
  }

Loading

because the type of the id column is rendered as character varying (with whitespace). There's also timestamp without time zone, timestamp with time zone, character varying, double precision, and likely others.

[aktary]

This bug renders the package basically useless. I'll take a look at putting in a PR.

FWIW, I think this is due to the difference between how sqlite3 and postgres report metadata

[Nabhag8848]

for postgres: cleaning up types worked:

import { writeFileSync } from 'fs';
import { join } from 'path';
import { ERDBuilder } from 'typeorm-erd';
import { AppDataSource } from '../datasource/app.datasource';

const cleanPostgresTypes = (mermaid: string): string => {
  return mermaid
    .replace(/character varying/g, 'varchar')
    .replace(/timestamp without time zone/g, 'timestamp')
    .replace(/timestamp with time zone/g, 'timestamptz')
    .replace(/double precision/g, 'float8')
    .replace(/integer/g, 'int')
    .replace(/bigint/g, 'int8')
    .replace(/smallint/g, 'int2')
    .replace(/boolean/g, 'bool')
    .replace(/text/g, 'text');
};

const main = async () => {
  if (!AppDataSource.isInitialized) {
    await AppDataSource.initialize();
  }

  const erd = new ERDBuilder('mermaid', AppDataSource);
  await erd.initialize();
  const mermaidErd = await erd.render();
  const cleanedErd = cleanPostgresTypes(mermaidErd);

  const readmeContent = `# Server

## Database ERD

\`\`\`mermaid
${cleanedErd}
\`\`\`
`;

  writeFileSync(join(process.cwd(), 'README.md'), readmeContent);

  if (AppDataSource.isInitialized) {
    await AppDataSource.destroy();
  }
};

main().catch((error) => {
  console.error(error);
  process.exit(1);
});