@@ -48,11 +48,15 @@ class SelectLocationViewModel(
48
48
selectedConstraintProviders ->
49
49
val selectedOwnershipItem = selectedOwnership.toNullableOwnership()
50
50
val selectedProvidersCount =
51
- filterSelectedProvidersByOwnership(
52
- selectedConstraintProviders.toSelectedProviders(allProviders),
53
- selectedOwnershipItem
54
- )
55
- ?.size
51
+ when (selectedConstraintProviders) {
52
+ is Constraint .Any -> null
53
+ is Constraint .Only ->
54
+ filterSelectedProvidersByOwnership(
55
+ selectedConstraintProviders.toSelectedProviders(allProviders),
56
+ selectedOwnershipItem
57
+ )
58
+ .size
59
+ }
56
60
57
61
val filteredRelayCountries =
58
62
relayCountries.filterOnSearchTerm(searchTerm, selectedItem)
@@ -97,15 +101,13 @@ class SelectLocationViewModel(
97
101
}
98
102
99
103
private fun filterSelectedProvidersByOwnership (
100
- selectedProviders : List <Provider >? ,
104
+ selectedProviders : List <Provider >,
101
105
selectedOwnership : Ownership ?
102
- ): List <Provider >? =
103
- selectedProviders?.let {
104
- when (selectedOwnership) {
105
- Ownership .MullvadOwned -> selectedProviders.filter { it.mullvadOwned }
106
- Ownership .Rented -> selectedProviders.filterNot { it.mullvadOwned }
107
- else -> selectedProviders
108
- }
106
+ ): List <Provider > =
107
+ when (selectedOwnership) {
108
+ Ownership .MullvadOwned -> selectedProviders.filter { it.mullvadOwned }
109
+ Ownership .Rented -> selectedProviders.filterNot { it.mullvadOwned }
110
+ else -> selectedProviders
109
111
}
110
112
111
113
fun removeOwnerFilter () {
0 commit comments