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"
@@ skipping 42 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) &&
+      ChromeTranslateClient::OverrideStoredPermanentCountry(country)) {
+    SendCountryToJs(true);
+  }
+}
+
 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) {
+  content::WebContents* web_contents = web_ui()->GetWebContents();
+  Profile* profile =
+      Profile::FromBrowserContext(web_contents->GetBrowserContext());
+  PrefService* prefs = profile->GetOriginalProfile()->GetPrefs();
+  std::unique_ptr<translate::TranslatePrefs> translate_prefs(
+      ChromeTranslateClient::CreateTranslatePrefs(prefs));
+
+  std::string country = translate_prefs->GetCountry();

[Alexei Svitkine (slow), 2016/04/20 20:43:54]

I actually don't quite follow this logic. What does the translate prefs
GetCountry() have to do with the variations country? I think the two are
separate, so if you want to fill in the existing variations country, I think you
need to get it from variations service. Or am I missing something here?

[hamelphi, 2016/04/20 21:39:46]

You are right. I don't need to go through ChromeTranslateClient at all. I can
access variations directly. Fixed.

+
+  base::DictionaryValue dict;
+  if (!country.empty()) {
+    dict.Set("country", new base::StringValue(translate_prefs->GetCountry()));

[Alexei Svitkine (slow), 2016/04/20 20:43:54]

Nit: You have |country| in a local var already.

[hamelphi, 2016/04/20 21:39:46]

Done.

+    dict.Set("update", new base::FundamentalValue(is_update));
+  }
+  SendMessageToJs("countryUpdated", dict);
+}