Reland 201932: Add API function chrome.notifications.getAll

chrome/browser/notifications/balloon_notification_ui_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/balloon_notification_ui_manager.h"
 
 #include "base/logging.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/prefs/pref_service.h"
 #include "base/stl_util.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/fullscreen.h"
 #include "chrome/browser/idle.h"
 #include "chrome/browser/notifications/balloon_collection.h"
 #include "chrome/browser/notifications/notification.h"
+#include "chrome/browser/profiles/profile.h"
 #include "chrome/common/chrome_notification_types.h"
 #include "chrome/common/pref_names.h"
 #include "content/public/browser/notification_service.h"
 
 BalloonNotificationUIManager::BalloonNotificationUIManager(
     PrefService* local_state)
     : NotificationUIManagerImpl(),
       NotificationPrefsManager(local_state),
       balloon_collection_(NULL) {
   position_pref_.Init(
@@ skipping 26 matching lines @@
 }
 
 bool BalloonNotificationUIManager::CancelById(const std::string& id) {
   // See if this ID hasn't been shown yet.
   if (NotificationUIManagerImpl::CancelById(id))
     return true;
   // If it has been shown, remove it from the balloon collections.
   return balloon_collection_->RemoveById(id);
 }
 
+std::set<std::string>
+BalloonNotificationUIManager::GetAllIdsByProfileAndSourceOrigin(
+    Profile* profile,
+    const GURL& source) {
+  std::set<std::string> notification_ids =
+      NotificationUIManagerImpl::GetAllIdsByProfileAndSourceOrigin(profile,
+                                                                   source);
+
+  const BalloonCollection::Balloons& balloons =
+      balloon_collection_->GetActiveBalloons();
+  for (BalloonCollection::Balloons::const_iterator iter = balloons.begin();
+       iter != balloons.end(); ++iter) {
+    if (profile->IsSameProfile((*iter)->profile()) &&
+        source == (*iter)->notification().origin_url()) {
+      notification_ids.insert((*iter)->notification().notification_id());
+    }
+  }
+  return notification_ids;
+}
+
 bool BalloonNotificationUIManager::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);
   return balloon_collection_->RemoveBySourceOrigin(source) || removed;
 }
 
 bool BalloonNotificationUIManager::CancelAllByProfile(Profile* profile) {
   // Same pattern as CancelAllBySourceOrigin.
   bool removed = NotificationUIManagerImpl::CancelAllByProfile(profile);
@@ skipping 75 matching lines @@
     BalloonNotificationUIManager::GetInstanceForTesting() {
   if (NotificationUIManager::DelegatesToMessageCenter()) {
     LOG(ERROR) << "Attempt to run a test that requires "
                << "BalloonNotificationUIManager while delegating to a "
                << "native MessageCenter. Test will fail. Ask dimich@";
     return NULL;
   }
   return static_cast<BalloonNotificationUIManager*>(
       g_browser_process->notification_ui_manager());
 }