如何在Jetpack Compose中获取NavHost当前显示的屏幕对象？

要解决动态根据当前屏幕更新Scaffold的FAB和顶部栏的问题，我们可以通过类型安全的密封类路由+通用扩展函数实现，无需为每个屏幕单独调用toRoute<T>()：

步骤1：定义统一的屏幕路由密封类 #

先创建一个所有屏幕都继承的密封类，确保所有路由都实现序列化（适配Navigation的类型安全路由要求）：

import kotlinx.serialization.Serializable

@Serializable
sealed class Screen {
    @Serializable
    object MainScreen : Screen()
    
    @Serializable
    data class DetailScreen(val itemId: Int) : Screen()
    
    // 其他屏幕路由同理添加
}

步骤2：编写通用的路由转换扩展函数 #

实现一个扩展函数，将NavBackStackEntry通用转换为Screen实例，无需指定具体泛型：

import androidx.navigation.NavBackStackEntry
import kotlinx.serialization.decodeFromString
import kotlinx.serialization.json.Json

fun NavBackStackEntry.toScreen(): Screen {
    return try {
        // 直接解码路由字符串为Screen密封类实例
        Json.decodeFromString(destination.route ?: "")
    } catch (e: Exception) {
        // 异常情况下默认返回主屏幕，可根据需求调整
        Screen.MainScreen
    }
}

步骤3：在Compose中动态获取并使用当前屏幕 #

在你的Compose代码中，监听当前返回栈条目，转换为currentScreen后，通过when表达式动态控制Scaffold的UI：

import androidx.compose.floatingactionbutton.FloatingActionButton
import androidx.compose.material.Icon
import androidx.compose.material.Scaffold
import androidx.compose.material.TopAppBar
import androidx.compose.material.icons.Icons
import androidx.compose.material.icons.filled.Add
import androidx.compose.material.icons.filled.Edit
import androidx.compose.material3.Text
import androidx.compose.runtime.Composable
import androidx.compose.runtime.getValue
import androidx.compose.ui.Modifier
import androidx.navigation.compose.NavHost
import androidx.navigation.compose.composable
import androidx.navigation.compose.rememberNavController
import androidx.navigation.compose.currentBackStackEntryAsState

@Composable
fun MainNavigation() {
    val navController = rememberNavController()
    // 监听当前返回栈条目变化
    val currentBackStackEntry by navController.currentBackStackEntryAsState()
    // 通用转换为Screen实例
    val currentScreen = currentBackStackEntry?.toScreen() ?: Screen.MainScreen

    Scaffold(
        topBar = {
            when (currentScreen) {
                is Screen.MainScreen -> TopAppBar(title = { Text("首页") })
                is Screen.DetailScreen -> TopAppBar(title = { Text("详情页：${currentScreen.itemId}") })
            }
        },
        floatingActionButton = {
            when (currentScreen) {
                is Screen.MainScreen -> {
                    FloatingActionButton(onClick = { 
                        // 导航到详情页示例
                        navController.navigate(Screen.DetailScreen(itemId = 1)) 
                    }) {
                        Icon(Icons.Default.Add, contentDescription = "添加")
                    }
                }
                is Screen.DetailScreen -> {
                    FloatingActionButton(onClick = { /* 详情页操作逻辑 */ }) {
                        Icon(Icons.Default.Edit, contentDescription = "编辑")
                    }
                }
            }
        }
    ) { innerPadding ->
        NavHost(
            navController = navController,
            modifier = Modifier.padding(innerPadding),
            startDestination = Screen.MainScreen
        ) {
            composable<Screen.MainScreen> {
                // 主屏幕内容
            }
            composable<Screen.DetailScreen> {
                // 详情页内容
            }
        }
    }
}

注意事项 #

• 确保项目已引入kotlinx-serialization依赖，在模块的build.gradle.kts中添加：

  plugins {
      id("kotlinx-serialization")
  }
  
  dependencies {
      implementation("org.jetbrains.kotlinx:kotlinx-serialization-json:1.6.0")
      implementation("androidx.navigation:navigation-compose:2.7.7")
  }
  

• 异常处理部分可根据实际需求调整，比如添加日志或默认返回其他屏幕。

内容的提问来源于stack exchange，提问作者alof