test: add -pdf-ready-to-print event to WebContents for testing (#43651)

Co-authored-by: trop[bot] <37223003+trop[bot]@users.noreply.github.com>
Co-authored-by: Shelley Vohr <shelley.vohr@gmail.com>
This commit is contained in:
trop[bot] 2024-09-09 21:56:16 -05:00 committed by GitHub
parent 4b5d29201c
commit 96fe03b200
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
5 changed files with 30 additions and 11 deletions

View file

@ -744,6 +744,7 @@ source_set("electron_lib") {
"//components/pdf/common:util", "//components/pdf/common:util",
"//components/pdf/renderer", "//components/pdf/renderer",
"//pdf", "//pdf",
"//pdf:content_restriction",
] ]
sources += [ sources += [
"shell/browser/electron_pdf_document_helper_client.cc", "shell/browser/electron_pdf_document_helper_client.cc",

View file

@ -3739,6 +3739,10 @@ void WebContents::OnInputEvent(const blink::WebInputEvent& event) {
Emit("input-event", event); Emit("input-event", event);
} }
void WebContents::PDFReadyToPrint() {
Emit("-pdf-ready-to-print");
}
void WebContents::RunJavaScriptDialog(content::WebContents* web_contents, void WebContents::RunJavaScriptDialog(content::WebContents* web_contents,
content::RenderFrameHost* rfh, content::RenderFrameHost* rfh,
content::JavaScriptDialogType dialog_type, content::JavaScriptDialogType dialog_type,

View file

@ -478,6 +478,8 @@ class WebContents : public ExclusiveAccessContext,
void SetBackgroundColor(std::optional<SkColor> color); void SetBackgroundColor(std::optional<SkColor> color);
void PDFReadyToPrint();
SkRegion* draggable_region() { SkRegion* draggable_region() {
return force_non_draggable_ ? nullptr : draggable_region_.get(); return force_non_draggable_ ? nullptr : draggable_region_.get();
} }

View file

@ -5,17 +5,29 @@
#include "shell/browser/electron_pdf_document_helper_client.h" #include "shell/browser/electron_pdf_document_helper_client.h"
#include "content/public/browser/web_contents.h" #include "content/public/browser/web_contents.h"
#include "pdf/content_restriction.h"
#include "shell/browser/api/electron_api_web_contents.h"
ElectronPDFDocumentHelperClient::ElectronPDFDocumentHelperClient() = default; ElectronPDFDocumentHelperClient::ElectronPDFDocumentHelperClient() = default;
ElectronPDFDocumentHelperClient::~ElectronPDFDocumentHelperClient() = default; ElectronPDFDocumentHelperClient::~ElectronPDFDocumentHelperClient() = default;
void ElectronPDFDocumentHelperClient::UpdateContentRestrictions( void ElectronPDFDocumentHelperClient::UpdateContentRestrictions(
content::RenderFrameHost* render_frame_host, content::RenderFrameHost* render_frame_host,
int content_restrictions) {} int content_restrictions) {
void ElectronPDFDocumentHelperClient::OnPDFHasUnsupportedFeature( // UpdateContentRestrictions potentially gets called twice from
content::WebContents* contents) {} // pdf/pdf_view_web_plugin.cc. The first time it is potentially called is
void ElectronPDFDocumentHelperClient::OnSaveURL( // when loading starts and it is called with a restriction on printing. The
content::WebContents* contents) {} // second time it is called is when loading is finished and if printing is
void ElectronPDFDocumentHelperClient::SetPluginCanSave( // allowed there won't be a printing restriction passed, so we can use this
content::RenderFrameHost* render_frame_host, // second call to notify that the pdf document is ready to print.
bool can_save) {} if (!(content_restrictions & chrome_pdf::kContentRestrictionPrint)) {
content::WebContents* web_contents =
content::WebContents::FromRenderFrameHost(render_frame_host);
electron::api::WebContents* api_web_contents =
electron::api::WebContents::From(
web_contents->GetOutermostWebContents());
if (api_web_contents) {
api_web_contents->PDFReadyToPrint();
}
}
}

View file

@ -19,10 +19,10 @@ class ElectronPDFDocumentHelperClient : public pdf::PDFDocumentHelperClient {
// pdf::PDFDocumentHelperClient // pdf::PDFDocumentHelperClient
void UpdateContentRestrictions(content::RenderFrameHost* render_frame_host, void UpdateContentRestrictions(content::RenderFrameHost* render_frame_host,
int content_restrictions) override; int content_restrictions) override;
void OnPDFHasUnsupportedFeature(content::WebContents* contents) override; void OnPDFHasUnsupportedFeature(content::WebContents* contents) override {}
void OnSaveURL(content::WebContents* contents) override; void OnSaveURL(content::WebContents* contents) override {}
void SetPluginCanSave(content::RenderFrameHost* render_frame_host, void SetPluginCanSave(content::RenderFrameHost* render_frame_host,
bool can_save) override; bool can_save) override {}
}; };
#endif // ELECTRON_SHELL_BROWSER_ELECTRON_PDF_DOCUMENT_HELPER_CLIENT_H_ #endif // ELECTRON_SHELL_BROWSER_ELECTRON_PDF_DOCUMENT_HELPER_CLIENT_H_