Implement PrintBrowser functionality.

chrome/browser/printing/print_preview_dialog_controller.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_preview_dialog_controller.h"
 
 #include <stddef.h>
 
 #include <algorithm>
 #include <string>
 #include <vector>
 
 #include "base/auto_reset.h"
+#include "base/command_line.h"
 #include "base/macros.h"
 #include "base/path_service.h"
 #include "base/strings/utf_string_conversions.h"
 #include "build/build_config.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/chrome_notification_types.h"
 #include "chrome/browser/extensions/chrome_extension_web_contents_observer.h"
 #include "chrome/browser/printing/print_view_manager.h"
 #include "chrome/browser/task_manager/web_contents_tags.h"
 #include "chrome/browser/ui/browser.h"
+#include "chrome/browser/ui/browser_commands.h"
 #include "chrome/browser/ui/browser_finder.h"
+#include "chrome/browser/ui/browser_list.h"
+#include "chrome/browser/ui/browser_navigator.h"
 #include "chrome/browser/ui/browser_window.h"
+#include "chrome/browser/ui/tab_contents/core_tab_helper.h"
 #include "chrome/browser/ui/webui/chrome_web_contents_handler.h"
 #include "chrome/browser/ui/webui/constrained_web_dialog_ui.h"
 #include "chrome/browser/ui/webui/print_preview/print_preview_ui.h"
 #include "chrome/common/chrome_paths.h"
+#include "chrome/common/chrome_switches.h"
 #include "chrome/common/url_constants.h"
 #include "components/guest_view/browser/guest_view_base.h"
 #include "components/web_modal/web_contents_modal_dialog_host.h"
 #include "content/public/browser/host_zoom_map.h"
 #include "content/public/browser/navigation_controller.h"
 #include "content/public/browser/navigation_details.h"
 #include "content/public/browser/navigation_entry.h"
 #include "content/public/browser/notification_details.h"
 #include "content/public/browser/notification_source.h"
 #include "content/public/browser/web_contents.h"
@@ skipping 296 matching lines @@
 
       // Cloud print sign-in causes a reload.
       if (!waiting_for_new_preview_page_ &&
           ui::PageTransitionCoreTypeIs(transition_type,
                                        ui::PAGE_TRANSITION_RELOAD) &&
           nav_type == content::NAVIGATION_TYPE_EXISTING_PAGE &&
           IsPrintPreviewURL(details->previous_url)) {
         return;
       }
     }
+#if BUILDFLAG(ENABLE_PRINT_BROWSER)
+    // If in PrintBrowser mode, navigation away from the print UI is fine
+    base::CommandLine* command_line = base::CommandLine::ForCurrentProcess();
+    bool print_browser_enabled =
+        command_line->HasSwitch(switches::kEnablePrintBrowser);
+    if (print_browser_enabled)
+      return;
+#endif
     NOTREACHED();
     return;
   }
   if (details) {
     ui::PageTransition type = details->entry->GetTransitionType();
     content::NavigationType nav_type = details->type;
     if (nav_type == content::NAVIGATION_TYPE_EXISTING_PAGE &&
         (ui::PageTransitionTypeIncludingQualifiersIs(
              type,
              ui::PageTransitionFromInt(ui::PAGE_TRANSITION_TYPED |
                                        ui::PAGE_TRANSITION_FROM_ADDRESS_BAR)) ||
          ui::PageTransitionTypeIncludingQualifiersIs(type,
                                                      ui::PAGE_TRANSITION_LINK)))
       return;
   }
 
   RemoveInitiator(contents);
 }
 
 WebContents* PrintPreviewDialogController::CreatePrintPreviewDialog(
     WebContents* initiator) {
   base::AutoReset<bool> auto_reset(&is_creating_print_preview_dialog_, true);
 
-  // The dialog delegates are deleted when the dialog is closed.
-  ConstrainedWebDialogDelegate* web_dialog_delegate =
-      ShowConstrainedWebDialog(initiator->GetBrowserContext(),
-                               new PrintPreviewDialogDelegate(initiator),
-                               initiator);
+  WebContents* preview_dialog;
 
-  WebContents* preview_dialog = web_dialog_delegate->GetWebContents();
+#if BUILDFLAG(ENABLE_PRINT_BROWSER)
+  base::CommandLine* command_line = base::CommandLine::ForCurrentProcess();
+  bool print_browser_enabled =
+      command_line->HasSwitch(switches::kEnablePrintBrowser);
+  if (!print_browser_enabled) {
+#endif
+    // The dialog delegates are deleted when the dialog is closed.
+    ConstrainedWebDialogDelegate* web_dialog_delegate =
+        ShowConstrainedWebDialog(initiator->GetBrowserContext(),
+                                 new PrintPreviewDialogDelegate(initiator),
+                                 initiator);
+
+    preview_dialog = web_dialog_delegate->GetWebContents();
+#if BUILDFLAG(ENABLE_PRINT_BROWSER)
+  } else {
+    // If in PrintBrowser mode, swap |initiator| with a newly-created
+    // |preview_dialog| and navigate |preview_dialog| to the print UI
+    Browser* browser =
+        chrome::FindBrowserWithWindow(initiator->GetTopLevelNativeWindow());
+    preview_dialog =
+        WebContents::Create(WebContents::CreateParams(browser->profile()));
+    CoreTabHelper::FromWebContents(initiator)->delegate()->SwapTabContents(

[rbpotter, 2016/12/16 20:05:28]

I think you need to ensure the initiator web contents are destroyed at an
appropriate point after calling this.

+        initiator, preview_dialog, false, false);
+    browser->OpenURL(content::OpenURLParams(
+        GURL(chrome::kChromeUIPrintURL), content::Referrer(),
+        WindowOpenDisposition::CURRENT_TAB, ui::PAGE_TRANSITION_AUTO_TOPLEVEL,
+        false));
+  }
+#endif
 
   // Clear the zoom level for the print preview dialog so it isn't affected by
   // the default zoom level. This also controls the zoom level of the OOP PDF
   // extension when iframed by the print preview dialog.
   GURL print_url(chrome::kChromeUIPrintURL);
   content::HostZoomMap::Get(preview_dialog->GetSiteInstance())
       ->SetZoomLevelForHostAndScheme(print_url.scheme(), print_url.host(), 0);
   PrintViewManager::CreateForWebContents(preview_dialog);
   extensions::ChromeExtensionWebContentsObserver::CreateForWebContents(
       preview_dialog);
@@ skipping 101 matching lines @@
   if (initiator) {
     RemoveObservers(initiator);
     PrintViewManager::FromWebContents(initiator)->PrintPreviewDone();
   }
 
   preview_dialog_map_.erase(preview_dialog);
   RemoveObservers(preview_dialog);
 }
 
 }  // namespace printing