chrome/browser/printing/print_preview_message_handler.cc

   1 // Copyright (c) 2012 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 "chrome/browser/printing/print_preview_message_handler.h"
   6
   7 #include <vector>
   8
   9 #include "base/bind.h"
  10 #include "base/memory/ref_counted.h"
  11 #include "base/memory/ref_counted_memory.h"
  12 #include "base/memory/shared_memory.h"
  13 #include "chrome/browser/browser_process.h"
  14 #include "chrome/browser/printing/print_job_manager.h"
  15 #include "chrome/browser/printing/print_preview_dialog_controller.h"
  16 #include "chrome/browser/printing/print_view_manager.h"
  17 #include "chrome/browser/printing/printer_query.h"
  18 #include "chrome/browser/ui/webui/print_preview/print_preview_ui.h"
  19 #include "components/printing/common/print_messages.h"
  20 #include "content/public/browser/browser_thread.h"
  21 #include "content/public/browser/render_view_host.h"
  22 #include "content/public/browser/web_contents.h"
  23 #include "content/public/browser/web_ui.h"
  24 #include "printing/page_size_margins.h"
  25 #include "printing/print_job_constants.h"
  26
  27 using content::BrowserThread;
  28 using content::WebContents;
  29
  30 DEFINE_WEB_CONTENTS_USER_DATA_KEY(printing::PrintPreviewMessageHandler);
  31
  32 namespace {
  33
  34 void StopWorker(int document_cookie) {
  35   if (document_cookie <= 0)
  36     return;
  37   scoped_refptr<printing::PrintQueriesQueue> queue =
  38       g_browser_process->print_job_manager()->queue();
  39   scoped_refptr<printing::PrinterQuery> printer_query =
  40       queue->PopPrinterQuery(document_cookie);
  41   if (printer_query.get()) {
  42     BrowserThread::PostTask(BrowserThread::IO, FROM_HERE,
  43                             base::Bind(&printing::PrinterQuery::StopWorker,
  44                                        printer_query));
  45   }
  46 }
  47
  48 base::RefCountedBytes* GetDataFromHandle(base::SharedMemoryHandle handle,
  49                                          uint32 data_size) {
  50   scoped_ptr<base::SharedMemory> shared_buf(
  51       new base::SharedMemory(handle, true));
  52   if (!shared_buf->Map(data_size)) {
  53     NOTREACHED();
  54     return NULL;
  55   }
  56
  57   unsigned char* data_begin = static_cast<unsigned char*>(shared_buf->memory());
  58   std::vector<unsigned char> data(data_begin, data_begin + data_size);
  59   return base::RefCountedBytes::TakeVector(&data);
  60 }
  61
  62 }  // namespace
  63
  64 namespace printing {
  65
  66 PrintPreviewMessageHandler::PrintPreviewMessageHandler(
  67     WebContents* web_contents)
  68     : content::WebContentsObserver(web_contents) {
  69   DCHECK(web_contents);
  70 }
  71
  72 PrintPreviewMessageHandler::~PrintPreviewMessageHandler() {
  73 }
  74
  75 WebContents* PrintPreviewMessageHandler::GetPrintPreviewDialog() {
  76   PrintPreviewDialogController* dialog_controller =
  77       PrintPreviewDialogController::GetInstance();
  78   if (!dialog_controller)
  79     return NULL;
  80   return dialog_controller->GetPrintPreviewForContents(web_contents());
  81 }
  82
  83 PrintPreviewUI* PrintPreviewMessageHandler::GetPrintPreviewUI() {
  84   WebContents* dialog = GetPrintPreviewDialog();
  85   if (!dialog || !dialog->GetWebUI())
  86     return NULL;
  87   return static_cast<PrintPreviewUI*>(dialog->GetWebUI()->GetController());
  88 }
  89
  90 void PrintPreviewMessageHandler::OnRequestPrintPreview(
  91     const PrintHostMsg_RequestPrintPreview_Params& params) {
  92   if (params.webnode_only) {
  93     printing::PrintViewManager::FromWebContents(web_contents())->
  94         PrintPreviewForWebNode();
  95   }
  96   PrintPreviewDialogController::PrintPreview(web_contents());
  97   PrintPreviewUI::SetInitialParams(GetPrintPreviewDialog(), params);
  98 }
  99
 100 void PrintPreviewMessageHandler::OnDidGetPreviewPageCount(
 101     const PrintHostMsg_DidGetPreviewPageCount_Params& params) {
 102   if (params.page_count <= 0) {
 103     NOTREACHED();
 104     return;
 105   }
 106
 107   PrintPreviewUI* print_preview_ui = GetPrintPreviewUI();
 108   if (!print_preview_ui)
 109     return;
 110
 111   if (!params.is_modifiable || params.clear_preview_data)
 112     print_preview_ui->ClearAllPreviewData();
 113
 114   print_preview_ui->OnDidGetPreviewPageCount(params);
 115 }
 116
 117 void PrintPreviewMessageHandler::OnDidPreviewPage(
 118     const PrintHostMsg_DidPreviewPage_Params& params) {
 119   int page_number = params.page_number;
 120   if (page_number < FIRST_PAGE_INDEX || !params.data_size)
 121     return;
 122
 123   PrintPreviewUI* print_preview_ui = GetPrintPreviewUI();
 124   if (!print_preview_ui)
 125     return;
 126
 127   base::RefCountedBytes* data_bytes =
 128       GetDataFromHandle(params.metafile_data_handle, params.data_size);
 129   DCHECK(data_bytes);
 130
 131   print_preview_ui->SetPrintPreviewDataForIndex(page_number, data_bytes);
 132   print_preview_ui->OnDidPreviewPage(page_number, params.preview_request_id);
 133 }
 134
 135 void PrintPreviewMessageHandler::OnMetafileReadyForPrinting(
 136     const PrintHostMsg_DidPreviewDocument_Params& params) {
 137   // Always try to stop the worker.
 138   StopWorker(params.document_cookie);
 139
 140   if (params.expected_pages_count <= 0) {
 141     NOTREACHED();
 142     return;
 143   }
 144
 145   PrintPreviewUI* print_preview_ui = GetPrintPreviewUI();
 146   if (!print_preview_ui)
 147     return;
 148
 149   // TODO(joth): This seems like a good match for using RefCountedStaticMemory
 150   // to avoid the memory copy, but the SetPrintPreviewData call chain below
 151   // needs updating to accept the RefCountedMemory* base class.
 152   base::RefCountedBytes* data_bytes =
 153       GetDataFromHandle(params.metafile_data_handle, params.data_size);
 154   if (!data_bytes || !data_bytes->size())
 155     return;
 156
 157   print_preview_ui->SetPrintPreviewDataForIndex(COMPLETE_PREVIEW_DOCUMENT_INDEX,
 158                                                 data_bytes);
 159   print_preview_ui->OnPreviewDataIsAvailable(
 160       params.expected_pages_count, params.preview_request_id);
 161 }
 162
 163 void PrintPreviewMessageHandler::OnPrintPreviewFailed(int document_cookie) {
 164   StopWorker(document_cookie);
 165
 166   PrintPreviewUI* print_preview_ui = GetPrintPreviewUI();
 167   if (!print_preview_ui)
 168     return;
 169   print_preview_ui->OnPrintPreviewFailed();
 170 }
 171
 172 void PrintPreviewMessageHandler::OnDidGetDefaultPageLayout(
 173     const PageSizeMargins& page_layout_in_points,
 174     const gfx::Rect& printable_area_in_points,
 175     bool has_custom_page_size_style) {
 176   PrintPreviewUI* print_preview_ui = GetPrintPreviewUI();
 177   if (!print_preview_ui)
 178     return;
 179   print_preview_ui->OnDidGetDefaultPageLayout(page_layout_in_points,
 180                                               printable_area_in_points,
 181                                               has_custom_page_size_style);
 182 }
 183
 184 void PrintPreviewMessageHandler::OnPrintPreviewCancelled(int document_cookie) {
 185   // Always need to stop the worker.
 186   StopWorker(document_cookie);
 187 }
 188
 189 void PrintPreviewMessageHandler::OnInvalidPrinterSettings(int document_cookie) {
 190   StopWorker(document_cookie);
 191   PrintPreviewUI* print_preview_ui = GetPrintPreviewUI();
 192   if (!print_preview_ui)
 193     return;
 194   print_preview_ui->OnInvalidPrinterSettings();
 195 }
 196
 197 void PrintPreviewMessageHandler::OnSetOptionsFromDocument(
 198     const PrintHostMsg_SetOptionsFromDocument_Params& params) {
 199   PrintPreviewUI* print_preview_ui = GetPrintPreviewUI();
 200   if (!print_preview_ui)
 201     return;
 202   print_preview_ui->OnSetOptionsFromDocument(params);
 203 }
 204
 205 bool PrintPreviewMessageHandler::OnMessageReceived(
 206     const IPC::Message& message) {
 207   bool handled = true;
 208   IPC_BEGIN_MESSAGE_MAP(PrintPreviewMessageHandler, message)
 209     IPC_MESSAGE_HANDLER(PrintHostMsg_RequestPrintPreview,
 210                         OnRequestPrintPreview)
 211     IPC_MESSAGE_HANDLER(PrintHostMsg_DidGetPreviewPageCount,
 212                         OnDidGetPreviewPageCount)
 213     IPC_MESSAGE_HANDLER(PrintHostMsg_DidPreviewPage,
 214                         OnDidPreviewPage)
 215     IPC_MESSAGE_HANDLER(PrintHostMsg_MetafileReadyForPrinting,
 216                         OnMetafileReadyForPrinting)
 217     IPC_MESSAGE_HANDLER(PrintHostMsg_PrintPreviewFailed,
 218                         OnPrintPreviewFailed)
 219     IPC_MESSAGE_HANDLER(PrintHostMsg_DidGetDefaultPageLayout,
 220                         OnDidGetDefaultPageLayout)
 221     IPC_MESSAGE_HANDLER(PrintHostMsg_PrintPreviewCancelled,
 222                         OnPrintPreviewCancelled)
 223     IPC_MESSAGE_HANDLER(PrintHostMsg_PrintPreviewInvalidPrinterSettings,
 224                         OnInvalidPrinterSettings)
 225     IPC_MESSAGE_HANDLER(PrintHostMsg_SetOptionsFromDocument,
 226                         OnSetOptionsFromDocument)
 227     IPC_MESSAGE_UNHANDLED(handled = false)
 228   IPC_END_MESSAGE_MAP()
 229   return handled;
 230 }
 231
 232 }  // namespace printing