chrome/browser/ui/views/toolbar/wrench_toolbar_button.cc

   1 // Copyright 2013 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 "chrome/browser/ui/views/toolbar/wrench_toolbar_button.h"
   6
   7 #include "base/location.h"
   8 #include "base/single_thread_task_runner.h"
   9 #include "base/thread_task_runner_handle.h"
  10 #include "base/time/time.h"
  11 #include "chrome/browser/ui/browser.h"
  12 #include "chrome/browser/ui/views/extensions/browser_action_drag_data.h"
  13 #include "chrome/browser/ui/views/toolbar/toolbar_view.h"
  14 #include "extensions/common/feature_switch.h"
  15 #include "grit/theme_resources.h"
  16 #include "ui/base/resource/resource_bundle.h"
  17 #include "ui/base/theme_provider.h"
  18 #include "ui/views/controls/button/label_button_border.h"
  19 #include "ui/views/metrics.h"
  20 #include "ui/views/painter.h"
  21
  22 // static
  23 bool WrenchToolbarButton::g_open_wrench_immediately_for_testing = false;
  24
  25 WrenchToolbarButton::WrenchToolbarButton(ToolbarView* toolbar_view)
  26     : views::MenuButton(NULL, base::string16(), toolbar_view, false),
  27       wrench_icon_painter_(new WrenchIconPainter(this)),
  28       toolbar_view_(toolbar_view),
  29       allow_extension_dragging_(
  30           extensions::FeatureSwitch::extension_action_redesign()->IsEnabled()),
  31       overflowed_toolbar_action_wants_to_run_for_testing_(false),
  32       weak_factory_(this) {
  33 }
  34
  35 WrenchToolbarButton::~WrenchToolbarButton() {
  36 }
  37
  38 void WrenchToolbarButton::SetSeverity(WrenchIconPainter::Severity severity,
  39                                       bool animate) {
  40   wrench_icon_painter_->SetSeverity(severity, animate);
  41   SchedulePaint();
  42 }
  43
  44 void WrenchToolbarButton::SetOverflowedToolbarActionWantsToRun(
  45     bool wants_to_run) {
  46   overflowed_toolbar_action_wants_to_run_for_testing_ = wants_to_run;
  47   SchedulePaint();
  48 }
  49
  50 gfx::Size WrenchToolbarButton::GetPreferredSize() const {
  51   return ResourceBundle::GetSharedInstance().
  52       GetImageSkiaNamed(IDR_TOOLBAR_BEZEL_HOVER)->size();
  53 }
  54
  55 void WrenchToolbarButton::ScheduleWrenchIconPaint() {
  56   SchedulePaint();
  57 }
  58
  59 const char* WrenchToolbarButton::GetClassName() const {
  60   return "WrenchToolbarButton";
  61 }
  62
  63 bool WrenchToolbarButton::GetDropFormats(
  64     int* formats, std::set<ui::OSExchangeData::CustomFormat>* custom_formats) {
  65   return allow_extension_dragging_ ?
  66       BrowserActionDragData::GetDropFormats(custom_formats) :
  67       views::View::GetDropFormats(formats, custom_formats);
  68 }
  69 bool WrenchToolbarButton::AreDropTypesRequired() {
  70   return allow_extension_dragging_ ?
  71       BrowserActionDragData::AreDropTypesRequired() :
  72       views::View::AreDropTypesRequired();
  73 }
  74 bool WrenchToolbarButton::CanDrop(const ui::OSExchangeData& data) {
  75   return allow_extension_dragging_ ?
  76       BrowserActionDragData::CanDrop(data,
  77                                      toolbar_view_->browser()->profile()) :
  78       views::View::CanDrop(data);
  79 }
  80
  81 void WrenchToolbarButton::OnDragEntered(const ui::DropTargetEvent& event) {
  82   DCHECK(allow_extension_dragging_);
  83   DCHECK(!weak_factory_.HasWeakPtrs());
  84   if (!g_open_wrench_immediately_for_testing) {
  85     base::ThreadTaskRunnerHandle::Get()->PostDelayedTask(
  86         FROM_HERE, base::Bind(&WrenchToolbarButton::ShowOverflowMenu,
  87                               weak_factory_.GetWeakPtr()),
  88         base::TimeDelta::FromMilliseconds(views::GetMenuShowDelay()));
  89   } else {
  90     ShowOverflowMenu();
  91   }
  92 }
  93
  94 int WrenchToolbarButton::OnDragUpdated(const ui::DropTargetEvent& event) {
  95   DCHECK(allow_extension_dragging_);
  96   return ui::DragDropTypes::DRAG_MOVE;
  97 }
  98
  99 void WrenchToolbarButton::OnDragExited() {
 100   DCHECK(allow_extension_dragging_);
 101   weak_factory_.InvalidateWeakPtrs();
 102 }
 103
 104 int WrenchToolbarButton::OnPerformDrop(const ui::DropTargetEvent& event) {
 105   DCHECK(allow_extension_dragging_);
 106   return ui::DragDropTypes::DRAG_MOVE;
 107 }
 108
 109 void WrenchToolbarButton::OnPaint(gfx::Canvas* canvas) {
 110   views::MenuButton::OnPaint(canvas);
 111   wrench_icon_painter_->Paint(canvas,
 112                               GetThemeProvider(),
 113                               gfx::Rect(size()),
 114                               WrenchIconPainter::BEZEL_NONE);
 115 }
 116
 117 void WrenchToolbarButton::ShowOverflowMenu() {
 118   toolbar_view_->ShowAppMenu(true);  // For drop.
 119 }