| // Copyright 2017 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. |
| |
| #ifndef COMPONENTS_PRINTING_BROWSER_PRINT_COMPOSITE_CLIENT_H_ |
| #define COMPONENTS_PRINTING_BROWSER_PRINT_COMPOSITE_CLIENT_H_ |
| |
| #include <map> |
| #include <memory> |
| |
| #include "base/optional.h" |
| #include "components/printing/service/public/cpp/pdf_service_mojo_types.h" |
| #include "components/printing/service/public/interfaces/pdf_compositor.mojom.h" |
| #include "content/public/browser/web_contents_observer.h" |
| #include "content/public/browser/web_contents_user_data.h" |
| #include "printing/common/pdf_metafile_utils.h" |
| #include "services/service_manager/public/cpp/connector.h" |
| |
| struct PrintHostMsg_DidPrintContent_Params; |
| |
| namespace printing { |
| |
| // Class to manage print requests and their communication with pdf |
| // compositor service. |
| // Each composite request have a separate interface pointer to connect |
| // with remote service. The request and its subframe printing results are |
| // tracked by its document cookie and print page number. |
| class PrintCompositeClient |
| : public content::WebContentsUserData<PrintCompositeClient>, |
| public content::WebContentsObserver { |
| public: |
| explicit PrintCompositeClient(content::WebContents* web_contents); |
| ~PrintCompositeClient() override; |
| |
| // content::WebContentsObserver |
| bool OnMessageReceived(const IPC::Message& message, |
| content::RenderFrameHost* render_frame_host) override; |
| |
| // IPC message handler. |
| void OnDidPrintFrameContent( |
| content::RenderFrameHost* render_frame_host, |
| int document_cookie, |
| const PrintHostMsg_DidPrintContent_Params& params); |
| |
| // Instructs the specified subframe to print. |
| void PrintCrossProcessSubframe(const gfx::Rect& rect, |
| int document_cookie, |
| content::RenderFrameHost* subframe_host); |
| |
| // NOTE: |handle| must be a READ-ONLY base::SharedMemoryHandle, i.e. one |
| // acquired by base::SharedMemory::GetReadOnlyHandle(). |
| |
| // Printing single pages is only used by print preview for early return of |
| // rendered results. In this case, the pages share the content with printed |
| // document. The entire document will always be printed and sent at the end. |
| // This is for compositing such a single preview page. |
| void DoCompositePageToPdf( |
| int cookie, |
| content::RenderFrameHost* render_frame_host, |
| int page_num, |
| base::SharedMemoryHandle handle, |
| uint32_t data_size, |
| const ContentToProxyIdMap& subframe_content_info, |
| mojom::PdfCompositor::CompositePageToPdfCallback callback); |
| |
| // Used for compositing the entire document for print preview or actual |
| // printing. |
| void DoCompositeDocumentToPdf( |
| int cookie, |
| content::RenderFrameHost* render_frame_host, |
| base::SharedMemoryHandle handle, |
| uint32_t data_size, |
| const ContentToProxyIdMap& subframe_content_info, |
| mojom::PdfCompositor::CompositeDocumentToPdfCallback callback); |
| |
| void set_for_preview(bool for_preview) { for_preview_ = for_preview; } |
| bool for_preview() const { return for_preview_; } |
| |
| // Converts a ContentToProxyIdMap to ContentToFrameMap. |
| // ContentToProxyIdMap maps content id to its corresponding render frame proxy |
| // routing id. This is generated when the content holder was created; |
| // ContentToFrameMap maps content id to its render frame's global unique id. |
| // The global unique id has the render process id concatenated with render |
| // frame routing id, which can uniquely identify a render frame. |
| static ContentToFrameMap ConvertContentInfoMap( |
| content::WebContents* web_contents, |
| content::RenderFrameHost* render_frame_host, |
| const ContentToProxyIdMap& content_proxy_map); |
| |
| private: |
| // Since page number is always non-negative, use this value to indicate it is |
| // for the whole document -- no page number specified. |
| static constexpr int kPageNumForWholeDoc = -1; |
| |
| // Callback functions for getting the replies. |
| void OnDidCompositePageToPdf( |
| printing::mojom::PdfCompositor::CompositePageToPdfCallback callback, |
| printing::mojom::PdfCompositor::Status status, |
| mojo::ScopedSharedBufferHandle handle); |
| |
| void OnDidCompositeDocumentToPdf( |
| int document_cookie, |
| printing::mojom::PdfCompositor::CompositeDocumentToPdfCallback callback, |
| printing::mojom::PdfCompositor::Status status, |
| mojo::ScopedSharedBufferHandle handle); |
| |
| // Get the request or create a new one if none exists. |
| // Since printed pages always share content with it document, they share the |
| // same composite request. |
| mojom::PdfCompositorPtr& GetCompositeRequest(int cookie); |
| |
| // Remove an existing request from |compositor_map_|. |
| void RemoveCompositeRequest(int cookie); |
| |
| mojom::PdfCompositorPtr CreateCompositeRequest(); |
| |
| // Whether this client is created for print preview dialog. |
| bool for_preview_; |
| |
| std::unique_ptr<service_manager::Connector> connector_; |
| |
| // Stores the mapping between document cookies and their corresponding |
| // requests. |
| std::map<int, mojom::PdfCompositorPtr> compositor_map_; |
| |
| DISALLOW_COPY_AND_ASSIGN(PrintCompositeClient); |
| }; |
| |
| } // namespace printing |
| |
| #endif // COMPONENTS_PRINTING_BROWSER_PRINT_COMPOSITE_CLIENT_H_ |