From 14815678aaa5020e81f0cca9274f289e6ac7e2d5 Mon Sep 17 00:00:00 2001
From: jacquarg <guillaume.jacquart@hoodbrains.com>
Date: Tue, 12 Apr 2022 18:55:02 +0200
Subject: 5248 hide whitelist app trackers and fix counts when privacy
 protection disabled.

---
 .../domain/usecases/TrackersStatisticsUseCase.kt   | 33 +++++++++++-----------
 1 file changed, 16 insertions(+), 17 deletions(-)

(limited to 'app/src/main/java/foundation/e/privacycentralapp/domain')

diff --git a/app/src/main/java/foundation/e/privacycentralapp/domain/usecases/TrackersStatisticsUseCase.kt b/app/src/main/java/foundation/e/privacycentralapp/domain/usecases/TrackersStatisticsUseCase.kt
index 55887ed..6b4e098 100644
--- a/app/src/main/java/foundation/e/privacycentralapp/domain/usecases/TrackersStatisticsUseCase.kt
+++ b/app/src/main/java/foundation/e/privacycentralapp/domain/usecases/TrackersStatisticsUseCase.kt
@@ -28,6 +28,7 @@ import foundation.e.privacymodules.trackers.Tracker
 import kotlinx.coroutines.channels.awaitClose
 import kotlinx.coroutines.flow.Flow
 import kotlinx.coroutines.flow.callbackFlow
+import kotlinx.coroutines.flow.flowOf
 import kotlinx.coroutines.flow.map
 import java.time.ZonedDateTime
 import java.time.format.DateTimeFormatter
@@ -142,7 +143,8 @@ class TrackersStatisticsUseCase(
                 apps.map { app ->
                     AppWithCounts(
                         app,
-                        blockTrackersPrivacyModule.isWhitelisted(app.uid),
+                        !blockTrackersPrivacyModule.isBlockingEnabled() ||
+                            blockTrackersPrivacyModule.isWhitelisted(app.uid),
                         appListsRepository.foldForHiddenSystemApp(app.uid) {
                             trackersCounts.getOrDefault(it, 0)
                         },
@@ -154,23 +156,20 @@ class TrackersStatisticsUseCase(
             }
     }
 
-    fun getNonBLockedTrackersCount(): Int {
-        return if (blockTrackersPrivacyModule.isBlockingEnabled()) {
-            val whiteListedTrackers = mutableSetOf<Tracker>()
-
-            val whiteListedAppUids = blockTrackersPrivacyModule.getWhiteListedApp()
-
-            appListsRepository.getVisibleAndHiddenApps().forEach { app ->
-                if (app.uid in whiteListedAppUids) {
-                    whiteListedTrackers.addAll(getTrackers(app.uid))
-                } else {
-                    whiteListedTrackers.addAll(blockTrackersPrivacyModule.getWhiteList(app.uid))
+    fun getNonBlockedTrackersCount(): Flow<Int> {
+        return if (blockTrackersPrivacyModule.isBlockingEnabled())
+            appListsRepository.getVisibleAndHiddenApps().map { apps ->
+                val whiteListedTrackers = mutableSetOf<Tracker>()
+                val whiteListedAppUids = blockTrackersPrivacyModule.getWhiteListedApp()
+                apps.forEach { app ->
+                    if (app.uid in whiteListedAppUids) {
+                        whiteListedTrackers.addAll(getTrackers(app.uid))
+                    } else {
+                        whiteListedTrackers.addAll(blockTrackersPrivacyModule.getWhiteList(app.uid))
+                    }
                 }
+                whiteListedTrackers.size
             }
-
-            whiteListedTrackers.size
-        } else {
-            trackTrackersPrivacyModule.getTrackersCount()
-        }
+        else flowOf(trackTrackersPrivacyModule.getTrackersCount())
     }
 }
-- 
cgit v1.2.3