Make WebUI objects not derive from WebUI. WebUI objects own the controller. This is the ownership...

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"
 #include "base/memory/ref_counted_memory.h"
 #include "base/shared_memory.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/printing/print_job_manager.h"
 #include "chrome/browser/printing/print_preview_tab_controller.h"
 #include "chrome/browser/printing/print_view_manager.h"
 #include "chrome/browser/printing/printer_query.h"
 #include "chrome/browser/ui/tab_contents/tab_contents_wrapper.h"
 #include "chrome/browser/ui/webui/print_preview/print_preview_ui.h"
 #include "chrome/common/print_messages.h"
 #include "content/browser/renderer_host/render_view_host.h"
+#include "content/browser/webui/web_ui.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/web_contents.h"
 #include "printing/page_size_margins.h"
 #include "printing/print_job_constants.h"
 
 using content::BrowserThread;
 using content::NavigationController;
 using content::WebContents;
 
 namespace {
@@ skipping 58 matching lines @@
   // Always need to stop the worker.
   StopWorker(document_cookie);
 
   // 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_contents()->GetWebUI())
     return NULL;
 
   return static_cast<PrintPreviewUI*>(
-      print_preview_tab->web_contents()->GetWebUI());
+      print_preview_tab->web_contents()->GetWebUI()->GetController());
 }
 
 void PrintPreviewMessageHandler::OnRequestPrintPreview(
     bool source_is_modifiable) {
   PrintPreviewTabController::PrintPreview(tab_contents_wrapper());
   PrintPreviewUI::SetSourceIsModifiable(GetPrintPreviewTab(),
                                         source_is_modifiable);
 }
 
 void PrintPreviewMessageHandler::OnDidGetPreviewPageCount(
     const PrintHostMsg_DidGetPreviewPageCount_Params& params) {
   if (params.page_count <= 0) {
     NOTREACHED();
     return;
   }
 
   TabContentsWrapper* print_preview_tab = GetPrintPreviewTab();
   if (!print_preview_tab || !print_preview_tab->web_contents()->GetWebUI())
     return;
 
   PrintPreviewUI* print_preview_ui = static_cast<PrintPreviewUI*>(
-      print_preview_tab->web_contents()->GetWebUI());
+      print_preview_tab->web_contents()->GetWebUI()->GetController());
 
   if (!params.is_modifiable || params.clear_preview_data)
     print_preview_ui->ClearAllPreviewData();
 
   print_preview_ui->OnDidGetPreviewPageCount(params);
 }
 
 void PrintPreviewMessageHandler::OnDidPreviewPage(
     const PrintHostMsg_DidPreviewPage_Params& params) {
   TabContentsWrapper* print_preview_tab = GetPrintPreviewTab();
   if (!print_preview_tab || !print_preview_tab->web_contents()->GetWebUI())
     return;
 
   PrintPreviewUI* print_preview_ui = static_cast<PrintPreviewUI*>(
-      print_preview_tab->web_contents()->GetWebUI());
+      print_preview_tab->web_contents()->GetWebUI()->GetController());
   int page_number = params.page_number;
   if (page_number >= FIRST_PAGE_INDEX && params.data_size) {
     RefCountedBytes* data_bytes =
         GetDataFromHandle(params.metafile_data_handle, params.data_size);
     DCHECK(data_bytes);
 
     print_preview_ui->SetPrintPreviewDataForIndex(page_number, data_bytes);
     print_preview_ui->OnDidPreviewPage(page_number, params.preview_request_id);
   }
 }
 
 void PrintPreviewMessageHandler::OnMetafileReadyForPrinting(
     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.
   TabContentsWrapper* print_preview_tab = GetPrintPreviewTab();
   // User might have closed it already.
   if (!print_preview_tab || !print_preview_tab->web_contents()->GetWebUI())
     return;
 
   PrintPreviewUI* print_preview_ui = static_cast<PrintPreviewUI*>(
-      print_preview_tab->web_contents()->GetWebUI());
+      print_preview_tab->web_contents()->GetWebUI()->GetController());
 
   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);
@@ skipping 22 matching lines @@
 }
 
 void PrintPreviewMessageHandler::OnDidGetDefaultPageLayout(
     const PageSizeMargins& page_layout_in_points,
     bool has_custom_page_size_style) {
   TabContentsWrapper* print_preview_tab = GetPrintPreviewTab();
   if (!print_preview_tab || !print_preview_tab->web_contents()->GetWebUI())
     return;
 
   PrintPreviewUI* print_preview_ui = static_cast<PrintPreviewUI*>(
-      print_preview_tab->web_contents()->GetWebUI());
+      print_preview_tab->web_contents()->GetWebUI()->GetController());
   print_preview_ui->OnDidGetDefaultPageLayout(page_layout_in_points,
                                               has_custom_page_size_style);
 }
 
 void PrintPreviewMessageHandler::OnPrintPreviewCancelled(int document_cookie) {
   // Always need to stop the worker.
   StopWorker(document_cookie);
 }
 
 void PrintPreviewMessageHandler::OnInvalidPrinterSettings(int document_cookie) {
@@ skipping 40 matching lines @@
     return;
   }
   // If |tab| is navigating and it has a print preview tab, notify |tab| to
   // consider print preview done so it unfreezes the renderer in the case of
   // window.print().
   if (preview_tab)
     tab->print_view_manager()->PrintPreviewDone();
 }
 
 }  // namespace printing