[ios Mojo] Supporting code for define WebUI function.

ios/web/webui/crw_web_ui_manager.mm

 // Copyright 2015 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/webui/crw_web_ui_manager.h"
 
+#include "base/json/string_escape.h"
 #include "base/mac/bind_objc_block.h"
 #include "base/mac/scoped_nsobject.h"
 #include "base/memory/scoped_vector.h"
+#include "base/strings/stringprintf.h"
 #import "base/strings/sys_string_conversions.h"
+#include "base/strings/utf_string_conversions.h"
 #include "base/values.h"
 #import "ios/web/net/request_group_util.h"
 #include "ios/web/public/browser_state.h"
 #include "ios/web/public/web_client.h"
 #import "ios/web/public/web_state/web_state_observer_bridge.h"
 #include "ios/web/web_state/web_state_impl.h"
 #import "ios/web/webui/crw_web_ui_page_builder.h"
 #include "ios/web/webui/url_fetcher_block_adapter.h"
 #import "net/base/mac/url_conversions.h"
 
@@ skipping 12 matching lines @@
 - (void)loadWebUIPageForURL:(const GURL&)webUIURL
           completionHandler:(void (^)(NSString*))completionHandler;
 
 // Retrieves resource for URL and invokes completionHandler with the result.
 - (void)fetchResourceWithURL:(const GURL&)URL
            completionHandler:(void (^)(NSData*))completionHandler;
 
 // Handles JavaScript message from the WebUI page.
 - (BOOL)handleWebUIJSMessage:(const base::DictionaryValue&)message;
 
+// Handles webui.requestFavicon JavaScript message from the WebUI page.
+- (BOOL)handleRequestFavicon:(const base::ListValue*)arguments;
+
+// Handles webui.loadMojo JavaScript message from the WebUI page.
+- (BOOL)handleLoadMojo:(const base::ListValue*)arguments;
+
 // Removes favicon callback from web state.
 - (void)resetWebState;
 
 // Removes fetcher from vector of active fetchers.
 - (void)removeFetcher:(web::URLFetcherBlockAdapter*)fetcher;
 
 @end
 
 @implementation CRWWebUIManager {
   // Set of live WebUI fetchers for retrieving data.
@@ skipping 101 matching lines @@
         [weakSelf removeFetcher:fetcher];
       };
 
   _fetchers.push_back(
       [self fetcherForURL:URL completionHandler:fetcherCompletion]);
   _fetchers.back()->Start();
 }
 
 - (BOOL)handleWebUIJSMessage:(const base::DictionaryValue&)message {
   std::string command;
-  if (!message.GetString("message", &command) ||
-      command != "webui.requestFavicon") {
-    DLOG(WARNING) << "Unexpected message received" << command;
+  if (!message.GetString("message", &command)) {
+    DLOG(WARNING) << "Malformed message received";
     return NO;
   }
   const base::ListValue* arguments = nullptr;
   if (!message.GetList("arguments", &arguments)) {
     DLOG(WARNING) << "JS message parameter not found: arguments";
     return NO;
   }
+
+  if (!arguments) {
+    DLOG(WARNING) << "No arguments provided to " << command;
+    return NO;
+  }
+
+  if (command == "webui.requestFavicon")
+    return [self handleRequestFavicon:arguments];
+  if (command == "webui.loadMojo")
+    return [self handleLoadMojo:arguments];
+
+  DLOG(WARNING) << "Unknown webui command received: " << command;
+  return NO;
+}
+
+- (BOOL)handleRequestFavicon:(const base::ListValue*)arguments {
   std::string favicon;
   if (!arguments->GetString(0, &favicon)) {
     DLOG(WARNING) << "JS message parameter not found: Favicon URL";
     return NO;
   }
   GURL faviconURL(favicon);
   DCHECK(faviconURL.is_valid());
   // Retrieve favicon resource and set favicon background image via JavaScript.
   base::WeakNSObject<CRWWebUIManager> weakSelf(self);
   void (^faviconHandler)(NSData*) = ^void(NSData* data) {
     base::scoped_nsobject<CRWWebUIManager> strongSelf([weakSelf retain]);
     if (!strongSelf)
       return;
     NSString* base64EncodedResource = [data base64EncodedStringWithOptions:0];
     NSString* dataURLString = [NSString
         stringWithFormat:@"data:image/png;base64,%@", base64EncodedResource];
     NSString* faviconURLString = base::SysUTF8ToNSString(faviconURL.spec());
     NSString* script =
         [NSString stringWithFormat:@"chrome.setFaviconBackground('%@', '%@');",
                                    faviconURLString, dataURLString];
-    [strongSelf webState]->ExecuteJavaScriptAsync(
-        base::SysNSStringToUTF16(script));
+    [strongSelf webState]->ExecuteJavaScript(base::SysNSStringToUTF16(script));
   };
   [self fetchResourceWithURL:faviconURL completionHandler:faviconHandler];
   return YES;
 }
 
+- (BOOL)handleLoadMojo:(const base::ListValue*)arguments {
+  std::string moduleName;
+  if (!arguments->GetString(0, &moduleName)) {
+    DLOG(WARNING) << "JS message parameter not found: Module name";
+    return NO;
+  }
+  std::string loadID;
+  if (!arguments->GetString(1, &loadID)) {
+    DLOG(WARNING) << "JS message parameter not found: Load ID";
+    return NO;
+  }
+
+  // Load and inject the script.
+  GURL resourceURL(self.webState->GetLastCommittedURL().Resolve(moduleName));
+  base::WeakNSObject<CRWWebUIManager> weakSelf(self);
+  [self fetchResourceWithURL:resourceURL completionHandler:^(NSData* data) {
+    std::string script;
+    if (data) {
+      script = base::SysNSStringToUTF8(
+          [[NSString alloc] initWithData:data encoding:NSUTF8StringEncoding]);
+      // WebUIIOSDataSourceImpl returns the default resource (which is the HTML
+      // page itself) if the resource URL isn't found. Fail with error instead
+      // of silently injecting garbage (leading to a not-very-useful syntax
+      // error from the JS side).
+      if (script.find("<!doctype") != std::string::npos) {
+        NOTREACHED() << "cannot load " << moduleName;
+        script.clear();
+      }
+    }
+
+    // Append with completion function call.
+    if (script.empty()) {
+      DLOG(ERROR) << "Unable to find a module named " << moduleName;
+      script = "__crWeb.webUIModuleLoadNotifier.moduleLoadFailed";
+    } else {
+      script += "__crWeb.webUIModuleLoadNotifier.moduleLoadCompleted";
+    }
+    base::StringAppendF(&script, "(%s, %s);",
+                        base::GetQuotedJSONString(moduleName).c_str(),
+                        base::GetQuotedJSONString(loadID).c_str());
+
+    [weakSelf webState]->ExecuteJavaScript(base::UTF8ToUTF16(script));
+  }];
+
+  return YES;
+}
+
 - (void)resetWebState {
   if (_webState) {
     _webState->RemoveScriptCommandCallback(kScriptCommandPrefix);
   }
   _webState = nullptr;
 }
 
 - (web::WebStateImpl*)webState {
   return _webState;
 }
 
 - (void)removeFetcher:(web::URLFetcherBlockAdapter*)fetcher {
   _fetchers.erase(std::find(_fetchers.begin(), _fetchers.end(), fetcher));
 }
 
 #pragma mark - Testing-Only Methods
 
 - (std::unique_ptr<web::URLFetcherBlockAdapter>)
     fetcherForURL:(const GURL&)URL
 completionHandler:(web::URLFetcherBlockAdapterCompletion)handler {
   return std::unique_ptr<web::URLFetcherBlockAdapter>(
       new web::URLFetcherBlockAdapter(
           URL, _webState->GetBrowserState()->GetRequestContext(), handler));
 }
 
 @end