ShortcutsDataSource: add table view section header

1 files changed, 35 insertions, 0 deletions

diff --git a/ios/MullvadVPN/ShortcutsDataSource.swift b/ios/MullvadVPN/ShortcutsDataSource.swift
index d021e6f458..1719af12c1 100644
--- a/ios/MullvadVPN/ShortcutsDataSource.swift
+++ b/ios/MullvadVPN/ShortcutsDataSource.swift
@@ -21,6 +21,17 @@ final class ShortcutsDataSource: NSObject, UITableViewDataSource, UITableViewDel
         }
     }
 
+    private enum HeaderFooterReuseIdentifier: String, CaseIterable {
+        case spacer
+
+        var reusableViewClass: AnyClass {
+            switch self {
+            case .spacer:
+                return EmptyTableViewHeaderFooterView.self
+            }
+        }
+    }
+
     enum Section: String {
         case shortcuts
     }
@@ -42,6 +53,12 @@ final class ShortcutsDataSource: NSObject, UITableViewDataSource, UITableViewDel
                 forCellReuseIdentifier: cellIdentifier.rawValue
             )
         }
+        HeaderFooterReuseIdentifier.allCases.forEach { reuseIdentifier in
+            tableView.register(
+                reuseIdentifier.reusableViewClass,
+                forHeaderFooterViewReuseIdentifier: reuseIdentifier.rawValue
+            )
+        }
         tableView.dataSource = self
         tableView.delegate = self
     }
@@ -91,6 +108,24 @@ final class ShortcutsDataSource: NSObject, UITableViewDataSource, UITableViewDel
         delegate?.shortcutsDataSource(self, didSelectItem: item)
         tableView.deselectRow(at: indexPath, animated: true)
     }
+
+    func tableView(_ tableView: UITableView, viewForHeaderInSection section: Int) -> UIView? {
+        return tableView.dequeueReusableHeaderFooterView(
+            withIdentifier: HeaderFooterReuseIdentifier.spacer.rawValue
+        )
+    }
+
+    func tableView(_ tableView: UITableView, viewForFooterInSection section: Int) -> UIView? {
+        return nil
+    }
+
+    func tableView(_ tableView: UITableView, heightForHeaderInSection section: Int) -> CGFloat {
+        return UIMetrics.sectionSpacing
+    }
+
+    func tableView(_ tableView: UITableView, heightForFooterInSection section: Int) -> CGFloat {
+        return 0
+    }
 }
 
 extension ShortcutsDataSource.Item {