Implement the ability to close alerts programatically

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 96 matching lines @@
 // A Cocoa class that represents the delegate of NSUserNotificationCenter and
 // can forward commands to C++.
 @interface NotificationCenterDelegate
     : NSObject<NSUserNotificationCenterDelegate> {
 }
 @end
 
 // Interface to communicate with the Alert XPC service.
 @interface NotificationRemoteDispatcher : NSObject
 
+// 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)closeNotification:(NSString*)notificationId
+              withProfile:(NSString*)profileId;
+
+// Close all notifications.
+- (void)closeAllNotifications;
+
+// Return true if an alert with the given |notificationId| and |profileId|
+// is being displayed. This is based on cached results so it might not
+// be totally accurate.
+- (BOOL)isAlertDisplayed:(NSString*)notificationId
+             withProfile:(NSString*)profileId;
 @end
 
 // /////////////////////////////////////////////////////////////////////////////
 
 NotificationPlatformBridgeMac::NotificationPlatformBridgeMac(
     NSUserNotificationCenter* notification_center)
     : delegate_([NotificationCenterDelegate alloc]),
       notification_center_(notification_center),
 #if BUILDFLAG(ENABLE_XPC_NOTIFICATIONS)
       notification_remote_dispatcher_(
           [[NotificationRemoteDispatcher alloc] init])
 #else
       notification_remote_dispatcher_(nullptr)
 #endif  // ENABLE_XPC_NOTIFICATIONS
 {
   [notification_center_ setDelegate:delegate_.get()];
 }
 
 NotificationPlatformBridgeMac::~NotificationPlatformBridgeMac() {
   [notification_center_ setDelegate:nil];
 
-  // TODO(miguelg) remove only alerts shown by the XPC service.
   // TODO(miguelg) do not remove banners if possible.
   [notification_center_ removeAllDeliveredNotifications];
+#if BUILDFLAG(ENABLE_XPC_NOTIFICATIONS)
+  [notification_remote_dispatcher_ closeAllNotifications];
+#endif  // BUILDFLAG(ENABLE_XPC_NOTIFICATIONS)
 }
 
 void NotificationPlatformBridgeMac::Display(
     NotificationCommon::Type notification_type,
     const std::string& notification_id,
     const std::string& profile_id,
     bool incognito,
     const Notification& notification) {
   base::scoped_nsobject<NotificationBuilder> builder(
       [[NotificationBuilder alloc]
@@ skipping 71 matching lines @@
   }
 #else
   NSUserNotification* toast = [builder buildUserNotification];
   [notification_center_ deliverNotification:toast];
 #endif  // ENABLE_XPC_NOTIFICATIONS
 }
 
 void NotificationPlatformBridgeMac::Close(const std::string& profile_id,
                                           const std::string& notification_id) {
   NSString* candidate_id = base::SysUTF8ToNSString(notification_id);
+  NSString* current_profile_id = base::SysUTF8ToNSString(profile_id);
+#if BUILDFLAG(ENABLE_XPC_NOTIFICATIONS)
+  if ([notification_remote_dispatcher_ isAlertDisplayed:candidate_id
+                                            withProfile:current_profile_id]) {
+    [notification_remote_dispatcher_ closeNotification:candidate_id
+                                           withProfile:current_profile_id];
+    return;
+  }
+#endif  // ENABLE_XPC_NOTIFICATIONS
 
-  NSString* current_profile_id = base::SysUTF8ToNSString(profile_id);
   for (NSUserNotification* toast in
        [notification_center_ deliveredNotifications]) {
     NSString* toast_id =
         [toast.userInfo objectForKey:notification_constants::kNotificationId];
 
     NSString* persistent_profile_id = [toast.userInfo
         objectForKey:notification_constants::kNotificationProfileId];
 
     if ([toast_id isEqualToString:candidate_id] &&
         [persistent_profile_id isEqualToString:current_profile_id]) {
@@ skipping 150 matching lines @@
      shouldPresentNotification:(NSUserNotification*)nsNotification {
   // Always display notifications, regardless of whether the app is foreground.
   return YES;
 }
 
 @end
 
 @implementation NotificationRemoteDispatcher {
   // The connection to the XPC server in charge of delivering alerts.
   base::scoped_nsobject<NSXPCConnection> xpcConnection_;
+  base::scoped_nsobject<NSMutableSet> alertsDisplayed_;
 }
 
 - (instancetype)init {
   if ((self = [super init])) {
+    alertsDisplayed_.reset([[NSMutableSet alloc] init]);

[Robert Sesek, 2016/10/10 18:22:47]

What is the purpose of this set? It looks like things are only ever added or
removed from it, never consulted.

[Miguel Garcia, 2016/10/11 11:55:41]

It is consulted through isAlertDisplayed which is used in the ::Close( method to
figure out if what we should be closing is an alert or a banner.

[Robert Sesek, 2016/10/11 15:28:41]

Sorry, missed that due to a find-in-page bug on Mac.

     xpcConnection_.reset([[NSXPCConnection alloc]
         initWithServiceName:
             [NSString
                 stringWithFormat:notification_constants::kAlertXPCServiceName,
                                  [base::mac::OuterBundle() bundleIdentifier]]]);
     xpcConnection_.get().remoteObjectInterface =
         [NSXPCInterface interfaceWithProtocol:@protocol(NotificationDelivery)];
 
     xpcConnection_.get().interruptionHandler = ^{
       LOG(WARNING) << "connection interrupted: interruptionHandler: ";
@@ skipping 11 matching lines @@
     xpcConnection_.get().exportedInterface =
         [NSXPCInterface interfaceWithProtocol:@protocol(NotificationReply)];
     xpcConnection_.get().exportedObject = self;
     [xpcConnection_ resume];
   }
 
   return self;
 }
 
 - (void)dispatchNotification:(NSDictionary*)data {
+  NSString* notificationId =
+      [data objectForKey:notification_constants::kNotificationId];
+  NSString* profileId =
+      [data objectForKey:notification_constants::kNotificationProfileId];
+  NSString* alertKey = [self _createAlertKeyWithNotification:notificationId
+                                                 withProfile:profileId];
+  [alertsDisplayed_ addObject:alertKey];
   [[xpcConnection_ remoteObjectProxy] deliverNotification:data];
 }
 
+- (void)closeNotification:(NSString*)notificationId
+              withProfile:(NSString*)profileId {
+  NSString* alertKey = [self _createAlertKeyWithNotification:notificationId
+                                                 withProfile:profileId];
+  [alertsDisplayed_ removeObject:alertKey];
+  [[xpcConnection_ remoteObjectProxy] closeNotification:notificationId
+                                            withProfile:profileId];
+}
+
+- (void)closeAllNotifications {
+  [alertsDisplayed_ removeAllObjects];
+  [[xpcConnection_ remoteObjectProxy] closeAllNotifications];
+}
+
+- (BOOL)isAlertDisplayed:(NSString*)notificationId
+             withProfile:(NSString*)profileId {
+  NSString* alertKey = [self _createAlertKeyWithNotification:notificationId
+                                                 withProfile:profileId];
+  return [alertsDisplayed_ containsObject:alertKey];
+}
+
+// Creates a unique identifier for the notification. Only used for internal
+// tracking.
+- (NSString*)_createAlertKeyWithNotification:(NSString*)notificationId

[Robert Sesek, 2016/10/10 18:22:47]

We don't name private methods with _.

[Miguel Garcia, 2016/10/11 11:55:41]

Done.

+                                 withProfile:(NSString*)profileId {
+  return [NSString stringWithFormat:@"%@:%@", notificationId, profileId];
+}
+
 // NotificationReply implementation
 - (void)notificationClick:(NSDictionary*)notificationResponseData {
   NotificationPlatformBridgeMac::ProcessNotificationResponse(
       notificationResponseData);
 }
 
 @end