add SignOut if user canceled account linking (clean provider token)

firebase · Oct 20, 2016 · 7883877 · 7883877
1 parent 9d17f34
commit 7883877
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/FirebaseAuthUI/FIRAuthUIBaseViewController.m b/FirebaseAuthUI/FIRAuthUIBaseViewController.m
@@ -230,7 +230,9 @@ - (void)showSignInAlertWithEmail:(NSString *)email
     UIAlertAction *cancelAction =
         [UIAlertAction actionWithTitle:[FIRAuthUIStrings cancel]
                                  style:UIAlertActionStyleCancel
-                               handler:nil];
+                               handler:^(UIAlertAction * _Nonnull action) {
+                                 [self.authUI signOutWithError:nil];
+                               }];
     [alertController addAction:cancelAction];
     [self presentViewController:alertController animated:YES completion:nil];
   } else {
@@ -241,7 +243,9 @@ - (void)showSignInAlertWithEmail:(NSString *)email
                          cancelButtonTitle:[FIRAuthUIStrings cancel]
                          otherButtonTitles:provider.signInLabel, nil];
     FIRAuthUIAlertViewDelegate *delegate =
-        [[FIRAuthUIAlertViewDelegate alloc] initWithCancelHandler:nil otherHandlers:@[ handler ]];
+        [[FIRAuthUIAlertViewDelegate alloc] initWithCancelHandler:^{
+          [self.authUI signOutWithError:nil];
+        }  otherHandlers:@[ handler ]];
     alertView.delegate = delegate;
     [alertView show];
   }