| OLD | NEW |
| (Empty) |
| 1 // Copyright 2014 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 "components/translate/content/browser/content_translate_driver.h" | |
| 6 | |
| 7 #include "base/bind.h" | |
| 8 #include "base/location.h" | |
| 9 #include "base/logging.h" | |
| 10 #include "base/single_thread_task_runner.h" | |
| 11 #include "base/threading/thread_task_runner_handle.h" | |
| 12 #include "components/translate/content/common/translate_messages.h" | |
| 13 #include "components/translate/core/browser/translate_download_manager.h" | |
| 14 #include "components/translate/core/browser/translate_manager.h" | |
| 15 #include "content/public/browser/browser_context.h" | |
| 16 #include "content/public/browser/navigation_controller.h" | |
| 17 #include "content/public/browser/navigation_details.h" | |
| 18 #include "content/public/browser/navigation_entry.h" | |
| 19 #include "content/public/browser/page_navigator.h" | |
| 20 #include "content/public/browser/render_frame_host.h" | |
| 21 #include "content/public/browser/web_contents.h" | |
| 22 #include "content/public/common/referrer.h" | |
| 23 #include "net/http/http_status_code.h" | |
| 24 #include "url/gurl.h" | |
| 25 | |
| 26 namespace { | |
| 27 | |
| 28 // The maximum number of attempts we'll do to see if the page has finshed | |
| 29 // loading before giving up the translation | |
| 30 const int kMaxTranslateLoadCheckAttempts = 20; | |
| 31 | |
| 32 } // namespace | |
| 33 | |
| 34 namespace translate { | |
| 35 | |
| 36 ContentTranslateDriver::ContentTranslateDriver( | |
| 37 content::NavigationController* nav_controller) | |
| 38 : content::WebContentsObserver(nav_controller->GetWebContents()), | |
| 39 navigation_controller_(nav_controller), | |
| 40 translate_manager_(NULL), | |
| 41 max_reload_check_attempts_(kMaxTranslateLoadCheckAttempts), | |
| 42 weak_pointer_factory_(this) { | |
| 43 DCHECK(navigation_controller_); | |
| 44 } | |
| 45 | |
| 46 ContentTranslateDriver::~ContentTranslateDriver() {} | |
| 47 | |
| 48 void ContentTranslateDriver::AddObserver(Observer* observer) { | |
| 49 observer_list_.AddObserver(observer); | |
| 50 } | |
| 51 | |
| 52 void ContentTranslateDriver::RemoveObserver(Observer* observer) { | |
| 53 observer_list_.RemoveObserver(observer); | |
| 54 } | |
| 55 | |
| 56 void ContentTranslateDriver::InitiateTranslation(const std::string& page_lang, | |
| 57 int attempt) { | |
| 58 if (translate_manager_->GetLanguageState().translation_pending()) | |
| 59 return; | |
| 60 | |
| 61 // During a reload we need web content to be available before the | |
| 62 // translate script is executed. Otherwise we will run the translate script on | |
| 63 // an empty DOM which will fail. Therefore we wait a bit to see if the page | |
| 64 // has finished. | |
| 65 if (web_contents()->IsLoading() && attempt < max_reload_check_attempts_) { | |
| 66 int backoff = attempt * kMaxTranslateLoadCheckAttempts; | |
| 67 base::ThreadTaskRunnerHandle::Get()->PostDelayedTask( | |
| 68 FROM_HERE, | |
| 69 base::Bind(&ContentTranslateDriver::InitiateTranslation, | |
| 70 weak_pointer_factory_.GetWeakPtr(), page_lang, attempt + 1), | |
| 71 base::TimeDelta::FromMilliseconds(backoff)); | |
| 72 return; | |
| 73 } | |
| 74 | |
| 75 translate_manager_->InitiateTranslation( | |
| 76 translate::TranslateDownloadManager::GetLanguageCode(page_lang)); | |
| 77 } | |
| 78 | |
| 79 // TranslateDriver methods | |
| 80 | |
| 81 bool ContentTranslateDriver::IsLinkNavigation() { | |
| 82 return navigation_controller_ && | |
| 83 navigation_controller_->GetLastCommittedEntry() && | |
| 84 ui::PageTransitionCoreTypeIs( | |
| 85 navigation_controller_->GetLastCommittedEntry() | |
| 86 ->GetTransitionType(), | |
| 87 ui::PAGE_TRANSITION_LINK); | |
| 88 } | |
| 89 | |
| 90 void ContentTranslateDriver::OnTranslateEnabledChanged() { | |
| 91 content::WebContents* web_contents = navigation_controller_->GetWebContents(); | |
| 92 FOR_EACH_OBSERVER( | |
| 93 Observer, observer_list_, OnTranslateEnabledChanged(web_contents)); | |
| 94 } | |
| 95 | |
| 96 void ContentTranslateDriver::OnIsPageTranslatedChanged() { | |
| 97 content::WebContents* web_contents = | |
| 98 navigation_controller_->GetWebContents(); | |
| 99 FOR_EACH_OBSERVER( | |
| 100 Observer, observer_list_, OnIsPageTranslatedChanged(web_contents)); | |
| 101 } | |
| 102 | |
| 103 void ContentTranslateDriver::TranslatePage(int page_seq_no, | |
| 104 const std::string& translate_script, | |
| 105 const std::string& source_lang, | |
| 106 const std::string& target_lang) { | |
| 107 content::WebContents* web_contents = navigation_controller_->GetWebContents(); | |
| 108 web_contents->GetMainFrame()->Send(new ChromeFrameMsg_TranslatePage( | |
| 109 web_contents->GetMainFrame()->GetRoutingID(), page_seq_no, | |
| 110 translate_script, source_lang, target_lang)); | |
| 111 } | |
| 112 | |
| 113 void ContentTranslateDriver::RevertTranslation(int page_seq_no) { | |
| 114 content::WebContents* web_contents = navigation_controller_->GetWebContents(); | |
| 115 web_contents->GetMainFrame()->Send(new ChromeFrameMsg_RevertTranslation( | |
| 116 web_contents->GetMainFrame()->GetRoutingID(), page_seq_no)); | |
| 117 } | |
| 118 | |
| 119 bool ContentTranslateDriver::IsOffTheRecord() { | |
| 120 return navigation_controller_->GetBrowserContext()->IsOffTheRecord(); | |
| 121 } | |
| 122 | |
| 123 const std::string& ContentTranslateDriver::GetContentsMimeType() { | |
| 124 return navigation_controller_->GetWebContents()->GetContentsMimeType(); | |
| 125 } | |
| 126 | |
| 127 const GURL& ContentTranslateDriver::GetLastCommittedURL() { | |
| 128 return navigation_controller_->GetWebContents()->GetLastCommittedURL(); | |
| 129 } | |
| 130 | |
| 131 const GURL& ContentTranslateDriver::GetVisibleURL() { | |
| 132 return navigation_controller_->GetWebContents()->GetVisibleURL(); | |
| 133 } | |
| 134 | |
| 135 bool ContentTranslateDriver::HasCurrentPage() { | |
| 136 return (navigation_controller_->GetLastCommittedEntry() != NULL); | |
| 137 } | |
| 138 | |
| 139 void ContentTranslateDriver::OpenUrlInNewTab(const GURL& url) { | |
| 140 content::OpenURLParams params(url, | |
| 141 content::Referrer(), | |
| 142 NEW_FOREGROUND_TAB, | |
| 143 ui::PAGE_TRANSITION_LINK, | |
| 144 false); | |
| 145 navigation_controller_->GetWebContents()->OpenURL(params); | |
| 146 } | |
| 147 | |
| 148 // content::WebContentsObserver methods | |
| 149 | |
| 150 void ContentTranslateDriver::NavigationEntryCommitted( | |
| 151 const content::LoadCommittedDetails& load_details) { | |
| 152 // Check whether this is a reload: When doing a page reload, the | |
| 153 // TranslateLanguageDetermined IPC is not sent so the translation needs to be | |
| 154 // explicitly initiated. | |
| 155 | |
| 156 content::NavigationEntry* entry = | |
| 157 web_contents()->GetController().GetLastCommittedEntry(); | |
| 158 if (!entry) { | |
| 159 NOTREACHED(); | |
| 160 return; | |
| 161 } | |
| 162 | |
| 163 // If the navigation happened while offline don't show the translate | |
| 164 // bar since there will be nothing to translate. | |
| 165 if (load_details.http_status_code == 0 || | |
| 166 load_details.http_status_code == net::HTTP_INTERNAL_SERVER_ERROR) { | |
| 167 return; | |
| 168 } | |
| 169 | |
| 170 if (!load_details.is_main_frame && | |
| 171 translate_manager_->GetLanguageState().translation_declined()) { | |
| 172 // Some sites (such as Google map) may trigger sub-frame navigations | |
| 173 // when the user interacts with the page. We don't want to show a new | |
| 174 // infobar if the user already dismissed one in that case. | |
| 175 return; | |
| 176 } | |
| 177 | |
| 178 // If not a reload, return. | |
| 179 if (!ui::PageTransitionCoreTypeIs(entry->GetTransitionType(), | |
| 180 ui::PAGE_TRANSITION_RELOAD) && | |
| 181 load_details.type != content::NAVIGATION_TYPE_SAME_PAGE) { | |
| 182 return; | |
| 183 } | |
| 184 | |
| 185 if (!translate_manager_->GetLanguageState().page_needs_translation()) | |
| 186 return; | |
| 187 | |
| 188 // Note that we delay it as the ordering of the processing of this callback | |
| 189 // by WebContentsObservers is undefined and might result in the current | |
| 190 // infobars being removed. Since the translation initiation process might add | |
| 191 // an infobar, it must be done after that. | |
| 192 base::ThreadTaskRunnerHandle::Get()->PostTask( | |
| 193 FROM_HERE, | |
| 194 base::Bind(&ContentTranslateDriver::InitiateTranslation, | |
| 195 weak_pointer_factory_.GetWeakPtr(), | |
| 196 translate_manager_->GetLanguageState().original_language(), | |
| 197 0)); | |
| 198 } | |
| 199 | |
| 200 void ContentTranslateDriver::DidNavigateAnyFrame( | |
| 201 content::RenderFrameHost* render_frame_host, | |
| 202 const content::LoadCommittedDetails& details, | |
| 203 const content::FrameNavigateParams& params) { | |
| 204 // Let the LanguageState clear its state. | |
| 205 const bool reload = | |
| 206 ui::PageTransitionCoreTypeIs(details.entry->GetTransitionType(), | |
| 207 ui::PAGE_TRANSITION_RELOAD) || | |
| 208 details.type == content::NAVIGATION_TYPE_SAME_PAGE; | |
| 209 translate_manager_->GetLanguageState().DidNavigate( | |
| 210 details.is_in_page, details.is_main_frame, reload); | |
| 211 } | |
| 212 | |
| 213 bool ContentTranslateDriver::OnMessageReceived( | |
| 214 const IPC::Message& message, | |
| 215 content::RenderFrameHost* render_frame_host) { | |
| 216 bool handled = true; | |
| 217 IPC_BEGIN_MESSAGE_MAP(ContentTranslateDriver, message) | |
| 218 IPC_MESSAGE_HANDLER(ChromeFrameHostMsg_TranslateAssignedSequenceNumber, | |
| 219 OnTranslateAssignedSequenceNumber) | |
| 220 IPC_MESSAGE_HANDLER(ChromeFrameHostMsg_TranslateLanguageDetermined, | |
| 221 OnLanguageDetermined) | |
| 222 IPC_MESSAGE_HANDLER(ChromeFrameHostMsg_PageTranslated, OnPageTranslated) | |
| 223 IPC_MESSAGE_UNHANDLED(handled = false) | |
| 224 IPC_END_MESSAGE_MAP() | |
| 225 return handled; | |
| 226 } | |
| 227 | |
| 228 void ContentTranslateDriver::OnTranslateAssignedSequenceNumber( | |
| 229 int page_seq_no) { | |
| 230 translate_manager_->set_current_seq_no(page_seq_no); | |
| 231 } | |
| 232 | |
| 233 void ContentTranslateDriver::OnLanguageDetermined( | |
| 234 const LanguageDetectionDetails& details, | |
| 235 bool page_needs_translation) { | |
| 236 translate_manager_->GetLanguageState().LanguageDetermined( | |
| 237 details.adopted_language, page_needs_translation); | |
| 238 | |
| 239 if (web_contents()) | |
| 240 translate_manager_->InitiateTranslation(details.adopted_language); | |
| 241 | |
| 242 FOR_EACH_OBSERVER(Observer, observer_list_, OnLanguageDetermined(details)); | |
| 243 } | |
| 244 | |
| 245 void ContentTranslateDriver::OnPageTranslated( | |
| 246 const std::string& original_lang, | |
| 247 const std::string& translated_lang, | |
| 248 TranslateErrors::Type error_type) { | |
| 249 translate_manager_->PageTranslated( | |
| 250 original_lang, translated_lang, error_type); | |
| 251 FOR_EACH_OBSERVER( | |
| 252 Observer, | |
| 253 observer_list_, | |
| 254 OnPageTranslated(original_lang, translated_lang, error_type)); | |
| 255 } | |
| 256 | |
| 257 } // namespace translate | |
| OLD | NEW |