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