Merge pull request #38 from GeneralMagicio/delete-poll

delete poll

Author: RamRamez

prisma/migrations/20250402170145_add_cascade_delete/migration.sql

@@ -0,0 +1,11 @@
+-- DropForeignKey
+ALTER TABLE "UserAction" DROP CONSTRAINT "UserAction_pollId_fkey";
+
+-- DropForeignKey
+ALTER TABLE "Vote" DROP CONSTRAINT "Vote_pollId_fkey";
+
+-- AddForeignKey
+ALTER TABLE "UserAction" ADD CONSTRAINT "UserAction_pollId_fkey" FOREIGN KEY ("pollId") REFERENCES "Poll"("pollId") ON DELETE CASCADE ON UPDATE CASCADE;
+
+-- AddForeignKey
+ALTER TABLE "Vote" ADD CONSTRAINT "Vote_pollId_fkey" FOREIGN KEY ("pollId") REFERENCES "Poll"("pollId") ON DELETE CASCADE ON UPDATE CASCADE;

prisma/schema.prisma

@@ -32,7 +32,7 @@ model UserAction {
   pollId  Int
   type      ActionType
   user      User     @relation(fields: [userId], references: [id])
-  poll      Poll       @relation(fields: [pollId], references: [pollId])
+  poll      Poll       @relation(fields: [pollId], references: [pollId],onDelete: Cascade)
  
 }
 
@@ -62,7 +62,7 @@ model Vote {
   weightDistribution Json
   proof             String
   user             User     @relation(fields: [userId], references: [id])
-  poll             Poll     @relation(fields: [pollId], references: [pollId])
+  poll             Poll     @relation(fields: [pollId], references: [pollId],onDelete: Cascade)
 }
 
 enum ActionType {

src/poll/poll.controller.ts

@@ -46,4 +46,18 @@ export class PollController {
         .json({ message: 'Internal server error', error: error.message });
     }
   }
+
+  @Delete(':id')
+  async deletePoll(@Param('id') id: number, @Res() res: Response) {
+    const userId = 1; // need to implement Auth
+    try {
+      const poll = await this.pollService.deletePoll(userId, Number(id));
+
+      return res.status(200).json({ message: 'Poll deleted', poll: poll });
+    } catch (error) {
+      return res
+        .status(500)
+        .json({ message: 'Internal server error', error: error.message });
+    }
+  }
 }

src/poll/poll.service.ts

@@ -134,4 +134,37 @@ export class PollService {
       throw new Error('Database query failed');
     }
   }
+
+  async deletePoll(userId: number, pollId: number) {
+    const poll = await this.databaseService.poll.findUnique({
+      where: { pollId },
+    });
+
+    if (!poll) {
+      throw new Error('Poll not found');
+    }
+    if (poll.authorUserId !== userId) {
+      throw new Error('User Not Authorized');
+    }
+
+    return this.databaseService.$transaction(async (tx) => {
+      const deleted = await tx.poll.delete({
+        where: {
+          pollId,
+        },
+      });
+
+      // Update user's pollsCreatedCount
+      await tx.user.update({
+        where: { id: deleted.authorUserId },
+        data: {
+          pollsCreatedCount: {
+            decrement: 1,
+          },
+        },
+      });
+
+      return deleted;
+    });
+  }
 }