Copy the implementations of TranslateUIDelegate to TransalteInfobarDelegate

chrome/browser/translate/translate_infobar_delegate.cc

 // Copyright (c) 2012 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/translate/translate_infobar_delegate.h"
 
 #include <algorithm>
 
 #include "base/i18n/string_compare.h"
 #include "base/metrics/histogram.h"
 #include "base/prefs/pref_service.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/infobars/infobar_service.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/translate/translate_accept_languages.h"
+#include "chrome/browser/translate/translate_browser_metrics.h"
 #include "chrome/browser/translate/translate_manager.h"
 #include "chrome/browser/translate/translate_tab_helper.h"
+#include "chrome/browser/translate/translate_ui_delegate.h"
 #include "components/translate/common/translate_constants.h"
 #include "content/public/browser/navigation_details.h"
 #include "content/public/browser/navigation_entry.h"
 #include "content/public/browser/web_contents.h"
 #include "grit/generated_resources.h"
 #include "grit/theme_resources.h"
 #include "third_party/icu/source/i18n/unicode/coll.h"
 #include "ui/base/l10n/l10n_util.h"
 
 using content::NavigationEntry;
 
 namespace {
+
 const char kCloseInfobar[] = "Translate.DeclineTranslateCloseInfobar";
 const char kShowErrorInfobar[] = "Translate.ShowErrorInfobar";
 
 }  // namespace
 
 const size_t TranslateInfoBarDelegate::kNoIndex = TranslateUIDelegate::NO_INDEX;
 
 TranslateInfoBarDelegate::~TranslateInfoBarDelegate() {
 }
 
@@ skipping 52 matching lines @@
   } else {
     DCHECK(replace_existing_infobar);
     infobar_service->ReplaceInfoBar(old_delegate,
                                     infobar.PassAs<InfoBarDelegate>());
   }
 }
 
 
 void TranslateInfoBarDelegate::UpdateOriginalLanguageIndex(
     size_t language_index) {
-  ui_delegate_.UpdateOriginalLanguageIndex(language_index);
+  // Note: TranslateUIDelegate should be updated if this implementation is
+  // updated.
+  if (original_language_index_ == language_index)
+    return;
+
+  UMA_HISTOGRAM_BOOLEAN(TranslateBrowserMetrics::GetMetricsName(
+      TranslateBrowserMetrics::UMA_MODIFY_ORIGINAL_LANG), true);
+  original_language_index_ = language_index;
 }
 
 void TranslateInfoBarDelegate::UpdateTargetLanguageIndex(
     size_t language_index) {
-  ui_delegate_.UpdateTargetLanguageIndex(language_index);
+  // Note: TranslateUIDelegate should be updated if this implementation is
+  // updated.
+  if (target_language_index_ == language_index)
+    return;
+
+  DCHECK_LT(language_index, num_languages());
+  UMA_HISTOGRAM_BOOLEAN(TranslateBrowserMetrics::GetMetricsName(
+      TranslateBrowserMetrics::UMA_MODIFY_TARGET_LANG), true);
+  target_language_index_ = language_index;
 }
 
 void TranslateInfoBarDelegate::Translate() {
-  ui_delegate_.Translate();
+  // Note: TranslateUIDelegate should be updated if this implementation is
+  // updated.
+  if (!web_contents()->GetBrowserContext()->IsOffTheRecord()) {
+    prefs_.ResetTranslationDeniedCount(original_language_code());
+    prefs_.IncrementTranslationAcceptedCount(original_language_code());
+  }
+  TranslateManager::GetInstance()->TranslatePage(web_contents(),
+                                                 original_language_code(),
+                                                 target_language_code());
+
+  UMA_HISTOGRAM_BOOLEAN(TranslateBrowserMetrics::GetMetricsName(
+      TranslateBrowserMetrics::UMA_PERFORM_TRANSLATE), true);
 }
 
 void TranslateInfoBarDelegate::RevertTranslation() {
-  ui_delegate_.RevertTranslation();
+  // Note: TranslateUIDelegate should be updated if this implementation is
+  // updated.
+  TranslateManager::GetInstance()->RevertTranslation(web_contents());
+  UMA_HISTOGRAM_BOOLEAN(TranslateBrowserMetrics::GetMetricsName(
+      TranslateBrowserMetrics::UMA_REVERT_TRANSLATION), true);
   RemoveSelf();
 }
 
 void TranslateInfoBarDelegate::ReportLanguageDetectionError() {
   TranslateManager::GetInstance()->ReportLanguageDetectionError(
       web_contents());
 }
 
 void TranslateInfoBarDelegate::TranslationDeclined() {
-  ui_delegate_.TranslationDeclined();
+  // Note: TranslateUIDelegate should be updated if this implementation is
+  // updated.
+  if (!web_contents()->GetBrowserContext()->IsOffTheRecord()) {
+    prefs_.ResetTranslationAcceptedCount(original_language_code());
+    prefs_.IncrementTranslationDeniedCount(original_language_code());
+  }
+
+  // Remember that the user declined the translation so as to prevent showing a
+  // translate infobar for that page again.  (TranslateManager initiates
+  // translations when getting a LANGUAGE_DETERMINED from the page, which
+  // happens when a load stops. That could happen multiple times, including
+  // after the user already declined the translation.)
+  TranslateTabHelper::FromWebContents(web_contents())->
+      language_state().set_translation_declined(true);
+
+  UMA_HISTOGRAM_BOOLEAN(TranslateBrowserMetrics::GetMetricsName(
+      TranslateBrowserMetrics::UMA_DECLINE_TRANSLATE), true);
 }
 
 bool TranslateInfoBarDelegate::IsTranslatableLanguageByPrefs() {
   Profile* profile =
       Profile::FromBrowserContext(web_contents()->GetBrowserContext());
   Profile* original_profile = profile->GetOriginalProfile();
   return TranslatePrefs::CanTranslateLanguage(original_profile,
                                               original_language_code());
 }
 
 void TranslateInfoBarDelegate::ToggleTranslatableLanguageByPrefs() {
-  if (ui_delegate_.IsLanguageBlocked()) {
-    ui_delegate_.SetLanguageBlocked(false);
+  // Note: TranslateUIDelegate should be updated if this implementation is
+  // updated.
+  if (prefs_.IsBlockedLanguage(original_language_code())) {
+    prefs_.UnblockLanguage(original_language_code());
   } else {
-    ui_delegate_.SetLanguageBlocked(true);
+    prefs_.BlockLanguage(original_language_code());
+    TranslateTabHelper* translate_tab_helper =
+        TranslateTabHelper::FromWebContents(web_contents());
+    DCHECK(translate_tab_helper);
+    translate_tab_helper->language_state().SetTranslateEnabled(false);
     RemoveSelf();
   }
+
+  UMA_HISTOGRAM_BOOLEAN(TranslateBrowserMetrics::GetMetricsName(
+      TranslateBrowserMetrics::UMA_NEVER_TRANSLATE_LANG), true);
 }
 
 bool TranslateInfoBarDelegate::IsSiteBlacklisted() {
-  return ui_delegate_.IsSiteBlacklisted();
+  // Note: TranslateUIDelegate should be updated if this implementation is
+  // updated.
+  std::string host = TranslateUIDelegate::GetPageHost(web_contents());
+  return !host.empty() && prefs_.IsSiteBlacklisted(host);
 }
 
 void TranslateInfoBarDelegate::ToggleSiteBlacklist() {
-  if (ui_delegate_.IsSiteBlacklisted()) {
-    ui_delegate_.SetSiteBlacklist(false);
+  // Note: TranslateUIDelegate should be updated if this implementation is
+  // updated.
+  std::string host = TranslateUIDelegate::GetPageHost(web_contents());
+  if (host.empty())
+    return;
+
+  if (prefs_.IsSiteBlacklisted(host)) {
+    prefs_.RemoveSiteFromBlacklist(host);
   } else {
-    ui_delegate_.SetSiteBlacklist(true);
+    prefs_.BlacklistSite(host);
+    TranslateTabHelper* translate_tab_helper =
+        TranslateTabHelper::FromWebContents(web_contents());
+    DCHECK(translate_tab_helper);
+    translate_tab_helper->language_state().SetTranslateEnabled(false);
     RemoveSelf();
   }
+
+  UMA_HISTOGRAM_BOOLEAN(TranslateBrowserMetrics::GetMetricsName(
+      TranslateBrowserMetrics::UMA_NEVER_TRANSLATE_SITE), true);
 }
 
 bool TranslateInfoBarDelegate::ShouldAlwaysTranslate() {
-  return ui_delegate_.ShouldAlwaysTranslate();
+  // Note: TranslateUIDelegate should be updated if this implementation is
+  // updated.
+  return prefs_.IsLanguagePairWhitelisted(original_language_code(),
+                                          target_language_code());
 }
 
 void TranslateInfoBarDelegate::ToggleAlwaysTranslate() {
-  ui_delegate_.SetAlwaysTranslate(!ui_delegate_.ShouldAlwaysTranslate());
+  // Note: TranslateUIDelegate should be updated if this implementation is
+  // updated.
+  const std::string& original_lang = original_language_code();
+  const std::string& target_lang = target_language_code();
+  if (prefs_.IsLanguagePairWhitelisted(original_lang, target_lang))
+    prefs_.RemoveLanguagePairFromWhitelist(original_lang, target_lang);
+  else
+    prefs_.WhitelistLanguagePair(original_lang, target_lang);
+
+  UMA_HISTOGRAM_BOOLEAN(TranslateBrowserMetrics::GetMetricsName(
+      TranslateBrowserMetrics::UMA_ALWAYS_TRANSLATE_LANG), true);
 }
 
 void TranslateInfoBarDelegate::AlwaysTranslatePageLanguage() {
-  DCHECK(!ui_delegate_.ShouldAlwaysTranslate());
-  ui_delegate_.SetAlwaysTranslate(true);
+  // Note: TranslateUIDelegate should be updated if this implementation is
+  // updated.
+  const std::string& original_lang = original_language_code();
+  const std::string& target_lang = target_language_code();
+  DCHECK(!prefs_.IsLanguagePairWhitelisted(original_lang, target_lang));
+  prefs_.WhitelistLanguagePair(original_lang, target_lang);
+  UMA_HISTOGRAM_BOOLEAN(TranslateBrowserMetrics::GetMetricsName(
+      TranslateBrowserMetrics::UMA_ALWAYS_TRANSLATE_LANG), true);
+
   Translate();
 }
 
 void TranslateInfoBarDelegate::NeverTranslatePageLanguage() {
-  DCHECK(!ui_delegate_.IsLanguageBlocked());
-  ui_delegate_.SetLanguageBlocked(true);
+  // Note: TranslateUIDelegate should be updated if this implementation is
+  // updated.
+  prefs_.BlockLanguage(original_language_code());
+  TranslateTabHelper* translate_tab_helper =
+      TranslateTabHelper::FromWebContents(web_contents());
+  DCHECK(translate_tab_helper);
+  translate_tab_helper->language_state().SetTranslateEnabled(false);
+  UMA_HISTOGRAM_BOOLEAN(TranslateBrowserMetrics::GetMetricsName(
+      TranslateBrowserMetrics::UMA_NEVER_TRANSLATE_LANG), true);
+
   RemoveSelf();
 }
 
 string16 TranslateInfoBarDelegate::GetMessageInfoBarText() {
   if (infobar_type_ == TRANSLATING) {
     string16 target_language_name = language_name_at(target_language_index());
     return l10n_util::GetStringFUTF16(IDS_TRANSLATE_INFOBAR_TRANSLATING_TO,
                                       target_language_name);
   }
 
@@ skipping 113 matching lines @@
     Type infobar_type,
     TranslateInfoBarDelegate* old_delegate,
     const std::string& original_language,
     const std::string& target_language,
     TranslateErrors::Type error_type,
     PrefService* prefs,
     ShortcutConfiguration shortcut_config)
     : InfoBarDelegate(infobar_service),
       infobar_type_(infobar_type),
       background_animation_(NONE),
-      ui_delegate_(web_contents(), original_language, target_language),
+      original_language_index_(kNoIndex),
+      target_language_index_(kNoIndex),
       error_type_(error_type),
       prefs_(prefs),
       shortcut_config_(shortcut_config) {
   DCHECK_NE((infobar_type_ == TRANSLATION_ERROR),
             (error_type_ == TranslateErrors::NONE));
 
   if (old_delegate && (old_delegate->is_error() != is_error()))
     background_animation_ = is_error() ? NORMAL_TO_ERROR : ERROR_TO_NORMAL;
+
+  TranslateUIDelegate::GetSortedLanguageNames(
+      g_browser_process->GetApplicationLocale(), &languages_);
+
+  for (std::vector<LanguageNamePair>::const_iterator iter = languages_.begin();
+       iter != languages_.end(); ++iter) {
+    std::string language_code = iter->first;
+    if (language_code == original_language)
+      original_language_index_ = iter - languages_.begin();
+    if (language_code == target_language)
+      target_language_index_ = iter - languages_.begin();
+  }
 }
 
 void TranslateInfoBarDelegate::InfoBarDismissed() {
   if (infobar_type_ != BEFORE_TRANSLATE)
     return;
 
   // The user closed the infobar without clicking the translate button.
   TranslationDeclined();
   UMA_HISTOGRAM_BOOLEAN(kCloseInfobar, true);
 }
@@ skipping 13 matching lines @@
   if (!details.is_navigation_to_different_page() && !details.is_main_frame)
     return false;
 
   return InfoBarDelegate::ShouldExpireInternal(details);
 }
 
 TranslateInfoBarDelegate*
     TranslateInfoBarDelegate::AsTranslateInfoBarDelegate() {
   return this;
 }