WebUI for WKWebView

ios/web/web_state/ui/crw_web_controller.mm

 // Copyright 2012 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_web_controller.h"
 
 #import <objc/runtime.h>
 #include <cmath>
 
 #include "base/ios/block_types.h"
@@ skipping 345 matching lines @@
 // entry).
 - (CRWSessionEntry*)currentSessionEntry;
 - (web::NavigationItem*)currentNavItem;
 // Returns the referrer for currentURL as a string. May return nil.
 - (web::Referrer)currentSessionEntryReferrer;
 // The data and HTTP headers associated to the current entry. These are nil
 // unless the request was a POST.
 - (NSData*)currentPOSTData;
 - (NSDictionary*)currentHttpHeaders;
 
-// Have the current web view load the request. If needed, it will add
-// information to let the network stack access the requestGroupID.
-- (void)loadRequest:(NSMutableURLRequest*)request;
 // Finds all the scrollviews in the view hierarchy and makes sure they do not
 // interfere with scroll to top when tapping the statusbar.
 - (void)optOutScrollsToTopForSubviews;
 // Tears down the old native controller, and then replaces it with the new one.
 - (void)setNativeController:(id<CRWNativeContent>)nativeController;
 // Returns whether |url| should be opened.
 - (BOOL)shouldOpenURL:(const GURL&)url
       mainDocumentURL:(const GURL&)mainDocumentURL
           linkClicked:(BOOL)linkClicked;
 // Called when |url| needs to be opened in a matching native app.
@@ skipping 793 matching lines @@
   return _webViewProxy.get();
 }
 
 - (UIView*)viewForPrinting {
   // TODO(ios): crbug.com/227944. Printing is not supported for native
   // controllers.
   return self.webView;
 }
 
 - (void)loadRequest:(NSMutableURLRequest*)request {
-  DCHECK(web::GetWebClient());
-  GURL url = net::GURLWithNSURL(request.URL);
-  // TODO(stuartmorgan): See if WKWebView has the same issue; if not, this can
-  // move into the subclass.
-  if (web::GetWebClient()->IsAppSpecificURL(url)) {
-    // Sub requests of a chrome:// page will not contain the user agent.
-    // Instead use the username part of the URL to allow the network stack to
-    // associate a request to the correct tab.
-    request.URL = web::AddRequestGroupIDToURL(
-        request.URL, _webStateImpl->GetRequestGroupID());
-  }
-  [self loadWebRequest:request];
+  // Subclasses must implement this method.
+  NOTREACHED();
 }
 
 - (void)registerLoadRequest:(const GURL&)requestURL
                    referrer:(const web::Referrer&)referrer
                  transition:(ui::PageTransition)transition {
   // Transfer time is registered so that further transitions within the time
   // envelope are not also registered as links.
   _lastTransferTimeInSeconds = CFAbsoluteTimeGetCurrent();
   // Before changing phases, the delegate should be informed that any existing
   // request is being cancelled before completion.
@@ skipping 293 matching lines @@
   _webStateImpl->SetCacheMode(net::RequestTracker::CACHE_NORMAL);
 }
 
 - (void)loadCurrentURL {
   // If the content view doesn't exist, the tab has either been evicted, or
   // never displayed. Bail, and let the URL be loaded when the tab is shown.
   if (!_containerView)
     return;
 
   // Reset current WebUI if one exists.
-  _webStateImpl->ClearWebUI();
+  [self clearWebUI];
 
   // Precaution, so that the outgoing URL is registered, to reduce the risk of
   // it being seen as a fresh URL later by the same method (and new page change
   // erroneously reported).
   [self checkForUnexpectedURLChange];
 
   // Abort any outstanding page load. This ensures the delegate gets informed
   // about the outgoing page, and further messages from the page are suppressed.
   if (_loadPhase != web::PAGE_LOADED)
     [self abortLoad];
 
   DCHECK(!_isHalted);
   // Remove the interstitial before doing anything else.
   [self clearInterstitials];
 
-  const GURL currentUrl = [self currentNavigationURL];
+  const GURL currentURL = [self currentNavigationURL];
   // If it's a chrome URL, but not a native one, create the WebUI instance.
-  if (web::GetWebClient()->IsAppSpecificURL(currentUrl) &&
-      ![_nativeProvider hasControllerForURL:currentUrl]) {
-    _webStateImpl->CreateWebUI(currentUrl);
+  if (web::GetWebClient()->IsAppSpecificURL(currentURL) &&
+      ![_nativeProvider hasControllerForURL:currentURL]) {
+    [self createWebUIForURL:currentURL];
   }
 
   // Loading a new url, must check here if it's a native chrome URL and
   // replace the appropriate view if so, or transition back to a web view from
   // a native view.
-  if ([self shouldLoadURLInNativeView:currentUrl]) {
+  if ([self shouldLoadURLInNativeView:currentURL]) {
     [self loadCurrentURLInNativeView];
   } else {
     [self loadCurrentURLInWebView];
   }
 
   // Once a URL has been loaded, any cached-based reconstruction state has
   // either been handled or obsoleted.
   _expectedReconstructionURL = GURL();
 }
 
@@ skipping 514 matching lines @@
       std::string messageContent;
       base::ListValue* arguments = nullptr;
       if (!message->GetString("message", &messageContent)) {
         DLOG(WARNING) << "JS message parameter not found: message";
         return NO;
       }
       if (!message->GetList("arguments", &arguments)) {
         DLOG(WARNING) << "JS message parameter not found: arguments";
         return NO;
       }
+      _webStateImpl->OnScriptCommandReceived(
+          messageContent, *message, currentURL,
+          context[web::kUserIsInteractingKey]);
       _webStateImpl->ProcessWebUIMessage(currentURL, messageContent,
                                          *arguments);
+      return YES;
     }
   }
-  return YES;
+
+  DLOG(WARNING)
+      << "chrome.send message not handled because WebUI was not found.";
+  return NO;
 }
 
 - (BOOL)handleConsoleMessage:(base::DictionaryValue*)message
                      context:(NSDictionary*)context {
   // Do not log if JS logging is off.
   if (![[NSUserDefaults standardUserDefaults] boolForKey:web::kLogJavaScript]) {
     return YES;
   }
 
   std::string method;
@@ skipping 808 matching lines @@
       }
     }
     return;
   }
 
   [self loadCompleteWithSuccess:NO];
   [self loadErrorInNativeView:error];
 }
 
 #pragma mark -
+#pragma mark WebUI
+
+- (void)createWebUIForURL:(const GURL&)URL {
+  _webStateImpl->CreateWebUI(URL);
+}
+
+- (void)clearWebUI {
+  _webStateImpl->ClearWebUI();
+}
+
+#pragma mark -
 #pragma mark UIGestureRecognizerDelegate
 
 - (BOOL)gestureRecognizer:(UIGestureRecognizer*)gestureRecognizer
     shouldRecognizeSimultaneouslyWithGestureRecognizer:
         (UIGestureRecognizer*)otherGestureRecognizer {
   // Allows the custom UILongPressGestureRecognizer to fire simultaneously with
   // other recognizers.
   return YES;
 }
 
@@ skipping 654 matching lines @@
 - (void)loadHTML:(NSString*)html forURL:(const GURL&)url {
   // Remove the interstitial before doing anything else.
   [self clearInterstitials];
 
   DLOG_IF(WARNING, !self.webView)
       << "self.webView null while trying to load HTML";
   _loadPhase = web::LOAD_REQUESTED;
   [self loadWebHTMLString:html forURL:url];
 }
 
+- (void)loadHTML:(NSString*)HTML forAppSpecificURL:(const GURL&)URL {
+  CHECK(web::GetWebClient()->IsAppSpecificURL(URL));
+  [self loadHTML:HTML forURL:URL];
+}
+
 - (void)stopLoading {
   web::RecordAction(UserMetricsAction("Stop"));
   // Discard the pending and transient entried before notifying the tab model
   // observers of the change via |-abortLoad|.
   [[self sessionController] discardNonCommittedEntries];
   [self abortLoad];
   // If discarding the non-committed entries results in an app-specific URL,
   // reload it in its native view.
   if (!_nativeController &&
       [self shouldLoadURLInNativeView:[self currentNavigationURL]]) {
@@ skipping 98 matching lines @@
   if (!_externalRequest || !_externalRequest->window_name)
     return @"";
   return _externalRequest->window_name;
 }
 
 - (void)resetExternalRequest {
   _externalRequest.reset();
 }
 
 @end