Chromium Code Reviews| Index: components/pdf/browser/pdf_web_contents_helper.cc |
| diff --git a/components/pdf/browser/pdf_web_contents_helper.cc b/components/pdf/browser/pdf_web_contents_helper.cc |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..342564e043e3adeb28b0337546a7b48637f98de2 |
| --- /dev/null |
| +++ b/components/pdf/browser/pdf_web_contents_helper.cc |
| @@ -0,0 +1,107 @@ |
| +// 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 "components/pdf/browser/pdf_web_contents_helper.h" |
| + |
| +#include "base/bind.h" |
| +#include "base/strings/utf_string_conversions.h" |
| +#include "components/pdf/browser/open_pdf_in_reader_prompt_client.h" |
| +#include "components/pdf/browser/pdf_web_contents_helper_client.h" |
| +#include "components/pdf/common/pdf_messages.h" |
| +#include "content/public/browser/navigation_details.h" |
| + |
| +DEFINE_WEB_CONTENTS_USER_DATA_KEY(pdf::PDFWebContentsHelper); |
| + |
| +namespace pdf { |
| + |
| +// static |
| +void PDFWebContentsHelper::CreateForWebContentsWithClient( |
| + content::WebContents* contents, |
| + scoped_ptr<PDFWebContentsHelperClient> client) { |
| + if (FromWebContents(contents)) |
| + return; |
| + contents->SetUserData(UserDataKey(), |
| + new PDFWebContentsHelper(contents, client.Pass())); |
| +} |
| + |
| +PDFWebContentsHelper::PDFWebContentsHelper( |
| + content::WebContents* web_contents, |
| + scoped_ptr<PDFWebContentsHelperClient> client) |
| + : content::WebContentsObserver(web_contents), client_(client.Pass()) { |
| +} |
| + |
| +PDFWebContentsHelper::~PDFWebContentsHelper() { |
| +} |
| + |
| +void PDFWebContentsHelper::ShowOpenInReaderPrompt( |
| + scoped_ptr<OpenPDFInReaderPromptClient> prompt) { |
| + open_in_reader_prompt_ = prompt.Pass(); |
| + UpdateLocationBar(); |
| +} |
| + |
| +bool PDFWebContentsHelper::OnMessageReceived(const IPC::Message& message) { |
| + bool handled = true; |
| + IPC_BEGIN_MESSAGE_MAP(PDFWebContentsHelper, message) |
| + IPC_MESSAGE_HANDLER(PDFHostMsg_PDFHasUnsupportedFeature, |
|
Tom Sepez
2014/08/27 17:24:49
nit: generally, we indent the IPC_MESSAGE_HANDLER/
sadrul
2014/08/27 23:39:20
Done, sorry. I addressed this in an earlier patchs
|
| + OnHasUnsupportedFeature) |
| + IPC_MESSAGE_HANDLER(PDFHostMsg_PDFSaveURLAs, OnSaveURLAs) |
| + IPC_MESSAGE_HANDLER(PDFHostMsg_PDFUpdateContentRestrictions, |
| + OnUpdateContentRestrictions) |
| + IPC_MESSAGE_HANDLER_DELAY_REPLY(PDFHostMsg_PDFModalPromptForPassword, |
| + OnModalPromptForPassword) |
| + IPC_MESSAGE_UNHANDLED(handled = false) |
| + IPC_END_MESSAGE_MAP() |
| + return handled; |
| +} |
| + |
| +void PDFWebContentsHelper::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 PDFWebContentsHelper::UpdateLocationBar() { |
| + client_->UpdateLocationBar(web_contents()); |
| +} |
| + |
| +void PDFWebContentsHelper::OnHasUnsupportedFeature() { |
| + client_->OnPDFHasUnsupportedFeature(web_contents()); |
| +} |
| + |
| +void PDFWebContentsHelper::OnSaveURLAs(const GURL& url, |
| + const content::Referrer& referrer) { |
| + client_->OnSaveURL(web_contents()); |
| + web_contents()->SaveFrame(url, referrer); |
| +} |
| + |
| +void PDFWebContentsHelper::OnUpdateContentRestrictions( |
| + int content_restrictions) { |
| + client_->UpdateContentRestrictions(web_contents(), content_restrictions); |
| +} |
| + |
| +void PDFWebContentsHelper::OnModalPromptForPasswordClosed( |
| + IPC::Message* reply_message, |
| + bool success, |
| + const base::string16& actual_value) { |
| + PDFHostMsg_PDFModalPromptForPassword::WriteReplyParams( |
| + reply_message, base::UTF16ToUTF8(actual_value)); |
| + Send(reply_message); |
| +} |
| + |
| +void PDFWebContentsHelper::OnModalPromptForPassword( |
| + const std::string& prompt, |
| + IPC::Message* reply_message) { |
| + base::Callback<void(bool, const base::string16&)> callback = |
| + base::Bind(&PDFWebContentsHelper::OnModalPromptForPasswordClosed, |
| + base::Unretained(this), |
| + reply_message); |
| + client_->OnShowPDFPasswordDialog( |
| + web_contents(), base::UTF8ToUTF16(prompt), callback); |
| +} |
| + |
| +} // namespace pdf |