From 24ede3bbac09747c770fd503cc4da015a0c2450f Mon Sep 17 00:00:00 2001
From: InsanusMokrassar <ovsyannikov.alexey95@gmail.com>
Date: Fri, 21 Jun 2024 14:30:01 +0600
Subject: [PATCH] add navigation node factory providing

---
 .../src/commonMain/kotlin/ComposeProviders.kt |  5 +++
 .../src/commonMain/kotlin/InitNavigation.kt   | 37 ++++++++++---------
 2 files changed, 25 insertions(+), 17 deletions(-)

diff --git a/compose/src/commonMain/kotlin/ComposeProviders.kt b/compose/src/commonMain/kotlin/ComposeProviders.kt
index 7344b2d2..683eabf3 100644
--- a/compose/src/commonMain/kotlin/ComposeProviders.kt
+++ b/compose/src/commonMain/kotlin/ComposeProviders.kt
@@ -1,5 +1,7 @@
 package dev.inmo.navigation.compose
 
+import androidx.compose.runtime.Composable
+import androidx.compose.runtime.ComposeCompilerApi
 import androidx.compose.runtime.ProvidableCompositionLocal
 import androidx.compose.runtime.compositionLocalOf
 import dev.inmo.navigation.core.NavigationChain
@@ -8,6 +10,7 @@ import dev.inmo.navigation.core.NavigationNodeFactory
 
 val LocalNavigationNodeFactory = compositionLocalOf<NavigationNodeFactory<*>> { NavigationNodeFactory<Any> { _, _ -> null } }
 
+internal val InternalLocalNavigationNodeFactory: ProvidableCompositionLocal<NavigationNodeFactory<*>> = compositionLocalOf<NavigationNodeFactory<*>> { NavigationNodeFactory<Any> { _, _ -> null } }
 internal val InternalLocalNavigationChainProvider: ProvidableCompositionLocal<NavigationChain<*>> = compositionLocalOf<NavigationChain<*>> { NavigationChain<Any>(null, { _, _ -> null }) }
 internal val InternalLocalNavigationNodeProvider: ProvidableCompositionLocal<NavigationNode<*, *>> = compositionLocalOf {
     NavigationNode.Empty<Any, Any>(
@@ -16,6 +19,8 @@ internal val InternalLocalNavigationNodeProvider: ProvidableCompositionLocal<Nav
     )
 }
 
+@Suppress("UNCHECKED_CAST")
+internal fun <Base> InternalLocalNavigationNodeFactory(): ProvidableCompositionLocal<NavigationNodeFactory<Base>> = InternalLocalNavigationNodeFactory as ProvidableCompositionLocal<NavigationNodeFactory<Base>>
 @Suppress("UNCHECKED_CAST")
 internal fun <Base> LocalNavigationChainProvider(): ProvidableCompositionLocal<NavigationChain<Base>> = InternalLocalNavigationChainProvider as ProvidableCompositionLocal<NavigationChain<Base>>
 @Suppress("UNCHECKED_CAST")
diff --git a/compose/src/commonMain/kotlin/InitNavigation.kt b/compose/src/commonMain/kotlin/InitNavigation.kt
index 2c888685..ac03309a 100644
--- a/compose/src/commonMain/kotlin/InitNavigation.kt
+++ b/compose/src/commonMain/kotlin/InitNavigation.kt
@@ -1,6 +1,7 @@
 package dev.inmo.navigation.compose
 
 import androidx.compose.runtime.Composable
+import androidx.compose.runtime.CompositionLocalProvider
 import androidx.compose.runtime.remember
 import androidx.compose.runtime.rememberCoroutineScope
 import dev.inmo.kslog.common.TagLogger
@@ -55,26 +56,28 @@ fun <Base> initNavigation(
             }
         }
     }
-    logger.d { "Hierarchy saving enabled" }
+    CompositionLocalProvider(InternalLocalNavigationNodeFactory<Base>() provides resultNodesFactory) {
+        logger.d { "Hierarchy saving enabled" }
 
-    val existsChain = configsRepo.get()
+        val existsChain = configsRepo.get()
 
-    logger.d {
-        if (existsChain == null) {
-            "Can't find exists chain. Using default one: $defaultStartChain"
-        } else {
-            "Took exists stored chain $existsChain"
+        logger.d {
+            if (existsChain == null) {
+                "Can't find exists chain. Using default one: $defaultStartChain"
+            } else {
+                "Took exists stored chain $existsChain"
+            }
         }
-    }
 
-    val restoredRootChain = remember {
-        restoreHierarchy<Base>(
-            existsChain ?: defaultStartChain,
-            factory = resultNodesFactory,
-            rootChain = rootChain,
-            dropRedundantChainsOnRestore = dropRedundantChainsOnRestore
-        )
+        val restoredRootChain = remember {
+            restoreHierarchy<Base>(
+                existsChain ?: defaultStartChain,
+                factory = resultNodesFactory,
+                rootChain = rootChain,
+                dropRedundantChainsOnRestore = dropRedundantChainsOnRestore
+            )
+        }
+        restoredRootChain ?.start(subscope)
+        restoredRootChain ?.StartInCompose({ savingJob.cancel() })
     }
-    restoredRootChain ?.start(subscope)
-    restoredRootChain ?.StartInCompose({ savingJob.cancel() })
 }