chore: add waitUntilCompletes option to LogoutUseCase (#2185)

wireapp · Nov 3, 2023 · 76b3f95 · 76b3f95
1 parent 821e139
commit 76b3f95
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/logic/src/commonMain/kotlin/com/wire/kalium/logic/feature/auth/LogoutUseCase.kt b/logic/src/commonMain/kotlin/com/wire/kalium/logic/feature/auth/LogoutUseCase.kt
@@ -43,9 +43,10 @@ import kotlinx.coroutines.launch
 interface LogoutUseCase {
     /**
      * @param reason the reason for the logout performed
+     * @param waitUntilCompletes if true, the logout suspend fun will wait until all the logout operations are completed
      * @see [LogoutReason]
      */
-    suspend operator fun invoke(reason: LogoutReason)
+    suspend operator fun invoke(reason: LogoutReason, waitUntilCompletes: Boolean = false)
 }
 
 internal class LogoutUseCaseImpl @Suppress("LongParameterList") constructor(
@@ -68,7 +69,7 @@ internal class LogoutUseCaseImpl @Suppress("LongParameterList") constructor(
     //                 Perhaps [UserSessionScope] (or another specialised class) can observe
     //                 the [LogoutRepository.observeLogout] and invalidating everything in [CoreLogic] level.
 
-    override suspend operator fun invoke(reason: LogoutReason) {
+    override suspend operator fun invoke(reason: LogoutReason, waitUntilCompletes: Boolean) {
         globalCoroutineScope.launch {
             deregisterTokenUseCase()
 
@@ -90,19 +91,21 @@ internal class LogoutUseCaseImpl @Suppress("LongParameterList") constructor(
                         wipeTokenAndMetadata()
                     }
                 }
+
                 LogoutReason.SESSION_EXPIRED -> {
                     if (kaliumConfigs.wipeOnCookieInvalid) {
                         wipeAllData()
                     } else {
                         clearCurrentClientIdAndFirebaseTokenFlag()
                     }
                 }
+
                 LogoutReason.SELF_SOFT_LOGOUT -> clearCurrentClientIdAndFirebaseTokenFlag()
             }
 
             userSessionScopeProvider.get(userId)?.cancel()
             userSessionScopeProvider.delete(userId)
-        }
+        }.let { if (waitUntilCompletes) it.join() else it }
     }
 
     private suspend fun clearCurrentClientIdAndFirebaseTokenFlag() {