From 805d19a646df9580693d226dd5414678da66e654 Mon Sep 17 00:00:00 2001
From: hwangsungmin <hwangsungmin_00@naver.com>
Date: Sat, 5 Oct 2024 15:48:43 +0900
Subject: [PATCH] [fix] str to txt, file download

---
 .../volare/controller/ScriptController.java   |  8 +++
 .../example/volare/service/ScriptService.java | 62 +++++++++++++++++++
 2 files changed, 70 insertions(+)

diff --git a/volare/src/main/java/com/example/volare/controller/ScriptController.java b/volare/src/main/java/com/example/volare/controller/ScriptController.java
index 7e36453..4649b69 100644
--- a/volare/src/main/java/com/example/volare/controller/ScriptController.java
+++ b/volare/src/main/java/com/example/volare/controller/ScriptController.java
@@ -10,10 +10,12 @@
 import com.fasterxml.jackson.core.JsonProcessingException;
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
+import org.springframework.core.io.InputStreamResource;
 import org.springframework.http.ResponseEntity;
 import org.springframework.security.core.annotation.AuthenticationPrincipal;
 import org.springframework.web.bind.annotation.*;
 
+import java.io.IOException;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
@@ -78,4 +80,10 @@ public ResponseEntity<Map<String, Object>> updateScript(
     public ApiResponse<StatisticsDTO.ScriptDetailsDTO> getScriptDetails(@PathVariable Long scriptId) {
         return ApiResponse.onSuccess(scriptService.getScriptInfo(scriptId));
     }
+
+    // Script ID로 txt 파일 다운로드
+    @GetMapping("/{scriptId}/download")
+    public ResponseEntity<InputStreamResource> downloadScriptTxtFile(@PathVariable Long scriptId) throws IOException {
+        return scriptService.downloadScriptTxtFile(scriptId);
+    }
 }
diff --git a/volare/src/main/java/com/example/volare/service/ScriptService.java b/volare/src/main/java/com/example/volare/service/ScriptService.java
index 3be917f..72f0410 100644
--- a/volare/src/main/java/com/example/volare/service/ScriptService.java
+++ b/volare/src/main/java/com/example/volare/service/ScriptService.java
@@ -14,11 +14,21 @@
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.cache.annotation.Cacheable;
+import org.springframework.core.io.InputStreamResource;
+import org.springframework.http.HttpHeaders;
+import org.springframework.http.MediaType;
+import org.springframework.http.ResponseEntity;
 import org.springframework.stereotype.Service;
 
+import java.io.FileInputStream;
+import java.io.IOException;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.nio.file.Paths;
 import java.time.LocalDateTime;
 import java.util.List;
 import java.util.Map;
+import java.util.Optional;
 import java.util.stream.Collectors;
 import com.example.volare.model.ScriptScene;
 
@@ -161,4 +171,56 @@ public List<Map<String, Object>> updateScriptItems(List<Map<String, Object>> uLi
         return uList;
     }
 
+    // Script 데이터를 txt 파일로 변환하는 메서드 추가
+    public Path generateScriptTxtFile(Long scriptId) throws IOException {
+        // Script 조회
+        Optional<Script> optionalScript = scriptRepository.findById(scriptId);
+        if (optionalScript.isEmpty()) {
+            throw new GeneralHandler(ErrorStatus._BAD_REQUEST);
+        }
+
+        Script script = optionalScript.get();
+
+        // 파일 경로 설정
+        Path filePath = Paths.get("script_" + scriptId + ".txt");
+
+        // Script 데이터를 텍스트 파일에 쓰기
+        StringBuilder sb = new StringBuilder();
+        sb.append("Script ID: ").append(script.getId()).append("\n");
+        sb.append("Novel Title: ").append(script.getNovel().getTitle()).append("\n");
+        sb.append("Type: ").append(script.getType()).append("\n");
+        sb.append("Script File Content: \n").append(script.getScriptFile()).append("\n");
+        sb.append("Characters: \n");
+
+        for (String character : script.getCharacters()) {
+            sb.append("- ").append(character).append("\n");
+        }
+
+        sb.append("Scenes: \n");
+        for (ScriptScene scene : script.getScriptScenes()) {
+            sb.append("Scene Location: ").append(scene.getLocation()).append("\n");
+            // 필요한 경우 더 많은 Scene 정보를 출력할 수 있습니다.
+        }
+
+        Files.write(filePath, sb.toString().getBytes());
+
+        return filePath;
+    }
+
+    // 파일 다운로드를 위한 API 응답 처리
+    public ResponseEntity<InputStreamResource> downloadScriptTxtFile(Long scriptId) throws IOException {
+        // 텍스트 파일 생성
+        Path filePath = generateScriptTxtFile(scriptId);
+
+        // 파일을 InputStream으로 읽어들임
+        InputStreamResource resource = new InputStreamResource(new FileInputStream(filePath.toFile()));
+
+        // 파일 다운로드 응답 반환
+        return ResponseEntity.ok()
+                .header(HttpHeaders.CONTENT_DISPOSITION, "attachment;filename=" + filePath.getFileName())
+                .contentType(MediaType.TEXT_PLAIN)
+                .contentLength(filePath.toFile().length())
+                .body(resource);
+    }
+
 }