Print Preview: Make print preview tab modal.

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/memory/ref_counted.h"
 #include "base/memory/ref_counted_memory.h"
@@ skipping 153 matching lines @@
     PrintHostMsg_DidGetPreviewPageCount_Params temp_params;
     temp_params.page_count = params.expected_pages_count;
     temp_params.document_cookie = params.document_cookie;
     temp_params.is_modifiable = params.modifiable;
     temp_params.preview_request_id = params.preview_request_id;
     print_preview_ui->OnDidGetPreviewPageCount(temp_params);
     print_preview_ui->OnReusePreviewData(params.preview_request_id);
     return;
   }
 
-  print_preview_tab->print_view_manager()->OverrideTitle(tab_contents());
-
   // TODO(joth): This seems like a good match for using RefCountedStaticMemory
   // to avoid the memory copy, but the SetPrintPreviewData call chain below
   // needs updating to accept the RefCountedMemory* base class.
   RefCountedBytes* data_bytes =
       GetDataFromHandle(params.metafile_data_handle, params.data_size);
   if (!data_bytes)
     return;
 
   print_preview_ui->SetPrintPreviewDataForIndex(COMPLETE_PREVIEW_DOCUMENT_INDEX,
                                                 data_bytes);
@@ skipping 53 matching lines @@
                         OnInvalidPrinterSettings)
     IPC_MESSAGE_UNHANDLED(handled = false)
   IPC_END_MESSAGE_MAP()
   return handled;
 }
 
 void PrintPreviewMessageHandler::DidStartLoading() {
   if (tab_contents()->delegate() &&
       PrintPreviewTabController::IsPrintPreviewTab(tab_contents_wrapper())) {
     tab_contents()->SetContentRestrictions(CONTENT_RESTRICTION_PRINT);
+
+    // Need to get the initiator tab to enabled advanced printing.
+    PrintPreviewTabController* tab_controller =
+        PrintPreviewTabController::GetInstance();
+    if (tab_controller) {
+      TabContentsWrapper* initiator_tab =
+          tab_controller->GetInitiatorTab(tab_contents_wrapper());

[kmadhusu, 2011/10/13 18:52:18]

Store the tab_contents_wrapper() in a local variable and use it in line 245 and
here.

[Lei Zhang, 2011/10/13 21:12:12]

Done.

+      if (initiator_tab) {
+        TabContents* contents = initiator_tab->tab_contents();
+        contents->SetContentRestrictions(contents->content_restrictions());
+      }
+    }
   }
 }
 
+void PrintPreviewMessageHandler::NavigateToPendingEntry(
+    const GURL& url,
+    NavigationController::ReloadType reload_type) {

[kmadhusu, 2011/10/13 18:52:18]

Do you need to call StopWorker() here?

[Lei Zhang, 2011/10/13 21:12:12]

I don't think so. Why do you say that?

+  TabContentsWrapper* tab = tab_contents_wrapper();
+  TabContentsWrapper* preview_tab = GetPrintPreviewTab();
+  if (tab == preview_tab) {
+    NOTREACHED();
+    return;
+  }
+  if (preview_tab)
+    tab->print_view_manager()->PrintPreviewDone();
+}
+
 }  // namespace printing