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
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
/*
* Copyright (C) 2023 MURENA SAS
*
* 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 <https://www.gnu.org/licenses/>.
*/
package foundation.e.advancedprivacy.features.trackers.trackerdetails
import android.net.Uri
import androidx.annotation.StringRes
import androidx.lifecycle.ViewModel
import androidx.lifecycle.viewModelScope
import foundation.e.advancedprivacy.domain.entities.ApplicationDescription
import foundation.e.advancedprivacy.domain.entities.TrackerMode
import foundation.e.advancedprivacy.domain.usecases.GetQuickPrivacyStateUseCase
import foundation.e.advancedprivacy.domain.usecases.TrackerDetailsUseCase
import foundation.e.advancedprivacy.domain.usecases.TrackersStateUseCase
import foundation.e.advancedprivacy.domain.usecases.TrackersStatisticsUseCase
import foundation.e.advancedprivacy.features.trackers.URL_LEARN_MORE_ABOUT_TRACKERS
import foundation.e.advancedprivacy.trackers.domain.entities.Tracker
import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.flow.MutableSharedFlow
import kotlinx.coroutines.flow.MutableStateFlow
import kotlinx.coroutines.flow.asSharedFlow
import kotlinx.coroutines.flow.asStateFlow
import kotlinx.coroutines.flow.map
import kotlinx.coroutines.flow.merge
import kotlinx.coroutines.flow.update
import kotlinx.coroutines.launch
import kotlinx.coroutines.withContext
class TrackerDetailsViewModel(
private val tracker: Tracker,
private val trackersStateUseCase: TrackersStateUseCase,
private val trackersStatisticsUseCase: TrackersStatisticsUseCase,
private val trackerDetailsUseCase: TrackerDetailsUseCase,
private val getQuickPrivacyStateUseCase: GetQuickPrivacyStateUseCase
) : ViewModel() {
private val _state = MutableStateFlow(TrackerDetailsState(tracker = tracker))
val state = _state.asStateFlow()
private val _singleEvents = MutableSharedFlow<SingleEvent>()
val singleEvents = _singleEvents.asSharedFlow()
suspend fun doOnStartedState() = withContext(Dispatchers.IO) {
merge(
getQuickPrivacyStateUseCase.trackerMode.map {
_state.update { s -> s.copy(isTrackersBlockingEnabled = it != TrackerMode.VULNERABLE) }
},
trackersStatisticsUseCase.listenUpdates().map { fetchStatistics() }
).collect { }
}
fun onToggleUnblockApp(app: ApplicationDescription, isBlocked: Boolean) {
viewModelScope.launch(Dispatchers.IO) {
if (!state.value.isTrackersBlockingEnabled) {
_singleEvents.emit(SingleEvent.ToastTrackersControlDisabled)
}
trackersStateUseCase.blockTracker(app, tracker, isBlocked)
updateWhitelist()
}
}
fun onToggleBlockAll(isBlocked: Boolean) {
viewModelScope.launch(Dispatchers.IO) {
if (!state.value.isTrackersBlockingEnabled) {
_singleEvents.emit(SingleEvent.ToastTrackersControlDisabled)
}
trackerDetailsUseCase.toggleTrackerWhitelist(tracker, isBlocked)
_state.update {
it.copy(
isBlockAllActivated = !trackersStateUseCase.isWhitelisted(tracker)
)
}
updateWhitelist()
}
}
fun onClickLearnMore() {
viewModelScope.launch {
_singleEvents.emit(SingleEvent.OpenUrl(Uri.parse(URL_LEARN_MORE_ABOUT_TRACKERS)))
}
}
private suspend fun fetchStatistics() = withContext(Dispatchers.IO) {
val (blocked, leaked) = trackerDetailsUseCase.getCalls(tracker)
val appsWhitWhiteListState = trackerDetailsUseCase.getAppsWithBlockedState(tracker)
_state.update { s ->
s.copy(
isBlockAllActivated = !trackersStateUseCase.isWhitelisted(tracker),
detectedCount = appsWhitWhiteListState.size,
blockedCount = blocked,
leakedCount = leaked,
appList = appsWhitWhiteListState,
)
}
}
private suspend fun updateWhitelist() {
_state.update { s ->
s.copy(
isBlockAllActivated = !trackersStateUseCase.isWhitelisted(tracker),
appList = trackerDetailsUseCase.enrichWithBlockedState(
s.appList.map { it.first }, tracker
)
)
}
}
sealed class SingleEvent {
data class ErrorEvent(@StringRes val errorResId: Int) : SingleEvent()
object ToastTrackersControlDisabled : SingleEvent()
data class OpenUrl(val url: Uri) : SingleEvent()
}
}
|