chrome/browser/tab_contents/language_state.cc

   1 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "chrome/browser/tab_contents/language_state.h"
   6
   7 #include "chrome/browser/tab_contents/language_state_observer.h"
   8 #include "content/public/browser/navigation_controller.h"
   9 #include "content/public/browser/navigation_details.h"
  10 #include "content/public/browser/navigation_entry.h"
  11 #include "content/public/browser/web_contents.h"
  12
  13 using content::NavigationController;
  14
  15 LanguageState::LanguageState(NavigationController* nav_controller)
  16     : is_page_translated_(false),
  17       navigation_controller_(nav_controller),
  18       page_needs_translation_(false),
  19       translation_pending_(false),
  20       translation_declined_(false),
  21       in_page_navigation_(false),
  22       translate_enabled_(false),
  23       observer_(NULL) {
  24 }
  25
  26 LanguageState::~LanguageState() {
  27 }
  28
  29 void LanguageState::DidNavigate(
  30     const content::LoadCommittedDetails& details) {
  31   in_page_navigation_ = details.is_in_page;
  32   if (in_page_navigation_ || !details.is_main_frame)
  33     return;  // Don't reset our states, the page has not changed.
  34
  35   bool reload =
  36       details.entry->GetTransitionType() == content::PAGE_TRANSITION_RELOAD ||
  37       details.type == content::NAVIGATION_TYPE_SAME_PAGE;
  38   if (reload) {
  39     // We might not get a LanguageDetermined notifications on reloads. Make sure
  40     // to keep the original language and to set current_lang_ so
  41     // IsPageTranslated() returns false.
  42     current_lang_ = original_lang_;
  43   } else {
  44     prev_original_lang_ = original_lang_;
  45     prev_current_lang_ = current_lang_;
  46     original_lang_.clear();
  47     current_lang_.clear();
  48   }
  49
  50   SetIsPageTranslated(false);
  51
  52   translation_pending_ = false;
  53   translation_declined_ = false;
  54
  55   SetTranslateEnabled(false);
  56 }
  57
  58 void LanguageState::LanguageDetermined(const std::string& page_language,
  59                                        bool page_needs_translation) {
  60   if (in_page_navigation_ && !original_lang_.empty()) {
  61     // In-page navigation, we don't expect our states to change.
  62     // Note that we'll set the languages if original_lang_ is empty.  This might
  63     // happen if the we did not get called on the top-page.
  64     return;
  65   }
  66   page_needs_translation_ = page_needs_translation;
  67   original_lang_ = page_language;
  68   current_lang_ = page_language;
  69   SetIsPageTranslated(false);
  70 }
  71
  72 bool LanguageState::InTranslateNavigation() const {
  73   // The user is in the same translate session if
  74   //   - no translation is pending
  75   //   - this page is in the same language as the previous page
  76   //   - the previous page had been translated
  77   //   - the new page was navigated through a link.
  78   return
  79       !translation_pending_ &&
  80       prev_original_lang_ == original_lang_ &&
  81       prev_original_lang_ != prev_current_lang_ &&
  82       navigation_controller_ &&
  83       navigation_controller_->GetActiveEntry() &&
  84       navigation_controller_->GetActiveEntry()->GetTransitionType() ==
  85           content::PAGE_TRANSITION_LINK;
  86 }
  87
  88 void LanguageState::SetCurrentLanguage(const std::string& language) {
  89   current_lang_ = language;
  90   SetIsPageTranslated(current_lang_ != original_lang_);
  91 }
  92
  93 std::string LanguageState::AutoTranslateTo() const {
  94   if (InTranslateNavigation() && !is_page_translated_)
  95     return prev_current_lang_;
  96
  97   return std::string();
  98 }
  99
 100 void LanguageState::SetTranslateEnabled(bool value) {
 101   if (translate_enabled_ == value)
 102     return;
 103
 104   translate_enabled_ = value;
 105   if (observer_) {
 106     content::WebContents* web_contents =
 107         navigation_controller_->GetWebContents();
 108     observer_->OnTranslateEnabledChanged(web_contents);
 109   }
 110 }
 111
 112 bool LanguageState::HasLanguageChanged() const {
 113   return original_lang_ != prev_original_lang_;
 114 }
 115
 116 void LanguageState::SetIsPageTranslated(bool value) {
 117   if (is_page_translated_ == value)
 118     return;
 119
 120   is_page_translated_ = value;
 121   if (observer_) {
 122     content::WebContents* web_contents =
 123         navigation_controller_->GetWebContents();
 124     observer_->OnIsPageTranslatedChanged(web_contents);
 125   }
 126
 127   // With the translation done, the translate feature must be enabled.
 128   if (is_page_translated_)
 129     SetTranslateEnabled(true);
 130 }