Print Preview: Hook up the print button to initiate printing without displaying a print dialog.

content/browser/renderer_host/render_message_filter.cc

Index: content/browser/renderer_host/render_message_filter.cc
diff --git a/content/browser/renderer_host/render_message_filter.cc b/content/browser/renderer_host/render_message_filter.cc
index 4308e7a83d00fa79d475ce837da6bd17de0a9822..de513890c511f82e0966a1b8e875ba4e793f0db1 100644
--- a/content/browser/renderer_host/render_message_filter.cc
+++ b/content/browser/renderer_host/render_message_filter.cc
@@ -462,7 +462,11 @@ bool RenderMessageFilter::OnMessageReceived(const IPC::Message& message,
     IPC_MESSAGE_HANDLER_DELAY_REPLY(ViewHostMsg_ResolveProxy, OnResolveProxy)
     IPC_MESSAGE_HANDLER_DELAY_REPLY(ViewHostMsg_GetDefaultPrintSettings,
                                     OnGetDefaultPrintSettings)
+    IPC_MESSAGE_HANDLER(ViewHostMsg_GetCurrentPrintSettings,
+                        OnGetCurrentPrintSettings)
     IPC_MESSAGE_HANDLER_DELAY_REPLY(ViewHostMsg_ScriptedPrint, OnScriptedPrint)
+    IPC_MESSAGE_HANDLER_DELAY_REPLY(ViewHostMsg_UpdatePrintSettings,
+                                    OnUpdatePrintSettings)
 #if defined(OS_MACOSX)
     IPC_MESSAGE_HANDLER(ViewHostMsg_AllocTransportDIB, OnAllocTransportDIB)
     IPC_MESSAGE_HANDLER(ViewHostMsg_FreeTransportDIB, OnFreeTransportDIB)
@@ -1163,6 +1167,65 @@ void RenderMessageFilter::OnGetDefaultPrintSettingsReply(
   }
 }
 
+void RenderMessageFilter::OnGetCurrentPrintSettings(int document_cookie,
+    ViewMsg_Print_Params* params) {
+  scoped_refptr<printing::PrinterQuery> printer_query;
+  print_job_manager_->PopPrinterQuery(document_cookie, &printer_query);
+  if (!printer_query.get()) {
+    memset(params, 0, sizeof(ViewMsg_Print_Params));
+  } else {
+    RenderParamsFromPrintSettings(printer_query->settings(), params);
+    params->document_cookie = printer_query->cookie();
+  }
+  // If printing was enabled.
+  if (printer_query.get()) {
+    // If user hasn't cancelled.

[Lei Zhang, 2011/02/24 03:49:32]

Can't you move this block into the else block above?

[kmadhusu, 2011/03/01 01:55:50]

Fixed.

+    if (printer_query->cookie() && printer_query->settings().dpi()) {
+      print_job_manager_->QueuePrinterQuery(printer_query.get());
+    } else {
+      printer_query->StopWorker();
+    }
+  }
+}
+
+void RenderMessageFilter::OnUpdatePrintSettings(int document_cookie,
+    const std::string& job_settings, IPC::Message* reply_msg) {
+  scoped_refptr<printing::PrinterQuery> printer_query;
+  print_job_manager_->PopPrinterQuery(document_cookie, &printer_query);
+  if (printer_query.get()) {
+    CancelableTask* task = NewRunnableMethod(
+        this,
+        &RenderMessageFilter::OnUpdatePrintSettingsReply,
+        printer_query,
+        reply_msg);
+    printer_query->SetSettings(job_settings, task);
+  }
+}
+
+void RenderMessageFilter::OnUpdatePrintSettingsReply(
+    scoped_refptr<printing::PrinterQuery> printer_query,
+    IPC::Message* reply_msg) {
+  ViewMsg_Print_Params params;
+  if (!printer_query.get() ||
+      printer_query->last_status() != printing::PrintingContext::OK) {
+    memset(&params, 0, sizeof(params));
+  } else {
+    RenderParamsFromPrintSettings(printer_query->settings(), &params);
+    params.document_cookie = printer_query->cookie();
+  }
+  ViewHostMsg_UpdatePrintSettings::WriteReplyParams(reply_msg, params);
+  Send(reply_msg);
+  // If printing was enabled.
+  if (printer_query.get()) {
+    // If user hasn't cancelled.
+    if (printer_query->cookie() && printer_query->settings().dpi()) {
+      print_job_manager_->QueuePrinterQuery(printer_query.get());
+    } else {
+      printer_query->StopWorker();
+    }
+  }
+}
+
 void RenderMessageFilter::OnScriptedPrint(
     const ViewHostMsg_ScriptedPrint_Params& params,
     IPC::Message* reply_msg) {