Add logout logic

TortugaPower · Apr 12, 2022 · aac5f00 · aac5f00
1 parent e766957
commit aac5f00
Show file tree

Hide file tree

Showing 6 changed files with 38 additions and 4 deletions.
diff --git a/BookPlayer/Coordinators/AccountCoordinator.swift b/BookPlayer/Coordinators/AccountCoordinator.swift
@@ -28,7 +28,7 @@ class AccountCoordinator: Coordinator {
 
   override func start() {
     let vc = AccountViewController.instantiate(from: .Profile)
-    let viewModel = AccountViewModel()
+    let viewModel = AccountViewModel(accountService: self.accountService)
     viewModel.coordinator = self
     vc.viewModel = viewModel
     vc.navigationItem.largeTitleDisplayMode = .never

diff --git a/BookPlayer/Profile/Account Screen/AccountViewController.swift b/BookPlayer/Profile/Account Screen/AccountViewController.swift
@@ -23,6 +23,10 @@ class AccountViewController: BaseTableViewController<AccountCoordinator, Account
   @IBOutlet weak var completeAccountButton: UIButton!
   @IBOutlet var secondaryLabels: [UILabel]!
 
+  enum AccountSection: Int {
+    case info = 0, pro, logout, delete
+  }
+
   override func viewDidLoad() {
     super.viewDidLoad()
 
@@ -51,6 +55,19 @@ class AccountViewController: BaseTableViewController<AccountCoordinator, Account
     return CGFloat.leastNormalMagnitude
   }
 
+  override func tableView(_ tableView: UITableView, didSelectRowAt indexPath: IndexPath) {
+    tableView.deselectRow(at: indexPath as IndexPath, animated: true)
+
+    switch indexPath.section {
+    case AccountSection.logout.rawValue:
+      self.viewModel.handleLogout()
+    case AccountSection.delete.rawValue:
+      self.viewModel.handleDelete()
+    default:
+      break
+    }
+  }
+
   @IBAction func didPressClose(_ sender: UIBarButtonItem) {
     self.viewModel.dismiss()
   }

diff --git a/BookPlayer/Profile/Account Screen/AccountViewModel.swift b/BookPlayer/Profile/Account Screen/AccountViewModel.swift
@@ -7,9 +7,25 @@
 //
 
 import Foundation
+import BookPlayerKit
 
 class AccountViewModel: BaseViewModel<AccountCoordinator> {
+  let accountService: AccountServiceProtocol
+
+  init(accountService: AccountServiceProtocol) {
+    self.accountService = accountService
+  }
+
   func showCompleteAccount() {
     self.coordinator.showCompleteAccount()
   }
+
+  func handleLogout() {
+    self.accountService.logout()
+    self.dismiss()
+  }
+
+  func handleDelete() {
+    // TODO: handle delete account
+  }
 }
diff --git a/BookPlayer/Profile/Login Screen/LoginViewModel.swift b/BookPlayer/Profile/Login Screen/LoginViewModel.swift
@@ -34,7 +34,6 @@ class LoginViewModel: BaseViewModel<LoginCoordinator> {
       Purchases.shared.logIn(appleIDCredential.user) { _, _, _ in }
 
       // TODO: network call to create the user in backend
-      NotificationCenter.default.post(name: .accountUpdate, object: self)
 
       self.coordinator.showCompleteAccount()
     default:

diff --git a/BookPlayer/Settings/Plus Screen/PlusViewModel.swift b/BookPlayer/Settings/Plus Screen/PlusViewModel.swift
@@ -48,7 +48,5 @@ final class PlusViewModel {
       hasSubscription: nil,
       accessToken: nil
     )
-
-    NotificationCenter.default.post(name: .accountUpdate, object: self)
   }
 }
diff --git a/Shared/Services/AccountService.swift b/Shared/Services/AccountService.swift
@@ -94,6 +94,8 @@ public final class AccountService: AccountServiceProtocol {
     }
 
     self.dataManager.saveContext()
+
+    NotificationCenter.default.post(name: .accountUpdate, object: self)
   }
 
   public func logout() {
@@ -105,6 +107,8 @@ public final class AccountService: AccountServiceProtocol {
     account.accessToken = ""
 
     self.dataManager.saveContext()
+
+    NotificationCenter.default.post(name: .accountUpdate, object: self)
   }
 
   public func deleteAccount() {