components/pdf/browser/pdf_web_contents_helper.cc

   1 // Copyright (c) 2012 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/pdf/browser/pdf_web_contents_helper.h"
   6
   7 #include "base/bind.h"
   8 #include "base/strings/utf_string_conversions.h"
   9 #include "components/pdf/browser/open_pdf_in_reader_prompt_client.h"
  10 #include "components/pdf/browser/pdf_web_contents_helper_client.h"
  11 #include "components/pdf/common/pdf_messages.h"
  12 #include "content/public/browser/navigation_details.h"
  13
  14 DEFINE_WEB_CONTENTS_USER_DATA_KEY(pdf::PDFWebContentsHelper);
  15
  16 namespace pdf {
  17
  18 // static
  19 void PDFWebContentsHelper::CreateForWebContentsWithClient(
  20     content::WebContents* contents,
  21     scoped_ptr<PDFWebContentsHelperClient> client) {
  22   if (FromWebContents(contents))
  23     return;
  24   contents->SetUserData(UserDataKey(),
  25                         new PDFWebContentsHelper(contents, client.Pass()));
  26 }
  27
  28 PDFWebContentsHelper::PDFWebContentsHelper(
  29     content::WebContents* web_contents,
  30     scoped_ptr<PDFWebContentsHelperClient> client)
  31     : content::WebContentsObserver(web_contents), client_(client.Pass()) {
  32 }
  33
  34 PDFWebContentsHelper::~PDFWebContentsHelper() {
  35 }
  36
  37 void PDFWebContentsHelper::ShowOpenInReaderPrompt(
  38     scoped_ptr<OpenPDFInReaderPromptClient> prompt) {
  39   open_in_reader_prompt_ = prompt.Pass();
  40   UpdateLocationBar();
  41 }
  42
  43 bool PDFWebContentsHelper::OnMessageReceived(const IPC::Message& message) {
  44   bool handled = true;
  45   IPC_BEGIN_MESSAGE_MAP(PDFWebContentsHelper, message)
  46     IPC_MESSAGE_HANDLER(PDFHostMsg_PDFHasUnsupportedFeature,
  47                         OnHasUnsupportedFeature)
  48     IPC_MESSAGE_HANDLER(PDFHostMsg_PDFSaveURLAs, OnSaveURLAs)
  49     IPC_MESSAGE_HANDLER(PDFHostMsg_PDFUpdateContentRestrictions,
  50                         OnUpdateContentRestrictions)
  51     IPC_MESSAGE_HANDLER_DELAY_REPLY(PDFHostMsg_PDFModalPromptForPassword,
  52                                     OnModalPromptForPassword)
  53     IPC_MESSAGE_UNHANDLED(handled = false)
  54   IPC_END_MESSAGE_MAP()
  55   return handled;
  56 }
  57
  58 void PDFWebContentsHelper::DidNavigateMainFrame(
  59     const content::LoadCommittedDetails& details,
  60     const content::FrameNavigateParams& params) {
  61   if (open_in_reader_prompt_.get() &&
  62       open_in_reader_prompt_->ShouldExpire(details)) {
  63     open_in_reader_prompt_.reset();
  64     UpdateLocationBar();
  65   }
  66 }
  67
  68 void PDFWebContentsHelper::UpdateLocationBar() {
  69   client_->UpdateLocationBar(web_contents());
  70 }
  71
  72 void PDFWebContentsHelper::OnHasUnsupportedFeature() {
  73   client_->OnPDFHasUnsupportedFeature(web_contents());
  74 }
  75
  76 void PDFWebContentsHelper::OnSaveURLAs(const GURL& url,
  77                                        const content::Referrer& referrer) {
  78   client_->OnSaveURL(web_contents());
  79   web_contents()->SaveFrame(url, referrer);
  80 }
  81
  82 void PDFWebContentsHelper::OnUpdateContentRestrictions(
  83     int content_restrictions) {
  84   client_->UpdateContentRestrictions(web_contents(), content_restrictions);
  85 }
  86
  87 void PDFWebContentsHelper::OnModalPromptForPasswordClosed(
  88     IPC::Message* reply_message,
  89     bool success,
  90     const base::string16& actual_value) {
  91   PDFHostMsg_PDFModalPromptForPassword::WriteReplyParams(
  92       reply_message, base::UTF16ToUTF8(actual_value));
  93   Send(reply_message);
  94 }
  95
  96 void PDFWebContentsHelper::OnModalPromptForPassword(
  97     const std::string& prompt,
  98     IPC::Message* reply_message) {
  99   base::Callback<void(bool, const base::string16&)> callback =
 100       base::Bind(&PDFWebContentsHelper::OnModalPromptForPasswordClosed,
 101                  base::Unretained(this),
 102                  reply_message);
 103   client_->OnShowPDFPasswordDialog(
 104       web_contents(), base::UTF8ToUTF16(prompt), callback);
 105 }
 106
 107 }  // namespace pdf