Settings: add AccountObserver

1 files changed, 17 insertions, 9 deletions

diff --git a/ios/MullvadVPN/SettingsViewController.swift b/ios/MullvadVPN/SettingsViewController.swift
index 58a382e60e..723f3c1828 100644
--- a/ios/MullvadVPN/SettingsViewController.swift
+++ b/ios/MullvadVPN/SettingsViewController.swift
@@ -13,7 +13,7 @@ protocol SettingsViewControllerDelegate: class {
     func settingsViewControllerDidFinish(_ controller: SettingsViewController)
 }
 
-class SettingsViewController: UITableViewController {
+class SettingsViewController: UITableViewController, AccountObserver {
 
     weak var delegate: SettingsViewControllerDelegate?
 
@@ -59,16 +59,24 @@ class SettingsViewController: UITableViewController {
         navigationItem.largeTitleDisplayMode = .always
         navigationItem.rightBarButtonItem = UIBarButtonItem(barButtonSystemItem: .done, target: self, action: #selector(handleDismiss))
 
-        accountExpiryObserver = NotificationCenter.default.addObserver(
-            forName: Account.didUpdateAccountExpiryNotification,
-            object: Account.shared,
-            queue: OperationQueue.main) { [weak self] (note) in
-                guard let accountRow = self?.accountRow else { return }
+        Account.shared.addObserver(self)
+        setupDataSource()
+    }
 
-                self?.staticDataSource.reloadRows([accountRow], with: .none)
-        }
+    // MARK: - AccountObserver
 
-        setupDataSource()
+    func account(_ account: Account, didUpdateExpiry expiry: Date) {
+        guard let accountRow = accountRow else { return }
+
+        staticDataSource.reloadRows([accountRow], with: .none)
+    }
+
+    func account(_ account: Account, didLoginWithToken token: String, expiry: Date) {
+        // no-op
+    }
+
+    func accountDidLogout(_ account: Account) {
+        // no-op
     }
 
     // MARK: - IBActions