@momo-kits/native-kits 0.158.1-beta.1 → 0.158.1-beta.2-debug

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

@@ -0,0 +1,333 @@
+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 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
+
+class Navigator(
+    private val navController: NavController,
+    private val maxApi: IMaxApi?
+) {
+    private val scope = CoroutineScope(SupervisorJob() + Dispatchers.Main)
+
+    fun push(screenName: String, content: @Composable () -> Unit, options: NavigationOptions? = null) {
+        val route = DynamicScreenRegistry.register(screenName, content, options)
+        navController.navigate(DynamicScreenRoute(route.id))
+    }
+
+    fun replace(screenName: String, content: @Composable () -> Unit, options: NavigationOptions? = null) {
+        if (navController.previousBackStackEntry != null){
+            val latestScreen = DynamicScreenRegistry.getLatestScreen()
+            latestScreen?.let { DynamicScreenRegistry.unregisterScreen(it.id) }
+            navController.popBackStack()
+            push(screenName, content, options)
+        }
+    }
+
+    fun onBackSafe(callBack: (() -> Unit)? = null) {
+        val latestScreen = DynamicScreenRegistry.getLatestScreen()
+        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) {
+        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)
+        OverplayComponentRegistry.hardClearAfterDismiss()
+    }
+
+    private suspend fun dismissScreen() {
+        if (navController.previousBackStackEntry != null){
+            navController.popBackStack()
+            delay(300L)
+            DynamicScreenRegistry.getLatestScreen()?.let { it1 ->
+                DynamicScreenRegistry.unregisterScreen(it1.id)
+            }
+        } else {
+            maxApi?.dismiss {  }
+        }
+    }
+
+    fun present(screenName: String, content: @Composable () -> Unit, options: NavigationOptions? = null) {
+        val route = DynamicScreenRegistry.register(screenName, content, options)
+        navController.navigate(DynamicDialogRoute(route.id))
+    }
+
+    fun reset(screenName: String, content: @Composable () -> Unit, options: NavigationOptions? = null) {
+        DynamicScreenRegistry.unregisterAll()
+
+        val route = DynamicScreenRegistry.register(screenName, content, options)
+        navController.navigate(DynamicScreenRoute(route.id)) {
+            popUpTo(0) { inclusive = true }
+        }
+    }
+
+    fun showModal(
+        content: @Composable () -> Unit,
+        barrierDismissible: Boolean = true,
+        onDismiss: (() -> Unit)? = null
+    ){
+        val id = DynamicScreenRegistry.getLatestScreen()?.id ?: -1
+        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 = DynamicScreenRegistry.getLatestScreen()?.id ?: -1
+        OverplayComponentRegistry.registerOverplay(id, content, OverplayComponentType.BOTTOM_SHEET, isSurface, barrierDismissible, onDismiss, bottomSheetHeader)
+    }
+
+    fun showSnackBar(snackBar: SnackBar, onDismiss: (() -> Unit)? = null) {
+        val id = DynamicScreenRegistry.getLatestScreen()?.id ?: -1
+        scope.launch {
+            OverplayComponentRegistry.registerOverplay(
+                id = id,
+                content = {
+                    SnackBar(snackBar, onDismiss)
+                },
+                type = OverplayComponentType.SNACK_BAR,
+                onDismiss = onDismiss
+            )
+        }
+    }
+
+    fun hideSnackBar() {
+        scope.launch {
+            OverplayComponentRegistry.clear()
+            delay(250L)
+            OverplayComponentRegistry.hardClearAfterDismiss()
+        }
+    }
+
+    fun 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
+)
+
+object DynamicScreenRegistry {
+    private val screens = mutableMapOf<Int, DynamicScreen>()
+    private var idCounter = 1
+
+    fun register(screenName: String, content: @Composable () -> Unit, options: NavigationOptions?): DynamicScreenRoute {
+        val id = idCounter++
+        screens[id] = DynamicScreen(
+            id = id,
+            name = screenName,
+            content = content,
+            options = options
+        )
+        return DynamicScreenRoute(id)
+    }
+
+    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
+    }
+}
+
+
+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
+    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 -> {}
+        }
+    }
+}