ash uber tray: Allow customizing each item depending on whether the user is logged in or not.

ash/system/tray/system_tray.cc

 // Copyright (c) 2012 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 "ash/system/tray/system_tray.h"
 
+#include "ash/shell.h"
 #include "ash/shell/panel_window.h"
+#include "ash/system/tray/system_tray_delegate.h"
 #include "ash/system/tray/system_tray_item.h"
+#include "ash/system/user/login_status.h"
 #include "ash/wm/shadow_types.h"
 #include "base/logging.h"
 #include "base/utf_string_conversions.h"
 #include "third_party/skia/include/core/SkColor.h"
 #include "ui/views/border.h"
 #include "ui/views/bubble/bubble_delegate.h"
 #include "ui/views/controls/label.h"
 #include "ui/views/layout/box_layout.h"
 #include "ui/views/view.h"
 
@@ skipping 24 matching lines @@
         (*it)->DestroyDefaultView();
     }
   }
 
  private:
   // Overridden from views::BubbleDelegateView.
   virtual void Init() OVERRIDE {
     SetLayoutManager(new views::BoxLayout(views::BoxLayout::kVertical,
           0, 0, 1));
 
+    ash::SystemTrayDelegate* delegate =
+        ash::Shell::GetInstance()->tray_delegate();
+    ash::user::LoginStatus login_status = delegate->GetUserLoginStatus();
     for (std::vector<ash::SystemTrayItem*>::iterator it = items_.begin();
         it != items_.end();
         ++it) {
-      views::View* view = detailed_ ? (*it)->CreateDetailedView() :
-                                      (*it)->CreateDefaultView();
+      views::View* view = detailed_ ? (*it)->CreateDetailedView(login_status) :
+                                      (*it)->CreateDefaultView(login_status);
+      if (!view)
+        continue;
       if (it != items_.begin())
         view->set_border(views::Border::CreateSolidSidedBorder(
               1, 0, 0, 0, SkColorSetARGB(25, 0, 0, 0)));
       AddChildView(view);
     }
   }
 
   ash::SystemTray* tray_;
   std::vector<ash::SystemTrayItem*> items_;
   bool detailed_;
@@ skipping 20 matching lines @@
   for (std::vector<SystemTrayItem*>::iterator it = items_.begin();
       it != items_.end();
       ++it) {
     (*it)->DestroyTrayView();
   }
 }
 
 void SystemTray::AddTrayItem(SystemTrayItem* item) {
   items_.push_back(item);
 
-  views::View* tray_item = item->CreateTrayView();
+  SystemTrayDelegate* delegate = Shell::GetInstance()->tray_delegate();
+  views::View* tray_item = item->CreateTrayView(delegate->GetUserLoginStatus());
   if (tray_item) {
     AddChildViewAt(tray_item, 0);
     PreferredSizeChanged();
   }
 }
 
 void SystemTray::RemoveTrayItem(SystemTrayItem* item) {
   NOTIMPLEMENTED();
 }
 
 void SystemTray::ShowDetailedView(SystemTrayItem* item) {
   if (popup_)
     popup_->Close();
   popup_ = NULL;
 
   std::vector<SystemTrayItem*> items;
   items.push_back(item);
   ShowItems(items, true);
 }
 
+void SystemTray::UpdateAfterLoginStatusChange(user::LoginStatus login_status) {
+  if (popup_)
+    popup_->CloseNow();
+
+  for (std::vector<SystemTrayItem*>::iterator it = items_.begin();
+      it != items_.end();
+      ++it) {
+    (*it)->DestroyTrayView();
+  }
+  RemoveAllChildViews(true);
+
+  for (std::vector<SystemTrayItem*>::iterator it = items_.begin();
+      it != items_.end();
+      ++it) {
+    views::View* view = (*it)->CreateTrayView(login_status);
+    if (view)
+      AddChildViewAt(view, 0);
+  }
+  PreferredSizeChanged();
+}
+
 void SystemTray::ShowItems(std::vector<SystemTrayItem*>& items, bool detailed) {
   CHECK(!popup_);
   SystemTrayBubble* bubble = new SystemTrayBubble(this, items, detailed);
   popup_ = views::BubbleDelegateView::CreateBubble(bubble);
   popup_->AddObserver(this);
   bubble->Show();
 }
 
 bool SystemTray::OnMousePressed(const views::MouseEvent& event) {
   if (popup_)
     popup_->Show();
   else
     ShowItems(items_, false);
   return true;
 }
 
 void SystemTray::OnWidgetClosing(views::Widget* widget) {
   CHECK_EQ(popup_, widget);
   popup_ = NULL;
 }
 
 }  // namespace ash