Upstream //ios/chrome/browser/translate

ios/chrome/browser/translate/chrome_ios_translate_client.mm

+// Copyright 2014 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 "ios/chrome/browser/translate/chrome_ios_translate_client.h"
+
+#include <vector>
+
+#include "base/logging.h"
+#include "base/prefs/pref_service.h"
+#include "components/infobars/core/infobar.h"
+#include "components/translate/core/browser/page_translated_details.h"
+#include "components/translate/core/browser/translate_accept_languages.h"
+#include "components/translate/core/browser/translate_infobar_delegate.h"
+#include "components/translate/core/browser/translate_manager.h"
+#include "components/translate/core/browser/translate_prefs.h"
+#include "components/translate/core/browser/translate_step.h"
+#include "ios/chrome/browser/infobars/infobar.h"
+#include "ios/chrome/browser/infobars/infobar_controller.h"
+#include "ios/chrome/browser/pref_names.h"
+#import "ios/chrome/browser/translate/after_translate_infobar_controller.h"
+#import "ios/chrome/browser/translate/before_translate_infobar_controller.h"
+#import "ios/chrome/browser/translate/never_translate_infobar_controller.h"
+#include "ios/chrome/browser/translate/translate_accept_languages_factory.h"
+#import "ios/chrome/browser/translate/translate_message_infobar_controller.h"
+#include "ios/chrome/browser/translate/translate_service_ios.h"
+#include "ios/chrome/grit/ios_theme_resources.h"
+#include "ios/public/provider/chrome/browser/browser_state/chrome_browser_state.h"
+#include "ios/public/provider/chrome/browser/chrome_browser_provider.h"
+#include "ios/web/public/browser_state.h"
+#include "ios/web/public/web_state/web_state.h"
+#include "url/gurl.h"
+
+DEFINE_WEB_STATE_USER_DATA_KEY(ChromeIOSTranslateClient);
+
+ChromeIOSTranslateClient::ChromeIOSTranslateClient(web::WebState* web_state)
+    : web::WebStateObserver(web_state),
+      translate_manager_(
+          new translate::TranslateManager(this, ios::prefs::kAcceptLanguages)),
+      translate_driver_(web_state,
+                        web_state->GetNavigationManager(),
+                        translate_manager_.get()) {
+}
+
+ChromeIOSTranslateClient::~ChromeIOSTranslateClient() {
+}
+
+// static
+scoped_ptr<translate::TranslatePrefs>
+ChromeIOSTranslateClient::CreateTranslatePrefs(PrefService* prefs) {
+  return scoped_ptr<translate::TranslatePrefs>(new translate::TranslatePrefs(
+      prefs, ios::prefs::kAcceptLanguages, nullptr));
+}
+
+translate::TranslateManager* ChromeIOSTranslateClient::GetTranslateManager() {
+  return translate_manager_.get();
+}
+
+// TranslateClient implementation:
+
+scoped_ptr<infobars::InfoBar> ChromeIOSTranslateClient::CreateInfoBar(
+    scoped_ptr<translate::TranslateInfoBarDelegate> delegate) const {
+  translate::TranslateStep step = delegate->translate_step();
+
+  scoped_ptr<InfoBarIOS> infobar(new InfoBarIOS(delegate.Pass()));
+  base::scoped_nsobject<InfoBarController> controller;
+  switch (step) {
+    case translate::TRANSLATE_STEP_AFTER_TRANSLATE:
+      controller.reset([[AfterTranslateInfoBarController alloc]
+          initWithDelegate:infobar.get()]);
+      break;
+    case translate::TRANSLATE_STEP_BEFORE_TRANSLATE:
+      controller.reset([[BeforeTranslateInfoBarController alloc]
+          initWithDelegate:infobar.get()]);
+      break;
+    case translate::TRANSLATE_STEP_NEVER_TRANSLATE:
+      controller.reset([[NeverTranslateInfoBarController alloc]
+          initWithDelegate:infobar.get()]);
+      break;
+    case translate::TRANSLATE_STEP_TRANSLATING:
+    case translate::TRANSLATE_STEP_TRANSLATE_ERROR:
+      controller.reset([[TranslateMessageInfoBarController alloc]
+          initWithDelegate:infobar.get()]);
+      break;
+    default:
+      NOTREACHED();
+  }
+  infobar->SetController(controller);
+  return infobar.Pass();
+}
+
+void ChromeIOSTranslateClient::ShowTranslateUI(
+    translate::TranslateStep step,
+    const std::string source_language,
+    const std::string target_language,
+    translate::TranslateErrors::Type error_type,
+    bool triggered_from_menu) {
+  DCHECK(web_state());
+  if (error_type != translate::TranslateErrors::NONE)
+    step = translate::TRANSLATE_STEP_TRANSLATE_ERROR;
+
+  // Infobar UI.
+  translate::TranslateInfoBarDelegate::Create(
+      step != translate::TRANSLATE_STEP_BEFORE_TRANSLATE,
+      translate_manager_->GetWeakPtr(),
+      ios::GetChromeBrowserProvider()->GetInfoBarManager(web_state()),
+      web_state()->GetBrowserState()->IsOffTheRecord(), step, source_language,
+      target_language, error_type, triggered_from_menu);
+}
+
+translate::TranslateDriver* ChromeIOSTranslateClient::GetTranslateDriver() {
+  return &translate_driver_;
+}
+
+PrefService* ChromeIOSTranslateClient::GetPrefs() {
+  DCHECK(web_state());
+  ios::ChromeBrowserState* chrome_browser_state =
+      ios::ChromeBrowserState::FromBrowserState(web_state()->GetBrowserState());
+  return chrome_browser_state->GetOriginalChromeBrowserState()->GetPrefs();
+}
+
+scoped_ptr<translate::TranslatePrefs>
+ChromeIOSTranslateClient::GetTranslatePrefs() {
+  DCHECK(web_state());
+  ios::ChromeBrowserState* chrome_browser_state =
+      ios::ChromeBrowserState::FromBrowserState(web_state()->GetBrowserState());
+  return CreateTranslatePrefs(chrome_browser_state->GetPrefs());
+}
+
+translate::TranslateAcceptLanguages*
+ChromeIOSTranslateClient::GetTranslateAcceptLanguages() {
+  DCHECK(web_state());
+  return TranslateAcceptLanguagesFactory::GetForBrowserState(
+      web_state()->GetBrowserState());
+}
+
+int ChromeIOSTranslateClient::GetInfobarIconID() const {
+  return IDR_INFOBAR_TRANSLATE_IOS;
+}
+
+bool ChromeIOSTranslateClient::IsTranslatableURL(const GURL& url) {
+  return TranslateServiceIOS::IsTranslatableURL(url);
+}
+
+void ChromeIOSTranslateClient::ShowReportLanguageDetectionErrorUI(
+    const GURL& report_url) {
+  NOTREACHED();
+}
+
+void ChromeIOSTranslateClient::WebStateDestroyed() {
+  // Translation process can be interrupted.
+  // Destroying the TranslateManager now guarantees that it never has to deal
+  // with nullptr WebState.
+  translate_manager_.reset();
+}