Review NavHost in "NavHost.kt" in androidx.navigation.compose to see whether there is more ViewModel-related code to copy and adapt here, and merge its changes since last time

Author: ShreckYe

navigation/src/jsMain/kotlin/com/huanshankeji/androidx/navigation/compose/NavHost.js.kt

@@ -66,21 +66,19 @@ actual fun NavHost(
 
     // Find the ComposeNavigator, returning early if it isn't found
     // (such as is the case when using TestNavHostController)
-    val composeNavigator = navController.navigatorProvider.get<Navigator<out NavDestination>>(
-        ComposeNavigator.NAME
-    ) as? ComposeNavigator ?: return
+    val composeNavigator =
+        navController.navigatorProvider.get<Navigator<out NavDestination>>(ComposeNavigator.NAME)
+                as? ComposeNavigator ?: return
 
     //val currentBackStack by composeNavigator.backStack.collectAsState()
 
-//    BackHandler(currentBackStack.size > 1) {
-//        navController.popBackStack()
-//    }
+    // `progress`, `isPredictiveBack`, etc.
 
     /*
     DisposableEffect(lifecycleOwner) {
         // Setup the navController with proper owners
         navController.setLifecycleOwner(lifecycleOwner)
-        onDispose { }
+        onDispose {}
     }
     */
 
@@ -100,6 +98,13 @@ actual fun NavHost(
     val backStackEntry: NavBackStackEntry? = visibleEntries.lastOrNull()
 
     if (backStackEntry != null) {
+        // TODO remove this if not needed
+        DisposableEffect(true) {
+            onDispose {
+                visibleEntries.forEach { entry -> composeNavigator.onTransitionComplete(entry) }
+            }
+        }
+
         // `fillMaxSize` is added here to make the Box align to the size of its parent
         // TODO consider adding a version of `NavHost` without `modifier` and `contentAlignment`
         // Originally it was `transition.AnimatedContent` here.
@@ -110,8 +115,9 @@ actual fun NavHost(
             // while in the scope of the composable, we provide the navBackStackEntry as the
             // ViewModelStoreOwner and LifecycleOwner
             currentEntry?.LocalOwnersProvider(saveableStateHolder) {
-                (currentEntry.destination as ComposeNavigator.Destination)
-                    .content(currentEntry)
+                (currentEntry.destination as ComposeNavigator.Destination).content(
+                    currentEntry
+                )
             }
             */
             currentEntry?.let {
@@ -120,19 +126,18 @@ actual fun NavHost(
             }
         }
 
+        // TODO remove this if not needed
         DisposableEffect(true) {
             onDispose {
-                visibleEntries.forEach { entry ->
-                    composeNavigator.onTransitionComplete(entry)
-                }
+                visibleEntries.forEach { entry -> composeNavigator.onTransitionComplete(entry) }
             }
         }
     }
 
     /*
-    val dialogNavigator = navController.navigatorProvider.get<Navigator<out NavDestination>>(
-        DialogNavigator.NAME
-    ) as? DialogNavigator ?: return
+    val dialogNavigator =
+        navController.navigatorProvider.get<Navigator<out NavDestination>>(DialogNavigator.NAME)
+                as? DialogNavigator ?: return
 
     // Show any dialog destinations
     DialogHost(dialogNavigator)