adding section header view in select location

- each section should have its own distinct header view.
- custom list section has an action that should take user to add/edit custom list.it will be coming during upcoming changes.
- refactoring location cell
- removing LocationCellFactory

1 files changed, 11 insertions, 4 deletions

diff --git a/ios/MullvadVPN/View controllers/SelectLocation/LocationViewController.swift b/ios/MullvadVPN/View controllers/SelectLocation/LocationViewController.swift
index 3c37884544..baa3cce181 100644
--- a/ios/MullvadVPN/View controllers/SelectLocation/LocationViewController.swift
+++ b/ios/MullvadVPN/View controllers/SelectLocation/LocationViewController.swift
@@ -83,7 +83,7 @@ final class LocationViewController: UIViewController {
             })
         )
 
-        setUpDataSource()
+        setUpDataSources()
         setUpTableView()
         setUpTopContent()
 
@@ -119,17 +119,24 @@ final class LocationViewController: UIViewController {
 
     // MARK: - Private
 
-    private func setUpDataSource() {
+    private func setUpDataSources() {
+        let allLocationDataSource = AllLocationDataSource()
+        let customListsDataSource = CustomListsDataSource(repository: customListRepository)
         dataSource = LocationDataSource(
             tableView: tableView,
-            allLocations: AllLocationDataSource(),
-            customLists: CustomListsDataSource(repository: customListRepository)
+            allLocations: allLocationDataSource,
+            customLists: customListsDataSource
         )
 
         dataSource?.didSelectRelayLocations = { [weak self] locations in
             self?.didSelectRelays?(locations)
         }
 
+        dataSource?.didTapEditCustomLists = { [weak self] in
+            guard let self else { return }
+            delegate?.didRequestRouteToCustomLists(self)
+        }
+
         if let cachedRelays {
             dataSource?.setRelays(cachedRelays.relays, selectedLocations: relayLocations, filter: filter)
         }