Add chrome://site-tiles-internals/

chrome/browser/ui/webui/site_tiles_internals_ui.cc

+// Copyright 2016 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/site_tiles_internals_ui.h"
+
+#include "base/bind.h"
+#include "base/memory/ptr_util.h"
+#include "chrome/browser/favicon/favicon_service_factory.h"
+#include "chrome/browser/history/top_sites_factory.h"
+#include "chrome/browser/profiles/profile.h"
+#include "chrome/browser/search/suggestions/image_decoder_impl.h"
+#include "chrome/browser/search/suggestions/suggestions_service_factory.h"
+#include "chrome/browser/search_engines/template_url_service_factory.h"
+#include "chrome/common/url_constants.h"
+#include "components/grit/components_resources.h"
+#include "components/history/core/browser/top_sites.h"
+#include "components/image_fetcher/image_fetcher_impl.h"
+#include "components/ntp_tiles/field_trial.h"
+#include "components/ntp_tiles/icon_cacher.h"
+#include "components/ntp_tiles/most_visited_sites.h"
+#include "components/ntp_tiles/webui/site_tiles_internals_message_handler.h"
+#include "components/ntp_tiles/webui/site_tiles_internals_message_handler_client.h"
+#include "content/public/browser/browser_thread.h"
+#include "content/public/browser/web_ui.h"
+#include "content/public/browser/web_ui_data_source.h"
+#include "content/public/browser/web_ui_message_handler.h"
+
+#if defined(OS_ANDROID)
+#include "chrome/browser/android/ntp/popular_sites.h"
+#endif
+
+namespace {
+
+// The implementation for the chrome://site-tiles-internals page.
+class ChromeSiteTilesInternalsMessageHandlerClient
+    : public content::WebUIMessageHandler,
+      public ntp_tiles::SiteTilesInternalsMessageHandlerClient {
+ public:
+  ChromeSiteTilesInternalsMessageHandlerClient() {}
+
+ private:
+  // content::WebUIMessageHandler:
+  void RegisterMessages() override;
+
+  // ntp_tiles::SiteTilesInternalsMessageHandlerClient
+  bool IsSourceEnabled(ntp_tiles::NTPTileSource source) override;
+  std::unique_ptr<ntp_tiles::MostVisitedSites> MakeMostVisitedSites() override;
+  std::unique_ptr<ntp_tiles::PopularSites> MakePopularSites() override;
+  PrefService* GetPrefs() override;
+  void RegisterMessageCallback(
+      const std::string& message,
+      const base::Callback<void(const base::ListValue*)>& callback) override;
+  void CallJavascriptFunctionVector(
+      const std::string& name,
+      const std::vector<const base::Value*>& values) override;
+
+  ntp_tiles::SiteTilesInternalsMessageHandler handler_;
+
+  DISALLOW_COPY_AND_ASSIGN(ChromeSiteTilesInternalsMessageHandlerClient);
+};
+
+void ChromeSiteTilesInternalsMessageHandlerClient::RegisterMessages() {
+  handler_.RegisterMessages(this);
+}
+
+bool ChromeSiteTilesInternalsMessageHandlerClient::IsSourceEnabled(
+    ntp_tiles::NTPTileSource source) {
+  switch (source) {
+    case ntp_tiles::NTPTileSource::TOP_SITES:
+    case ntp_tiles::NTPTileSource::SUGGESTIONS_SERVICE:
+      return true;
+#if defined(OS_ANDROID)
+    case ntp_tiles::NTPTileSource::POPULAR:
+      return ntp_tiles::ShouldShowPopularSites();
+    case ntp_tiles::NTPTileSource::WHITELIST:
+      return true;
+#else
+    case ntp_tiles::NTPTileSource::POPULAR:
+      return false;
+    case ntp_tiles::NTPTileSource::WHITELIST:
+      // TODO(sfiera): support WHITELIST.

[Marc Treib, 2016/12/12 09:42:18]

I guess this is because desktop doesn't pass in the supervisor yet?

[sfiera, 2016/12/12 10:09:50]

Right.

I guess we're not going to see WHITELIST tiles on this page on Android either
because MakeMostVisitedSites() below doesn't pass the supervisor, but I'm
inclined to let that slide until the factory exists.

+      return false;
+#endif
+  }
+  NOTREACHED();
+  return false;
+}
+
+std::unique_ptr<ntp_tiles::MostVisitedSites>
+ChromeSiteTilesInternalsMessageHandlerClient::MakeMostVisitedSites() {
+  // TODO(sfiera): share with Android and Instant in a factory.
+  auto* profile = Profile::FromWebUI(web_ui());
+  return base::MakeUnique<ntp_tiles::MostVisitedSites>(
+      GetPrefs(), TopSitesFactory::GetForProfile(profile),
+      suggestions::SuggestionsServiceFactory::GetForProfile(profile),
+      MakePopularSites(),
+      base::MakeUnique<ntp_tiles::IconCacher>(
+          FaviconServiceFactory::GetForProfile(
+              profile, ServiceAccessType::IMPLICIT_ACCESS),
+          base::MakeUnique<image_fetcher::ImageFetcherImpl>(
+              base::MakeUnique<suggestions::ImageDecoderImpl>(),
+              profile->GetRequestContext())),
+      /*supervisor=*/nullptr);
+}
+
+std::unique_ptr<ntp_tiles::PopularSites>
+ChromeSiteTilesInternalsMessageHandlerClient::MakePopularSites() {
+#if defined(OS_ANDROID)
+  return ChromePopularSites::NewForProfile(Profile::FromWebUI(web_ui()));
+#else
+  return nullptr;
+#endif
+}
+
+PrefService* ChromeSiteTilesInternalsMessageHandlerClient::GetPrefs() {
+  return Profile::FromWebUI(web_ui())->GetPrefs();
+}
+
+void ChromeSiteTilesInternalsMessageHandlerClient::RegisterMessageCallback(
+    const std::string& message,
+    const base::Callback<void(const base::ListValue*)>& callback) {
+  web_ui()->RegisterMessageCallback(message, callback);
+}
+
+void ChromeSiteTilesInternalsMessageHandlerClient::CallJavascriptFunctionVector(
+    const std::string& name,
+    const std::vector<const base::Value*>& values) {
+  web_ui()->CallJavascriptFunctionUnsafe(name, values);
+}
+
+content::WebUIDataSource* CreateSiteTilesInternalsHTMLSource() {
+  content::WebUIDataSource* source =
+      content::WebUIDataSource::Create(chrome::kChromeUISiteTilesInternalsHost);
+
+  source->AddResourcePath("site_tiles_internals.js",
+                          IDR_SITE_TILES_INTERNALS_JS);
+  source->AddResourcePath("site_tiles_internals.css",
+                          IDR_SITE_TILES_INTERNALS_CSS);
+  source->SetDefaultResource(IDR_SITE_TILES_INTERNALS_HTML);
+  return source;
+}
+
+}  // namespace
+
+SiteTilesInternalsUI::SiteTilesInternalsUI(content::WebUI* web_ui)
+    : WebUIController(web_ui) {
+  content::WebUIDataSource::Add(Profile::FromWebUI(web_ui),
+                                CreateSiteTilesInternalsHTMLSource());
+  web_ui->AddMessageHandler(new ChromeSiteTilesInternalsMessageHandlerClient);
+}
+
+SiteTilesInternalsUI::~SiteTilesInternalsUI() {}