expose the request so it can be modified

uzegonemad · May 23, 2015 · 39c26a5 · 39c26a5
1 parent 09b72bd
commit 39c26a5
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 4 deletions.
diff --git a/TOWebViewController/TOWebViewController.h b/TOWebViewController/TOWebViewController.h
@@ -51,6 +51,11 @@
  */
 @property (nonatomic,strong)    NSURL *url;
 
+/** 
+ Get/set the request
+ */
+@property (nonatomic,strong)    NSMutableURLRequest *urlRequest;
+
 /**
  The web view used to display the HTML content. You can access it through this
  read-only property if you need to anything specific, such as having it execute arbitrary JS code.

diff --git a/TOWebViewController/TOWebViewController.m b/TOWebViewController/TOWebViewController.m
@@ -258,6 +258,9 @@ - (void)setup
 
     //Set the initial default style as full screen (But this can be easily overridden)
     self.modalPresentationStyle = UIModalPresentationFullScreen;
+
+    //Set the URL request
+    self.urlRequest = [[NSMutableURLRequest alloc] init];
 }
 
 - (void)loadView
@@ -491,7 +494,10 @@ - (void)viewWillAppear:(BOOL)animated
 
     //start loading the initial page
     if (self.url && self.webView.request == nil)
-        [self.webView loadRequest:[NSURLRequest requestWithURL:self.url]];
+    {
+        [self.urlRequest setURL:self.url];
+        [self.webView loadRequest:self.urlRequest];
+    }
 }
 
 - (void)viewWillDisappear:(BOOL)animated
@@ -596,7 +602,8 @@ - (void)setUrl:(NSURL *)url
     if (self.webView.loading)
         [self.webView stopLoading];
 
-    [self.webView loadRequest:[NSURLRequest requestWithURL:self.url]];
+    [self.urlRequest setURL:self.url];
+    [self.webView loadRequest:self.urlRequest];
 }
 
 - (void)setLoadingBarTintColor:(UIColor *)loadingBarTintColor
@@ -803,8 +810,7 @@ - (void)reloadStopButtonTapped:(id)sender
         //off our stored self.url property instead
         if (self.webView.request.URL.absoluteString.length == 0 && self.url)
         {
-            NSURLRequest *request = [NSURLRequest requestWithURL:self.url];
-            [self.webView loadRequest:request];
+            [self.webView loadRequest:self.urlRequest];
         }
         else {
             [self.webView reload];