DO NOT COMMIT: hack hack hacking on preloading md-settings

chrome/browser/ui/webui/settings/md_settings_ui.cc

 // 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 "chrome/browser/ui/webui/settings/md_settings_ui.h"
 
 #include <stddef.h>
 
-#include <string>
+#include <vector>
 
+#include "base/bind.h"
 #include "base/metrics/histogram_macros.h"
+#include "base/strings/string_piece.h"
+#include "base/strings/string_split.h"
+#include "base/strings/string_util.h"
+#include "base/strings/stringprintf.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/ui/webui/settings/about_handler.h"
 #include "chrome/browser/ui/webui/settings/appearance_handler.h"
 #include "chrome/browser/ui/webui/settings/browser_lifetime_handler.h"
 #include "chrome/browser/ui/webui/settings/downloads_handler.h"
 #include "chrome/browser/ui/webui/settings/font_handler.h"
 #include "chrome/browser/ui/webui/settings/languages_handler.h"
 #include "chrome/browser/ui/webui/settings/md_settings_localized_strings_provider.h"
 #include "chrome/browser/ui/webui/settings/metrics_reporting_handler.h"
 #include "chrome/browser/ui/webui/settings/people_handler.h"
 #include "chrome/browser/ui/webui/settings/profile_info_handler.h"
 #include "chrome/browser/ui/webui/settings/protocol_handlers_handler.h"
 #include "chrome/browser/ui/webui/settings/reset_settings_handler.h"
 #include "chrome/browser/ui/webui/settings/search_engines_handler.h"
 #include "chrome/browser/ui/webui/settings/settings_clear_browsing_data_handler.h"
 #include "chrome/browser/ui/webui/settings/settings_cookies_view_handler.h"
 #include "chrome/browser/ui/webui/settings/settings_media_devices_selection_handler.h"
 #include "chrome/browser/ui/webui/settings/settings_page_ui_handler.h"
 #include "chrome/browser/ui/webui/settings/settings_startup_pages_handler.h"
 #include "chrome/browser/ui/webui/settings/site_settings_handler.h"
 #include "chrome/common/url_constants.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/browser/web_ui.h"
-#include "content/public/browser/web_ui_data_source.h"
+
+// TODO(dbeam): make content/public interface for this.
+#include "content/browser/webui/shared_resources_data_source.h"
+
 #include "grit/settings_resources.h"
 #include "grit/settings_resources_map.h"
+#include "ui/base/resource/resource_bundle.h"
 
 #if defined(OS_CHROMEOS)
 #include "ash/common/system/chromeos/palette/palette_utils.h"
 #include "chrome/browser/chromeos/login/quick_unlock/quick_unlock_utils.h"
 #include "chrome/browser/ui/webui/settings/chromeos/accessibility_handler.h"
 #include "chrome/browser/ui/webui/settings/chromeos/change_picture_handler.h"
 #include "chrome/browser/ui/webui/settings/chromeos/cups_printers_handler.h"
 #include "chrome/browser/ui/webui/settings/chromeos/device_keyboard_handler.h"
 #include "chrome/browser/ui/webui/settings/chromeos/device_pointer_handler.h"
 #include "chrome/browser/ui/webui/settings/chromeos/easy_unlock_settings_handler.h"
 #include "chrome/browser/ui/webui/settings/chromeos/internet_handler.h"
 #else  // !defined(OS_CHROMEOS)
 #include "chrome/browser/ui/webui/settings/settings_default_browser_handler.h"
 #include "chrome/browser/ui/webui/settings/settings_manage_profile_handler.h"
 #include "chrome/browser/ui/webui/settings/system_handler.h"
 #endif  // defined(OS_CHROMEOS)
 
 #if defined(USE_NSS_CERTS)
 #include "chrome/browser/ui/webui/settings/certificates_handler.h"
 #elif defined(OS_WIN) || defined(OS_MACOSX)
 #include "chrome/browser/ui/webui/settings/native_certificates_handler.h"
 #endif  // defined(USE_NSS_CERTS)
 
+#include "third_party/re2/src/re2/re2.h"
+
 namespace settings {
 
 MdSettingsUI::MdSettingsUI(content::WebUI* web_ui, const GURL& url)
     : content::WebUIController(web_ui),
-      WebContentsObserver(web_ui->GetWebContents()) {
+      WebContentsObserver(web_ui->GetWebContents()),
+      html_source_(nullptr) {
+      // weak_ptr_factory_(this) {
   Profile* profile = Profile::FromWebUI(web_ui);
   AddSettingsPageUIHandler(new AppearanceHandler(web_ui));
 
 #if defined(USE_NSS_CERTS)
   AddSettingsPageUIHandler(new CertificatesHandler(false));
 #elif defined(OS_WIN) || defined(OS_MACOSX)
   AddSettingsPageUIHandler(new NativeCertificatesHandler());
 #endif  // defined(USE_NSS_CERTS)
 
   AddSettingsPageUIHandler(new ClearBrowsingDataHandler(web_ui));
@@ skipping 25 matching lines @@
   AddSettingsPageUIHandler(new DefaultBrowserHandler(web_ui));
   AddSettingsPageUIHandler(new ManageProfileHandler(profile));
   AddSettingsPageUIHandler(new SystemHandler());
 #endif
 
   // Host must be derived from the visible URL, since this might be serving
   // either chrome://settings or chrome://md-settings.
   CHECK(url.GetOrigin() == GURL(chrome::kChromeUISettingsURL).GetOrigin() ||
         url.GetOrigin() == GURL(chrome::kChromeUIMdSettingsURL).GetOrigin());
 
-  content::WebUIDataSource* html_source =
-      content::WebUIDataSource::Create(url.host());
+  html_source_ = content::WebUIDataSource::Create(url.host());
+  html_source_->SetRequestFilter(
+      base::Bind(&MdSettingsUI::RequestFilter, base::Unretained(this)));
 
 #if defined(OS_CHROMEOS)
   chromeos::settings::EasyUnlockSettingsHandler* easy_unlock_handler =
-      chromeos::settings::EasyUnlockSettingsHandler::Create(html_source,
+      chromeos::settings::EasyUnlockSettingsHandler::Create(html_source_,
                                                             profile);
   if (easy_unlock_handler)
     AddSettingsPageUIHandler(easy_unlock_handler);
 
-  html_source->AddBoolean("noteAllowed", ash::IsPaletteEnabled());
-  html_source->AddBoolean("quickUnlockEnabled",
-                          chromeos::IsQuickUnlockEnabled());
+  html_source_->AddBoolean("noteAllowed", ash::IsPaletteEnabled());
+  html_source_->AddBoolean("quickUnlockEnabled",
+                           chromeos::IsQuickUnlockEnabled());
 #endif
 
-  AddSettingsPageUIHandler(AboutHandler::Create(html_source, profile));
-  AddSettingsPageUIHandler(ResetSettingsHandler::Create(html_source, profile));
+  AddSettingsPageUIHandler(AboutHandler::Create(html_source_, profile));
+  AddSettingsPageUIHandler(ResetSettingsHandler::Create(html_source_, profile));
 
   // Add all settings resources.
   for (size_t i = 0; i < kSettingsResourcesSize; ++i) {
-    html_source->AddResourcePath(kSettingsResources[i].name,
-                                 kSettingsResources[i].value);
+    html_source_->AddResourcePath(kSettingsResources[i].name,
+                                  kSettingsResources[i].value);
   }
 
-  AddLocalizedStrings(html_source, profile);
-  html_source->SetDefaultResource(IDR_SETTINGS_SETTINGS_HTML);
+  AddLocalizedStrings(html_source_, profile);
+  html_source_->SetDefaultResource(IDR_SETTINGS_SETTINGS_HTML);
+
+  base::TimeTicks then = base::TimeTicks::Now();
+
+  std::vector<GURL> urls;
+  GatherPreloadUrls(GURL("chrome://md-settings/settings.html"), &urls);
+
+  for (const GURL& url : urls) {
+    const std::string spec = url.spec();
+
+    std::string rel;
+    std::string as;
+    if (base::EndsWith(spec, ".js", base::CompareCase::INSENSITIVE_ASCII)) {
+      rel = "preload";
+      as = " as=\"script\"";
+    } else if (base::EndsWith(spec, ".css",
+                              base::CompareCase::INSENSITIVE_ASCII)) {
+      rel = "preload";
+      as = " as=\"style\"";
+    } else if (base::EndsWith(spec, ".html",
+                              base::CompareCase::INSENSITIVE_ASCII)) {
+      continue;
+      rel = "import";
+      // as="document" doesn't work yet.
+    }
+
+    preloads_.append(base::StringPrintf(
+        "<link rel=\"%s\" href=\"%s\"%s>\n",
+        rel.c_str(), spec.c_str(), as.c_str()));
+  }
+
+  LOG(ERROR) << "time gathering: " << (base::TimeTicks::Now() - then).InMilliseconds();
 
   content::WebUIDataSource::Add(web_ui->GetWebContents()->GetBrowserContext(),
-                                html_source);
+                                html_source_);
 }
 
 MdSettingsUI::~MdSettingsUI() {
 }
 
+bool MdSettingsUI::RequestFilter(
+    const std::string& path,
+    const content::WebUIDataSource::GotDataCallback& callback) {
+  if (!path.empty())
+    return false;
+
+  base::StringPiece settings_html =
+      ui::ResourceBundle::GetSharedInstance().GetRawDataResource(
+          IDR_SETTINGS_SETTINGS_HTML);
+  
+  std::string preloads_placeholder("<!-- preloads -->");
+  size_t preloads_start = settings_html.find(preloads_placeholder);
+
+  std::string response;
+  settings_html.substr(0, preloads_start).AppendToString(&response);
+  response.append(preloads_);
+  settings_html.substr(preloads_start + preloads_placeholder.size())
+      .AppendToString(&response);
+
+  callback.Run(base::RefCountedString::TakeString(&response));
+  return true;
+}
+
+void MdSettingsUI::GatherPreloadUrls(const GURL& url, std::vector<GURL>* urls) {
+  CHECK(url.SchemeIs(content::kChromeUIScheme) &&
+        (url.host() == content::kChromeUIResourcesHost ||
+         url.host() == chrome::kChromeUISettingsHost ||
+         url.host() == chrome::kChromeUIMdSettingsHost));
+
+  std::string path = url.path();
+  CHECK(!path.empty());
+
+  if (path[0] == '/')
+    path.erase(0, 1);
+
+  int idr = -1;
+
+  if (url.host() == content::kChromeUIResourcesHost) {
+    const auto& map = content::SharedResourcesDataSource::GetResourcesMap();
+    auto it = map.find(path);
+    if (it != map.end())
+      idr = it->second;
+  } else {
+    const auto& map = html_source_->path_to_idr_map();
+    auto it = map.find(path);
+    if (it != map.end())
+      idr = it->second;
+  }
+
+  if (idr == -1) {
+    LOG(ERROR) << "@@ unknown url: " << url << ", path: " << path << " @@";
+    return;
+  }
+
+  std::string resource = ui::ResourceBundle::GetSharedInstance()
+      .GetRawDataResource(idr).as_string();
+
+  re2::StringPiece re2_resource(resource);
+
+  re2::RE2 regex(
+      "<link[^>]+?rel=\"import\"[^>]+?href=\"([^\"]+)\"|"
+      "<link[^>]+?rel=\"stylesheet\"[^>]+?href=\"([^\"]+)\"|"
+      "<script[^>]+?src=\"([^\"]+)");
+
+  re2::StringPiece html, css, js;
+
+  while (re2::RE2::FindAndConsume(&re2_resource, regex, &html, &css, &js)) {
+    CHECK_EQ(html.empty() + css.empty() + js.empty(), 2);
+
+    std::string new_path = html.as_string() + css.as_string() + js.as_string();
+    CHECK(!new_path.empty());
+
+    GURL new_url = url.Resolve(new_path);
+
+    if (std::find(urls->begin(), urls->end(), new_url) != urls->end())
+      continue;
+
+    urls->push_back(new_url);
+    GatherPreloadUrls(new_url, urls);
+  }
+}
+
 void MdSettingsUI::AddSettingsPageUIHandler(SettingsPageUIHandler* handler) {
   DCHECK(handler);
   handlers_.insert(handler);
   web_ui()->AddMessageHandler(handler);  // |handler| is owned by |web_ui()|.
 }
 
 void MdSettingsUI::DidStartProvisionalLoadForFrame(
     content::RenderFrameHost* render_frame_host,
     const GURL& validated_url,
     bool is_error_page,
     bool is_iframe_srcdoc) {
   load_start_time_ = base::Time::Now();
 }
 
 void MdSettingsUI::DocumentLoadedInFrame(
     content::RenderFrameHost* render_frame_host) {
   UMA_HISTOGRAM_TIMES("Settings.LoadDocumentTime.MD",
                       base::Time::Now() - load_start_time_);
 }
 
 void MdSettingsUI::DocumentOnLoadCompletedInMainFrame() {
   UMA_HISTOGRAM_TIMES("Settings.LoadCompletedTime.MD",
                       base::Time::Now() - load_start_time_);
 }
 
 }  // namespace settings