Merge pull request #55 from marshmallowing/feat/image

#45 Feat: Presigned URL 기능 구현

Author: marshmallowing

build.gradle

@@ -55,6 +55,9 @@ dependencies {
 	implementation 'org.webjars:sockjs-client:1.0.2'
 	implementation 'org.webjars:stomp-websocket:2.3.3'
 	implementation 'org.webjars:jquery:3.1.1-1'
+
+	// s3 관련
+	implementation 'org.springframework.cloud:spring-cloud-starter-aws:2.2.6.RELEASE'
 }
 
 tasks.named('test') {

src/main/java/com/memesphere/domain/image/controller/ImageController.java

@@ -0,0 +1,31 @@
+package com.memesphere.domain.image.controller;
+
+import com.memesphere.domain.image.service.ImageService;
+import com.memesphere.global.apipayload.ApiResponse;
+import io.swagger.v3.oas.annotations.Operation;
+import lombok.RequiredArgsConstructor;
+import org.springframework.http.MediaType;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RequestParam;
+import org.springframework.web.bind.annotation.RestController;
+import org.springframework.web.multipart.MultipartFile;
+
+import java.io.IOException;
+
+@RequestMapping("/s3")
+@RestController
+@RequiredArgsConstructor
+public class ImageController {
+
+    private final ImageService imageService;
+
+    @PostMapping(value="/upload", consumes = MediaType.MULTIPART_FORM_DATA_VALUE, produces = MediaType.APPLICATION_JSON_VALUE)
+    @Operation(summary = "이미지 Presigned URL 발급",
+            description = "사용자가 이미지를 업로드하고, S3에 저장한 후 해당 이미지에 대한 presigned URL을 발급받습니다. \n" +
+                    "업로드할 파일(JPG, JPEG, PNG만 가능)은 'file'이라는 키로 multipart/form-data 형식으로 첨부해야 합니다.")
+    public ApiResponse<String> uploadFile(@RequestParam("file") MultipartFile file) throws IOException {
+        String url = imageService.uploadFile(file);
+        return ApiResponse.onSuccess(url);
+    }
+}

src/main/java/com/memesphere/domain/image/service/ImageService.java

@@ -0,0 +1,80 @@
+package com.memesphere.domain.image.service;
+
+import com.amazonaws.SdkClientException;
+import com.amazonaws.services.s3.AmazonS3;
+import com.amazonaws.services.s3.model.ObjectMetadata;
+import com.memesphere.global.apipayload.code.status.ErrorStatus;
+import com.memesphere.global.apipayload.exception.GeneralException;
+import lombok.RequiredArgsConstructor;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.stereotype.Service;
+import org.springframework.web.multipart.MultipartFile;
+
+import java.io.IOException;
+import java.util.Arrays;
+import java.util.List;
+import java.util.UUID;
+
+@Service
+@RequiredArgsConstructor
+public class ImageService {
+
+    @Value("${cloud.aws.s3.bucket}")
+    private String bucket;
+
+    private final AmazonS3 amazonS3;
+    private String defaultUrl = "https://s3.amazonaws.com/";
+
+    // S3에 파일을 업로드
+    public String uploadFile(MultipartFile file) throws IOException {
+        if(file.isEmpty()){
+            throw new GeneralException(ErrorStatus.EMPTY_FILE_EXCEPTION);
+        }
+
+        validateImageFileExtension(file.getOriginalFilename());
+
+        String fileName=generateFileName(file);
+        try{
+            amazonS3.putObject(bucket, fileName,file.getInputStream(),getObjectMetadata(file));
+            return defaultUrl+fileName;
+        } catch(SdkClientException e){
+            throw new IOException("error uploading file", e);
+        }
+    }
+
+    // 파일 확장자 유효성 검사
+    private void validateImageFileExtension(String filename) {
+        String extension = getFileExtension(filename);
+
+        List<String> allowedExtensions = Arrays.asList("jpg", "jpeg", "png");
+
+        if (!allowedExtensions.contains(extension)) {
+            throw new GeneralException(ErrorStatus.INVALID_FILE_EXTENTION);
+        }
+    }
+
+    // 파일 확장자 추출
+    private String getFileExtension(String filename) {
+        int lastDotIndex = filename.lastIndexOf(".");
+
+        // 확장자 없을 시 예외 발생
+        if (lastDotIndex == -1) {
+            throw new GeneralException(ErrorStatus.INVALID_FILE_EXTENTION);
+        }
+
+        return filename.substring(lastDotIndex + 1).toLowerCase();
+    }
+
+    // MultipartFile의 사이즈와 타입을 S3에 전달하기 위한 메타데이터 생성
+    private ObjectMetadata getObjectMetadata(MultipartFile file){
+        ObjectMetadata objectMetadata = new ObjectMetadata();
+        objectMetadata.setContentLength(file.getSize());
+        objectMetadata.setContentType(file.getContentType());
+        return objectMetadata;
+    }
+
+    // filename 설정 (UUID 이용)
+    private String generateFileName(MultipartFile file){
+        return UUID.randomUUID().toString() + "-"+file.getOriginalFilename();
+    }
+}

src/main/java/com/memesphere/global/apipayload/code/status/ErrorStatus.java

@@ -30,7 +30,11 @@ public enum ErrorStatus implements BaseCode {
     USER_NOT_FOUND(HttpStatus.NOT_FOUND, "USER NOT FOUND", "유저를 찾을 수 없습니다."),
     PASSWORD_NOT_MATCH(HttpStatus.NOT_FOUND, "PASSWORD NOT MATCH", "비밀번호가 틀렸습니다."),
     USER_ALREADY_EXISTS(HttpStatus.BAD_REQUEST, "USER ALREADY EXIST ", "이미 존재하는 회원입니다."),
-    NICKNAME_ALREADY_EXISTS(HttpStatus.BAD_REQUEST, "NICKNAME ALREADY EXIST ", "이미 사용 중인 닉네임입니다.");
+    NICKNAME_ALREADY_EXISTS(HttpStatus.BAD_REQUEST, "NICKNAME ALREADY EXIST ", "이미 사용 중인 닉네임입니다."),
+
+    // 이미지 에러
+    EMPTY_FILE_EXCEPTION(HttpStatus.BAD_REQUEST, "EMPTY_FILE", "이미지가 빈 파일입니다."),
+    INVALID_FILE_EXTENTION(HttpStatus.BAD_REQUEST, "INVALID FILE EXTENSION", "지원되지 않는 파일 형식입니다.");
 
     private final HttpStatus httpStatus;
     private final String code;

src/main/java/com/memesphere/global/config/AmazonConfig.java

@@ -0,0 +1,40 @@
+package com.memesphere.global.config;
+
+import com.amazonaws.auth.AWSCredentials;
+import com.amazonaws.auth.AWSCredentialsProvider;
+import com.amazonaws.auth.AWSStaticCredentialsProvider;
+import com.amazonaws.auth.BasicAWSCredentials;
+import com.amazonaws.services.s3.AmazonS3;
+import com.amazonaws.services.s3.AmazonS3ClientBuilder;
+import jakarta.annotation.PostConstruct;
+import lombok.Getter;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+
+@Configuration
+@Getter
+public class AmazonConfig {
+    private AWSCredentials awsCredentials;
+
+    @Value("${cloud.aws.credentials.accessKey}")
+    private String accessKey;
+
+    @Value("${cloud.aws.credentials.secretKey}")
+    private String secretKey;
+
+    @Value("${cloud.aws.region.static}")
+    private String region;
+
+    @PostConstruct
+    public void init() { this.awsCredentials = new BasicAWSCredentials(accessKey, secretKey); }
+
+    @Bean
+    public AmazonS3 amazonS3() {
+        AWSCredentials awsCredentials1=new BasicAWSCredentials(accessKey, secretKey);
+        return AmazonS3ClientBuilder.standard()
+                .withRegion(region)
+                .withCredentials(new AWSStaticCredentialsProvider(awsCredentials))
+                .build();
+    }
+}