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

chrome/browser/notifications/notification_ui_manager.cc

 // 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.
 
 #include "chrome/browser/notifications/notification_ui_manager.h"
 
 #include "base/logging.h"
 #include "base/scoped_ptr.h"
 #include "base/stl_util-inl.h"
 #include "chrome/browser/notifications/balloon_collection.h"
@@ skipping 48 matching lines @@
     return;
   }
 
   VLOG(1) << "Added notification. URL: "
           << notification.content_url().spec();
   show_queue_.push_back(
       new QueuedNotification(notification, profile));
   CheckAndShowNotifications();
 }
 
-bool NotificationUIManager::Cancel(const Notification& notification) {
-  // First look through the notifications that haven't been shown.  If not
-  // found there, call to the active balloon collection to tear it down.
+bool NotificationUIManager::CancelById(const std::string& id) {
+  // See if this ID hasn't been shown yet.
   NotificationDeque::iterator iter;
   for (iter = show_queue_.begin(); iter != show_queue_.end(); ++iter) {
-    if (notification.IsSame((*iter)->notification())) {
+    if ((*iter)->notification().notification_id() == id) {
       show_queue_.erase(iter);
       return true;
     }
   }
-  return balloon_collection_->Remove(notification);
+  // If it has been shown, remove it from the balloon collections.
+  return balloon_collection_->RemoveById(id);
+}
+
+bool NotificationUIManager::CancelAllBySourceOrigin(const GURL& source) {
+  // Same pattern as CancelById, but more complicated than the above
+  // because there may be multiple notifications from the same source.
+  bool removed = false;
+  NotificationDeque::iterator iter;
+  for (iter = show_queue_.begin(); iter != show_queue_.end(); ++iter) {
+    if ((*iter)->notification().origin_url() == source) {
+      iter = show_queue_.erase(iter);
+      removed = true;
+    } else {
+      ++iter;
+    }
+  }
+
+  return balloon_collection_->RemoveBySourceOrigin(source) || removed;
 }
 
 void NotificationUIManager::CheckAndShowNotifications() {
   // TODO(johnnyg): http://crbug.com/25061 - Check for user idle/presentation.
   ShowNotifications();
 }
 
 void NotificationUIManager::ShowNotifications() {
   while (!show_queue_.empty() && balloon_collection_->HasSpace()) {
     scoped_ptr<QueuedNotification> queued_notification(show_queue_.front());
@@ skipping 33 matching lines @@
        ++balloon_iter) {
     if (origin == (*balloon_iter)->notification().origin_url() &&
         replace_id == (*balloon_iter)->notification().replace_id()) {
       (*balloon_iter)->Update(notification);
       return true;
     }
   }
 
   return false;
 }