Share ViewModel Between Composable

DefaultNavHost.kt

@Composable
fun DefaultNavHost(
    modifier: Modifier = Modifier
) {
    val navController = rememberNavController()

    NavHost(
        navController = navController,
        startDestination = HOME_ROUTE,
        modifier = modifier,
    ) {

        composable(HOME_ROUTE) {
            HomeScreen { navController.navigate(SECOND_ROUTE) }
        }

        composable(SECOND_ROUTE) {
            SecondScreen(
                viewModel = hiltViewModel(),
                onClickBack = navController::popBackStack,
                onClickNext = { navController.navigate(THIRD_ROUTE) }
            )
        }

        composable(THIRD_ROUTE) {
            val secondViewModel = navController.hiltNavGraphViewModel<SecondViewModel>(SECOND_ROUTE, currentEntry)
            ThirdScreen(
                viewModel = secondViewModel,
                onClickBack = navController::popBackStack,
            )
        }

    }
}

NavGraphViewModel.kt

@Composable
inline fun <reified VM : ViewModel> NavController.hiltNavGraphViewModel(
    route: String, currentEntry: NavBackStackEntry
): VM {
    val routeEntry = remember(currentEntry) { getBackStackEntry(route) }
    return hiltViewModel<VM>(routeEntry)
}