iOS Firebase：如何确保用户注册后完善信息才可访问应用功能？

这个问题我之前做iOS Firebase项目时也碰到过，核心矛盾在于Firebase Auth的状态只判断用户是否完成登录，但我们的业务要求用户必须补充完姓名等信息才能进入功能页面。下面是我实际用过的解决方案，分步骤拆解：

1. 扩展认证状态枚举 #

首先，我们需要给AuthStatus增加一个中间状态，用来区分「已登录但未完善资料」的情况：

enum AuthStatus {
    case signedOut
    case signedIn
    case needsProfileSetup // 新增状态：已登录但需要补充资料
}

2. 修改监听器，加入用户Profile检查 #

原来的监听器只判断Auth状态，现在要在用户登录后，去Cloud Firestore查询该用户的Profile文档是否存在（或者是否包含必填字段）。这里假设我们用用户的UID作为Firestore文档ID：

func initialAuthHandleObserver(completionHandler: @escaping (AuthStatus) -> Void) {
    Auth.auth().addStateDidChangeListener { (auth, user) in
        guard let currentUser = user else {
            completionHandler(.signedOut)
            return
        }
        
        // 用户已登录，检查Firestore中的Profile
        let userDocRef = Firestore.firestore().collection("users").document(currentUser.uid)
        userDocRef.getDocument { snapshot, error in
            if let error = error {
                // 处理查询错误，比如网络问题，这里可以默认引导到资料完善页，或者重试
                print("Failed to fetch user profile: \(error.localizedDescription)")
                completionHandler(.needsProfileSetup)
                return
            }
            
            // 判断Profile是否存在且包含必填字段（比如name）
            if let snapshot = snapshot, snapshot.exists, let _ = snapshot.data()?["name"] as? String {
                completionHandler(.signedIn)
            } else {
                completionHandler(.needsProfileSetup)
            }
        }
    }
}

3. 调整页面跳转逻辑 #

在应用启动时，根据返回的AuthStatus来跳转对应页面：

// 示例：在AppDelegate或SceneDelegate中处理
initialAuthHandleObserver { status in
    DispatchQueue.main.async {
        switch status {
        case .signedOut:
            self.window?.rootViewController = UINavigationController(rootViewController: LoginViewController())
        case .needsProfileSetup:
            self.window?.rootViewController = UINavigationController(rootViewController: ProfileSetupViewController())
        case .signedIn:
            self.window?.rootViewController = UINavigationController(rootViewController: HomeViewController())
        }
    }
}

4. 处理资料完善后的状态同步 #

当用户在ProfileSetupViewController中提交姓名等信息后，要确保成功写入Firestore，之后可以直接跳转首页，或者触发一次状态检查：

// ProfileSetupViewController中提交资料的示例
func submitProfile(name: String) {
    guard let currentUser = Auth.auth().currentUser else { return }
    let userDocRef = Firestore.firestore().collection("users").document(currentUser.uid)
    
    userDocRef.setData(["name": name, "createdAt": Timestamp()]) { error in
        if let error = error {
            print("Failed to save profile: \(error.localizedDescription)")
            // 提示用户重试
        } else {
            // 资料保存成功，跳转到首页
            DispatchQueue.main.async {
                self.navigationController?.pushViewController(HomeViewController(), animated: true)
            }
        }
    }
}

边缘情况处理 #

• 用户在完善资料时退出应用：下次启动时，监听器会再次检查Firestore，发现Profile不存在/不完整，自动引导到资料完善页。
• Firestore查询失败：比如网络问题，我们可以默认引导到资料完善页，或者在页面中加入重试逻辑，避免用户卡在登录状态。

这样就能确保用户必须完成资料补充后，才能进入应用功能页面，完美解决你提到的问题。

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