fix(prisma): improve the seed command

code100x · Mar 31, 2024 · 00b842a · 00b842a
1 parent 3405f9f
commit 00b842a
Showing 1 changed file with 31 additions and 15 deletions.
diff --git a/prisma/seed.ts b/prisma/seed.ts
@@ -1,23 +1,39 @@
 import db from '../src/db/index';
 
 async function main() {
-  await db.user.createMany({
-    data: [
-      {
-        id: '2',
-        email: 'testuser',
-        name: 'Test User 1',
-        disableDrm: false,
-      },
-      {
-        id: '1',
-        email: 'testuser2',
-        name: 'Test User 2',
-        disableDrm: false,
-      },
-    ],
+  await db.user.upsert({
+    where: {
+      id: '2',
+      email: 'testuser',
+    },
+    create: {
+      id: '2',
+      email: 'testuser',
+      name: 'Test User 1',
+      disableDrm: false,
+    },
+    update: {},
+  });
+
+  await db.user.upsert({
+    where: {
+      id: '1',
+      email: 'testuser2',
+    },
+    create: {
+      id: '1',
+      email: 'testuser2',
+      name: 'Test User 2',
+      disableDrm: false,
+    },
+    update: {},
   });
 
+  if (await db.course.findMany()) {
+    console.error('DB is already seeded!');
+    process.exit(0);
+  }
+
   await db.course.createMany({
     data: [
       {