Cleaning up user data on logout and passing a weak client delegate wr…

…apper to rust.
element-hq · Mar 1, 2022 · 0d33e1e · 0d33e1e
1 parent e50f025
commit 0d33e1e
Show file tree

Hide file tree

Showing 3 changed files with 38 additions and 3 deletions.
diff --git a/ElementX/Sources/AppCoordinator.swift b/ElementX/Sources/AppCoordinator.swift
@@ -64,6 +64,10 @@ class AppCoordinator: AuthenticationCoordinatorDelegate, Coordinator {
     }
 
     func authenticationCoordinatorDidTearDownUserSession(_ authenticationCoordinator: AuthenticationCoordinator) {
+        if let presentedCoordinator = childCoordinators.first {
+            remove(childCoordinator: presentedCoordinator)
+        }
+
         mainNavigationController.setViewControllers([splashViewController], animated: false)
         authenticationCoordinator.start()
     }

diff --git a/ElementX/Sources/Modules/Authentication/AuthenticationCoordinator.swift b/ElementX/Sources/Modules/Authentication/AuthenticationCoordinator.swift
@@ -74,7 +74,13 @@ class AuthenticationCoordinator: Coordinator {
 
     func logout() {
         keychainController.removeAllTokens()
+
+        if let userIdentifier = userSession?.userIdentifier {
+            deleteBaseDirectoryForUsername(userIdentifier)
+        }
+
         userSession = nil
+
         delegate?.authenticationCoordinatorDidTearDownUserSession(self)
     }
 
@@ -183,4 +189,14 @@ class AuthenticationCoordinator: Coordinator {
 
         return url.path
     }
+
+    private func deleteBaseDirectoryForUsername(_ username: String) {
+        guard var url = FileManager.default.urls(for: .cachesDirectory, in: .userDomainMask).first else {
+            fatalError("Should always be able to retrieve the caches directory")
+        }
+
+        url = url.appendingPathComponent(username)
+
+        try? FileManager.default.removeItem(at: url)
+    }
 }
diff --git a/ElementX/Sources/Modules/Authentication/UserSession.swift b/ElementX/Sources/Modules/Authentication/UserSession.swift
@@ -18,18 +18,33 @@ enum UserSessionError: Error {
     case failedRetrievingAvatar
 }
 
+private class WeakUserSessionWrapper: ClientDelegate {
+    weak var userSession: UserSession?
+
+    init(userSession: UserSession) {
+        self.userSession = userSession
+    }
+
+    func didReceiveSyncUpdate() {
+        userSession?.didReceiveSyncUpdate()
+    }
+}
+
 class UserSession: ClientDelegate {
 
     private let client: Client
 
+    deinit {
+        client.setDelegate(delegate: nil)
+    }
+
     let callbacks = PassthroughSubject<UserSessionCallback, Never>()
 
     init(client: Client) {
         self.client = client
 
-        if !client.hasFirstSynced() {
-            client.startSync(delegate: self)
-        }
+        client.setDelegate(delegate: WeakUserSessionWrapper(userSession: self))
+        client.startSync()
     }
 
     var userIdentifier: String {