Allows to manually change the stored permanent country in Variations from the translate-internals d…

chrome/browser/ui/webui/translate_internals/translate_internals_handler.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/ui/webui/translate_internals/translate_internals_handler.h"
 
 #include <map>
 #include <vector>
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
 #include "base/values.h"
+#include "chrome/browser/browser_process.h"
 #include "chrome/browser/chrome_notification_types.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/translate/chrome_translate_client.h"
 #include "chrome/browser/translate/translate_service.h"
 #include "chrome/common/pref_names.h"
 #include "components/prefs/pref_service.h"
 #include "components/translate/core/browser/translate_download_manager.h"
 #include "components/translate/core/browser/translate_error_details.h"
 #include "components/translate/core/browser/translate_event_details.h"
 #include "components/translate/core/browser/translate_prefs.h"
 #include "components/translate/core/common/language_detection_details.h"
 #include "components/translate/core/common/translate_pref_names.h"
+#include "components/variations/service/variations_service.h"
 #include "content/public/browser/notification_details.h"
 #include "content/public/browser/notification_service.h"
 #include "content/public/browser/notification_source.h"
 #include "content/public/browser/notification_types.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/browser/web_ui.h"
 
+
 TranslateInternalsHandler::TranslateInternalsHandler() {
   notification_registrar_.Add(this,
                               chrome::NOTIFICATION_TAB_LANGUAGE_DETERMINED,
                               content::NotificationService::AllSources());
 
   error_subscription_ =
       translate::TranslateManager::RegisterTranslateErrorCallback(
           base::Bind(&TranslateInternalsHandler::OnTranslateError,
                      base::Unretained(this)));
 
@@ skipping 11 matching lines @@
 TranslateInternalsHandler::~TranslateInternalsHandler() {
   // |event_subscription_| and |error_subscription_| are deleted automatically
   // and un-register the callbacks automatically.
 }
 
 void TranslateInternalsHandler::RegisterMessages() {
   web_ui()->RegisterMessageCallback("removePrefItem", base::Bind(
       &TranslateInternalsHandler::OnRemovePrefItem, base::Unretained(this)));
   web_ui()->RegisterMessageCallback("requestInfo", base::Bind(
       &TranslateInternalsHandler::OnRequestInfo, base::Unretained(this)));
+  web_ui()->RegisterMessageCallback(
+      "overrideCountry",
+      base::Bind(&TranslateInternalsHandler::OnOverrideCountry,
+                 base::Unretained(this)));
 }
 
 void TranslateInternalsHandler::Observe(
     int type,
     const content::NotificationSource& source,
     const content::NotificationDetails& details) {
   DCHECK_EQ(chrome::NOTIFICATION_TAB_LANGUAGE_DETERMINED, type);
   const translate::LanguageDetectionDetails* language_detection_details =
       content::Details<const translate::LanguageDetectionDetails>(details)
           .ptr();
@@ skipping 89 matching lines @@
     translate_prefs->RemoveLanguagePairFromWhitelist(from, to);
   } else if (pref_name == "too_often_denied") {
     translate_prefs->ResetDenialState();
   } else {
     return;
   }
 
   SendPrefsToJs();
 }
 
+void TranslateInternalsHandler::OnOverrideCountry(const base::ListValue* args) {
+  std::string country;
+  if (args->GetString(0, &country)) {
+    variations::VariationsService* variations_service =
+        g_browser_process->variations_service();
+    if (variations_service)

[Alexei Svitkine (slow), 2016/04/20 21:46:44]

Nit: {} since body is multi-line.

Alternatively, maybe you can name it |service| and avoid needing to wrap?

[hamelphi, 2016/04/21 14:25:09]

Done.

+      SendCountryToJs(
+          variations_service->ForceSetStoredPermanentCountry(country));
+  }
+}
+
 void TranslateInternalsHandler::OnRequestInfo(const base::ListValue* /*args*/) {
   SendPrefsToJs();
   SendSupportedLanguagesToJs();
+  SendCountryToJs(false);
 }
 
 void TranslateInternalsHandler::SendMessageToJs(const std::string& message,
                                                 const base::Value& value) {
   const char func[] = "cr.translateInternals.messageHandler";
   base::StringValue message_data(message);
   web_ui()->CallJavascriptFunction(func, message_data, value);
 }
 
 void TranslateInternalsHandler::SendPrefsToJs() {
@@ skipping 40 matching lines @@
     if (translate::TranslateDownloadManager::IsAlphaLanguage(lang))
       alpha_languages_list->Append(new base::StringValue(lang));
   }
 
   dict.Set("languages", languages_list);
   dict.Set("alpha_languages", alpha_languages_list);
   dict.Set("last_updated",
            new base::FundamentalValue(last_updated.ToJsTime()));
   SendMessageToJs("supportedLanguagesUpdated", dict);
 }
+
+void TranslateInternalsHandler::SendCountryToJs(bool is_update) {
+  std::string country = "";

[Alexei Svitkine (slow), 2016/04/20 21:46:44]

Nit: std::string default ctor is the empty string, so need for "= "

[hamelphi, 2016/04/21 14:25:09]

Done.

+  variations::VariationsService* variations_service =
+      g_browser_process->variations_service();
+  if (variations_service) {

[Alexei Svitkine (slow), 2016/04/20 21:46:44]

Nit: No {}'s if the body is multi-line.

[hamelphi, 2016/04/21 14:25:09]

Done.

+    country = variations_service->GetStoredPermanentCountry();
+  }
+  base::DictionaryValue dict;
+  if (!country.empty()) {
+    dict.Set("country", new base::StringValue(country));
+    dict.Set("update", new base::FundamentalValue(is_update));
+  }
+  SendMessageToJs("countryUpdated", dict);
+}