Update album fetch logic

app/src/main/kotlin/it/vfsfitvnm/vimusic/Database.kt

@@ -138,7 +138,7 @@ interface Database {
     fun artist(id: String): Flow<Artist?>
 
     @Query("SELECT * FROM Album WHERE id = :id")
-    fun album(id: String): Flow<Album?>
+    fun album(id: String): Album?
 
     @Query("UPDATE Song SET totalPlayTimeMs = totalPlayTimeMs + :addition WHERE id = :id")
     fun incrementTotalPlayTimeMs(id: String, addition: Long)

app/src/main/kotlin/it/vfsfitvnm/vimusic/service/BitmapProvider.kt

@@ -72,7 +72,6 @@ class BitmapProvider(
                     onSuccess = { _, result ->
                         lastBitmap = (result.drawable as BitmapDrawable).bitmap
                         onDone(bitmap)
-                        println("invoking listener")
                         listener?.invoke(lastBitmap)
                     }
                 )

app/src/main/kotlin/it/vfsfitvnm/vimusic/ui/screens/AlbumScreen.kt

@@ -27,6 +27,7 @@ import androidx.compose.foundation.text.BasicText
 import androidx.compose.runtime.Composable
 import androidx.compose.runtime.collectAsState
 import androidx.compose.runtime.getValue
+import androidx.compose.runtime.produceState
 import androidx.compose.runtime.remember
 import androidx.compose.ui.Alignment
 import androidx.compose.ui.Modifier
@@ -78,23 +79,22 @@ import it.vfsfitvnm.youtubemusic.YouTube
 import java.text.DateFormat
 import java.util.Date
 import kotlinx.coroutines.Dispatchers
-import kotlinx.coroutines.flow.distinctUntilChanged
-import kotlinx.coroutines.flow.map
 import kotlinx.coroutines.runBlocking
+import kotlinx.coroutines.withContext
 
 @ExperimentalAnimationApi
 @Composable
 fun AlbumScreen(browseId: String) {
     val lazyListState = rememberLazyListState()
 
-    val albumResult by remember(browseId) {
-        Database.album(browseId).map { album ->
-            album
-                ?.takeIf { album.timestamp != null }
+    val albumResult by produceState<Result<Album>?>(initialValue = null, browseId) {
+        value = withContext(Dispatchers.IO) {
+            Database.album(browseId)
+                ?.takeIf { it.timestamp != null }
                 ?.let(Result.Companion::success)
                 ?: fetchAlbum(browseId)
-        }.distinctUntilChanged()
-    }.collectAsState(initial = null, context = Dispatchers.IO)
+        }
+    }
 
     val songs by remember(browseId) {
         Database.albumSongs(browseId)
@@ -404,9 +404,9 @@ private fun LoadingOrError(
 }
 
 private suspend fun fetchAlbum(browseId: String): Result<Album>? {
-    return YouTube.playlistOrAlbum(browseId)
+    return YouTube.album(browseId)
         ?.map { youtubeAlbum ->
-            Album(
+            val album = Album(
                 id = browseId,
                 title = youtubeAlbum.title,
                 thumbnailUrl = youtubeAlbum.thumbnail?.url,
@@ -414,19 +414,23 @@ private suspend fun fetchAlbum(browseId: String): Result<Album>? {
                 authorsText = youtubeAlbum.authors?.joinToString("") { it.name },
                 shareUrl = youtubeAlbum.url,
                 timestamp = System.currentTimeMillis()
-            ).also(Database::upsert).also {
-                youtubeAlbum.withAudioSources()?.getOrNull()?.items?.forEachIndexed { position, albumItem ->
-                    albumItem.toMediaItem(browseId, youtubeAlbum)?.let { mediaItem ->
-                        Database.insert(mediaItem)
-                        Database.upsert(
-                            SongAlbumMap(
-                                songId = mediaItem.mediaId,
-                                albumId = browseId,
-                                position = position
-                            )
+            )
+
+            Database.upsert(album)
+
+            youtubeAlbum.items?.forEachIndexed { position, albumItem ->
+                albumItem.toMediaItem(browseId, youtubeAlbum)?.let { mediaItem ->
+                    Database.insert(mediaItem)
+                    Database.upsert(
+                        SongAlbumMap(
+                            songId = mediaItem.mediaId,
+                            albumId = browseId,
+                            position = position
                         )
-                    }
+                    )
                 }
             }
+
+            album
         }
 }

app/src/main/kotlin/it/vfsfitvnm/vimusic/ui/screens/PlaylistScreen.kt

@@ -102,7 +102,7 @@ fun PlaylistScreen(browseId: String) {
 
             val onLoad = relaunchableEffect(Unit) {
                 playlist = withContext(Dispatchers.IO) {
-                    YouTube.playlistOrAlbum(browseId)?.map {
+                    YouTube.playlist(browseId)?.map {
                         it.next()
                     }?.map { playlist ->
                         playlist.copy(items = playlist.items?.filter { it.info.endpoint != null })

youtube-music/src/main/kotlin/it/vfsfitvnm/youtubemusic/YouTube.kt

@@ -882,17 +882,23 @@ object YouTube {
                 }.recoverIfCancelled()?.getOrNull()
             } ?: this
         }
+    }
 
-        suspend fun withAudioSources(): Result<PlaylistOrAlbum>? {
-            return url?.let { Url(it).parameters["list"] }?.let { playlistId ->
-                playlistOrAlbum("VL$playlistId")?.map { playlist ->
-                    copy(items = playlist.items)
+    suspend fun album(browseId: String): Result<PlaylistOrAlbum>? {
+        return playlistOrAlbum(browseId)?.map { album ->
+           album.url?.let { Url(it).parameters["list"] }?.let { playlistId ->
+                playlistOrAlbum("VL$playlistId")?.getOrNull()?.let { playlist ->
+                    album.copy(items = playlist.items)
                 }
-            }
+            } ?: album
         }
     }
 
-    suspend fun playlistOrAlbum(browseId: String): Result<PlaylistOrAlbum>? {
+    suspend fun playlist(browseId: String): Result<PlaylistOrAlbum>? {
+        return playlistOrAlbum(browseId)
+    }
+
+    private suspend fun playlistOrAlbum(browseId: String): Result<PlaylistOrAlbum>? {
         return browse(browseId)?.map { body ->
             PlaylistOrAlbum(
                 title = body