[Auth] Use result type to wrap completion handlers internally

FirebaseAuth/Sources/Swift/Auth/Auth.swift

@@ -300,9 +300,9 @@ extension Auth: AuthInterop {
       Task {
         do {
           let response = try await self.backend.call(with: request)
-          Auth.wrapMainAsync(callback: completion, withParam: response.signinMethods, error: nil)
+          Auth.wrapMainAsync(callback: completion, with: .success(response.signinMethods))
         } catch {
-          Auth.wrapMainAsync(callback: completion, withParam: nil, error: error)
+          Auth.wrapMainAsync(callback: completion, with: .failure(error))
         }
       }
     }
@@ -365,9 +365,9 @@ extension Auth: AuthInterop {
             withEmail: email,
             password: password
           )
-          decoratedCallback(authData, nil)
+          decoratedCallback(.success(authData))
         } catch {
-          decoratedCallback(nil, error)
+          decoratedCallback(.failure(error))
         }
       }
     }
@@ -463,9 +463,9 @@ extension Auth: AuthInterop {
         do {
           let authData = try await self.internalSignInAndRetrieveData(withCredential: credential,
                                                                       isReauthentication: false)
-          decoratedCallback(authData, nil)
+          decoratedCallback(.success(authData))
         } catch {
-          decoratedCallback(nil, error)
+          decoratedCallback(.failure(error))
         }
       }
     }
@@ -544,9 +544,9 @@ extension Auth: AuthInterop {
               withCredential: credential,
               isReauthentication: false
             )
-            decoratedCallback(authData, nil)
+            decoratedCallback(.success(authData))
           } catch {
-            decoratedCallback(nil, error)
+            decoratedCallback(.failure(error))
           }
         }
       }
@@ -642,9 +642,9 @@ extension Auth: AuthInterop {
         do {
           let authData = try await self.internalSignInAndRetrieveData(withCredential: credential,
                                                                       isReauthentication: false)
-          decoratedCallback(authData, nil)
+          decoratedCallback(.success(authData))
         } catch {
-          decoratedCallback(nil, error)
+          decoratedCallback(.failure(error))
         }
       }
     }
@@ -710,7 +710,7 @@ extension Auth: AuthInterop {
       let decoratedCallback = self.signInFlowAuthDataResultCallback(byDecorating: completion)
       if let currentUser = self._currentUser, currentUser.isAnonymous {
         let result = AuthDataResult(withUser: currentUser, additionalUserInfo: nil)
-        decoratedCallback(result, nil)
+        decoratedCallback(.success(result))
         return
       }
       let request = SignUpNewUserRequest(requestConfiguration: self.requestConfiguration)
@@ -728,10 +728,11 @@ extension Auth: AuthInterop {
                                                       profile: nil,
                                                       username: nil,
                                                       isNewUser: true)
-          decoratedCallback(AuthDataResult(withUser: user, additionalUserInfo: additionalUserInfo),
-                            nil)
+          decoratedCallback(
+            .success(AuthDataResult(withUser: user, additionalUserInfo: additionalUserInfo))
+          )
         } catch {
-          decoratedCallback(nil, error)
+          decoratedCallback(.failure(error))
         }
       }
     }
@@ -790,10 +791,11 @@ extension Auth: AuthInterop {
                                                       profile: nil,
                                                       username: nil,
                                                       isNewUser: response.isNewUser)
-          decoratedCallback(AuthDataResult(withUser: user, additionalUserInfo: additionalUserInfo),
-                            nil)
+          decoratedCallback(
+            .success(AuthDataResult(withUser: user, additionalUserInfo: additionalUserInfo))
+          )
         } catch {
-          decoratedCallback(nil, error)
+          decoratedCallback(.failure(error))
         }
       }
     }
@@ -866,7 +868,7 @@ extension Auth: AuthInterop {
                                  action: AuthRecaptchaAction.signUpPassword) { response, error in
           if let error {
             DispatchQueue.main.async {
-              decoratedCallback(nil, error)
+              decoratedCallback(.failure(error))
             }
             return
           }
@@ -881,7 +883,8 @@ extension Auth: AuthInterop {
 
   func internalCreateUserWithEmail(request: SignUpNewUserRequest,
                                    inResponse: SignUpNewUserResponse? = nil,
-                                   decoratedCallback: @escaping (AuthDataResult?, Error?) -> Void) {
+                                   decoratedCallback: @escaping (Result<AuthDataResult, Error>)
+                                     -> Void) {
     Task {
       do {
         var response: SignUpNewUserResponse
@@ -900,11 +903,11 @@ extension Auth: AuthInterop {
                                                     profile: nil,
                                                     username: nil,
                                                     isNewUser: true)
-        decoratedCallback(AuthDataResult(withUser: user,
-                                         additionalUserInfo: additionalUserInfo),
-                          nil)
+        decoratedCallback(
+          .success(AuthDataResult(withUser: user, additionalUserInfo: additionalUserInfo))
+        )
       } catch {
-        decoratedCallback(nil, error)
+        decoratedCallback(.failure(error))
       }
     }
   }
@@ -1009,9 +1012,9 @@ extension Auth: AuthInterop {
           let actionCodeInfo = ActionCodeInfo(withOperation: operation,
                                               email: email,
                                               newEmail: response.verifiedEmail)
-          Auth.wrapMainAsync(callback: completion, withParam: actionCodeInfo, error: nil)
+          Auth.wrapMainAsync(callback: completion, with: .success(actionCodeInfo))
         } catch {
-          Auth.wrapMainAsync(callback: completion, withParam: nil, error: error)
+          Auth.wrapMainAsync(callback: completion, with: .failure(error))
         }
       }
     }
@@ -2238,23 +2241,19 @@ extension Auth: AuthInterop {
   /// Invoked asynchronously on the main thread in the future.
   /// - Returns: Returns a block that updates the current user.
   func signInFlowAuthDataResultCallback(byDecorating callback:
-    ((AuthDataResult?, Error?) -> Void)?) -> (AuthDataResult?, Error?) -> Void {
-    let authDataCallback: (((AuthDataResult?, Error?) -> Void)?, AuthDataResult?, Error?) -> Void =
-      { callback, result, error in
-        Auth.wrapMainAsync(callback: callback, withParam: result, error: error)
-      }
-    return { authResult, error in
-      if let error {
-        authDataCallback(callback, nil, error)
-        return
-      }
-      do {
-        try self.updateCurrentUser(authResult?.user, byForce: false, savingToDisk: true)
-      } catch {
-        authDataCallback(callback, nil, error)
-        return
+    ((AuthDataResult?, Error?) -> Void)?) -> (Result<AuthDataResult, Error>) -> Void {
+    return { result in
+      switch result {
+      case let .success(authResult):
+        do {
+          try self.updateCurrentUser(authResult.user, byForce: false, savingToDisk: true)
+          Auth.wrapMainAsync(callback: callback, with: .success(authResult))
+        } catch {
+          Auth.wrapMainAsync(callback: callback, with: .failure(error))
+        }
+      case let .failure(error):
+        Auth.wrapMainAsync(callback: callback, with: .failure(error))
       }
-      authDataCallback(callback, authResult, nil)
     }
   }
 
@@ -2278,11 +2277,14 @@ extension Auth: AuthInterop {
   }
 
   class func wrapMainAsync<T: Any>(callback: ((T?, Error?) -> Void)?,
-                                   withParam param: T?,
-                                   error: Error?) -> Void {
-    if let callback {
-      DispatchQueue.main.async {
-        callback(param, error)
+                                   with result: Result<T, Error>) -> Void {
+    guard let callback else { return }
+    DispatchQueue.main.async {
+      switch result {
+      case let .success(success):
+        callback(success, nil)
+      case let .failure(error):
+        callback(nil, error)
       }
     }
   }

FirebaseAuth/Sources/Swift/AuthProvider/PhoneAuthProvider.swift

@@ -86,9 +86,9 @@ import Foundation
               uiDelegate: uiDelegate,
               multiFactorSession: multiFactorSession
             )
-            Auth.wrapMainAsync(callback: completion, withParam: verificationID, error: nil)
+            Auth.wrapMainAsync(callback: completion, with: .success(verificationID))
           } catch {
-            Auth.wrapMainAsync(callback: completion, withParam: nil, error: error)
+            Auth.wrapMainAsync(callback: completion, with: .failure(error))
           }
         }
       }
@@ -184,7 +184,7 @@ import Foundation
     private func internalVerify(phoneNumber: String,
                                 uiDelegate: AuthUIDelegate?,
                                 multiFactorSession: MultiFactorSession? = nil) async throws
-      -> String? {
+      -> String {
       guard phoneNumber.count > 0 else {
         throw AuthErrorUtils.missingPhoneNumberError(message: nil)
       }
@@ -209,7 +209,7 @@ import Foundation
     private func verifyClAndSendVerificationCode(toPhoneNumber phoneNumber: String,
                                                  retryOnInvalidAppCredential: Bool,
                                                  uiDelegate: AuthUIDelegate?) async throws
-      -> String? {
+      -> String {
       let codeIdentity = try await verifyClient(withUIDelegate: uiDelegate)
       let request = SendVerificationCodeRequest(phoneNumber: phoneNumber,
                                                 codeIdentity: codeIdentity,
@@ -236,7 +236,7 @@ import Foundation
                                                  retryOnInvalidAppCredential: Bool,
                                                  multiFactorSession session: MultiFactorSession?,
                                                  uiDelegate: AuthUIDelegate?) async throws
-      -> String? {
+      -> String {
       if let settings = auth.settings,
          settings.isAppVerificationDisabledForTesting {
         let request = SendVerificationCodeRequest(
@@ -264,15 +264,20 @@ import Foundation
                                                   enrollmentInfo: startMFARequestInfo,
                                                   requestConfiguration: auth.requestConfiguration)
           let response = try await auth.backend.call(with: request)
-          return response.phoneSessionInfo?.sessionInfo
+          guard let sessionInfo = response.phoneSessionInfo?.sessionInfo else {
+            throw AuthErrorUtils.error(code: .missingMultiFactorInfo)
+          }
+          return sessionInfo
         } else {
           let request = StartMFASignInRequest(MFAPendingCredential: session.mfaPendingCredential,
                                               MFAEnrollmentID: session.multiFactorInfo?.uid,
                                               signInInfo: startMFARequestInfo,
                                               requestConfiguration: auth.requestConfiguration)
-
           let response = try await auth.backend.call(with: request)
-          return response.responseInfo?.sessionInfo
+          guard let sessionInfo = response.responseInfo.sessionInfo else {
+            throw AuthErrorUtils.error(code: .multiFactorInfoNotFound)
+          }
+          return sessionInfo
         }
       } catch {
         return try await handleVerifyErrorWithRetry(
@@ -289,7 +294,7 @@ import Foundation
                                             phoneNumber: String,
                                             retryOnInvalidAppCredential: Bool,
                                             multiFactorSession session: MultiFactorSession?,
-                                            uiDelegate: AuthUIDelegate?) async throws -> String? {
+                                            uiDelegate: AuthUIDelegate?) async throws -> String {
       if (error as NSError).code == AuthErrorCode.invalidAppCredential.rawValue {
         if retryOnInvalidAppCredential {
           auth.appCredentialManager.clearCredential()

FirebaseAuth/Sources/Swift/Backend/RPC/MultiFactor/SignIn/StartMFASignInResponse.swift

@@ -16,7 +16,7 @@ import Foundation
 
 @available(iOS 13, tvOS 13, macOS 10.15, macCatalyst 13, watchOS 7, *)
 struct StartMFASignInResponse: AuthRPCResponse {
-  var responseInfo: AuthProtoStartMFAPhoneResponseInfo?
+  let responseInfo: AuthProtoStartMFAPhoneResponseInfo
 
   init(dictionary: [String: AnyHashable]) throws {
     if let data = dictionary["phoneResponseInfo"] as? [String: AnyHashable] {

FirebaseAuth/Sources/Swift/Backend/RPC/SendVerificationTokenResponse.swift

@@ -15,9 +15,12 @@
 import Foundation
 
 struct SendVerificationCodeResponse: AuthRPCResponse {
-  var verificationID: String?
+  let verificationID: String
 
   init(dictionary: [String: AnyHashable]) throws {
-    verificationID = dictionary["sessionInfo"] as? String
+    guard let verificationID = dictionary["sessionInfo"] as? String else {
+      throw AuthErrorUtils.unexpectedResponse(deserializedResponse: dictionary)
+    }
+    self.verificationID = verificationID
   }
 }

FirebaseAuth/Sources/Swift/MultiFactor/MultiFactorResolver.swift

@@ -80,7 +80,7 @@ import Foundation
           let result = AuthDataResult(withUser: user, additionalUserInfo: nil)
           let decoratedCallback = self.auth
             .signInFlowAuthDataResultCallback(byDecorating: completion)
-          decoratedCallback(result, nil)
+          decoratedCallback(.success(result))
         } catch {
           if let completion {
             completion(nil, error)

FirebaseAuth/Tests/Unit/RPCBaseTests.swift

@@ -98,7 +98,7 @@ class RPCBaseTests: XCTestCase {
         XCTFail("decodedRequest is not a dictionary")
       }
       // Dummy response to unblock await.
-      let _ = try self.rpcIssuer?.respond(withJSON: [:])
+      let _ = try self.rpcIssuer?.respond(withJSON: ["sessionInfo": "verification_id_123"])
     }
     let _ = try await authBackend.call(with: request)
   }