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:
		
					parent
					
						
							
								4b5d29201c
							
						
					
				
			
			
				commit
				
					
						96fe03b200
					
				
			
		
					 5 changed files with 30 additions and 11 deletions
				
			
		
							
								
								
									
										1
									
								
								BUILD.gn
									
										
									
									
									
								
							
							
						
						
									
										1
									
								
								BUILD.gn
									
										
									
									
									
								
							|  | @ -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", | ||||||
|  |  | ||||||
|  | @ -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, | ||||||
|  |  | ||||||
|  | @ -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(); | ||||||
|   } |   } | ||||||
|  |  | ||||||
|  | @ -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(); | ||||||
|  |     } | ||||||
|  |   } | ||||||
|  | } | ||||||
|  |  | ||||||
|  | @ -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_
 | ||||||
|  |  | ||||||
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue
	
	![37223003+trop[bot]@users.noreply.github.com](/assets/img/avatar_default.png) trop[bot]
				trop[bot]