WebUI for WKWebView

ios/web/web_state/ui/crw_wk_web_view_web_controller.mm

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #import "ios/web/web_state/ui/crw_wk_web_view_web_controller.h"
 
 #import <WebKit/WebKit.h>
 
 #include "base/ios/weak_nsobject.h"
 #include "base/json/json_reader.h"
 #import "base/mac/scoped_nsobject.h"
 #include "base/macros.h"
 #include "base/strings/sys_string_conversions.h"
 #include "base/values.h"
 #import "ios/web/crw_network_activity_indicator_manager.h"
 #import "ios/web/navigation/crw_session_controller.h"
+#include "ios/web/navigation/web_load_params.h"
 #include "ios/web/public/web_client.h"
+#import "ios/web/public/web_state/crw_native_content_provider.h"
 #import "ios/web/public/web_state/js/crw_js_injection_manager.h"
 #import "ios/web/ui_web_view_util.h"
 #include "ios/web/web_state/blocked_popup_info.h"
 #import "ios/web/web_state/js/crw_js_window_id_manager.h"
 #import "ios/web/web_state/js/page_script_util.h"
 #import "ios/web/web_state/ui/crw_web_controller+protected.h"
 #import "ios/web/web_state/ui/crw_wk_web_view_crash_detector.h"
 #import "ios/web/web_state/ui/web_view_js_utils.h"
 #import "ios/web/web_state/ui/wk_web_view_configuration_provider.h"
 #import "ios/web/web_state/web_state_impl.h"
 #import "ios/web/web_state/web_view_creation_utils.h"
+#import "ios/web/webui/crw_web_ui_manager.h"
 #import "net/base/mac/url_conversions.h"
 
 #if !defined(ENABLE_CHROME_NET_STACK_FOR_WKWEBVIEW)
 #include "ios/web/public/cert_store.h"
 #include "ios/web/public/navigation_item.h"
 #include "ios/web/public/ssl_status.h"
 #import "ios/web/web_state/wk_web_view_ssl_error_util.h"
 #include "net/ssl/ssl_info.h"
 #endif
 
@@ skipping 43 matching lines @@
   base::scoped_nsobject<NSString> _currentReferrerString;
 
   // Backs the property of the same name.
   base::scoped_nsobject<NSString> _documentMIMEType;
 
   // Whether the web page is currently performing window.history.pushState or
   // window.history.replaceState
   // Set to YES on window.history.willChangeState message. To NO on
   // window.history.didPushState or window.history.didReplaceState.
   BOOL _changingHistoryState;
+
+  // CRWWebUIManager object for loading WebUI pages.
+  base::scoped_nsobject<CRWWebUIManager> web_ui_manager_;
 }
 
 // Response's MIME type of the last known navigation.
 @property(nonatomic, copy) NSString* documentMIMEType;
 
 // Dictionary where keys are the names of WKWebView properties and values are
 // selector names which should be called when a corresponding property has
 // changed. e.g. @{ @"URL" : @"webViewURLDidChange" } means that
 // -[self webViewURLDidChange] must be called every time when WKWebView.URL is
 // changed.
@@ skipping 232 matching lines @@
 
   if ([self.documentMIMEType isEqualToString:@"text/html"] ||
       [self.documentMIMEType isEqualToString:@"application/xhtml+xml"] ||
       [self.documentMIMEType isEqualToString:@"application/xml"]) {
     return web::WEB_VIEW_DOCUMENT_TYPE_HTML;
   }
 
   return web::WEB_VIEW_DOCUMENT_TYPE_GENERIC;
 }
 
-- (void)loadWebRequest:(NSURLRequest*)request {
+- (void)loadRequest:(NSMutableURLRequest*)request {
   [_wkWebView loadRequest:request];
 }
 
 - (void)loadWebHTMLString:(NSString*)html forURL:(const GURL&)URL {
   [_wkWebView loadHTMLString:html baseURL:net::NSURLWithGURL(URL)];
 }
 
 - (BOOL)scriptHasBeenInjectedForClass:(Class)jsInjectionManagerClass
                        presenceBeacon:(NSString*)beacon {
   return [_injectedScriptManagers containsObject:jsInjectionManagerClass];
@@ skipping 431 matching lines @@
 - (BOOL)handleWindowHistoryDidReplaceStateMessage:
     (base::DictionaryValue*)message
                                          context:(NSDictionary*)context {
   DCHECK(_changingHistoryState);
   _changingHistoryState = NO;
   return [super handleWindowHistoryDidReplaceStateMessage:message
                                                   context:context];
 }
 
 #pragma mark -
+#pragma mark WebUI
+
+- (void)createWebUIForURL:(const GURL&)URL {
+  [super createWebUIForURL:URL];
+  web_ui_manager_.reset(
+      [[CRWWebUIManager alloc] initWithWebState:self.webStateImpl]);
+}
+
+- (void)clearWebUI {
+  [super clearWebUI];
+  web_ui_manager_.reset();
+}
+
+#pragma mark -
 #pragma mark KVO Observation
 
 - (void)observeValueForKeyPath:(NSString*)keyPath
                       ofObject:(id)object
                         change:(NSDictionary*)change
                        context:(void*)context {
   NSString* dispatcherSelectorName = self.wkWebViewObservers[keyPath];
   DCHECK(dispatcherSelectorName);
   if (dispatcherSelectorName)
     [self performSelector:NSSelectorFromString(dispatcherSelectorName)];
@@ skipping 126 matching lines @@
 }
 
 // TODO(stuartmorgan): Move all the guesswork around these states out of the
 // superclass, and wire these up to the remaining methods.
 - (void)webView:(WKWebView *)webView
     didStartProvisionalNavigation:(WKNavigation *)navigation {
   GURL webViewURL = net::GURLWithNSURL(webView.URL);
   // If this navigation has not yet been registered, do so. loadPhase check is
   // necessary because lastRegisteredRequestURL may be the same as the
   // webViewURL on a new tab created by window.open (default is about::blank).
+  // TODO(jyquinn): Audit [CRWWebController loadCurrentURL] for other tasks that
+  // should be performed here.
   if (self.lastRegisteredRequestURL != webViewURL ||
       self.loadPhase != web::LOAD_REQUESTED) {
-    [self registerLoadRequest:webViewURL];
+    // Reset current WebUI if one exists.
+    [self clearWebUI];
+    // If webViewURL is a chrome URL, abort the current load and initialize the
+    // load from the web controller.
+    if (web::GetWebClient()->IsAppSpecificURL(webViewURL)) {
+      [self abortWebLoad];
+      web::WebLoadParams params(webViewURL);
+      [self loadWithParams:params];
+      return;
+    } else {
+      [self registerLoadRequest:webViewURL];
+    }
   }
   // Ensure the URL is registered and loadPhase is as expected.
   DCHECK(self.lastRegisteredRequestURL == webViewURL);
   DCHECK(self.loadPhase == web::LOAD_REQUESTED);
 }
 
 - (void)webView:(WKWebView *)webView
     didReceiveServerRedirectForProvisionalNavigation:
         (WKNavigation *)navigation {
   [self registerLoadRequest:net::GURLWithNSURL(webView.URL)
@@ skipping 146 matching lines @@
                               placeholderText:defaultText
                                    requestURL:
             net::GURLWithNSURL(frame.request.URL)
                             completionHandler:completionHandler];
   } else if (completionHandler) {
     completionHandler(nil);
   }
 }
 
 @end