Merge pull request #84 from capgoing/feat/#69

✨ [feat] Picture Quiz 생성 로직 구현 API

Author: khyaejin

src/main/java/com/going/server/domain/quiz/dto/PictureQuizDto.java

@@ -2,8 +2,12 @@
 
 import lombok.Builder;
 import lombok.Getter;
+import java.util.Set;
 
 @Builder
 @Getter
 public class PictureQuizDto {
+    private final String imageUrl;
+    private final Set<String> shuffled;
+    private final String answer;
 }

src/main/java/com/going/server/domain/quiz/exception/QuizErrorCode.java

@@ -0,0 +1,23 @@
+package com.going.server.domain.quiz.exception;
+
+import com.going.server.global.exception.BaseErrorCode;
+import lombok.AllArgsConstructor;
+import lombok.Getter;
+import org.springframework.http.HttpStatus;
+
+import static com.going.server.global.constant.StaticValue.BAD_REQUEST;
+import static com.going.server.global.constant.StaticValue.INTERNAL_SERVER_ERROR;
+
+@Getter
+@AllArgsConstructor
+public enum QuizErrorCode implements BaseErrorCode {
+    // 400
+    UNSUPPORTED_QUIZ_MODE(BAD_REQUEST, "QUIZ_400_1", "지원하지 않는 mode 입니다."),
+
+    // 500
+    QUIZ_IMAGE_GENERATION_FAILED(INTERNAL_SERVER_ERROR, "QUIZ_500_1", "퀴즈 이미지 생성 중 서버 오류가 발생했습니다.");
+
+    private final int httpStatus;
+    private final String code;
+    private final String message;
+}

src/main/java/com/going/server/domain/quiz/exception/QuizImageException.java

@@ -0,0 +1,7 @@
+package com.going.server.domain.quiz.exception;
+
+import com.going.server.global.exception.BaseException;
+
+public class QuizImageException extends BaseException {
+    public QuizImageException(){super(QuizErrorCode.QUIZ_IMAGE_GENERATION_FAILED);}
+}

src/main/java/com/going/server/domain/quiz/exception/UnsupportedQuizModeException.java

@@ -0,0 +1,7 @@
+package com.going.server.domain.quiz.exception;
+
+import com.going.server.global.exception.BaseException;
+
+public class UnsupportedQuizModeException extends BaseException {
+    public UnsupportedQuizModeException() {super(QuizErrorCode.UNSUPPORTED_QUIZ_MODE);}
+}

src/main/java/com/going/server/domain/quiz/generate/ListenUpQuizGenerator.java

@@ -15,10 +15,11 @@ public ListenUpQuizDto generate(Graph graph) {
         Random random = new Random();
         List<ListenUpQuizDto.ListenUpQuiz> quizzes = new ArrayList<>();
         Set<String> usedSentences = new HashSet<>();
-        List<String> options = new ArrayList<>();
+        List<String> candidateSentences = new ArrayList<>(); // 전처리한 includeSentence
 
         // 1. 그래프 노드에서 문장 추출
         for (GraphNode node : graph.getNodes()) {
+            // IncludeSentence가 비어있는 경우 넘어가기
             if (node.getIncludeSentence() == null || node.getIncludeSentence().isBlank()) continue;
 
             // "." 으로 문장 나누기
@@ -32,18 +33,18 @@ public ListenUpQuizDto generate(Graph graph) {
                 String[] words = sentence.split("\\s+");
                 if (words.length < 5) continue; // 5단어 미만은 스킵
 
-                options.add(sentence);
+                candidateSentences.add(sentence);
             }
         }
 
         // 2. 단어 수 기준 정렬 (5단어에 가까운 순서)
-        options.sort(Comparator.comparingInt(
+        candidateSentences.sort(Comparator.comparingInt(
                 s -> Math.abs(s.trim().split("\\s+").length - 5)
         ));
 
         int count = 0;
 
-        for (String sentence : options) {
+        for (String sentence : candidateSentences) {
             if (count >= 3) break;
 
             String[] words = sentence.split("\\s+");

src/main/java/com/going/server/domain/quiz/generate/PictureQuizGenerator.java

@@ -1,15 +1,68 @@
 package com.going.server.domain.quiz.generate;
 
 import com.going.server.domain.graph.entity.Graph;
+import com.going.server.domain.graph.entity.GraphNode;
 import com.going.server.domain.quiz.dto.PictureQuizDto;
 import org.springframework.stereotype.Component;
 
+import java.util.*;
+
 @Component
 public class PictureQuizGenerator implements QuizGenerator<PictureQuizDto> {
 
     @Override
     public PictureQuizDto generate(Graph graph) {
-        // TODO: picture 퀴즈 생성 로직
-        return null;
+        Random random = new Random();
+        Set<String> shuffled = new HashSet<>(); // 보기로 제공할 랜덤 문장들(중복 방지)
+        int shuffledListSize = 3; // 총 보기 수 (정답 포함)
+        List<String> candidateSentences = new ArrayList<>(); // 전처리한 includeSentence
+
+        // 그래프 노드를 돌면서 문장 후보 수집
+        for (GraphNode node : graph.getNodes()) {
+            // IncludeSentence가 비어있는 경우 넘어가기
+            if (node.getIncludeSentence() == null || node.getIncludeSentence().isBlank()) continue;
+
+            // "." 으로 문장 나누기
+            String[] splitSentences = node.getIncludeSentence().split("\\.");
+
+            for (String rawSentence : splitSentences) {
+                String sentence = rawSentence.trim();
+                if (sentence.isBlank()) continue; // 공백은 스킵
+                if (sentence.split("\\s+").length < 5) continue; // 5단어 미만은 스킵
+
+                candidateSentences.add(sentence);
+            }
+        }
+
+        // 후보 문장이 부족할 경우 방어
+        if (candidateSentences.size() < shuffledListSize) {
+            throw new IllegalStateException("생성 가능한 문장이 부족합니다.");
+        }
+
+        // 랜덤으로 문장 선택
+        Collections.shuffle(candidateSentences, random);
+
+        // 보기용 문장 추가
+        Set<String> selectedSentences = new LinkedHashSet<>();
+        for (int i = 0; i < shuffledListSize; i++) {
+            selectedSentences.add(candidateSentences.get(i));
+        }
+
+        // 무작위로 정답 보기 설정
+        int answerIndex = random.nextInt(shuffledListSize);
+        String answer = new ArrayList<>(selectedSentences).get(answerIndex);
+
+        // TODO : 정답 보기 기반으로 이미지 생성
+
+        // TODO : S3에 이미지 업로드
+
+        // TODO : S3 업로드 후 이미지 URL 받아오기
+        String imageUrl = "https://s3-ap-northeast-2.amazonaws.com/~.png"; // 예시 url
+
+        return PictureQuizDto.builder()
+                .imageUrl(imageUrl)
+                .shuffled(selectedSentences)
+                .answer(answer)
+                .build();
     }
 }

src/main/java/com/going/server/domain/quiz/service/QuizServiceImpl.java

@@ -1,25 +1,15 @@
 package com.going.server.domain.quiz.service;
 
-import com.going.server.domain.graph.dto.EdgeDto;
-import com.going.server.domain.graph.dto.KnowledgeGraphDto;
-import com.going.server.domain.graph.dto.NodeDto;
 import com.going.server.domain.graph.entity.Graph;
-import com.going.server.domain.graph.entity.GraphEdge;
-import com.going.server.domain.graph.entity.GraphNode;
 import com.going.server.domain.graph.repository.GraphRepository;
-import com.going.server.domain.quiz.dto.ConnectQuizDto;
-import com.going.server.domain.quiz.dto.ListenUpQuizDto;
-import com.going.server.domain.quiz.dto.PictureQuizDto;
 import com.going.server.domain.quiz.dto.QuizCreateResponseDto;
+import com.going.server.domain.quiz.exception.UnsupportedQuizModeException;
 import com.going.server.domain.quiz.generate.ConnectQuizGenerator;
 import com.going.server.domain.quiz.generate.ListenUpQuizGenerator;
 import com.going.server.domain.quiz.generate.PictureQuizGenerator;
-import com.going.server.domain.quiz.generate.QuizGenerator;
 import lombok.AllArgsConstructor;
 import org.springframework.stereotype.Service;
 
-import java.util.*;
-
 @Service
 @AllArgsConstructor
 public class QuizServiceImpl implements QuizService{
@@ -39,7 +29,7 @@ public QuizCreateResponseDto quizCreate(String graphIdStr, String mode) {
             case "listenUp" -> listenUpQuizGenerator.generate(graph);
             case "connect" -> connectQuizGenerator.generate(graph);
             case "picture" -> pictureQuizGenerator.generate(graph);
-            default -> throw new IllegalArgumentException("지원하지 않는 모드입니다: " + mode);
+            default -> throw new UnsupportedQuizModeException();
         };
 
         return new QuizCreateResponseDto<>(graphIdStr, mode, quizDto);