Changes to fire a delegate method when the dialog is dismissed

oauth-io · Oct 22, 2015 · 6416187 · 6416187
1 parent e40b81b
commit 6416187
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/OAuthiOS/Src/OAuthIOModal.h b/OAuthiOS/Src/OAuthIOModal.h
@@ -28,6 +28,7 @@
 - (void)didReceiveOAuthIOCode:(NSString *)code;
 - (void)didAuthenticateServerSide:(NSString *)body andResponse:(NSURLResponse *) response;
 - (void)didFailAuthenticationServerSide:(NSString *)body andResponse:(NSURLResponse *)response andError:(NSError *)error;
+- (void)didDismissModalDialog;
 @end
 
 @interface OAuthIOModal : UIViewController<UIWebViewDelegate>

diff --git a/OAuthiOS/Src/OAuthIOModal.m b/OAuthiOS/Src/OAuthIOModal.m
@@ -396,6 +396,7 @@ - (void)showWithProvider:(NSString *)provider options:(NSDictionary*)options
         [[NSUserDefaults standardUserDefaults] synchronize];
     }
     [_rootViewController presentViewController:self animated:YES completion:^{
+
         [_browser loadRequest:url];
     }];
 }
@@ -442,7 +443,11 @@ - (BOOL)webView:(UIWebView *)webView shouldStartLoadWithRequest:(NSURLRequest *)
     {
         if ([url.scheme isEqual:@"oauthio"] && [url.host isEqual:@"localhost"])
         {
-            [self dismissViewControllerAnimated:YES completion:nil];
+            [self dismissViewControllerAnimated:YES completion:^(){
+                if([self.delegate respondsToSelector:@selector(didDismissModalDialog)]) {
+                    [self.delegate didDismissModalDialog];
+                }
+            }];
             [self getTokens:[url absoluteString]];
             return (NO);
         }