From aa0a5d2269441577dd4ce485134b2d47fdeb029a Mon Sep 17 00:00:00 2001
From: Stuart Breckenridge <stuartb@hey.com>
Date: Tue, 2 Feb 2021 11:54:47 +0800
Subject: [PATCH] Checks for notification permissions

If notifications are authorised, the toggle will work.

Otherwise, an alert is shown giving the user required information to enable notifications.
---
 ...idebarViewController+ContextualMenus.swift | 44 ++++++++++++++++++-
 1 file changed, 42 insertions(+), 2 deletions(-)

diff --git a/Mac/MainWindow/Sidebar/SidebarViewController+ContextualMenus.swift b/Mac/MainWindow/Sidebar/SidebarViewController+ContextualMenus.swift
index c51698b8c..d28fb85c1 100644
--- a/Mac/MainWindow/Sidebar/SidebarViewController+ContextualMenus.swift
+++ b/Mac/MainWindow/Sidebar/SidebarViewController+ContextualMenus.swift
@@ -10,6 +10,7 @@ import AppKit
 import Articles
 import Account
 import RSCore
+import UserNotifications
 
 extension Notification.Name {
 	public static let DidUpdateFeedPreferencesFromContextMenu = Notification.Name(rawValue: "DidUpdateFeedPreferencesFromContextMenu")
@@ -108,8 +109,28 @@ extension SidebarViewController {
 			  let feed = item.representedObject as? WebFeed else {
 			return
 		}
-		feed.isNotifyAboutNewArticles?.toggle()
-		NotificationCenter.default.post(Notification(name: .DidUpdateFeedPreferencesFromContextMenu))
+		UNUserNotificationCenter.current().getNotificationSettings { (settings) in
+			if settings.authorizationStatus == .denied {
+				self.showNotificationsNotEnabledAlert()
+			} else if settings.authorizationStatus == .authorized {
+				DispatchQueue.main.async {
+					feed.isNotifyAboutNewArticles?.toggle()
+					NotificationCenter.default.post(Notification(name: .DidUpdateFeedPreferencesFromContextMenu))
+				}
+			} else {
+				UNUserNotificationCenter.current().requestAuthorization(options: [.badge, .sound, .alert]) { (granted, error) in
+					if granted {
+						DispatchQueue.main.async {
+							feed.isNotifyAboutNewArticles?.toggle()
+							NotificationCenter.default.post(Notification(name: .DidUpdateFeedPreferencesFromContextMenu))
+							NSApplication.shared.registerForRemoteNotifications()
+						}
+					} else {
+						self.showNotificationsNotEnabledAlert()
+					}
+				}
+			}
+		}
 	}
 	
 	@objc func toggleArticleExtractorFromContextMenu(_ sender: Any?) {
@@ -121,6 +142,25 @@ extension SidebarViewController {
 		NotificationCenter.default.post(Notification(name: .DidUpdateFeedPreferencesFromContextMenu))
 	}
 	
+	func showNotificationsNotEnabledAlert() {
+		DispatchQueue.main.async {
+			let alert = NSAlert()
+			alert.messageText = NSLocalizedString("Notifications are not enabled", comment: "Notifications are not enabled.")
+			alert.informativeText = NSLocalizedString("You can enable NetNewsWire notifications in System Preferences.", comment: "Notifications are not enabled.")
+			alert.addButton(withTitle: NSLocalizedString("Open System Preferences", comment: "Open System Preferences"))
+			alert.addButton(withTitle: NSLocalizedString("Dismiss", comment: "Dismiss"))
+			let userChoice = alert.runModal()
+			if userChoice == .alertFirstButtonReturn {
+				let config = NSWorkspace.OpenConfiguration()
+				config.activates = true
+				// If System Preferences is already open, and no delay is provided here, then it appears in the foreground and immediately disappears.
+				DispatchQueue.main.asyncAfter(wallDeadline: .now() + 0.2, execute: {
+					NSWorkspace.shared.open(URL(string: "x-apple.systempreferences:com.apple.preference.notifications")!, configuration: config)
+				})
+			}
+		}
+	}
+	
 }
 
 extension SidebarViewController: RenameWindowControllerDelegate {