| 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..59a17a88180cdf4471062aad0d4b028c3b598116
|
| --- /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,
|
| + 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
|
|
|