Glimpse: MediaRepository: Introduce medias(List<Uri>)

And use instead of the hacky toFlowOfRequestStatusList()

Change-Id: Ibf4d4641c430c1b5bc1ca3b89e9b3e04218b7474

Author: luca020400

app/src/main/java/org/lineageos/glimpse/datasources/LocalDataSource.kt

@@ -24,6 +24,7 @@ import org.lineageos.glimpse.models.Thumbnail
 import org.lineageos.glimpse.query.Query
 import org.lineageos.glimpse.query.and
 import org.lineageos.glimpse.query.eq
+import org.lineageos.glimpse.query.`in`
 import org.lineageos.glimpse.query.or
 import org.lineageos.glimpse.query.query
 import org.lineageos.glimpse.utils.MimeUtils
@@ -265,6 +266,26 @@ class LocalDataSource(
         } ?: RequestStatus.Error(MediaError.NOT_FOUND)
     }
 
+    override fun medias(mediaUris: List<Uri>) = contentResolver.queryFlow(
+        filesUri,
+        mediaProjection,
+        bundleOf(
+            ContentResolver.QUERY_ARG_SQL_SELECTION to query {
+                (MediaStore.Files.FileColumns._ID `in` List(mediaUris.size) {
+                    Query.ARG
+                }) and isImageOrVideo
+            },
+            ContentResolver.QUERY_ARG_SQL_SELECTION_ARGS to mediaUris.map {
+                it.lastPathSegment!!
+            }.toTypedArray(),
+            ContentResolver.QUERY_ARG_SORT_COLUMNS to arrayOf(
+                "${MediaStore.Files.FileColumns.DATE_MODIFIED} DESC",
+            ),
+        ),
+    ).mapEachRow(mapMedia).mapLatest {
+        RequestStatus.Success<_, MediaError>(it)
+    }
+
     companion object {
         private const val ALBUMS_PATH = "albums"
 

app/src/main/java/org/lineageos/glimpse/datasources/MediaDataSource.kt

@@ -1,5 +1,5 @@
 /*
- * SPDX-FileCopyrightText: 2024 The LineageOS Project
+ * SPDX-FileCopyrightText: 2024-2025 The LineageOS Project
  * SPDX-License-Identifier: Apache-2.0
  */
 
@@ -75,4 +75,9 @@ interface MediaDataSource {
      * Get the media information of the given media.
      */
     fun media(mediaUri: Uri): Flow<MediaRequestStatus<Media>>
+
+    /**
+     * Get the media information of the given medias.
+     */
+    fun medias(mediaUris: List<Uri>): Flow<MediaRequestStatus<List<Media>>>
 }

app/src/main/java/org/lineageos/glimpse/repository/MediaRepository.kt

@@ -1,5 +1,5 @@
 /*
- * SPDX-FileCopyrightText: 2023-2024 The LineageOS Project
+ * SPDX-FileCopyrightText: 2023-2025 The LineageOS Project
  * SPDX-License-Identifier: Apache-2.0
  */
 
@@ -79,4 +79,9 @@ class MediaRepository(
      * @see MediaDataSource.media
      */
     fun media(mediaUri: Uri) = localDataSource.media(mediaUri)
+
+    /**
+     * @see MediaDataSource.medias
+     */
+    fun medias(mediaUris: List<Uri>) = localDataSource.medias(mediaUris)
 }

app/src/main/java/org/lineageos/glimpse/viewmodels/LocalPlayerViewModel.kt

@@ -15,7 +15,6 @@ import androidx.media3.common.MediaItem
 import androidx.media3.exoplayer.ExoPlayer
 import kotlinx.coroutines.Dispatchers
 import kotlinx.coroutines.ExperimentalCoroutinesApi
-import kotlinx.coroutines.flow.Flow
 import kotlinx.coroutines.flow.MutableStateFlow
 import kotlinx.coroutines.flow.SharingStarted
 import kotlinx.coroutines.flow.asStateFlow
@@ -114,46 +113,18 @@ class LocalPlayerViewModel(
             initialValue = RequestStatus.Loading(),
         )
 
-    private fun <T, E> List<Flow<RequestStatus<T, E>>>.toFlowOfRequestStatusList() = flow {
-        if (isEmpty()) {
-            emit(RequestStatus.Success(emptyList()))
-            return@flow
-        }
-
-        emit(RequestStatus.Loading())
-
-        combine(this@toFlowOfRequestStatusList) { statusArray ->
-            // Check if any item is still loading
-            if (statusArray.any { it is RequestStatus.Loading }) {
-                return@combine RequestStatus.Loading<List<T>, E>()
-            }
-
-            // Collect all successful data, ignoring errors
-            statusArray
-                .filterIsInstance<RequestStatus.Success<T, E>>()
-                .map { it.data }
-                .let {
-                    RequestStatus.Success(it)
-                }
-        }.collect { combinedStatus ->
-            emit(combinedStatus)
-        }
-    }
-
     /**
      * Collect secure media via Uri to update its list after deletion/restore.
      * Needed because we can't use the album to observe for changes.
-     *
-     * NOTE: This _will_ be slow for large amounts of media.
      */
     @OptIn(ExperimentalCoroutinesApi::class)
     val secureMedias = parsedIntent
         .flatMapLatest {
             when (it) {
                 is IntentsViewModel.ParsedIntent.SecureReviewIntent -> {
-                    it.medias.map { media ->
-                        mediaRepository.media(media.uri)
-                    }.toFlowOfRequestStatusList()
+                    mediaRepository.medias(it.medias.map { media ->
+                        media.uri
+                    })
                 }
 
                 else -> flowOf(RequestStatus.Loading())