Fix: 리라이팅 결과 저장 시 이미지 첨부

Fix: 리라이팅 결과 저장 시 이미지 첨부

Author: junsoo22

src/main/java/com/tave/PromptMate/controller/CommunityController.java

@@ -1,8 +1,6 @@
 package com.tave.PromptMate.controller;
 
 import com.tave.PromptMate.auth.dto.request.CustomUserDetails;
-import com.tave.PromptMate.common.S3Uploader;
-import com.tave.PromptMate.domain.Community;
 import com.tave.PromptMate.domain.Platform;
 import com.tave.PromptMate.domain.PromptCategory;
 import com.tave.PromptMate.dto.community.*;
@@ -12,13 +10,10 @@
 import io.swagger.v3.oas.annotations.tags.Tag;
 import jakarta.validation.Valid;
 import lombok.RequiredArgsConstructor;
-import org.springframework.http.MediaType;
 import org.springframework.http.ResponseEntity;
 import org.springframework.security.core.annotation.AuthenticationPrincipal;
 import org.springframework.web.bind.annotation.*;
-import org.springframework.web.multipart.MultipartFile;
 
-import java.io.IOException;
 import java.net.URI;
 import java.util.List;
 
@@ -29,31 +24,16 @@
 public class CommunityController {
 
     private final CommunityService communityService;
-    private final S3Uploader s3Uploader;
     private final CommunityRecentService communityRecentService;
 
     // 커뮤니티 글 작성
-    @PostMapping(value="/posts", consumes= MediaType.MULTIPART_FORM_DATA_VALUE)
-    @Operation(summary = "게시글 작성", description = "게시글 작성 입니다.")
+    @PostMapping("/posts")
+    @Operation(summary = "게시글 작성", description = "게시글 작성 입니다. 이미지는 라이브러리 저장 시 첨부됩니다.")
     public ResponseEntity<CommunityPostResponse> createPost(
             @AuthenticationPrincipal CustomUserDetails principal,
-            @RequestParam("rewriteResultId") Long rewriteResultId,
-            @RequestParam("title") String title,
-            @RequestParam("description") String description,
-            @RequestParam("visibility") Community.Visibility visibility,
-            @RequestPart(value="image" ,required = false) MultipartFile image
-            ) throws IOException {
-        Long userId=principal.getUserId();
-
-        // 이미지 업로드
-        String imageUrl = null;
-        if (image != null && !image.isEmpty()) {
-            imageUrl = s3Uploader.uploadImage(image, "community");
-        }
-
-        CreateCommunityPostRequest request = new CreateCommunityPostRequest(
-                rewriteResultId, title, description, visibility,imageUrl
-        );
+            @Valid @RequestBody CreateCommunityPostRequest request
+    ) {
+        Long userId = principal.getUserId();
 
         CommunityPostResponse response = communityService.createPost(request, userId);
 

src/main/java/com/tave/PromptMate/controller/LibraryController.java

@@ -1,19 +1,24 @@
 package com.tave.PromptMate.controller;
 
 import com.tave.PromptMate.auth.dto.request.CustomUserDetails;
+import com.tave.PromptMate.common.S3Uploader;
+import com.tave.PromptMate.domain.Platform;
+import com.tave.PromptMate.domain.PromptCategory;
 import com.tave.PromptMate.dto.community.CommunityPostResponse;
 import com.tave.PromptMate.dto.library.CreateLibraryRequest;
 import com.tave.PromptMate.dto.library.LibraryResponse;
 import com.tave.PromptMate.service.LibraryService;
 import io.swagger.v3.oas.annotations.Operation;
 import io.swagger.v3.oas.annotations.tags.Tag;
-import jakarta.validation.Valid;
 import lombok.RequiredArgsConstructor;
 import org.springframework.data.domain.Page;
+import org.springframework.http.MediaType;
 import org.springframework.http.ResponseEntity;
 import org.springframework.security.core.annotation.AuthenticationPrincipal;
 import org.springframework.web.bind.annotation.*;
+import org.springframework.web.multipart.MultipartFile;
 
+import java.io.IOException;
 import java.net.URI;
 import java.util.List;
 
@@ -24,15 +29,32 @@
 public class LibraryController {
 
     private final LibraryService libraryService;
+    private final S3Uploader s3Uploader;
 
     // 라이브러리에 리라이팅 결과 저장하기
-    @PostMapping
-    @Operation(summary = "리라이팅 결과 저장", description = "리라이팅된 결과를 라이브러리에 저장합니다.")
+    @PostMapping(consumes = MediaType.MULTIPART_FORM_DATA_VALUE)
+    @Operation(summary = "리라이팅 결과 저장", description = "리라이팅된 결과를 라이브러리에 저장합니다. 이미지 첨부 가능.")
     public ResponseEntity<LibraryResponse> save(
             @AuthenticationPrincipal CustomUserDetails principal,
-            @Valid @RequestBody CreateLibraryRequest req){
-       Long userId= principal.getUserId();
-       LibraryResponse res=libraryService.save(userId,req);
+            @RequestParam("rewriteResultId") Long rewriteResultId,
+            @RequestParam(value = "savedTitle", required = false) String savedTitle,
+            @RequestParam("platform") Platform platform,
+            @RequestParam("category") PromptCategory category,
+            @RequestPart(value = "image", required = false) MultipartFile image
+    ) throws IOException {
+        Long userId = principal.getUserId();
+
+        // 이미지 업로드
+        String imageUrl = null;
+        if (image != null && !image.isEmpty()) {
+            imageUrl = s3Uploader.uploadImage(image, "library");
+        }
+
+        CreateLibraryRequest req = new CreateLibraryRequest(
+                rewriteResultId, savedTitle, platform, category, imageUrl
+        );
+
+        LibraryResponse res = libraryService.save(userId, req);
 
         return ResponseEntity.created(URI.create("/api/libraries/" + res.id()))
                 .body(res);

src/main/java/com/tave/PromptMate/domain/Library.java

@@ -42,6 +42,9 @@ public class Library {
     @Column(nullable = false)
     private PromptCategory category;
 
+    @Column(length = 500)
+    private String imageUrl;
+
     @Column(name = "created_at", nullable = false, updatable = false)
     private LocalDateTime createdAt;
 

src/main/java/com/tave/PromptMate/dto/community/CreateCommunityPostRequest.java

@@ -16,7 +16,5 @@ public record CreateCommunityPostRequest(
         String description,
 
         @NotNull
-        Visibility visibility,
-
-        String imageUrl
+        Visibility visibility
 ) {}

src/main/java/com/tave/PromptMate/dto/library/CreateLibraryRequest.java

@@ -8,5 +8,6 @@ public record CreateLibraryRequest(
         @NotNull Long rewriteResultId,
         String savedTitle,
         Platform platform,
-        PromptCategory category
+        PromptCategory category,
+        String imageUrl
 ) {}

src/main/java/com/tave/PromptMate/dto/library/LibraryMapper.java

@@ -13,6 +13,7 @@ public static LibraryResponse toResponse(Library e) {
                 e.getRewriteResult().getContent(),
                 e.getPlatform(),
                 e.getCategory(),
+                e.getImageUrl(),
                 e.getCreatedAt()
         );
     }

src/main/java/com/tave/PromptMate/dto/library/LibraryResponse.java

@@ -13,5 +13,6 @@ public record LibraryResponse(
         String content,
         Platform platform,
         PromptCategory category,
+        String imageUrl,
         LocalDateTime createdAt
 ) {}

src/main/java/com/tave/PromptMate/service/CommunityService.java

@@ -43,8 +43,7 @@ public CommunityPostResponse createPost(CreateCommunityPostRequest req, Long use
                 req.visibility(),
                 lib.getPlatform(),
                 lib.getCategory(),
-                req.imageUrl()
-
+                lib.getImageUrl()
         );
 
         Community savedCommunity = communityRepository.save(community);

src/main/java/com/tave/PromptMate/service/LibraryService.java

@@ -55,6 +55,7 @@ public LibraryResponse save(Long userId,CreateLibraryRequest req) {
                 .savedTitle(req.savedTitle())
                 .platform(req.platform())
                 .category(req.category())
+                .imageUrl(req.imageUrl())
                 .build();
 
         return LibraryMapper.toResponse(libraryRepository.save(library));