Halloween db schema

Author: ekzyis

prisma/migrations/20251022202342_halloween/migration.sql

@@ -0,0 +1,30 @@
+-- CreateTable
+CREATE TABLE "Infection" (
+    "id" SERIAL NOT NULL,
+    "created_at" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
+    "infecteeId" INTEGER NOT NULL,
+    "infectorId" INTEGER,
+
+    CONSTRAINT "Infection_pkey" PRIMARY KEY ("id"),
+    CONSTRAINT "Infection_patient_zero_check" CHECK (
+        CASE
+            WHEN id <= 4 THEN true
+            ELSE "infectorId" IS NOT NULL
+        END
+    )
+);
+
+-- CreateIndex
+CREATE UNIQUE INDEX "Infection_infecteeId_unique" ON "Infection"("infecteeId");
+
+-- CreateIndex
+CREATE INDEX "Infection_infecteeId_idx" ON "Infection"("infecteeId");
+
+-- CreateIndex
+CREATE INDEX "Infection_infectorId_idx" ON "Infection"("infectorId");
+
+-- AddForeignKey
+ALTER TABLE "Infection" ADD CONSTRAINT "Infection_infecteeId_fkey" FOREIGN KEY ("infecteeId") REFERENCES "users"("id") ON DELETE CASCADE ON UPDATE CASCADE;
+
+-- AddForeignKey
+ALTER TABLE "Infection" ADD CONSTRAINT "Infection_infectorId_fkey" FOREIGN KEY ("infectorId") REFERENCES "users"("id") ON DELETE CASCADE ON UPDATE CASCADE;

prisma/schema.prisma

@@ -153,6 +153,8 @@ model User {
   DirectPaymentSent         DirectPayment[]      @relation("DirectPaymentSent")
   UserSubTrust              UserSubTrust[]
   CommentsViewAt            CommentsViewAt[]
+  infectedBy                Infection? @relation("Infection_infectee")
+  infectedByMe              Infection[] @relation("Infection_infector")
 
   @@index([photoId])
   @@index([createdAt], map: "users.created_at_index")
@@ -163,6 +165,18 @@ model User {
   @@map("users")
 }
 
+model Infection {
+  id         Int      @id @default(autoincrement())
+  createdAt  DateTime @default(now()) @map("created_at")
+  infecteeId Int      @unique(map: "Infection_infecteeId_unique")
+  infectorId Int?
+  infectee   User     @relation("Infection_infectee", fields: [infecteeId], references: [id], onDelete: Cascade)
+  infector   User?    @relation("Infection_infector", fields: [infectorId], references: [id], onDelete: Cascade)
+
+  @@index([infecteeId])
+  @@index([infectorId])
+}
+
 enum OneDayReferralType {
   REFERRAL
   POST