[#14] Feat: tomorrow, days, specific date

Author: 김산

src/main/java/com/example/helper/constant/Messages.java

@@ -5,6 +5,7 @@ public enum Messages {
     NO_EXIST_MEAL_ERROR("조건에 맞는 식단이 존재하지 않습니다."),
     NO_MEAL_KOR("식단 준비중입니다."),
     NO_MEAL_ENG("The meal is being prepared."),
+    INVALID_DATE("유효하지 않은 날짜입니다."),
     DUMMY_MEAL_KOR("2023-01-27 조식\n\n제2학생회관1층\n\n흰밥*김가루양념밥\n"),
     DUMMY_MEAL_ENG("2023-01-27 Breakfast\n\nStudent Union Bldg.2 1st floor\n\nWhite rice*Seasoned rice with seaweed\n");
     private String message;

src/main/java/com/example/helper/constant/SpecMealInputsKor.java

@@ -34,6 +34,6 @@ public static Integer getTypeByString(String str){
             if (each.getInputs().equals(str)) {
                 return each.getInputValue();
             }
-        return null;
+        return -1;
     }
 }

src/main/java/com/example/helper/controller/MealController.java

@@ -7,6 +7,8 @@
 import com.example.helper.service.MealService;
 import com.fasterxml.jackson.core.JsonProcessingException;
 import com.fasterxml.jackson.databind.ObjectMapper;
+import java.time.LocalDateTime;
+import java.time.ZoneId;
 import java.util.Map.Entry;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.BeanUtils;
@@ -95,7 +97,8 @@ public Map<String, Object> readSpecKorMeal(@RequestBody Map<String, Object> requ
         String dateCustom = params.get("dateCustom").toString();
         String bld =  params.get("bld").toString();
 
-        String specMeal = mealService.getSpecKorMeal(dateCustom, bld);
+        LocalDateTime currentDateTime = LocalDateTime.now(ZoneId.of("Asia/Seoul"));
+        String specMeal = mealService.getSpecKorMeal(dateCustom, bld, currentDateTime);
         Map<String, Object> responseBody = mealService.responseMeal(specMeal);
 
         return responseBody;

src/main/java/com/example/helper/service/MealService.java

@@ -102,36 +102,50 @@ public String getNowEngMeal() {
         return getNowMeal(Types.LANG_ENG.getType());
     }
 
-    public String getSpecKorMeal(String dateCustom, String bld) {
+    public String getSpecKorMeal(String dateCustom, String bld, LocalDateTime currentDateTime) {
+        // test를 위해 LocalDateTime 객체를 argument로 받도록 변경
+
         if (!specInputValidation(dateCustom, bld)) {
             return Messages.NO_MEAL_KOR.getMessages();
         }
 
-        LocalDateTime currentDateTime = LocalDateTime.now(ZoneId.of("Asia/Seoul"));
-
-        if(dateCustom.equals(SpecMealInputsKor.TODAY.getInputs())) {
-            // 오늘
-            String date = currentDateTime.getYear() + "-";
-            date += String.format("%02d", currentDateTime.getMonth().getValue()) + "-";
-            date += String.format("%02d", currentDateTime.getDayOfMonth()) + "";
-
-            Optional<Meal> result = sqlMealRepository.findByDate(
-                    Types.BLDG2_1ST.getType(),
-                    Types.LANG_KOR.getType(),
-                    SpecMealInputsKor.getTypeByString(bld),
-                    date);
-        }
-        else if(dateCustom.equals(SpecMealInputsKor.TOMORROW.getInputs())) {
-            // 내일
-        }
-        else if(dateCustom.length() == 1) {
-            // 요일
+        try {
+            if (dateCustom.equals(SpecMealInputsKor.TOMORROW.getInputs())) {
+                // 내일
+                currentDateTime = currentDateTime.plusDays(1);
+            } else if (dateCustom.length() == 1) {
+                // 요일
+                Integer dateDiff = getDateDifference(dateCustom, currentDateTime);
+                currentDateTime = currentDateTime.plusDays(dateDiff);
+            } else if ((dateCustom.charAt(dateCustom.length() - 1) + "").equals(SpecMealInputsKor.DAY.getInputs())) {
+                // 특정날짜
+                currentDateTime = currentDateTime.withDayOfMonth(
+                        Integer.parseInt(dateCustom.substring(0, dateCustom.length() - 1)));
+            }
+        } catch (Exception e) {
+            return Messages.INVALID_DATE.getMessages();
         }
-        else if((dateCustom.charAt(dateCustom.length() - 1) + "").equals(SpecMealInputsKor.DAY.getInputs())) {
-            // 특정날짜
+
+        String date = currentDateTime.getYear() + "-";
+        date += String.format("%02d", currentDateTime.getMonth().getValue()) + "-";
+        date += String.format("%02d", currentDateTime.getDayOfMonth()) + "";
+
+        Optional<Meal> result = sqlMealRepository.findByDate(
+                Types.BLDG2_1ST.getType(),
+                Types.LANG_KOR.getType(),
+                SpecMealInputsKor.getTypeByString(bld),
+                date);
+
+        // TODO : 존재하지 않는 식단이면, error message 반환.
+        if(result.isEmpty()) {
+            return Messages.NO_MEAL_KOR.getMessages();
         }
 
-        return "2023-01-27 조식\n\n제2학생회관1층\n\n흰밥*김가루양념밥\n";
+        return result.get().generateMenu();
+    }
+
+    public Integer getDateDifference(String day, LocalDateTime currentDateTime) {
+        return (Integer) currentDateTime.getDayOfWeek().getValue() - SpecMealInputsKor.getTypeByString(day);
     }
 
     public String getSpecEngMeal(String dateCustom, String bld) {