Print Preview: Make print preview tab modal. (try 2)

chrome/browser/ui/webui/print_preview_ui.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_ui.h"
 
 #include <map>
 
 #include "base/lazy_instance.h"
 #include "base/metrics/histogram.h"
 #include "base/string_util.h"
 #include "base/synchronization/lock.h"
 #include "base/values.h"
+#include "chrome/browser/browser_process.h"
+#include "chrome/browser/printing/background_printing_manager.h"
 #include "chrome/browser/printing/print_preview_data_service.h"
 #include "chrome/browser/profiles/profile.h"
+#include "chrome/browser/ui/tab_contents/tab_contents_wrapper.h"
+#include "chrome/browser/ui/webui/html_dialog_ui.h"
 #include "chrome/browser/ui/webui/print_preview_data_source.h"
 #include "chrome/browser/ui/webui/print_preview_handler.h"
 #include "chrome/common/print_messages.h"
 #include "content/browser/tab_contents/tab_contents.h"
 #include "printing/page_size_margins.h"
 #include "printing/print_job_constants.h"
 
 using printing::PageSizeMargins;
 
 namespace {
@@ skipping 34 matching lines @@
   base::Lock lock_;
 };
 
 // Written to on the UI thread, read from any thread.
 base::LazyInstance<PrintPreviewRequestIdMapWithLock>
     g_print_preview_request_id_map(base::LINKER_INITIALIZED);
 
 }  // namespace
 
 PrintPreviewUI::PrintPreviewUI(TabContents* contents)
-    : ChromeWebUI(contents),
-      initial_preview_start_time_(base::TimeTicks::Now()) {
+    : ConstrainedHtmlUI(contents),
+      initial_preview_start_time_(base::TimeTicks::Now()),
+      tab_closed_(false) {
   // WebUI owns |handler_|.
   handler_ = new PrintPreviewHandler();
   AddMessageHandler(handler_->Attach(this));
 
   // Set up the chrome://print/ data source.
   Profile* profile = Profile::FromBrowserContext(contents->browser_context());
   profile->GetChromeURLDataManager()->AddDataSource(
       new PrintPreviewDataSource());
 
   preview_ui_addr_str_ = GetPrintPreviewUIAddress();
@@ skipping 52 matching lines @@
 }
 
 std::string PrintPreviewUI::GetPrintPreviewUIAddress() const {
   // Store the PrintPreviewUIAddress as a string.
   // "0x" + deadc0de + '\0' = 2 + 2 * sizeof(this) + 1;
   char preview_ui_addr[2 + (2 * sizeof(this)) + 1];
   base::snprintf(preview_ui_addr, sizeof(preview_ui_addr), "%p", this);
   return preview_ui_addr;
 }
 
-void PrintPreviewUI::OnInitiatorTabCrashed() {
-  StringValue initiator_tab_url(initiator_url_);
-  CallJavascriptFunction("onInitiatorTabCrashed", initiator_tab_url);
+void PrintPreviewUI::OnPrintPreviewTabClosed() {
+  TabContentsWrapper* preview_tab =
+      TabContentsWrapper::GetCurrentWrapperForContents(tab_contents());
+  printing::BackgroundPrintingManager* background_printing_manager =
+      g_browser_process->background_printing_manager();
+  if (background_printing_manager->HasPrintPreviewTab(preview_tab))
+    return;
+  OnClosePrintPreviewTab();
 }
 
 void PrintPreviewUI::OnInitiatorTabClosed() {
-  StringValue initiator_tab_url(initiator_url_);
-  CallJavascriptFunction("onInitiatorTabClosed", initiator_tab_url);
+  TabContentsWrapper* preview_tab =
+      TabContentsWrapper::GetCurrentWrapperForContents(tab_contents());
+  printing::BackgroundPrintingManager* background_printing_manager =
+      g_browser_process->background_printing_manager();
+  if (background_printing_manager->HasPrintPreviewTab(preview_tab))
+    CallJavascriptFunction("cancelPendingPrintRequest");
+  else
+    OnClosePrintPreviewTab();
 }
 
 void PrintPreviewUI::OnPrintPreviewRequest(int request_id) {
   g_print_preview_request_id_map.Get().Set(preview_ui_addr_str_, request_id);
 }
 
 void PrintPreviewUI::OnShowSystemDialog() {
   CallJavascriptFunction("onSystemDialogLinkClicked");
 }
 
 void PrintPreviewUI::OnDidGetPreviewPageCount(
     const PrintHostMsg_DidGetPreviewPageCount_Params& params) {
   DCHECK_GT(params.page_count, 0);
   base::FundamentalValue count(params.page_count);
   base::FundamentalValue request_id(params.preview_request_id);
-  CallJavascriptFunction("onDidGetPreviewPageCount", count,request_id);
+  CallJavascriptFunction("onDidGetPreviewPageCount", count, request_id);
 }
 
 void PrintPreviewUI::OnDidGetDefaultPageLayout(
     const PageSizeMargins& page_layout) {
   if (page_layout.margin_top < 0 || page_layout.margin_left < 0 ||
       page_layout.margin_bottom < 0 || page_layout.margin_right < 0 ||
       page_layout.content_width < 0 || page_layout.content_height < 0) {
     NOTREACHED();
     return;
   }
@@ skipping 60 matching lines @@
   CallJavascriptFunction("printPreviewFailed");
 }
 
 void PrintPreviewUI::OnInvalidPrinterSettings() {
   CallJavascriptFunction("invalidPrinterSettings");
 }
 
 PrintPreviewDataService* PrintPreviewUI::print_preview_data_service() {
   return PrintPreviewDataService::GetInstance();
 }
+
+void PrintPreviewUI::OnHidePreviewTab() {
+  TabContentsWrapper* preview_tab =
+      TabContentsWrapper::GetCurrentWrapperForContents(tab_contents());
+  printing::BackgroundPrintingManager* background_printing_manager =
+      g_browser_process->background_printing_manager();
+  if (background_printing_manager->HasPrintPreviewTab(preview_tab))
+    return;
+
+  ConstrainedHtmlUIDelegate* delegate = GetConstrainedDelegate();
+  if (!delegate)
+    return;
+  delegate->ReleaseTabContentsOnDialogClose();
+  background_printing_manager->OwnPrintPreviewTab(preview_tab);
+  OnClosePrintPreviewTab();
+}
+
+void PrintPreviewUI::OnClosePrintPreviewTab() {
+  if (tab_closed_)
+    return;
+  tab_closed_ = true;
+  ConstrainedHtmlUIDelegate* delegate = GetConstrainedDelegate();
+  if (!delegate)
+    return;
+  delegate->GetHtmlDialogUIDelegate()->OnDialogClosed("");
+  delegate->OnDialogCloseFromWebUI();
+}
+
+void PrintPreviewUI::OnReloadPrintersList() {
+  CallJavascriptFunction("reloadPrintersList");
+}