electron/brightray/browser/notification_presenter_mac.mm

#import "browser/notification_presenter_mac.h"

#import "base/strings/sys_string_conversions.h"
#import "content/public/browser/render_view_host.h"
#import "content/public/common/show_desktop_notification_params.h"

#import <Foundation/Foundation.h>

@interface BRYUserNotificationCenterDelegate : NSObject <NSUserNotificationCenterDelegate>
@end

namespace brightray {

namespace {

struct NotificationID {
  int render_process_id;
  int render_view_id;
  int notification_id;
};

NSString * const kRenderProcessIDKey = @"RenderProcessID";
NSString * const kRenderViewIDKey = @"RenderViewID";
NSString * const kNotificationIDKey = @"NotificationID";

scoped_nsobject<NSUserNotification> CreateUserNotification(
    const content::ShowDesktopNotificationHostMsgParams& params,
    int render_process_id,
    int render_view_id) {
  auto notification = [[NSUserNotification alloc] init];
  notification.title = base::SysUTF16ToNSString(params.title);
  notification.informativeText = base::SysUTF16ToNSString(params.body);
  notification.userInfo = @{
    kRenderProcessIDKey: @(render_process_id),
    kRenderViewIDKey: @(render_view_id),
    kNotificationIDKey: @(params.notification_id),
  };

  return scoped_nsobject<NSUserNotification>(notification);
}

NotificationID GetID(NSUserNotification* notification) {
  NotificationID ID;
  ID.render_process_id = [[notification.userInfo objectForKey:kRenderProcessIDKey] intValue];
  ID.render_view_id = [[notification.userInfo objectForKey:kRenderViewIDKey] intValue];
  ID.notification_id = [[notification.userInfo objectForKey:kNotificationIDKey] intValue];
  return ID;
}

}

NotificationPresenter* NotificationPresenter::Create() {
  return new NotificationPresenterMac;
}

NotificationPresenterMac::NotificationPresenterMac()
    : delegate_([[BRYUserNotificationCenterDelegate alloc] init]) {
  NSUserNotificationCenter.defaultUserNotificationCenter.delegate = delegate_;
}

NotificationPresenterMac::~NotificationPresenterMac() {
}

void NotificationPresenterMac::ShowNotification(
    const content::ShowDesktopNotificationHostMsgParams& params,
    int render_process_id,
    int render_view_id) {
  auto notification = CreateUserNotification(params, render_process_id, render_view_id);
  [NSUserNotificationCenter.defaultUserNotificationCenter deliverNotification:notification];
}

}

@implementation BRYUserNotificationCenterDelegate

- (void)userNotificationCenter:(NSUserNotificationCenter *)center didDeliverNotification:(NSUserNotification *)notification {
  auto ID = brightray::GetID(notification);

  auto host = content::RenderViewHost::FromID(ID.render_process_id, ID.render_view_id);
  if (!host)
    return;

  host->DesktopNotificationPostDisplay(ID.notification_id);
}

- (void)userNotificationCenter:(NSUserNotificationCenter *)center didActivateNotification:(NSUserNotification *)notification {
  auto ID = brightray::GetID(notification);

  auto host = content::RenderViewHost::FromID(ID.render_process_id, ID.render_view_id);
  if (!host)
    return;

  host->DesktopNotificationPostClick(ID.notification_id);
}

- (BOOL)userNotificationCenter:(NSUserNotificationCenter *)center shouldPresentNotification:(NSUserNotification *)notification {
  // Display notifications even if the app is active.
  return YES;
}

@end