Cleanup: Convert PrintPreviewTabController to use TabContentsWrapper.

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 51 matching lines @@
 
 PrintPreviewMessageHandler::PrintPreviewMessageHandler(
     TabContents* tab_contents)
     : TabContentsObserver(tab_contents) {
   DCHECK(tab_contents);
 }
 
 PrintPreviewMessageHandler::~PrintPreviewMessageHandler() {
 }
 
-TabContents* PrintPreviewMessageHandler::GetPrintPreviewTab() {
-  printing::PrintPreviewTabController* tab_controller =
-      printing::PrintPreviewTabController::GetInstance();
+TabContentsWrapper* PrintPreviewMessageHandler::GetPrintPreviewTab() {
+  PrintPreviewTabController* tab_controller =
+      PrintPreviewTabController::GetInstance();
   if (!tab_controller)
     return NULL;
 
-  return tab_controller->GetPrintPreviewForTab(tab_contents());
+  return tab_controller->GetPrintPreviewForTab(tab_contents_wrapper());
+}
+
+TabContentsWrapper* PrintPreviewMessageHandler::tab_contents_wrapper() {
+  return TabContentsWrapper::GetCurrentWrapperForContents(tab_contents());
 }
 
 void PrintPreviewMessageHandler::OnRequestPrintPreview() {
-  PrintPreviewTabController::PrintPreview(tab_contents());
+  PrintPreviewTabController::PrintPreview(tab_contents_wrapper());
 }
 
 void PrintPreviewMessageHandler::OnDidGetPreviewPageCount(
     const PrintHostMsg_DidGetPreviewPageCount_Params& params) {
   if (params.page_count <= 0) {
     NOTREACHED();
     return;
   }
 
-  TabContents* print_preview_tab = GetPrintPreviewTab();
+  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->OnDidGetPreviewPageCount(params);
 }
 
 void PrintPreviewMessageHandler::OnDidPreviewPage(
     const PrintHostMsg_DidPreviewPage_Params& params) {
-  TabContents* print_preview_tab = GetPrintPreviewTab();
+  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());
   int page_number = params.page_number;
 
   if (page_number == FIRST_PAGE_INDEX)
     print_preview_ui->ClearAllPreviewData();
 
@@ skipping 11 matching lines @@
     const PrintHostMsg_DidPreviewDocument_Params& params) {
   // Always try to stop the worker.
   StopWorker(params.document_cookie);
 
   if (params.expected_pages_count <= 0) {
     NOTREACHED();
     return;
   }
 
   // Get the print preview tab.
-  TabContents* print_preview_tab = GetPrintPreviewTab();
+  TabContentsWrapper* print_preview_tab = GetPrintPreviewTab();
   // User might have closed it already.
-  if (!print_preview_tab)
+  if (!print_preview_tab || !print_preview_tab->web_ui())
     return;
 
   PrintPreviewUI* print_preview_ui =
       static_cast<PrintPreviewUI*>(print_preview_tab->web_ui());
 
-  TabContentsWrapper* wrapper =
-      TabContentsWrapper::GetCurrentWrapperForContents(print_preview_tab);
-
   if (params.reuse_existing_data) {
     // Need to match normal rendering where we are expected to send this.
     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;
   }
 
-  wrapper->print_view_manager()->OverrideTitle(tab_contents());
+  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);
   print_preview_ui->OnPreviewDataIsAvailable(
       params.expected_pages_count, params.preview_request_id);
 }
 
 void PrintPreviewMessageHandler::OnPrintPreviewFailed(int document_cookie) {
   // Always need to stop the worker.
   StopWorker(document_cookie);
 
   // Inform the print preview tab of the failure.
-  TabContents* print_preview_tab = GetPrintPreviewTab();
+  TabContentsWrapper* print_preview_tab = GetPrintPreviewTab();
   // User might have closed it already.
-  if (!print_preview_tab)
+  if (!print_preview_tab || !print_preview_tab->web_ui())
     return;
 
-  TabContentsWrapper* wrapper =
-      TabContentsWrapper::GetCurrentWrapperForContents(print_preview_tab);
-
   if (g_browser_process->background_printing_manager()->
-          HasTabContents(wrapper)) {
+          HasTabContents(print_preview_tab)) {
     // Preview tab was hidden to serve the print request.
-    delete wrapper;
+    delete print_preview_tab;
   } else {
     PrintPreviewUI* print_preview_ui =
       static_cast<PrintPreviewUI*>(print_preview_tab->web_ui());
     print_preview_ui->OnPrintPreviewFailed();
   }
 }
 
 void PrintPreviewMessageHandler::OnDidGetDefaultPageLayout(
     const PageSizeMargins& page_layout_in_points) {
-  TabContents* print_preview_tab = GetPrintPreviewTab();
-  if (!print_preview_tab)
+  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->OnDidGetDefaultPageLayout(page_layout_in_points);
 }
 
 void PrintPreviewMessageHandler::OnPrintPreviewCancelled(int document_cookie) {
   // Always need to stop the worker.
   StopWorker(document_cookie);
@@ skipping 17 matching lines @@
                         OnDidGetDefaultPageLayout)
     IPC_MESSAGE_HANDLER(PrintHostMsg_PrintPreviewCancelled,
                         OnPrintPreviewCancelled)
     IPC_MESSAGE_UNHANDLED(handled = false)
   IPC_END_MESSAGE_MAP()
   return handled;
 }
 
 void PrintPreviewMessageHandler::DidStartLoading() {
   if (tab_contents()->delegate() &&
-      printing::PrintPreviewTabController::IsPrintPreviewTab(tab_contents())) {
+      PrintPreviewTabController::IsPrintPreviewTab(tab_contents_wrapper())) {
     tab_contents()->SetContentRestrictions(CONTENT_RESTRICTION_PRINT);
   }
 }
 
 }  // namespace printing