feat: TreeResponse에서 Type, Level과 텍스트 추가

Author: Na1gae

src/tree/constants.ts

@@ -113,3 +113,9 @@ export const TREE_TYPES_MAP: TreeType[] = [
     ],
   },
 ];
+
+export function getTreeLevel(recommendationCount: number): number{
+  if (recommendationCount <= 10) return 1
+  if (recommendationCount <= 20) return 2
+  else return 3
+}

src/tree/repository.ts

@@ -3,6 +3,7 @@ import { PrismaService } from '../prisma/prisma.service';
 import { Restaurant, SavedRestaurant, User } from '@prisma/client';
 import { Coordinate, PlantTreeDto } from './dto';
 import { TreeDetail } from './types';
+import { getTreeLevel, TREE_TYPES_MAP } from './constants';
 
 const MIN_ZOOM_LEVEL = 11;
 
@@ -48,10 +49,20 @@ export class TreeRepository {
     prismaRes: SavedRestaurant & { user: User; restaurant: Restaurant },
   ): TreeDetail {
     const { user, restaurant, ...tree } = prismaRes;
+    const { recommendedByUsers, treeType } = tree
+    const treeData = TREE_TYPES_MAP[treeType ?? 0];
+    const height = recommendedByUsers?.length ?? 0
+    const treeLevelData = treeData.levels[getTreeLevel(height + 1)]
+
     return {
       user,
       restaurant,
-      tree,
+      tree: {
+        ...tree,
+        level: treeLevelData.level,
+        imageUrl: treeLevelData.imageUrl,
+        treeTypeName: treeData.name
+      }
     };
   }
 

src/tree/types.ts

@@ -3,5 +3,9 @@ import { Restaurant, SavedRestaurant, User } from '@prisma/client';
 export type TreeDetail = {
   user: User;
   restaurant: Restaurant;
-  tree: SavedRestaurant;
+  tree: SavedRestaurant & {
+    level: number,
+    imageUrl: string,
+    treeTypeName: string
+  }
 };