Add action and store to support updating notification settings

woocommerce · Dec 20, 2024 · 9d38e56 · 9d38e56
1 parent 9963dc7
commit 9d38e56
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/Yosemite/Yosemite/Actions/AccountAction.swift b/Yosemite/Yosemite/Actions/AccountAction.swift
@@ -15,5 +15,6 @@ public enum AccountAction: Action {
     case synchronizeSitesAndReturnSelectedSiteInfo(siteAddress: String, onCompletion: (Result<Site, Error>) -> Void)
     case synchronizeSitePlan(siteID: Int64, onCompletion: (Result<Void, Error>) -> Void)
     case updateAccountSettings(userID: Int64, tracksOptOut: Bool, onCompletion: (Result<Void, Error>) -> Void)
+    case updateNotificationSettings(notificationSettings: NotificationSettings, onCompletion: (Result<Void, Error>) -> Void)
     case closeAccount(onCompletion: (Result<Void, Error>) -> Void)
 }
diff --git a/Yosemite/Yosemite/Stores/AccountStore.swift b/Yosemite/Yosemite/Stores/AccountStore.swift
@@ -56,6 +56,8 @@ public class AccountStore: Store {
             synchronizeSitePlan(siteID: siteID, onCompletion: onCompletion)
         case .updateAccountSettings(let userID, let tracksOptOut, let onCompletion):
             updateAccountSettings(userID: userID, tracksOptOut: tracksOptOut, onCompletion: onCompletion)
+        case .updateNotificationSettings(let notificationSettings, let onCompletion):
+            updateNotificationSettings(notificationSettings: notificationSettings, onCompletion: onCompletion)
         case .closeAccount(let onCompletion):
             closeAccount(onCompletion: onCompletion)
         }
@@ -223,6 +225,17 @@ private extension AccountStore {
         }
     }
 
+    func updateNotificationSettings(notificationSettings: NotificationSettings, onCompletion: @escaping (Result<Void, Error>) -> Void) {
+        Task {
+            do {
+                try await remote.updateNotificationSettings(with: notificationSettings)
+                onCompletion(.success(()))
+            } catch {
+                onCompletion(.failure(error))
+            }
+        }
+    }
+
     func closeAccount(onCompletion: @escaping (Result<Void, Error>) -> Void) {
         Task {
             do {