Fix viewmodel input

Author: fornewid

feature/detail/impl/src/main/java/soup/movie/feature/detail/impl/DetailNavGraph.kt

@@ -17,13 +17,13 @@ package soup.movie.feature.detail.impl
 
 import androidx.compose.runtime.Composable
 import androidx.compose.runtime.remember
-import androidx.hilt.navigation.compose.hiltViewModel
+import androidx.hilt.lifecycle.viewmodel.compose.hiltViewModel
 import androidx.navigation3.runtime.NavKey
 import androidx.navigation3.runtime.entryProvider
 import androidx.navigation3.ui.NavDisplay
 import kotlinx.serialization.Serializable
 
-private sealed interface DetailScreen : NavKey {
+sealed interface DetailScreen : NavKey {
 
     @Serializable
     data class Home(val movieId: String) : DetailScreen
@@ -44,7 +44,11 @@ fun DetailNavGraph(movieId: String) {
     // Define entry provider for detail destinations
     val entryProvider = entryProvider<NavKey> {
         entry<DetailScreen.Home> { key ->
-            val viewModel = hiltViewModel<DetailViewModel>()
+            val viewModel = hiltViewModel<DetailViewModel, DetailViewModel.Factory>(
+                creationCallback = { factory ->
+                    factory.create(key)
+                }
+            )
             DetailScreen(
                 viewModel = viewModel,
                 onPosterClick = {

feature/detail/impl/src/main/java/soup/movie/feature/detail/impl/DetailViewModel.kt

@@ -15,9 +15,11 @@
  */
 package soup.movie.feature.detail.impl
 
-import androidx.lifecycle.SavedStateHandle
 import androidx.lifecycle.ViewModel
 import androidx.lifecycle.viewModelScope
+import dagger.assisted.Assisted
+import dagger.assisted.AssistedFactory
+import dagger.assisted.AssistedInject
 import dagger.hilt.android.lifecycle.HiltViewModel
 import kotlinx.coroutines.CoroutineDispatcher
 import kotlinx.coroutines.flow.MutableStateFlow
@@ -35,16 +37,15 @@ import soup.movie.model.MovieDetailModel
 import soup.movie.model.MovieModel
 import soup.movie.model.OpenDateAlarmModel
 import java.time.temporal.ChronoUnit
-import javax.inject.Inject
 
-@HiltViewModel
-class DetailViewModel @Inject constructor(
-    savedStateHandle: SavedStateHandle,
+@HiltViewModel(assistedFactory = DetailViewModel.Factory::class)
+class DetailViewModel @AssistedInject constructor(
+    @Assisted private val input: DetailScreen.Home,
     private val repository: MovieRepository,
     @DefaultDispatcher private val defaultDispatcher: CoroutineDispatcher,
 ) : ViewModel() {
 
-    private val movieId: String = savedStateHandle["movieId"]!!
+    private val movieId: String = input.movieId
 
     private val _uiModel = MutableStateFlow<DetailUiModel>(DetailUiModel.None)
     val uiModel: StateFlow<DetailUiModel> = _uiModel
@@ -221,4 +222,9 @@ class DetailViewModel @Inject constructor(
             return 0
         }
     }
+
+    @AssistedFactory
+    interface Factory {
+        fun create(input: DetailScreen.Home): DetailViewModel
+    }
 }