Print Preview: Hook up the print button to initiate a normal print operation....

chrome/renderer/print_web_view_helper.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/renderer/print_web_view_helper.h"
 
 #include <string>
 
 #include "base/logging.h"
 #include "base/process_util.h"
@@ skipping 14 matching lines @@
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebURLRequest.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebView.h"
 #include "ui/base/l10n/l10n_util.h"
 #include "webkit/glue/webkit_glue.h"
 
 using printing::ConvertPixelsToPoint;
 using printing::ConvertPixelsToPointDouble;
 using printing::ConvertUnit;
 using printing::ConvertUnitDouble;
 using WebKit::WebConsoleMessage;
+using WebKit::WebDocument;
+using WebKit::WebElement;
 using WebKit::WebFrame;
 using WebKit::WebNode;
 using WebKit::WebRect;
 using WebKit::WebSize;
 using WebKit::WebScreenInfo;
 using WebKit::WebString;
 using WebKit::WebURLRequest;
 using WebKit::WebView;
 
 namespace {
@@ skipping 69 matching lines @@
 
 void PrintWebViewHelper::PrintNode(WebNode* node,
                                    bool script_initiated,
                                    bool is_preview) {
   Print(node->document().frame(), node, script_initiated, is_preview);
 }
 
 bool PrintWebViewHelper::OnMessageReceived(const IPC::Message& message) {
   bool handled = true;
   IPC_BEGIN_MESSAGE_MAP(PrintWebViewHelper, message)
+    IPC_MESSAGE_HANDLER(ViewMsg_PrintForPrintPreview,
+                        OnPrintForPrintPreview)
     IPC_MESSAGE_HANDLER(ViewMsg_PrintPages, OnPrintPages)
     IPC_MESSAGE_HANDLER(ViewMsg_PrintingDone, OnPrintingDone)
     IPC_MESSAGE_HANDLER(ViewMsg_PrintPreview, OnPrintPreview)
     IPC_MESSAGE_HANDLER(ViewMsg_PrintNodeUnderContextMenu,
                         OnPrintNodeUnderContextMenu)
     IPC_MESSAGE_UNHANDLED(handled = false)
     IPC_END_MESSAGE_MAP()
   return handled;
 }
 
+void PrintWebViewHelper::OnPrintForPrintPreview() {
+  if (!render_view()->webview())
+    return;
+  WebFrame* main_frame = render_view()->webview()->mainFrame();
+  if (!main_frame)
+    return;
+
+  WebDocument document = main_frame->document();
+  // <object> with id="pdf-viewer" is created in
+  // chrome/browser/resources/print_preview.js
+  WebElement element = document.getElementById("pdf-viewer");
+  if (element.isNull()) {
+    NOTREACHED();
+    return;
+  }
+
+  PrintNode(&element, false, false);
+}
+
 void PrintWebViewHelper::OnPrint(bool is_preview) {
   DCHECK(render_view()->webview());
   if (!render_view()->webview())
     return;
 
   // If the user has selected text in the currently focused frame we print
   // only that frame (this makes print selection work for multiple frames).
   if (render_view()->webview()->focusedFrame()->hasSelection())
     PrintFrame(render_view()->webview()->focusedFrame(), false, is_preview);
   else
@@ skipping 431 matching lines @@
         shared_buf.GiveToProcess(base::GetCurrentProcessHandle(),
                                  shared_mem_handle);
         return true;
       }
     }
   }
   NOTREACHED();
   return false;
 }
 #endif