| // 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/ui/pdf/pdf_tab_helper.h" |
| |
| #include "base/strings/utf_string_conversions.h" |
| #include "chrome/browser/download/download_stats.h" |
| #include "chrome/browser/ui/browser.h" |
| #include "chrome/browser/ui/browser_finder.h" |
| #include "chrome/browser/ui/browser_window.h" |
| #include "chrome/browser/ui/omnibox/location_bar.h" |
| #include "chrome/browser/ui/pdf/open_pdf_in_reader_prompt_delegate.h" |
| #include "chrome/browser/ui/pdf/pdf_unsupported_feature.h" |
| #include "chrome/browser/ui/tab_contents/core_tab_helper.h" |
| #include "chrome/common/render_messages.h" |
| #include "content/public/browser/navigation_details.h" |
| |
| #if defined(TOOLKIT_GTK) |
| #include "chrome/browser/ui/app_modal_dialogs/javascript_dialog_manager.h" |
| #include "content/public/browser/javascript_dialog_manager.h" |
| #endif |
| |
| DEFINE_WEB_CONTENTS_USER_DATA_KEY(PDFTabHelper); |
| |
| PDFTabHelper::PDFTabHelper(content::WebContents* web_contents) |
| : content::WebContentsObserver(web_contents) { |
| } |
| |
| PDFTabHelper::~PDFTabHelper() { |
| } |
| |
| void PDFTabHelper::ShowOpenInReaderPrompt( |
| scoped_ptr<OpenPDFInReaderPromptDelegate> prompt) { |
| open_in_reader_prompt_ = prompt.Pass(); |
| UpdateLocationBar(); |
| } |
| |
| bool PDFTabHelper::OnMessageReceived(const IPC::Message& message) { |
| bool handled = true; |
| IPC_BEGIN_MESSAGE_MAP(PDFTabHelper, message) |
| IPC_MESSAGE_HANDLER(ChromeViewHostMsg_PDFHasUnsupportedFeature, |
| OnHasUnsupportedFeature) |
| IPC_MESSAGE_HANDLER(ChromeViewHostMsg_PDFSaveURLAs, OnSaveURLAs) |
| IPC_MESSAGE_HANDLER(ChromeViewHostMsg_PDFUpdateContentRestrictions, |
| OnUpdateContentRestrictions) |
| IPC_MESSAGE_HANDLER_DELAY_REPLY(ChromeViewHostMsg_PDFModalPromptForPassword, |
| OnModalPromptForPassword) |
| IPC_MESSAGE_UNHANDLED(handled = false) |
| IPC_END_MESSAGE_MAP() |
| return handled; |
| } |
| |
| void PDFTabHelper::DidNavigateMainFrame( |
| const content::LoadCommittedDetails& details, |
| const content::FrameNavigateParams& params) { |
| if (open_in_reader_prompt_.get() && |
| open_in_reader_prompt_->ShouldExpire(details)) { |
| open_in_reader_prompt_.reset(); |
| UpdateLocationBar(); |
| } |
| } |
| |
| void PDFTabHelper::UpdateLocationBar() { |
| Browser* browser = chrome::FindBrowserWithWebContents(web_contents()); |
| if (!browser) |
| return; |
| |
| BrowserWindow* window = browser->window(); |
| if (!window) |
| return; |
| |
| LocationBar* location_bar = window->GetLocationBar(); |
| if (!location_bar) |
| return; |
| |
| location_bar->UpdateOpenPDFInReaderPrompt(); |
| } |
| |
| void PDFTabHelper::OnHasUnsupportedFeature() { |
| PDFHasUnsupportedFeature(web_contents()); |
| } |
| |
| void PDFTabHelper::OnSaveURLAs(const GURL& url, |
| const content::Referrer& referrer) { |
| RecordDownloadSource(DOWNLOAD_INITIATED_BY_PDF_SAVE); |
| web_contents()->SaveFrame(url, referrer); |
| } |
| |
| void PDFTabHelper::OnUpdateContentRestrictions(int content_restrictions) { |
| CoreTabHelper* core_tab_helper = |
| CoreTabHelper::FromWebContents(web_contents()); |
| core_tab_helper->UpdateContentRestrictions(content_restrictions); |
| } |
| |
| void PDFTabHelper::OnModalPromptForPasswordClosed( |
| IPC::Message* reply_message, |
| bool success, |
| const base::string16& actual_value) { |
| ChromeViewHostMsg_PDFModalPromptForPassword::WriteReplyParams( |
| reply_message, UTF16ToUTF8(actual_value)); |
| Send(reply_message); |
| } |
| |
| void PDFTabHelper::OnModalPromptForPassword(const std::string& prompt, |
| IPC::Message* reply_message) { |
| base::Callback<void(bool, const base::string16&)> callback = |
| base::Bind(&PDFTabHelper::OnModalPromptForPasswordClosed, |
| base::Unretained(this), reply_message); |
| #if !defined(TOOLKIT_GTK) |
| ShowPDFPasswordDialog(web_contents(), base::UTF8ToUTF16(prompt), callback); |
| #else |
| // GTK is going away, so it's not worth the effort to create a password dialog |
| // for it. Cheat (for now) until the GTK code is removed. |
| bool did_suppress_message; |
| GetJavaScriptDialogManagerInstance()->RunJavaScriptDialog( |
| web_contents(), |
| GURL(), |
| std::string(), |
| content::JAVASCRIPT_MESSAGE_TYPE_PROMPT, |
| base::UTF8ToUTF16(prompt), |
| base::string16(), |
| callback, |
| &did_suppress_message); |
| #endif // TOOLKIT_GTK |
| } |