Print preview: Disable the right context menu items in print preview.

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 <vector>
 
 #include "base/bind.h"
 #include "base/memory/ref_counted.h"
 #include "base/memory/ref_counted_memory.h"
 #include "base/shared_memory.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/printing/print_job_manager.h"
 #include "chrome/browser/printing/print_preview_tab_controller.h"
 #include "chrome/browser/printing/print_view_manager.h"
 #include "chrome/browser/printing/printer_query.h"
 #include "chrome/browser/ui/tab_contents/tab_contents_wrapper.h"
 #include "chrome/browser/ui/webui/print_preview_ui.h"
 #include "chrome/common/print_messages.h"
 #include "content/browser/renderer_host/render_view_host.h"
 #include "content/browser/tab_contents/tab_contents.h"
 #include "content/public/browser/browser_thread.h"
+#include "content/public/common/content_restriction.h"
 #include "printing/page_size_margins.h"
 #include "printing/print_job_constants.h"
 
 using content::BrowserThread;
 
 namespace {
 
 void StopWorker(int document_cookie) {
   if (document_cookie <= 0)
     return;
@@ skipping 202 matching lines @@
                         OnDidGetDefaultPageLayout)
     IPC_MESSAGE_HANDLER(PrintHostMsg_PrintPreviewCancelled,
                         OnPrintPreviewCancelled)
     IPC_MESSAGE_HANDLER(PrintHostMsg_PrintPreviewInvalidPrinterSettings,
                         OnInvalidPrinterSettings)
     IPC_MESSAGE_UNHANDLED(handled = false)
   IPC_END_MESSAGE_MAP()
   return handled;
 }
 
+void PrintPreviewMessageHandler::DidStartLoading() {
+  // This helps prevents the print preview tab from being able to print.

[Lei Zhang, 2011/12/09 07:07:49]

Should not have removed this earlier.

+  // It also disables the print menu entry.
+  if (tab_contents()->delegate() &&
+      PrintPreviewTabController::IsPrintPreviewTab(tab_contents_wrapper())) {
+    tab_contents()->SetContentRestrictions(content::CONTENT_RESTRICTION_PRINT);
+  }
+}
+
 void PrintPreviewMessageHandler::NavigateToPendingEntry(
     const GURL& url,
     NavigationController::ReloadType reload_type) {
   TabContentsWrapper* tab = tab_contents_wrapper();
   TabContentsWrapper* preview_tab = GetPrintPreviewTab();
   if (tab == preview_tab) {
     // Cloud print sign-in reloads the page.
     DCHECK(PrintPreviewTabController::IsPrintPreviewURL(url));
     DCHECK_EQ(NavigationController::RELOAD, reload_type);
     return;
   }
   // If |tab| is navigating and it has a print preview tab, notify |tab| to
   // consider print preview done so it unfreezes the renderer in the case of
   // window.print().
   if (preview_tab)
     tab->print_view_manager()->PrintPreviewDone();
 }
 
 }  // namespace printing