Fixed issues with user settings

zentalk · May 26, 2022 · d408575 · d408575
1 parent 86cbce0
commit d408575
Show file tree

Hide file tree

Showing 4 changed files with 30 additions and 0 deletions.
diff --git a/Jotify.xcworkspace/xcuserdata/harrisonleath.xcuserdatad/UserInterfaceState.xcuserstate b/Jotify.xcworkspace/xcuserdata/harrisonleath.xcuserdatad/UserInterfaceState.xcuserstate
diff --git a/Jotify/Controller/Settings/AccountSettingsController.swift b/Jotify/Controller/Settings/AccountSettingsController.swift
@@ -111,6 +111,7 @@ class AccountSettingsController: SettingsController {
                                     alertController.addAction(UIAlertAction(title: "OK", style: .default, handler: nil))
                                     self.present(alertController, animated: true, completion: nil)
                                 }
+                                User.updateSettings()
                             }
                         } else {
                             let alertController = UIAlertController(title: "Error", message: "There was an error resetting settings. Please try again later.", preferredStyle: .alert)

diff --git a/Jotify/Controller/SetupController.swift b/Jotify/Controller/SetupController.swift
@@ -78,15 +78,43 @@ class SetupController {
                 SetupController.firstLauch = true
                 setupWidget()
                 setupDefaults()
+                checkForRestorePurchase()
             } else {
                 // first launch
                 print("first launch")
                 SetupController.firstLauch = true
                 setupWidget()
                 setupDefaults()
+                checkForRestorePurchase()
             }
             defaults.set(currentVersion, forKey: shortVersionKey)
         }
     }
 
+    func checkForRestorePurchase() {
+        IAPManager.shared.restorePurchases { (result) in
+            DispatchQueue.main.async {
+                switch result {
+                case .success(let success):
+                    if success {
+                        //unlock premium
+                        DataManager.updateUserSettings(setting: "hasPremium", value: true) { success in
+                            if !success! {
+                                print("Error granting premium from restore")
+                            }
+                            print("User settings updated...")
+                            User.settings?.hasPremium = true
+                        }
+                    } else {
+                        //no products were found
+                        print("Nothing to automatically restore...")
+                    }
+
+                case .failure(let error):
+                    //there was an error
+                    print("\(error) restoring IAP")
+                }
+            }
+        }
+    }
 }
diff --git a/Jotify/Model/Firebase/DataManager.swift b/Jotify/Model/Firebase/DataManager.swift
@@ -23,6 +23,7 @@ class DataManager {
             "defaultView": 0,
             "referrals": 0,
             "referralLink": "",
+            "hasPremium": false
         ]) { (error) in
             if let error = error {
                 print("Error deleting document: \(error.localizedDescription)")