feat: Nullable keys in ERD

keonik · May 26, 2022 · f7d7304 · f7d7304
1 parent c792db7
commit f7d7304
Show file tree

Hide file tree

Showing 14 changed files with 63 additions and 9 deletions.
diff --git a/ERD.svg b/ERD.svg
diff --git a/__tests__/36.png b/__tests__/36.png
diff --git a/__tests__/66.png b/__tests__/66.png
diff --git a/__tests__/73.svg b/__tests__/73.svg
diff --git a/__tests__/CompositeTypes.svg b/__tests__/CompositeTypes.svg
diff --git a/__tests__/Enums.svg b/__tests__/Enums.svg
diff --git a/__tests__/ManyToMany.svg b/__tests__/ManyToMany.svg
diff --git a/__tests__/Mappings.svg b/__tests__/Mappings.svg
diff --git a/__tests__/MongoDB.svg b/__tests__/MongoDB.svg
diff --git a/__tests__/nullables.svg b/__tests__/nullables.svg
diff --git a/__tests__/nullables.test.ts b/__tests__/nullables.test.ts
@@ -0,0 +1,24 @@
+import * as child_process from 'child_process';
+
+test('nullables.prisma', async () => {
+    const fileName = 'nullables.svg';
+    const folderName = '__tests__';
+    child_process.execSync(`rm -f ${folderName}/${fileName}`);
+    child_process.execSync(
+        `npx prisma generate --schema ./prisma/nullables.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 with the correct content
+    expect(svgAsString).toContain(`<svg`);
+    expect(svgAsString).toContain(`nullable`);
+    expect(svgAsString).toContain(`inviteeEmail`);
+    expect(svgAsString).toContain(`cancelCode`);
+    expect(svgAsString).toContain(`name`);
+});
diff --git a/__tests__/simple-1-n.svg b/__tests__/simple-1-n.svg
diff --git a/prisma/nullables.prisma b/prisma/nullables.prisma
@@ -0,0 +1,27 @@
+// 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__/nullables.svg"
+}
+
+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[]
+}
diff --git a/src/generate.ts b/src/generate.ts
@@ -187,7 +187,9 @@ ${
                       `    ${field.type.trimStart()} ${field.name.replace(
                           /^_/,
                           'z_'
-                      )}`
+                      )} ${field.isId ? 'PK' : ''} ${
+                          field.isRequired ? '' : '"nullable"'
+                      }`
               )
               .join('\n')
 }