[feat] #62 upload jpg to S3

lsakee · Sangwook123 · commit f014f1ae9b39 · 2024-07-19T12:34:17.000+09:00
diff --git a/feature/upload/src/main/java/com/record/upload/UploadViewModel.kt b/feature/upload/src/main/java/com/record/upload/UploadViewModel.kt
@@ -5,9 +5,8 @@ import android.util.Log
 import androidx.lifecycle.viewModelScope
 import com.record.ui.base.BaseViewModel
 import com.record.upload.extension.GalleryVideo
-import com.record.upload.extension.extractAndUploadThumbnail
-import com.record.upload.extension.reencodeVideo
 import com.record.upload.extension.uploadFileToS3PresignedUrl
+import com.record.upload.extension.uploadFileToS3ThumbnailPresignedUrl
 import com.record.upload.repository.UploadRepository
 import dagger.hilt.android.lifecycle.HiltViewModel
 import kotlinx.coroutines.launch
@@ -38,12 +37,15 @@ class UploadViewModel @Inject constructor(
         }
     }
 
-    fun uploadVideoToS3Bucket(file: File) =
+    fun uploadVideoToS3Bucket(context: Context,file: File) =
         viewModelScope.launch {
-            Log.d("outputfile", "$file ")
-            Log.d("outputfile", "${uiState.value.bucketUrl} ")
             uploadFileToS3PresignedUrl(uiState.value.bucketUrl, file) { success, message ->
                 println(message)
+                Log.d("messageFile", "$message ")
+            }
+            uploadFileToS3ThumbnailPresignedUrl(context,uiState.value.thumbnailUrl, file) { success, message ->
+                println(message)
+                Log.d("messageThumbnail", "$message ")
             }
         }
 
diff --git a/feature/upload/src/main/java/com/record/upload/VideoPickerScreen.kt b/feature/upload/src/main/java/com/record/upload/VideoPickerScreen.kt
@@ -67,7 +67,6 @@ import com.record.upload.component.bottomsheet.DefinedContentBottomSheet
 import com.record.upload.component.bottomsheet.SelectedVideoBottomSheet
 import com.record.upload.extension.GalleryVideo
 import com.record.upload.extension.getAllVideos
-import com.record.upload.extension.getVideoEncodingInfo
 import kotlinx.coroutines.flow.collectLatest
 import kotlinx.coroutines.launch
 import timber.log.Timber
@@ -96,7 +95,7 @@ fun VideoPickerRoute(
         onClickContentChip = viewModel::setSelectedList,
         setVideo = viewModel::setVideo,
         uploadVideoS3Bucket = {
-            viewModel.uploadVideoToS3Bucket(it)
+            viewModel.uploadVideoToS3Bucket(context,it)
         },
     )
 }
diff --git a/feature/upload/src/main/java/com/record/upload/extension/UploadExtension.kt b/feature/upload/src/main/java/com/record/upload/extension/UploadExtension.kt
@@ -2,6 +2,7 @@ package com.record.upload.extension
 
 import android.content.ContentUris
 import android.content.Context
+import android.graphics.Bitmap
 import android.media.MediaMetadataRetriever
 import android.net.Uri
 import android.provider.MediaStore
@@ -12,8 +13,6 @@ import com.abedelazizshe.lightcompressorlibrary.VideoQuality
 import com.abedelazizshe.lightcompressorlibrary.config.Configuration
 import com.abedelazizshe.lightcompressorlibrary.config.SaveLocation
 import com.abedelazizshe.lightcompressorlibrary.config.SharedStorageConfiguration
-import com.arthenica.mobileffmpeg.Config
-import com.arthenica.mobileffmpeg.FFmpeg
 import kotlinx.coroutines.CoroutineScope
 import kotlinx.coroutines.Dispatchers
 import kotlinx.coroutines.launch
@@ -22,6 +21,7 @@ import okhttp3.OkHttpClient
 import okhttp3.Request
 import okhttp3.RequestBody
 import java.io.File
+import java.io.FileOutputStream
 import java.io.IOException
 
 fun getAllVideos(
@@ -166,52 +166,8 @@ fun formatDuration(durationMillis: Long): String {
     return String.format("%d:%02d", minutes, seconds)
 }
 
-fun getVideoEncodingInfo(context: Context, videoFile: File) {
-    val retriever = MediaMetadataRetriever()
-    try {
-        retriever.setDataSource(context, Uri.fromFile(videoFile))
-        val mimeType = retriever.extractMetadata(MediaMetadataRetriever.METADATA_KEY_MIMETYPE)
-        val duration = retriever.extractMetadata(MediaMetadataRetriever.METADATA_KEY_DURATION)
-        val width = retriever.extractMetadata(MediaMetadataRetriever.METADATA_KEY_VIDEO_WIDTH)
-        val height = retriever.extractMetadata(MediaMetadataRetriever.METADATA_KEY_VIDEO_HEIGHT)
-        val bitrate = retriever.extractMetadata(MediaMetadataRetriever.METADATA_KEY_BITRATE)
-
-        Log.d("VideoInfo", "MIME Type: $mimeType")
-        Log.d("VideoInfo", "Duration: $duration ms")
-        Log.d("VideoInfo", "Width: $width pixels")
-        Log.d("VideoInfo", "Height: $height pixels")
-        Log.d("VideoInfo", "Bitrate: $bitrate bps")
-    } catch (e: Exception) {
-        e.printStackTrace()
-    } finally {
-        retriever.release()
-    }
-}
-
-fun reencodeVideo(inputFile: File, outputFile: File, callback: (Boolean, String) -> Unit) {
-    val command = arrayOf(
-        "-i", inputFile.absolutePath,
-        "-c:v", "libx264", // 비디오 코덱 설정
-        "-b:v", "1000k",  // 비디오 비트레이트 설정
-        "-c:a", "aac",    // 오디오 코덱 설정
-        "-b:a", "192k",   // 오디오 비트레이트 설정
-        "-movflags", "faststart", // 웹 스트리밍을 위해 메타데이터를 파일의 시작 부분에 배치
-        outputFile.absolutePath
-    )
-    Config.setLogLevel(Config.getLogLevel())
-    Config.enableLogCallback { logMessage ->
-        Log.d("FFmpegLog", logMessage.text)
-    }
-    FFmpeg.executeAsync(command) { executionId, returnCode ->
-        if (returnCode == Config.RETURN_CODE_SUCCESS) {
-            callback(true, "Re-encoding successful")
-        } else {
-            callback(false, "Re-encoding failed with return code $returnCode")
-        }
-    }
-}
-
 fun uploadFileToS3PresignedUrl(presignedUrl: String, file: File, callback: (Boolean, String) -> Unit) {
+
     val client = OkHttpClient()
     val mediaType = "application/octet-stream".toMediaTypeOrNull()
     val requestBody = RequestBody.create(mediaType, file)
@@ -236,49 +192,62 @@ fun uploadFileToS3PresignedUrl(presignedUrl: String, file: File, callback: (Bool
     })
 }
 
-fun extractAndUploadThumbnail(videoFile: File, thumbnailFile: File, presignedUrl: String, callback: (Boolean, String) -> Unit) {
-    val extractCommand = arrayOf(
-        "ffmpeg",
-        "-i", videoFile.absolutePath,
-        "-ss", "00:00:01.000",
-        "-vframes", "1",
-        thumbnailFile.absolutePath
-    )
+fun uploadFileToS3ThumbnailPresignedUrl(context: Context,presignedUrl: String, file: File, callback: (Boolean, String) -> Unit) {
+    val videoPath = file.absolutePath
+    val outputImagePath = File(context.cacheDir,"${ file.name }.jpg")
+    getVideoFrameAt1Sec(videoPath, outputImagePath.absolutePath)
+    val client = OkHttpClient()
+    val mediaType = "application/octet-stream".toMediaTypeOrNull()
+    val requestBody = RequestBody.create(mediaType, outputImagePath)
 
-    try {
-        val process = ProcessBuilder(*extractCommand).start()
-        process.waitFor()
-        if (process.exitValue() == 0) {
-            val client = OkHttpClient()
-            val mediaType = "image/jpeg".toMediaTypeOrNull()
-            val requestBody = RequestBody.create(mediaType, thumbnailFile)
+    val request = Request.Builder()
+        .url(presignedUrl)
+        .put(requestBody)
+        .build()
 
-            val request = Request.Builder()
-                .url(presignedUrl)
-                .put(requestBody)
-                .build()
+    client.newCall(request).enqueue(object : okhttp3.Callback {
+        override fun onFailure(call: okhttp3.Call, e: IOException) {
+            callback(false, "Upload failed: ${e.message}")
+        }
 
-            client.newCall(request).enqueue(object : okhttp3.Callback {
-                override fun onFailure(call: okhttp3.Call, e: IOException) {
-                    callback(false, "Upload failed: ${e.message}")
-                }
+        override fun onResponse(call: okhttp3.Call, response: okhttp3.Response) {
+            if (response.isSuccessful) {
+                callback(true, "Upload successful")
+            } else {
+                callback(false, "Upload failed: ${response.message}")
+            }
+        }
+    })
+}
+fun getVideoFrameAt1Sec(videoPath: String, outputImagePath: String) {
+    val retriever = MediaMetadataRetriever()
+    try {
+        retriever.setDataSource(videoPath)
+        val timeUs = 1 * 1000000 // 1초를 마이크로초로 변환
+        val bitmap: Bitmap? = retriever.getFrameAtTime(timeUs.toLong(), MediaMetadataRetriever.OPTION_CLOSEST_SYNC)
 
-                override fun onResponse(call: okhttp3.Call, response: okhttp3.Response) {
-                    if (response.isSuccessful) {
-                        callback(true, "Upload successful")
-                    } else {
-                        callback(false, "Upload failed: ${response.message}")
-                    }
-                }
-            })
-        } else {
-            callback(false, "Thumbnail extraction failed with return code ${process.exitValue()}")
+        if (bitmap != null) {
+            saveBitmapAsJpeg(bitmap, outputImagePath)
         }
-    } catch (e: IOException) {
-        e.printStackTrace()
-        callback(false, "Thumbnail extraction failed with exception: ${e.message}")
-    } catch (e: InterruptedException) {
+    } catch (e: Exception) {
         e.printStackTrace()
-        callback(false, "Thumbnail extraction failed with exception: ${e.message}")
+    } finally {
+        retriever.release()
     }
 }
+
+fun saveBitmapAsJpeg(bitmap: Bitmap, outputPath: String) {
+    var out: FileOutputStream? = null
+    try {
+        out = FileOutputStream(outputPath)
+        bitmap.compress(Bitmap.CompressFormat.JPEG, 90, out)
+    } catch (e: IOException) {
+        e.printStackTrace()
+    } finally {
+        try {
+            out?.close()
+        } catch (e: IOException) {
+            e.printStackTrace()
+        }
+    }
+}
