[auth-swift] Fix two flow porting errors in AuthURLPresenter.swift (#…

…12981)
firebase · May 20, 2024 · a5af29d · a5af29d
1 parent 4a0f631
commit a5af29d
Showing 1 changed file with 15 additions and 2 deletions.
diff --git a/FirebaseAuth/Sources/Swift/Utilities/AuthURLPresenter.swift b/FirebaseAuth/Sources/Swift/Utilities/AuthURLPresenter.swift
@@ -84,6 +84,19 @@
       return false
     }
 
+    // MARK: SFSafariViewControllerDelegate
+
+    func safariViewControllerDidFinish(_ controller: SFSafariViewController) {
+      kAuthGlobalWorkQueue.async {
+        if controller == self.safariViewController {
+          // TODO: Ensure that the SFSafariViewController is actually removed from the screen
+          // before invoking finishPresentation
+          self.finishPresentation(withURL: nil,
+                                  error: AuthErrorUtils.webContextCancelledError(message: nil))
+        }
+      }
+    }
+
     // MARK: AuthWebViewControllerDelegate
 
     func webViewControllerDidCancel(_ controller: AuthWebViewController) {
@@ -156,8 +169,8 @@
       let webViewController = self.webViewController
       self.webViewController = nil
       if safariViewController != nil || webViewController != nil {
-        uiDelegate?.dismiss(animated: true) {
-          kAuthGlobalWorkQueue.async {
+        DispatchQueue.main.async {
+          uiDelegate?.dismiss(animated: true) {
             self.isPresenting = false
             if let completion {
               completion(url, error)