From c67f14b211b98bd6db92bc767602c79191c72525 Mon Sep 17 00:00:00 2001
From: Hakan Dilek <hakandilek@gmail.com>
Date: Tue, 20 Jun 2023 22:44:11 +0200
Subject: [PATCH 1/2] feat: introduce ignoreViews parameter

---
 .gitignore                    |  3 ++-
 README.md                     | 11 +++++++++
 __tests__/ignoreViews.test.ts | 41 ++++++++++++++++++++++++++++++++
 prisma/ignore-views.prisma    | 44 +++++++++++++++++++++++++++++++++++
 src/generate.ts               | 26 ++++++++++++++++++++-
 5 files changed, 123 insertions(+), 2 deletions(-)
 create mode 100644 __tests__/ignoreViews.test.ts
 create mode 100644 prisma/ignore-views.prisma

diff --git a/.gitignore b/.gitignore
index 8bf9f74..19f65a7 100644
--- a/.gitignore
+++ b/.gitignore
@@ -7,4 +7,5 @@ prisma/debug
 coverage
 !__test__/*.ts
 __tests__/*.svg
-__tests__/*.png
\ No newline at end of file
+__tests__/*.png
+.idea
diff --git a/README.md b/README.md
index 31b0781..beb973c 100644
--- a/README.md
+++ b/README.md
@@ -151,6 +151,17 @@ generator erd {
 }
 ```
 
+### Ignore views
+
+If you enable this option, view entities will be hidden.
+This is useful if you want to reduce the number of entities and focus on the tables without displaying all the views.
+
+```prisma
+generator erd {
+  provider = "prisma-erd-generator"
+  ignoreViews = true
+}
+```
 ### Include relation from field
 
 By default this module skips relation fields in the result diagram. For example fields `userId` and `productId` will not be generated from this prisma schema.
diff --git a/__tests__/ignoreViews.test.ts b/__tests__/ignoreViews.test.ts
new file mode 100644
index 0000000..2562bdd
--- /dev/null
+++ b/__tests__/ignoreViews.test.ts
@@ -0,0 +1,41 @@
+import * as child_process from 'child_process';
+
+test('ignore-views.prisma', async () => {
+    const fileName = 'ignoreViews.svg';
+    const folderName = '__tests__';
+    child_process.execSync(`rm -f ${folderName}/${fileName}`);
+    child_process.execSync(
+        `prisma generate --schema ./prisma/ignore-views.prisma`
+    );
+    const listFile = child_process.execSync(`ls -la ${folderName}/${fileName}`);
+    // did it generate a file
+    expect(listFile.toString()).toContain(fileName);
+
+    const svgAsString = child_process
+        .execSync(`cat ${folderName}/${fileName}`)
+        .toString();
+
+    // did it generate a file without enum
+    expect(svgAsString).toContain(`<svg`);
+    // include tables
+    expect(svgAsString).toContain(`Booking`);
+    expect(svgAsString).toContain(`Event`);
+    // include table columns
+    expect(svgAsString).toContain(`name`);
+    expect(svgAsString).toContain(`startDate`);
+    expect(svgAsString).toContain(`status`);
+    expect(svgAsString).toContain(`inviteeEmail`);
+    expect(svgAsString).toContain(`startDateUTC`);
+    expect(svgAsString).toContain(`cancelCode`);
+    // include enum names
+    expect(svgAsString).toContain(`Status`);
+    // include enums
+    expect(svgAsString).toContain(`PENDING`);
+    expect(svgAsString).toContain(`CONFIRMED`);
+    expect(svgAsString).toContain(`CANCELLED`);
+    // exclude views
+    expect(svgAsString).not.toContain(`Attendance`);
+    expect(svgAsString).not.toContain(`email`);
+    expect(svgAsString).not.toContain(`eventDate`);
+    expect(svgAsString).not.toContain(`bookingStatus`);
+});
diff --git a/prisma/ignore-views.prisma b/prisma/ignore-views.prisma
new file mode 100644
index 0000000..bcdd9dd
--- /dev/null
+++ b/prisma/ignore-views.prisma
@@ -0,0 +1,44 @@
+// This is your Prisma schema file,
+// learn more about it in the docs: https://pris.ly/d/prisma-schema
+
+datasource db {
+    provider = "postgresql"
+    url      = env("DATABASE_URL")
+}
+
+generator erd {
+    provider        = "node ./dist/index.js"
+    output          = "../__tests__/ignoreViews.svg"
+    theme           = "forest"
+    previewFeatures = ["views"]
+    ignoreViews     = true
+}
+
+model Booking {
+    id           Int      @id @default(autoincrement())
+    inviteeEmail String
+    startDateUTC DateTime
+    cancelCode   String
+    events       Event[]
+}
+
+model Event {
+    id        Int       @id @default(autoincrement())
+    name      String
+    startDate DateTime
+    bookings  Booking[]
+    status    Status    @default(PENDING)
+}
+
+enum Status {
+    PENDING
+    CANCELLED
+    CONFIRMED
+}
+
+view Attendance {
+    id            Int      @unique
+    email         String
+    eventDate     DateTime
+    bookingStatus Status
+}
diff --git a/src/generate.ts b/src/generate.ts
index 2274ed6..cc0a162 100644
--- a/src/generate.ts
+++ b/src/generate.ts
@@ -22,9 +22,25 @@ export interface DMLModel {
     } | null;
 }
 
+export interface DMLView {
+    name: string;
+    isEmbedded: boolean;
+    dbName: string | null;
+    fields: DMLField[];
+    idFields: any[];
+    uniqueFields: any[];
+    uniqueIndexes: any[];
+    isGenerated: boolean;
+    primaryKey: {
+        name: string | null;
+        fields: string[];
+    } | null;
+}
+
 export interface DMLRendererOptions {
     tableOnly?: boolean;
     ignoreEnums?: boolean;
+    ignoreViews?: boolean;
     includeRelationFromFields?: boolean;
 }
 
@@ -55,7 +71,7 @@ export interface DMLField {
     isRequired: boolean;
     isUnique: boolean;
     isUpdatedAt: boolean;
-    kind: 'scalar' | 'object' | 'enum';
+    kind: 'scalar' | 'object' | 'enum' | 'view';
     type: string;
     relationFromFields?: any[];
     relationName?: string;
@@ -75,6 +91,7 @@ export interface DMLEnum {
 export interface DML {
     enums: DMLEnum[];
     models: DMLModel[];
+    views: DMLView[];
     types: DMLType[];
 }
 
@@ -137,6 +154,7 @@ function renderDml(dml: DML, options?: DMLRendererOptions) {
     const {
         tableOnly = false,
         ignoreEnums = false,
+        ignoreViews = false,
         includeRelationFromFields = false,
     } = options ?? {};
 
@@ -196,9 +214,13 @@ ${
     for (const model of modellikes) {
         for (const field of model.fields) {
             const isEnum = field.kind === 'enum';
+            const isView = field.kind === 'view';
             if (isEnum && (tableOnly || ignoreEnums)) {
                 continue;
             }
+            if (isView && (tableOnly || ignoreViews)) {
+                continue;
+            }
 
             const relationshipName = `${isEnum ? 'enum:' : ''}${field.name}`;
             const thisSide = `"${model.dbName || model.name}"`;
@@ -373,6 +395,7 @@ export default async (options: GeneratorOptions) => {
         );
         const tableOnly = config.tableOnly === 'true';
         const ignoreEnums = config.ignoreEnums === 'true';
+        const ignoreViews = config.ignoreViews === 'true';
         const includeRelationFromFields =
             config.includeRelationFromFields === 'true';
         const disabled = Boolean(process.env.DISABLE_ERD);
@@ -428,6 +451,7 @@ export default async (options: GeneratorOptions) => {
         const mermaid = renderDml(dml, {
             tableOnly,
             ignoreEnums,
+            ignoreViews,
             includeRelationFromFields,
         });
         if (debug && mermaid) {

From 0a27c5edef1e5b51901f7498da34c859568d679a Mon Sep 17 00:00:00 2001
From: Hakan Dilek <hakandilek@gmail.com>
Date: Tue, 20 Jun 2023 23:11:07 +0200
Subject: [PATCH 2/2]  Add @hakandilek as a contributor

---
 .all-contributorsrc | 9 +++++++++
 README.md           | 3 ++-
 2 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/.all-contributorsrc b/.all-contributorsrc
index e0f322d..efdb3b7 100644
--- a/.all-contributorsrc
+++ b/.all-contributorsrc
@@ -153,6 +153,15 @@
       "contributions": [
         "code"
       ]
+    },
+    {
+      "login": "hakandilek",
+      "name": "Hakan Dilek",
+      "avatar_url": "https://avatars.githubusercontent.com/u/1072473?v=4",
+      "profile": "http://www.dilek.me/",
+      "contributions": [
+        "code"
+      ]
     }
   ],
   "contributorsPerLine": 7,
diff --git a/README.md b/README.md
index beb973c..c335541 100644
--- a/README.md
+++ b/README.md
@@ -1,7 +1,7 @@
 # Prisma Entity Relationship Diagram Generator
 
 <!-- ALL-CONTRIBUTORS-BADGE:START - Do not remove or modify this section -->
-[![All Contributors](https://img.shields.io/badge/all_contributors-15-orange.svg?style=flat-square)](#contributors-)
+[![All Contributors](https://img.shields.io/badge/all_contributors-16-orange.svg?style=flat-square)](#contributors-)
 <!-- ALL-CONTRIBUTORS-BADGE:END -->
 
 Prisma generator to create an ER Diagram every time you generate your prisma client.
@@ -233,6 +233,7 @@ Thanks goes to these wonderful people ([emoji key](https://allcontributors.org/d
     </tr>
     <tr>
       <td align="center" valign="top" width="14.28%"><a href="https://www.chintristan.io/"><img src="https://avatars.githubusercontent.com/u/23557893?v=4?s=100" width="100px;" alt="Tristan Chin"/><br /><sub><b>Tristan Chin</b></sub></a><br /><a href="https://github.com/keonik/prisma-erd-generator/commits?author=maxijonson" title="Code">💻</a></td>
+      <td align="center" valign="top" width="14.28%"><a href="http://www.dilek.me/"><img src="https://avatars.githubusercontent.com/u/1072473?v=4?s=100" width="100px;" alt="Hakan Dilek"/><br /><sub><b>Hakan Dilek</b></sub></a><br /><a href="https://github.com/keonik/prisma-erd-generator/commits?author=hakandilek" title="Code">💻</a></td>
     </tr>
   </tbody>
 </table>