fix logout issue

lib/data/provider/auth_fb.dart

@@ -12,6 +12,8 @@ import 'package:fcs/helpers/firebase_helper.dart';
 import 'package:firebase_auth/firebase_auth.dart' as fb;
 import 'package:logging/logging.dart';
 
+import '../services/services.dart';
+
 class AuthFb {
   final log = Logger('AuthFb');
 
@@ -21,6 +23,10 @@ class AuthFb {
   late StreamController<User?> controller;
   static final fb.FirebaseAuth _fb = fb.FirebaseAuth.instance;
   static String _verificationId = '';
+  static bool _logIn = false;
+
+  StreamSubscription<DocumentSnapshot>? userListener;
+  StreamSubscription<DocumentSnapshot>? userAuthListener;
 
   Future<fcs.AuthResult> sendSmsCodeToPhoneNumber(String phoneNumber) {
     Completer<fcs.AuthResult> completer = Completer();
@@ -99,9 +105,13 @@ class AuthFb {
     return Future.value(fcs.AuthResult(authStatus: AuthStatus.AUTH_VERIFIED));
   }
 
-  Future<void> signout() async {
-    if (userListener != null) await userListener!.cancel();
-    return _fb.signOut();
+  Future<void> signoutStart() async {
+    await userListener?.cancel();
+    await userAuthListener?.cancel();
+  }
+
+  Future<void> signoutEnd() async {
+    await _fb.signOut();
   }
 
   Future<void> _addUserToStream({bool refreshIdToken = false}) async {
@@ -111,7 +121,6 @@ class AuthFb {
         await getClaims(refreshIdToken: refreshIdToken);
 
     log.info("Claims:$claims");
-    if (claims == null) return;
 
     String? cid = claims["cid"];
     User? user;
@@ -123,6 +132,11 @@ class AuthFb {
       return;
     }
 
+    loadUserClaim(claims, user);
+    controller.add(user);
+  }
+
+  loadUserClaim(Map claims, User user) {
     // add privileges
     String? privileges = claims["pr"];
     if (privileges != null && privileges != "") {
@@ -130,7 +144,6 @@ class AuthFb {
     } else {
       user.privileges = [];
     }
-    controller.add(user);
   }
 
   Future<User?> _getUserFromFirestore(String userID) async {
@@ -205,13 +218,12 @@ class AuthFb {
     fb.User? firebaseUser = _fb.currentUser;
     if (firebaseUser == null) return null;
     Map? claims = await getClaims();
-    if (claims == null) return null;
     String cid = claims["cid"];
     return cid;
   }
 
   Future<void> _startUserListener() async {
-    if (userListener != null) userListener!.cancel();
+    _startAuthListener();
     String? _userID = await _getCurrentUserID();
     if (_userID == null) {
       return;
@@ -221,6 +233,7 @@ class AuthFb {
         .collection(user_collection)
         .doc(_userID)
         .snapshots();
+    userListener?.cancel();
     userListener = snapshot.listen((snap) async {
       User user = User.fromMap(snap.data() as Map<String, dynamic>, snap.id);
 
@@ -229,14 +242,12 @@ class AuthFb {
         userListener?.cancel();
         return;
       }
+
+      if (!_logIn) return;
       try {
         // get privilege from claim
-        fb.IdTokenResult idToken = await firebaseUser.getIdTokenResult(true);
-
-        String? privileges = idToken.claims?["pr"] ?? '';
-        if (privileges != null && privileges != "") {
-          user.privileges = privileges.split(":").toList();
-        }
+        Map<dynamic, dynamic> claims = await getClaims(refreshIdToken: true);
+        loadUserClaim(claims, user);
         controller.add(user);
       } catch (e) {
         controller.add(null);
@@ -244,14 +255,47 @@ class AuthFb {
     });
   }
 
-  StreamSubscription<DocumentSnapshot>? userListener;
+  Future<void> _startAuthListener() async {
+    String? authId = _fb.currentUser?.uid;
+    if (authId == null) return;
+
+    Stream<DocumentSnapshot> snapshot = FirebaseFirestore.instance
+        .collection(authCollection)
+        .doc(authId)
+        .snapshots();
+    userAuthListener?.cancel();
+    userAuthListener = snapshot.listen((snap) async {
+      if (snap.exists) {
+        Map<String, dynamic> map = snap.data() as Map<String, dynamic>;
+        String userID = map['user_id'] ?? "";
+
+        User? user = await Services.instance.userService.getUser(userID);
+        if (user == null) return;
+
+        if (_fb.currentUser == null) {
+          userAuthListener?.cancel();
+          return;
+        }
+
+        Map<dynamic, dynamic> claims = await getClaims(refreshIdToken: true);
+        loadUserClaim(claims, user);
+
+        log.info("_startAuthListener: $user");
+        if (_logIn) {
+          controller.add(user);
+        }
+      }
+    });
+  }
+
   Stream<User?> user() {
     // ignore: close_sinks
     StreamSubscription<fb.User?>? authListener;
 
     Future<void> _start() async {
       await authListener?.cancel();
-      authListener = _fb.authStateChanges().listen((firebaseUser) {
+      authListener = _fb.authStateChanges().listen((firebaseUser) async {
+        _logIn = firebaseUser != null;
         if (firebaseUser == null) {
           controller.add(null);
         } else {