Allow printing/saving a plugin through right click menu, even if it's in a fr...

chrome/renderer/print_web_view_helper.cc

 // Copyright (c) 2010 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 "app/l10n_util.h"
 #include "base/logging.h"
 #include "base/process_util.h"
 #include "base/utf_string_conversions.h"
 #include "chrome/common/render_messages.h"
 #include "chrome/common/render_messages_params.h"
 #include "chrome/renderer/render_view.h"
 #include "grit/generated_resources.h"
 #include "printing/units.h"
 #include "third_party/WebKit/WebKit/chromium/public/WebConsoleMessage.h"
+#include "third_party/WebKit/WebKit/chromium/public/WebDocument.h"
+#include "third_party/WebKit/WebKit/chromium/public/WebElement.h"
 #include "third_party/WebKit/WebKit/chromium/public/WebFrame.h"
+#include "third_party/WebKit/WebKit/chromium/public/WebNode.h"
 #include "third_party/WebKit/WebKit/chromium/public/WebRect.h"
 #include "third_party/WebKit/WebKit/chromium/public/WebScreenInfo.h"
 #include "third_party/WebKit/WebKit/chromium/public/WebSize.h"
 #include "third_party/WebKit/WebKit/chromium/public/WebURLRequest.h"
 #include "third_party/WebKit/WebKit/chromium/public/WebView.h"
 #include "webkit/glue/webkit_glue.h"
 
 using printing::ConvertPixelsToPoint;
 using printing::ConvertPixelsToPointDouble;
 using printing::ConvertUnit;
 using printing::ConvertUnitDouble;
 using WebKit::WebConsoleMessage;
 using WebKit::WebFrame;
+using WebKit::WebPoint;

[viettrungluu, 2011/01/13 01:04:40]

To be consistent with the file, maybe you should add a |using WebKit::WebNode;|
(and use it, obviously)?

 using WebKit::WebRect;
 using WebKit::WebSize;
 using WebKit::WebScreenInfo;
 using WebKit::WebString;
 using WebKit::WebURLRequest;
 using WebKit::WebView;
 
 PrepareFrameAndViewForPrint::PrepareFrameAndViewForPrint(
     const ViewMsg_Print_Params& print_params,
     WebFrame* frame,
+    WebKit::WebNode* node,
     WebView* web_view)
         : frame_(frame), web_view_(web_view), expected_pages_count_(0),
           use_browser_overlays_(true) {
   int dpi = static_cast<int>(print_params.dpi);
 #if defined(OS_MACOSX)
   // On the Mac, the printable area is in points, don't do any scaling based
   // on dpi.
   dpi = printing::kPointsPerInch;
 #endif  // defined(OS_MACOSX)
   print_canvas_size_.set_width(
@@ skipping 11 matching lines @@
   // This is important for sites that try to fill the page.
   gfx::Size print_layout_size(print_canvas_size_);
   print_layout_size.set_height(static_cast<int>(
       static_cast<double>(print_layout_size.height()) * 1.25));
 
   prev_view_size_ = web_view->size();
 
   web_view->resize(print_layout_size);
 
   expected_pages_count_ = frame->printBegin(
-      print_canvas_size_, static_cast<int>(print_params.dpi),
+      print_canvas_size_,
+      node,
+      static_cast<int>(print_params.dpi),
       &use_browser_overlays_);
 }
 
 PrepareFrameAndViewForPrint::~PrepareFrameAndViewForPrint() {
   frame_->printEnd();
   web_view_->resize(prev_view_size_);
 }
 
 
 PrintWebViewHelper::PrintWebViewHelper(RenderView* render_view)
     : render_view_(render_view),
       print_web_view_(NULL),
       user_cancelled_scripted_print_count_(0),
       is_preview_(false) {}
 
 PrintWebViewHelper::~PrintWebViewHelper() {}
 
 void PrintWebViewHelper::Print(WebFrame* frame,
                                bool script_initiated,
                                bool is_preview) {
+  Print(frame, NULL, script_initiated, is_preview);
+}
+
+void PrintWebViewHelper::Print(WebKit::WebNode* node,
+                               bool script_initiated,
+                               bool is_preview) {
+  Print(node->document().frame(), node, script_initiated, is_preview);
+}
+
+void PrintWebViewHelper::Print(WebKit::WebFrame* frame,
+                               WebKit::WebNode* node,
+                               bool script_initiated,
+                               bool is_preview) {
   const int kMinSecondsToIgnoreJavascriptInitiatedPrint = 2;
   const int kMaxSecondsToIgnoreJavascriptInitiatedPrint = 2 * 60;  // 2 Minutes.
 
   // If still not finished with earlier print request simply ignore.
   if (IsPrinting())
     return;
 
   // Check if there is script repeatedly trying to print and ignore it if too
   // frequent.  We use exponential wait time so for a page that calls print() in
   // a loop the user will need to cancel the print dialog after 2 seconds, 4
@@ skipping 12 matching lines @@
           WebConsoleMessage::LevelWarning,
           message));
       return;
     }
   }
 
   bool print_cancelled = false;
   is_preview_ = is_preview;
 
   // Initialize print settings.
-  if (!InitPrintSettings(frame))
+  if (!InitPrintSettings(frame, node))
     return;  // Failed to init print page settings.
 
   int expected_pages_count = 0;
   bool use_browser_overlays = true;
 
   // Prepare once to calculate the estimated page count.  This must be in
   // a scope for itself (see comments on PrepareFrameAndViewForPrint).
   {
     PrepareFrameAndViewForPrint prep_frame_view(
-        (*print_pages_params_).params, frame, frame->view());
+        (*print_pages_params_).params, frame, node, frame->view());
     expected_pages_count = prep_frame_view.GetExpectedPageCount();
     if (expected_pages_count)
       use_browser_overlays = prep_frame_view.ShouldUseBrowserOverlays();
   }
 
   // Some full screen plugins can say they don't want to print.
   if (expected_pages_count) {
     if (!is_preview_) {
       // Ask the browser to show UI to retrieve the final print settings.
       if (!GetPrintSettingsFromUser(frame, expected_pages_count,
                                     use_browser_overlays)) {
         print_cancelled = true;
       }
     }
 
     // Render Pages for printing.
     if (!print_cancelled) {
       if (is_preview_)
         RenderPagesForPreview(frame);
       else
-        RenderPagesForPrint(frame);
+        RenderPagesForPrint(frame, node);
 
       // Reset cancel counter on first successful print.
       user_cancelled_scripted_print_count_ = 0;
       return;  // All went well.
     } else {
       if (script_initiated) {
         ++user_cancelled_scripted_print_count_;
         last_cancelled_script_print_ = base::Time::Now();
       }
     }
@@ skipping 48 matching lines @@
 
   // When loading is done this will call DidStopLoading that will do the
   // actual printing.
   print_web_view_->mainFrame()->loadRequest(WebURLRequest(url));
 
   return true;
 }
 
 #if defined(OS_MACOSX) || defined(OS_WIN)
 void PrintWebViewHelper::PrintPages(const ViewMsg_PrintPages_Params& params,
-                                    WebFrame* frame) {
+                                    WebFrame* frame,
+                                    WebKit::WebNode* node) {
   ViewMsg_Print_Params printParams = params.params;
-  UpdatePrintableSizeInPrintParameters(frame, &printParams);
+  UpdatePrintableSizeInPrintParameters(frame, node, &printParams);
 
   PrepareFrameAndViewForPrint prep_frame_view(printParams,
                                               frame,
+                                              node,
                                               frame->view());
   int page_count = prep_frame_view.GetExpectedPageCount();
 
   Send(new ViewHostMsg_DidGetPrintedPagesCount(routing_id(),
                                                printParams.document_cookie,
                                                page_count));
   if (!page_count)
     return;
 
   const gfx::Size& canvas_size = prep_frame_view.GetPrintCanvasSize();
@@ skipping 18 matching lines @@
 bool PrintWebViewHelper::Send(IPC::Message* msg) {
   return render_view_->Send(msg);
 }
 
 int32 PrintWebViewHelper::routing_id() {
   return render_view_->routing_id();
 }
 
 void PrintWebViewHelper::didStopLoading() {
   DCHECK(print_pages_params_.get() != NULL);
-  PrintPages(*print_pages_params_.get(), print_web_view_->mainFrame());
+  PrintPages(*print_pages_params_.get(), print_web_view_->mainFrame(), NULL);
 }
 
 void PrintWebViewHelper::GetPageSizeAndMarginsInPoints(
     WebFrame* frame,
     int page_index,
     const ViewMsg_Print_Params& default_params,
     double* content_width_in_points,
     double* content_height_in_points,
     double* margin_top_in_points,
     double* margin_right_in_points,
@@ skipping 64 matching lines @@
         ConvertPixelsToPointDouble(margin_right_in_pixels);
   if (margin_bottom_in_points)
     *margin_bottom_in_points =
         ConvertPixelsToPointDouble(margin_bottom_in_pixels);
   if (margin_left_in_points)
     *margin_left_in_points =
         ConvertPixelsToPointDouble(margin_left_in_pixels);
 }
 
 void PrintWebViewHelper::UpdatePrintableSizeInPrintParameters(
-    WebFrame* frame, ViewMsg_Print_Params* params) {
+    WebFrame* frame,
+    WebKit::WebNode* node,
+    ViewMsg_Print_Params* params) {
   double content_width_in_points;
   double content_height_in_points;
   double margin_top_in_points;
   double margin_right_in_points;
   double margin_bottom_in_points;
   double margin_left_in_points;
-  PrepareFrameAndViewForPrint prepare(*params, frame, frame->view());
+  PrepareFrameAndViewForPrint prepare(*params, frame, node, frame->view());
   PrintWebViewHelper::GetPageSizeAndMarginsInPoints(frame, 0, *params,
       &content_width_in_points, &content_height_in_points,
       &margin_top_in_points, &margin_right_in_points,
       &margin_bottom_in_points, &margin_left_in_points);
 
 #if defined(OS_MACOSX)
   params->page_size = gfx::Size(
       static_cast<int>(content_width_in_points +
           margin_left_in_points + margin_right_in_points),
       static_cast<int>(content_height_in_points +
           margin_top_in_points + margin_bottom_in_points));
 
   params->printable_size = gfx::Size(static_cast<int>(content_width_in_points),
       static_cast<int>(content_height_in_points));
 #else
   params->printable_size = gfx::Size(
       static_cast<int>(ConvertUnitDouble(
           content_width_in_points, printing::kPointsPerInch, params->dpi)),
       static_cast<int>(ConvertUnitDouble(
           content_height_in_points, printing::kPointsPerInch, params->dpi)));
 #endif
   params->margin_top = static_cast<int>(margin_top_in_points);
   params->margin_left = static_cast<int>(margin_left_in_points);
 }
 
-bool PrintWebViewHelper::InitPrintSettings(WebFrame* frame) {
+bool PrintWebViewHelper::InitPrintSettings(WebFrame* frame,
+                                           WebKit::WebNode* node) {
   ViewMsg_PrintPages_Params settings;
-  if (GetDefaultPrintSettings(frame, &settings.params)) {
+  if (GetDefaultPrintSettings(frame, node, &settings.params)) {
     print_pages_params_.reset(new ViewMsg_PrintPages_Params(settings));
     print_pages_params_->pages.clear();
     return true;
   }
   return false;
 }
 
 bool PrintWebViewHelper::GetDefaultPrintSettings(
-    WebFrame* frame, ViewMsg_Print_Params* params) {
+    WebFrame* frame,
+    WebKit::WebNode* node,
+    ViewMsg_Print_Params* params) {
   IPC::SyncMessage* msg =
       new ViewHostMsg_GetDefaultPrintSettings(routing_id(), params);
   if (!Send(msg)) {
     NOTREACHED();
     return false;
   }
   // Check if the printer returned any settings, if the settings is empty, we
   // can safely assume there are no printer drivers configured. So we safely
   // terminate.
   if (params->IsEmpty()) {
     render_view_->runModalAlertDialog(
         frame,
         l10n_util::GetStringUTF16(IDS_DEFAULT_PRINTER_NOT_FOUND_WARNING));
     return false;
   }
   if (!(params->dpi && params->document_cookie)) {
     // Invalid print page settings.
     NOTREACHED();
     return false;
   }
-  UpdatePrintableSizeInPrintParameters(frame, params);
+  UpdatePrintableSizeInPrintParameters(frame, node, params);
   return true;
 }
 
 bool PrintWebViewHelper::GetPrintSettingsFromUser(WebFrame* frame,
                                                   int expected_pages_count,
                                                   bool use_browser_overlays) {
   ViewHostMsg_ScriptedPrint_Params params;
   ViewMsg_PrintPages_Params print_settings;
 
   // The routing id is sent across as it is needed to look up the
@@ skipping 15 matching lines @@
   msg->EnableMessagePumping();
   if (Send(msg)) {
     print_pages_params_.reset(new ViewMsg_PrintPages_Params(print_settings));
   } else {
     // Send() failed.
     NOTREACHED();
     return false;
   }
   return (print_settings.params.dpi && print_settings.params.document_cookie);
 }
 

[viettrungluu, 2011/01/13 01:04:40]

Could you change that to |WebFrame* frame| while you're here?

-void PrintWebViewHelper::RenderPagesForPrint(WebFrame *frame) {
+void PrintWebViewHelper::RenderPagesForPrint(WebFrame *frame,
+                                             WebKit::WebNode* node) {
   ViewMsg_PrintPages_Params print_settings = *print_pages_params_;
   if (print_settings.params.selection_only) {
     CopyAndPrint(frame);
   } else {
     // TODO: Always copy before printing.
-    PrintPages(print_settings, frame);
+    PrintPages(print_settings, frame, node);
   }
 }
 
 void PrintWebViewHelper::RenderPagesForPreview(WebFrame *frame) {
   ViewMsg_PrintPages_Params print_settings = *print_pages_params_;
   ViewHostMsg_DidPreviewDocument_Params print_params;
   CreatePreviewDocument(print_settings, frame, &print_params);
   Send(new ViewHostMsg_PagesReadyForPreview(routing_id(), print_params));
 }
 
@@ skipping 21 matching lines @@
         shared_buf.GiveToProcess(base::GetCurrentProcessHandle(),
                                  shared_mem_handle);
         return true;
       }
     }
   }
   NOTREACHED();
   return false;
 }
 #endif