Keep reference view pressed while extension actions have a popup

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

 // Copyright 2013 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/browser_action_view.h"
 
 #include <string>
 
 #include "base/strings/utf_string_conversions.h"
 #include "chrome/browser/chrome_notification_types.h"
@@ skipping 112 matching lines @@
 void BrowserActionView::ButtonPressed(views::Button* sender,
                                       const ui::Event& event) {
   view_controller_->ExecuteActionByUser();
 }
 
 void BrowserActionView::UpdateState() {
   int tab_id = view_controller_->GetCurrentTabId();
   if (tab_id < 0)
     return;
 
-  if (!IsEnabled(tab_id)) {
+  if (!IsEnabled(tab_id))
     SetState(views::CustomButton::STATE_DISABLED);
-  } else {
-    SetState(menu_visible_ ?
-             views::CustomButton::STATE_PRESSED :
-             views::CustomButton::STATE_NORMAL);
-  }
 
   gfx::ImageSkia icon = *view_controller_->GetIcon(tab_id).ToImageSkia();
 
   if (!icon.isNull()) {
     if (!extension_action()->GetIsVisible(tab_id))
       icon = gfx::ImageSkiaOperations::CreateTransparentImage(icon, .25);
 
     ThemeService* theme = ThemeServiceFactory::GetForProfile(
         view_controller_->browser()->profile());
 
@@ skipping 98 matching lines @@
     LabelButton::OnGestureEvent(event);
 }
 
 scoped_ptr<LabelButtonBorder> BrowserActionView::CreateDefaultBorder() const {
   scoped_ptr<LabelButtonBorder> border = LabelButton::CreateDefaultBorder();
   border->set_insets(gfx::Insets(kBorderInset, kBorderInset,
                                  kBorderInset, kBorderInset));
   return border.Pass();
 }
 
-void BrowserActionView::SetButtonPushed() {
-  SetState(views::CustomButton::STATE_PRESSED);
-  menu_visible_ = true;
-}
-
-void BrowserActionView::SetButtonNotPushed() {
-  SetState(views::CustomButton::STATE_NORMAL);
-  menu_visible_ = false;
-}
-
 bool BrowserActionView::IsEnabled(int tab_id) const {
   return view_controller_->extension_action()->GetIsVisible(tab_id);
 }
 
 gfx::ImageSkia BrowserActionView::GetIconWithBadge() {
   int tab_id = view_controller_->GetCurrentTabId();
   gfx::Size spacing(0, ToolbarView::kVertSpacing);
   gfx::ImageSkia icon = *view_controller_->GetIcon(tab_id).ToImageSkia();
   if (!IsEnabled(tab_id))
     icon = gfx::ImageSkiaOperations::CreateTransparentImage(icon, .25);
@@ skipping 49 matching lines @@
   return delegate_->GetCurrentWebContents();
 }
 
 void BrowserActionView::HideActivePopup() {
   delegate_->HideActivePopup();
 }
 
 void BrowserActionView::OnPopupShown(bool grant_tab_permissions) {
   delegate_->SetPopupOwner(this);
   if (grant_tab_permissions)
-    SetButtonPushed();
+    SetMenuVisible(true);
 }
 
 void BrowserActionView::CleanupPopup() {
   // We need to do these actions synchronously (instead of closing and then
   // performing the rest of the cleanup in OnWidgetDestroyed()) because
   // OnWidgetDestroyed() can be called asynchronously from Close(), and we need
   // to keep the delegate's popup owner up-to-date.
-  SetButtonNotPushed();
+  SetMenuVisible(false);
   delegate_->SetPopupOwner(NULL);
 }
 
 void BrowserActionView::OnWillShowContextMenus() {
-  SetButtonPushed();
+  SetMenuVisible(true);
 }
 
 void BrowserActionView::OnContextMenuDone() {
-  SetButtonNotPushed();
+  SetMenuVisible(false);
 }
+
+void BrowserActionView::SetMenuVisible(bool menu_visible) {
+  // We set the state of the menu button we're using as a reference view, which
+  // is either this or the overflow reference view.
+  // This cast is safe because GetReferenceViewForPopup returns either |this|
+  // or delegate_->GetOverflowReferenceView(), which returns a MenuButton.
+  views::MenuButton* reference_view =
+      static_cast<views::MenuButton*>(GetReferenceViewForPopup());
+  if (menu_visible)
+    reference_view->AttachMenu();
+  else
+    reference_view->DetachMenu();
+}