-
Notifications
You must be signed in to change notification settings - Fork 19
/
GetE2EICertificateViewModel.kt
95 lines (86 loc) 路 3.89 KB
/
GetE2EICertificateViewModel.kt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
/*
* Wire
* Copyright (C) 2024 Wire Swiss GmbH
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see http://www.gnu.org/licenses/.
*/
package com.wire.android.ui.e2eiEnrollment
import androidx.lifecycle.ViewModel
import com.wire.android.di.KaliumCoreLogic
import com.wire.android.feature.e2ei.OAuthUseCase
import com.wire.android.util.dispatchers.DispatcherProvider
import com.wire.kalium.logic.CoreFailure
import com.wire.kalium.logic.CoreLogic
import com.wire.kalium.logic.E2EIFailure
import com.wire.kalium.logic.feature.e2ei.usecase.E2EIEnrollmentResult
import com.wire.kalium.logic.feature.session.CurrentSessionResult
import com.wire.kalium.logic.feature.session.CurrentSessionUseCase
import com.wire.kalium.logic.functional.Either
import com.wire.kalium.logic.functional.fold
import dagger.hilt.android.lifecycle.HiltViewModel
import kotlinx.coroutines.CoroutineScope
import kotlinx.coroutines.SupervisorJob
import kotlinx.coroutines.flow.MutableSharedFlow
import kotlinx.coroutines.launch
import javax.inject.Inject
@HiltViewModel
class GetE2EICertificateViewModel @Inject constructor(
@KaliumCoreLogic private val coreLogic: CoreLogic,
private val currentSession: CurrentSessionUseCase,
val dispatcherProvider: DispatcherProvider
) : ViewModel() {
private val scope = CoroutineScope(SupervisorJob() + dispatcherProvider.default())
val requestOAuthFlow = MutableSharedFlow<E2EIEnrollmentResult.Initialized>()
val enrollmentResultFlow = MutableSharedFlow<Either<CoreFailure, E2EIEnrollmentResult>>()
fun handleOAuthResult(oAuthResult: OAuthUseCase.OAuthResult, initialEnrollmentResult: E2EIEnrollmentResult.Initialized) {
scope.launch {
when (oAuthResult) {
is OAuthUseCase.OAuthResult.Success -> finalizeEnrollment(oAuthResult, initialEnrollmentResult)
is OAuthUseCase.OAuthResult.Failed -> enrollmentResultFlow.emit(Either.Left(E2EIFailure.OAuth(oAuthResult.reason)))
}
}
}
fun getCertificate(isNewClient: Boolean) {
scope.launch {
val currentSessionResult = currentSession()
if (currentSessionResult is CurrentSessionResult.Success && currentSessionResult.accountInfo.isValid()) {
coreLogic.getSessionScope(currentSessionResult.accountInfo.userId)
.users
.enrollE2EI
.initialEnrollment(isNewClientRegistration = isNewClient)
.fold({
enrollmentResultFlow.emit(Either.Left(it))
}, {
requestOAuthFlow.emit(it)
})
}
}
}
private suspend fun finalizeEnrollment(
oAuthResult: OAuthUseCase.OAuthResult.Success,
initialEnrollmentResult: E2EIEnrollmentResult.Initialized
) {
val currentSessionResult = currentSession()
if (currentSessionResult is CurrentSessionResult.Success && currentSessionResult.accountInfo.isValid()) {
val enrollmentResult = coreLogic.getSessionScope(currentSessionResult.accountInfo.userId)
.users
.enrollE2EI.finalizeEnrollment(
oAuthResult.idToken,
oAuthResult.authState,
initialEnrollmentResult
)
enrollmentResultFlow.emit(enrollmentResult)
}
}
}