add map db name option

Author: marcjulian

README.md

@@ -145,6 +145,7 @@ Ref: Post.authorId > User.id
 | `output`              | Output directory for the DBML file              | `string`  | `./dbml`      |
 | `outputName`          | Name for the DBML file                          | `string`  | `dbml.schema` |
 | `manyToMany`          | Create Many-To-Many join table                  | `boolean` | `true`        |
+| `mapToDbSchema`       | Use mapped table name                           | `boolean` | `true`        |
 
 Use additional options in the `schema.prisma`
 

prisma/schema.prisma

@@ -2,26 +2,29 @@
 // learn more about it in the docs: https://pris.ly/d/prisma-schema
 
 datasource db {
-  provider = "sqlite"
+  provider = "mysql"
   url      = env("DATABASE_URL")
 }
 
 generator client {
-  provider        = "prisma-client-js"
+  provider = "prisma-client-js"
 }
 
 generator dbml {
-  provider = "node ./dist/generator.js"
+  provider    = "node ./dist/generator.js"
+  mapToDbSchema = "false"
 }
 
 model User {
   id        Int      @id @default(autoincrement())
-  createdAt DateTime @default(now())
-  updatedAt DateTime @updatedAt
+  createdAt DateTime @default(now()) @db.DateTime(0)
+  updatedAt DateTime @updatedAt @map("updated_at")
   email     String   @unique
-  name      String?
+  name      String?  @db.VarChar(255)
   posts     Post[]
   profile   Profile?
+
+  @@map("user")
 }
 
 /// User profile
@@ -30,6 +33,8 @@ model Profile {
   bio    String?
   user   User    @relation(fields: [userId], references: [id], onDelete: Cascade, onUpdate: Cascade)
   userId Int     @unique
+
+  @@map("profile")
 }
 
 model Post {

src/cli/dbml-generator.ts

@@ -14,6 +14,7 @@ export async function generate(options: GeneratorOptions) {
   const outputDir = parseEnvValue(output!);
   const dbmlFileName = config.outputName || defaultDBMLFileName;
   const allowManyToMany = config.manyToMany === 'false' ? false : true;
+  const mapToDbSchema = config.mapToDbSchema === 'false' ? false: true;
   const projectOptions = await getProjectOptions(config);
 
   try {
@@ -24,6 +25,7 @@ export async function generate(options: GeneratorOptions) {
 
     const dbmlSchema = generateDBMLSchema(
       options.dmmf,
+      mapToDbSchema,
       allowManyToMany,
       projectOptions
     );

src/generator/dbml.ts

@@ -11,10 +11,11 @@ export const autoGeneratedComment = `//// --------------------------------------
 
 export function generateDBMLSchema(
   dmmf: DMMF.Document,
+  mapToDbSchema: boolean,
   allowManyToMany: boolean = true,
   projectOptions?: ProjectOptions
 ): string {
-  const tables = generateTables(dmmf.datamodel.models);
+  const tables = generateTables(dmmf.datamodel.models, mapToDbSchema);
   const manyToManyTables = allowManyToMany
     ? generateManyToManyTables(dmmf.datamodel.models)
     : [];

src/generator/table.ts

@@ -1,15 +1,25 @@
 import { DBMLKeywords, PrismaScalars } from './../keywords';
 import { DMMF } from '@prisma/generator-helper';
 
-export function generateTables(models: DMMF.Model[]): string[] {
-  return models.map(
-    (model) =>
-      `${DBMLKeywords.Table} ${model.name} {\n` +
+export function generateTables(
+  models: DMMF.Model[],
+  mapToDbSchema: boolean
+): string[] {
+  return models.map((model) => {
+    let modelName = model.name;
+
+    if (mapToDbSchema && model.dbName) {
+      modelName = model.dbName;
+    }
+
+    return (
+      `${DBMLKeywords.Table} ${modelName} {\n` +
       generateFields(model.fields) +
       generateTableIndexes(model) +
       generateTableDocumentation(model) +
       '\n}'
-  );
+    );
+  });
 }
 
 const generateTableIndexes = (model: DMMF.Model): string => {