Open the WrenchMenu on mouseover when dragging a browser action

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

 // Copyright 2014 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/ui/views/toolbar/extension_toolbar_menu_view.h"
 
 #include "chrome/browser/ui/views/frame/browser_view.h"
 #include "chrome/browser/ui/views/toolbar/browser_actions_container.h"
 #include "chrome/browser/ui/views/toolbar/toolbar_view.h"
+#include "chrome/browser/ui/views/toolbar/wrench_menu.h"
 #include "ui/views/controls/menu/menu_item_view.h"
 
 namespace {
 
 // Bottom padding to make sure we have enough room for the icons.
 // TODO(devlin): Figure out why the bottom few pixels of the last row in the
 // overflow menu are cut off (so we can remove this).
 const int kVerticalPadding = 8;
 
 }  // namespace
 
-ExtensionToolbarMenuView::ExtensionToolbarMenuView(Browser* browser)
-    : browser_(browser) {
-  BrowserView* browser_view = BrowserView::GetBrowserViewForBrowser(browser);
+ExtensionToolbarMenuView::ExtensionToolbarMenuView(Browser* browser,
+                                                   WrenchMenu* wrench_menu)
+    : browser_(browser),
+      wrench_menu_(wrench_menu),
+      container_(NULL),
+      browser_actions_container_observer_(this),
+      weak_factory_(this) {
+  BrowserActionsContainer* main =
+      BrowserView::GetBrowserViewForBrowser(browser)
+          ->toolbar()->browser_actions();
   container_ = new BrowserActionsContainer(
       browser_,
       NULL,  // No owner view, means no extra keybindings are registered.
-      browser_view->GetToolbarView()->browser_actions());
+      main);
   container_->Init();
   AddChildView(container_);
+
+  // If we were opened for a drop command, we have to wait for the drop to
+  // finish so we can close the wrench menu.
+  if (wrench_menu_->for_drop()) {
+    browser_actions_container_observer_.Add(container_);
+    browser_actions_container_observer_.Add(main);
+  }
 }
 
 ExtensionToolbarMenuView::~ExtensionToolbarMenuView() {
 }
 
 gfx::Size ExtensionToolbarMenuView::GetPreferredSize() const {
   gfx::Size sz = container_->GetPreferredSize();
   if (sz.height() == 0)
     return sz;
 
   sz.Enlarge(0, kVerticalPadding);
   return sz;
 }
 
 void ExtensionToolbarMenuView::Layout() {
   // All buttons are given the same width.
   gfx::Size sz = container_->GetPreferredSize();
   int height = sz.height() + kVerticalPadding / 2;
   SetBounds(views::MenuItemView::label_start(), 0, sz.width(), height);
   container_->SetBounds(0, 0, sz.width(), height);
 }
+
+void ExtensionToolbarMenuView::OnBrowserActionDragDone() {
+  DCHECK(wrench_menu_->for_drop());
+  // We have to post this in a message loop because it results in the menu being
+  // closed, which deletes this view. Since this is in an observer method, that
+  // could break the BrowserActionsContainer's observer list if done here.

[sky, 2014/07/18 19:36:53]

observer list handles removal while notifying.

[Devlin, 2014/07/18 21:38:31]

Oh, nifty.  Coulda sworn it didn't...  Done.

+  base::MessageLoop::current()->PostTask(
+      FROM_HERE,
+      base::Bind(&ExtensionToolbarMenuView::CloseWrenchMenu,
+                 weak_factory_.GetWeakPtr()));
+}
+
+void ExtensionToolbarMenuView::CloseWrenchMenu() {
+  wrench_menu_->CloseMenu();
+}