Tunnel creation: After saving the new tunnel, show the detail view

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

1 files changed, 14 insertions, 2 deletions

diff --git a/WireGuard/WireGuard/UI/iOS/TunnelsListTableViewController.swift b/WireGuard/WireGuard/UI/iOS/TunnelsListTableViewController.swift
index ec8d381..abbf0ac 100644
--- a/WireGuard/WireGuard/UI/iOS/TunnelsListTableViewController.swift
+++ b/WireGuard/WireGuard/UI/iOS/TunnelsListTableViewController.swift
@@ -41,6 +41,7 @@ class TunnelsListTableViewController: UITableViewController {
             UIAlertAction(title: "Create from scratch", style: .default) { [weak self] (action) in
                 if let s = self, let tunnelsManager = s.tunnelsManager {
                     let editVC = TunnelEditTableViewController(tunnelsManager: tunnelsManager)
+                    editVC.delegate = s
                     let editNC = UINavigationController(rootViewController: editVC)
                     s.present(editNC, animated: true)
                 }
@@ -55,6 +56,17 @@ class TunnelsListTableViewController: UITableViewController {
     }
 }
 
+// MARK: TunnelEditTableViewControllerDelegate
+
+extension TunnelsListTableViewController: TunnelEditTableViewControllerDelegate {
+    func saved(tunnel: TunnelContainer) {
+        guard let tunnelsManager = tunnelsManager else { return }
+        let tunnelDetailVC = TunnelDetailTableViewController(tunnelsManager: tunnelsManager,
+                                                             tunnel: tunnel)
+        showDetailViewController(tunnelDetailVC, sender: self) // Shall get propagated up to the split-vc
+    }
+}
+
 // MARK: UITableViewDataSource
 
 extension TunnelsListTableViewController {
@@ -81,9 +93,9 @@ extension TunnelsListTableViewController {
 extension TunnelsListTableViewController {
     override func tableView(_ tableView: UITableView, didSelectRowAt indexPath: IndexPath) {
         guard let tunnelsManager = tunnelsManager else { return }
-        let tunnelConfiguration = tunnelsManager.tunnel(at: indexPath.row).tunnelProvider.tunnelConfiguration
+        let tunnel = tunnelsManager.tunnel(at: indexPath.row)
         let tunnelDetailVC = TunnelDetailTableViewController(tunnelsManager: tunnelsManager,
-                                                             tunnelConfiguration: tunnelConfiguration)
+                                                             tunnel: tunnel)
         showDetailViewController(tunnelDetailVC, sender: self) // Shall get propagated up to the split-vc
     }
 }