feat: deleteUser

Author: Vandivier

scripts/deleteUser.js

@@ -0,0 +1,111 @@
+// deletes a user and related entities like subscription
+// by userId or email
+const path = require("path")
+
+require("dotenv").config({ path: path.resolve(__dirname, "../.env.local") })
+
+const { PrismaClient } = require("@prisma/client")
+
+const prisma = new PrismaClient()
+
+const args = process.argv.slice(2)
+
+if (
+  args.length !== 2 ||
+  (!args.includes("--userId") && !args.includes("--email"))
+) {
+  console.error(
+    "Please provide exactly one flag: --userId or --email followed by the respective value."
+  )
+  process.exit(1)
+}
+
+let userId, email
+
+if (args.includes("--userId")) {
+  userId = args[args.indexOf("--userId") + 1]
+  if (!/^\d+$/.test(userId)) {
+    console.error("Invalid userId provided.")
+    process.exit(1)
+  }
+} else if (args.includes("--email")) {
+  email = args[args.indexOf("--email") + 1]
+  if (!/^[^\s@]+@[^\s@]+\.[^\s@]+$/.test(email)) {
+    console.error("Invalid email provided.")
+    process.exit(1)
+  }
+}
+
+const deleteUser = async ({ userId, email }) => {
+  try {
+    let user
+    if (userId) {
+      user = await prisma.user.findUnique({
+        where: { id: parseInt(userId, 10) },
+      })
+    } else if (email) {
+      user = await prisma.user.findUnique({
+        where: { email: email },
+      })
+    }
+
+    if (!user) {
+      console.error("User not found.")
+      process.exit(1)
+    }
+
+    const userIdToDelete = user.id
+
+    // Delete related UserChecklistItems first to avoid foreign key constraints
+    await prisma.userChecklistItem.deleteMany({
+      where: { userId: userIdToDelete },
+    })
+    console.log(`Deleted userChecklistItem entries for user ${userIdToDelete}`)
+
+    // Delete user-related entities
+    const relatedEntities = [
+      { model: "userChecklist", relation: "userId" },
+      { model: "subscription", relation: "userId" },
+      { model: "session", relation: "userId" },
+      { model: "token", relation: "userId" },
+      { model: "transaction", relation: "userId" },
+      { model: "quizResult", relation: "userId" },
+      { model: "contribution", relation: "userId" },
+    ]
+
+    for (const entity of relatedEntities) {
+      const deleteManyMethod = prisma[entity.model].deleteMany
+      if (deleteManyMethod) {
+        await deleteManyMethod({
+          where: { [entity.relation]: userIdToDelete },
+        })
+        console.log(
+          `Deleted ${entity.model} entries for user ${userIdToDelete}`
+        )
+      } else {
+        console.log(`No delete method found for ${entity.model}`)
+      }
+    }
+
+    // Delete the user
+    await prisma.user.delete({
+      where: { id: userIdToDelete },
+    })
+
+    console.log(`Deleted user ${userIdToDelete}`)
+  } catch (error) {
+    if (error.code === "P2025") {
+      console.log(`Entity not found: ${error.meta.cause}`)
+    } else {
+      console.error("An error occurred while deleting the user:", error)
+    }
+  } finally {
+    await prisma.$disconnect()
+  }
+}
+
+if (userId) {
+  deleteUser({ userId })
+} else if (email) {
+  deleteUser({ email })
+}