Switch PrintViewManager, PrintPreviewMessageHandler to use WebContentsUserData.

chrome/browser/printing/print_view_manager.cc

 // Copyright (c) 2012 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/command_line.h"
@@ skipping 27 matching lines @@
 #include "grit/generated_resources.h"
 #include "printing/metafile.h"
 #include "printing/metafile_impl.h"
 #include "printing/print_destination_interface.h"
 #include "printing/printed_document.h"
 #include "ui/base/l10n/l10n_util.h"
 
 using base::TimeDelta;
 using content::BrowserThread;
 
+int printing::PrintViewManager::kUserDataKey;
+
 namespace {
 
 // Keeps track of pending scripted print preview closures.
 // No locking, only access on the UI thread.
 typedef std::map<content::RenderProcessHost*, base::Closure>
     ScriptedPrintPreviewClosureMap;
 static base::LazyInstance<ScriptedPrintPreviewClosureMap>
     g_scripted_print_preview_closure_map = LAZY_INSTANCE_INITIALIZER;
 
 // Limits memory usage by raster to 64 MiB.
 const int kMaxRasterSizeInPixels = 16*1024*1024;
 
 }  // namespace
 
 namespace printing {
 
-PrintViewManager::PrintViewManager(TabContents* tab)
-    : content::WebContentsObserver(tab->web_contents()),
-      tab_(tab),
+PrintViewManager::PrintViewManager(content::WebContents* web_contents)
+    : content::WebContentsObserver(web_contents),
       number_pages_(0),
       printing_succeeded_(false),
       inside_inner_message_loop_(false),
       observer_(NULL),
       cookie_(0),
       print_preview_state_(NOT_PREVIEWING),
       scripted_print_preview_rph_(NULL),
       tab_content_blocked_(false) {
 #if defined(OS_POSIX) && !defined(OS_MACOSX)
   expecting_first_page_ = true;
 #endif
   registrar_.Add(this, chrome::NOTIFICATION_CONTENT_BLOCKED_STATE_CHANGED,
-                 content::Source<TabContents>(tab));
+                 content::Source<content::WebContents>(web_contents));
+  Profile* profile =
+      Profile::FromBrowserContext(web_contents->GetBrowserContext());
   printing_enabled_.Init(prefs::kPrintingEnabled,
-                         tab->profile()->GetPrefs(),
+                         profile->GetPrefs(),
                          this);
 }
 
 PrintViewManager::~PrintViewManager() {
   DCHECK_EQ(NOT_PREVIEWING, print_preview_state_);
   ReleasePrinterQuery();
   DisconnectFromCurrentPrintJob();
 }
 
 bool PrintViewManager::PrintNow() {
   return PrintNowInternal(new PrintMsg_PrintPages(routing_id()));
 }
 
 bool PrintViewManager::PrintForSystemDialogNow() {
   return PrintNowInternal(new PrintMsg_PrintForSystemDialog(routing_id()));
 }
 
 bool PrintViewManager::AdvancedPrintNow() {
   PrintPreviewTabController* tab_controller =
       PrintPreviewTabController::GetInstance();
   if (!tab_controller)
     return false;
-  TabContents* print_preview_tab = tab_controller->GetPrintPreviewForTab(tab_);
+  TabContents* print_preview_tab = tab_controller->GetPrintPreviewForTab(
+      TabContents::FromWebContents(web_contents()));
   if (print_preview_tab) {
     // Preview tab exist for current tab or current tab is preview tab.
     if (!print_preview_tab->web_contents()->GetWebUI())
       return false;
     PrintPreviewUI* print_preview_ui = static_cast<PrintPreviewUI*>(
         print_preview_tab->web_contents()->GetWebUI()->GetController());
     print_preview_ui->OnShowSystemDialog();
     return true;
   } else {
     return PrintNow();
@@ skipping 172 matching lines @@
   ShouldQuitFromInnerMessageLoop();
 }
 
 void PrintViewManager::OnPrintingFailed(int cookie) {
   if (cookie != cookie_) {
     NOTREACHED();
     return;
   }
 
   chrome::ShowPrintErrorDialog(
-      tab_->web_contents()->GetView()->GetTopLevelNativeWindow());
+      web_contents()->GetView()->GetTopLevelNativeWindow());
 
   ReleasePrinterQuery();
 
   content::NotificationService::current()->Notify(
       chrome::NOTIFICATION_PRINT_JOB_RELEASED,
-      content::Source<TabContents>(tab_),
+      content::Source<TabContents>(
+          TabContents::FromWebContents(web_contents())),
       content::NotificationService::NoDetails());
 }
 
 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 = web_contents()->GetRenderProcessHost();
 
@@ skipping 19 matching lines @@
   }
 
   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;
 
-  tab_controller->PrintPreview(tab_);
+  TabContents* tab_contents = TabContents::FromWebContents(web_contents());
+  tab_controller->PrintPreview(tab_contents);
   PrintPreviewUI::SetSourceIsModifiable(
-      tab_controller->GetPrintPreviewForTab(tab_),
+      tab_controller->GetPrintPreviewForTab(tab_contents),
       source_is_modifiable);
 }
 
 void PrintViewManager::OnScriptedPrintPreviewReply(IPC::Message* reply_msg) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
   Send(reply_msg);
 }
 
 void PrintViewManager::DidStartLoading(
     content::RenderViewHost* render_view_host) {
@@ skipping 42 matching lines @@
 }
 
 void PrintViewManager::OnNotifyPrintJobEvent(
     const JobEventDetails& event_details) {
   switch (event_details.type()) {
     case JobEventDetails::FAILED: {
       TerminatePrintJob(true);
 
       content::NotificationService::current()->Notify(
           chrome::NOTIFICATION_PRINT_JOB_RELEASED,
-          content::Source<TabContents>(tab_),
+          content::Source<TabContents>(
+              TabContents::FromWebContents(web_contents())),
           content::NotificationService::NoDetails());
       break;
     }
     case JobEventDetails::USER_INIT_DONE:
     case JobEventDetails::DEFAULT_INIT_DONE:
     case JobEventDetails::USER_INIT_CANCELED: {
       NOTREACHED();
       break;
     }
     case JobEventDetails::ALL_PAGES_REQUESTED: {
       ShouldQuitFromInnerMessageLoop();
       break;
     }
     case JobEventDetails::NEW_DOC:
     case JobEventDetails::NEW_PAGE:
     case JobEventDetails::PAGE_DONE:
     case JobEventDetails::DOC_DONE: {
       // Don't care about the actual printing process.
       break;
     }
     case JobEventDetails::JOB_DONE: {
       // Printing is done, we don't need it anymore.
       // print_job_->is_job_pending() may still be true, depending on the order
       // of object registration.
       printing_succeeded_ = true;
       ReleasePrintJob();
 
       content::NotificationService::current()->Notify(
           chrome::NOTIFICATION_PRINT_JOB_RELEASED,
-          content::Source<TabContents>(tab_),
+          content::Source<TabContents>(
+              TabContents::FromWebContents(web_contents())),
           content::NotificationService::NoDetails());
       break;
     }
     default: {
       NOTREACHED();
       break;
     }
   }
 }
 
@@ skipping 226 matching lines @@
   scoped_refptr<printing::PrinterQuery> printer_query;
   print_job_manager->PopPrinterQuery(cookie, &printer_query);
   if (!printer_query.get())
     return;
   BrowserThread::PostTask(
       BrowserThread::IO, FROM_HERE,
       base::Bind(&PrinterQuery::StopWorker, printer_query.get()));
 }
 
 }  // namespace printing