src/chrome/browser/ui/views/color_chooser_win.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include <windows.h>
   6
   7 #include "chrome/browser/platform_util.h"
   8 #include "chrome/browser/ui/browser_dialogs.h"
   9 #include "chrome/browser/ui/host_desktop.h"
  10 #include "chrome/browser/ui/views/color_chooser_aura.h"
  11 #include "chrome/browser/ui/views/color_chooser_dialog.h"
  12 #include "content/public/browser/color_chooser.h"
  13 #include "content/public/browser/render_view_host.h"
  14 #include "content/public/browser/render_widget_host_view.h"
  15 #include "content/public/browser/web_contents.h"
  16 #include "ui/views/color_chooser/color_chooser_listener.h"
  17
  18 class ColorChooserWin : public content::ColorChooser,
  19                         public views::ColorChooserListener {
  20  public:
  21   static ColorChooserWin* Open(content::WebContents* web_contents,
  22                                SkColor initial_color);
  23
  24   ColorChooserWin(content::WebContents* web_contents,
  25                   SkColor initial_color);
  26   ~ColorChooserWin();
  27
  28   // content::ColorChooser overrides:
  29   virtual void End() override;
  30   virtual void SetSelectedColor(SkColor color) override {}
  31
  32   // views::ColorChooserListener overrides:
  33   virtual void OnColorChosen(SkColor color);
  34   virtual void OnColorChooserDialogClosed();
  35
  36  private:
  37   static ColorChooserWin* current_color_chooser_;
  38
  39   // The web contents invoking the color chooser.  No ownership. because it will
  40   // outlive this class.
  41   content::WebContents* web_contents_;
  42
  43   // The color chooser dialog which maintains the native color chooser UI.
  44   scoped_refptr<ColorChooserDialog> color_chooser_dialog_;
  45 };
  46
  47 ColorChooserWin* ColorChooserWin::current_color_chooser_ = NULL;
  48
  49 ColorChooserWin* ColorChooserWin::Open(content::WebContents* web_contents,
  50                                        SkColor initial_color) {
  51   if (current_color_chooser_)
  52     return NULL;
  53   current_color_chooser_ = new ColorChooserWin(web_contents, initial_color);
  54   return current_color_chooser_;
  55 }
  56
  57 ColorChooserWin::ColorChooserWin(content::WebContents* web_contents,
  58                                  SkColor initial_color)
  59     : web_contents_(web_contents) {
  60   gfx::NativeWindow owning_window = platform_util::GetTopLevel(
  61       web_contents->GetRenderViewHost()->GetView()->GetNativeView());
  62   color_chooser_dialog_ = new ColorChooserDialog(this,
  63                                                  initial_color,
  64                                                  owning_window);
  65 }
  66
  67 ColorChooserWin::~ColorChooserWin() {
  68   // Always call End() before destroying.
  69   DCHECK(!color_chooser_dialog_);
  70 }
  71
  72 void ColorChooserWin::End() {
  73   // The ColorChooserDialog's listener is going away.  Ideally we'd
  74   // programmatically close the dialog at this point.  Since that's impossible,
  75   // we instead tell the dialog its listener is going away, so that the dialog
  76   // doesn't try to communicate with a destroyed listener later.  (We also tell
  77   // the renderer the dialog is closed, since from the renderer's perspective
  78   // it effectively is.)
  79   OnColorChooserDialogClosed();
  80 }
  81
  82 void ColorChooserWin::OnColorChosen(SkColor color) {
  83   if (web_contents_)
  84     web_contents_->DidChooseColorInColorChooser(color);
  85 }
  86
  87 void ColorChooserWin::OnColorChooserDialogClosed() {
  88   if (color_chooser_dialog_.get()) {
  89     color_chooser_dialog_->ListenerDestroyed();
  90     color_chooser_dialog_ = NULL;
  91   }
  92   DCHECK(current_color_chooser_ == this);
  93   current_color_chooser_ = NULL;
  94   if (web_contents_)
  95     web_contents_->DidEndColorChooser();
  96 }
  97
  98 namespace chrome {
  99
 100 content::ColorChooser* ShowColorChooser(content::WebContents* web_contents,
 101                                         SkColor initial_color) {
 102   gfx::NativeView native_view = web_contents->GetNativeView();
 103   if (GetHostDesktopTypeForNativeView(native_view) == HOST_DESKTOP_TYPE_ASH)
 104     return ColorChooserAura::Open(web_contents, initial_color);
 105
 106   return ColorChooserWin::Open(web_contents, initial_color);
 107 }
 108
 109 }  // namespace chrome