Remove Notifications Associated with an Extension When the Extension is Unchecked from the Message …

ui/message_center/message_center_impl.h

 // Copyright (c) 2013 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 UI_MESSAGE_CENTER_MESSAGE_CENTER_IMPL_H_
 #define UI_MESSAGE_CENTER_MESSAGE_CENTER_IMPL_H_
 
 #include <string>
 #include <vector>
 
 #include "base/memory/scoped_vector.h"
 #include "base/memory/weak_ptr.h"
 #include "base/stl_util.h"
 #include "base/time/time.h"
 #include "base/timer/timer.h"
 #include "ui/message_center/message_center.h"
 #include "ui/message_center/message_center_observer.h"
 #include "ui/message_center/message_center_types.h"
 #include "ui/message_center/notification_blocker.h"
+#include "ui/message_center/notifier_settings.h"
 
 namespace message_center {
 class NotificationDelegate;
 class MessageCenterImpl;
 
 namespace internal {
 class ChangeQueue;
 class PopupTimersController;
 
 // A class that manages timeout behavior for notification popups.  One instance
@@ skipping 100 matching lines @@
   PopupTimerCollection popup_timers_;
   STLValueDeleter<PopupTimerCollection> popup_deleter_;
 
   DISALLOW_COPY_AND_ASSIGN(PopupTimersController);
 };
 
 }  // namespace internal
 
 // The default implementation of MessageCenter.
 class MessageCenterImpl : public MessageCenter,
-                          public NotificationBlocker::Observer {
+                          public NotificationBlocker::Observer,
+                          public message_center::NotifierSettingsObserver {
  public:
   MessageCenterImpl();
   virtual ~MessageCenterImpl();
 
   // MessageCenter overrides:
   virtual void AddObserver(MessageCenterObserver* observer) OVERRIDE;
   virtual void RemoveObserver(MessageCenterObserver* observer) OVERRIDE;
   virtual void AddNotificationBlocker(NotificationBlocker* blocker) OVERRIDE;
   virtual void RemoveNotificationBlocker(NotificationBlocker* blocker) OVERRIDE;
   virtual void SetVisibility(Visibility visible) OVERRIDE;
@@ skipping 36 matching lines @@
   virtual NotifierSettingsProvider* GetNotifierSettingsProvider() OVERRIDE;
   virtual void SetQuietMode(bool in_quiet_mode) OVERRIDE;
   virtual void EnterQuietModeWithExpire(
       const base::TimeDelta& expires_in) OVERRIDE;
   virtual void RestartPopupTimers() OVERRIDE;
   virtual void PausePopupTimers() OVERRIDE;
 
   // NotificationBlocker::Observer overrides:
   virtual void OnBlockingStateChanged(NotificationBlocker* blocker) OVERRIDE;
 
+  // message_center::NotifierSettingsObserver overrides:
+  virtual void UpdateIconImage(const NotifierId& notifier_id,
+                               const gfx::Image& icon) OVERRIDE;
+  virtual void NotifierGroupChanged() OVERRIDE;
+  virtual void NotifierEnabledChanged(const NotifierId& notifier_id,
+                                      bool enabled) OVERRIDE;
+
  protected:
   virtual void DisableTimersForTest() OVERRIDE;
 
  private:
   struct NotificationCache {
     NotificationCache();
     ~NotificationCache();
     void Rebuild(const NotificationList::Notifications& notifications);
     void RecountUnread();
 
     NotificationList::Notifications visible_notifications;
     size_t unread_count;
   };
 
   void RemoveNotifications(bool by_user, const NotificationBlockers& blockers);
+  void RemoveNotificationsForNotifierId(const NotifierId& notifier_id);
 
   scoped_ptr<NotificationList> notification_list_;
   NotificationCache notification_cache_;
   ObserverList<MessageCenterObserver> observer_list_;
   scoped_ptr<internal::PopupTimersController> popup_timers_controller_;
   scoped_ptr<base::OneShotTimer<MessageCenterImpl> > quiet_mode_timer_;
   NotifierSettingsProvider* settings_provider_;
   std::vector<NotificationBlocker*> blockers_;
 
   // Queue for the notifications to delay the addition/updates when the message
   // center is visible.
   scoped_ptr<internal::ChangeQueue> notification_queue_;
 
   DISALLOW_COPY_AND_ASSIGN(MessageCenterImpl);
 };
 
 }  // namespace message_center
 
 #endif  // UI_MESSAGE_CENTER_MESSAGE_CENTER_H_