android_webview/browser/renderer_host/print_manager.cc

   1 // Copyright 2013 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "android_webview/browser/renderer_host/print_manager.h"
   6
   7 #include "android_webview/browser/aw_browser_context.h"
   8 #include "android_webview/common/render_view_messages.h"
   9 #include "base/android/scoped_java_ref.h"
  10 #include "base/callback.h"
  11 #include "base/files/file_util.h"
  12 #include "base/logging.h"
  13 #include "components/printing/common/print_messages.h"
  14 #include "content/public/browser/browser_thread.h"
  15 #include "printing/print_settings.h"
  16
  17 using content::BrowserThread;
  18 using printing::PrintSettings;
  19
  20 DEFINE_WEB_CONTENTS_USER_DATA_KEY(android_webview::PrintManager);
  21
  22 namespace android_webview {
  23
  24 // static
  25 PrintManager* PrintManager::CreateForWebContents(
  26     content::WebContents* contents,
  27     PrintSettings* settings,
  28     int fd,
  29     PrintManagerDelegate* delegate) {
  30   PrintManager* print_manager =
  31       new PrintManager(contents, settings, fd, delegate);
  32   contents->SetUserData(UserDataKey(), print_manager);
  33   return print_manager;
  34 }
  35
  36 PrintManager::PrintManager(content::WebContents* contents,
  37                            PrintSettings* settings,
  38                            int fd,
  39                            PrintManagerDelegate* delegate)
  40     : content::WebContentsObserver(contents),
  41       settings_(settings),
  42       fd_(fd),
  43       delegate_(delegate),
  44       number_pages_(0),
  45       cookie_(1),
  46       printing_(false) {
  47   DCHECK(delegate_);
  48 }
  49
  50 PrintManager::~PrintManager() {}
  51
  52 bool PrintManager::OnMessageReceived(const IPC::Message& message) {
  53   bool handled = true;
  54   IPC_BEGIN_MESSAGE_MAP(PrintManager, message)
  55     IPC_MESSAGE_HANDLER(PrintHostMsg_DidGetPrintedPagesCount,
  56                         OnDidGetPrintedPagesCount)
  57     IPC_MESSAGE_HANDLER(PrintHostMsg_DidGetDocumentCookie,
  58                         OnDidGetDocumentCookie)
  59     IPC_MESSAGE_HANDLER(PrintHostMsg_PrintingFailed, OnPrintingFailed)
  60     IPC_MESSAGE_HANDLER_DELAY_REPLY(PrintHostMsg_GetDefaultPrintSettings,
  61                                     OnGetDefaultPrintSettings)
  62     IPC_MESSAGE_UNHANDLED(handled = false)
  63   IPC_END_MESSAGE_MAP()
  64
  65   return handled ? true : WebContentsObserver::OnMessageReceived(message);
  66 }
  67
  68 void PrintManager::OnAllocateTempFileForPrinting(
  69     base::FileDescriptor* temp_file_fd,
  70     int* sequence_number) {
  71   // we don't really use the sequence number.
  72   *sequence_number = 0;
  73   temp_file_fd->fd = fd_;
  74   temp_file_fd->auto_close = false;
  75 }
  76
  77 void PrintManager::OnTempFileForPrintingWritten(int sequence_number) {
  78   delegate_->DidExportPdf(true);
  79 }
  80
  81 void PrintManager::OnDidGetPrintedPagesCount(int cookie,
  82                                              int number_pages) {
  83   DCHECK_GT(cookie, 0);
  84   DCHECK_GT(number_pages, 0);
  85   number_pages_ = number_pages;
  86 }
  87
  88 void PrintManager::OnDidGetDocumentCookie(int cookie) {
  89   cookie_ = cookie;
  90 }
  91
  92 void PrintManager::OnGetDefaultPrintSettings(IPC::Message* reply_msg) {
  93   // Unlike the printing_message_filter, we do process this in ui thread
  94   DCHECK_CURRENTLY_ON(BrowserThread::UI);
  95   OnGetDefaultPrintSettingsReply(reply_msg);
  96 }
  97
  98 void RenderParamsFromPrintSettings(const printing::PrintSettings& settings,
  99                                    PrintMsg_Print_Params* params) {
 100   params->page_size = settings.page_setup_device_units().physical_size();
 101   params->content_size.SetSize(
 102       settings.page_setup_device_units().content_area().width(),
 103       settings.page_setup_device_units().content_area().height());
 104   params->printable_area.SetRect(
 105       settings.page_setup_device_units().printable_area().x(),
 106       settings.page_setup_device_units().printable_area().y(),
 107       settings.page_setup_device_units().printable_area().width(),
 108       settings.page_setup_device_units().printable_area().height());
 109   params->margin_top = settings.page_setup_device_units().content_area().y();
 110   params->margin_left = settings.page_setup_device_units().content_area().x();
 111   params->dpi = settings.dpi();
 112   // Currently hardcoded at 1.25. See PrintSettings' constructor.
 113   params->min_shrink = settings.min_shrink();
 114   // Currently hardcoded at 2.0. See PrintSettings' constructor.
 115   params->max_shrink = settings.max_shrink();
 116   // Currently hardcoded at 72dpi. See PrintSettings' constructor.
 117   params->desired_dpi = settings.desired_dpi();
 118   // Always use an invalid cookie.
 119   params->document_cookie = 0;
 120   params->selection_only = settings.selection_only();
 121   params->supports_alpha_blend = settings.supports_alpha_blend();
 122   params->should_print_backgrounds = settings.should_print_backgrounds();
 123   params->display_header_footer = settings.display_header_footer();
 124   params->title = settings.title();
 125   params->url = settings.url();
 126 }
 127
 128 void PrintManager::OnGetDefaultPrintSettingsReply(IPC::Message* reply_msg) {
 129   PrintMsg_Print_Params params;
 130   RenderParamsFromPrintSettings(*settings_, &params);
 131   params.document_cookie = cookie_;
 132   PrintHostMsg_GetDefaultPrintSettings::WriteReplyParams(reply_msg, params);
 133   Send(reply_msg);
 134 }
 135
 136 void PrintManager::OnPrintingFailed(int cookie) {
 137   if (cookie != cookie_) {
 138     NOTREACHED();
 139     return;
 140   }
 141   delegate_->DidExportPdf(false);
 142 }
 143
 144 bool PrintManager::PrintNow() {
 145   DCHECK_CURRENTLY_ON(BrowserThread::UI);
 146   if (printing_)
 147     return false;
 148
 149   printing_ = true;
 150   return Send(new PrintMsg_PrintPages(routing_id()));
 151 }
 152
 153 }  // namespace android_webview