Koin Composable nav graph viewmodel API proposal

MainNavigation.kt

//A new viewmodel would be provided each time this destination is met in the graph
//Passing the backStackEntry is actually optional because it will be the LocalViewModelStoreOwner
private fun NavGraphBuilder.viewModelToEntry() {
    composable(route = "route") { backStackEntry ->
        Screen(
            viewModel = getViewModel(backStackEntry)
        )
    }
}

//A ViewModel instance will be shared between two screens by scoping it to a parent's route
//The ViewModel will be recycled when the navgraph exits this subgraph.
private fun NavGraphBuilder.viewModelToSubGraph(navController: NavController){
    val parentRoute = "parent"
    val childARoute = "childA"
    val childBRoute = "childB"
    navigation(route = parentRoute, startDestination = childARoute) {
        val viewModel = getViewModel()
        composable(route = childARoute) {
            val parentEntry = remember {
                navController.getBackStackEntry(parentRoute)
            }
            ScreenA(
                viewModel = getViewModel(parentEntry)
            )
        }
        composable(route = childBRoute) {
            val parentEntry = remember {
                navController.getBackStackEntry(parentRoute)
            }
            ScreenB(
                viewModel = getViewModel(parentEntry)
            )
        }
    }
}

ViewModelComposeExt.kt

@OptIn(KoinInternalApi::class)
@Composable
inline fun <reified T : ViewModel> getViewModel(
    viewModelStoreOwner: ViewModelStoreOwner = LocalViewModelStoreOwner.current!!,
    qualifier: Qualifier? = null,
    scope: Scope = GlobalContext.get().scopeRegistry.rootScope,
    noinline parameters: ParametersDefinition? = null,
): T {
    return remember(qualifier, parameters) {
        scope.getViewModel(qualifier, {
            ViewModelOwner.from(viewModelStoreOwner)
        }, parameters)
    }
}