NestedNavigation.kt

package com.uragiristereo.composeplayground.experiments.nestednavigation

import androidx.compose.animation.ExperimentalAnimationApi
import androidx.compose.foundation.background
import androidx.compose.foundation.border
import androidx.compose.foundation.layout.Box
import androidx.compose.foundation.layout.Column
import androidx.compose.foundation.layout.fillMaxSize
import androidx.compose.material.Button
import androidx.compose.material.Scaffold
import androidx.compose.material.Text
import androidx.compose.material.TopAppBar
import androidx.compose.runtime.Composable
import androidx.compose.runtime.CompositionLocalProvider
import androidx.compose.runtime.compositionLocalOf
import androidx.compose.ui.Alignment
import androidx.compose.ui.Modifier
import androidx.compose.ui.graphics.Color
import androidx.compose.ui.unit.dp
import androidx.navigation.NavHostController
import soup.compose.material.motion.materialElevationScaleIn
import soup.compose.material.motion.materialElevationScaleOut
import soup.compose.material.motion.navigation.MaterialMotionNavHost
import soup.compose.material.motion.navigation.composable
import soup.compose.material.motion.navigation.rememberMaterialMotionNavController
import soup.compose.material.motion.translateXIn
import soup.compose.material.motion.translateXOut

val LocalNavigation = compositionLocalOf<NavHostController> { error("no NavHost provided!") }

@ExperimentalAnimationApi
@Composable
fun NestedNavigation() {
    val navHost = rememberMaterialMotionNavController()

    CompositionLocalProvider(LocalNavigation provides navHost) {
        MaterialMotionNavHost(
            navController = navHost,
            startDestination = "first",
        ) {
            composable(
                route = "first",
                enterMotionSpec = { _, _ ->
                    translateXIn({ it }, 250)
                },
                popExitMotionSpec = { _, _ ->
                    translateXOut({ it }, 250)
                },
                exitMotionSpec = { _, _ ->
                    materialElevationScaleOut()
                },
                popEnterMotionSpec = { _, _ ->
                    materialElevationScaleIn()
                },
            ) {
                NextScreen(
                    title = "First",
                    color = Color.Cyan,
                    nextRoute = "second",
                )
            }

            composable(
                route = "second",
                enterMotionSpec = { _, _ ->
                    translateXIn({ it }, 250)
                },
                popExitMotionSpec = { _, _ ->
                    translateXOut({ it }, 250)
                },
                exitMotionSpec = { _, _ ->
                    materialElevationScaleOut()
                },
                popEnterMotionSpec = { _, _ ->
                    materialElevationScaleIn()
                },
            ) {
                NextScreen(
                    title = "Second",
                    color = Color.Red,
                    nextRoute = "third",
                )
            }

            composable(
                route = "third",
                enterMotionSpec = { _, _ ->
                    translateXIn({ it }, 250)
                },
                popExitMotionSpec = { _, _ ->
                    translateXOut({ it }, 250)
                },
                exitMotionSpec = { _, _ ->
                    materialElevationScaleOut()
                },
                popEnterMotionSpec = { _, _ ->
                    materialElevationScaleIn()
                },
            ) {
                NextScreen(
                    title = "Third",
                    color = Color.Green,
                    nextRoute = "fourth",
                )
            }

            composable(
                route = "fourth",
                enterMotionSpec = { _, _ ->
                    translateXIn({ it }, 250)
                },
                popExitMotionSpec = { _, _ ->
                    translateXOut({ it }, 250)
                },
                exitMotionSpec = { _, _ ->
                    materialElevationScaleOut()
                },
                popEnterMotionSpec = { _, _ ->
                    materialElevationScaleIn()
                },
            ) {
                NextScreen(
                    title = "Fourth",
                    color = Color.Yellow,
                )
            }
        }
    }
}

@Composable
fun NextScreen(
    title: String,
    color: Color,
    nextRoute: String? = null,
) {
    val nav = LocalNavigation.current

    Scaffold(
        topBar = {
            TopAppBar(
                title = { Text(text = title) },
            )
        }
    ) {
        Box(
            contentAlignment = Alignment.Center,
            modifier = Modifier
                .fillMaxSize()
                .background(color = color)
                .border(width = 4.dp, color = color),
        ) {
            Column(
                horizontalAlignment = Alignment.CenterHorizontally,
            ) {
                Text(
                    text = title,
                    color = Color.Black,
                )

                Button(
                    enabled = nextRoute != null,
                    onClick = {
                        nav.navigate(nextRoute ?: "")
                    },
                    content = { Text(text = "go to $nextRoute") },
                )

                Button(
                    onClick = {
                        nav.navigateUp()
//                        nav.popBackStack()
                    },
                    content = { Text(text = "back") },
                )
            }
        }
    }
}