go-notify-darwin

noti.m (10977B)

---

 //go:build darwin && cgo
 
 #import <stdlib.h>
 #import "_cgo_export.h"
 
 #if  ! __has_feature(objc_arc)
     #error This file must be compiled with ARC. Use -fobjc-arc flag (or convert project to ARC).
 #endif
 
 @import Foundation;
 @import UserNotifications;
 
 // Log levels mirror Go's slog package.
 const int DEBUG = -4;
 const int INFO  = 0;
 const int WARN  = 4;
 const int ERROR = 8;
 
 // notify_log macro provides a var-args wrapper that can do formatting to the Go side log function.
 #define notify_log(level, fmt, ...) golog(level, [[NSString stringWithFormat:@fmt, ##__VA_ARGS__] UTF8String])
 
 @interface NotifyUNDelegate : NSObject <UNUserNotificationCenterDelegate>
 { }
 - (void)userNotificationCenter:(UNUserNotificationCenter *)center didReceiveNotificationResponse:(UNNotificationResponse *)response withCompletionHandler:(void (^)(void))completionHandler;
 - (void)userNotificationCenter:(UNUserNotificationCenter *)center willPresentNotification:(UNNotification *)notification withCompletionHandler:(void (^)(UNNotificationPresentationOptions options))completionHandler;
 - (void)userNotificationCenter:(UNUserNotificationCenter *)center openSettingsForNotification:(UNNotification *)notification;
 @end
 
 @implementation NotifyUNDelegate
 - (void)userNotificationCenter:(UNUserNotificationCenter *)center didReceiveNotificationResponse:(UNNotificationResponse *)response withCompletionHandler:(void (^)(void))completionHandler {
 	notify_log(DEBUG, "NotifyUNDelegate: didReceiveNotificationResponse %@", response);
 
 	NSError  *error; 
 	NSData   *jsonData   = [NSJSONSerialization dataWithJSONObject: response.notification.request.content.userInfo options: NSJSONWritingPrettyPrinted error: &error];
 	NSString *actionID   = response.actionIdentifier;
 	NSString *categoryID = response.notification.request.content.categoryIdentifier;
 	NSString *userText   = @"";
 
 	// NOTE(jfm): macOS passes a different object for text actions (UNTextInputNotificationResponse).
 	// Since we declare the method to use UNNotificationResponse, we must use reflection to grab the 
 	// userText property when the concrete object is actually UNTextInputNotificationResponse.
 	// In the case of a non-text action the userText will be the empty string. 
 
 	@try {
 		userText = [response valueForKey:@"userText"];
 	}
 	@catch (NSException *e) {
 		// ignore
 	}
 
 	// Category, Action, UserText, UserData (as JSON data).
 	gocallback([categoryID UTF8String], [actionID UTF8String], [userText UTF8String], [[[NSString alloc] initWithBytes:[jsonData bytes] length:[jsonData length] encoding:NSUTF8StringEncoding] UTF8String]);
 
 	completionHandler();
 }
 
 - (void)userNotificationCenter:(UNUserNotificationCenter *)center willPresentNotification:(UNNotification *)notification withCompletionHandler:(void (^)(UNNotificationPresentationOptions options))completionHandler {
 	notify_log(DEBUG, "NotifyUNDelegate: willPresentNotification");
 }
 
 - (void)userNotificationCenter:(UNUserNotificationCenter *)center openSettingsForNotification:(UNNotification *)notification {
 	notify_log(DEBUG, "NotifyUNDelegate: openSettingsForNotification");
 }
 @end
 
 UNUserNotificationCenter *nc;
 NotifyUNDelegate *del;
 BOOL enabled;
 BOOL hasBundle;
 
 // noti_setup registers the app delegate.
 void
 noti_setup() {
 	notify_log(DEBUG, "Getting application bundle");
 	enabled = NO;
 	hasBundle = NO;
 	NSBundle *main = [NSBundle mainBundle];
 	if (main.bundleIdentifier == nil) {
 		notify_log(ERROR, "No app bundle.");
 		return;
 	}
 	hasBundle = YES;
 	notify_log(DEBUG, "Bundle ID: %@", main.bundleIdentifier);
 	nc = [UNUserNotificationCenter currentNotificationCenter];
 	del = [NotifyUNDelegate new];
 	nc.delegate = del;
 }
 
 // noti_category_register adds a notification category.
 void
 noti_category_register(Category category) {
 	NSString *category_id = [[NSString alloc] initWithUTF8String:category.id];
 	NSString *preview     = [[NSString alloc] initWithUTF8String:category.preview];
 	NSString *summary     = [[NSString alloc] initWithUTF8String:category.summary];
 
 	if (!hasBundle) {
 		return;
 	}
 
 	NSMutableArray *actions_array = [NSMutableArray new];
 
 	/*
 		Iterate both types of actions and add them to the notification center.
 	*/
 
 	for (uintptr_t ii = 0; ii < category.actions_len; ii++) {
 		Action a = category.actions[ii];
 
 		NSString *a_id    = [[NSString alloc] initWithUTF8String:a.id];
 		NSString *a_icon  = [[NSString alloc] initWithUTF8String:a.icon];
 		NSString *a_title = [[NSString alloc] initWithUTF8String:a.title];
 
 		NSString *a_icon_path = [[NSBundle mainBundle] pathForResource:a_icon.stringByDeletingPathExtension ofType:a_icon.pathExtension]; 
 
 		[actions_array addObject:[UNNotificationAction actionWithIdentifier:a_id title:a_title options:0 icon:[UNNotificationActionIcon iconWithTemplateImageName:a_icon_path]]];
 	}
 
 	for (uintptr_t ii = 0; ii < category.text_input_actions_len; ii++) {
 		TextInputAction tia = category.text_input_actions[ii];
 
 		NSString *tia_id           = [[NSString alloc] initWithUTF8String:tia.id];
 		NSString *tia_icon         = [[NSString alloc] initWithUTF8String:tia.icon];
 		NSString *tia_title        = [[NSString alloc] initWithUTF8String:tia.title];
 		NSString *tia_placeholder  = [[NSString alloc] initWithUTF8String:tia.placeholder];
 		NSString *tia_button_title = [[NSString alloc] initWithUTF8String:tia.button_title];
 
 		NSString *tia_icon_path = [[NSBundle mainBundle] pathForResource:tia_icon.stringByDeletingPathExtension ofType:tia_icon.pathExtension]; 
 
 		[actions_array addObject:[UNTextInputNotificationAction actionWithIdentifier:tia_id title:tia_title options:UNNotificationActionOptionForeground icon:[UNNotificationActionIcon iconWithTemplateImageName:tia_icon_path] textInputButtonTitle:tia_button_title textInputPlaceholder:tia_placeholder]];
 	}
 
 	UNNotificationCategory *un_category = [UNNotificationCategory categoryWithIdentifier:category_id actions:actions_array intentIdentifiers:[NSArray new] hiddenPreviewsBodyPlaceholder:preview categorySummaryFormat:summary options:0];
 
 
 	// We must take care to merge the categories into a single set
 	// otherwise we will overwrite the existing categories with this
 	// one.
 
 	UNUserNotificationCenter *nc = [UNUserNotificationCenter currentNotificationCenter];
 
 	[nc getNotificationCategoriesWithCompletionHandler: ^(NSSet<UNNotificationCategory *> * categories) {
 		NSMutableSet *set = [NSMutableSet setWithSet:categories];
 		[set addObject:un_category];
 		[nc setNotificationCategories:set];
 	}];
 
 	return;
 }
 
 // noti_notify fires a notification.
 void
 noti_notify(Notification n) {
 	if (!hasBundle) {
 		notify_log(ERROR, "no bundle");
 		return;
 	}
 
 	notify_log(DEBUG, "creating notification");
 
 	dispatch_semaphore_t semaphore = dispatch_semaphore_create(0);
 
 	UNMutableNotificationContent *note = [[UNMutableNotificationContent alloc] init];
 
 	note.title              = [[NSString alloc] initWithUTF8String: n.title];
 	note.subtitle           = [[NSString alloc] initWithUTF8String: n.sub_title];
 	note.body               = [[NSString alloc] initWithUTF8String: n.body];
 	note.categoryIdentifier = [[NSString alloc] initWithUTF8String: n.category_id];
 
 	/*
 
 		Process attachments.
 
 	*/
 
 	NSMutableArray *attachments_array = [NSMutableArray new];
 
 	for (uintptr_t ii = 0; ii < n.attachments_len; ii++) {
 
 		NSError *error;
 		NSString *attachment_id   = @""; // id will be generated for us.
 		NSString *attachment_name = [[NSString alloc] initWithUTF8String:n.attachments[ii]];
 		NSString *attachment_path = [[NSBundle mainBundle] pathForResource:attachment_name.stringByDeletingPathExtension ofType:attachment_name.pathExtension];
 		NSURL    *url             = [NSURL fileURLWithPath:attachment_path?attachment_path:attachment_name];
 
 		// TODO: add ability to express clipping rectangle options dictionary.
 		UNNotificationAttachment *attachment = [UNNotificationAttachment attachmentWithIdentifier:attachment_id URL:url options:nil error:&error];
 
 		notify_log(DEBUG, "attachment: %@", attachment);
 
 		if (attachment) {
 			[attachments_array addObject:attachment];
 		} else {
 			notify_log(ERROR, "attachment not found for %@", attachment_name);
 		}
 	}
 
 	note.attachments = attachments_array;
 	
 	/*
 
 		Process user data.
 
 	*/
 
 	NSError      *json_error;
 	NSData       *user_data_data = [[[NSString alloc] initWithUTF8String:n.user_data] dataUsingEncoding:NSUTF8StringEncoding];
 	NSDictionary *user_data_dict = [NSJSONSerialization JSONObjectWithData:user_data_data options:NSJSONReadingFragmentsAllowed error:&json_error];
 
 	if (json_error) {
 		notify_log(ERROR, "unmarshalling user data into dictionary: %@", json_error);
 	}
 
 	note.userInfo = user_data_dict;
 
 	notify_log(DEBUG, "user info: %@", user_data_dict);
 
 	/*
 
 		Build and register ad-hoc category.
 		This logic kicks in the there is not category id set on the notification 
 		and there is one or more actions.
 
 	*/
 
 	if ([note.categoryIdentifier length] == 0 && (n.actions_len > 0 || n.text_input_actions_len > 0)) {
 		notify_log(DEBUG, "creating adhoc category for notification");
 
 		NSString *adhoc_category_id = [[NSUUID UUID] UUIDString];
 
 		noti_category_register((Category){
 			.id = [adhoc_category_id UTF8String],
 			.summary = "adhoc category",
 			.preview = "",
 			.actions = n.actions,
 			.actions_len = n.actions_len,
 			.text_input_actions = n.text_input_actions,
 			.text_input_actions_len = n.text_input_actions_len,
 		});
 
 		note.categoryIdentifier = adhoc_category_id;
 		
 	}
 
 	/*
 	
 		Submit request.
 
 	*/
 
 	notify_log(DEBUG, "creating request");
 
 	UNNotificationRequest *req = [UNNotificationRequest requestWithIdentifier:[NSBundle mainBundle].bundleIdentifier content: note trigger:nil];
 
 	notify_log(DEBUG, "requesting authorization");
 
 	[nc requestAuthorizationWithOptions: UNAuthorizationOptionBadge | UNAuthorizationOptionSound | UNAuthorizationOptionAlert completionHandler: ^(BOOL granted, NSError *error){
 		notify_log(DEBUG, "granted = %s", granted?"true":"false");
 
 		if (error) {
 			notify_log(ERROR, "authorizing: %@", error);
 			return;
 		}
 
 		enabled = granted;
 
 		if (enabled == YES) {
   		notify_log(DEBUG, "adding notification request");
 
   		[nc addNotificationRequest:req withCompletionHandler: ^(NSError *error) {
 				if (error) {
 	  			notify_log(ERROR, "adding notification: error: %@", error);
 				}
   			dispatch_semaphore_signal(semaphore);
   		}];
 
 		} else {
 			notify_log(ERROR, "permission denied");
 			dispatch_semaphore_signal(semaphore);
 		}
 	}];
 
 	dispatch_semaphore_wait(semaphore, DISPATCH_TIME_FOREVER);
 
 	return;
 }
 
 // noti_cancel cancels the specified notification.
 void
 noti_cancel() {
 	if (!hasBundle) {
 		notify_log(ERROR, "no bundle");
 		return;
 	}
 
 	NSString *nid = [NSBundle mainBundle].bundleIdentifier;
 
   @try {
 			[nc removePendingNotificationRequestsWithIdentifiers: @[(NSString*)nid]];
 			[nc removeDeliveredNotificationsWithIdentifiers: @[(NSString*)nid]];
   }
 
   @catch(NSException *ne) {
       notify_log(ERROR, "caught exception when cancelling notification %@: %@", nid, ne);
   }
 }