2019-04-28 22:43:54 +02:00
|
|
|
package org.schabi.newpipe.local.subscription
|
|
|
|
|
|
|
|
import android.content.Context
|
|
|
|
import io.reactivex.Completable
|
|
|
|
import io.reactivex.android.schedulers.AndroidSchedulers
|
|
|
|
import io.reactivex.schedulers.Schedulers
|
|
|
|
import org.schabi.newpipe.NewPipeDatabase
|
|
|
|
import org.schabi.newpipe.database.subscription.SubscriptionDAO
|
|
|
|
import org.schabi.newpipe.database.subscription.SubscriptionEntity
|
2019-12-16 08:36:04 +01:00
|
|
|
import org.schabi.newpipe.extractor.ListInfo
|
2019-04-28 22:43:54 +02:00
|
|
|
import org.schabi.newpipe.extractor.channel.ChannelInfo
|
2019-12-16 08:36:04 +01:00
|
|
|
import org.schabi.newpipe.extractor.feed.FeedInfo
|
|
|
|
import org.schabi.newpipe.extractor.stream.StreamInfoItem
|
2019-04-28 22:43:54 +02:00
|
|
|
import org.schabi.newpipe.local.feed.FeedDatabaseManager
|
|
|
|
|
|
|
|
class SubscriptionManager(context: Context) {
|
|
|
|
private val database = NewPipeDatabase.getInstance(context)
|
|
|
|
private val subscriptionTable = database.subscriptionDAO()
|
|
|
|
private val feedDatabaseManager = FeedDatabaseManager(context)
|
|
|
|
|
|
|
|
fun subscriptionTable(): SubscriptionDAO = subscriptionTable
|
|
|
|
fun subscriptions() = subscriptionTable.all
|
|
|
|
|
2020-04-05 21:25:44 +02:00
|
|
|
fun filterByName(filter: String) = subscriptionTable.filterByName(filter)
|
|
|
|
|
2019-04-28 22:43:54 +02:00
|
|
|
fun upsertAll(infoList: List<ChannelInfo>): List<SubscriptionEntity> {
|
|
|
|
val listEntities = subscriptionTable.upsertAll(
|
|
|
|
infoList.map { SubscriptionEntity.from(it) })
|
|
|
|
|
|
|
|
database.runInTransaction {
|
|
|
|
infoList.forEachIndexed { index, info ->
|
|
|
|
feedDatabaseManager.upsertAll(listEntities[index].uid, info.relatedItems)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return listEntities
|
|
|
|
}
|
|
|
|
|
|
|
|
fun updateChannelInfo(info: ChannelInfo): Completable = subscriptionTable.getSubscription(info.serviceId, info.url)
|
|
|
|
.flatMapCompletable {
|
|
|
|
Completable.fromRunnable {
|
|
|
|
it.setData(info.name, info.avatarUrl, info.description, info.subscriberCount)
|
|
|
|
subscriptionTable.update(it)
|
|
|
|
feedDatabaseManager.upsertAll(it.uid, info.relatedItems)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-12-16 08:36:04 +01:00
|
|
|
fun updateFromInfo(subscriptionId: Long, info: ListInfo<StreamInfoItem>) {
|
2019-04-28 22:43:54 +02:00
|
|
|
val subscriptionEntity = subscriptionTable.getSubscription(subscriptionId)
|
2019-12-16 08:36:04 +01:00
|
|
|
|
|
|
|
if (info is FeedInfo) {
|
|
|
|
subscriptionEntity.name = info.name
|
|
|
|
} else if (info is ChannelInfo) {
|
|
|
|
subscriptionEntity.setData(info.name, info.avatarUrl, info.description, info.subscriberCount)
|
|
|
|
}
|
2019-04-28 22:43:54 +02:00
|
|
|
|
|
|
|
subscriptionTable.update(subscriptionEntity)
|
|
|
|
}
|
|
|
|
|
|
|
|
fun deleteSubscription(serviceId: Int, url: String): Completable {
|
|
|
|
return Completable.fromCallable { subscriptionTable.deleteSubscription(serviceId, url) }
|
|
|
|
.subscribeOn(Schedulers.io())
|
|
|
|
.observeOn(AndroidSchedulers.mainThread())
|
|
|
|
}
|
|
|
|
|
|
|
|
fun insertSubscription(subscriptionEntity: SubscriptionEntity, info: ChannelInfo) {
|
|
|
|
database.runInTransaction {
|
|
|
|
val subscriptionId = subscriptionTable.insert(subscriptionEntity)
|
|
|
|
feedDatabaseManager.upsertAll(subscriptionId, info.relatedItems)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
fun deleteSubscription(subscriptionEntity: SubscriptionEntity) {
|
|
|
|
subscriptionTable.delete(subscriptionEntity)
|
|
|
|
}
|
|
|
|
}
|