tray_user.cc revision 03b57e008b61dfcb1fbad3aea950ae0e001748b0
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 "ash/system/user/tray_user.h"
6
7#include "ash/ash_switches.h"
8#include "ash/root_window_controller.h"
9#include "ash/session/session_state_delegate.h"
10#include "ash/shelf/shelf_layout_manager.h"
11#include "ash/shell_delegate.h"
12#include "ash/system/tray/system_tray.h"
13#include "ash/system/tray/system_tray_delegate.h"
14#include "ash/system/tray/system_tray_notifier.h"
15#include "ash/system/tray/tray_constants.h"
16#include "ash/system/tray/tray_item_view.h"
17#include "ash/system/tray/tray_utils.h"
18#include "ash/system/user/accounts_detailed_view.h"
19#include "ash/system/user/rounded_image_view.h"
20#include "ash/system/user/user_view.h"
21#include "base/logging.h"
22#include "base/strings/string16.h"
23#include "components/user_manager/user_info.h"
24#include "grit/ash_strings.h"
25#include "ui/aura/window.h"
26#include "ui/base/l10n/l10n_util.h"
27#include "ui/gfx/image/image.h"
28#include "ui/views/border.h"
29#include "ui/views/controls/label.h"
30#include "ui/views/layout/box_layout.h"
31#include "ui/views/view.h"
32#include "ui/views/widget/widget.h"
33
34namespace {
35
36const int kUserLabelToIconPadding = 5;
37
38}  // namespace
39
40namespace ash {
41
42TrayUser::TrayUser(SystemTray* system_tray, MultiProfileIndex index)
43    : SystemTrayItem(system_tray),
44      multiprofile_index_(index),
45      user_(NULL),
46      layout_view_(NULL),
47      avatar_(NULL),
48      label_(NULL) {
49  Shell::GetInstance()->system_tray_notifier()->AddUserObserver(this);
50}
51
52TrayUser::~TrayUser() {
53  Shell::GetInstance()->system_tray_notifier()->RemoveUserObserver(this);
54}
55
56TrayUser::TestState TrayUser::GetStateForTest() const {
57  if (!user_)
58    return HIDDEN;
59  return user_->GetStateForTest();
60}
61
62gfx::Size TrayUser::GetLayoutSizeForTest() const {
63  if (!layout_view_) {
64    return gfx::Size(0, 0);
65  } else {
66    return layout_view_->size();
67  }
68}
69
70gfx::Rect TrayUser::GetUserPanelBoundsInScreenForTest() const {
71  DCHECK(user_);
72  return user_->GetBoundsInScreenOfUserButtonForTest();
73}
74
75void TrayUser::UpdateAfterLoginStatusChangeForTest(user::LoginStatus status) {
76  UpdateAfterLoginStatusChange(status);
77}
78
79views::View* TrayUser::CreateTrayView(user::LoginStatus status) {
80  CHECK(layout_view_ == NULL);
81
82  layout_view_ = new views::View;
83  layout_view_->SetLayoutManager(
84      new views::BoxLayout(views::BoxLayout::kHorizontal,
85                           0, 0, kUserLabelToIconPadding));
86  UpdateAfterLoginStatusChange(status);
87  return layout_view_;
88}
89
90views::View* TrayUser::CreateDefaultView(user::LoginStatus status) {
91  if (status == user::LOGGED_IN_NONE)
92    return NULL;
93  const SessionStateDelegate* session_state_delegate =
94      Shell::GetInstance()->session_state_delegate();
95
96  // If the screen is locked or a system modal dialog box is shown, show only
97  // the currently active user.
98  if (multiprofile_index_ &&
99      (session_state_delegate->IsUserSessionBlocked() ||
100       Shell::GetInstance()->IsSystemModalWindowOpen()))
101    return NULL;
102
103  CHECK(user_ == NULL);
104
105  int logged_in_users = session_state_delegate->NumberOfLoggedInUsers();
106
107  // Do not show more UserView's then there are logged in users.
108  if (multiprofile_index_ >= logged_in_users)
109    return NULL;
110
111  user_ = new tray::UserView(this, status, multiprofile_index_, false);
112  return user_;
113}
114
115views::View* TrayUser::CreateDetailedView(user::LoginStatus status) {
116  return new tray::AccountsDetailedView(this, status);
117}
118
119void TrayUser::DestroyTrayView() {
120  layout_view_ = NULL;
121  avatar_ = NULL;
122  label_ = NULL;
123}
124
125void TrayUser::DestroyDefaultView() {
126  user_ = NULL;
127}
128
129void TrayUser::DestroyDetailedView() {
130}
131
132void TrayUser::UpdateAfterLoginStatusChange(user::LoginStatus status) {
133  // Only the active user is represented in the tray.
134  if (!layout_view_)
135    return;
136  if (GetTrayIndex() > 0)
137    return;
138  bool need_label = false;
139  bool need_avatar = false;
140  switch (status) {
141    case user::LOGGED_IN_LOCKED:
142    case user::LOGGED_IN_USER:
143    case user::LOGGED_IN_OWNER:
144    case user::LOGGED_IN_PUBLIC:
145      need_avatar = true;
146      break;
147    case user::LOGGED_IN_SUPERVISED:
148      need_avatar = true;
149      need_label = true;
150      break;
151    case user::LOGGED_IN_GUEST:
152      need_label = true;
153      break;
154    case user::LOGGED_IN_RETAIL_MODE:
155    case user::LOGGED_IN_KIOSK_APP:
156    case user::LOGGED_IN_NONE:
157      break;
158  }
159
160  if ((need_avatar != (avatar_ != NULL)) ||
161      (need_label != (label_ != NULL))) {
162    layout_view_->RemoveAllChildViews(true);
163    if (need_label) {
164      label_ = new views::Label;
165      SetupLabelForTray(label_);
166      layout_view_->AddChildView(label_);
167    } else {
168      label_ = NULL;
169    }
170    if (need_avatar) {
171      avatar_ = new tray::RoundedImageView(kTrayAvatarCornerRadius, true);
172      layout_view_->AddChildView(avatar_);
173    } else {
174      avatar_ = NULL;
175    }
176  }
177
178  if (status == user::LOGGED_IN_SUPERVISED) {
179    label_->SetText(
180        l10n_util::GetStringUTF16(IDS_ASH_STATUS_TRAY_SUPERVISED_LABEL));
181  } else if (status == user::LOGGED_IN_GUEST) {
182    label_->SetText(l10n_util::GetStringUTF16(IDS_ASH_STATUS_TRAY_GUEST_LABEL));
183  }
184
185  if (avatar_) {
186    avatar_->SetCornerRadii(
187        0, kTrayAvatarCornerRadius, kTrayAvatarCornerRadius, 0);
188    avatar_->SetBorder(views::Border::NullBorder());
189  }
190  UpdateAvatarImage(status);
191
192  // Update layout after setting label_ and avatar_ with new login status.
193  UpdateLayoutOfItem();
194}
195
196void TrayUser::UpdateAfterShelfAlignmentChange(ShelfAlignment alignment) {
197  // Inactive users won't have a layout.
198  if (!layout_view_)
199    return;
200  if (alignment == SHELF_ALIGNMENT_BOTTOM ||
201      alignment == SHELF_ALIGNMENT_TOP) {
202    if (avatar_) {
203      avatar_->SetBorder(views::Border::NullBorder());
204      avatar_->SetCornerRadii(
205          0, kTrayAvatarCornerRadius, kTrayAvatarCornerRadius, 0);
206    }
207    if (label_) {
208      // If label_ hasn't figured out its size yet, do that first.
209      if (label_->GetContentsBounds().height() == 0)
210        label_->SizeToPreferredSize();
211      int height = label_->GetContentsBounds().height();
212      int vertical_pad = (kTrayItemSize - height) / 2;
213      int remainder = height % 2;
214      label_->SetBorder(views::Border::CreateEmptyBorder(
215          vertical_pad + remainder,
216          kTrayLabelItemHorizontalPaddingBottomAlignment,
217          vertical_pad,
218          kTrayLabelItemHorizontalPaddingBottomAlignment));
219    }
220    layout_view_->SetLayoutManager(
221        new views::BoxLayout(views::BoxLayout::kHorizontal,
222                             0, 0, kUserLabelToIconPadding));
223  } else {
224    if (avatar_) {
225      avatar_->SetBorder(views::Border::NullBorder());
226      avatar_->SetCornerRadii(
227          0, 0, kTrayAvatarCornerRadius, kTrayAvatarCornerRadius);
228    }
229    if (label_) {
230      label_->SetBorder(views::Border::CreateEmptyBorder(
231          kTrayLabelItemVerticalPaddingVerticalAlignment,
232          kTrayLabelItemHorizontalPaddingBottomAlignment,
233          kTrayLabelItemVerticalPaddingVerticalAlignment,
234          kTrayLabelItemHorizontalPaddingBottomAlignment));
235    }
236    layout_view_->SetLayoutManager(
237        new views::BoxLayout(views::BoxLayout::kVertical,
238                             0, 0, kUserLabelToIconPadding));
239  }
240}
241
242void TrayUser::OnUserUpdate() {
243  UpdateAvatarImage(Shell::GetInstance()->system_tray_delegate()->
244      GetUserLoginStatus());
245}
246
247void TrayUser::OnUserAddedToSession() {
248  SessionStateDelegate* session_state_delegate =
249      Shell::GetInstance()->session_state_delegate();
250  // Only create views for user items which are logged in.
251  if (GetTrayIndex() >= session_state_delegate->NumberOfLoggedInUsers())
252    return;
253
254  // Enforce a layout change that newly added items become visible.
255  UpdateLayoutOfItem();
256
257  // Update the user item.
258  UpdateAvatarImage(
259      Shell::GetInstance()->system_tray_delegate()->GetUserLoginStatus());
260}
261
262void TrayUser::UpdateAvatarImage(user::LoginStatus status) {
263  SessionStateDelegate* session_state_delegate =
264      Shell::GetInstance()->session_state_delegate();
265  if (!avatar_ ||
266      GetTrayIndex() >= session_state_delegate->NumberOfLoggedInUsers())
267    return;
268
269  content::BrowserContext* context = session_state_delegate->
270      GetBrowserContextByIndex(GetTrayIndex());
271  avatar_->SetImage(session_state_delegate->GetUserInfo(context)->GetImage(),
272                    gfx::Size(kTrayAvatarSize, kTrayAvatarSize));
273
274  // Unit tests might come here with no images for some users.
275  if (avatar_->size().IsEmpty())
276    avatar_->SetSize(gfx::Size(kTrayAvatarSize, kTrayAvatarSize));
277}
278
279MultiProfileIndex TrayUser::GetTrayIndex() {
280  Shell* shell = Shell::GetInstance();
281  // If multi profile is not enabled we can use the normal index.
282  if (!shell->delegate()->IsMultiProfilesEnabled())
283    return multiprofile_index_;
284  // In case of multi profile we need to mirror the indices since the system
285  // tray items are in the reverse order then the menu items.
286  return shell->session_state_delegate()->GetMaximumNumberOfLoggedInUsers() -
287             1 - multiprofile_index_;
288}
289
290void TrayUser::UpdateLayoutOfItem() {
291  RootWindowController* controller = GetRootWindowController(
292      system_tray()->GetWidget()->GetNativeWindow()->GetRootWindow());
293  if (controller && controller->shelf()) {
294    UpdateAfterShelfAlignmentChange(
295        controller->GetShelfLayoutManager()->GetAlignment());
296  }
297}
298
299}  // namespace ash
300