2021-07-12 10:14:26 +02:00
|
|
|
package audio.funkwhale.ffa.repositories
|
2019-08-19 16:50:33 +02:00
|
|
|
|
|
|
|
import android.content.Context
|
2021-08-22 09:48:33 +02:00
|
|
|
import audio.funkwhale.ffa.model.FFAResponse
|
|
|
|
import audio.funkwhale.ffa.model.Track
|
|
|
|
import audio.funkwhale.ffa.model.TracksCache
|
|
|
|
import audio.funkwhale.ffa.model.TracksResponse
|
|
|
|
import audio.funkwhale.ffa.utils.OAuth
|
|
|
|
import audio.funkwhale.ffa.utils.getMetadata
|
2022-06-16 15:08:29 +02:00
|
|
|
import audio.funkwhale.ffa.utils.gsonDeserializerOf
|
2021-08-22 09:48:33 +02:00
|
|
|
import audio.funkwhale.ffa.utils.mustNormalizeUrl
|
2020-06-13 17:44:50 +02:00
|
|
|
import com.google.android.exoplayer2.offline.Download
|
2021-08-09 06:50:46 +02:00
|
|
|
import com.google.android.exoplayer2.offline.DownloadManager
|
|
|
|
import com.google.android.exoplayer2.upstream.cache.Cache
|
2019-08-19 16:50:33 +02:00
|
|
|
import com.google.gson.reflect.TypeToken
|
2019-10-31 16:17:37 +01:00
|
|
|
import kotlinx.coroutines.flow.map
|
|
|
|
import kotlinx.coroutines.flow.toList
|
2019-08-19 16:50:33 +02:00
|
|
|
import kotlinx.coroutines.runBlocking
|
2021-08-09 06:50:46 +02:00
|
|
|
import org.koin.core.qualifier.named
|
|
|
|
import org.koin.java.KoinJavaComponent.inject
|
2019-08-19 16:50:33 +02:00
|
|
|
|
2021-07-23 14:10:13 +02:00
|
|
|
class TracksRepository(override val context: Context?, albumId: Int) :
|
|
|
|
Repository<Track, TracksCache>() {
|
|
|
|
|
2021-08-09 06:50:46 +02:00
|
|
|
private val exoCache: Cache by inject(Cache::class.java, named("exoCache"))
|
|
|
|
private val oAuth: OAuth by inject(OAuth::class.java)
|
|
|
|
private val exoDownloadManager: DownloadManager by inject(DownloadManager::class.java)
|
2021-07-30 10:57:49 +02:00
|
|
|
|
2019-08-19 16:50:33 +02:00
|
|
|
override val cacheId = "tracks-album-$albumId"
|
2021-07-23 14:10:13 +02:00
|
|
|
|
2021-08-22 09:48:33 +02:00
|
|
|
override val upstream = HttpUpstream<Track, FFAResponse<Track>>(
|
2021-07-23 14:10:13 +02:00
|
|
|
context,
|
|
|
|
HttpUpstream.Behavior.AtOnce,
|
|
|
|
"/api/v1/tracks/?playable=true&album=$albumId&ordering=disc_number,position",
|
2021-07-30 10:57:49 +02:00
|
|
|
object : TypeToken<TracksResponse>() {}.type,
|
|
|
|
oAuth
|
2021-07-23 14:10:13 +02:00
|
|
|
)
|
2019-08-19 16:50:33 +02:00
|
|
|
|
|
|
|
override fun cache(data: List<Track>) = TracksCache(data)
|
2022-06-12 14:48:32 +02:00
|
|
|
override fun uncache(json: String) =
|
|
|
|
gsonDeserializerOf(TracksCache::class.java).deserialize(json)
|
2019-08-19 16:50:33 +02:00
|
|
|
|
2020-06-13 19:50:02 +02:00
|
|
|
companion object {
|
2021-08-09 06:50:46 +02:00
|
|
|
fun getDownloadedIds(exoDownloadManager: DownloadManager): List<Int>? {
|
2020-06-22 21:48:31 +02:00
|
|
|
val ids: MutableList<Int> = mutableListOf()
|
2022-06-21 10:03:46 +02:00
|
|
|
exoDownloadManager.downloadIndex.getDownloads()
|
|
|
|
.use { cursor ->
|
|
|
|
while (cursor.moveToNext()) {
|
|
|
|
val download = cursor.download
|
|
|
|
download.getMetadata()?.let {
|
|
|
|
if (download.state == Download.STATE_COMPLETED) {
|
|
|
|
ids.add(it.id)
|
|
|
|
}
|
|
|
|
}
|
2020-06-13 19:50:02 +02:00
|
|
|
}
|
|
|
|
}
|
2020-06-22 21:48:31 +02:00
|
|
|
return ids
|
2020-06-13 19:50:02 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-08-19 16:50:33 +02:00
|
|
|
override fun onDataFetched(data: List<Track>): List<Track> = runBlocking {
|
2020-07-11 16:56:09 +02:00
|
|
|
val favorites = FavoritedRepository(context).fetch(Origin.Cache.origin)
|
2019-10-31 16:17:37 +01:00
|
|
|
.map { it.data }
|
|
|
|
.toList()
|
|
|
|
.flatten()
|
2019-08-19 16:50:33 +02:00
|
|
|
|
2021-08-09 06:50:46 +02:00
|
|
|
val downloaded = getDownloadedIds(exoDownloadManager) ?: listOf()
|
2020-06-13 17:44:50 +02:00
|
|
|
|
2019-08-19 16:50:33 +02:00
|
|
|
data.map { track ->
|
2019-10-23 20:21:18 +02:00
|
|
|
track.favorite = favorites.contains(track.id)
|
2020-06-13 17:44:50 +02:00
|
|
|
track.downloaded = downloaded.contains(track.id)
|
2020-06-20 15:42:10 +02:00
|
|
|
|
|
|
|
track.bestUpload()?.let { upload ->
|
|
|
|
val url = mustNormalizeUrl(upload.listen_url)
|
|
|
|
|
2021-08-09 06:50:46 +02:00
|
|
|
track.cached = exoCache.isCached(url, 0, upload.duration * 1000L)
|
2020-06-20 15:42:10 +02:00
|
|
|
}
|
|
|
|
|
2019-08-19 16:50:33 +02:00
|
|
|
track
|
2020-09-01 18:25:10 +02:00
|
|
|
}.sortedWith(compareBy({ it.disc_number }, { it.position }))
|
2019-08-19 16:50:33 +02:00
|
|
|
}
|
2021-07-02 13:55:49 +02:00
|
|
|
}
|