Merge pull request #49 from NU-BIM/refactor/album/presignedurl

S3 Presigned URL 생성 시 컨텐츠 타입 검증 추가

src/main/java/com/soyeon/nubim/domain/album/AlbumPhotoUploadUrlService.java

@@ -5,20 +5,32 @@
 import org.springframework.stereotype.Service;
 
 import com.soyeon.nubim.common.util.aws.S3PresignedUrlGenerator;
+import com.soyeon.nubim.domain.album.exception.InvalidContentTypeException;
 
 import lombok.RequiredArgsConstructor;
 
 @Service
 @RequiredArgsConstructor
 public class AlbumPhotoUploadUrlService {
 
+	public static final String MIME_TYPE_IMAGE = "image";
 	private final S3PresignedUrlGenerator s3PresignedUrlGenerator;
 
 	public List<String> generatePhotoUploadUrlsWithRandomPath(List<String> contentTypes) {
+		validateContentTypes(contentTypes);
 		return s3PresignedUrlGenerator.generatePresignedUrls(contentTypes);
 	}
 
 	public List<String> generatePhotoUploadUrlsWithCustomPath(List<String> contentTypes, String uploadPath) {
+		validateContentTypes(contentTypes);
 		return s3PresignedUrlGenerator.generatePresignedUrls(contentTypes, uploadPath);
 	}
+
+	private void validateContentTypes(List<String> contentTypes) {
+		for (String contentType : contentTypes) {
+			if (!contentType.startsWith(MIME_TYPE_IMAGE)) {
+				throw new InvalidContentTypeException("Invalid content type: " + contentType);
+			}
+		}
+	}
 }

src/main/java/com/soyeon/nubim/domain/album/exception/InvalidContentTypeException.java

@@ -0,0 +1,10 @@
+package com.soyeon.nubim.domain.album.exception;
+
+import org.springframework.http.HttpStatus;
+import org.springframework.web.server.ResponseStatusException;
+
+public class InvalidContentTypeException extends ResponseStatusException {
+	public InvalidContentTypeException(String message) {
+		super(HttpStatus.UNSUPPORTED_MEDIA_TYPE, "Invalid content type: " + message);
+	}
+}