feat: add filters

tylersai · Jan 14, 2025 · 1af80cf · 1af80cf
1 parent d94bd46
commit 1af80cf
Show file tree

Hide file tree

Showing 3 changed files with 28 additions and 15 deletions.
diff --git a/src/movie/dto/create-movie.ts b/src/movie/dto/create-movie.ts
@@ -14,23 +14,23 @@ class CreateMoviePayload {
 
   @IsString()
   @IsOptional()
-  overview: string;
+  overview?: string;
 
   @IsDateString()
   @IsOptional()
-  release_date: string;
+  release_date?: string;
 
   @IsString()
   @IsOptional()
-  poster_path: string;
+  poster_path?: string;
 
   @IsNumberString()
   @IsOptional()
-  vote_average: number;
+  vote_average?: number;
 
   @IsInt()
   @IsOptional()
-  vote_count: number;
+  vote_count?: number;
 }
 
 export default CreateMoviePayload;
diff --git a/src/movie/movie.controller.ts b/src/movie/movie.controller.ts
@@ -6,6 +6,7 @@ import {
   Param,
   Post,
   Put,
+  Query,
 } from '@nestjs/common';
 import { MovieService } from './movie.service';
 import CreateMoviePayload from './dto/create-movie';
@@ -16,8 +17,8 @@ export class MovieController {
   constructor(private readonly movieService: MovieService) {}
 
   @Get()
-  getAll() {
-    return this.movieService.findAll();
+  getAll(@Query() filters: Record<string, string>) {
+    return this.movieService.findAll(filters);
   }
 
   @Get(':id')

diff --git a/src/movie/movie.service.ts b/src/movie/movie.service.ts
@@ -11,8 +11,19 @@ import UpdateMoviePayload from './dto/update-movie';
 export class MovieService {
   constructor(@InjectModel(Movie.name) private movieModel: Model<Movie>) {}
 
-  async findAll(): Promise<Movie[]> {
-    return this.movieModel.find().exec();
+  async findAll(filters: Record<string, string>): Promise<Movie[]> {
+    const query: Record<string, any> = {};
+    if (filters.title) {
+      query.title = { $regex: filters.title, $options: 'i' };
+    }
+    if (filters.overview) {
+      query.overview = { $regex: filters.overview, $options: 'i' };
+    }
+    if (filters.rating && filters.rating.split(':').length == 2) {
+      const rts = filters.rating.split(':');
+      query.vote_average = { [rts[0]]: rts[1] }; // $eq, $gt, $lt, $gte, $lte etc
+    }
+    return this.movieModel.find(query).exec();
   }
 
   async findById(id: string): Promise<Movie> {
@@ -36,11 +47,12 @@ export class MovieService {
   }
 
   async deleteById(id: string): Promise<RespMessage> {
-    const movie = await this.findById(id);
-    movie.deleteOne();
-    return {
-      statusCode: HttpStatus.OK,
-      message: 'Deleted successfully',
-    };
+    if (isValidObjectId(id)) {
+      await this.movieModel.deleteOne({ _id: id }).exec();
+      return {
+        statusCode: HttpStatus.OK,
+        message: 'Deleted successfully',
+      };
+    } else throw new NotFoundException();
   }
 }