You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
50 lines
1.8 KiB
Kotlin
50 lines
1.8 KiB
Kotlin
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
package org.mozilla.fenix.components
|
|
|
|
import android.content.Context
|
|
import androidx.annotation.VisibleForTesting
|
|
import androidx.paging.DataSource
|
|
import kotlinx.coroutines.Dispatchers
|
|
import kotlinx.coroutines.withContext
|
|
import mozilla.components.feature.sitepermissions.SitePermissions
|
|
import mozilla.components.feature.sitepermissions.SitePermissionsStorage
|
|
import org.mozilla.fenix.ext.components
|
|
import org.mozilla.fenix.utils.Mockable
|
|
import kotlin.coroutines.CoroutineContext
|
|
|
|
@Mockable
|
|
class PermissionStorage(
|
|
private val context: Context,
|
|
@VisibleForTesting internal val dispatcher: CoroutineContext = Dispatchers.IO,
|
|
@VisibleForTesting internal val permissionsStorage: SitePermissionsStorage =
|
|
context.components.sitePermissionsStorage
|
|
) {
|
|
|
|
suspend fun add(sitePermissions: SitePermissions) = withContext(dispatcher) {
|
|
permissionsStorage.save(sitePermissions)
|
|
}
|
|
|
|
suspend fun findSitePermissionsBy(origin: String): SitePermissions? = withContext(dispatcher) {
|
|
permissionsStorage.findSitePermissionsBy(origin)
|
|
}
|
|
|
|
suspend fun updateSitePermissions(sitePermissions: SitePermissions) = withContext(dispatcher) {
|
|
permissionsStorage.update(sitePermissions)
|
|
}
|
|
|
|
fun getSitePermissionsPaged(): DataSource.Factory<Int, SitePermissions> {
|
|
return permissionsStorage.getSitePermissionsPaged()
|
|
}
|
|
|
|
suspend fun deleteSitePermissions(sitePermissions: SitePermissions) = withContext(dispatcher) {
|
|
permissionsStorage.remove(sitePermissions)
|
|
}
|
|
|
|
suspend fun deleteAllSitePermissions() = withContext(dispatcher) {
|
|
permissionsStorage.removeAll()
|
|
}
|
|
}
|