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

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "chrome/browser/ui/views/frame/browser_non_client_frame_view.h"
   6
   7 #include "chrome/browser/browser_process.h"
   8 #include "chrome/browser/profiles/avatar_menu.h"
   9 #include "chrome/browser/profiles/profile.h"
  10 #include "chrome/browser/profiles/profile_avatar_icon_util.h"
  11 #include "chrome/browser/profiles/profile_info_cache.h"
  12 #include "chrome/browser/profiles/profile_manager.h"
  13 #include "chrome/browser/profiles/profiles_state.h"
  14 #include "chrome/browser/ui/view_ids.h"
  15 #include "chrome/browser/ui/views/frame/browser_view.h"
  16 #include "chrome/browser/ui/views/frame/taskbar_decorator.h"
  17 #include "chrome/browser/ui/views/profiles/avatar_label.h"
  18 #include "chrome/browser/ui/views/profiles/avatar_menu_button.h"
  19 #include "chrome/browser/ui/views/profiles/new_avatar_button.h"
  20 #include "components/signin/core/common/profile_management_switches.h"
  21 #include "grit/generated_resources.h"
  22 #include "grit/theme_resources.h"
  23 #include "third_party/skia/include/core/SkColor.h"
  24 #include "ui/base/resource/resource_bundle.h"
  25 #include "ui/base/theme_provider.h"
  26 #include "ui/gfx/image/image.h"
  27 #include "ui/views/background.h"
  28
  29 BrowserNonClientFrameView::BrowserNonClientFrameView(BrowserFrame* frame,
  30                                                      BrowserView* browser_view)
  31     : frame_(frame),
  32       browser_view_(browser_view),
  33       avatar_button_(NULL),
  34       avatar_label_(NULL),
  35       new_avatar_button_(NULL) {
  36 }
  37
  38 BrowserNonClientFrameView::~BrowserNonClientFrameView() {
  39 }
  40
  41 void BrowserNonClientFrameView::VisibilityChanged(views::View* starting_from,
  42                                                   bool is_visible) {
  43   if (!is_visible)
  44     return;
  45   // The first time UpdateAvatarInfo() is called the window is not visible so
  46   // DrawTaskBarDecoration() has no effect. Therefore we need to call it again
  47   // once the window is visible.
  48   if (!browser_view_->IsRegularOrGuestSession() ||
  49       !switches::IsNewAvatarMenu())
  50     UpdateAvatarInfo();
  51 }
  52
  53 void BrowserNonClientFrameView::OnThemeChanged() {
  54   if (avatar_label_)
  55     avatar_label_->UpdateLabelStyle();
  56 }
  57
  58 void BrowserNonClientFrameView::UpdateAvatarInfo() {
  59   if (browser_view_->ShouldShowAvatar()) {
  60     if (!avatar_button_) {
  61       Profile* profile = browser_view_->browser()->profile();
  62       if (profile->IsSupervised() && !avatar_label_) {
  63         avatar_label_ = new AvatarLabel(browser_view_);
  64         avatar_label_->set_id(VIEW_ID_AVATAR_LABEL);
  65         AddChildView(avatar_label_);
  66       }
  67       avatar_button_ = new AvatarMenuButton(
  68           browser_view_->browser(), !browser_view_->IsRegularOrGuestSession());
  69       avatar_button_->set_id(VIEW_ID_AVATAR_BUTTON);
  70       AddChildView(avatar_button_);
  71       // Invalidate here because adding a child does not invalidate the layout.
  72       InvalidateLayout();
  73       frame_->GetRootView()->Layout();
  74     }
  75   } else if (avatar_button_) {
  76     // The avatar label can just be there if there is also an avatar button.
  77     if (avatar_label_) {
  78       RemoveChildView(avatar_label_);
  79       delete avatar_label_;
  80       avatar_label_ = NULL;
  81     }
  82     RemoveChildView(avatar_button_);
  83     delete avatar_button_;
  84     avatar_button_ = NULL;
  85     frame_->GetRootView()->Layout();
  86   }
  87
  88   ui::ResourceBundle& rb = ui::ResourceBundle::GetSharedInstance();
  89   gfx::Image avatar;
  90   gfx::Image taskbar_badge_avatar;
  91   base::string16 text;
  92   bool is_rectangle = false;
  93   if (browser_view_->IsGuestSession()) {
  94     avatar = rb.
  95         GetImageNamed(profiles::GetPlaceholderAvatarIconResourceID());
  96   } else if (browser_view_->IsOffTheRecord()) {
  97     avatar = rb.GetImageNamed(IDR_OTR_ICON);
  98     // TODO(nkostylev): Allow this on ChromeOS once the ChromeOS test
  99     // environment handles profile directories correctly.
 100 #if !defined(OS_CHROMEOS)
 101     bool is_badge_rectangle = false;
 102     // The taskbar badge should be the profile avatar, not the OTR avatar.
 103     AvatarMenu::GetImageForMenuButton(browser_view_->browser()->profile(),
 104                                       &taskbar_badge_avatar,
 105                                       &is_badge_rectangle);
 106 #endif
 107   } else if (avatar_button_ || AvatarMenu::ShouldShowAvatarMenu()) {
 108     ProfileInfoCache& cache =
 109         g_browser_process->profile_manager()->GetProfileInfoCache();
 110     Profile* profile = browser_view_->browser()->profile();
 111     size_t index = cache.GetIndexOfProfileWithPath(profile->GetPath());
 112     if (index == std::string::npos)
 113       return;
 114     text = cache.GetNameOfProfileAtIndex(index);
 115
 116     AvatarMenu::GetImageForMenuButton(browser_view_->browser()->profile(),
 117                                       &avatar,
 118                                       &is_rectangle);
 119     // Disable the menu when we should not show the menu.
 120     if (avatar_button_ && !AvatarMenu::ShouldShowAvatarMenu())
 121       avatar_button_->SetEnabled(false);
 122   }
 123   if (avatar_button_)
 124     avatar_button_->SetAvatarIcon(avatar, is_rectangle);
 125
 126   // For popups and panels which don't have the avatar button, we still
 127   // need to draw the taskbar decoration. Even though we have an icon on the
 128   // window's relaunch details, we draw over it because the user may have pinned
 129   // the badge-less Chrome shortcut which will cause windows to ignore the
 130   // relaunch details.
 131   // TODO(calamity): ideally this should not be necessary but due to issues with
 132   // the default shortcut being pinned, we add the runtime badge for safety.
 133   // See crbug.com/313800.
 134   chrome::DrawTaskbarDecoration(
 135       frame_->GetNativeWindow(),
 136       AvatarMenu::ShouldShowAvatarMenu()
 137           ? (taskbar_badge_avatar.IsEmpty() ? &avatar : &taskbar_badge_avatar)
 138           : NULL);
 139 }
 140
 141 void BrowserNonClientFrameView::UpdateNewStyleAvatarInfo(
 142     views::ButtonListener* listener,
 143     const NewAvatarButton::AvatarButtonStyle style) {
 144   DCHECK(switches::IsNewAvatarMenu());
 145   // This should never be called in incognito mode.
 146   DCHECK(browser_view_->IsRegularOrGuestSession());
 147
 148   if (browser_view_->ShouldShowAvatar()) {
 149     if (!new_avatar_button_) {
 150       base::string16 profile_name = profiles::GetAvatarNameForProfile(
 151           browser_view_->browser()->profile()->GetPath());
 152       new_avatar_button_ = new NewAvatarButton(
 153           listener, profile_name, style, browser_view_->browser());
 154       new_avatar_button_->set_id(VIEW_ID_NEW_AVATAR_BUTTON);
 155       AddChildView(new_avatar_button_);
 156       frame_->GetRootView()->Layout();
 157     }
 158   } else if (new_avatar_button_) {
 159     delete new_avatar_button_;
 160     new_avatar_button_ = NULL;
 161     frame_->GetRootView()->Layout();
 162   }
 163 }