Print Preview: Making margin selection sticky (part 2/2)

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"
@@ skipping 81 matching lines @@
 
   // Inform the print preview tab of the failure.
   TabContentsWrapper* print_preview_tab = GetPrintPreviewTab();
   // User might have closed it already.
   if (!print_preview_tab || !print_preview_tab->web_ui())
     return NULL;
 
   return static_cast<PrintPreviewUI*>(print_preview_tab->web_ui());
 }
 
-void PrintPreviewMessageHandler::OnRequestPrintPreview() {
+void PrintPreviewMessageHandler::OnRequestPrintPreview(
+    bool source_is_modifiable) {
   PrintPreviewTabController::PrintPreview(tab_contents_wrapper());
+
+  TabContentsWrapper* print_preview_tab = GetPrintPreviewTab();
+  if (!print_preview_tab || !print_preview_tab->web_ui())
+    return;
+  PrintPreviewUI* print_preview_ui =
+      static_cast<PrintPreviewUI*>(print_preview_tab->web_ui());
+  print_preview_ui->SetSourceIsModifiable(source_is_modifiable);
 }
 
 void PrintPreviewMessageHandler::OnDidGetPreviewPageCount(
     const PrintHostMsg_DidGetPreviewPageCount_Params& params) {
   if (params.page_count <= 0) {
     NOTREACHED();
     return;
   }
 
   TabContentsWrapper* print_preview_tab = GetPrintPreviewTab();
@@ skipping 131 matching lines @@
 }
 
 void PrintPreviewMessageHandler::DidStartLoading() {
   if (tab_contents()->delegate() &&
       PrintPreviewTabController::IsPrintPreviewTab(tab_contents_wrapper())) {
     tab_contents()->SetContentRestrictions(content::CONTENT_RESTRICTION_PRINT);
   }
 }
 
 }  // namespace printing