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

chrome/browser/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.
 
 // Handles the visible notification (or balloons).
 
 #ifndef CHROME_BROWSER_NOTIFICATIONS_BALLOON_COLLECTION_IMPL_H_
 #define CHROME_BROWSER_NOTIFICATIONS_BALLOON_COLLECTION_IMPL_H_
 #pragma once
 
 #include <deque>
 
 #include "base/basictypes.h"
 #include "base/message_loop.h"
 #include "chrome/browser/notifications/balloon_collection.h"
+#include "chrome/browser/notifications/balloon_collection_base.h"
 #include "gfx/point.h"
 #include "gfx/rect.h"
 
 // Mac balloons grow from the top down and have close buttons on top, so
 // offsetting is not necessary for easy multiple-closing.  Other platforms grow
 // from the bottom up and have close buttons on top, so it is necessary.
 #if defined(OS_MACOSX)
 #define USE_OFFSETS 0
 #else
 #define USE_OFFSETS 1
 #endif
 
 // A balloon collection represents a set of notification balloons being
 // shown on the screen.  It positions new notifications according to
 // a layout, and monitors for balloons being closed, which it reports
 // up to its parent, the notification UI manager.
 class BalloonCollectionImpl : public BalloonCollection
 #if USE_OFFSETS
                             , public MessageLoopForUI::Observer
 #endif
 {
  public:
   BalloonCollectionImpl();
   virtual ~BalloonCollectionImpl();
 
   // BalloonCollection interface.
   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& source_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(); }
 
   // MessageLoopForUI::Observer interface.
 #if defined(OS_WIN)
   virtual void WillProcessMessage(const MSG& event) {}
   virtual void DidProcessMessage(const MSG& event);
 #endif
 #if defined(OS_LINUX)
   virtual void WillProcessEvent(GdkEvent* event) {}
   virtual void DidProcessEvent(GdkEvent* event);
 #endif
@@ skipping 66 matching lines @@
     gfx::Rect work_area_;
     DISALLOW_COPY_AND_ASSIGN(Layout);
   };
 
   // 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);
 
  private:
-  // The number of balloons being displayed.
-  int count() const { return balloons_.size(); }
-
   // Adjusts the positions of the balloons (e.g., when one is closed).
   // Implemented by each platform for specific UI requirements.
   void PositionBalloons(bool is_reposition);
 
   // Cross-platform internal implementation for PositionBalloons.
   void PositionBalloonsInternal(bool is_reposition);
 
 #if defined(OS_MACOSX)
   // Get the work area on Mac OS, without inverting the coordinates.
   static gfx::Rect GetMacWorkArea();
 #endif
 
+  // Base implementation for the collection of active balloons.
+  BalloonCollectionBase base_;
+
+  // The layout parameters for balloons in this collection.
+  Layout layout_;
+
 #if USE_OFFSETS
   // Start and stop observing all UI events.
   void AddMessageLoopObserver();
   void RemoveMessageLoopObserver();
 
   // Cancel all offset and reposition the balloons normally.
   void CancelOffsets();
 
   // Handles a mouse motion while the balloons are temporarily offset.
   void HandleMouseMoveEvent();
 
   // Is the current cursor in the balloon area?
   bool IsCursorInBalloonCollection() const;
-#endif
 
-  // Queue of active balloons.
-  typedef std::deque<Balloon*> Balloons;
-  Balloons balloons_;
-
-  // The layout parameters for balloons in this collection.
-  Layout layout_;
-
-#if USE_OFFSETS
   // Factory for generating delayed reposition tasks on mouse motion.
   ScopedRunnableMethodFactory<BalloonCollectionImpl> reposition_factory_;
 
   // Is the balloon collection currently listening for UI events?
   bool added_as_message_loop_observer_;
 #endif
 
   DISALLOW_COPY_AND_ASSIGN(BalloonCollectionImpl);
 };
 
 #endif  // CHROME_BROWSER_NOTIFICATIONS_BALLOON_COLLECTION_IMPL_H_