2 files changed, 73 insertions, 0 deletions
diff --git a/app/src/main/java/foundation/e/privacycentralapp/domain/entities/AppWithCounts.kt b/app/src/main/java/foundation/e/privacycentralapp/domain/entities/AppWithCounts.kt
new file mode 100644
index 0000000..682dfc8
--- /dev/null
+++ b/app/src/main/java/foundation/e/privacycentralapp/domain/entities/AppWithCounts.kt
@@ -0,0 +1,50 @@
+/*
+ * Copyright (C) 2022 E FOUNDATION
+ *
+ * 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.privacycentralapp.domain.entities
+
+import android.graphics.drawable.Drawable
+import foundation.e.privacymodules.permissions.data.ApplicationDescription
+
+data class AppWithCounts(
+    val packageName: String,
+    val uid: Int,
+    var label: CharSequence?,
+    var icon: Drawable?,
+    val isWhitelisted: Boolean = false,
+    val trackersCount: Int = 0,
+    val whiteListedTrackersCount: Int = 0
+) {
+    constructor(
+        app: ApplicationDescription,
+        isWhitelisted: Boolean,
+        trackersCount: Int,
+        whiteListedTrackersCount: Int
+    ) :
+        this(
+            packageName = app.packageName,
+            uid = app.uid,
+            label = app.label,
+            icon = app.icon,
+            isWhitelisted = isWhitelisted,
+            trackersCount = trackersCount,
+            whiteListedTrackersCount = whiteListedTrackersCount
+        )
+
+    val blockedTrackersCount get() = if (isWhitelisted) 0
+    else trackersCount - whiteListedTrackersCount
+}
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 9a8b12a..ad8f565 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
@@ -20,18 +20,22 @@ package foundation.e.privacycentralapp.domain.usecases
 import android.content.res.Resources
 import foundation.e.privacycentralapp.R
 import foundation.e.privacycentralapp.data.repositories.AppListsRepository
+import foundation.e.privacycentralapp.domain.entities.AppWithCounts
 import foundation.e.privacycentralapp.domain.entities.TrackersPeriodicStatistics
+import foundation.e.privacymodules.trackers.IBlockTrackersPrivacyModule
 import foundation.e.privacymodules.trackers.ITrackTrackersPrivacyModule
 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.map
 import java.time.ZonedDateTime
 import java.time.format.DateTimeFormatter
 import java.time.temporal.ChronoUnit
 
 class TrackersStatisticsUseCase(
     private val trackTrackersPrivacyModule: ITrackTrackersPrivacyModule,
+    private val blockTrackersPrivacyModule: IBlockTrackersPrivacyModule,
     private val appListsRepository: AppListsRepository,
     private val resources: Resources
 ) {
@@ -124,4 +128,23 @@ class TrackersStatisticsUseCase(
 
         return trackers.sortedBy { it.label.lowercase() }
     }
+
+    fun getAppsWithCounts(): Flow<List<AppWithCounts>> {
+        val trackersCounts = trackTrackersPrivacyModule.getTrackersCountByApp()
+        return appListsRepository.getVisibleApps()
+            .map { apps ->
+                apps.map { app ->
+                    AppWithCounts(
+                        app,
+                        blockTrackersPrivacyModule.isWhitelisted(app.uid),
+                        appListsRepository.foldForHiddenSystemApp(app.uid) {
+                            trackersCounts.getOrDefault(it, 0)
+                        },
+                        appListsRepository.foldForHiddenSystemApp(app.uid) {
+                            blockTrackersPrivacyModule.getWhiteList(it).size
+                        }
+                    )
+                }
+            }
+    }
 }