tmppvghivng.kotlin

@Serializable
data class PickedImage(val url: String, val width: Int, val height: Int)

@Composable
fun ImagePickerScreen(navController: NavController) {
    LazyVerticalGrid(columns = GridCells.Fixed(3)) {
        items(images) { image ->
            AsyncImage(
                model = image.url,
                contentDescription = null,
                modifier = Modifier
                    .fillMaxWidth()
                    .aspectRatio(1f)
                    .clickable {
                        navController.previousBackStackEntry
                            ?.savedStateHandle
                            ?.set("picked_image", PickedImage(
                                url = image.url,
                                width = image.width,
                                height = image.height
                            ))
                        navController.popBackStack()
                    }
            )
        }
    }
}

@Composable
fun ImageListScreen(navController: NavController) {
    val pickedImage = navController.currentBackStackEntry
        ?.savedStateHandle
        ?.getStateFlow<PickedImage?>("picked_image", null)
        ?.collectAsState()
}