mirrors/electron
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages 1 Wiki Activity Actions 1

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

Browse source
This commit is contained in:
John Kleinschmidt 2024-08-23 13:34:26 -04:00 • committed by GitHub
parent 9ce0ca74c3
commit c1eee18e41
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
6 changed files with 34 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

23
shell/browser/electron_pdf_document_helper_client.cc
View file

@ -5,6 +5,29 @@
#include "shell/browser/electron_pdf_document_helper_client.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;
void ElectronPDFDocumentHelperClient::UpdateContentRestrictions(
content::RenderFrameHost* render_frame_host,
int content_restrictions) {
// UpdateContentRestrictions potentially gets called twice from
// pdf/pdf_view_web_plugin.cc. The first time it is potentially called is
// when loading starts and it is called with a restriction on printing. The
// second time it is called is when loading is finished and if printing is
// allowed there won't be a printing restriction passed, so we can use this
// second call to notify that the pdf document is ready to print.
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();
}
}
}