feature: ask for notification permission on news page (#320)

KlubJagiellonski · Mar 16, 2023 · e6c493b · e6c493b
1 parent 3711254
commit e6c493b
Show file tree

Hide file tree

Showing 4 changed files with 37 additions and 9 deletions.
diff --git a/BuyPolish/Pola/AppDelegate.swift b/BuyPolish/Pola/AppDelegate.swift
@@ -12,8 +12,12 @@ class AppDelegate: UIResponder, UIApplicationDelegate {
         analyticsProvider.configure()
         applyAppearance()
 
+        let notificationProvider = DI.container.resolve(NotificationProvider.self)!
+        notificationProvider.register(in: application)
+        self.notificationProvider = notificationProvider
+
         window = UIWindow(frame: UIScreen.main.bounds)
-        window?.rootViewController = RootViewController(analyticsProvider: analyticsProvider)
+        window?.rootViewController = RootViewController(analyticsProvider: analyticsProvider, notificationProvider: notificationProvider)
         window?.backgroundColor = R.color.backgroundWindowColor()
         window?.makeKeyAndVisible()
 
@@ -32,9 +36,6 @@ class AppDelegate: UIResponder, UIApplicationDelegate {
             _ = handleShortcutItem(shortcutItem)
         }
 
-        notificationProvider = DI.container.resolve(NotificationProvider.self)
-        notificationProvider?.register(in: application)
-
         return true
     }
 

diff --git a/BuyPolish/Pola/Manager/NotificationManager/NotificationManager.swift b/BuyPolish/Pola/Manager/NotificationManager/NotificationManager.swift
@@ -3,16 +3,27 @@ import Foundation
 import UIKit
 
 final class NotificationManager: NSObject, NotificationProvider {
+    weak var application: UIApplication?
     func register(in application: UIApplication) {
         UNUserNotificationCenter.current().delegate = self
 
+        self.application = application
+    }
+
+    func requestAuthorization() {
         let authOptions: UNAuthorizationOptions = [.alert, .badge, .sound]
         UNUserNotificationCenter.current().requestAuthorization(
-            options: authOptions,
-            completionHandler: { _, _ in }
-        )
+            options: authOptions) { [weak self] success, _ in
+                if success {
+                    DispatchQueue.main.async {
+                        self?.authorizationGranted()
+                    }
+                }
+        }
+    }
 
-        application.registerForRemoteNotifications()
+    private func authorizationGranted() {
+        application?.registerForRemoteNotifications()
 
         Messaging.messaging().token { token, error in
             if let error = error {

diff --git a/BuyPolish/Pola/Manager/NotificationManager/NotificationProvider.swift b/BuyPolish/Pola/Manager/NotificationManager/NotificationProvider.swift
@@ -1,4 +1,5 @@
 import UIKit
 protocol NotificationProvider {
     func register(in application: UIApplication)
+    func requestAuthorization()
 }
diff --git a/BuyPolish/Pola/UI/RootViewController.swift b/BuyPolish/Pola/UI/RootViewController.swift
@@ -4,16 +4,19 @@ import UIKit
 final class RootViewController: UITabBarController {
     private let scanCodeViewController: ScanCodeViewController
     private let analytics: AnalyticsHelper
+    private let notificationProvider: NotificationProvider
 
     private enum TabOrder: Int, CaseIterable {
         case scan = 0
         case search
         case news
     }
 
-    init(analyticsProvider: AnalyticsProvider) {
+    init(analyticsProvider: AnalyticsProvider,
+         notificationProvider: NotificationProvider) {
         analytics = AnalyticsHelper(provider: analyticsProvider)
         scanCodeViewController = DI.container.resolve(ScanCodeViewController.self)!
+        self.notificationProvider = notificationProvider
         super.init(nibName: nil, bundle: nil)
 
         let strings = R.string.localizable.self
@@ -71,6 +74,18 @@ final class RootViewController: UITabBarController {
     }
 
     override func tabBar(_: UITabBar, didSelect item: UITabBarItem) {
+        requestNotifictionAuthorizationIfNeeded(selectedTab: item)
+        trackTabChanged(item: item)
+    }
+
+    private func requestNotifictionAuthorizationIfNeeded(selectedTab: UITabBarItem) {
+        guard TabOrder(rawValue: selectedTab.tag) == .news else {
+            return
+        }
+        notificationProvider.requestAuthorization()
+    }
+
+    private func trackTabChanged(item: UITabBarItem) {
         let tab = TabOrder(rawValue: item.tag).map { order -> AnalyticsMainTab in
             switch order {
             case .scan: