Tunnels list: Update when a tunnel is added

Signed-off-by: Roopesh Chander <roop@roopc.net>

1 files changed, 13 insertions, 2 deletions

diff --git a/WireGuard/WireGuard/UI/iOS/TunnelsListTableViewController.swift b/WireGuard/WireGuard/UI/iOS/TunnelsListTableViewController.swift
index f13a900..2a57393 100644
--- a/WireGuard/WireGuard/UI/iOS/TunnelsListTableViewController.swift
+++ b/WireGuard/WireGuard/UI/iOS/TunnelsListTableViewController.swift
@@ -30,8 +30,11 @@ class TunnelsListTableViewController: UITableViewController {
 
         TunnelsManager.create { [weak self] tunnelsManager in
             guard let tunnelsManager = tunnelsManager else { return }
-            self?.tunnelsManager = tunnelsManager
-            self?.tableView.reloadData()
+            if let s = self {
+                tunnelsManager.delegate = s
+                s.tunnelsManager = tunnelsManager
+                s.tableView.reloadData()
+            }
         }
     }
 
@@ -78,6 +81,14 @@ extension TunnelsListTableViewController {
     }
 }
 
+// MARK: TunnelsManagerDelegate
+
+extension TunnelsListTableViewController: TunnelsManagerDelegate {
+    func tunnelsAdded(atIndex index: Int, numberOfTunnels: Int) {
+        self.tableView.insertRows(at: [IndexPath(row: index, section: 0)], with: .automatic)
+    }
+}
+
 class TunnelsListTableViewCell: UITableViewCell {
     static let id: String = "TunnelsListTableViewCell"
     var tunnelName: String {