From 88ed132be5d11157e539f6a9e751eba70b977321 Mon Sep 17 00:00:00 2001
From: Aamir Nazir <aamir.nu@gmail.com>
Date: Mon, 1 Aug 2016 23:31:18 +0500
Subject: [PATCH] SVWebViewController: Added custom protocol for event of
 webview dismissed

---
 SVWebViewController/SVModalWebViewController.m | 1 +
 SVWebViewController/SVWebViewController.h      | 6 ++++++
 SVWebViewController/SVWebViewController.m      | 3 +++
 3 files changed, 10 insertions(+)

diff --git a/SVWebViewController/SVModalWebViewController.m b/SVWebViewController/SVModalWebViewController.m
index 83f2337..ca21980 100644
--- a/SVWebViewController/SVModalWebViewController.m
+++ b/SVWebViewController/SVModalWebViewController.m
@@ -61,6 +61,7 @@ - (void)viewWillAppear:(BOOL)animated {
 
 - (void)setWebViewDelegate:(id<UIWebViewDelegate>)webViewDelegate {
     self.webViewController.delegate = webViewDelegate;
+    self.webViewController.webDelegate = webViewDelegate;
 }
 
 - (id<UIWebViewDelegate>)webViewDelegate {
diff --git a/SVWebViewController/SVWebViewController.h b/SVWebViewController/SVWebViewController.h
index b41ce2c..b260955 100644
--- a/SVWebViewController/SVWebViewController.h
+++ b/SVWebViewController/SVWebViewController.h
@@ -6,6 +6,11 @@
 //
 //  https://github.com/samvermette/SVWebViewController
 
+@protocol webDismissalDelegate <NSObject>
+@optional
+- (void) webviewDismissed;
+@end
+
 @interface SVWebViewController : UIViewController
 
 - (instancetype)initWithAddress:(NSString*)urlString;
@@ -13,5 +18,6 @@
 - (instancetype)initWithURLRequest:(NSURLRequest *)request;
 
 @property (nonatomic, weak) id<UIWebViewDelegate> delegate;
+@property (nonatomic, weak) id<webDismissalDelegate> webDelegate;
 
 @end
diff --git a/SVWebViewController/SVWebViewController.m b/SVWebViewController/SVWebViewController.m
index b9b2aea..22406ce 100644
--- a/SVWebViewController/SVWebViewController.m
+++ b/SVWebViewController/SVWebViewController.m
@@ -306,6 +306,9 @@ - (void)actionButtonTapped:(id)sender {
 
 - (void)doneButtonTapped:(id)sùender {
     [self dismissViewControllerAnimated:YES completion:NULL];
+
+    if ([self.webDelegate respondsToSelector:@selector(webviewDismissed)])
+        [self.webDelegate webviewDismissed];
 }
 
 @end