Merge pull request #20 from GeneralMagicio/add-getPollDetails

Add get poll details

Author: RamRamez

src/poll/createPoll.dto.ts src/poll/Poll.dto.tssrc/poll/createPoll.dto.ts renamed to src/poll/Poll.dto.ts

@@ -2,9 +2,12 @@ import {
   IsArray,
   IsBoolean,
   IsDateString,
+  IsEnum,
+  IsInt,
   IsNotEmpty,
   IsOptional,
   IsString,
+  Min,
 } from 'class-validator';
 
 export class CreatePollDto {
@@ -38,3 +41,35 @@ export class CreatePollDto {
   @IsOptional()
   isAnonymous?: boolean;
 }
+
+export class GetPollsDto {
+  @IsOptional()
+  @IsInt()
+  @Min(1)
+  page?: number = 1;
+
+  @IsOptional()
+  @IsInt()
+  @Min(1)
+  limit?: number = 10;
+
+  @IsOptional()
+  @IsBoolean()
+  isActive?: boolean;
+
+  @IsOptional()
+  @IsBoolean()
+  userVoted?: boolean;
+
+  @IsOptional()
+  @IsBoolean()
+  userCreated?: boolean;
+
+  @IsOptional()
+  @IsEnum(['endDate', 'participantCount', 'creationDate'])
+  sortBy?: 'endDate' | 'participantCount' | 'creationDate';
+
+  @IsOptional()
+  @IsEnum(['asc', 'desc'])
+  sortOrder?: 'asc' | 'desc';
+}

src/poll/poll.controller.ts

@@ -1,7 +1,18 @@
-import { Controller, Get, Post, Body, Param, Delete } from '@nestjs/common';
+import {
+  Controller,
+  Get,
+  Post,
+  Body,
+  Param,
+  Delete,
+  Req,
+  Query,
+  Res,
+} from '@nestjs/common';
+import { Response } from 'express';
 import { PollService } from './poll.service';
 import { Prisma } from '@prisma/client';
-import { CreatePollDto } from './createPoll.dto';
+import { CreatePollDto, GetPollsDto } from './Poll.dto';
 
 @Controller('poll')
 export class PollController {
@@ -14,12 +25,25 @@ export class PollController {
   }
 
   @Get()
-  getPolls() {
-    return this.pollService.getPolls();
+  getPolls(@Req() req, @Query() query: GetPollsDto) {
+    const userId = 1;
+    return this.pollService.getPolls(userId, query);
   }
 
   @Get(':id')
-  getPollDetails(@Param('id') id: string) {
-    return this.pollService.getPollDetails(Number(id));
+  async getPollDetails(@Param('id') id: number, @Res() res: Response) {
+    try {
+      const poll = await this.pollService.getPollDetails(Number(id));
+
+      if (!poll) {
+        return res.status(404).json({ message: 'No poll found' });
+      }
+
+      return res.status(200).json(poll);
+    } catch (error) {
+      return res
+        .status(500)
+        .json({ message: 'Internal server error', error: error.message });
+    }
   }
 }

src/poll/poll.service.ts

@@ -1,7 +1,7 @@
 import { BadRequestException, Injectable } from '@nestjs/common';
 import { ActionType, Prisma } from '@prisma/client';
 import { DatabaseService } from 'src/database/database.service';
-import { CreatePollDto } from './createPoll.dto';
+import { CreatePollDto, GetPollsDto } from './Poll.dto';
 
 @Injectable()
 export class PollService {
@@ -66,15 +66,72 @@ export class PollService {
     });
   }
 
-  async getPolls() {
-    return this.databaseService.poll.findMany({});
+  async getPolls(userId: number, query: GetPollsDto) {
+    const {
+      page = 1,
+      limit = 10,
+      isActive,
+      userVoted,
+      userCreated,
+      sortBy = 'endDate',
+      sortOrder = 'asc',
+    } = query;
+    const skip = (page - 1) * limit;
+    const now = new Date();
+    const filters: any = {};
+    if (isActive) {
+      filters.startDate = { lte: now };
+      filters.endDate = { gt: now };
+    }
+    if (userCreated) {
+      filters.authorUserId = userId;
+    }
+
+    // Get polls user voted in
+    let votedPollIds: number[] = [];
+    if (userVoted) {
+      const userVotes = await this.databaseService.vote.findMany({
+        where: { userId },
+        select: { pollId: true },
+      });
+      votedPollIds = userVotes.map((v) => v.pollId);
+      filters.pollId = { in: votedPollIds };
+    }
+
+    // Sorting options
+    const orderBy: any = {};
+    if (sortBy) {
+      orderBy[sortBy] = sortOrder;
+    } else {
+      orderBy.endDate = 'asc';
+    }
+
+    // Fetch polls with pagination
+    const [polls, total] = await this.databaseService.$transaction([
+      this.databaseService.poll.findMany({
+        where: filters,
+        orderBy,
+        skip,
+        take: Number(limit),
+      }),
+      this.databaseService.poll.count({ where: filters }),
+    ]);
+
+    return { polls, total };
   }
 
   async getPollDetails(id: number) {
-    return this.databaseService.poll.findUnique({
-      where: {
-        pollId: id,
-      },
-    });
+    try {
+      const poll = await this.databaseService.poll.findUnique({
+        where: { pollId: id },
+      });
+      const user = await this.databaseService.user.findUnique({
+        where: { id: poll?.authorUserId },
+      });
+
+      return { user, poll };
+    } catch (error) {
+      throw new Error('Database query failed');
+    }
   }
 }