如何在Firebase authStateChanges中动态关联学校ID？

针对你遇到的authStateChanges()无法获取学校ID的问题，以下是三种可行的解决方案，适配不同场景需求：

方案1：将学校ID存入Firebase用户自定义Claims #

通过Firebase Auth的自定义Claims，把学校ID绑定到用户身份上，这样authStateChanges()返回的User对象就能直接携带学校ID信息。

修改AuthService代码 #

1. 更新谷歌登录方法，设置自定义Claims

Future<dynamic> signInWithGoogle(String schoolID) async {
  try {
    final GoogleSignInAccount? account = await _googleSignIn.signIn();
    final GoogleSignInAuthentication auth = await account!.authentication;
    final AuthCredential authCredential = GoogleAuthProvider.credential(
        idToken: auth.idToken, accessToken: auth.accessToken);
    UserCredential result = await _auth.signInWithCredential(authCredential);
    User? firebaseUser = result.user;
    
    // 将学校ID写入用户自定义Claims
    if (firebaseUser != null) {
      await _auth.setCustomUserClaims(firebaseUser.uid, {'schoolUID': schoolID});
      // 重新加载用户信息，确保Claims立即生效
      await firebaseUser.reload();
      firebaseUser = _auth.currentUser;
    }

    CurrentUser? currentUser = _userFromFirebase(firebaseUser, schoolID);

    bool userExists = await DatabaseService(user: currentUser)
        .checkIfUserExists(currentUser!.uid);
    if (!userExists) {
      DatabaseService(user: currentUser)
          .userCreationData(firebaseUser, schoolID);
    }
    return currentUser;
  } on FirebaseAuthException catch (e) {
    return e.message;
  }
}

2. 修改用户流生成逻辑，从Claims提取学校ID

CurrentUser? _userFromFirebase(User? user) {
  if (user == null) return null;
  
  // 从自定义Claims中读取学校ID
  String? schoolUID = user.claims?['schoolUID'] as String?;
  // 处理学校ID为空的情况（比如用户首次登录未设置）
  if (schoolUID == null) return null;
  
  return CurrentUser(uid: user.uid, schoolUID: schoolUID);
}

Stream<CurrentUser?> get user {
  return _auth
      .authStateChanges()
      .map((User? user) => _userFromFirebase(user));
}

方案2：本地存储+流合并（SharedPreferences） #

登录时将学校ID保存在本地，通过流合并的方式，把authStateChanges()的用户流和本地存储的学校ID流合并，生成包含完整信息的CurrentUser流。

步骤1：添加依赖 #

在pubspec.yaml中添加：

dependencies:
  shared_preferences: ^2.2.2
  rxdart: ^0.27.7

步骤2：修改AuthService代码 #

import 'package:shared_preferences/shared_preferences.dart';
import 'package:rxdart/rxdart.dart';

class AuthService {
  final FirebaseAuth _auth = FirebaseAuth.instance;
  final GoogleSignIn _googleSignIn = GoogleSignIn();
  final _schoolIdController = BehaviorSubject<String?>();

  AuthService() {
    // 初始化时加载本地保存的学校ID
    _loadSavedSchoolId();
  }

  Future<void> _loadSavedSchoolId() async {
    SharedPreferences prefs = await SharedPreferences.getInstance();
    _schoolIdController.add(prefs.getString('schoolUID'));
  }

  Future<void> _saveSchoolId(String schoolId) async {
    SharedPreferences prefs = await SharedPreferences.getInstance();
    await prefs.setString('schoolUID', schoolId);
    _schoolIdController.add(schoolId);
  }

  CurrentUser? _userFromFirebase(User? user, String? schoolUID) {
    return user != null && schoolUID != null
        ? CurrentUser(uid: user.uid, schoolUID: schoolUID)
        : null;
  }

  Stream<CurrentUser?> get user {
    // 合并用户状态流与学校ID流
    return CombineLatestStream.combine2(
      _auth.authStateChanges(),
      _schoolIdController.stream,
      (User? user, String? schoolId) => _userFromFirebase(user, schoolId),
    );
  }

  Future<dynamic> signInWithGoogle(String schoolID) async {
    try {
      final GoogleSignInAccount? account = await _googleSignIn.signIn();
      final GoogleSignInAuthentication auth = await account!.authentication;
      final AuthCredential authCredential = GoogleAuthProvider.credential(
          idToken: auth.idToken, accessToken: auth.accessToken);
      UserCredential result = await _auth.signInWithCredential(authCredential);
      CurrentUser? currentUser = _userFromFirebase(result.user, schoolID);

      // 保存学校ID到本地
      await _saveSchoolId(schoolID);

      bool userExists = await DatabaseService(user: currentUser)
          .checkIfUserExists(currentUser!.uid);
      if (!userExists) {
        DatabaseService(user: currentUser)
            .userCreationData(result.user, schoolID);
      }
      return currentUser;
    } on FirebaseAuthException catch (e) {
      return e.message;
    }
  }

  // 登出时清除本地学校ID
  Future<void> signOut() async {
    await _auth.signOut();
    await _googleSignIn.signOut();
    SharedPreferences prefs = await SharedPreferences.getInstance();
    await prefs.remove('schoolUID');
    _schoolIdController.add(null);
  }
}

方案3：从Firestore数据库查询学校ID #

利用你已有的用户创建逻辑，登录时将学校ID存入Firestore的用户文档，当authStateChanges()触发时，通过用户UID查询数据库获取学校ID。

修改AuthService的用户流逻辑 #

Stream<CurrentUser?> get user {
  return _auth.authStateChanges().asyncMap((User? user) async {
    if (user == null) return null;
    
    // 从Firestore查询用户关联的学校ID
    DocumentSnapshot userDoc = await FirebaseFirestore.instance
        .collection('users')
        .doc(user.uid)
        .get();
    
    if (!userDoc.exists) return null;
    
    String schoolUID = userDoc['schoolUID'];
    return CurrentUser(uid: user.uid, schoolUID: schoolUID);
  });
}

前提：确保你的userCreationData方法已经将学校ID写入用户文档，例如：

Future<void> userCreationData(User? user, String schoolID) async {
  await FirebaseFirestore.instance
      .collection('users')
      .doc(user!.uid)
      .set({
        'uid': user.uid,
        'email': user.email,
        'schoolUID': schoolID,
        // 其他用户字段
      });
}

方案对比 #

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