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"
@@ skipping 41 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);
 }
 
+void BalloonNotificationUIManager::GetAllIdsByProfileAndSourceOrigin(
+    Profile* profile,
+    const GURL& source,
+    std::set<std::string>* notification_ids) {
+  DCHECK(notification_ids);
+  NotificationUIManagerImpl::GetAllIdsByProfileAndSourceOrigin(
+      profile, source, notification_ids);
+
+  const BalloonCollection::Balloons& balloons =
+      balloon_collection_->GetActiveBalloons();
+  for (BalloonCollection::Balloons::const_iterator iter = balloons.begin();
+       iter != balloons.end();

[Dmitry Titov, 2013/05/16 20:39:15]

this loop can be in 2 lines.

[dewittj, 2013/05/16 23:40:34]

ALL HAIL CLANG-FORMAT-DIFF

+       ++iter) {
+    if (profile == (*iter)->profile() &&
+        source == (*iter)->notification().origin_url()) {
+      notification_ids->insert((*iter)->notification().notification_id());
+    }
+  }
+}
+
 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());
 }