feat(many-to-many): option to disable m-n table

* add test if disabled

Author: marcjulian

README.md

@@ -107,10 +107,11 @@ Ref: Post.authorId > User.id
 
 ## Additional Options
 
-| Option       |  Description                       | Type     |  Default      |
-| ------------ | ---------------------------------- | -------- | ------------- |
-| `output`     | Output directory for the DBML file | `string` | `./dbml`      |
-| `outputName` | Name for the DBML file             | `string` | `dbml.schema` |
+| Option       |  Description                       | Type      |  Default      |
+| ------------ | ---------------------------------- | --------- | ------------- |
+| `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`        |
 
 Use additional options in the `prisma.schema`
 

__tests__/dbml.test.ts

@@ -183,4 +183,34 @@ Table AuthorToBook {
 
     expect(dbml).toEqual(expectedDbml);
   });
+
+  test('generating dbml schema with many-to-many relationship', async () => {
+    const dmmf = await generateDMMF(datamodelDbmlManyToMany);
+
+    const expectedDbml = `${autoGeneratedComment}
+
+Table Post {
+  id Int [pk, increment]
+  categories Category
+}
+
+Table Category {
+  id Int [pk, increment]
+  posts Post
+}
+
+Table Author {
+  id Int [pk, increment]
+  books Book
+}
+
+Table Book {
+  id Int [pk, increment]
+  authors Author
+}`;
+
+    const dbml = generateDBMLSchema(dmmf, false);
+
+    expect(dbml).toEqual(expectedDbml);
+  });
 });

prisma/dbml/schema.dbml

@@ -2,32 +2,40 @@
 //// THIS FILE WAS AUTOMATICALLY GENERATED (DO NOT MODIFY)
 //// ------------------------------------------------------
 
-Table Post {
+Table User {
   id Int [pk, increment]
-  categories Category
+  createdAt DateTime [default: `now()`, not null]
+  updatedAt DateTime [not null]
+  email String [unique, not null]
+  name String
+  posts Post
+  profile Profile
+  role Role [not null, default: 'USER', note: 'user role']
 }
 
-Table Category {
+Table Profile {
   id Int [pk, increment]
-  posts Post
+  bio String
+  user User [not null]
+  userId Int [unique, not null]
+
+  Note: 'User profile'
 }
 
-Table Author {
+Table Post {
   id Int [pk, increment]
-  books Book
+  title String [not null, default: '']
+  content String
+  published Boolean [not null, default: false]
+  author User
+  authorId Int
 }
 
-Table Book {
-  id String [pk]
-  authors Author
+Enum Role {
+  ADMIN
+  USER
 }
 
-Table CategoryToPost {
-  categoryId Int [ref: > Category.id]
-  postId Int [ref: > Post.id]
-}
+Ref: Profile.userId - User.id
 
-Table AuthorToBook {
-  bookId String [ref: > Book.id]
-  authorId Int [ref: > Author.id]
-}
+Ref: Post.authorId > User.id

src/cli/dbml-generator.ts

@@ -12,12 +12,14 @@ export async function generate(options: GeneratorOptions) {
   const dbmlFileName =
     options.generator.config.outputName || defaultDBMLFileName;
 
+  const allowManyToMany = options.generator.config.manyToMany || true;
+
   try {
     await mkdir(outputDir, { recursive: true });
 
     // await writeFile('./test.json', JSON.stringify(options.dmmf.datamodel));
 
-    const dbmlSchema = generateDBMLSchema(options.dmmf);
+    const dbmlSchema = generateDBMLSchema(options.dmmf, manyToMany);
 
     await writeFile(join(outputDir, dbmlFileName), dbmlSchema);
   } catch (e) {

src/generator/dbml.ts

@@ -1,16 +1,21 @@
 import { DMMF } from '@prisma/generator-helper';
 import { generateTables } from './table';
 import { generateEnums } from './enums';
-import { generateRelations } from './relations';
+import { generateRelations, manyToOne } from './relations';
 import { generateManyToManyTables } from './many-to-many-tables';
 
 export const autoGeneratedComment = `//// ------------------------------------------------------
 //// THIS FILE WAS AUTOMATICALLY GENERATED (DO NOT MODIFY)
 //// ------------------------------------------------------`;
 
-export function generateDBMLSchema(dmmf: DMMF.Document): string {
+export function generateDBMLSchema(
+  dmmf: DMMF.Document,
+  allowManyToMany: boolean = true
+): string {
   const tables = generateTables(dmmf.datamodel.models);
-  const manyToManyTables = generateManyToManyTables(dmmf.datamodel.models);
+  const manyToManyTables = allowManyToMany
+    ? generateManyToManyTables(dmmf.datamodel.models)
+    : [];
   const enums = generateEnums(dmmf.datamodel.enums);
   const refs = generateRelations(dmmf.datamodel.models);