Refactor the avatar button/icon class

chrome/browser/ui/views/profiles/avatar_menu_button.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/profiles/avatar_menu_button.h"
 
 #include "base/command_line.h"
 #include "base/prefs/pref_service.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/chrome_notification_types.h"
@@ skipping 17 matching lines @@
 #include "ui/views/widget/widget.h"
 
 static inline int Round(double x) {
   return static_cast<int>(x + 0.5);
 }
 
 // static
 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),
       button_on_right_(false) {
   // In RTL mode, the avatar icon should be looking the opposite direction.
   EnableCanvasFlippingForRTLUI(true);
 
   SetEventTargeter(
       scoped_ptr<views::ViewTargeter>(new views::ViewTargeter(this)));
+
+  Update();
 }
 
 AvatarMenuButton::~AvatarMenuButton() {
 }
 
 const char* AvatarMenuButton::GetClassName() const {
   return kViewClassName;
 }
 
 void AvatarMenuButton::OnPaint(gfx::Canvas* canvas) {
@@ skipping 78 matching lines @@
 #endif
     } else {
       AvatarMenu::GetImageForMenuButton(profile->GetPath(),
                                         avatar,
                                         is_rectangle);
     }
   }
   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 {
   CHECK_EQ(target, this);
   return !disabled_ &&
          views::ViewTargeterDelegate::DoesIntersectRect(target, rect);
 }
 
 // views::MenuButtonListener implementation
 void AvatarMenuButton::OnMenuButtonClicked(views::View* source,
                                            const gfx::Point& point) {
   if (!disabled_)
-    chrome::ShowAvatarMenu(browser_);
+    chrome::ShowAvatarMenu(browser());
 }