fix: add audio asset metadata (WPB-3334)

app/src/main/kotlin/com/wire/android/ui/home/conversations/MessageComposerViewModel.kt

@@ -47,6 +47,7 @@ import com.wire.android.ui.navArgs
 import com.wire.android.util.FileManager
 import com.wire.android.util.ImageUtil
 import com.wire.android.util.dispatchers.DispatcherProvider
+import com.wire.android.util.getAudioLengthInMs
 import com.wire.kalium.logic.configuration.FileSharingStatus
 import com.wire.kalium.logic.data.asset.AttachmentType
 import com.wire.kalium.logic.data.asset.KaliumFileSystem
@@ -305,7 +306,8 @@ class MessageComposerViewModel @Inject constructor(
                                 assetWidth = imgWidth,
                                 assetHeight = imgHeight,
                                 assetDataSize = dataSize,
-                                assetMimeType = mimeType
+                                assetMimeType = mimeType,
+                                audioLengthInMs = 0L
                             )
                         }
 
@@ -320,7 +322,11 @@ class MessageComposerViewModel @Inject constructor(
                                     assetMimeType = mimeType,
                                     assetDataSize = dataSize,
                                     assetHeight = null,
-                                    assetWidth = null
+                                    assetWidth = null,
+                                    audioLengthInMs = getAudioLengthInMs(
+                                        dataPath = dataPath,
+                                        mimeType = mimeType
+                                    )
                                 )
                             } catch (e: OutOfMemoryError) {
                                 appLogger.e("There was an OutOfMemory error while uploading the asset")

app/src/main/kotlin/com/wire/android/ui/home/messagecomposer/recordaudio/RecordAudioViewModel.kt

@@ -26,15 +26,18 @@
 import com.wire.android.media.audiomessage.AudioState
 import com.wire.android.media.audiomessage.RecordAudioMessagePlayer
 import com.wire.android.ui.home.conversations.model.UriAsset
+import com.wire.android.util.AUDIO_MIME_TYPE
 import com.wire.android.util.CurrentScreen
 import com.wire.android.util.CurrentScreenManager
+import com.wire.android.util.getAudioLengthInMs
 import com.wire.android.util.ui.UIText
 import com.wire.kalium.logic.feature.call.usecase.ObserveEstablishedCallsUseCase
 import dagger.hilt.android.lifecycle.HiltViewModel
 import kotlinx.coroutines.flow.MutableSharedFlow
 import kotlinx.coroutines.flow.SharedFlow
 import kotlinx.coroutines.flow.asSharedFlow
 import kotlinx.coroutines.launch
+import okio.Path.Companion.toPath
 import java.io.File
 import javax.inject.Inject
 import kotlin.io.path.deleteIfExists
@@ -143,10 +146,21 @@
 
     fun stopRecording() {
         if (state.buttonState == RecordAudioButtonState.RECORDING) {
+            audioMediaRecorder.stop()
+
             state = state.copy(
-                buttonState = RecordAudioButtonState.READY_TO_SEND
+                buttonState = RecordAudioButtonState.READY_TO_SEND,
+                audioState = AudioState.DEFAULT.copy(
+                    totalTimeInMs = AudioState.TotalTimeInMs.Known(
+                        state.outputFile?.let {
+                            getAudioLengthInMs(
+                                dataPath = it.path.toPath(),
+                                mimeType = AUDIO_MIME_TYPE
+                            ).toInt()
+                        } ?: 0
+                    )
+                )
             )
-            audioMediaRecorder.stop()
         }
         audioMediaRecorder.release()
     }

app/src/main/kotlin/com/wire/android/ui/sharing/ImportMediaAuthenticatedViewModel.kt

@@ -31,6 +31,7 @@
 import com.wire.android.util.FileManager
 import com.wire.android.util.ImageUtil
 import com.wire.android.util.dispatchers.DispatcherProvider
+import com.wire.android.util.getAudioLengthInMs
 import com.wire.android.util.getMetadataFromUri
 import com.wire.android.util.getMimeType
 import com.wire.android.util.isImageFile
@@ -321,6 +322,10 @@
                         assetMimeType = importedAsset.mimeType,
                         assetWidth = if (isImage) (importedAsset as ImportedMediaAsset.Image).width else 0,
                         assetHeight = if (isImage) (importedAsset as ImportedMediaAsset.Image).height else 0,
+                        audioLengthInMs = getAudioLengthInMs(
+                            dataPath = importedAsset.dataPath,
+                            mimeType = importedAsset.mimeType
+                        )
                     ).also {
                         if (it is ScheduleNewAssetMessageResult.Failure) {
                             appLogger.e("Failed to import asset message to conversationId=${conversation.conversationId.toLogString()} ")

app/src/main/kotlin/com/wire/android/util/FileUtil.kt

@@ -30,6 +30,7 @@ import android.content.Context
 import android.content.Intent
 import android.database.Cursor
 import android.graphics.drawable.Drawable
+import android.media.MediaMetadataRetriever
 import android.net.Uri
 import android.os.Build
 import android.os.Environment
@@ -54,6 +55,7 @@ import com.wire.android.util.ImageUtil.ImageSizeClass
 import com.wire.android.util.ImageUtil.ImageSizeClass.Medium
 import com.wire.android.util.dispatchers.DefaultDispatcherProvider
 import com.wire.android.util.dispatchers.DispatcherProvider
+import com.wire.kalium.logic.data.asset.isAudioMimeType
 import com.wire.kalium.logic.util.buildFileName
 import com.wire.kalium.logic.util.splitFileExtensionAndCopyCounter
 import kotlinx.coroutines.withContext
@@ -62,6 +64,7 @@ import java.io.File
 import java.io.FileNotFoundException
 import java.io.InputStream
 import java.util.Locale
+import kotlin.time.Duration.Companion.milliseconds
 
 /**
  * Gets the uri of any drawable or given resource
@@ -414,6 +417,18 @@ fun findFirstUniqueName(dir: File, desiredName: String): String {
     return currentName
 }
 
+fun getAudioLengthInMs(dataPath: Path, mimeType: String): Long =
+    if (isAudioMimeType(mimeType)) {
+        val retriever = MediaMetadataRetriever()
+        retriever.setDataSource(dataPath.toFile().absolutePath)
+        val rawDuration = retriever
+            .extractMetadata(MediaMetadataRetriever.METADATA_KEY_DURATION)
+            ?.toLong() ?: 0L
+        rawDuration.milliseconds.inWholeMilliseconds
+    } else {
+        0L
+    }
+
 private const val ATTACHMENT_FILENAME = "attachment"
 private const val DATA_COPY_BUFFER_SIZE = 2048
 const val SDK_VERSION = 33

app/src/test/kotlin/com/wire/android/ui/home/conversations/MessageComposerViewModelArrangement.kt

@@ -239,6 +239,7 @@ internal class MessageComposerViewModelArrangement {
                 any(),
                 any(),
                 any(),
+                any(),
                 any()
             )
         } returns ScheduleNewAssetMessageResult.Success("some-message-id")

app/src/test/kotlin/com/wire/android/ui/home/conversations/MessageComposerViewModelTest.kt

@@ -192,6 +192,7 @@ class MessageComposerViewModelTest {
                     any(),
                     any(),
                     any(),
+                    any(),
                     any()
                 )
             }
@@ -228,6 +229,7 @@ class MessageComposerViewModelTest {
                     any(),
                     any(),
                     any(),
+                    any(),
                     any()
                 )
             }
@@ -254,6 +256,7 @@ class MessageComposerViewModelTest {
                     any(),
                     any(),
                     any(),
+                    any(),
                     any()
                 )
             }
@@ -293,6 +296,7 @@ class MessageComposerViewModelTest {
                     any(),
                     any(),
                     any(),
+                    any(),
                     any()
                 )
             }
@@ -333,6 +337,7 @@ class MessageComposerViewModelTest {
                     any(),
                     any(),
                     any(),
+                    any(),
                     any()
                 )
             }
@@ -374,6 +379,7 @@ class MessageComposerViewModelTest {
                     any(),
                     any(),
                     any(),
+                    any(),
                     any()
                 )
             }
@@ -418,6 +424,7 @@ class MessageComposerViewModelTest {
                         any(),
                         any(),
                         any(),
+                        any(),
                         any()
                     )
                 }
@@ -455,6 +462,7 @@ class MessageComposerViewModelTest {
                     any(),
                     any(),
                     any(),
+                    any(),
                     any()
                 )
             }
@@ -559,6 +567,7 @@ class MessageComposerViewModelTest {
                     any(),
                     any(),
                     any(),
+                    any(),
                     any()
                 )
             }