refactor: MediaServiceImplをリファクタリング

src/main/kotlin/dev/usbharu/hideout/core/service/media/MediaServiceImpl.kt

@@ -6,6 +6,7 @@ import dev.usbharu.hideout.core.domain.model.media.Media
 import dev.usbharu.hideout.core.domain.model.media.MediaRepository
 import dev.usbharu.hideout.core.service.media.converter.MediaProcessService
 import dev.usbharu.hideout.mastodon.interfaces.api.media.MediaRequest
+import dev.usbharu.hideout.util.withDelete
 import io.ktor.client.*
 import io.ktor.client.request.*
 import io.ktor.client.statement.*
@@ -38,32 +39,32 @@ class MediaServiceImpl(
         )
 
         val tempFile = Files.createTempFile("hideout-tmp-file", ".tmp")
-        AutoCloseable { println(tempFile);Files.delete(tempFile) }.use {
+
+
+
+        tempFile.withDelete().use {
             Files.newOutputStream(tempFile).use { outputStream ->
                 mediaRequest.file.inputStream.use {
                     it.transferTo(outputStream)
                 }
             }
-            val mimeType = fileTypeDeterminationService.fileType(
+            val mimeType = fileTypeDeterminationService.fileType(tempFile, fileName)
-                tempFile,
+
-                fileName,
+            val process = findMediaProcessor(mimeType).process(
-            )
-            val process = try {
-                mediaProcessServices.first { it.isSupport(mimeType) }.process(
                 mimeType,
                 fileName,
                 tempFile,
                 null
             )
-            } catch (e: NoSuchElementException) {
+
-                throw UnsupportedMediaException("MediaType: $mimeType isn't supported.")
-            }
             val dataMediaSave = MediaSaveRequest(
                 process.filePath.fileName.toString(),
                 "",
                 process.filePath,
                 process.thumbnailPath
             )
+            dataMediaSave.filePath.withDelete().use {
+                dataMediaSave.thumbnailPath.withDelete().use {
                     val save = try {
                         mediaDataStore.save(dataMediaSave)
                     } catch (e: Exception) {
@@ -77,27 +78,7 @@ class MediaServiceImpl(
                         throw MediaSaveException("Failed to save the media.")
                     }
                     save as SuccessSavedMedia
-            val blurHash = withContext(Dispatchers.IO) {
+                    val blurHash = generateBlurhash(process)
-                if (process.thumbnailPath != null && process.thumbnailMimeType != null) {
-                    val iterator =
-                        ImageIO.getImageReadersByMIMEType(process.thumbnailMimeType.type + "/" + process.thumbnailMimeType.subtype)
-                    for (imageReader in iterator) {
-                        try {
-                            ImageIO.createImageInputStream(process.thumbnailPath.toFile()).use {
-                                imageReader.input = it
-                                val read = imageReader.read(0)
-                                return@withContext mediaBlurhashService.generateBlurhash(read)
-                            }
-                        } catch (e: Exception) {
-                            logger.warn("Failed to read thumbnail", e)
-                        }
-
-                    }
-                    ""
-                } else {
-                    ""
-                }
-            }
                     return mediaRepository.save(
                         EntityMedia(
                             id = mediaRepository.generateId(),
@@ -111,6 +92,8 @@ class MediaServiceImpl(
                         )
                     )
                 }
+            }
+        }
 
 
     }
@@ -178,6 +161,47 @@ class MediaServiceImpl(
         )
     }
 
+    protected fun findMediaProcessor(mimeType: MimeType): MediaProcessService {
+        try {
+            return mediaProcessServices.first {
+                try {
+                    it.isSupport(mimeType)
+                } catch (e: Exception) {
+                    false
+                }
+            }
+        } catch (e: NoSuchElementException) {
+            throw UnsupportedMediaException("MediaType: $mimeType isn't supported.")
+        }
+    }
+
+    protected fun generateBlurhash(process: ProcessedMediaPath): String {
+        val path = if (process.thumbnailPath != null && process.thumbnailMimeType != null) {
+            process.thumbnailPath
+        } else {
+            process.filePath
+        }
+        val mimeType = if (process.thumbnailPath != null && process.thumbnailMimeType != null) {
+            process.thumbnailMimeType
+        } else {
+            process.fileMimeType
+        }
+
+        val imageReadersByMIMEType = ImageIO.getImageReadersByMIMEType(mimeType.type + "/" + mimeType.subtype)
+        for (imageReader in imageReadersByMIMEType) {
+            try {
+                val bufferedImage = ImageIO.createImageInputStream(path.toFile()).use {
+                    imageReader.input = it
+                    imageReader.read(0)
+                }
+                return mediaBlurhashService.generateBlurhash(bufferedImage)
+            } catch (e: Exception) {
+                logger.warn("Failed to read thumbnail", e)
+            }
+        }
+        return ""
+    }
+
     companion object {
         private val logger = LoggerFactory.getLogger(MediaServiceImpl::class.java)
     }

src/main/kotlin/dev/usbharu/hideout/util/TempFileUtil.kt

@@ -0,0 +1,12 @@
+package dev.usbharu.hideout.util
+
+import java.nio.file.Files
+import java.nio.file.Path
+
+fun Path?.withDelete(): TempFile = TempFile(this)
+
+class TempFile(val path: Path?) : AutoCloseable {
+    override fun close() {
+        path?.let { Files.deleteIfExists(it) }
+    }
+}