Remove most of chrome/ and content/ usage from TranslateInfoBarDelegate

chrome/browser/translate/translate_infobar_delegate.cc

 // Copyright 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/infobars/infobar_service.h"
-#include "chrome/browser/translate/translate_tab_helper.h"
 #include "components/infobars/core/infobar.h"
+#include "components/translate/core/browser/language_state.h"
 #include "components/translate/core/browser/translate_accept_languages.h"
+#include "components/translate/core/browser/translate_client.h"
 #include "components/translate/core/browser/translate_download_manager.h"
+#include "components/translate/core/browser/translate_driver.h"
 #include "components/translate/core/browser/translate_manager.h"
 #include "components/translate/core/common/translate_constants.h"
-#include "content/public/browser/browser_context.h"
 #include "content/public/browser/web_contents.h"
 #include "grit/component_strings.h"
 #include "grit/theme_resources.h"
 #include "third_party/icu/source/i18n/unicode/coll.h"
 #include "ui/base/l10n/l10n_util.h"
 
 namespace {
 
 // Counts used to decide whether infobars should be shown.
 // Android and iOS implementations do not offer a drop down (for space reasons),
@@ skipping 14 matching lines @@
 #endif
 
 }  // namespace
 
 const size_t TranslateInfoBarDelegate::kNoIndex = TranslateUIDelegate::NO_INDEX;
 
 TranslateInfoBarDelegate::~TranslateInfoBarDelegate() {
 }
 
 // static
-void TranslateInfoBarDelegate::Create(bool replace_existing_infobar,
-                                      content::WebContents* web_contents,
-                                      translate::TranslateStep step,
-                                      const std::string& original_language,
-                                      const std::string& target_language,
-                                      TranslateErrors::Type error_type,
-                                      PrefService* prefs,
-                                      bool triggered_from_menu) {
+void TranslateInfoBarDelegate::Create(
+    bool replace_existing_infobar,
+    const base::WeakPtr<TranslateManager>& translate_manager,
+    InfoBarService* infobar_service,
+    bool is_off_the_record,
+    translate::TranslateStep step,
+    const std::string& original_language,
+    const std::string& target_language,
+    TranslateErrors::Type error_type,
+    PrefService* prefs,
+    bool triggered_from_menu) {
+  DCHECK(translate_manager);
+  DCHECK(infobar_service);
+
   // Check preconditions.
   if (step != translate::TRANSLATE_STEP_TRANSLATE_ERROR) {
     DCHECK(TranslateDownloadManager::IsSupportedLanguage(target_language));
     if (!TranslateDownloadManager::IsSupportedLanguage(original_language)) {
       // The original language can only be "unknown" for the "translating"
       // infobar, which is the case when the user started a translation from the
       // context menu.
       DCHECK(step == translate::TRANSLATE_STEP_TRANSLATING ||
              step == translate::TRANSLATE_STEP_AFTER_TRANSLATE);
       DCHECK_EQ(translate::kUnknownLanguageCode, original_language);
     }
   }
 
   // Do not create the after translate infobar if we are auto translating.
-  if ((step == translate::TRANSLATE_STEP_AFTER_TRANSLATE) ||
-      (step == translate::TRANSLATE_STEP_TRANSLATING)) {
-    TranslateTabHelper* translate_tab_helper =
-        TranslateTabHelper::FromWebContents(web_contents);
-    if (!translate_tab_helper ||
-        translate_tab_helper->GetLanguageState().InTranslateNavigation())
-      return;
+  if (((step == translate::TRANSLATE_STEP_AFTER_TRANSLATE) ||
+       (step == translate::TRANSLATE_STEP_TRANSLATING)) &&
+      translate_manager->translate_client()->GetTranslateDriver()
+          ->GetLanguageState().InTranslateNavigation()) {
+    return;
   }
 
   // Find any existing translate infobar delegate.
   infobars::InfoBar* old_infobar = NULL;
-  InfoBarService* infobar_service =
-      InfoBarService::FromWebContents(web_contents);
   TranslateInfoBarDelegate* old_delegate = NULL;
   for (size_t i = 0; i < infobar_service->infobar_count(); ++i) {
     old_infobar = infobar_service->infobar_at(i);
     old_delegate = old_infobar->delegate()->AsTranslateInfoBarDelegate();
     if (old_delegate) {
       if (!replace_existing_infobar)
         return;
       break;
     }
   }
 
   // Add the new delegate.
   scoped_ptr<infobars::InfoBar> infobar(CreateInfoBar(
       scoped_ptr<TranslateInfoBarDelegate>(new TranslateInfoBarDelegate(
-          web_contents, step, old_delegate, original_language,
-          target_language, error_type, prefs,
+          translate_manager, is_off_the_record, step, old_delegate,
+          original_language, target_language, error_type, prefs,
           triggered_from_menu))));
   if (old_delegate)
     infobar_service->ReplaceInfoBar(old_infobar, infobar.Pass());
   else
     infobar_service->AddInfoBar(infobar.Pass());
 }
 
 
 void TranslateInfoBarDelegate::UpdateOriginalLanguageIndex(
     size_t language_index) {
   ui_delegate_.UpdateOriginalLanguageIndex(language_index);
 }
 
 void TranslateInfoBarDelegate::UpdateTargetLanguageIndex(
     size_t language_index) {
   ui_delegate_.UpdateTargetLanguageIndex(language_index);
 }
 
 void TranslateInfoBarDelegate::Translate() {
   ui_delegate_.Translate();
 }
 
 void TranslateInfoBarDelegate::RevertTranslation() {
   ui_delegate_.RevertTranslation();
   infobar()->RemoveSelf();
 }
 
 void TranslateInfoBarDelegate::ReportLanguageDetectionError() {
-  TranslateManager* manager =
-      TranslateTabHelper::GetManagerFromWebContents(GetWebContents());
-  if (!manager)
-    return;
-  manager->ReportLanguageDetectionError();
+  if (translate_manager_)
+    translate_manager_->ReportLanguageDetectionError();
 }
 
 void TranslateInfoBarDelegate::TranslationDeclined() {
   ui_delegate_.TranslationDeclined(false);
 }
 
 bool TranslateInfoBarDelegate::IsTranslatableLanguageByPrefs() {
-  TranslateTabHelper* translate_tab_helper =
-      TranslateTabHelper::FromWebContents(GetWebContents());
-  scoped_ptr<TranslatePrefs> translate_prefs(
-      TranslateTabHelper::CreateTranslatePrefs(
-          translate_tab_helper->GetPrefs()));
+  TranslateClient* client = GetTranslateClient();
+  scoped_ptr<TranslatePrefs> translate_prefs(client->GetTranslatePrefs());
   TranslateAcceptLanguages* accept_languages =
-      translate_tab_helper->GetTranslateAcceptLanguages();
+      client->GetTranslateAcceptLanguages();
   return translate_prefs->CanTranslateLanguage(accept_languages,
                                                original_language_code());
 }
 
 void TranslateInfoBarDelegate::ToggleTranslatableLanguageByPrefs() {
   if (ui_delegate_.IsLanguageBlocked()) {
     ui_delegate_.SetLanguageBlocked(false);
   } else {
     ui_delegate_.SetLanguageBlocked(true);
     infobar()->RemoveSelf();
@@ skipping 83 matching lines @@
   return base::string16();
 }
 
 void TranslateInfoBarDelegate::MessageInfoBarButtonPressed() {
   DCHECK_EQ(translate::TRANSLATE_STEP_TRANSLATE_ERROR, step_);
   if (error_type_ == TranslateErrors::UNSUPPORTED_LANGUAGE) {
     RevertTranslation();
     return;
   }
   // This is the "Try again..." case.
-  TranslateManager* manager =
-      TranslateTabHelper::GetManagerFromWebContents(GetWebContents());
-  DCHECK(manager);
-  manager->TranslatePage(
+  DCHECK(translate_manager_);
+  translate_manager_->TranslatePage(
       original_language_code(), target_language_code(), false);
 }
 
 bool TranslateInfoBarDelegate::ShouldShowMessageInfoBarButton() {
   return !GetMessageInfoBarButtonText().empty();
 }
 
 bool TranslateInfoBarDelegate::ShouldShowNeverTranslateShortcut() {
   DCHECK_EQ(translate::TRANSLATE_STEP_BEFORE_TRANSLATE, step_);
-  return !GetWebContents()->GetBrowserContext()->IsOffTheRecord() &&
+  return !is_off_the_record_ &&
       (prefs_->GetTranslationDeniedCount(original_language_code()) >=
           kNeverTranslateMinCount);
 }
 
 bool TranslateInfoBarDelegate::ShouldShowAlwaysTranslateShortcut() {
   DCHECK_EQ(translate::TRANSLATE_STEP_BEFORE_TRANSLATE, step_);
-  return !GetWebContents()->GetBrowserContext()->IsOffTheRecord() &&
+  return !is_off_the_record_ &&
       (prefs_->GetTranslationAcceptedCount(original_language_code()) >=
           kAlwaysTranslateMinCount);
 }
 
 content::WebContents* TranslateInfoBarDelegate::GetWebContents() {
   return InfoBarService::WebContentsFromInfoBar(infobar());
 }
 
 // static
 void TranslateInfoBarDelegate::GetAfterTranslateStrings(
@@ skipping 24 matching lines @@
   *swap_languages = (offsets[0] > offsets[1]);
   if (*swap_languages)
     std::swap(offsets[0], offsets[1]);
 
   strings->push_back(text.substr(0, offsets[0]));
   strings->push_back(text.substr(offsets[0], offsets[1] - offsets[0]));
   strings->push_back(text.substr(offsets[1]));
 }
 
 TranslateInfoBarDelegate::TranslateInfoBarDelegate(
-    content::WebContents* web_contents,
+    const base::WeakPtr<TranslateManager>& translate_manager,
+    bool is_off_the_record,
     translate::TranslateStep step,
     TranslateInfoBarDelegate* old_delegate,
     const std::string& original_language,
     const std::string& target_language,
     TranslateErrors::Type error_type,
     PrefService* prefs,
     bool triggered_from_menu)
     : infobars::InfoBarDelegate(),
+      is_off_the_record_(is_off_the_record),
       step_(step),
       background_animation_(NONE),
-      ui_delegate_(TranslateTabHelper::FromWebContents(web_contents),
-                   TranslateTabHelper::GetManagerFromWebContents(web_contents),
+      ui_delegate_(translate_manager->translate_client(),
+                   translate_manager.get(),

[blundell, 2014/05/16 15:11:03]

hmm, is it ok that this guy holds on to a raw pointer? could address in followup
if it's actually a problem since you're not changing the existing situation.

[droger, 2014/05/16 15:14:11]

You're right, it's probably not OK conceptually.

                    original_language,
                    target_language),
+      translate_manager_(translate_manager),
       error_type_(error_type),
-      prefs_(TranslateTabHelper::CreateTranslatePrefs(prefs)),
+      prefs_(translate_manager->translate_client()->GetTranslatePrefs()),
       triggered_from_menu_(triggered_from_menu) {
   DCHECK_NE((step_ == translate::TRANSLATE_STEP_TRANSLATE_ERROR),
             (error_type_ == TranslateErrors::NONE));
+  DCHECK(translate_manager_);
 
   if (old_delegate && (old_delegate->is_error() != is_error()))
     background_animation_ = is_error() ? NORMAL_TO_ERROR : ERROR_TO_NORMAL;
 }
 
+TranslateClient* TranslateInfoBarDelegate::GetTranslateClient() {
+  if (!translate_manager_)
+    return NULL;
+
+  return translate_manager_->translate_client();
+}
+
 // TranslateInfoBarDelegate::CreateInfoBar() is implemented in platform-specific
 // files.
 
 void TranslateInfoBarDelegate::InfoBarDismissed() {
   if (step_ != translate::TRANSLATE_STEP_BEFORE_TRANSLATE)
     return;
 
   // The user closed the infobar without clicking the translate button.
   TranslationDeclined();
   UMA_HISTOGRAM_BOOLEAN("Translate.DeclineTranslateCloseInfobar", true);
@@ skipping 15 matching lines @@
   if (!details.is_navigation_to_different_page && !details.is_main_frame)
     return false;
 
   return infobars::InfoBarDelegate::ShouldExpireInternal(details);
 }
 
 TranslateInfoBarDelegate*
     TranslateInfoBarDelegate::AsTranslateInfoBarDelegate() {
   return this;
 }