Hook up LocalNtpSource to OneGoogleBarService

chrome/browser/search/local_ntp_source.cc

 // Copyright 2013 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/search/local_ntp_source.h"
 
-#include <stddef.h>
-
 #include "base/base64.h"
 #include "base/command_line.h"
+#include "base/feature_list.h"
 #include "base/json/json_string_value_serializer.h"
+#include "base/json/json_writer.h"
 #include "base/logging.h"
 #include "base/memory/ptr_util.h"
 #include "base/memory/ref_counted_memory.h"
 #include "base/strings/string_util.h"
 #include "base/strings/stringprintf.h"
 #include "base/values.h"
 #include "build/build_config.h"
+#include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/search/instant_io_context.h"
 #include "chrome/browser/search/local_files_ntp_source.h"
+#include "chrome/browser/search/one_google_bar/one_google_bar_data.h"
+#include "chrome/browser/search/one_google_bar/one_google_bar_service.h"
+#include "chrome/browser/search/one_google_bar/one_google_bar_service_factory.h"
 #include "chrome/browser/search_engines/template_url_service_factory.h"
 #include "chrome/browser/themes/theme_properties.h"
 #include "chrome/browser/themes/theme_service.h"
 #include "chrome/browser/themes/theme_service_factory.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/common/url_constants.h"
 #include "chrome/grit/browser_resources.h"
 #include "chrome/grit/generated_resources.h"
 #include "chrome/grit/theme_resources.h"
 #include "components/search_engines/template_url_service.h"
 #include "components/search_engines/template_url_service_observer.h"
 #include "components/strings/grit/components_strings.h"
 #include "content/public/browser/browser_thread.h"
 #include "crypto/secure_hash.h"
 #include "net/base/hash_value.h"
 #include "net/url_request/url_request.h"
 #include "third_party/skia/include/core/SkColor.h"
 #include "ui/base/l10n/l10n_util.h"
 #include "ui/base/resource/resource_bundle.h"
 #include "ui/base/webui/web_ui_util.h"
 #include "ui/resources/grit/ui_resources.h"
 #include "url/gurl.h"
 
 namespace {
 
+base::Feature kOneGoogleBarOnLocalNtpFeature{"OneGoogleBarOnLocalNtp",
+                                             base::FEATURE_DISABLED_BY_DEFAULT};
+
 // Signifies a locally constructed resource, i.e. not from grit/.
 const int kLocalResource = -1;
 
 const char kConfigDataFilename[] = "config.js";
 const char kThemeCSSFilename[] = "theme.css";
 const char kMainHtmlFilename[] = "local-ntp.html";
+const char kOneGoogleBarScriptFilename[] = "one-google.js";
+const char kOneGoogleBarInHeadStyleFilename[] = "one-google/in-head.css";
+const char kOneGoogleBarInHeadScriptFilename[] = "one-google/in-head.js";
+const char kOneGoogleBarAfterBarScriptFilename[] = "one-google/after-bar.js";
+const char kOneGoogleBarEndOfBodyScriptFilename[] = "one-google/end-of-body.js";
 
 const struct Resource{
   const char* filename;
   int identifier;
   const char* mime_type;
 } kResources[] = {
     {kMainHtmlFilename, kLocalResource, "text/html"},
     {"local-ntp.js", IDR_LOCAL_NTP_JS, "application/javascript"},
     {kConfigDataFilename, kLocalResource, "application/javascript"},
     {kThemeCSSFilename, kLocalResource, "text/css"},
     {"local-ntp.css", IDR_LOCAL_NTP_CSS, "text/css"},
     {"images/close_3_mask.png", IDR_CLOSE_3_MASK, "image/png"},
     {"images/close_4_button.png", IDR_CLOSE_4_BUTTON, "image/png"},
     {"images/ntp_default_favicon.png", IDR_NTP_DEFAULT_FAVICON, "image/png"},
+    {kOneGoogleBarScriptFilename, kLocalResource, "text/javascript"},
+    {kOneGoogleBarInHeadStyleFilename, kLocalResource, "text/css"},
+    {kOneGoogleBarInHeadScriptFilename, kLocalResource, "text/javascript"},
+    {kOneGoogleBarAfterBarScriptFilename, kLocalResource, "text/javascript"},
+    {kOneGoogleBarEndOfBodyScriptFilename, kLocalResource, "text/javascript"},
 };
 
 // Strips any query parameters from the specified path.
 std::string StripParameters(const std::string& path) {
   return path.substr(0, path.find("?"));
 }
 
 // Adds a localized string keyed by resource id to the dictionary.
 void AddString(base::DictionaryValue* dictionary,
                const std::string& key,
@@ skipping 77 matching lines @@
 
 bool DefaultSearchProviderIsGoogleImpl(
     const TemplateURLService* template_url_service) {
   const TemplateURL* default_provider =
       template_url_service->GetDefaultSearchProvider();
   return default_provider && (default_provider->GetEngineType(
                                   template_url_service->search_terms_data()) ==
                               SEARCH_ENGINE_GOOGLE);
 }
 
+std::unique_ptr<base::DictionaryValue> ConvertOGBDataToDict(
+    const OneGoogleBarData& og) {
+  auto result = base::MakeUnique<base::DictionaryValue>();
+  // Only provide the html parts here. The js and css are injected separately
+  // via <script src=...> and <link rel="stylesheet" href=...>.
+  result->SetString("html", og.bar_html);
+  result->SetString("end_of_body_html", og.end_of_body_html);
+  return result;
+}
+
 }  // namespace
 
 class LocalNtpSource::GoogleSearchProviderTracker
     : public TemplateURLServiceObserver {
  public:
   using SearchProviderIsGoogleChangedCallback =
       base::Callback<void(bool is_google)>;
 
   GoogleSearchProviderTracker(
       TemplateURLService* service,
@@ skipping 25 matching lines @@
   }
 
   TemplateURLService* service_;
   SearchProviderIsGoogleChangedCallback callback_;
 
   bool is_google_;
 };
 
 LocalNtpSource::LocalNtpSource(Profile* profile)
     : profile_(profile),
+      one_google_bar_service_(nullptr),
+      one_google_bar_service_observer_(this),
       default_search_provider_is_google_(false),
       default_search_provider_is_google_io_thread_(false),
       weak_ptr_factory_(this) {
   DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
 
+  if (base::FeatureList::IsEnabled(kOneGoogleBarOnLocalNtpFeature)) {
+    one_google_bar_service_ =
+        OneGoogleBarServiceFactory::GetForProfile(profile_);
+  }
+
+  // |one_google_bar_service_| is null in incognito, or when the feature is
+  // disabled.
+  if (one_google_bar_service_)
+    one_google_bar_service_observer_.Add(one_google_bar_service_);
+
   TemplateURLService* template_url_service =
       TemplateURLServiceFactory::GetForProfile(profile_);
   if (template_url_service) {
     google_tracker_ = base::MakeUnique<GoogleSearchProviderTracker>(
         template_url_service,
         base::Bind(&LocalNtpSource::DefaultSearchProviderIsGoogleChanged,
                    base::Unretained(this)));
     DefaultSearchProviderIsGoogleChanged(
         google_tracker_->DefaultSearchProviderIsGoogle());
   }
@@ skipping 17 matching lines @@
         GetConfigData(default_search_provider_is_google_);
     callback.Run(base::RefCountedString::TakeString(&config_data_js));
     return;
   }
   if (stripped_path == kThemeCSSFilename) {
     std::string theme_css = GetThemeCSS(profile_);
     callback.Run(base::RefCountedString::TakeString(&theme_css));
     return;
   }
 
+  if (base::StartsWith(stripped_path, "one-google",
+                       base::CompareCase::SENSITIVE)) {
+    if (!one_google_bar_service_) {
+      callback.Run(nullptr);
+      return;
+    }
+
+    const base::Optional<OneGoogleBarData>& data =
+        one_google_bar_service_->one_google_bar_data();
+
+    // The OneGoogleBar injector helper.
+    if (stripped_path == kOneGoogleBarScriptFilename) {
+      one_google_callbacks_.push_back(callback);
+
+      // If there already is (cached) OGB data, serve it immediately.
+      if (data.has_value())
+        ServeOneGoogleBar(*data);
+
+      // In any case, request a refresh.
+      one_google_bar_service_->Refresh();
+    } else {
+      // The actual OneGoogleBar sources.
+      std::string result;
+      if (data.has_value()) {
+        if (stripped_path == kOneGoogleBarInHeadStyleFilename) {
+          result = data->in_head_style;
+        } else if (stripped_path == kOneGoogleBarInHeadScriptFilename) {
+          result = data->in_head_script;
+        } else if (stripped_path == kOneGoogleBarAfterBarScriptFilename) {
+          result = data->after_bar_script;
+        } else if (stripped_path == kOneGoogleBarEndOfBodyScriptFilename) {
+          result = data->end_of_body_script;
+        }
+      }
+      callback.Run(base::RefCountedString::TakeString(&result));
+    }
+
+    return;
+  }
+
 #if !defined(GOOGLE_CHROME_BUILD)
   base::CommandLine* command_line = base::CommandLine::ForCurrentProcess();
   if (command_line->HasSwitch(switches::kLocalNtpReload)) {
     if (stripped_path == "local-ntp.html" || stripped_path == "local-ntp.js" ||
         stripped_path == "local-ntp.css") {
       base::ReplaceChars(stripped_path, "-", "_", &stripped_path);
       local_ntp::SendLocalFileResource(stripped_path, callback);
       return;
     }
   }
@@ skipping 76 matching lines @@
     // While live-editing the local NTP files, turn off CSP.
     return "script-src *;";
   }
 #endif  // !defined(GOOGLE_CHROME_BUILD)
 
   return "script-src 'strict-dynamic' "
          "'sha256-" +
          GetIntegritySha256Value(
              GetConfigData(default_search_provider_is_google_io_thread_)) +
          "' "
-         "'sha256-g38WaUaxnOIWY7E2LtLZ5ff9r5sn1dBj80jevt/kmx0=';";
+         "'sha256-ROPmcormZEipZzy3Ff+o345FFrhHWsAZjBpGIyZzCYY=';";
 }
 
 std::string LocalNtpSource::GetContentSecurityPolicyChildSrc() const {
   DCHECK_CURRENTLY_ON(content::BrowserThread::IO);
 
-  // Allow embedding of most visited iframes.
+  if (one_google_bar_service_) {
+    // Allow embedding of the most visited iframe, as well as the account
+    // switcher and the notifications dropdown from the One Google Bar.
+    // TODO(treib): Figure out a way to also allow staging instances.
+    return base::StringPrintf(
+        "child-src %s https://accounts.google.com/ https://docs.google.com "
+        "https://notifications.google.com;",
+        chrome::kChromeSearchMostVisitedUrl);
+  }
+  // Allow embedding of the most visited iframe.
   return base::StringPrintf("child-src %s;",
                             chrome::kChromeSearchMostVisitedUrl);
 }
 
+void LocalNtpSource::OnOneGoogleBarDataChanged() {
+  const base::Optional<OneGoogleBarData>& data =
+      one_google_bar_service_->one_google_bar_data();
+  if (data.has_value())
+    ServeOneGoogleBar(*data);
+  else
+    ServeNullOneGoogleBar();
+}
+
+void LocalNtpSource::OnOneGoogleBarFetchFailed() {
+  ServeNullOneGoogleBar();
+}
+
+void LocalNtpSource::OnOneGoogleBarServiceShuttingDown() {
+  one_google_bar_service_observer_.RemoveAll();
+  one_google_bar_service_ = nullptr;
+}
+
+void LocalNtpSource::ServeOneGoogleBar(const OneGoogleBarData& data) {
+  DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
+
+  if (one_google_callbacks_.empty())
+    return;
+
+  std::string json;
+  base::JSONWriter::Write(*ConvertOGBDataToDict(data), &json);
+  for (auto& callback : one_google_callbacks_) {
+    std::string data = "var og = " + json + ";";
+    callback.Run(base::RefCountedString::TakeString(&data));
+  }
+  one_google_callbacks_.clear();
+}
+
+void LocalNtpSource::ServeNullOneGoogleBar() {
+  for (auto& callback : one_google_callbacks_)
+    callback.Run(nullptr);
+  one_google_callbacks_.clear();
+}
+
 void LocalNtpSource::DefaultSearchProviderIsGoogleChanged(bool is_google) {
   DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
 
   default_search_provider_is_google_ = is_google;
   content::BrowserThread::PostTask(
       content::BrowserThread::IO, FROM_HERE,
       base::Bind(&LocalNtpSource::SetDefaultSearchProviderIsGoogleOnIOThread,
                  weak_ptr_factory_.GetWeakPtr(), is_google));
 }
 
 void LocalNtpSource::SetDefaultSearchProviderIsGoogleOnIOThread(
     bool is_google) {
   DCHECK_CURRENTLY_ON(content::BrowserThread::IO);
 
   default_search_provider_is_google_io_thread_ = is_google;
 }