Show small notification icons in notification tray

ash/common/system/web_notification/web_notification_tray.h

 // 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.
 
 #ifndef ASH_COMMON_SYSTEM_WEB_NOTIFICATION_WEB_NOTIFICATION_TRAY_H_
 #define ASH_COMMON_SYSTEM_WEB_NOTIFICATION_WEB_NOTIFICATION_TRAY_H_
 
 #include <memory>
 
 #include "ash/ash_export.h"
 #include "ash/common/login_status.h"
 #include "ash/common/system/tray/tray_background_view.h"
 #include "base/gtest_prod_util.h"
 #include "base/macros.h"
 #include "base/memory/weak_ptr.h"
 #include "ui/base/models/simple_menu_model.h"
+#include "ui/gfx/animation/animation_container.h"
 #include "ui/message_center/message_center_tray.h"
 #include "ui/message_center/message_center_tray_delegate.h"
 #include "ui/views/bubble/tray_bubble_view.h"
 #include "ui/views/controls/button/button.h"
 
 // Status area tray for showing browser and app notifications. This hosts
 // a MessageCenter class which manages the notification list. This class
 // contains the Ash specific tray implementation.
 //
 // Note: These are not related to system notifications (i.e NotificationView
@@ skipping 10 matching lines @@
 class MessageCenter;
 class MessageCenterBubble;
 class MessagePopupCollection;
 }
 
 namespace ash {
 class AshPopupAlignmentDelegate;
 class SystemTray;
 class WebNotificationBubbleWrapper;
 class WebNotificationButton;
+class WebNotificationImage;
+class WebNotificationLabel;
 class WmWindow;
 
 class ASH_EXPORT WebNotificationTray
     : public TrayBackgroundView,
       public views::TrayBubbleView::Delegate,
       public message_center::MessageCenterTrayDelegate,
-      public views::ButtonListener,
       public base::SupportsWeakPtr<WebNotificationTray>,
       public ui::SimpleMenuModel::Delegate {
  public:
   WebNotificationTray(WmShelf* shelf,
                       WmWindow* status_area_window,
                       SystemTray* system_tray);
   ~WebNotificationTray() override;
 
+  static void DisableAnimationsForTest(bool disable);
+
   // Sets the height of the system tray bubble (or legacy notification bubble)
   // from the edge of the work area so that the web notification popups don't
   // overlap with the tray. Pass 0 if no bubble is shown.
   void SetTrayBubbleHeight(int height);
 
   // Returns the current tray bubble height or 0 if there is no bubble.
   int tray_bubble_height_for_test() const;
 
   // Returns true if it should block the auto hide behavior of the shelf.
   bool ShouldBlockShelfAutoHide() const;
@@ skipping 27 matching lines @@
   base::string16 GetAccessibleNameForBubble() override;
   gfx::Rect GetAnchorRect(views::Widget* anchor_widget,
                           AnchorType anchor_type,
                           AnchorAlignment anchor_alignment) const override;
   void OnBeforeBubbleWidgetInit(
       views::Widget* anchor_widget,
       views::Widget* bubble_widget,
       views::Widget::InitParams* params) const override;
   void HideBubble(const views::TrayBubbleView* bubble_view) override;
 
-  // Overridden from ButtonListener.
-  void ButtonPressed(views::Button* sender, const ui::Event& event) override;
-
   // Overridden from MessageCenterTrayDelegate.
   void OnMessageCenterTrayChanged() override;
   bool ShowMessageCenter() override;
   void HideMessageCenter() override;
   bool ShowPopups() override;
   void HidePopups() override;
   bool ShowNotifierSettings() override;
   bool IsContextMenuEnabled() const override;
   message_center::MessageCenterTray* GetMessageCenterTray() override;
 
-  // Overridden from SimpleMenuModel::Delegate.
+  // Overridden from ui::SimpleMenuModel::Delegate.
   bool IsCommandIdChecked(int command_id) const override;
   bool IsCommandIdEnabled(int command_id) const override;
   void ExecuteCommand(int command_id, int event_flags) override;
 
   message_center::MessageCenter* message_center() const;
 
  private:
   friend class WebNotificationTrayTest;
 
   FRIEND_TEST_ALL_PREFIXES(WebNotificationTrayTest, WebNotifications);
@@ skipping 35 matching lines @@
 
   // Testing accessors.
   bool IsPopupVisible() const;
   message_center::MessageCenterBubble* GetMessageCenterBubbleForTest();
 
   WmWindow* status_area_window_;
   SystemTray* system_tray_;
   std::unique_ptr<message_center::MessageCenterTray> message_center_tray_;
   std::unique_ptr<WebNotificationBubbleWrapper> message_center_bubble_;
   std::unique_ptr<message_center::MessagePopupCollection> popup_collection_;
-  WebNotificationButton* button_;
+  std::unique_ptr<WebNotificationImage> bell_icon_;
+  std::unique_ptr<WebNotificationLabel> counter_;
+
+  scoped_refptr<gfx::AnimationContainer> animation_container_ =
+      new gfx::AnimationContainer();
+
+  std::unordered_map<std::string, WebNotificationImage*> visible_small_icons_;
 
   bool show_message_center_on_unlock_;
 
   bool should_update_tray_content_;
 
   // True when the shelf auto hide behavior has to be blocked. Previously
   // this was done by checking |message_center_bubble_| but actually
   // the check can be called when creating this object, so it would cause
   // flickers of the shelf from hidden to shown. See: crbug.com/181213
   bool should_block_shelf_auto_hide_;
 
   std::unique_ptr<AshPopupAlignmentDelegate> popup_alignment_delegate_;
 
   DISALLOW_COPY_AND_ASSIGN(WebNotificationTray);
 };
 
 }  // namespace ash
 
 #endif  // ASH_COMMON_SYSTEM_WEB_NOTIFICATION_WEB_NOTIFICATION_TRAY_H_