1 files changed, 83 insertions, 0 deletions
diff --git a/app/src/main/java/foundation/e/advancedprivacy/domain/usecases/AppTrackersUseCase.kt b/app/src/main/java/foundation/e/advancedprivacy/domain/usecases/AppTrackersUseCase.kt
new file mode 100644
index 0000000..92550ab
--- /dev/null
+++ b/app/src/main/java/foundation/e/advancedprivacy/domain/usecases/AppTrackersUseCase.kt
@@ -0,0 +1,83 @@
+/*
+ * Copyright (C) 2023 MURENA SAS
+ *
+ * This program 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.
+ *
+ * This program 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 foundation.e.advancedprivacy.domain.usecases
+
+import foundation.e.advancedprivacy.data.repositories.AppListsRepository
+import foundation.e.advancedprivacy.domain.entities.ApplicationDescription
+import foundation.e.advancedprivacy.trackers.data.StatsDatabase
+import foundation.e.advancedprivacy.trackers.data.TrackersRepository
+import foundation.e.advancedprivacy.trackers.data.WhitelistRepository
+import foundation.e.advancedprivacy.trackers.domain.entities.Tracker
+import foundation.e.advancedprivacy.trackers.domain.usecases.FilterHostnameUseCase
+
+class AppTrackersUseCase(
+    private val whitelistRepository: WhitelistRepository,
+    private val trackersStateUseCase: TrackersStateUseCase,
+    private val appListsRepository: AppListsRepository,
+    private val statsDatabase: StatsDatabase,
+    private val trackersRepository: TrackersRepository,
+    private val filterHostnameUseCase: FilterHostnameUseCase,
+) {
+    suspend fun toggleAppWhitelist(app: ApplicationDescription, isBlocked: Boolean) {
+        appListsRepository.applyForHiddenApps(app) {
+            whitelistRepository.setWhiteListed(it.apId, !isBlocked)
+            val trackerIds = statsDatabase.getTrackerIds(listOf(app.apId))
+            whitelistRepository.setWhitelistedTrackersForApp(it.apId, trackerIds, !isBlocked)
+        }
+        trackersStateUseCase.updateAllTrackersBlockedState()
+    }
+
+    suspend fun clearWhitelist(app: ApplicationDescription) {
+        appListsRepository.applyForHiddenApps(
+            app
+        ) {
+            whitelistRepository.clearWhiteList(it.apId)
+        }
+        trackersStateUseCase.updateAllTrackersBlockedState()
+    }
+
+    suspend fun getCalls(app: ApplicationDescription): Pair<Int, Int> {
+        return appListsRepository.mapReduceForHiddenApps(
+            app = app,
+            map = {
+                statsDatabase.getCallsForApp(app.apId)
+            },
+            reduce = { zip ->
+                zip.unzip().let { (blocked, leaked) ->
+                    blocked.sum() to leaked.sum()
+                }
+            }
+        )
+    }
+
+    suspend fun getTrackersWithBlockedList(app: ApplicationDescription): List<Pair<Tracker, Boolean>> {
+        val realApIds = appListsRepository.getRealApps(app).map { it.apId }
+        val trackers = statsDatabase.getTrackerIds(realApIds)
+            .mapNotNull { trackersRepository.getTracker(it) }
+
+        return enrichWithBlockedState(app, trackers)
+    }
+
+    suspend fun enrichWithBlockedState(app: ApplicationDescription, trackers: List<Tracker>): List<Pair<Tracker, Boolean>> {
+        val realAppUids = appListsRepository.getRealApps(app).map { it.uid }
+        return trackers.map { tracker ->
+            tracker to !realAppUids.any { uid ->
+                filterHostnameUseCase.isWhitelisted(uid, tracker.id)
+            }
+        }.sortedBy { it.first.label.lowercase() }
+    }
+}