diff options
| author | David Göransson <david.goransson@mullvad.net> | 2024-07-25 16:15:26 +0200 |
|---|---|---|
| committer | David Göransson <david.goransson@mullvad.net> | 2024-07-25 16:40:18 +0200 |
| commit | 5586bcc894d149187b4677244007b39ac8692199 (patch) | |
| tree | 495afd930cc5b515b51bac04db4e2b02e471e1b0 /android/app/src | |
| parent | 0da1615d60ffb231fdcb9574a2ea6ba96521d1e9 (diff) | |
| download | mullvadvpn-5586bcc894d149187b4677244007b39ac8692199.tar.xz mullvadvpn-5586bcc894d149187b4677244007b39ac8692199.zip | |
Fix initial expands
Diffstat (limited to 'android/app/src')
| -rw-r--r-- | android/app/src/main/kotlin/net/mullvad/mullvadvpn/viewmodel/CustomListLocationsViewModel.kt | 13 |
1 files changed, 7 insertions, 6 deletions
diff --git a/android/app/src/main/kotlin/net/mullvad/mullvadvpn/viewmodel/CustomListLocationsViewModel.kt b/android/app/src/main/kotlin/net/mullvad/mullvadvpn/viewmodel/CustomListLocationsViewModel.kt index 3df2b2f623..7fa101a8e3 100644 --- a/android/app/src/main/kotlin/net/mullvad/mullvadvpn/viewmodel/CustomListLocationsViewModel.kt +++ b/android/app/src/main/kotlin/net/mullvad/mullvadvpn/viewmodel/CustomListLocationsViewModel.kt @@ -97,8 +97,9 @@ class CustomListLocationsViewModel( val (exp, filteredRelayCountries) = relayCountries.newFilterOnSearch(searchTerm) exp.toSet() to filteredRelayCountries } else { - initialExpands((_selectedLocations.value ?: emptyList()).toSet()) to - relayCountries + initialExpands( + _selectedLocations.value?.calculateLocationsToSave() ?: emptyList() + ) to relayCountries } } .onEach { _expandedItems.value = it.first } @@ -223,16 +224,16 @@ class CustomListLocationsViewModel( } private suspend fun fetchInitialSelectedLocations() { - val selectedLocations = - customListRelayItemsUseCase(navArgs.customListId).first().withDescendants().toSet() + val locations = customListRelayItemsUseCase(navArgs.customListId).first() + val selectedLocations = locations.withDescendants().toSet() _initialLocations.value = selectedLocations _selectedLocations.value = selectedLocations // Initial expand - _expandedItems.value = initialExpands(selectedLocations) + _expandedItems.value = initialExpands(locations) } - private fun initialExpands(locations: Set<RelayItem.Location>): Set<RelayItemId> = + private fun initialExpands(locations: List<RelayItem.Location>): Set<RelayItemId> = locations.flatMap { it.id.ancestors() }.toSet() private fun List<RelayItem.Location>.toRelayItems( |