基于Jetpack Compose与Firebase的多用户权限系统实现求助

你的当前实现可以从以下几个方面优化，提升健壮性和实时性：

1. 使用 Firebase Auth 内置 Flow 替代手动监听 #

Firebase Auth KTX 提供了 authStateFlow 扩展，无需手动管理 AuthStateListener，代码更简洁且符合 Kotlin Flow 最佳实践。

2. 改用实时快照监听替代单次获取 #

原代码用 get() 只能获取一次角色信息，若用户角色在 Firestore 中更新，UI 无法实时同步。改用 snapshots() 可以监听文档的实时变化。

3. 完善错误处理与空值安全 #

原代码缺少 Firestore 请求失败的处理逻辑，且对文档存在性和字段空值的判断不够严谨。

优化后的代码： #

override fun getUserRole(viewModelScope: CoroutineScope): StateFlow<Boolean> = 
    FirebaseAuth.getInstance().authStateFlow
        .flatMapLatest { user ->
            if (user == null) {
                // 用户未登录，直接返回非管理员状态
                flowOf(false)
            } else {
                usersRef.document(user.uid)
                    .snapshots() // 监听文档实时变化
                    .map { document ->
                        // 安全判断文档存在且角色为 ADMIN
                        document.exists() && document.getString("role") == "ADMIN"
                    }
                    .catch {
                        // 处理 Firestore 请求失败的情况，默认返回非管理员
                        emit(false)
                    }
            }
        }
        .stateIn(
            scope = viewModelScope,
            // 订阅者取消后保持 5 秒活跃，避免频繁重建
            started = SharingStarted.WhileSubscribed(5000),
            initialValue = false
        )

为避免在每个 ViewModel 中重复处理 Auth 状态和 Firestore 请求，建议创建一个全局的 用户数据仓库（UserRepository），统一管理用户状态和数据的获取。

1. 定义 UserData 数据类 #

data class UserData(
    val uid: String,
    val role: String, // "USER" 或 "ADMIN"
    // 添加其他用户字段，如昵称、邮箱等
)

2. 实现单例 UserRepository #

使用 Hilt 注入确保仓库全局唯一，统一处理 Auth 状态监听和 Firestore 数据获取：

class UserRepository(
    private val firebaseAuth: FirebaseAuth,
    private val firestore: FirebaseFirestore
) {
    private val usersRef = firestore.collection("users")

    // 暴露当前用户数据的 StateFlow，UI 可直接观察
    val currentUserData: StateFlow<UserData?> = firebaseAuth.authStateFlow
        .flatMapLatest { user ->
            if (user == null) {
                flowOf(null)
            } else {
                usersRef.document(user.uid)
                    .snapshots()
                    .map { document ->
                        if (document.exists()) {
                            // 将 Firestore 文档映射为 UserData，确保 uid 与 Auth 一致
                            document.toObject(UserData::class.java)?.copy(uid = user.uid)
                        } else {
                            // 用户在 Auth 存在但 Firestore 无记录，可根据业务逻辑处理（如引导完善资料）
                            null
                        }
                    }
                    .catch {
                        // 处理异常，返回 null
                        emit(null)
                    }
            }
        }
        .stateIn(
            scope = CoroutineScope(Dispatchers.IO + SupervisorJob()),
            started = SharingStarted.WhileSubscribed(5000),
            initialValue = null
        )

    // 辅助方法：快速判断当前用户是否为管理员
    fun isCurrentUserAdmin(): StateFlow<Boolean> = currentUserData
        .map { it?.role == "ADMIN" }
        .stateIn(
            scope = CoroutineScope(Dispatchers.IO + SupervisorJob()),
            started = SharingStarted.WhileSubscribed(5000),
            initialValue = false
        )
}

3. 配置 Hilt 注入 #

在 Firebase 模块中提供仓库实例，确保全局单例：

@Module
@InstallIn(SingletonComponent::class)
object FirebaseModule {
    @Provides
    fun provideFirebaseAuth(): FirebaseAuth = FirebaseAuth.getInstance()

    @Provides
    fun provideFirestore(): FirebaseFirestore = FirebaseFirestore.getInstance()

    @Provides
    fun provideUserRepository(
        auth: FirebaseAuth,
        firestore: FirebaseFirestore
    ): UserRepository = UserRepository(auth, firestore)
}

4. 在 ViewModel 和 UI 中使用 #

ViewModel 注入仓库： #

class HomeViewModel @Inject constructor(
    private val userRepository: UserRepository
) : ViewModel() {
    val currentUser = userRepository.currentUserData
    val isAdmin = userRepository.isCurrentUserAdmin()
}

Compose UI 中观察状态： #

@Composable
fun HomeScreen(viewModel: HomeViewModel = hiltViewModel()) {
    val currentUser by viewModel.currentUser.collectAsState()
    val isAdmin by viewModel.isAdmin.collectAsState()

    if (isAdmin) {
        // 渲染管理员专属界面
        AdminDashboard()
    } else {
        // 渲染普通用户界面
        UserDashboard()
    }
}

1. 状态封装：可以用密封类封装用户状态（加载中、成功、错误、未登录），让 UI 能更精准地处理不同场景：

sealed class UserState {
    object Loading : UserState()
    data class Success(val userData: UserData) : UserState()
    data class Error(val message: String) : UserState()
    object Unauthenticated : UserState()
}

2. Firestore 安全规则：确保只有用户本人能读取自己的文档，管理员可按需配置权限：

rules_version = '2';
service cloud.firestore {
  match /databases/{database}/documents {
    match /users/{userId} {
      allow read, write: if request.auth != null && request.auth.uid == userId;
      // 允许管理员修改用户角色（可选）
      allow write: if request.auth != null && get(/databases/$(database)/documents/users/$(request.auth.uid)).data.role == "ADMIN";
    }
  }
}

3. 缓存策略：若需要离线访问，可启用 Firestore 离线缓存，确保用户状态在离线时仍能正常显示。

内容的提问来源于stack exchange，提问作者Alex Diaz Llagostera