feat: restore user data (#61)

paul-nguyen-goldenowl · web-flow · commit 750ab5ce49a0 · 2022-11-20T21:37:47.000+07:00
diff --git a/GoMoney.xcodeproj/project.pbxproj b/GoMoney.xcodeproj/project.pbxproj
@@ -39,6 +39,7 @@
 		083EA336290659660079605F /* TransactionTag.swift in Sources */ = {isa = PBXBuildFile; fileRef = 083EA335290659660079605F /* TransactionTag.swift */; };
 		083EA339290676EB0079605F /* StatViewModel.swift in Sources */ = {isa = PBXBuildFile; fileRef = 083EA338290676EB0079605F /* StatViewModel.swift */; };
 		083EA33B2906933A0079605F /* StatLineChartCell.swift in Sources */ = {isa = PBXBuildFile; fileRef = 083EA33A2906933A0079605F /* StatLineChartCell.swift */; };
+		083EDBAE292A3ED50058F5D7 /* GMAuthViewController.swift in Sources */ = {isa = PBXBuildFile; fileRef = 083EDBAD292A3ED40058F5D7 /* GMAuthViewController.swift */; };
 		085F7526291215B80094A026 /* SettingsViewController.swift in Sources */ = {isa = PBXBuildFile; fileRef = 085F7525291215B80094A026 /* SettingsViewController.swift */; };
 		085F752A2912490E0094A026 /* SettingsTableViewCell.swift in Sources */ = {isa = PBXBuildFile; fileRef = 085F75292912490E0094A026 /* SettingsTableViewCell.swift */; };
 		085F752C29124CEC0094A026 /* BlockerToggle.swift in Sources */ = {isa = PBXBuildFile; fileRef = 085F752B29124CEC0094A026 /* BlockerToggle.swift */; };
@@ -203,6 +204,7 @@
 		083EA335290659660079605F /* TransactionTag.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = TransactionTag.swift; sourceTree = "<group>"; };
 		083EA338290676EB0079605F /* StatViewModel.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = StatViewModel.swift; sourceTree = "<group>"; };
 		083EA33A2906933A0079605F /* StatLineChartCell.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = StatLineChartCell.swift; sourceTree = "<group>"; };
+		083EDBAD292A3ED40058F5D7 /* GMAuthViewController.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = GMAuthViewController.swift; sourceTree = "<group>"; };
 		085F7525291215B80094A026 /* SettingsViewController.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = SettingsViewController.swift; sourceTree = "<group>"; };
 		085F75292912490E0094A026 /* SettingsTableViewCell.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = SettingsTableViewCell.swift; sourceTree = "<group>"; };
 		085F752B29124CEC0094A026 /* BlockerToggle.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = BlockerToggle.swift; sourceTree = "<group>"; };
@@ -703,6 +705,7 @@
 				08DB56E528F8FC7D00170C60 /* View */,
 				08C8731728F865B200DC859D /* SignUp */,
 				08C872EC28F6C5DE00DC859D /* SignIn */,
+				083EDBAD292A3ED40058F5D7 /* GMAuthViewController.swift */,
 			);
 			path = Auth;
 			sourceTree = "<group>";
@@ -1128,6 +1131,7 @@
 				083EA339290676EB0079605F /* StatViewModel.swift in Sources */,
 				082BF96828FD1D510094FE94 /* Podfile in Sources */,
 				088D691C29015865003D0660 /* AddExpenseField.swift in Sources */,
+				083EDBAE292A3ED50058F5D7 /* GMAuthViewController.swift in Sources */,
 				08F0688D2921DBEE005C58EC /* ExchangeCell.swift in Sources */,
 				083B14DF2914EF300058E36E /* SyncManager.swift in Sources */,
 				08C6894D2913B01C002071CC /* TrackingService.swift in Sources */,
diff --git a/GoMoney/Models/Expense.swift b/GoMoney/Models/Expense.swift
@@ -87,3 +87,15 @@ struct DateAmount {
     let date: Date
     var totalAmount: Double
 }
+
+/// Transaction structure on firestore.
+struct RemoteTransaction: Codable {
+    var _id: String
+    var type: String
+    var tag: String
+    var amount: Double
+    var note: String
+    var occuredOn: Date
+    var createdAt: Date?
+    var updatedAt: Date?
+}
diff --git a/GoMoney/Scences/Auth/GMAuthViewController.swift b/GoMoney/Scences/Auth/GMAuthViewController.swift
@@ -0,0 +1,72 @@
+import UIKit
+
+class GMAuthViewController: GMViewController {
+    func initData(completion: @escaping (String?) -> Void) {
+        GMLoadingView.shared.startLoadingAnimation(with: "Initing data ...")
+        TagService.shared.createDefaultDb { err in
+            GMLoadingView.shared.endLoadingAnimation()
+            if let err = err {
+                completion(err.localizedDescription)
+            } else {
+                completion(nil)
+            }
+        }
+    }
+
+    func initDataAndGoHome() {
+        initData { [weak self] err in
+            if let err = err {
+                self?.errorAlert(message: err)
+            } else {
+                self?.navigateToMainVC()
+            }
+        }
+    }
+
+    func navigateToMainVC() {
+        let homeVC = GMTabBarViewController()
+        if let delegate = view.window?.windowScene?.delegate as? SceneDelegate {
+            if let window = delegate.window {
+                window.rootViewController = homeVC
+
+                let options: UIView.AnimationOptions = .transitionCrossDissolve
+                let duration: TimeInterval = 0.5
+                UIView.transition(
+                    with: window,
+                    duration: duration,
+                    options: options,
+                    animations: {},
+                    completion: { _ in })
+            }
+        }
+    }
+
+    func restoreData(completion: @escaping (Error?) -> Void) {
+        GMLoadingView.shared.startLoadingAnimation(with: "Restoring data ...")
+        AuthService.shared.restoreUserData(completion: { err in
+            GMLoadingView.shared.endLoadingAnimation()
+            completion(err)
+        })
+    }
+
+    func restoreDataAndGoHome() {
+        restoreData { [weak self] err in
+            if let err = err {
+                self?.errorAlert(message: err.localizedDescription)
+            } else {
+                self?.navigateToMainVC()
+            }
+        }
+    }
+
+    func checkIfNewUser(completion: @escaping (Bool) -> Void) {
+        RemoteService.shared.checkIfUserExist { result in
+            switch result {
+            case .success(let exist):
+                completion(!exist)
+            case .failure(let err):
+                self.errorAlert(message: err.localizedDescription)
+            }
+        }
+    }
+}
diff --git a/GoMoney/Scences/Auth/SignIn/SignInPasswordVC.swift b/GoMoney/Scences/Auth/SignIn/SignInPasswordVC.swift
@@ -1,6 +1,6 @@
 import UIKit
 
-class SignInPasswordVC: GMViewController {
+class SignInPasswordVC: GMAuthViewController {
     // MARK: - Content
 
     private enum Content {
@@ -116,25 +116,18 @@ class SignInPasswordVC: GMViewController {
                 errorLabel.text = error
             } else {
                 errorLabel.text = ""
-                GMLoadingView.shared.startLoadingAnimation()
+                GMLoadingView.shared.startLoadingAnimation(with: "Logging in ...")
                 viewModel.signInWithEmailAndPassword(email: email, password: password) { [weak self] error in
-                    GMLoadingView.shared.endLoadingAnimation()
                     if error != nil {
+                        GMLoadingView.shared.endLoadingAnimation()
                         self?.errorLabel.text = error?.localizedDescription
                     } else {
-                        self?.onSuccessLogin()
+                        self?.restoreDataAndGoHome()
                     }
                 }
             }
         }
     }
-    
-    private func navigateToMainVC() {
-        let homeVC = GMTabBarViewController()
-        if let delegate = view.window?.windowScene?.delegate as? SceneDelegate {
-            delegate.window?.rootViewController = homeVC
-        }
-    }
 }
 
 extension SignInPasswordVC: UITextFieldDelegate {
diff --git a/GoMoney/Scences/Auth/SignIn/SignInViewController.swift b/GoMoney/Scences/Auth/SignIn/SignInViewController.swift
@@ -7,7 +7,7 @@
 
 import UIKit
 
-class SignInViewController: GMViewController {
+class SignInViewController: GMAuthViewController {
     private lazy var img: UIImageView = .build { view in
         view.image = UIImage(named: "onboard_1")
     }
@@ -108,18 +108,18 @@ class SignInViewController: GMViewController {
             with: self,
             completion: { [weak self] err in
                 if let err = err {
+                    GMLoadingView.shared.endLoadingAnimation()
                     self?.errorAlert(message: err)
                 } else {
-                    self?.navigateToMainVC()
+                    self?.checkIfNewUser { isNew in
+                        if isNew {
+                            self?.initDataAndGoHome()
+                        } else {
+                            self?.restoreDataAndGoHome()
+                        }
+                    }
                 }
             }
         )
     }
-    
-    private func navigateToMainVC() {
-        let homeVC = GMTabBarViewController()
-        if let delegate = view.window?.windowScene?.delegate as? SceneDelegate {
-            delegate.window?.rootViewController = homeVC
-        }
-    }
 }
diff --git a/GoMoney/Scences/Auth/SignUp/SignUpPasswordVC.swift b/GoMoney/Scences/Auth/SignUp/SignUpPasswordVC.swift
@@ -1,6 +1,6 @@
 import UIKit
 
-class SignUpPasswordVC: GMViewController {
+class SignUpPasswordVC: GMAuthViewController {
     private enum Constant {
         static let padding: CGFloat = 16
     }
@@ -121,7 +121,7 @@ class SignUpPasswordVC: GMViewController {
                         self?.errorLabel.text = error?.localizedDescription
                     } else {
                         // TODO: navigateToDetailVC
-                        self?.navigateToMainVC()
+                        self?.initDataAndGoHome()
                     }
                 }
             }
@@ -133,13 +133,6 @@ class SignUpPasswordVC: GMViewController {
         errorLabel.text = ""
     }
     
-    private func navigateToMainVC() {
-        let homeVC = GMTabBarViewController()
-        if let delegate = view.window?.windowScene?.delegate as? SceneDelegate {
-            delegate.window?.rootViewController = homeVC
-        }
-    }
-    
     private func navigateToDetailVC() {
         let vc = SignUpDetailViewController()
         navigationController?.pushViewController(vc, animated: true)
diff --git a/GoMoney/Scences/Profile/ProfileViewController.swift b/GoMoney/Scences/Profile/ProfileViewController.swift
@@ -169,7 +169,18 @@ class ProfileViewController: GMMainViewController {
         let navVC = UINavigationController(rootViewController: signInVC)
 
         if let delegate = view.window?.windowScene?.delegate as? SceneDelegate {
-            delegate.window?.rootViewController = navVC
+            if let window = delegate.window {
+                window.rootViewController = navVC
+
+                let options: UIView.AnimationOptions = .transitionCrossDissolve
+                let duration: TimeInterval = 0.5
+                UIView.transition(
+                    with: window,
+                    duration: duration,
+                    options: options,
+                    animations: {},
+                    completion: { _ in })
+            }
         }
     }
 
diff --git a/GoMoney/Service/AuthService.swift b/GoMoney/Service/AuthService.swift
@@ -120,4 +120,33 @@ class AuthService {
         // remove realm
         DataService.shared.dropAllTable()
     }
+
+    func restoreUserData(completion: @escaping (Error?) -> Void) {
+        RemoteService.shared.getAllTags { result in
+            switch result {
+            case .failure(let err):
+                completion(err)
+            case .success(let tags):
+                TagService.shared.setTags(tags: tags) { err in
+                    if let err = err {
+                        completion(err)
+                    }
+                    else {
+                        print("[restore] \(tags.count) tags")
+                        RemoteService.shared.getAllTransactions { result in
+                            switch result {
+                            case .success(let transactions):
+                                print("[restore] \(transactions.count) transactions")
+                                DataService.shared.addTransactions(transactions) { err in
+                                    completion(err)
+                                }
+                            case .failure(let err):
+                                completion(err)
+                            }
+                        }
+                    }
+                }
+            }
+        }
+    }
 }
diff --git a/GoMoney/Service/DataService.swift b/GoMoney/Service/DataService.swift
@@ -13,6 +13,7 @@ enum DataError: Error {
     case transactionNotFound(_ transaction: Expense)
     case noTransactions
     case userNotFound
+    case tagNotFound
 
     var localizedDescription: String {
         switch self {
@@ -22,6 +23,8 @@ enum DataError: Error {
             return "There is no transactions!"
         case .userNotFound:
             return "User not found!"
+        case .tagNotFound:
+            return "Tag not found!"
         }
     }
 }
@@ -111,6 +114,17 @@ class DataService {
         }
     }
 
+    func addTransactions(_ transactions: [Expense], completion: (Error?) -> Void) {
+        do {
+            try realm.write {
+                realm.add(transactions)
+            }
+            completion(nil)
+        } catch {
+            completion(error)
+        }
+    }
+
     func deleteExpense(expense: Expense, completion: ((Error?) -> Void)? = nil) {
         do {
             try realm.write {
diff --git a/GoMoney/Service/RemoteService.swift b/GoMoney/Service/RemoteService.swift
diff --git a/GoMoney/Service/TagService.swift b/GoMoney/Service/TagService.swift
diff --git a/GoMoney/Service/UserManager.swift b/GoMoney/Service/UserManager.swift

Original file line number	Diff line number	Diff line change
`@@ -1,6 +1,6 @@`
`1`	`1`	`import UIKit`
`2`	`2`
`3`		`-class SignInPasswordVC: GMViewController {`
	`3`	`+class SignInPasswordVC: GMAuthViewController {`
`4`	`4`	`// MARK: - Content`
`5`	`5`
`6`	`6`	`private enum Content {`
`@@ -116,25 +116,18 @@ class SignInPasswordVC: GMViewController {`
`116`	`116`	`errorLabel.text = error`
`117`	`117`	`} else {`
`118`	`118`	`errorLabel.text = ""`
`119`		`- GMLoadingView.shared.startLoadingAnimation()`
	`119`	`+ GMLoadingView.shared.startLoadingAnimation(with: "Logging in ...")`
`120`	`120`	`viewModel.signInWithEmailAndPassword(email: email, password: password) { [weak self] error in`
`121`		`- GMLoadingView.shared.endLoadingAnimation()`
`122`	`121`	`if error != nil {`
	`122`	`+ GMLoadingView.shared.endLoadingAnimation()`
`123`	`123`	`self?.errorLabel.text = error?.localizedDescription`
`124`	`124`	`} else {`
`125`		`- self?.onSuccessLogin()`
	`125`	`+ self?.restoreDataAndGoHome()`
`126`	`126`	`}`
`127`	`127`	`}`
`128`	`128`	`}`
`129`	`129`	`}`
`130`	`130`	`}`
`131`		`-`
`132`		`- private func navigateToMainVC() {`
`133`		`- let homeVC = GMTabBarViewController()`
`134`		`- if let delegate = view.window?.windowScene?.delegate as? SceneDelegate {`
`135`		`- delegate.window?.rootViewController = homeVC`
`136`		`- }`
`137`		`- }`
`138`	`131`	`}`
`139`	`132`
`140`	`133`	`extension SignInPasswordVC: UITextFieldDelegate {`
Original file line number	Diff line number	Diff line change
`@@ -7,7 +7,7 @@`
`7`	`7`
`8`	`8`	`import UIKit`
`9`	`9`
`10`		`-class SignInViewController: GMViewController {`
	`10`	`+class SignInViewController: GMAuthViewController {`
`11`	`11`	`private lazy var img: UIImageView = .build { view in`
`12`	`12`	`view.image = UIImage(named: "onboard_1")`
`13`	`13`	`}`
`@@ -108,18 +108,18 @@ class SignInViewController: GMViewController {`
`108`	`108`	`with: self,`
`109`	`109`	`completion: { [weak self] err in`
`110`	`110`	`if let err = err {`
	`111`	`+ GMLoadingView.shared.endLoadingAnimation()`
`111`	`112`	`self?.errorAlert(message: err)`
`112`	`113`	`} else {`
`113`		`- self?.navigateToMainVC()`
	`114`	`+ self?.checkIfNewUser { isNew in`
	`115`	`+ if isNew {`
	`116`	`+ self?.initDataAndGoHome()`
	`117`	`+ } else {`
	`118`	`+ self?.restoreDataAndGoHome()`
	`119`	`+ }`
	`120`	`+ }`
`114`	`121`	`}`
`115`	`122`	`}`
`116`	`123`	`)`
`117`	`124`	`}`
`118`		`-`
`119`		`- private func navigateToMainVC() {`
`120`		`- let homeVC = GMTabBarViewController()`
`121`		`- if let delegate = view.window?.windowScene?.delegate as? SceneDelegate {`
`122`		`- delegate.window?.rootViewController = homeVC`
`123`		`- }`
`124`		`- }`
`125`	`125`	`}`
Original file line number	Diff line number	Diff line change
`@@ -1,6 +1,6 @@`
`1`	`1`	`import UIKit`
`2`	`2`
`3`		`-class SignUpPasswordVC: GMViewController {`
	`3`	`+class SignUpPasswordVC: GMAuthViewController {`
`4`	`4`	`private enum Constant {`
`5`	`5`	`static let padding: CGFloat = 16`
`6`	`6`	`}`
`@@ -121,7 +121,7 @@ class SignUpPasswordVC: GMViewController {`
`121`	`121`	`self?.errorLabel.text = error?.localizedDescription`
`122`	`122`	`} else {`
`123`	`123`	`// TODO: navigateToDetailVC`
`124`		`- self?.navigateToMainVC()`
	`124`	`+ self?.initDataAndGoHome()`
`125`	`125`	`}`
`126`	`126`	`}`
`127`	`127`	`}`
`@@ -133,13 +133,6 @@ class SignUpPasswordVC: GMViewController {`
`133`	`133`	`errorLabel.text = ""`
`134`	`134`	`}`
`135`	`135`
`136`		`- private func navigateToMainVC() {`
`137`		`- let homeVC = GMTabBarViewController()`
`138`		`- if let delegate = view.window?.windowScene?.delegate as? SceneDelegate {`
`139`		`- delegate.window?.rootViewController = homeVC`
`140`		`- }`
`141`		`- }`
`142`		`-`
`143`	`136`	`private func navigateToDetailVC() {`
`144`	`137`	`let vc = SignUpDetailViewController()`
`145`	`138`	`navigationController?.pushViewController(vc, animated: true)`