1 files changed, 74 insertions, 30 deletions

diff --git a/app/src/main/java/foundation/e/privacycentralapp/DependencyContainer.kt b/app/src/main/java/foundation/e/privacycentralapp/DependencyContainer.kt
index 727d00d..6be3724 100644
--- a/app/src/main/java/foundation/e/privacycentralapp/DependencyContainer.kt
+++ b/app/src/main/java/foundation/e/privacycentralapp/DependencyContainer.kt
@@ -20,6 +20,10 @@ package foundation.e.privacycentralapp
 import android.app.Application
 import android.content.Context
 import android.os.Process
+import androidx.lifecycle.DEFAULT_ARGS_KEY
+import androidx.lifecycle.ViewModel
+import androidx.lifecycle.ViewModelProvider
+import androidx.lifecycle.viewmodel.CreationExtras
 import foundation.e.privacycentralapp.data.repositories.AppListsRepository
 import foundation.e.privacycentralapp.data.repositories.LocalStateRepository
 import foundation.e.privacycentralapp.data.repositories.TrackersRepository
@@ -30,11 +34,12 @@ import foundation.e.privacycentralapp.domain.usecases.IpScramblingStateUseCase
 import foundation.e.privacycentralapp.domain.usecases.TrackersStateUseCase
 import foundation.e.privacycentralapp.domain.usecases.TrackersStatisticsUseCase
 import foundation.e.privacycentralapp.dummy.CityDataSource
-import foundation.e.privacycentralapp.features.dashboard.DashBoardViewModelFactory
-import foundation.e.privacycentralapp.features.internetprivacy.InternetPrivacyViewModelFactory
-import foundation.e.privacycentralapp.features.location.FakeLocationViewModelFactory
-import foundation.e.privacycentralapp.features.trackers.TrackersViewModelFactory
-import foundation.e.privacycentralapp.features.trackers.apptrackers.AppTrackersViewModelFactory
+import foundation.e.privacycentralapp.features.dashboard.DashboardViewModel
+import foundation.e.privacycentralapp.features.internetprivacy.InternetPrivacyViewModel
+import foundation.e.privacycentralapp.features.location.FakeLocationViewModel
+import foundation.e.privacycentralapp.features.trackers.TrackersViewModel
+import foundation.e.privacycentralapp.features.trackers.apptrackers.AppTrackersFragment
+import foundation.e.privacycentralapp.features.trackers.apptrackers.AppTrackersViewModel
 import foundation.e.privacymodules.ipscrambler.IpScramblerModule
 import foundation.e.privacymodules.ipscramblermodule.IIpScramblerModule
 import foundation.e.privacymodules.location.FakeLocationModule
@@ -43,14 +48,15 @@ import foundation.e.privacymodules.permissions.PermissionsPrivacyModule
 import foundation.e.privacymodules.permissions.data.ApplicationDescription
 import foundation.e.privacymodules.trackers.api.BlockTrackersPrivacyModule
 import foundation.e.privacymodules.trackers.api.TrackTrackersPrivacyModule
+import kotlinx.coroutines.DelicateCoroutinesApi
 import kotlinx.coroutines.FlowPreview
 import kotlinx.coroutines.GlobalScope
 
 /**
  * Simple container to hold application wide dependencies.
  *
- * TODO: Test if this implementation is leaky.
  */
+@OptIn(DelicateCoroutinesApi::class)
 class DependencyContainer(val app: Application) {
     val context: Context by lazy { app.applicationContext }
 
@@ -102,32 +108,17 @@ class DependencyContainer(val app: Application) {
         )
     }
 
-    // ViewModelFactories
-    val dashBoardViewModelFactory by lazy {
-        DashBoardViewModelFactory(getQuickPrivacyStateUseCase, trackersStatisticsUseCase)
-    }
-
-    val fakeLocationViewModelFactory by lazy {
-        FakeLocationViewModelFactory(
-            getQuickPrivacyStateUseCase = getQuickPrivacyStateUseCase,
-            fakeLocationStateUseCase = fakeLocationStateUseCase
-        )
-    }
-
-    val internetPrivacyViewModelFactory by lazy {
-        InternetPrivacyViewModelFactory(ipScramblerModule, getQuickPrivacyStateUseCase, ipScramblingStateUseCase, appListUseCase)
-    }
-
-    val trackersViewModelFactory by lazy {
-        TrackersViewModelFactory(getQuickPrivacyStateUseCase, trackersStatisticsUseCase)
-    }
-
-    val appTrackersViewModelFactory by lazy {
-        AppTrackersViewModelFactory(trackersStateUseCase, trackersStatisticsUseCase, getQuickPrivacyStateUseCase)
-    }
+    val viewModelsFactory by lazy { ViewModelsFactory(
+        getQuickPrivacyStateUseCase = getQuickPrivacyStateUseCase,
+        trackersStatisticsUseCase = trackersStatisticsUseCase,
+        trackersStateUseCase = trackersStateUseCase,
+        fakeLocationStateUseCase = fakeLocationStateUseCase,
+        ipScramblerModule = ipScramblerModule,
+        ipScramblingStateUseCase = ipScramblingStateUseCase,
+        appListUseCase = appListUseCase
+    ) }
 
     // Background
-    @FlowPreview
     fun initBackgroundSingletons() {
         trackersStateUseCase
         ipScramblingStateUseCase
@@ -142,3 +133,56 @@ class DependencyContainer(val app: Application) {
         )
     }
 }
+
+class ViewModelsFactory(
+    private val getQuickPrivacyStateUseCase: GetQuickPrivacyStateUseCase,
+    private val trackersStatisticsUseCase: TrackersStatisticsUseCase,
+    private val trackersStateUseCase: TrackersStateUseCase,
+    private val fakeLocationStateUseCase: FakeLocationStateUseCase,
+    private val ipScramblerModule: IIpScramblerModule,
+    private val ipScramblingStateUseCase: IpScramblingStateUseCase,
+    private val appListUseCase: AppListUseCase
+): ViewModelProvider.Factory {
+
+    @Suppress("UNCHECKED_CAST")
+    override fun <T : ViewModel> create(modelClass: Class<T>, extras: CreationExtras): T {
+        return when (modelClass) {
+            AppTrackersViewModel::class.java -> {
+                val fallbackUid = android.os.Process.myPid()
+                val appUid = extras[DEFAULT_ARGS_KEY]?.
+                    getInt(AppTrackersFragment.PARAM_APP_UID, fallbackUid)?: fallbackUid
+
+                AppTrackersViewModel(
+                    appUid = appUid,
+                    trackersStateUseCase = trackersStateUseCase,
+                    trackersStatisticsUseCase = trackersStatisticsUseCase,
+                    getQuickPrivacyStateUseCase = getQuickPrivacyStateUseCase
+                )
+            }
+
+            TrackersViewModel::class.java ->
+                TrackersViewModel(
+                    getQuickPrivacyStateUseCase = getQuickPrivacyStateUseCase,
+                    trackersStatisticsUseCase = trackersStatisticsUseCase
+                )
+            FakeLocationViewModel::class.java ->
+                FakeLocationViewModel(
+                    getQuickPrivacyStateUseCase = getQuickPrivacyStateUseCase,
+                    fakeLocationStateUseCase = fakeLocationStateUseCase
+                )
+            InternetPrivacyViewModel::class.java ->
+                InternetPrivacyViewModel(
+                    ipScramblerModule = ipScramblerModule,
+                    getQuickPrivacyStateUseCase = getQuickPrivacyStateUseCase,
+                    ipScramblingStateUseCase = ipScramblingStateUseCase,
+                    appListUseCase = appListUseCase
+                )
+            DashboardViewModel::class.java ->
+                DashboardViewModel(
+                    getPrivacyStateUseCase = getQuickPrivacyStateUseCase,
+                    trackersStatisticsUseCase = trackersStatisticsUseCase
+                )
+            else -> throw IllegalArgumentException("Unknown class $modelClass")
+        } as T
+    }
+}