Added logic to change priority for outdated supplies with priority urgent

Author: gustavocs

src/shelter-supply/constants.ts

@@ -0,0 +1,3 @@
+const priorityExpiryInterval = 4; // In hours
+
+export { priorityExpiryInterval };

src/shelter-supply/shelter-supply.service.ts

@@ -1,7 +1,10 @@
-import { z } from 'zod';
 import { Injectable } from '@nestjs/common';
+import { z } from 'zod';
 
 import { PrismaService } from '../prisma/prisma.service';
+import { SupplyPriority } from '../supply/types';
+import { getDifferenceInHours } from '../utils';
+import { priorityExpiryInterval } from './constants';
 import { CreateShelterSupplySchema, UpdateShelterSupplySchema } from './types';
 
 @Injectable()
@@ -98,4 +101,23 @@ export class ShelterSupplyService {
       },
     });
   }
+
+  async checkAndUpdateOutdatedPriorities(shelterSupplies: Array<any>) {
+    shelterSupplies.forEach((s) => {
+      if (
+        s.priority === SupplyPriority.Urgent &&
+        getDifferenceInHours(
+          new Date(s.updatedAt || s.createdAt),
+          new Date(Date.now()),
+        ) >= priorityExpiryInterval
+      ) {
+        this.update({
+          data: {
+            priority: SupplyPriority.Needing,
+          },
+          where: { shelterId: s.shelterId, supplyId: s.supply.id },
+        });
+      }
+    });
+  }
 }

src/shelter/shelter.module.ts

@@ -3,10 +3,11 @@ import { Module } from '@nestjs/common';
 import { ShelterService } from './shelter.service';
 import { ShelterController } from './shelter.controller';
 import { PrismaModule } from '../prisma/prisma.module';
+import { ShelterSupplyService } from '../shelter-supply/shelter-supply.service';
 
 @Module({
   imports: [PrismaModule],
-  providers: [ShelterService],
+  providers: [ShelterService, ShelterSupplyService],
   controllers: [ShelterController],
 })
 export class ShelterModule {}

src/shelter/shelter.service.ts

@@ -11,10 +11,14 @@ import {
 } from './types';
 import { SeachQueryProps } from '@/decorators/search-query/types';
 import { SupplyPriority } from '../supply/types';
+import { ShelterSupplyService } from 'src/shelter-supply/shelter-supply.service';
 
 @Injectable()
 export class ShelterService {
-  constructor(private readonly prismaService: PrismaService) {}
+  constructor(
+    private readonly prismaService: PrismaService,
+    private readonly shelterSupplyService: ShelterSupplyService,
+  ) {}
 
   async store(body: z.infer<typeof CreateShelterSchema>) {
     const payload = CreateShelterSchema.parse(body);
@@ -102,7 +106,40 @@ export class ShelterService {
   async index(props: SeachQueryProps) {
     const { handleSearch } = props;
 
-    return await handleSearch<Prisma.ShelterSelect<DefaultArgs>>(
+    const partialResult = await handleSearch<Prisma.ShelterSelect<DefaultArgs>>(
+      this.prismaService.shelter,
+      {
+        select: {
+          id: true,
+          shelterSupplies: {
+            where: {
+              priority: {
+                gte: SupplyPriority.Urgent,
+              },
+            },
+            take: 10,
+            select: {
+              priority: true,
+              supply: {
+                select: {
+                  id: true,
+                },
+              },
+              createdAt: true,
+              updatedAt: true,
+            },
+          },
+        },
+      },
+    );
+
+    await this.shelterSupplyService.checkAndUpdateOutdatedPriorities(
+      partialResult.results.flatMap((r) =>
+        r.shelterSupplies.map((s) => ({ ...s, shelterId: r.id })),
+      ),
+    );
+
+    const result = await handleSearch<Prisma.ShelterSelect<DefaultArgs>>(
       this.prismaService.shelter,
       {
         select: {
@@ -142,5 +179,7 @@ export class ShelterService {
         },
       },
     );
+
+    return result;
   }
 }

src/utils/index.ts

@@ -6,6 +6,7 @@ import {
   getSearchWhere,
   parseStringToObject,
   capitalize,
+  getDifferenceInHours,
 } from './utils';
 
 export {
@@ -16,4 +17,5 @@ export {
   deepMerge,
   getSearchWhere,
   parseStringToObject,
+  getDifferenceInHours,
 };

src/utils/utils.ts

@@ -140,6 +140,15 @@ function getSearchWhere(search: string, or?: boolean) {
   } else return where;
 }
 
+function getDifferenceInHours(startDate: Date, endDate: Date) {
+  const timeDiff = endDate.getTime() - startDate.getTime();
+  return getHours(timeDiff);
+}
+
+function getHours(date: number) {
+  return date / (1000 * 60 * 60);
+}
+
 export {
   ServerResponse,
   removeNotNumbers,
@@ -148,4 +157,5 @@ export {
   parseStringToObject,
   getSearchWhere,
   capitalize,
+  getDifferenceInHours,
 };