Stop filtering edit view of custom lists

mullvad · May 2, 2024 · a5980da · a5980da
1 parent 0eb0832
commit a5980da
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/ios/MullvadVPN/View controllers/SelectLocation/LocationViewController.swift b/ios/MullvadVPN/View controllers/SelectLocation/LocationViewController.swift
@@ -130,13 +130,10 @@ final class LocationViewController: UIViewController {
     // MARK: - Private
 
     private func setUpDataSources() {
-        let allLocationDataSource = AllLocationDataSource()
-        let customListsDataSource = CustomListsDataSource(repository: customListRepository)
-
         dataSource = LocationDataSource(
             tableView: tableView,
-            allLocations: allLocationDataSource,
-            customLists: customListsDataSource
+            allLocations: AllLocationDataSource(),
+            customLists: CustomListsDataSource(repository: customListRepository)
         )
 
         dataSource?.didSelectRelayLocations = { [weak self] locations in
@@ -145,7 +142,12 @@ final class LocationViewController: UIViewController {
 
         dataSource?.didTapEditCustomLists = { [weak self] in
             guard let self else { return }
-            delegate?.didRequestRouteToCustomLists(self, nodes: allLocationDataSource.nodes)
+
+            if let cachedRelays {
+                let allLocationDataSource = AllLocationDataSource()
+                allLocationDataSource.reload(cachedRelays.relays, relays: cachedRelays.relays.wireguard.relays)
+                delegate?.didRequestRouteToCustomLists(self, nodes: allLocationDataSource.nodes)
+            }
         }
 
         if let cachedRelays {