Fix/develop bugs (#54)

* feat: update many supplies

* feat: created admin user role

Author: fagundesjg

prisma/migrations/20240511054108_/migration.sql

@@ -0,0 +1,2 @@
+-- AlterEnum
+ALTER TYPE "AccessLevel" ADD VALUE 'Admin';

prisma/schema.prisma

@@ -11,6 +11,7 @@ enum AccessLevel {
   User
   Staff
   DistributionCenter
+  Admin
 }
 
 model User {

src/guards/admin.guard.ts

@@ -0,0 +1,20 @@
+import { ExecutionContext, HttpException, Injectable } from '@nestjs/common';
+import { AuthGuard } from '@nestjs/passport';
+import { AccessLevel } from '@prisma/client';
+
+import { canActivate } from './utils';
+
+@Injectable()
+export class AdminGuard extends AuthGuard('jwt') {
+  constructor() {
+    super();
+  }
+
+  async canActivate(context: ExecutionContext): Promise<boolean> {
+    await super.canActivate(context);
+    const ok = await canActivate(context, [AccessLevel.Admin]);
+    if (ok) return true;
+
+    throw new HttpException('Acesso não autorizado', 401);
+  }
+}

src/guards/distribution-center.guard.ts

@@ -0,0 +1,23 @@
+import { ExecutionContext, HttpException, Injectable } from '@nestjs/common';
+import { AuthGuard } from '@nestjs/passport';
+import { AccessLevel } from '@prisma/client';
+
+import { canActivate } from './utils';
+
+@Injectable()
+export class DistributionCenterGuard extends AuthGuard('jwt') {
+  constructor() {
+    super();
+  }
+
+  async canActivate(context: ExecutionContext): Promise<boolean> {
+    await super.canActivate(context);
+    const ok = await canActivate(context, [
+      AccessLevel.Admin,
+      AccessLevel.DistributionCenter,
+    ]);
+    if (ok) return true;
+
+    throw new HttpException('Acesso não autorizado', 401);
+  }
+}

src/guards/staff.guard.ts

@@ -12,7 +12,10 @@ export class StaffGuard extends AuthGuard('jwt') {
 
   async canActivate(context: ExecutionContext): Promise<boolean> {
     await super.canActivate(context);
-    const ok = await canActivate(context, [AccessLevel.Staff]);
+    const ok = await canActivate(context, [
+      AccessLevel.Admin,
+      AccessLevel.Staff,
+    ]);
     if (ok) return true;
 
     throw new HttpException('Acesso não autorizado', 401);

src/guards/user.guard.ts

@@ -13,8 +13,10 @@ export class UserGuard extends AuthGuard('jwt') {
   async canActivate(context: ExecutionContext): Promise<boolean> {
     await super.canActivate(context);
     const ok = await canActivate(context, [
-      AccessLevel.Staff,
       AccessLevel.User,
+      AccessLevel.Staff,
+      AccessLevel.DistributionCenter,
+      AccessLevel.Admin,
     ]);
     if (ok) return true;
     throw new HttpException('Acesso não autorizado', 401);

src/shelter-managers/shelter-managers.controller.ts

@@ -13,7 +13,7 @@ import { ApiTags } from '@nestjs/swagger';
 
 import { ShelterManagersService } from './shelter-managers.service';
 import { ServerResponse } from '../utils';
-import { StaffGuard } from '@/guards/staff.guard';
+import { AdminGuard } from '@/guards/admin.guard';
 
 @ApiTags('Admin de Abrigo')
 @Controller('shelter/managers')
@@ -25,7 +25,7 @@ export class ShelterManagersController {
   ) {}
 
   @Post('')
-  @UseGuards(StaffGuard)
+  @UseGuards(AdminGuard)
   async store(@Body() body) {
     try {
       await this.shelterManagerServices.store(body);

src/shelter-supply/shelter-supply.controller.ts

@@ -7,11 +7,13 @@ import {
   Param,
   Post,
   Put,
+  UseGuards,
 } from '@nestjs/common';
 import { ApiTags } from '@nestjs/swagger';
 
 import { ShelterSupplyService } from './shelter-supply.service';
 import { ServerResponse } from '../utils';
+import { DistributionCenterGuard } from '@/guards/distribution-center.guard';
 
 @ApiTags('Suprimento de abrigos')
 @Controller('shelter/supplies')
@@ -67,4 +69,23 @@ export class ShelterSupplyController {
       throw new HttpException(err?.code ?? err?.name ?? `${err}`, 400);
     }
   }
+
+  @Put(':shelterId/supplies/many')
+  @UseGuards(DistributionCenterGuard)
+  async updateMany(@Body() body, @Param('shelterId') shelterId: string) {
+    try {
+      const data = await this.shelterSupplyService.updateMany({
+        shelterId,
+        ...body,
+      });
+      return new ServerResponse(
+        200,
+        'Successfully updated many shelter supplies',
+        data,
+      );
+    } catch (err: any) {
+      this.logger.error(`Failed to update many shelter supplies: ${err}`);
+      throw new HttpException(err?.code ?? err?.name ?? `${err}`, 400);
+    }
+  }
 }

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

@@ -2,7 +2,11 @@ import { z } from 'zod';
 import { Injectable } from '@nestjs/common';
 
 import { PrismaService } from '../prisma/prisma.service';
-import { CreateShelterSupplySchema, UpdateShelterSupplySchema } from './types';
+import {
+  CreateShelterSupplySchema,
+  UpdateManyShelterSupplySchema,
+  UpdateShelterSupplySchema,
+} from './types';
 import { SupplyPriority } from '../supply/types';
 
 @Injectable()
@@ -70,7 +74,24 @@ export class ShelterSupplyService {
       data: {
         ...data,
         quantity: priority !== SupplyPriority.UnderControl ? quantity : null,
-        createdAt: new Date().toISOString(),
+        updatedAt: new Date().toISOString(),
+      },
+    });
+  }
+
+  async updateMany(body: z.infer<typeof UpdateManyShelterSupplySchema>) {
+    const { ids, shelterId } = UpdateManyShelterSupplySchema.parse(body);
+
+    await this.prismaService.shelterSupply.updateMany({
+      where: {
+        shelterId: shelterId,
+        supplyId: {
+          in: ids,
+        },
+      },
+      data: {
+        priority: SupplyPriority.UnderControl,
+        updatedAt: new Date().toISOString(),
       },
     });
   }

src/shelter-supply/types.ts

@@ -44,8 +44,14 @@ const UpdateShelterSupplySchema = z.object({
   }),
 });
 
+const UpdateManyShelterSupplySchema = z.object({
+  ids: z.array(z.string()),
+  shelterId: z.string(),
+});
+
 export {
   ShelterSupplySchema,
   CreateShelterSupplySchema,
   UpdateShelterSupplySchema,
+  UpdateManyShelterSupplySchema,
 };