Reland 201932: Add API function chrome.notifications.getAll

chrome/browser/notifications/message_center_notification_manager.cc

 // 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.
 
 #include "chrome/browser/notifications/message_center_notification_manager.h"
 
 #include "base/logging.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/prefs/pref_service.h"
 #include "chrome/browser/extensions/extension_info_map.h"
@@ skipping 52 matching lines @@
 
   // If it has been shown, remove it.
   NotificationMap::iterator iter = profile_notifications_.find(id);
   if (iter == profile_notifications_.end())
     return false;
 
   message_center_->RemoveNotification(id, /* by_user */ false);
   return true;
 }
 
+std::set<std::string>
+MessageCenterNotificationManager::GetAllIdsByProfileAndSourceOrigin(
+    Profile* profile,
+    const GURL& source) {
+
+  std::set<std::string> notification_ids =
+      NotificationUIManagerImpl::GetAllIdsByProfileAndSourceOrigin(profile,
+                                                                   source);
+
+  for (NotificationMap::iterator iter = profile_notifications_.begin();
+       iter != profile_notifications_.end(); iter++) {
+    if ((*iter).second->notification().origin_url() == source &&
+        profile->IsSameProfile((*iter).second->profile())) {
+      notification_ids.insert(iter->first);
+    }
+  }
+  return notification_ids;
+}
+
 bool MessageCenterNotificationManager::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 = NotificationUIManagerImpl::CancelAllBySourceOrigin(source);
 
   for (NotificationMap::iterator loopiter = profile_notifications_.begin();
        loopiter != profile_notifications_.end(); ) {
     NotificationMap::iterator curiter = loopiter++;
     if ((*curiter).second->notification().origin_url() == source) {
       message_center_->RemoveNotification(curiter->first, /* by_user */ false);
       removed = true;
     }
   }
   return removed;
 }
 
 bool MessageCenterNotificationManager::CancelAllByProfile(Profile* profile) {
   // Same pattern as CancelAllBySourceOrigin.
   bool removed = NotificationUIManagerImpl::CancelAllByProfile(profile);
 
   for (NotificationMap::iterator loopiter = profile_notifications_.begin();
        loopiter != profile_notifications_.end(); ) {
     NotificationMap::iterator curiter = loopiter++;
-    if ((*curiter).second->profile() == profile) {
+    if (profile->IsSameProfile((*curiter).second->profile())) {
       message_center_->RemoveNotification(curiter->first, /* by_user */ false);
       removed = true;
     }
   }
   return removed;
 }
 
 void MessageCenterNotificationManager::CancelAll() {
   NotificationUIManagerImpl::CancelAll();
 
@@ skipping 380 matching lines @@
 
 MessageCenterNotificationManager::ProfileNotification*
     MessageCenterNotificationManager::FindProfileNotification(
         const std::string& id) const {
   NotificationMap::const_iterator iter = profile_notifications_.find(id);
   if (iter == profile_notifications_.end())
     return NULL;
 
   return (*iter).second;
 }