Print Preview: Set source modifiable bit for scripted printing.

chrome/browser/printing/print_view_manager.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_view_manager.h"
 
+#include <map>
+
 #include "base/bind.h"
 #include "base/lazy_instance.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/utf_string_conversions.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/printing/print_job.h"
 #include "chrome/browser/printing/print_job_manager.h"
 #include "chrome/browser/printing/print_preview_tab_controller.h"
 #include "chrome/browser/printing/printer_query.h"
 #include "chrome/browser/printing/print_view_manager_observer.h"
@@ skipping 234 matching lines @@
 
 void PrintViewManager::OnPrintingFailed(int cookie) {
   ReleasePrinterQuery(cookie);
 
   content::NotificationService::current()->Notify(
       chrome::NOTIFICATION_PRINT_JOB_RELEASED,
       content::Source<TabContents>(tab_contents()),
       content::NotificationService::NoDetails());
 }
 
-void PrintViewManager::OnScriptedPrintPreview(IPC::Message* reply_msg) {
+void PrintViewManager::OnScriptedPrintPreview(bool source_is_modifiable,
+                                              IPC::Message* reply_msg) {
   BrowserThread::CurrentlyOn(BrowserThread::UI);
   ScriptedPrintPreviewClosureMap& map =
       g_scripted_print_preview_closure_map.Get();
   content::RenderProcessHost* rph =
       tab_contents()->render_view_host()->process();
 
   // This should always be 0 once we get modal window.print().
   if (map.count(rph) != 0) {
     // Renderer already handling window.print() in another View.
     Send(reply_msg);
     return;
   }
   if (print_preview_state_ != NOT_PREVIEWING) {
     // If a user initiated print dialog is already open, ignore the scripted
     // print message.
     DCHECK_EQ(USER_INITIATED_PREVIEW, print_preview_state_);
     Send(reply_msg);
     return;
   }
 
+  PrintPreviewTabController* tab_controller =
+      PrintPreviewTabController::GetInstance();
+  if (!tab_controller) {
+    Send(reply_msg);
+    return;
+  }
+
   print_preview_state_ = SCRIPTED_PREVIEW;
   base::Closure callback =
       base::Bind(&PrintViewManager::OnScriptedPrintPreviewReply,
                  base::Unretained(this),
                  reply_msg);
   map[rph] = callback;
   scripted_print_preview_rph_ = rph;
 
-  PrintPreviewTabController::PrintPreview(tab_);
+  tab_controller->PrintPreview(tab_);
+  PrintPreviewUI::SetSourceIsModifiable(
+      tab_controller->GetPrintPreviewForTab(tab_),
+      source_is_modifiable);
 }
 
 void PrintViewManager::OnScriptedPrintPreviewReply(IPC::Message* reply_msg) {
   BrowserThread::CurrentlyOn(BrowserThread::UI);
   Send(reply_msg);
 }
 
 bool PrintViewManager::OnMessageReceived(const IPC::Message& message) {
   bool handled = true;
   IPC_BEGIN_MESSAGE_MAP(PrintViewManager, message)
@@ skipping 280 matching lines @@
 }
 
 bool PrintViewManager::PrintNowInternal(IPC::Message* message) {
   // Don't print / print preview interstitials.
   if (tab_contents()->showing_interstitial_page())
     return false;
   return Send(message);
 }
 
 }  // namespace printing