chore: refactor notifications to support categories (#265)

Author: ethanndickson

Coder-Desktop/Coder-Desktop/Coder_DesktopApp.swift

@@ -49,11 +49,10 @@ class AppDelegate: NSObject, NSApplicationDelegate {
     let state: AppState
     let fileSyncDaemon: MutagenDaemon
     let urlHandler: URLHandler
-    let notifDelegate: NotifDelegate
     let autoUpdater: UpdaterService
 
     override init() {
-        notifDelegate = NotifDelegate()
+        AppDelegate.registerNotificationCategories()
         vpn = CoderVPNService()
         autoUpdater = UpdaterService()
         let state = AppState(onChange: vpn.configureTunnelProviderProtocol)
@@ -79,8 +78,9 @@ class AppDelegate: NSObject, NSApplicationDelegate {
         }
         self.fileSyncDaemon = fileSyncDaemon
         urlHandler = URLHandler(state: state, vpn: vpn)
+        super.init()
         // `delegate` is weak
-        UNUserNotificationCenter.current().delegate = notifDelegate
+        UNUserNotificationCenter.current().delegate = self
     }
 
     func applicationDidFinishLaunching(_: Notification) {
@@ -164,7 +164,11 @@ class AppDelegate: NSObject, NSApplicationDelegate {
         do { try urlHandler.handle(url) } catch let handleError {
             Task {
                 do {
-                    try await sendNotification(title: "Failed to handle link", body: handleError.description)
+                    try await sendNotification(
+                        title: "Failed to handle link",
+                        body: handleError.description,
+                        category: .uriFailure
+                    )
                 } catch let notifError {
                     logger.error("Failed to send notification (\(handleError.description)): \(notifError)")
                 }

Coder-Desktop/Coder-Desktop/Notifications.swift

@@ -1,8 +1,12 @@
 import UserNotifications
 
-class NotifDelegate: NSObject, UNUserNotificationCenterDelegate {
-    override init() {
-        super.init()
+extension AppDelegate: UNUserNotificationCenterDelegate {
+    static func registerNotificationCategories() {
+        UNUserNotificationCenter.current().setNotificationCategories(
+            Set(NotificationCategory.allCases.map {
+                UNNotificationCategory(identifier: $0.rawValue, actions: [], intentIdentifiers: [], options: [])
+            })
+        )
     }
 
     // This function is required for notifications to appear as banners whilst the app is running.
@@ -15,7 +19,7 @@ class NotifDelegate: NSObject, UNUserNotificationCenterDelegate {
     }
 }
 
-func sendNotification(title: String, body: String) async throws {
+func sendNotification(title: String, body: String, category: NotificationCategory) async throws {
     let nc = UNUserNotificationCenter.current()
     let granted = try await nc.requestAuthorization(options: [.alert, .badge])
     guard granted else {
@@ -24,5 +28,10 @@ func sendNotification(title: String, body: String) async throws {
     let content = UNMutableNotificationContent()
     content.title = title
     content.body = body
+    content.categoryIdentifier = category.rawValue
     try await nc.add(.init(identifier: UUID().uuidString, content: content, trigger: nil))
 }
+
+enum NotificationCategory: String, CaseIterable {
+    case uriFailure = "URI_FAILURE"
+}