@momo-kits/native-kits 0.162.1-beta.2 → 0.162.1-beta.3-debug

package/compose/src/commonMain/kotlin/vn/momo/kits/navigation/Navigator.kt

@@ -0,0 +1,399 @@
+package vn.momo.kits.navigation
+
+import androidx.compose.foundation.layout.Box
+import androidx.compose.foundation.layout.fillMaxSize
+import androidx.compose.runtime.Composable
+import androidx.compose.runtime.MutableState
+import androidx.compose.runtime.mutableStateOf
+import androidx.compose.runtime.staticCompositionLocalOf
+import androidx.compose.ui.Alignment
+import androidx.compose.ui.Modifier
+import androidx.navigation.NavController
+import androidx.navigation.toRoute
+import kotlinx.coroutines.CoroutineScope
+import kotlinx.coroutines.Dispatchers
+import kotlinx.coroutines.SupervisorJob
+import kotlinx.coroutines.cancel
+import kotlinx.coroutines.delay
+import kotlinx.coroutines.launch
+import kotlinx.serialization.Serializable
+import vn.momo.kits.components.PopupAction
+import vn.momo.kits.components.PopupNotify
+import vn.momo.kits.navigation.BottomHeader.*
+import vn.momo.kits.navigation.component.SnackBar
+import vn.momo.maxapi.IMaxApi
+import kotlin.time.Duration.Companion.milliseconds
+
+class Navigator(
+    private val navController: NavController,
+    private val maxApi: IMaxApi?,
+    private val registry: DynamicScreenRegistry,
+    private val onNavigationEvent: ((NavigationEvent) -> Unit)? = null
+) {
+    private val scope = CoroutineScope(SupervisorJob() + Dispatchers.Main)
+
+    private fun emit(event: NavigationEvent) {
+        onNavigationEvent?.invoke(event)
+    }
+
+    /**
+     * The screen currently on top of THIS container's [navController] back stack.
+     *
+     * Use this instead of [DynamicScreenRegistry.getLatestScreen], which returns the globally
+     * last-registered screen across every NavigationContainer. With multiple containers sharing
+     * the single [DynamicScreenRegistry], the global "latest" is not guaranteed to belong to this
+     * navController, so back/dismiss/overlay logic could read the wrong options or unregister the
+     * wrong screen.
+     */
+    private fun currentScreen(): DynamicScreen? {
+        val entry = navController.currentBackStackEntry ?: return null
+        val id = runCatching { entry.toRoute<DynamicScreenRoute>().id }.getOrNull()
+            ?: runCatching { entry.toRoute<DynamicDialogRoute>().id }.getOrNull()
+            ?: return null
+        return registry.getScreen(id)
+    }
+
+    fun push(screenName: String, content: @Composable () -> Unit, options: NavigationOptions? = null) {
+        val route = registry.register(screenName, content, options)
+        emit(NavigationEvent.Push(screenName, route.id))
+        navController.navigate(DynamicScreenRoute(route.id))
+    }
+
+    fun replace(screenName: String, content: @Composable () -> Unit, options: NavigationOptions? = null) {
+        if (navController.previousBackStackEntry != null){
+            val replacedScreen = currentScreen()
+            emit(NavigationEvent.Replace(screenName, replacedScreen?.id ?: -1))
+            replacedScreen?.let { registry.unregisterScreen(it.id) }
+            navController.popBackStack()
+            push(screenName, content, options)
+        }
+    }
+
+    fun onBackSafe(callBack: (() -> Unit)? = null) {
+        val latestScreen = currentScreen()
+        val options = latestScreen?.options
+        if (options?.onBackHandler != null) {
+            options.onBackHandler.invoke()
+            return
+        }
+        val preventBack = options?.headerBackProps?.preventBack
+        if (preventBack != null) {
+            showModal(
+                content = {
+                    PopupNotify(preventBack.copy(
+                        primary = PopupAction(
+                            title = preventBack.primary?.title ?: "",
+                            onPress = {
+                                preventBack.primary?.onPress?.invoke()
+                                pop(2, callBack) // close popup notify, then close the screen
+                            }
+                        )
+                    ))
+                }
+            )
+        } else {
+            pop(1, callBack)
+        }
+    }
+
+    fun pop(count: Int = 1, callBack: (() -> Unit)? = null) {
+        emit(NavigationEvent.Pop(count, OverplayComponentRegistry.getOverplayType()))
+        scope.launch {
+            repeat(count) {
+                if (OverplayComponentRegistry.getOverplayType() == OverplayComponentType.SNACK_BAR){
+                    dismissScreen()
+                    OverplayComponentRegistry.hardClearAfterDismiss()
+                }
+                else if (OverplayComponentRegistry.isOverplayShowing()){
+                    dismissOverplay()
+                } else {
+                    dismissScreen()
+                }
+            }
+            callBack?.invoke()
+        }
+    }
+    private suspend fun dismissOverplay(isDelay: Boolean = true) {
+        OverplayComponentRegistry.clear()
+        if (isDelay) delay(300L.milliseconds)
+        OverplayComponentRegistry.hardClearAfterDismiss()
+    }
+
+    private suspend fun dismissScreen() {
+        if (navController.previousBackStackEntry != null){
+            // Capture the screen being dismissed BEFORE popping; after popBackStack the current
+            // entry is the revealed screen, so reading it post-pop would unregister the wrong one.
+            val dismissing = currentScreen()
+            navController.popBackStack()
+            delay(300L.milliseconds)
+            dismissing?.let { registry.unregisterScreen(it.id) }
+        } else {
+            maxApi?.dismiss {  }
+        }
+    }
+
+    fun present(screenName: String, content: @Composable () -> Unit, options: NavigationOptions? = null) {
+        val route = registry.register(screenName, content, options)
+        emit(NavigationEvent.Present(screenName, route.id))
+        navController.navigate(DynamicDialogRoute(route.id))
+    }
+
+    fun reset(screenName: String, content: @Composable () -> Unit, options: NavigationOptions? = null) {
+        registry.unregisterAll()
+
+        val route = registry.register(screenName, content, options)
+        emit(NavigationEvent.Reset(screenName, route.id))
+        navController.navigate(DynamicScreenRoute(route.id)) {
+            popUpTo(0) { inclusive = true }
+        }
+    }
+
+    fun showModal(
+        content: @Composable () -> Unit,
+        barrierDismissible: Boolean = true,
+        onDismiss: (() -> Unit)? = null
+    ){
+        val id = currentScreen()?.id ?: -1
+        emit(NavigationEvent.ShowModal(id, barrierDismissible))
+        OverplayComponentRegistry.registerOverplay(id, content, OverplayComponentType.MODAL, false, barrierDismissible, onDismiss)
+    }
+
+    fun showBottomSheet(
+        content: @Composable () -> Unit,
+        isSurface: Boolean = false,
+        barrierDismissible: Boolean = true,
+        onDismiss: (() -> Unit)? = null,
+        bottomSheetHeader: BottomHeader? = null
+    ){
+        val id = currentScreen()?.id ?: -1
+        emit(NavigationEvent.ShowBottomSheet(id, barrierDismissible))
+        OverplayComponentRegistry.registerOverplay(id, content, OverplayComponentType.BOTTOM_SHEET, isSurface, barrierDismissible, onDismiss, bottomSheetHeader)
+    }
+
+    fun showSnackBar(snackBar: SnackBar, onDismiss: (() -> Unit)? = null) {
+        val id = currentScreen()?.id ?: -1
+        emit(NavigationEvent.ShowSnackBar(id))
+        scope.launch {
+            OverplayComponentRegistry.registerOverplay(
+                id = id,
+                content = {
+                    SnackBar(snackBar, onDismiss)
+                },
+                type = OverplayComponentType.SNACK_BAR,
+                onDismiss = onDismiss
+            )
+        }
+    }
+
+    fun hideSnackBar() {
+        scope.launch {
+            OverplayComponentRegistry.clear()
+            delay(250L.milliseconds)
+            OverplayComponentRegistry.hardClearAfterDismiss()
+        }
+    }
+
+    fun dispose(){
+        emit(NavigationEvent.Dispose)
+        scope.cancel()
+    }
+}
+
+val LocalNavigator = staticCompositionLocalOf<Navigator> {
+    error("No NavigationStack provided")
+}
+
+@Serializable
+data class DynamicScreenRoute(val id: Int)
+
+@Serializable
+data class DynamicDialogRoute(val id: Int)
+
+data class DynamicScreen(
+    val id: Int,
+    val name: String,
+    val content: @Composable () -> Unit,
+    var options: NavigationOptions? = null
+)
+
+/**
+ * Per-container store of dynamic screens. Each [NavigationContainer] owns its own instance, so the
+ * number of live containers no longer matters: `getLatestScreen`/`unregisterAll` only ever see the
+ * screens of their own container.
+ *
+ * Ids are still drawn from a process-wide counter so they stay globally unique — this keeps the
+ * still-global [OverplayComponentRegistry] (which keys overlays by screen id) collision-free.
+ */
+class DynamicScreenRegistry {
+    private val screens = mutableMapOf<Int, DynamicScreen>()
+
+    fun register(screenName: String, content: @Composable () -> Unit, options: NavigationOptions?): DynamicScreenRoute {
+        val id = nextGlobalId()
+        screens[id] = DynamicScreen(
+            id = id,
+            name = screenName,
+            content = content,
+            options = options
+        )
+        return DynamicScreenRoute(id)
+    }
+
+    fun nextId(): Int = nextGlobalId()
+
+    fun bind(id: Int, screenName: String, content: @Composable () -> Unit, options: NavigationOptions?) {
+        screens[id] = DynamicScreen(
+            id = id,
+            name = screenName,
+            content = content,
+            options = options
+        )
+        advanceGlobalIdPast(id) // keep counter ahead after process-death restore
+    }
+
+    fun unregisterScreen(id: Int) {
+        screens.remove(id)
+    }
+
+    fun getLatestScreen(): DynamicScreen?{
+        return screens.entries.lastOrNull()?.value
+    }
+
+    fun unregisterAll(){
+        val ids = screens.keys.toList()
+        for (id in ids) {
+            unregisterScreen(id)
+        }
+    }
+
+    fun getScreen(id: Int): DynamicScreen? = screens[id]
+
+    fun setOptions(id: Int, options: NavigationOptions){
+        screens[id]?.options = options
+    }
+
+    companion object {
+        private var globalIdCounter = 1
+
+        private fun nextGlobalId(): Int = globalIdCounter++
+
+        private fun advanceGlobalIdPast(id: Int) {
+            if (id >= globalIdCounter) globalIdCounter = id + 1
+        }
+    }
+}
+
+val LocalDynamicScreenRegistry = staticCompositionLocalOf<DynamicScreenRegistry?> { null }
+
+
+sealed class OverplayComponentParams {
+    class Modal(
+        val onDismiss: (() -> Unit)? = null,
+        val barrierDismissible: Boolean = true
+    ) : OverplayComponentParams()
+
+    class BottomSheet(
+        val isSurface: Boolean = false,
+        val onDismiss: (() -> Unit)? = null,
+        val barrierDismissible: Boolean = true,
+        val bottomSheetHeader: BottomHeader? = null,
+    ) : OverplayComponentParams()
+
+    class SnackBar : OverplayComponentParams()
+}
+
+data class OverplayComponent(
+    val id: Int,
+    val type: OverplayComponentType? = null,
+    val content: @Composable () -> Unit,
+    val params: OverplayComponentParams? = null
+)
+
+enum class OverplayComponentType {
+    MODAL, BOTTOM_SHEET, SNACK_BAR
+}
+
+object OverplayComponentRegistry {
+    private var currentOverlayComponent : MutableState<OverplayComponent?> = mutableStateOf(null)
+    private var requestClose: (() -> Unit)? = null
+    internal fun bindClose(handler: (() -> Unit)?) {
+        requestClose = handler
+    }
+
+    fun registerOverplay(
+        id: Int,
+        content: @Composable () -> Unit,
+        type: OverplayComponentType,
+        isSurface: Boolean = false,
+        barrierDismissible: Boolean = true,
+        onDismiss: (() -> Unit)?,
+        bottomSheetHeader: BottomHeader? = null,
+    ){
+        val params = when(type){
+            OverplayComponentType.MODAL -> OverplayComponentParams.Modal(onDismiss, barrierDismissible)
+            OverplayComponentType.BOTTOM_SHEET -> OverplayComponentParams.BottomSheet(isSurface, onDismiss, barrierDismissible, bottomSheetHeader)
+            OverplayComponentType.SNACK_BAR -> OverplayComponentParams.SnackBar()
+        }
+
+        currentOverlayComponent.value = OverplayComponent(
+            id = id,
+            type = type,
+            content = content,
+            params = params
+        )
+    }
+
+    fun isOverplayShowing(): Boolean = currentOverlayComponent.value != null
+
+    fun getOverplayType(): OverplayComponentType? = currentOverlayComponent.value?.type
+
+    fun currentRootId(): Int? = currentOverlayComponent.value?.id
+
+    fun clear(){
+        if (requestClose != null) {
+            requestClose?.invoke()
+        } else {
+            currentOverlayComponent.value = null
+        }
+    }
+
+    internal fun hardClearAfterDismiss() {
+        currentOverlayComponent.value = null
+        requestClose = null
+    }
+
+    @Composable
+    internal fun OverlayComponent(){
+        val overplay = currentOverlayComponent.value ?: return
+
+        when (val params = overplay.params) {
+            is OverplayComponentParams.BottomSheet -> {
+                BottomSheet(
+                    content = overplay.content,
+                    header = params.bottomSheetHeader ?: Title(),
+                    isSurface = params.isSurface,
+                    barrierDismissible = params.barrierDismissible,
+                    onDismiss = params.onDismiss
+                )
+            }
+
+            is OverplayComponentParams.Modal -> {
+                ModalScreen(
+                    content = overplay.content,
+                    barrierDismissible = params.barrierDismissible,
+                    onDismiss = params.onDismiss
+                )
+            }
+
+            is OverplayComponentParams.SnackBar -> {
+                Box(
+                    modifier = Modifier.fillMaxSize(),
+                    contentAlignment = Alignment.BottomCenter
+                ) {
+                    overplay.content()
+                }
+            }
+
+            null -> {}
+        }
+    }
+}

package/compose/src/commonMain/kotlin/vn/momo/kits/navigation/ScrollToTop.kt

@@ -0,0 +1,8 @@
+package vn.momo.kits.navigation
+
+import androidx.compose.runtime.Composable
+
+@Composable
+internal expect fun RegisterScrollToTop(callback: (() -> Unit)?)
+
+