2019-04-28 20:43:54 +00:00
|
|
|
package org.schabi.newpipe.local.feed
|
|
|
|
|
|
|
|
import android.content.Context
|
2021-11-08 04:11:10 +00:00
|
|
|
import androidx.core.content.edit
|
2020-03-05 22:47:00 +00:00
|
|
|
import androidx.lifecycle.LiveData
|
2019-04-28 20:43:54 +00:00
|
|
|
import androidx.lifecycle.MutableLiveData
|
|
|
|
import androidx.lifecycle.ViewModel
|
|
|
|
import androidx.lifecycle.ViewModelProvider
|
2021-11-08 04:11:10 +00:00
|
|
|
import androidx.preference.PreferenceManager
|
2020-10-31 20:55:45 +00:00
|
|
|
import io.reactivex.rxjava3.android.schedulers.AndroidSchedulers
|
|
|
|
import io.reactivex.rxjava3.core.Flowable
|
|
|
|
import io.reactivex.rxjava3.functions.Function4
|
2020-04-05 14:11:03 +00:00
|
|
|
import io.reactivex.rxjava3.processors.BehaviorProcessor
|
2020-10-31 20:55:45 +00:00
|
|
|
import io.reactivex.rxjava3.schedulers.Schedulers
|
2021-11-10 21:16:17 +00:00
|
|
|
import org.schabi.newpipe.R
|
2020-01-28 05:59:49 +00:00
|
|
|
import org.schabi.newpipe.database.feed.model.FeedGroupEntity
|
2020-04-05 14:11:03 +00:00
|
|
|
import org.schabi.newpipe.database.stream.StreamWithState
|
|
|
|
import org.schabi.newpipe.local.feed.item.StreamItem
|
2019-04-28 20:43:54 +00:00
|
|
|
import org.schabi.newpipe.local.feed.service.FeedEventManager
|
2020-11-04 17:07:40 +00:00
|
|
|
import org.schabi.newpipe.local.feed.service.FeedEventManager.Event.ErrorResultEvent
|
|
|
|
import org.schabi.newpipe.local.feed.service.FeedEventManager.Event.IdleEvent
|
|
|
|
import org.schabi.newpipe.local.feed.service.FeedEventManager.Event.ProgressEvent
|
|
|
|
import org.schabi.newpipe.local.feed.service.FeedEventManager.Event.SuccessResultEvent
|
2019-04-28 20:43:54 +00:00
|
|
|
import org.schabi.newpipe.util.DEFAULT_THROTTLE_TIMEOUT
|
2020-10-31 20:55:45 +00:00
|
|
|
import java.time.OffsetDateTime
|
|
|
|
import java.util.concurrent.TimeUnit
|
2019-04-28 20:43:54 +00:00
|
|
|
|
2020-04-05 14:11:03 +00:00
|
|
|
class FeedViewModel(
|
2021-11-11 18:46:15 +00:00
|
|
|
private val applicationContext: Context,
|
2020-04-05 14:11:03 +00:00
|
|
|
groupId: Long = FeedGroupEntity.GROUP_ALL_ID,
|
|
|
|
initialShowPlayedItems: Boolean = true
|
|
|
|
) : ViewModel() {
|
2019-04-28 20:43:54 +00:00
|
|
|
private var feedDatabaseManager: FeedDatabaseManager = FeedDatabaseManager(applicationContext)
|
|
|
|
|
2020-04-05 14:11:03 +00:00
|
|
|
private val toggleShowPlayedItems = BehaviorProcessor.create<Boolean>()
|
|
|
|
private val streamItems = toggleShowPlayedItems
|
|
|
|
.startWithItem(initialShowPlayedItems)
|
|
|
|
.distinctUntilChanged()
|
|
|
|
.switchMap { showPlayedItems ->
|
|
|
|
feedDatabaseManager.getStreams(groupId, showPlayedItems)
|
|
|
|
}
|
|
|
|
|
2020-03-05 22:47:00 +00:00
|
|
|
private val mutableStateLiveData = MutableLiveData<FeedState>()
|
|
|
|
val stateLiveData: LiveData<FeedState> = mutableStateLiveData
|
2019-04-28 20:43:54 +00:00
|
|
|
|
|
|
|
private var combineDisposable = Flowable
|
2020-10-31 20:55:45 +00:00
|
|
|
.combineLatest(
|
|
|
|
FeedEventManager.events(),
|
2020-04-05 14:11:03 +00:00
|
|
|
streamItems,
|
2020-10-31 20:55:45 +00:00
|
|
|
feedDatabaseManager.notLoadedCount(groupId),
|
|
|
|
feedDatabaseManager.oldestSubscriptionUpdate(groupId),
|
2020-04-05 14:11:03 +00:00
|
|
|
|
|
|
|
Function4 { t1: FeedEventManager.Event, t2: List<StreamWithState>,
|
|
|
|
t3: Long, t4: List<OffsetDateTime> ->
|
2020-11-04 17:07:40 +00:00
|
|
|
return@Function4 CombineResultHolder(t1, t2, t3, t4.firstOrNull())
|
|
|
|
}
|
2020-10-31 20:55:45 +00:00
|
|
|
)
|
|
|
|
.throttleLatest(DEFAULT_THROTTLE_TIMEOUT, TimeUnit.MILLISECONDS)
|
|
|
|
.subscribeOn(Schedulers.io())
|
|
|
|
.observeOn(AndroidSchedulers.mainThread())
|
|
|
|
.subscribe { (event, listFromDB, notLoadedCount, oldestUpdate) ->
|
|
|
|
mutableStateLiveData.postValue(
|
|
|
|
when (event) {
|
2020-04-05 14:11:03 +00:00
|
|
|
is IdleEvent -> FeedState.LoadedState(listFromDB.map { e -> StreamItem(e) }, oldestUpdate, notLoadedCount)
|
2019-12-16 07:36:04 +00:00
|
|
|
is ProgressEvent -> FeedState.ProgressState(event.currentProgress, event.maxProgress, event.progressMessage)
|
2020-04-05 14:11:03 +00:00
|
|
|
is SuccessResultEvent -> FeedState.LoadedState(listFromDB.map { e -> StreamItem(e) }, oldestUpdate, notLoadedCount, event.itemsErrors)
|
2019-12-16 07:36:04 +00:00
|
|
|
is ErrorResultEvent -> FeedState.ErrorState(event.error)
|
2019-04-28 20:43:54 +00:00
|
|
|
}
|
2020-10-31 20:55:45 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
if (event is ErrorResultEvent || event is SuccessResultEvent) {
|
|
|
|
FeedEventManager.reset()
|
2019-04-28 20:43:54 +00:00
|
|
|
}
|
2020-10-31 20:55:45 +00:00
|
|
|
}
|
2019-04-28 20:43:54 +00:00
|
|
|
|
|
|
|
override fun onCleared() {
|
|
|
|
super.onCleared()
|
|
|
|
combineDisposable.dispose()
|
|
|
|
}
|
2019-12-16 07:36:04 +00:00
|
|
|
|
2020-04-05 14:11:03 +00:00
|
|
|
private data class CombineResultHolder(val t1: FeedEventManager.Event, val t2: List<StreamWithState>, val t3: Long, val t4: OffsetDateTime?)
|
|
|
|
|
|
|
|
fun togglePlayedItems(showPlayedItems: Boolean) {
|
|
|
|
toggleShowPlayedItems.onNext(showPlayedItems)
|
|
|
|
}
|
|
|
|
|
2021-11-11 18:46:15 +00:00
|
|
|
fun saveShowPlayedItemsToPreferences(showPlayedItems: Boolean) =
|
|
|
|
PreferenceManager.getDefaultSharedPreferences(applicationContext).edit {
|
|
|
|
this.putBoolean(applicationContext.getString(R.string.feed_show_played_items_key), showPlayedItems)
|
|
|
|
this.apply()
|
|
|
|
}
|
2021-11-08 04:11:10 +00:00
|
|
|
|
2021-11-11 18:46:15 +00:00
|
|
|
fun getShowPlayedItemsFromPreferences() = getShowPlayedItemsFromPreferences(applicationContext)
|
|
|
|
|
|
|
|
companion object {
|
|
|
|
private fun getShowPlayedItemsFromPreferences(context: Context) =
|
|
|
|
PreferenceManager.getDefaultSharedPreferences(context)
|
|
|
|
.getBoolean(context.getString(R.string.feed_show_played_items_key), true)
|
|
|
|
}
|
2021-11-08 04:11:10 +00:00
|
|
|
|
2020-04-05 14:11:03 +00:00
|
|
|
class Factory(
|
|
|
|
private val context: Context,
|
2021-11-11 18:46:15 +00:00
|
|
|
private val groupId: Long = FeedGroupEntity.GROUP_ALL_ID
|
2020-04-05 14:11:03 +00:00
|
|
|
) : ViewModelProvider.Factory {
|
|
|
|
@Suppress("UNCHECKED_CAST")
|
|
|
|
override fun <T : ViewModel?> create(modelClass: Class<T>): T {
|
2021-11-11 18:46:15 +00:00
|
|
|
return FeedViewModel(
|
|
|
|
context.applicationContext,
|
|
|
|
groupId,
|
|
|
|
// Read initial value from preferences
|
|
|
|
getShowPlayedItemsFromPreferences(context.applicationContext)
|
|
|
|
) as T
|
2020-04-05 14:11:03 +00:00
|
|
|
}
|
|
|
|
}
|
2020-05-01 18:13:01 +00:00
|
|
|
}
|