feat(planners): implement remove planner item

sparcs-kaist · Mar 20, 2024 · 757ba1f · 757ba1f
1 parent cc0d5e7
commit 757ba1f
Show file tree

Hide file tree

Showing 3 changed files with 88 additions and 3 deletions.
diff --git a/src/modules/planners/planners.controller.ts b/src/modules/planners/planners.controller.ts
@@ -49,10 +49,14 @@ export class PlannersController {
 
   @Post(':plannerId/remove-item')
   async removePlanner(
-    @Body() dto: PlannerRemoveItemDto,
+    @Body() removeItem: PlannerRemoveItemDto,
     @Param('plannerId') plannerId: number,
     @GetUser() user: session_userprofile,
   ): Promise<PlannerResponseDto> {
-    throw new Error('Not implemented');
+    return await this.plannersService.removePlannerItem(
+      plannerId,
+      removeItem,
+      user,
+    );
   }
 }
diff --git a/src/modules/planners/planners.service.ts b/src/modules/planners/planners.service.ts
@@ -1,8 +1,13 @@
-import { Injectable } from '@nestjs/common';
+import {
+  BadRequestException,
+  Injectable,
+  NotFoundException,
+} from '@nestjs/common';
 import { session_userprofile } from '@prisma/client';
 import {
   PlannerBodyDto,
   PlannerQueryDto,
+  PlannerRemoveItemDto,
 } from 'src/common/interfaces/dto/planner/planner.request.dto';
 import { PlannerResponseDto } from 'src/common/interfaces/dto/planner/planner.response.dto';
 import { toJsonPlanner } from 'src/common/interfaces/serializer/planner.serializer';
@@ -93,4 +98,47 @@ export class PlannersService {
 
     return toJsonPlanner(planner);
   }
+
+  public async removePlannerItem(
+    plannerId: number,
+    removeItem: PlannerRemoveItemDto,
+    user: session_userprofile,
+  ): Promise<PlannerResponseDto> {
+    switch (removeItem.item_type) {
+      case 'TAKEN':
+        throw new BadRequestException(
+          'Planner item with type "taken" can\'t be deleted',
+        );
+      case 'FUTURE': {
+        const futureItem =
+          await this.PlannerRepository.getFuturePlannerItemById(
+            user,
+            removeItem.item,
+          );
+        if (futureItem.planner_id !== plannerId) {
+          throw new NotFoundException();
+        }
+        await this.PlannerRepository.deleteFuturePlannerItem(futureItem);
+        break;
+      }
+      case 'ARBITRARY': {
+        const arbitraryItem =
+          await this.PlannerRepository.getArbitraryPlannerItemById(
+            user,
+            removeItem.item,
+          );
+        if (arbitraryItem.planner_id !== plannerId) {
+          throw new NotFoundException();
+        }
+        await this.PlannerRepository.deleteArbitraryPlannerItem(arbitraryItem);
+        break;
+      }
+    }
+
+    const planner = await this.PlannerRepository.getPlannerById(
+      user,
+      plannerId,
+    );
+    return toJsonPlanner(planner);
+  }
 }
diff --git a/src/prisma/repositories/planner.repository.ts b/src/prisma/repositories/planner.repository.ts
@@ -79,6 +79,23 @@ export class PlannerRepository {
     });
   }
 
+  public async getPlannerById(
+    user: session_userprofile,
+    id: number,
+  ): Promise<PlannerDetails> {
+    const planner = await this.prisma.planner_planner.findFirst({
+      ...plannerDetails,
+      where: {
+        user_id: user.id,
+        id: id,
+      },
+    });
+    if (!planner) {
+      throw new NotFoundException();
+    }
+    return planner;
+  }
+
   public async getRelatedPlanner(
     user: session_userprofile,
   ): Promise<PlannerBasic[]> {
@@ -190,6 +207,14 @@ export class PlannerRepository {
     });
   }
 
+  public async deleteFuturePlannerItem(target_item: FuturePlannerItem) {
+    return await this.prisma.planner_futureplanneritem.delete({
+      where: {
+        id: target_item.id,
+      },
+    });
+  }
+
   public async getArbitraryPlannerItemById(
     user: session_userprofile,
     id: number,
@@ -247,4 +272,12 @@ export class PlannerRepository {
       },
     });
   }
+
+  public async deleteArbitraryPlannerItem(target_item: ArbitraryPlannerItem) {
+    return await this.prisma.planner_arbitraryplanneritem.delete({
+      where: {
+        id: target_item.id,
+      },
+    });
+  }
 }