♻️ :: s3 refactor

src/main/kotlin/com/example/sharing/global/error/exception/ErrorCode.kt

@@ -6,6 +6,7 @@ enum class ErrorCode(
 ) {
     PASSWORD_MISMATCED(400, "Password Mis Matched"),
     FEIGN_BAD_REQUEST(400, "Feign Bad Request"),
+    BAD_FILE_EXTENSION(400, "Bad File Extension"),
 
     TOKEN_EXPIRED(401, "Token Expired"),
     TOKEN_INVALID(401, "Token Invalid"),

src/main/kotlin/com/example/sharing/global/utils/aws/s3/S3Utils.kt

@@ -5,6 +5,7 @@ import com.amazonaws.services.s3.model.CannedAccessControlList
 import com.amazonaws.services.s3.model.ObjectMetadata
 import com.amazonaws.services.s3.model.PutObjectRequest
 import com.amazonaws.util.IOUtils
+import com.example.sharing.global.utils.aws.s3.exception.BadFileExtensionException
 import org.springframework.beans.factory.annotation.Value
 import org.springframework.stereotype.Component
 import org.springframework.web.multipart.MultipartFile
@@ -17,15 +18,10 @@ class S3Utils(
 
     @Value("\${cloud.aws.s3.bucket}")
     private val bucketName: String,
-
-    @Value("\${cloud.aws.s3.base-image-url}")
-    private val baseImageUrl: String,
-
-    @Value("\${cloud.aws.s3.prefix}")
-    private val prefix: String
 ) {
     fun upload(file: MultipartFile): String {
-        val fileName = UUID.randomUUID().toString() + "_" + file.originalFilename
+        val ext = verificationFile(file)
+        val fileName = UUID.randomUUID().toString() + "." + ext
         val objMeta = ObjectMetadata()
 
         val bytes = IOUtils.toByteArray(file.inputStream)
@@ -41,4 +37,15 @@ class S3Utils(
 
         return s3Client.getUrl(bucketName, bucketName + fileName).toString()
     }
+
+    private fun verificationFile(file: MultipartFile): String {
+        val originalFilename = file.originalFilename
+        val ext = originalFilename!!.substring(originalFilename.lastIndexOf(".") + 1)
+
+        if (!(ext == "jpg" || ext == "jpeg" || ext == "png" || ext == "heic")) {
+            throw BadFileExtensionException.EXCEPTION
+        }
+        return ext
+    }
+
 }

src/main/kotlin/com/example/sharing/global/utils/aws/s3/exception/BadFileExtensionException.kt

@@ -0,0 +1,11 @@
+package com.example.sharing.global.utils.aws.s3.exception
+
+import com.example.sharing.global.error.exception.ErrorCode.*
+import com.example.sharing.global.error.exception.SharingException
+
+class BadFileExtensionException : SharingException(BAD_FILE_EXTENSION){
+    companion object {
+        @JvmField
+        val EXCEPTION = BadFileExtensionException()
+    }
+}