Refactor the avatar button/icon class.

Such that they are based on the same base class.

BUG=452524

Review URL: https://codereview.chromium.org/1009403002

Cr-Commit-Position: refs/heads/master@{#329249}

chrome/browser/ui/views/frame/browser_non_client_frame_view.cc

@@ -215,26 +215,6 @@ void BrowserNonClientFrameView::UpdateOldAvatarButton() {
     avatar_button_ = nullptr;
     frame_->GetRootView()->Layout();
   }
-
-  gfx::Image avatar;
-  gfx::Image taskbar_badge_avatar;
-  bool is_rectangle = false;
-
-  // Update the avatar button in the window frame and the taskbar overlay.
-  bool should_show_avatar_menu =
-      avatar_button_ || AvatarMenu::ShouldShowAvatarMenu();
-
-  if (!AvatarMenuButton::GetAvatarImages(
-          browser_view_->browser()->profile(), should_show_avatar_menu, &avatar,
-          &taskbar_badge_avatar, &is_rectangle)) {
-    return;
-  }
-
-  // Disable the menu when we should not show the menu.
-  if (avatar_button_ && !AvatarMenu::ShouldShowAvatarMenu())
-    avatar_button_->SetEnabled(false);
-  if (avatar_button_)
-    avatar_button_->SetAvatarIcon(avatar, is_rectangle);
 }
 
 void BrowserNonClientFrameView::UpdateNewAvatarButton(

chrome/browser/ui/views/profiles/avatar_base_button.cc

+// Copyright 2015 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/profiles/avatar_base_button.h"
+
+#include "base/files/file_path.h"
+#include "base/strings/string16.h"
+#include "chrome/browser/browser_process.h"
+#include "chrome/browser/profiles/profile_manager.h"
+#include "chrome/browser/ui/browser.h"
+#include "components/signin/core/common/profile_management_switches.h"
+
+AvatarBaseButton::AvatarBaseButton(Browser* browser)
+    : browser_(browser) {
+  // |profile_manager| might be null in tests.
+  ProfileManager* profile_manager = g_browser_process->profile_manager();
+  if (profile_manager)
+    profile_manager->GetProfileInfoCache().AddObserver(this);
+}
+
+AvatarBaseButton::~AvatarBaseButton() {
+  // |profile_manager| might be null in tests.
+  ProfileManager* profile_manager = g_browser_process->profile_manager();
+  if (profile_manager)
+    profile_manager->GetProfileInfoCache().RemoveObserver(this);
+}
+
+void AvatarBaseButton::OnProfileAdded(const base::FilePath& profile_path) {
+  Update();
+}
+
+void AvatarBaseButton::OnProfileWasRemoved(
+      const base::FilePath& profile_path,
+      const base::string16& profile_name) {
+  // If deleting the active profile, don't bother updating the avatar
+  // button, as the browser window is being closed anyway.
+  if (browser_->profile()->GetPath() != profile_path)
+    Update();
+}
+
+void AvatarBaseButton::OnProfileNameChanged(
+      const base::FilePath& profile_path,
+      const base::string16& old_profile_name) {
+  if (browser_->profile()->GetPath() == profile_path)
+    Update();
+}
+
+void AvatarBaseButton::OnProfileAvatarChanged(
+      const base::FilePath& profile_path) {
+  if (!switches::IsNewAvatarMenu() &&
+      browser_->profile()->GetPath() == profile_path)
+    Update();
+}
+
+void AvatarBaseButton::OnProfileSupervisedUserIdChanged(
+      const base::FilePath& profile_path) {
+  if (browser_->profile()->GetPath() == profile_path)
+    Update();
+}

chrome/browser/ui/views/profiles/avatar_base_button.h

+// Copyright 2015 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.
+
+#ifndef CHROME_BROWSER_UI_VIEWS_PROFILES_AVATAR_BASE_BUTTON_H_
+#define CHROME_BROWSER_UI_VIEWS_PROFILES_AVATAR_BASE_BUTTON_H_
+
+#include "chrome/browser/profiles/profile_info_cache_observer.h"
+
+class Browser;
+
+// This view manages the button that sits in the top of the window frame and
+// displays the active profile's info when using multi-profiles.
+class AvatarBaseButton : public ProfileInfoCacheObserver {
+ public:
+  explicit AvatarBaseButton(Browser* browser);
+  ~AvatarBaseButton() override;
+
+ protected:
+  Browser* browser() const { return browser_; }
+
+  // Called when the profile info cache has changed, which means we might
+  // have to update the icon/text of the button.
+  virtual void Update() = 0;
+
+ private:
+  // ProfileInfoCacheObserver:
+  void OnProfileAdded(const base::FilePath& profile_path) override;
+  void OnProfileWasRemoved(const base::FilePath& profile_path,
+                           const base::string16& profile_name) override;
+  void OnProfileNameChanged(const base::FilePath& profile_path,
+                            const base::string16& old_profile_name) override;
+  void OnProfileAvatarChanged(const base::FilePath& profile_path) override;
+  void OnProfileSupervisedUserIdChanged(
+      const base::FilePath& profile_path) override;
+
+  Browser* browser_;
+
+  DISALLOW_COPY_AND_ASSIGN(AvatarBaseButton);
+};
+
+#endif  // CHROME_BROWSER_UI_VIEWS_PROFILES_AVATAR_BASE_BUTTON_H_

chrome/browser/ui/views/profiles/avatar_menu_button.cc

@@ -35,8 +35,8 @@ static inline int Round(double x) {
 const char AvatarMenuButton::kViewClassName[] = "AvatarMenuButton";
 
 AvatarMenuButton::AvatarMenuButton(Browser* browser, bool disabled)
-    : MenuButton(NULL, base::string16(), this, false),
-      browser_(browser),
+    : AvatarBaseButton(browser),
+      MenuButton(NULL, base::string16(), this, false),
       disabled_(disabled),
       is_rectangle_(false),
       old_height_(0),
@@ -46,6 +46,8 @@ AvatarMenuButton::AvatarMenuButton(Browser* browser, bool disabled)
 
   SetEventTargeter(
       scoped_ptr<views::ViewTargeter>(new views::ViewTargeter(this)));
+
+  Update();
 }
 
 AvatarMenuButton::~AvatarMenuButton() {
@@ -144,6 +146,24 @@ bool AvatarMenuButton::GetAvatarImages(Profile* profile,
   return true;
 }
 
+void AvatarMenuButton::Update() {
+  // The browser can be null in tests.
+  if (!browser())
+    return;
+
+  const bool should_show_avatar_menu = AvatarMenu::ShouldShowAvatarMenu();
+  SetEnabled(should_show_avatar_menu);
+
+  gfx::Image avatar;
+  gfx::Image taskbar_badge_avatar;
+  bool is_rectangle = false;
+  if (AvatarMenuButton::GetAvatarImages(
+          browser()->profile(), should_show_avatar_menu, &avatar,
+          &taskbar_badge_avatar, &is_rectangle)) {
+    SetAvatarIcon(avatar, is_rectangle);
+  }
+}
+
 // views::ViewTargeterDelegate:
 bool AvatarMenuButton::DoesIntersectRect(const views::View* target,
                                          const gfx::Rect& rect) const {
@@ -156,5 +176,5 @@ bool AvatarMenuButton::DoesIntersectRect(const views::View* target,
 void AvatarMenuButton::OnMenuButtonClicked(views::View* source,
                                            const gfx::Point& point) {
   if (!disabled_)
-    chrome::ShowAvatarMenu(browser_);
+    chrome::ShowAvatarMenu(browser());
 }

chrome/browser/ui/views/profiles/avatar_menu_button.h

@@ -8,6 +8,7 @@
 #include <string>
 
 #include "base/compiler_specific.h"
+#include "chrome/browser/ui/views/profiles/avatar_base_button.h"
 #include "ui/base/models/simple_menu_model.h"
 #include "ui/views/controls/button/menu_button.h"
 #include "ui/views/controls/button/menu_button_listener.h"
@@ -25,7 +26,8 @@ class Profile;
 // A button used to show either the incognito avatar or the profile avatar.
 // The button can optionally have a menu attached to it.
 
-class AvatarMenuButton : public views::MenuButton,
+class AvatarMenuButton : public AvatarBaseButton,
+                         public views::MenuButton,
                          public views::MenuButtonListener,
                          public views::ViewTargeterDelegate {
  public:
@@ -61,6 +63,9 @@ class AvatarMenuButton : public views::MenuButton,
                               gfx::Image* avatar,
                               gfx::Image* taskbar_badge_avatar,
                               bool* is_rectangle);
+ protected:
+  // AvatarBaseButton:
+  void Update() override;
 
  private:
   // views::ViewTargeterDelegate:
@@ -71,7 +76,6 @@ class AvatarMenuButton : public views::MenuButton,
   void OnMenuButtonClicked(views::View* source,
                            const gfx::Point& point) override;
 
-  Browser* browser_;
   bool disabled_;
   scoped_ptr<ui::MenuModel> menu_model_;
 

chrome/browser/ui/views/profiles/new_avatar_button.cc

@@ -46,8 +46,8 @@ scoped_ptr<views::Border> CreateBorder(const int normal_image_set[],
 NewAvatarButton::NewAvatarButton(views::ButtonListener* listener,
                                  AvatarButtonStyle button_style,
                                  Browser* browser)
-    : LabelButton(listener, base::string16()),
-      browser_(browser),
+    : AvatarBaseButton(browser),
+      LabelButton(listener, base::string16()),
       has_auth_error_(false),
       suppress_mouse_released_action_(false) {
   set_triggerable_event_flags(
@@ -95,12 +95,10 @@ NewAvatarButton::NewAvatarButton(views::ButtonListener* listener,
         *rb->GetImageNamed(IDR_AVATAR_GLASS_BUTTON_AVATAR).ToImageSkia();
   }
 
-  g_browser_process->profile_manager()->GetProfileInfoCache().AddObserver(this);
-
   // Subscribe to authentication error changes so that the avatar button can
   // update itself.  Note that guest mode profiles won't have a token service.
   SigninErrorController* error =
-      profiles::GetSigninErrorController(browser_->profile());
+      profiles::GetSigninErrorController(browser->profile());
   if (error) {
     error->AddObserver(this);
     OnErrorChanged();  // This calls Update().
@@ -111,10 +109,8 @@ NewAvatarButton::NewAvatarButton(views::ButtonListener* listener,
 }
 
 NewAvatarButton::~NewAvatarButton() {
-  g_browser_process->profile_manager()->
-      GetProfileInfoCache().RemoveObserver(this);
   SigninErrorController* error =
-      profiles::GetSigninErrorController(browser_->profile());
+      profiles::GetSigninErrorController(browser()->profile());
   if (error)
     error->RemoveObserver(this);
 }
@@ -132,41 +128,6 @@ void NewAvatarButton::OnMouseReleased(const ui::MouseEvent& event) {
     LabelButton::OnMouseReleased(event);
 }
 
-void NewAvatarButton::OnProfileAdded(const base::FilePath& profile_path) {
-  Update();
-}
-
-void NewAvatarButton::OnProfileWasRemoved(
-      const base::FilePath& profile_path,
-      const base::string16& profile_name) {
-  // If deleting the active profile, don't bother updating the avatar
-  // button, as the browser window is being closed anyway.
-  if (browser_->profile()->GetPath() != profile_path)
-    Update();
-}
-
-void NewAvatarButton::OnProfileNameChanged(
-      const base::FilePath& profile_path,
-      const base::string16& old_profile_name) {
-  if (browser_->profile()->GetPath() == profile_path)
-    Update();
-}
-
-void NewAvatarButton::OnProfileSupervisedUserIdChanged(
-      const base::FilePath& profile_path) {
-  if (browser_->profile()->GetPath() == profile_path)
-    Update();
-}
-
-void NewAvatarButton::OnErrorChanged() {
-  // If there is an error, show an warning icon.
-  const SigninErrorController* error =
-      profiles::GetSigninErrorController(browser_->profile());
-  has_auth_error_ = error && error->HasError();
-
-  Update();
-}
-
 void NewAvatarButton::Update() {
   const ProfileInfoCache& cache =
       g_browser_process->profile_manager()->GetProfileInfoCache();
@@ -174,12 +135,12 @@ void NewAvatarButton::Update() {
   // If we have a single local profile, then use the generic avatar
   // button instead of the profile name. Never use the generic button if
   // the active profile is Guest.
-  bool use_generic_button = (!browser_->profile()->IsGuestSession() &&
+  bool use_generic_button = (!browser()->profile()->IsGuestSession() &&
                              cache.GetNumberOfProfiles() == 1 &&
                              !cache.ProfileIsAuthenticatedAtIndex(0));
 
   SetText(use_generic_button ? base::string16() :
-      profiles::GetAvatarButtonTextForProfile(browser_->profile()));
+      profiles::GetAvatarButtonTextForProfile(browser()->profile()));
 
   // If the button has no text, clear the text shadows to make sure the
   // image is centered correctly.
@@ -209,3 +170,12 @@ void NewAvatarButton::Update() {
 
   PreferredSizeChanged();
 }
+
+void NewAvatarButton::OnErrorChanged() {
+  // If there is an error, show a warning.
+  const SigninErrorController* error =
+      profiles::GetSigninErrorController(browser()->profile());
+  has_auth_error_ = error && error->HasError();
+
+  Update();
+}

chrome/browser/ui/views/profiles/new_avatar_button.h

@@ -5,15 +5,15 @@
 #ifndef CHROME_BROWSER_UI_VIEWS_PROFILES_NEW_AVATAR_BUTTON_H_
 #define CHROME_BROWSER_UI_VIEWS_PROFILES_NEW_AVATAR_BUTTON_H_
 
-#include "chrome/browser/profiles/profile_info_cache_observer.h"
+#include "chrome/browser/ui/views/profiles/avatar_base_button.h"
 #include "components/signin/core/browser/signin_error_controller.h"
 #include "ui/views/controls/button/label_button.h"
 
 class Browser;
 
 // Avatar button that displays the active profile's name in the caption area.
-class NewAvatarButton : public views::LabelButton,
-                        public ProfileInfoCacheObserver,
+class NewAvatarButton : public AvatarBaseButton,
+                        public views::LabelButton,
                         public SigninErrorController::Observer {
  public:
   // Different button styles that can be applied.
@@ -31,27 +31,16 @@ class NewAvatarButton : public views::LabelButton,
   bool OnMousePressed(const ui::MouseEvent& event) override;
   void OnMouseReleased(const ui::MouseEvent& event) override;
 
+ protected:
+  // AvatarBaseButton:
+  void Update() override;
+
  private:
   friend class ProfileChooserViewExtensionsTest;
 
-  // ProfileInfoCacheObserver:
-  void OnProfileAdded(const base::FilePath& profile_path) override;
-  void OnProfileWasRemoved(const base::FilePath& profile_path,
-                           const base::string16& profile_name) override;
-  void OnProfileNameChanged(const base::FilePath& profile_path,
-                            const base::string16& old_profile_name) override;
-  void OnProfileSupervisedUserIdChanged(
-      const base::FilePath& profile_path) override;
-
   // SigninErrorController::Observer:
   void OnErrorChanged() override;
 
-  // Called when the profile info cache has changed, which means we might
-  // have to update the icon/text of the button.
-  void Update();
-
-  Browser* browser_;
-
   // Whether the signed in profile has an authentication error. Used to display
   // an error icon next to the button text.
   bool has_auth_error_;

chrome/chrome_browser_ui.gypi

@@ -2309,6 +2309,8 @@
       'browser/ui/views/passwords/save_account_more_combobox_model.cc',
       'browser/ui/views/passwords/save_account_more_combobox_model.h',
       'browser/ui/views/process_singleton_dialog_linux.cc',
+      'browser/ui/views/profiles/avatar_base_button.cc',
+      'browser/ui/views/profiles/avatar_base_button.h',
       'browser/ui/views/profiles/avatar_menu_bubble_view.cc',
       'browser/ui/views/profiles/avatar_menu_bubble_view.h',
       'browser/ui/views/profiles/avatar_menu_button.cc',