Add view model to home screen

LibrePass · Jul 15, 2024 · aaf86fa · aaf86fa
1 parent eaed8ef
commit aaf86fa
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 6 deletions.
diff --git a/ui-logic/src/main/java/dev/medzik/librepass/android/ui/Home.kt b/ui-logic/src/main/java/dev/medzik/librepass/android/ui/Home.kt
@@ -1,9 +1,8 @@
 package dev.medzik.librepass.android.ui
 
 import androidx.compose.runtime.Composable
-import androidx.compose.ui.platform.LocalContext
+import androidx.hilt.navigation.compose.hiltViewModel
 import androidx.navigation.NavController
-import dev.medzik.librepass.android.database.injection.DatabaseProvider
 import dev.medzik.librepass.android.ui.vault.VaultHome
 import dev.medzik.librepass.android.ui.vault.VaultHomeScreen
 import kotlinx.serialization.Serializable
@@ -12,10 +11,11 @@ import kotlinx.serialization.Serializable
 object Home
 
 @Composable
-fun HomeScreen(navController: NavController) {
-    val context = LocalContext.current
-    val repository = DatabaseProvider.provideRepository(context)
-    val credentials = repository.credentials.get()
+fun HomeScreen(
+    navController: NavController,
+    viewModel: HomeScreenViewModel = hiltViewModel()
+) {
+    val credentials = viewModel.getCredentials()
 
     if (credentials != null) {
         val args = VaultHome(

diff --git a/ui-logic/src/main/java/dev/medzik/librepass/android/ui/HomeScreenViewModel.kt b/ui-logic/src/main/java/dev/medzik/librepass/android/ui/HomeScreenViewModel.kt
@@ -0,0 +1,16 @@
+package dev.medzik.librepass.android.ui
+
+import androidx.lifecycle.ViewModel
+import dagger.hilt.android.lifecycle.HiltViewModel
+import dev.medzik.librepass.android.database.Credentials
+import dev.medzik.librepass.android.database.Repository
+import javax.inject.Inject
+
+@HiltViewModel
+class HomeScreenViewModel @Inject constructor(
+    private val repository: Repository
+) : ViewModel() {
+    fun getCredentials(): Credentials? {
+        return repository.credentials.get()
+    }
+}