Retrive displayed notifications asynchronously

chrome/browser/notifications/notification_platform_bridge_mac.mm

 // Copyright 2015 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_platform_bridge_mac.h"
 
 #include <utility>
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
@@ skipping 102 matching lines @@
 // Deliver a notification to the XPC service to be displayed as an alert.
 - (void)dispatchNotification:(NSDictionary*)data;
 
 // Close a notification for a given |notificationId| and |profileId|.
 - (void)closeNotificationWithId:(NSString*)notificationId
                   withProfileId:(NSString*)profileId;
 
 // Close all notifications.
 - (void)closeAllNotifications;
 
+//  Get displayed alerts for |profileId| merge them with |localNotifications|
+// and post |callback| to the IO thread.
+- (void)
+getDisplayedAlertsForProfileId:(NSString*)profileId
+        withLocalNotifications:(NSArray*)localNotifications
+                      callback:(NotificationCommon::NotificationResultCallback)
+                                   callback;
+
 @end
 
 // /////////////////////////////////////////////////////////////////////////////
 NotificationPlatformBridgeMac::NotificationPlatformBridgeMac(
     NSUserNotificationCenter* notification_center,
     id<AlertDispatcher> alert_dispatcher)
     : delegate_([NotificationCenterDelegate alloc]),
       notification_center_([notification_center retain]),
       alert_dispatcher_([alert_dispatcher retain]) {
   [notification_center_ setDelegate:delegate_.get()];
@@ skipping 149 matching lines @@
   NSString* current_profile_id = base::SysUTF8ToNSString(profile_id);
   for (NSUserNotification* toast in
        [notification_center_ deliveredNotifications]) {
     NSString* toast_profile_id = [toast.userInfo
         objectForKey:notification_constants::kNotificationProfileId];
     if ([toast_profile_id isEqualToString:current_profile_id]) {
       notifications->insert(base::SysNSStringToUTF8([toast.userInfo
           objectForKey:notification_constants::kNotificationId]));
     }
   }
+
   return true;
 }
 
+void NotificationPlatformBridgeMac::GetDisplayedAsync(
+    const std::string& profile_id,
+    bool incognito,
+    const NotificationCommon::NotificationResultCallback& callback) const {
+#if BUILDFLAG(ENABLE_XPC_NOTIFICATIONS)
+  [alert_dispatcher_
+      getDisplayedAlertsForProfileId:base::SysUTF8ToNSString(profile_id)
+              withLocalNotifications:[notification_center_
+                                         deliveredNotifications]
+                            callback:callback];
+
+#endif  // ENABLE_XPC_NOTIFICATIONS
+}
+
 // static
 void NotificationPlatformBridgeMac::ProcessNotificationResponse(
     NSDictionary* response) {
   if (!NotificationPlatformBridgeMac::VerifyNotificationData(response))
     return;
 
   NSNumber* button_index =
       [response objectForKey:notification_constants::kNotificationButtonIndex];
   NSNumber* operation =
       [response objectForKey:notification_constants::kNotificationOperation];
@@ skipping 164 matching lines @@
 - (void)closeNotificationWithId:(NSString*)notificationId
                   withProfileId:(NSString*)profileId {
   [[xpcConnection_ remoteObjectProxy] closeNotificationWithId:notificationId
                                                 withProfileId:profileId];
 }
 
 - (void)closeAllNotifications {
   [[xpcConnection_ remoteObjectProxy] closeAllNotifications];
 }
 
+- (void)
+getDisplayedAlertsForProfileId:(NSString*)profileId
+        withLocalNotifications:(NSArray*)localNotifications
+                      callback:(NotificationCommon::NotificationResultCallback)
+                                   callback {
+  auto reply = ^(NSArray* alerts) {
+    std::unique_ptr<std::set<std::string>> displayedNotifications =
+        base::MakeUnique<std::set<std::string>>();
+
+    for (NSUserNotification* toast in localNotifications) {
+      NSString* toastProfileId = [toast.userInfo
+          objectForKey:notification_constants::kNotificationProfileId];
+      if ([toastProfileId isEqualToString:profileId]) {
+        displayedNotifications->insert(base::SysNSStringToUTF8([toast.userInfo
+            objectForKey:notification_constants::kNotificationId]));
+      }
+    }
+
+    for (NSString* alert in alerts) {
+      displayedNotifications->insert(base::SysNSStringToUTF8(alert));
+    }
+
+    content::BrowserThread::PostTask(
+        content::BrowserThread::IO, FROM_HERE,
+        base::Bind(callback, base::Passed(&displayedNotifications),
+                   true /* supports sync */));
+  };
+
+  [[xpcConnection_ remoteObjectProxy] getDisplayedAlertsForProfileId:profileId
+                                                           withReply:reply];
+}
+
 // NotificationReply implementation
 - (void)notificationClick:(NSDictionary*)notificationResponseData {
   NotificationPlatformBridgeMac::ProcessNotificationResponse(
       notificationResponseData);
 }
 
 @end