74 lines
2.7 KiB
Kotlin
74 lines
2.7 KiB
Kotlin
/*
|
|
* Copyright (C) 2020 Conny Duck
|
|
*
|
|
* This file is part of Pixelcat.
|
|
*
|
|
* Pixelcat is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* Pixelcat is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
package at.connyduck.pixelcat.components.notifications
|
|
|
|
import androidx.paging.ExperimentalPagingApi
|
|
import androidx.paging.LoadType
|
|
import androidx.paging.PagingState
|
|
import androidx.paging.RemoteMediator
|
|
import androidx.room.withTransaction
|
|
import at.connyduck.pixelcat.db.AppDatabase
|
|
import at.connyduck.pixelcat.db.entitity.NotificationEntity
|
|
import at.connyduck.pixelcat.db.entitity.toEntity
|
|
import at.connyduck.pixelcat.network.FediverseApi
|
|
|
|
@ExperimentalPagingApi
|
|
class NotificationsRemoteMediator(
|
|
private val accountId: Long,
|
|
private val api: FediverseApi,
|
|
private val db: AppDatabase
|
|
) : RemoteMediator<Int, NotificationEntity>() {
|
|
|
|
override suspend fun load(
|
|
loadType: LoadType,
|
|
state: PagingState<Int, NotificationEntity>
|
|
): MediatorResult {
|
|
val apiCall = when (loadType) {
|
|
LoadType.REFRESH -> {
|
|
api.notifications(limit = state.config.initialLoadSize, excludes = setOf("poll", "follow_request"))
|
|
}
|
|
LoadType.PREPEND -> {
|
|
return MediatorResult.Success(true)
|
|
}
|
|
LoadType.APPEND -> {
|
|
val maxId = state.pages.findLast { it.data.isNotEmpty() }?.data?.lastOrNull()?.id
|
|
api.notifications(maxId = maxId, limit = state.config.pageSize, excludes = setOf("poll", "follow_request"))
|
|
}
|
|
}
|
|
|
|
return apiCall.fold(
|
|
{ notificationResult ->
|
|
db.withTransaction {
|
|
if (loadType == LoadType.REFRESH) {
|
|
db.notificationsDao().clearAll(accountId)
|
|
}
|
|
db.notificationsDao().insertOrReplace(notificationResult.map { it.toEntity(accountId) })
|
|
}
|
|
MediatorResult.Success(endOfPaginationReached = notificationResult.isEmpty())
|
|
},
|
|
{
|
|
MediatorResult.Error(it)
|
|
}
|
|
)
|
|
}
|
|
|
|
override suspend fun initialize() = InitializeAction.SKIP_INITIAL_REFRESH
|
|
}
|