WebUI for WKWebView

ios/web/webui/crw_web_ui_page_builder.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.
 
 #include "ios/web/webui/crw_web_ui_page_builder.h"
 
 #include <map>
 #include <string>
 #include <vector>
 
@@ skipping 10 matching lines @@
 NSString* const kJSTagSuffix = @"\"></script>";
 // Prefix for stylesheet tags. Used to locate CSS subresources.
 NSString* const kCSSTagPrefix = @"<link rel=\"stylesheet\" href=\"";
 // Suffix for stylesheet tags. Used to locate CSS subresources.
 NSString* const kCSSTagSuffix = @"\">";
 // Template for creating inlined JavaScript tags.
 NSString* const kWebUIScriptTextTemplate = @"<script>%@</script>";
 // Template for creating inlined CSS tags.
 NSString* const kWebUIStyleTextTemplate = @"<style>%@</style>";
 // URL placeholder for WebUI messaging JavaScript.
-NSString* const kWebUICoreJSURL = @"chrome://resources/js/ios/core.js";
+NSString* const kWebUIJSURL = @"chrome://resources/js/ios/web_ui.js";
 }  // namespace
 
 @interface CRWWebUIPageBuilder ()
 
 // Builds WebUI page for URL with HTML as default resource.
 - (void)buildWebUIPageForHTML:(NSString*)HTML
                      webUIURL:(const GURL&)URL
             completionHandler:(web::WebUIPageCompletion)completionHandler;
 
 // Loads |resourceURL| by invoking _delegate.
@@ skipping 111 matching lines @@
     pendingSubresourceCount--;
     // When subresource loading is complete, flatten the default resource
     // and invoke the completion handler.
     if (!pendingSubresourceCount) {
       [weakSelf flattenHTML:webUIHTML withSubresources:subresources];
       completionHandler(webUIHTML);
     }
   } copy] autorelease];
 
   for (NSString* URLString in subresourceURLStrings) {
-    // chrome://resources/js/ios/core.js is skipped because it is
+    // chrome://resources/js/ios/web_ui.js is skipped because it is
     // retrieved via webUIJavaScript rather than the net stack.
-    if ([URLString isEqualToString:kWebUICoreJSURL]) {
+    if ([URLString isEqualToString:kWebUIJSURL]) {
       pendingSubresourceCount--;
       if (!pendingSubresourceCount) {
         [weakSelf flattenHTML:webUIHTML withSubresources:subresources];
         completionHandler(webUIHTML);
       }
       continue;
     }
     GURL URL(pageURL.Resolve(base::SysNSStringToUTF8(URLString)));
     // If the resolved URL is different from URLString, replace URLString in
     // webUIHTML so that it will be located appropriately when flattening
@@ skipping 96 matching lines @@
       stringByAppendingString:sourceTag];
   [HTML replaceOccurrencesOfString:sourceTag
                         withString:extendedTag
                            options:0
                              range:NSMakeRange(0, [HTML length])];
 }
 
 - (void)flattenHTML:(NSMutableString*)HTML
     withSubresources:(std::map<GURL, std::string>)subresources {
   // Add core.js script to resources.
-  GURL webUIJSURL("chrome://resources/js/ios/core.js");
+  // TODO(ios): Move inclusion of this resource into WebUI implementation
+  // rather than forking each HTML file (crbug.com/487000).
+  GURL webUIJSURL("chrome://resources/js/ios/web_ui.js");
   subresources[webUIJSURL] = base::SysNSStringToUTF8([self webUIJavaScript]);
   for (auto it = subresources.begin(); it != subresources.end(); it++) {
     NSString* linkTemplate = @"";
     NSString* textTemplate = @"";
     if ([self isCSSSubresourceURL:it->first]) {
       linkTemplate =
           [NSString stringWithFormat:@"%@%%@%@", kCSSTagPrefix, kCSSTagSuffix];
       textTemplate = kWebUIStyleTextTemplate;
     } else {  // JavaScript.
       linkTemplate =
@@ skipping 15 matching lines @@
 - (NSString*)webUIJavaScript {
   NSBundle* bundle = base::mac::FrameworkBundle();
   NSString* path = [bundle pathForResource:@"web_ui" ofType:@"js"];
   DCHECK(path) << "web_ui.js file not found";
   return [NSString stringWithContentsOfFile:path
                                    encoding:NSUTF8StringEncoding
                                       error:nil];
 }
 
 @end