From 2677efc9bf3fa8cb74c5dcdf4ed87ff86eaf4f0c Mon Sep 17 00:00:00 2001
From: Roopesh Chander <roop@roopc.net>
Date: Tue, 22 Jan 2019 19:23:05 +0530
Subject: macOS: Tunnel detail: Handle deletion outside app

Signed-off-by: Roopesh Chander <roop@roopc.net>
---
 .../UI/macOS/ViewController/TunnelsListTableViewController.swift | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

(limited to 'WireGuard')

diff --git a/WireGuard/WireGuard/UI/macOS/ViewController/TunnelsListTableViewController.swift b/WireGuard/WireGuard/UI/macOS/ViewController/TunnelsListTableViewController.swift
index 539b487..5ae89c4 100644
--- a/WireGuard/WireGuard/UI/macOS/ViewController/TunnelsListTableViewController.swift
+++ b/WireGuard/WireGuard/UI/macOS/ViewController/TunnelsListTableViewController.swift
@@ -155,10 +155,6 @@ class TunnelsListTableViewController: NSViewController {
                     ErrorPresenter.showErrorAlert(error: error, from: self)
                     return
                 }
-                let tunnelIndex = min(selectedTunnelIndex, self.tunnelsManager.numberOfTunnels() - 1)
-                if tunnelIndex >= 0 {
-                    self.selectTunnel(at: tunnelIndex)
-                }
             }
         }
     }
@@ -249,10 +245,15 @@ extension TunnelsListTableViewController {
     }
 
     func tunnelRemoved(at index: Int) {
+        let selectedTunnelIndex = tableView.selectedRow
         tableView.removeRows(at: IndexSet(integer: index), withAnimation: .slideLeft)
         if tunnelsManager.numberOfTunnels() == 0 {
             delegate?.tunnelsListEmpty()
         }
+        let tunnelIndex = min(selectedTunnelIndex, self.tunnelsManager.numberOfTunnels() - 1)
+        if tunnelIndex >= 0 {
+            self.selectTunnel(at: tunnelIndex)
+        }
     }
 }
 
-- 
cgit v1.2.3-59-g8ed1b