browser_action_overflow_menu_controller.cc revision 5d1f7b1de12d16ceb2c938c56701a3e8bfa558f7
1fa63f976522bd4faf19249e8c9ac4d3edda498d9Tom Stellard// Copyright (c) 2012 The Chromium Authors. All rights reserved. 2a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard// Use of this source code is governed by a BSD-style license that can be 3a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard// found in the LICENSE file. 4a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard 5a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard#include "chrome/browser/ui/views/extensions/browser_action_overflow_menu_controller.h" 6a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard 7a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard#include "base/message_loop/message_loop.h" 8a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard#include "base/strings/utf_string_conversions.h" 9a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard#include "chrome/browser/extensions/extension_action.h" 10b72ab79d73b29ec087d90cf2c698adbab4db5defTom Stellard#include "chrome/browser/extensions/extension_action_manager.h" 11a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard#include "chrome/browser/extensions/extension_context_menu_model.h" 12a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard#include "chrome/browser/profiles/profile.h" 13a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard#include "chrome/browser/ui/browser_list.h" 142f921101c0826dc52a2c69f85c3da0f7f6e8212aTom Stellard#include "chrome/browser/ui/views/extensions/browser_action_drag_data.h" 15a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard#include "chrome/browser/ui/views/toolbar/browser_action_view.h" 162f921101c0826dc52a2c69f85c3da0f7f6e8212aTom Stellard#include "chrome/browser/ui/views/toolbar/browser_actions_container.h" 172f921101c0826dc52a2c69f85c3da0f7f6e8212aTom Stellard#include "extensions/common/extension.h" 18a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard#include "ui/gfx/canvas.h" 19a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard#include "ui/views/controls/menu/menu_item_view.h" 20a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard#include "ui/views/controls/menu/menu_runner.h" 21a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard#include "ui/views/controls/menu/submenu_view.h" 222f921101c0826dc52a2c69f85c3da0f7f6e8212aTom Stellard#include "ui/views/widget/widget.h" 23a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard 24a75c6163e605f35b14f26930dd9227e4f337ec9eTom StellardBrowserActionOverflowMenuController::BrowserActionOverflowMenuController( 25a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard BrowserActionsContainer* owner, 26a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard Browser* browser, 27a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard views::MenuButton* menu_button, 282f921101c0826dc52a2c69f85c3da0f7f6e8212aTom Stellard const std::vector<BrowserActionView*>& views, 29a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard int start_index) 30a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard : owner_(owner), 31a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard browser_(browser), 32a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard observer_(NULL), 33a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard menu_button_(menu_button), 342f921101c0826dc52a2c69f85c3da0f7f6e8212aTom Stellard menu_(NULL), 35a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard views_(&views), 36a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard start_index_(start_index), 37a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard for_drop_(false) { 38a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard menu_ = new views::MenuItemView(this); 392f921101c0826dc52a2c69f85c3da0f7f6e8212aTom Stellard menu_runner_.reset(new views::MenuRunner(menu_)); 40a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard menu_->set_has_icons(true); 412f921101c0826dc52a2c69f85c3da0f7f6e8212aTom Stellard 42a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard size_t command_id = 1; // Menu id 0 is reserved, start with 1. 43a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard for (size_t i = start_index; i < views_->size(); ++i) { 44a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard BrowserActionView* view = (*views_)[i]; 45a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard menu_->AppendMenuItemWithIcon( 46a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard command_id, 47a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard base::UTF8ToUTF16(view->button()->extension()->name()), 482f921101c0826dc52a2c69f85c3da0f7f6e8212aTom Stellard view->GetIconWithBadge()); 49a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard 502f921101c0826dc52a2c69f85c3da0f7f6e8212aTom Stellard // Set the tooltip for this item. 51a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard base::string16 tooltip = base::UTF8ToUTF16( 52a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard extensions::ExtensionActionManager::Get(owner_->profile())-> 53a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard GetBrowserAction(*view->button()->extension())-> 54a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard GetTitle(owner_->GetCurrentTabId())); 55a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard menu_->SetTooltip(tooltip, command_id); 56a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard 572f921101c0826dc52a2c69f85c3da0f7f6e8212aTom Stellard ++command_id; 58a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard } 59a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard} 60a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard 61a75c6163e605f35b14f26930dd9227e4f337ec9eTom StellardBrowserActionOverflowMenuController::~BrowserActionOverflowMenuController() { 622f921101c0826dc52a2c69f85c3da0f7f6e8212aTom Stellard if (observer_) 63a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard observer_->NotifyMenuDeleted(this); 64a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard} 65a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard 66a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellardbool BrowserActionOverflowMenuController::RunMenu(views::Widget* window, 672f921101c0826dc52a2c69f85c3da0f7f6e8212aTom Stellard bool for_drop) { 68a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard for_drop_ = for_drop; 692f921101c0826dc52a2c69f85c3da0f7f6e8212aTom Stellard 702f921101c0826dc52a2c69f85c3da0f7f6e8212aTom Stellard gfx::Rect bounds = menu_button_->bounds(); 712f921101c0826dc52a2c69f85c3da0f7f6e8212aTom Stellard gfx::Point screen_loc; 722f921101c0826dc52a2c69f85c3da0f7f6e8212aTom Stellard views::View::ConvertPointToScreen(menu_button_, &screen_loc); 732f921101c0826dc52a2c69f85c3da0f7f6e8212aTom Stellard bounds.set_x(screen_loc.x()); 742f921101c0826dc52a2c69f85c3da0f7f6e8212aTom Stellard bounds.set_y(screen_loc.y()); 75a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard 762f921101c0826dc52a2c69f85c3da0f7f6e8212aTom Stellard views::MenuItemView::AnchorPosition anchor = views::MenuItemView::TOPRIGHT; 772f921101c0826dc52a2c69f85c3da0f7f6e8212aTom Stellard // As we maintain our own lifetime we can safely ignore the result. 78a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard ignore_result(menu_runner_->RunMenuAt(window, menu_button_, bounds, anchor, 792f921101c0826dc52a2c69f85c3da0f7f6e8212aTom Stellard ui::MENU_SOURCE_NONE, for_drop_ ? views::MenuRunner::FOR_DROP : 0)); 802f921101c0826dc52a2c69f85c3da0f7f6e8212aTom Stellard if (!for_drop_) { 81a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard // Give the context menu (if any) a chance to execute the user-selected 822f921101c0826dc52a2c69f85c3da0f7f6e8212aTom Stellard // command. 83a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard base::MessageLoop::current()->DeleteSoon(FROM_HERE, this); 842f921101c0826dc52a2c69f85c3da0f7f6e8212aTom Stellard } 852f921101c0826dc52a2c69f85c3da0f7f6e8212aTom Stellard return true; 86a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard} 872f921101c0826dc52a2c69f85c3da0f7f6e8212aTom Stellard 88a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellardvoid BrowserActionOverflowMenuController::CancelMenu() { 892f921101c0826dc52a2c69f85c3da0f7f6e8212aTom Stellard menu_->Cancel(); 902f921101c0826dc52a2c69f85c3da0f7f6e8212aTom Stellard} 91a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard 922f921101c0826dc52a2c69f85c3da0f7f6e8212aTom Stellardbool BrowserActionOverflowMenuController::IsCommandEnabled(int id) const { 932f921101c0826dc52a2c69f85c3da0f7f6e8212aTom Stellard BrowserActionView* view = (*views_)[start_index_ + id - 1]; 94a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard return view->button()->IsEnabled(owner_->GetCurrentTabId()); 95a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard} 962f921101c0826dc52a2c69f85c3da0f7f6e8212aTom Stellard 972f921101c0826dc52a2c69f85c3da0f7f6e8212aTom Stellardvoid BrowserActionOverflowMenuController::ExecuteCommand(int id) { 98a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard BrowserActionView* view = (*views_)[start_index_ + id - 1]; 99a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard owner_->OnBrowserActionExecuted(view->button()); 100a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard} 1012f921101c0826dc52a2c69f85c3da0f7f6e8212aTom Stellard 102a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellardbool BrowserActionOverflowMenuController::ShowContextMenu( 103a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard views::MenuItemView* source, 104a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard int id, 105a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard const gfx::Point& p, 1062f921101c0826dc52a2c69f85c3da0f7f6e8212aTom Stellard ui::MenuSourceType source_type) { 107a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard const extensions::Extension* extension = 108a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard (*views_)[start_index_ + id - 1]->button()->extension(); 1092f921101c0826dc52a2c69f85c3da0f7f6e8212aTom Stellard if (!extension->ShowConfigureContextMenus()) 110a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard return false; 111a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard 112a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard scoped_refptr<ExtensionContextMenuModel> context_menu_contents = 1132f921101c0826dc52a2c69f85c3da0f7f6e8212aTom Stellard new ExtensionContextMenuModel(extension, browser_, owner_); 114a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard views::MenuRunner context_menu_runner(context_menu_contents.get()); 1152f921101c0826dc52a2c69f85c3da0f7f6e8212aTom Stellard 116a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard // We can ignore the result as we delete ourself. 117a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard // This blocks until the user choses something or dismisses the menu. 1182f921101c0826dc52a2c69f85c3da0f7f6e8212aTom Stellard ignore_result(context_menu_runner.RunMenuAt(menu_button_->GetWidget(), 119a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard NULL, gfx::Rect(p, gfx::Size()), views::MenuItemView::TOPLEFT, 1202f921101c0826dc52a2c69f85c3da0f7f6e8212aTom Stellard source_type, 121a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard views::MenuRunner::HAS_MNEMONICS | views::MenuRunner::IS_NESTED | 122a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard views::MenuRunner::CONTEXT_MENU)); 1232f921101c0826dc52a2c69f85c3da0f7f6e8212aTom Stellard 124a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard // The user is done with the context menu, so we can close the underlying 1252f921101c0826dc52a2c69f85c3da0f7f6e8212aTom Stellard // menu. 126a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard menu_->Cancel(); 127a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard 128a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard return true; 1292f921101c0826dc52a2c69f85c3da0f7f6e8212aTom Stellard} 130a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard 131a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellardvoid BrowserActionOverflowMenuController::DropMenuClosed( 132a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard views::MenuItemView* menu) { 133a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard delete this; 134a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard} 135a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard 136a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellardbool BrowserActionOverflowMenuController::GetDropFormats( 137a75c6163e605f35b14f26930dd9227e4f337ec9eTom Stellard views::MenuItemView* menu, 138 int* formats, 139 std::set<OSExchangeData::CustomFormat>* custom_formats) { 140 custom_formats->insert(BrowserActionDragData::GetBrowserActionCustomFormat()); 141 return true; 142} 143 144bool BrowserActionOverflowMenuController::AreDropTypesRequired( 145 views::MenuItemView* menu) { 146 return true; 147} 148 149bool BrowserActionOverflowMenuController::CanDrop( 150 views::MenuItemView* menu, const OSExchangeData& data) { 151 BrowserActionDragData drop_data; 152 if (!drop_data.Read(data)) 153 return false; 154 return drop_data.IsFromProfile(owner_->profile()); 155} 156 157int BrowserActionOverflowMenuController::GetDropOperation( 158 views::MenuItemView* item, 159 const ui::DropTargetEvent& event, 160 DropPosition* position) { 161 // Don't allow dropping from the BrowserActionContainer into slot 0 of the 162 // overflow menu since once the move has taken place the item you are dragging 163 // falls right out of the menu again once the user releases the button 164 // (because we don't shrink the BrowserActionContainer when you do this). 165 if ((item->GetCommand() == 0) && (*position == DROP_BEFORE)) { 166 BrowserActionDragData drop_data; 167 if (!drop_data.Read(event.data())) 168 return ui::DragDropTypes::DRAG_NONE; 169 170 if (drop_data.index() < owner_->VisibleBrowserActions()) 171 return ui::DragDropTypes::DRAG_NONE; 172 } 173 174 return ui::DragDropTypes::DRAG_MOVE; 175} 176 177int BrowserActionOverflowMenuController::OnPerformDrop( 178 views::MenuItemView* menu, 179 DropPosition position, 180 const ui::DropTargetEvent& event) { 181 BrowserActionDragData drop_data; 182 if (!drop_data.Read(event.data())) 183 return ui::DragDropTypes::DRAG_NONE; 184 185 size_t drop_index; 186 ViewForId(menu->GetCommand(), &drop_index); 187 188 // When not dragging within the overflow menu (dragging an icon into the menu) 189 // subtract one to get the right index. 190 if (position == DROP_BEFORE && 191 drop_data.index() < owner_->VisibleBrowserActions()) 192 --drop_index; 193 194 owner_->MoveBrowserAction(drop_data.id(), drop_index); 195 196 if (for_drop_) 197 delete this; 198 return ui::DragDropTypes::DRAG_MOVE; 199} 200 201bool BrowserActionOverflowMenuController::CanDrag(views::MenuItemView* menu) { 202 return true; 203} 204 205void BrowserActionOverflowMenuController::WriteDragData( 206 views::MenuItemView* sender, OSExchangeData* data) { 207 size_t drag_index; 208 BrowserActionView* view = ViewForId(sender->GetCommand(), &drag_index); 209 std::string id = view->button()->extension()->id(); 210 211 BrowserActionDragData drag_data(id, drag_index); 212 drag_data.Write(owner_->profile(), data); 213} 214 215int BrowserActionOverflowMenuController::GetDragOperations( 216 views::MenuItemView* sender) { 217 return ui::DragDropTypes::DRAG_MOVE; 218} 219 220BrowserActionView* BrowserActionOverflowMenuController::ViewForId( 221 int id, size_t* index) { 222 // The index of the view being dragged (GetCommand gives a 1-based index into 223 // the overflow menu). 224 size_t view_index = owner_->VisibleBrowserActions() + id - 1; 225 if (index) 226 *index = view_index; 227 return owner_->GetBrowserActionViewAt(view_index); 228} 229