Cleanup: Convert PrintPreviewTabController to use TabContentsWrapper.

chrome/browser/ui/webui/print_preview_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/ui/webui/print_preview_handler.h"
 
 #include <string>
 
 #include "base/base64.h"
 #if !defined(OS_CHROMEOS)
@@ skipping 429 matching lines @@
   web_ui_->RegisterMessageCallback("closePrintPreviewTab",
       NewCallback(this, &PrintPreviewHandler::HandleClosePreviewTab));
   web_ui_->RegisterMessageCallback("hidePreview",
       NewCallback(this, &PrintPreviewHandler::HandleHidePreview));
   web_ui_->RegisterMessageCallback("cancelPendingPrintRequest",
       NewCallback(this, &PrintPreviewHandler::HandleCancelPendingPrintRequest));
   web_ui_->RegisterMessageCallback("saveLastPrinter",
       NewCallback(this, &PrintPreviewHandler::HandleSaveLastPrinter));
 }
 
-TabContents* PrintPreviewHandler::preview_tab() {
+TabContentsWrapper* PrintPreviewHandler::preview_tab_wrapper() const {
+  return TabContentsWrapper::GetCurrentWrapperForContents(preview_tab());
+}
+TabContents* PrintPreviewHandler::preview_tab() const {
   return web_ui_->tab_contents();
 }
 
 void PrintPreviewHandler::HandleGetDefaultPrinter(const ListValue*) {
   scoped_refptr<PrintSystemTaskProxy> task =
       new PrintSystemTaskProxy(AsWeakPtr(),
                                print_backend_.get(),
                                has_logged_printers_count_);
   BrowserThread::PostTask(
       BrowserThread::FILE, FROM_HERE,
@@ skipping 26 matching lines @@
   print_preview_ui->OnPrintPreviewRequest(request_id);
   // Add an additional key in order to identify |print_preview_ui| later on
   // when calling PrintPreviewUI::GetCurrentPrintPreviewStatus() on the IO
   // thread.
   settings->SetString(printing::kPreviewUIAddr,
                       print_preview_ui->GetPrintPreviewUIAddress());
 
   // Increment request count.
   ++regenerate_preview_request_count_;
 
-  TabContents* initiator_tab = GetInitiatorTab();
+  TabContentsWrapper* initiator_tab = GetInitiatorTab();
   if (!initiator_tab) {
     ReportUserActionHistogram(INITIATOR_TAB_CLOSED);
     print_preview_ui->OnInitiatorTabClosed();
     return;
   }
 
   // Retrieve the page title and url and send it to the renderer process if
   // headers and footers are to be displayed.
   bool display_header_footer = false;
   if (!settings->GetBoolean(printing::kSettingHeaderFooterEnabled,
                             &display_header_footer)) {
     NOTREACHED();
   }
   if (display_header_footer) {
     settings->SetString(printing::kSettingHeaderFooterTitle,
-                        initiator_tab->GetTitle());
+                        initiator_tab->tab_contents()->GetTitle());
     std::string url;
     NavigationEntry* entry = initiator_tab->controller().GetActiveEntry();
     if (entry)
       url = entry->virtual_url().spec();
     settings->SetString(printing::kSettingHeaderFooterURL, url);
   }
 
   VLOG(1) << "Print preview request start";
   RenderViewHost* rvh = initiator_tab->render_view_host();
   rvh->Send(new PrintMsg_PrintPreview(rvh->routing_id(), *settings));
 }
 
 void PrintPreviewHandler::HandlePrint(const ListValue* args) {
   ReportStats();
 
   // Record the number of times the user requests to regenerate preview data
   // before printing.
   UMA_HISTOGRAM_COUNTS("PrintPreview.RegeneratePreviewRequest.BeforePrint",
                        regenerate_preview_request_count_);
 
-  TabContents* initiator_tab = GetInitiatorTab();
+  TabContentsWrapper* initiator_tab = GetInitiatorTab();
   if (initiator_tab) {
     RenderViewHost* rvh = initiator_tab->render_view_host();
     rvh->Send(new PrintMsg_ResetScriptedPrintCount(rvh->routing_id()));
   }
 
   scoped_ptr<DictionaryValue> settings(GetSettingsDictionary(args));
   if (!settings.get())
     return;
 
   // Storing last used color setting.
@@ skipping 52 matching lines @@
     RenderViewHost* rvh = web_ui_->tab_contents()->render_view_host();
     rvh->Send(new PrintMsg_PrintForPrintPreview(rvh->routing_id(), *settings));
   }
 }
 
 void PrintPreviewHandler::HandleHidePreview(const ListValue*) {
   HidePreviewTab();
 }
 
 void PrintPreviewHandler::HandleCancelPendingPrintRequest(const ListValue*) {
-  TabContentsWrapper* wrapper = NULL;
-  TabContents* initiator_tab = GetInitiatorTab();
+  TabContentsWrapper* initiator_tab = GetInitiatorTab();
   if (initiator_tab) {
-    wrapper = TabContentsWrapper::GetCurrentWrapperForContents(initiator_tab);
     ClearInitiatorTabDetails();
   } else {
     // Initiator tab does not exists. Get the wrapper contents of current tab.
     Browser* browser = BrowserList::GetLastActive();
     if (browser)
-      wrapper = browser->GetSelectedTabContentsWrapper();
+      initiator_tab = browser->GetSelectedTabContentsWrapper();
   }
 
-  if (wrapper)
-    wrapper->print_view_manager()->PreviewPrintingRequestCancelled();
-  delete TabContentsWrapper::GetCurrentWrapperForContents(preview_tab());
+  if (initiator_tab)
+    initiator_tab->print_view_manager()->PreviewPrintingRequestCancelled();
+  delete preview_tab_wrapper();
 }
 
 void PrintPreviewHandler::HandleSaveLastPrinter(const ListValue* args) {
   std::string data_to_save;
   if (args->GetString(0, &data_to_save) && !data_to_save.empty()) {
     if (last_used_printer_name_ == NULL)
       last_used_printer_name_ = new std::string();
     *last_used_printer_name_ = data_to_save;
   }
   if (args->GetString(1, &data_to_save) && !data_to_save.empty()) {
@@ skipping 31 matching lines @@
                    GetCloudPrintServiceManageURL(),
                    GURL(),
                    NEW_FOREGROUND_TAB,
                    PageTransition::LINK);
 }
 
 void PrintPreviewHandler::HandleShowSystemDialog(const ListValue*) {
   ReportStats();
   ReportUserActionHistogram(FALLBACK_TO_ADVANCED_SETTINGS_DIALOG);
 
-  TabContents* initiator_tab = GetInitiatorTab();
+  TabContentsWrapper* initiator_tab = GetInitiatorTab();
   if (!initiator_tab)
     return;
 
-  TabContentsWrapper* wrapper =
-      TabContentsWrapper::GetCurrentWrapperForContents(initiator_tab);
-  printing::PrintViewManager* manager = wrapper->print_view_manager();
+  printing::PrintViewManager* manager = initiator_tab->print_view_manager();
   manager->set_observer(this);
   manager->PrintForSystemDialogNow();
 
   // Cancel the pending preview request if exists.
   PrintPreviewUI* print_preview_ui = static_cast<PrintPreviewUI*>(web_ui_);
   print_preview_ui->OnCancelPendingPreviewRequest();
 }
 
 void PrintPreviewHandler::HandleManagePrinters(const ListValue*) {
   ++manage_printers_dialog_request_count_;
   printing::PrinterManagerDialog::ShowPrinterManagerDialog();
 }
 
 void PrintPreviewHandler::HandleReloadCrashedInitiatorTab(const ListValue*) {
   ReportStats();
   ReportUserActionHistogram(INITIATOR_TAB_CRASHED);
 
-  TabContents* initiator_tab = GetInitiatorTab();
+  TabContentsWrapper* initiator_tab = GetInitiatorTab();
   if (!initiator_tab)
     return;
 
-  initiator_tab->OpenURL(
-      initiator_tab->GetURL(), GURL(), CURRENT_TAB, PageTransition::RELOAD);
+  TabContents* contents = initiator_tab->tab_contents();
+  contents->OpenURL(contents->GetURL(), GURL(), CURRENT_TAB,
+                    PageTransition::RELOAD);
   ActivateInitiatorTabAndClosePreviewTab();
 }
 
 void PrintPreviewHandler::HandleClosePreviewTab(const ListValue*) {
   ReportStats();
   ReportUserActionHistogram(CANCEL);
 
   // Record the number of times the user requests to regenerate preview data
   // before cancelling.
   UMA_HISTOGRAM_COUNTS("PrintPreview.RegeneratePreviewRequest.BeforeCancel",
                        regenerate_preview_request_count_);
 
   ActivateInitiatorTabAndClosePreviewTab();
 }
 
 void PrintPreviewHandler::ReportStats() {
   UMA_HISTOGRAM_COUNTS("PrintPreview.ManagePrinters",
                        manage_printers_dialog_request_count_);
 }
 
 void PrintPreviewHandler::ActivateInitiatorTabAndClosePreviewTab() {
-  TabContents* initiator_tab = GetInitiatorTab();
-  if (initiator_tab)
-    static_cast<RenderViewHostDelegate*>(initiator_tab)->Activate();
+  TabContentsWrapper* initiator_tab = GetInitiatorTab();
+  if (initiator_tab) {
+    static_cast<RenderViewHostDelegate*>(
+        initiator_tab->tab_contents())->Activate();
+  }
   ClosePrintPreviewTab();
 }
 
 void PrintPreviewHandler::SendPrinterCapabilities(
     const DictionaryValue& settings_info) {
   VLOG(1) << "Get printer capabilities finished";
   web_ui_->CallJavascriptFunction("updateWithPrinterCapabilities",
                                   settings_info);
 }
 
@@ skipping 78 matching lines @@
                       boundary,
                       base64_data.c_str(),
                       boundary);
 
   StringValue data_value(final_data);
 
   web_ui_->CallJavascriptFunction("printToCloud",
                                   data_value);
 }
 
-TabContents* PrintPreviewHandler::GetInitiatorTab() {
+TabContentsWrapper* PrintPreviewHandler::GetInitiatorTab() const {
   printing::PrintPreviewTabController* tab_controller =
       printing::PrintPreviewTabController::GetInstance();
   if (!tab_controller)
     return NULL;
-  return tab_controller->GetInitiatorTab(preview_tab());
+  return tab_controller->GetInitiatorTab(preview_tab_wrapper());
 }
 
 void PrintPreviewHandler::ClosePrintPreviewTab() {
   TabContentsWrapper* tab =
       TabContentsWrapper::GetCurrentWrapperForContents(preview_tab());
   if (!tab)
     return;
   Browser* preview_tab_browser = BrowserList::FindBrowserWithID(
       tab->restore_tab_helper()->window_id().id());
   if (!preview_tab_browser)
     return;
   TabStripModel* tabstrip = preview_tab_browser->tabstrip_model();
 
   // Keep print preview tab out of the recently closed tab list, because
   // re-opening that page will just display a non-functional print preview page.
   tabstrip->CloseTabContentsAt(tabstrip->GetIndexOfController(
       &preview_tab()->controller()), TabStripModel::CLOSE_NONE);
 }
 
 void PrintPreviewHandler::OnPrintDialogShown() {
-  static_cast<RenderViewHostDelegate*>(GetInitiatorTab())->Activate();
+  static_cast<RenderViewHostDelegate*>(
+      GetInitiatorTab()->tab_contents())->Activate();
   ClosePrintPreviewTab();
 }
 
 void PrintPreviewHandler::SelectFile(const FilePath& default_filename) {
   SelectFileDialog::FileTypeInfo file_type_info;
   file_type_info.extensions.resize(1);
   file_type_info.extensions[0].push_back(FILE_PATH_LITERAL("pdf"));
 
   // Initializing last_saved_path_ if it is not already initialized.
   if (!last_saved_path_) {
@@ skipping 14 matching lines @@
       last_saved_path_->Append(default_filename),
       &file_type_info,
       0,
       FILE_PATH_LITERAL(""),
       preview_tab(),
       platform_util::GetTopLevel(preview_tab()->GetNativeView()),
       NULL);
 }
 
 void PrintPreviewHandler::OnTabDestroyed() {
-  TabContents* initiator_tab = GetInitiatorTab();
+  TabContentsWrapper* initiator_tab = GetInitiatorTab();
   if (!initiator_tab)
     return;
 
-  TabContentsWrapper* wrapper =
-      TabContentsWrapper::GetCurrentWrapperForContents(initiator_tab);
-  wrapper->print_view_manager()->set_observer(NULL);
+  initiator_tab->print_view_manager()->set_observer(NULL);
 }
 
 void PrintPreviewHandler::OnPrintPreviewFailed() {
   if (reported_failed_preview_)
     return;
   reported_failed_preview_ = true;
   ReportUserActionHistogram(PREVIEW_FAILED);
 }
 
 void PrintPreviewHandler::ShowSystemDialog() {
@@ skipping 30 matching lines @@
 
 void PrintPreviewHandler::HidePreviewTab() {
   TabContentsWrapper* preview_tab_wrapper =
       TabContentsWrapper::GetCurrentWrapperForContents(preview_tab());
   if (GetBackgroundPrintingManager()->HasTabContents(preview_tab_wrapper))
     return;
   GetBackgroundPrintingManager()->OwnTabContents(preview_tab_wrapper);
 }
 
 void PrintPreviewHandler::ClearInitiatorTabDetails() {
-  TabContents* initiator_tab = GetInitiatorTab();
+  TabContentsWrapper* initiator_tab = GetInitiatorTab();
   if (!initiator_tab)
     return;
 
   // We no longer require the initiator tab details. Remove those details
   // associated with the preview tab to allow the initiator tab to create
   // another preview tab.
   printing::PrintPreviewTabController* tab_controller =
      printing::PrintPreviewTabController::GetInstance();
   if (tab_controller)
-    tab_controller->EraseInitiatorTabInfo(preview_tab());
+    tab_controller->EraseInitiatorTabInfo(preview_tab_wrapper());
 }