Enabled system print dialog on Windows Aura.

printing/printing_context_win.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 "printing/printing_context_win.h"
 
 #include <winspool.h>
 
 #include <algorithm>
 
 #include "base/i18n/file_util_icu.h"
 #include "base/i18n/time_formatting.h"
 #include "base/message_loop.h"
 #include "base/metrics/histogram.h"
 #include "base/time.h"
 #include "base/utf_string_conversions.h"
 #include "base/values.h"
 #include "base/win/metro.h"
 #include "printing/backend/print_backend.h"
 #include "printing/backend/printing_info_win.h"
 #include "printing/backend/win_helper.h"
 #include "printing/print_job_constants.h"
 #include "printing/print_settings_initializer_win.h"
 #include "printing/printed_document.h"
 #include "printing/units.h"
 #include "skia/ext/platform_device.h"
 #include "win8/util/win8_util.h"
 
+#if defined(USE_AURA)
+#include "ui/aura/remote_root_window_host_win.h"
+#include "ui/aura/root_window.h"
+#include "ui/aura/window.h"
+#endif
+
 using base::Time;
 
 namespace {
 
 // Constants for setting default PDF settings.
 const int kPDFDpi = 300;  // 300 dpi
 // LETTER: 8.5 x 11 inches
 const int kPDFLetterWidth = 8.5 * kPDFDpi;
 const int kPDFLetterHeight = 11 * kPDFDpi;
 // LEGAL: 8.5 x 14 inches
 const int kPDFLegalWidth = 8.5 * kPDFDpi;
 const int kPDFLegalHeight = 14 * kPDFDpi;
 // A4: 8.27 x 11.69 inches
 const int kPDFA4Width = 8.27 * kPDFDpi;
 const int kPDFA4Height = 11.69 * kPDFDpi;
 // A3: 11.69 x 16.54 inches
 const int kPDFA3Width = 11.69 * kPDFDpi;
 const int kPDFA3Height = 16.54 * kPDFDpi;
 
+HWND GetRootWindow(gfx::NativeView view) {
+  HWND window = NULL;
+#if defined(USE_AURA)
+  if (view)
+    window = view->GetRootWindow()->GetAcceleratedWidget();
+#else
+  if (view && IsWindow(view)) {
+    window = GetAncestor(view, GA_ROOTOWNER);
+  }
+#endif
+  if (!window) {
+    // TODO(maruel):  bug 1214347 Get the right browser window instead.
+    return GetDesktopWindow();
+  }
+  return window;
+}
+
 }  // anonymous namespace
 
 namespace printing {
 
 class PrintingContextWin::CallbackHandler : public IPrintDialogCallback,
                                             public IObjectWithSite {
  public:
   CallbackHandler(PrintingContextWin& owner, HWND owner_hwnd)
       : owner_(owner),
         owner_hwnd_(owner_hwnd),
@@ skipping 101 matching lines @@
     : PrintingContext(app_locale),
       context_(NULL),
       dialog_box_(NULL),
       print_dialog_func_(&PrintDlgEx) {
 }
 
 PrintingContextWin::~PrintingContextWin() {
   ReleaseContext();
 }
 
+// TODO(vitalybuka): Implement as ui::BaseShellDialog crbug.com/180997.
 void PrintingContextWin::AskUserForSettings(
     gfx::NativeView view, int max_pages, bool has_selection,
     const PrintSettingsCallback& callback) {
-#if !defined(USE_AURA)
   DCHECK(!in_print_job_);
-
   if (win8::IsSingleWindowMetroMode()) {
     // The system dialog can not be opened while running in Metro.
     // But we can programatically launch the Metro print device charm though.
     HMODULE metro_module = base::win::GetMetroModule();
     if (metro_module != NULL) {
       typedef void (*MetroShowPrintUI)();
       MetroShowPrintUI metro_show_print_ui =
           reinterpret_cast<MetroShowPrintUI>(
               ::GetProcAddress(metro_module, "MetroShowPrintUI"));
       if (metro_show_print_ui) {
         // TODO(mad): Remove this once we can send user metrics from the metro
         // driver. crbug.com/142330
         UMA_HISTOGRAM_ENUMERATION("Metro.Print", 1, 2);
         metro_show_print_ui();
       }
     }
     return callback.Run(CANCEL);
   }
   dialog_box_dismissed_ = false;
 
-  HWND window;
-  if (!view || !IsWindow(view)) {
-    // TODO(maruel):  bug 1214347 Get the right browser window instead.
-    window = GetDesktopWindow();
-  } else {
-    window = GetAncestor(view, GA_ROOTOWNER);
-  }
+  HWND window = GetRootWindow(view);
   DCHECK(window);
 
   // Show the OS-dependent dialog box.
   // If the user press
   // - OK, the settings are reset and reinitialized with the new settings. OK is
   //   returned.
   // - Apply then Cancel, the settings are reset and reinitialized with the new
   //   settings. CANCEL is returned.
   // - Cancel, the settings are not changed, the previous setting, if it was
   //   initialized before, are kept. CANCEL is returned.
@@ skipping 17 matching lines @@
     dialog_options.nPageRanges = 1;
     dialog_options.nMaxPageRanges = arraysize(ranges);
     dialog_options.nMinPage = 1;
     dialog_options.nMaxPage = max_pages;
     dialog_options.lpPageRanges = ranges;
   } else {
     // No need to bother, we don't know how many pages are available.
     dialog_options.Flags |= PD_NOPAGENUMS;
   }
 
-  if ((*print_dialog_func_)(&dialog_options) != S_OK) {
+  HRESULT hr = (*print_dialog_func_)(&dialog_options);
+  if (hr != S_OK) {
     ResetSettings();
     callback.Run(FAILED);
   }
 
   // TODO(maruel):  Support PD_PRINTTOFILE.
   callback.Run(ParseDialogResultEx(dialog_options));
-#endif
 }
 
 PrintingContext::Result PrintingContextWin::UseDefaultSettings() {
   DCHECK(!in_print_job_);
 
   PRINTDLG dialog_options = { sizeof(PRINTDLG) };
   dialog_options.Flags = PD_RETURNDC | PD_RETURNDEFAULT;
   if (PrintDlg(&dialog_options))
     return ParseDialogResult(dialog_options);
 
@@ skipping 496 matching lines @@
 
   if (dialog_options.hDevMode != NULL)
     GlobalFree(dialog_options.hDevMode);
   if (dialog_options.hDevNames != NULL)
     GlobalFree(dialog_options.hDevNames);
 
   return context_ ? OK : FAILED;
 }
 
 }  // namespace printing