Print Preview: Store preview data in the print preview handler....

chrome/browser/printing/print_preview_message_handler.cc

+// Copyright (c) 2011 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.
+
+#include "chrome/browser/printing/print_preview_message_handler.h"
+
+#include "chrome/browser/browser_process.h"
+#include "chrome/browser/browser_thread.h"
+#include "chrome/browser/dom_ui/print_preview_handler.h"
+#include "chrome/browser/dom_ui/print_preview_ui.h"
+#include "chrome/browser/dom_ui/print_preview_ui_html_source.h"
+#include "chrome/browser/printing/printer_query.h"
+#include "chrome/browser/printing/print_job_manager.h"
+#include "chrome/browser/printing/print_preview_tab_controller.h"
+#include "chrome/browser/renderer_host/render_view_host.h"
+#include "chrome/browser/tab_contents/tab_contents.h"
+#include "chrome/common/render_messages.h"
+#include "chrome/common/render_messages_params.h"
+#include "ipc/ipc_message_macros.h"
+
+namespace printing {
+
+PrintPreviewMessageHandler::PrintPreviewMessageHandler(TabContents* owner)
+    : owner_(owner) {
+  DCHECK(owner);
+}
+
+PrintPreviewMessageHandler::~PrintPreviewMessageHandler() {
+}
+
+TabContents* PrintPreviewMessageHandler::GetOrCreatePrintPreviewTab() {
+  // Get/Create preview tab for initiator tab.
+  printing::PrintPreviewTabController* tab_controller =
+      printing::PrintPreviewTabController::GetInstance();
+  if (!tab_controller)
+    return NULL;
+  return tab_controller->GetOrCreatePreviewTab(owner_,
+      owner_->controller().window_id().id());
+}
+
+void PrintPreviewMessageHandler::OnPagesReadyForPreview(
+    const ViewHostMsg_DidPreviewDocument_Params& params) {
+#if defined(OS_MACOSX)
+  base::SharedMemory* shared_buf =
+      new base::SharedMemory(params.metafile_data_handle, true);
+  if (!shared_buf->Map(params.data_size)) {
+    NOTREACHED();
+    return;
+  }
+#endif
+
+  // Get/Create print preview tab.
+  TabContents* print_preview_tab = GetOrCreatePrintPreviewTab();
+  DCHECK(print_preview_tab);
+
+#if defined(OS_MACOSX)
+  PrintPreviewUI* print_preview_ui =
+      static_cast<PrintPreviewUI*>(print_preview_tab->dom_ui());
+  print_preview_ui->html_source()->SetPrintPreviewData(
+      std::make_pair(shared_buf, params.data_size));
+#endif
+
+  scoped_refptr<printing::PrinterQuery> printer_query;
+  g_browser_process->print_job_manager()->PopPrinterQuery(
+      params.document_cookie, &printer_query);
+  if (printer_query.get()) {
+    BrowserThread::PostTask(
+        BrowserThread::IO, FROM_HERE,
+        NewRunnableMethod(printer_query.get(),
+                          &printing::PrinterQuery::StopWorker));
+  }
+
+  RenderViewHost* rvh = owner_->render_view_host();
+  rvh->Send(new ViewMsg_PrintingDone(rvh->routing_id(),
+                                     params.document_cookie,
+                                     true));
+}
+
+bool PrintPreviewMessageHandler::OnMessageReceived(
+    const IPC::Message& message) {
+  bool handled = true;
+  IPC_BEGIN_MESSAGE_MAP(PrintPreviewMessageHandler, message)
+    IPC_MESSAGE_HANDLER(ViewHostMsg_PagesReadyForPreview,
+                        OnPagesReadyForPreview)
+    IPC_MESSAGE_UNHANDLED(handled = false)
+  IPC_END_MESSAGE_MAP()
+  return handled;
+}
+
+}  // namespace printing