1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
// Copyright 2014 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 "chrome/browser/ui/pdf/chrome_pdf_web_contents_helper_client.h"
#include "chrome/browser/download/download_stats.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/location_bar/location_bar.h"
#include "chrome/browser/ui/pdf/pdf_unsupported_feature.h"
#include "chrome/browser/ui/tab_contents/core_tab_helper.h"
#include "extensions/browser/guest_view/mime_handler_view/mime_handler_view_guest.h"
namespace {
content::WebContents* GetWebContentsToUse(
content::WebContents* web_contents) {
// If we're viewing the PDF in a MimeHandlerViewGuest, use its embedder
// WebContents.
auto guest_view =
extensions::MimeHandlerViewGuest::FromWebContents(web_contents);
if (guest_view)
return guest_view->embedder_web_contents();
return web_contents;
}
} // namespace
ChromePDFWebContentsHelperClient::ChromePDFWebContentsHelperClient() {
}
ChromePDFWebContentsHelperClient::~ChromePDFWebContentsHelperClient() {
}
void ChromePDFWebContentsHelperClient::UpdateLocationBar(
content::WebContents* contents) {
Browser* browser = chrome::FindBrowserWithWebContents(contents);
if (!browser)
return;
BrowserWindow* window = browser->window();
if (!window)
return;
LocationBar* location_bar = window->GetLocationBar();
if (!location_bar)
return;
location_bar->UpdateOpenPDFInReaderPrompt();
}
void ChromePDFWebContentsHelperClient::UpdateContentRestrictions(
content::WebContents* contents,
int content_restrictions) {
CoreTabHelper* core_tab_helper =
CoreTabHelper::FromWebContents(GetWebContentsToUse(contents));
// |core_tab_helper| is NULL for WebViewGuest.
if (core_tab_helper)
core_tab_helper->UpdateContentRestrictions(content_restrictions);
}
void ChromePDFWebContentsHelperClient::OnPDFHasUnsupportedFeature(
content::WebContents* contents) {
PDFHasUnsupportedFeature(GetWebContentsToUse(contents));
}
void ChromePDFWebContentsHelperClient::OnSaveURL(
content::WebContents* contents) {
RecordDownloadSource(DOWNLOAD_INITIATED_BY_PDF_SAVE);
}
void ChromePDFWebContentsHelperClient::OnShowPDFPasswordDialog(
content::WebContents* contents,
const base::string16& prompt,
const pdf::PasswordDialogClosedCallback& callback) {
ShowPDFPasswordDialog(contents, prompt, callback);
}
|
