When an extension is uninstalled, close all desktop notifications from that e...

chrome/browser/chromeos/notifications/balloon_collection_impl.h

 // Copyright (c) 2010 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 CHROME_BROWSER_CHROMEOS_NOTIFICATIONS_BALLOON_COLLECTION_IMPL_H_
 #define CHROME_BROWSER_CHROMEOS_NOTIFICATIONS_BALLOON_COLLECTION_IMPL_H_
 #pragma once
 
 #include <string>
 
 #include "base/basictypes.h"
 #include "base/scoped_ptr.h"
 #include "chrome/browser/chromeos/notifications/balloon_view_host.h"
 #include "chrome/browser/notifications/balloon_collection.h"
+#include "chrome/browser/notifications/balloon_collection_base.h"
 #include "chrome/common/notification_registrar.h"
 #include "gfx/point.h"
 #include "gfx/rect.h"
 
 namespace gfx {
 class Size;
 }  // namespace gfx
 
 namespace chromeos {
 
@@ skipping 28 matching lines @@
    private:
     DISALLOW_COPY_AND_ASSIGN(NotificationUI);
   };
 
   BalloonCollectionImpl();
   virtual ~BalloonCollectionImpl();
 
   // BalloonCollectionInterface overrides
   virtual void Add(const Notification& notification,
                    Profile* profile);
-  virtual bool Remove(const Notification& notification);
+  virtual bool RemoveById(const std::string& id);
+  virtual bool RemoveBySourceOrigin(const GURL& origin);
   virtual bool HasSpace() const;
   virtual void ResizeBalloon(Balloon* balloon, const gfx::Size& size);
   virtual void DisplayChanged() {}
   virtual void OnBalloonClosed(Balloon* source);
-  virtual const Balloons& GetActiveBalloons() { return balloons_; }
+  virtual const Balloons& GetActiveBalloons() { return base_.balloons(); }
 
   // NotificationObserver overrides:
   virtual void Observe(NotificationType type,
                        const NotificationSource& source,
                        const NotificationDetails& details);
 
   // Adds a callback for DOMUI message. Returns true if the callback
   // is succssfully registered, or false otherwise. It fails to add if
   // there is no notification that matches NotificationDelegate::id(),
   // or a callback for given message already exists. The callback
@@ skipping 29 matching lines @@
   NotificationUI* notification_ui() {
     return notification_ui_.get();
   }
 
  protected:
   // Creates a new balloon. Overridable by unit tests.  The caller is
   // responsible for freeing the pointer returned.
   virtual Balloon* MakeBalloon(const Notification& notification,
                                Profile* profile);
 
+  // Base implementation for the collection of active balloons.
+  BalloonCollectionBase base_;
+
  private:
   friend class NotificationPanelTester;
 
   // Shutdown the notification ui.
   void Shutdown();
 
-  // The number of balloons being displayed.
-  int count() const { return balloons_.size(); }
-
-  Balloons::iterator FindBalloon(const Notification& notification);
-
-  // Queue of active balloons.
-  Balloons balloons_;
+  Balloon* FindBalloon(const Notification& notification) {
+    return base_.FindBalloon(notification);
+  }
 
   scoped_ptr<NotificationUI> notification_ui_;
 
   NotificationRegistrar registrar_;
 
   DISALLOW_COPY_AND_ASSIGN(BalloonCollectionImpl);
 };
 
 }  // namespace chromeos
 
 #endif  // CHROME_BROWSER_CHROMEOS_NOTIFICATIONS_BALLOON_COLLECTION_IMPL_H_