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); |
+} |