refact: index search

Author: fagundesjg

src/shelter/ShelterSearch.ts

@@ -1,11 +1,15 @@
 import { Prisma } from '@prisma/client';
+
 import { IFilterFormProps } from './types';
+import { PrismaService } from '../prisma/prisma.service';
 
 class ShelterSearch {
   private data: IFilterFormProps;
+  private prismaService: PrismaService;
 
-  constructor(props: IFilterFormProps) {
+  constructor(prismaService: PrismaService, props: IFilterFormProps) {
     this.data = props;
+    this.prismaService = prismaService;
   }
 
   get priority(): Prisma.ShelterWhereInput[] {
@@ -31,16 +35,83 @@ class ShelterSearch {
           return {
             capacity: null,
           };
-        else if (status === 'available') return {};
+        else if (status === 'available')
+          return {
+            capacity: {
+              gt: this.prismaService.shelter.fields.shelteredPeople,
+            },
+          };
         else
           return {
-            // capacity: {
-            //   lte: Prisma.ShelterScalarFieldEnum.shelteredPeople,
-            // },
+            capacity: {
+              lte: this.prismaService.shelter.fields.shelteredPeople,
+            },
           };
       });
     }
   }
+
+  get supplyCategoryIds(): Prisma.ShelterWhereInput {
+    if (!this.data.supplyCategoryIds) return {};
+    return {
+      shelterSupplies: {
+        some: {
+          supply: {
+            supplyCategoryId: {
+              in: this.data.supplyCategoryIds,
+            },
+          },
+        },
+      },
+    };
+  }
+
+  get supplyIds(): Prisma.ShelterWhereInput {
+    if (!this.data.supplyIds) return {};
+    return {
+      shelterSupplies: {
+        some: {
+          supply: {
+            id: {
+              in: this.data.supplyIds,
+            },
+          },
+        },
+      },
+    };
+  }
+
+  get search(): Prisma.ShelterWhereInput[] {
+    if (!this.data.search) return [];
+    else
+      return [
+        {
+          address: {
+            contains: this.data.search,
+            mode: 'insensitive',
+          },
+        },
+        {
+          name: {
+            contains: this.data.search,
+            mode: 'insensitive',
+          },
+        },
+      ];
+  }
+
+  get query(): Prisma.ShelterWhereInput {
+    if (Object.keys(this.data).length === 0) return {};
+    return {
+      OR: [
+        ...this.search,
+        ...this.shelterStatus,
+        this.supplyCategoryIds,
+        this.supplyIds,
+        ...this.search,
+      ],
+    };
+  }
 }
 
 export { ShelterSearch };

src/shelter/shelter.service.ts

@@ -1,6 +1,5 @@
 import { z } from 'zod';
 import { Injectable } from '@nestjs/common';
-import { Prisma } from '@prisma/client';
 import * as qs from 'qs';
 
 import { PrismaService } from '../prisma/prisma.service';
@@ -114,27 +113,7 @@ export class ShelterService {
       search: searchQuery,
     } = SearchSchema.parse(query);
     const queryData = qs.parse(searchQuery) as unknown as IFilterFormProps;
-    const { priority, search, shelterStatus, supplyCategoryIds, supplyIds } =
-      queryData;
-    const shelterSearch = new ShelterSearch(queryData);
-    const where: Prisma.ShelterWhereInput = {
-      OR: [
-        {
-          address: {
-            contains: search,
-            mode: 'insensitive',
-          },
-        },
-        {
-          name: {
-            contains: search,
-            mode: 'insensitive',
-          },
-        },
-        ...shelterSearch.priority,
-        ...shelterSearch.shelterStatus,
-      ],
-    };
+    const { query: where } = new ShelterSearch(this.prismaService, queryData);
 
     const count = await this.prismaService.shelter.count({ where });