| Index: chrome/browser/ui/pdf/pdf_tab_helper.cc
|
| ===================================================================
|
| --- chrome/browser/ui/pdf/pdf_tab_helper.cc (revision 212906)
|
| +++ chrome/browser/ui/pdf/pdf_tab_helper.cc (working copy)
|
| @@ -4,12 +4,14 @@
|
|
|
| #include "chrome/browser/ui/pdf/pdf_tab_helper.h"
|
|
|
| +#include "chrome/browser/download/download_util.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"
|
|
|
| @@ -32,7 +34,10 @@
|
| bool handled = true;
|
| IPC_BEGIN_MESSAGE_MAP(PDFTabHelper, message)
|
| IPC_MESSAGE_HANDLER(ChromeViewHostMsg_PDFHasUnsupportedFeature,
|
| - OnPDFHasUnsupportedFeature)
|
| + OnHasUnsupportedFeature)
|
| + IPC_MESSAGE_HANDLER(ChromeViewHostMsg_PDFSaveURLAs, OnSaveURLAs)
|
| + IPC_MESSAGE_HANDLER(ChromeViewHostMsg_PDFUpdateContentRestrictions,
|
| + OnUpdateContentRestrictions)
|
| IPC_MESSAGE_UNHANDLED(handled = false)
|
| IPC_END_MESSAGE_MAP()
|
| return handled;
|
| @@ -64,6 +69,18 @@
|
| location_bar->UpdateOpenPDFInReaderPrompt();
|
| }
|
|
|
| -void PDFTabHelper::OnPDFHasUnsupportedFeature() {
|
| +void PDFTabHelper::OnHasUnsupportedFeature() {
|
| PDFHasUnsupportedFeature(web_contents());
|
| }
|
| +
|
| +void PDFTabHelper::OnSaveURLAs(const GURL& url,
|
| + const content::Referrer& referrer) {
|
| + download_util::RecordDownloadSource(download_util::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);
|
| +}
|
|
|