TunnelsTracker: Simplify using TunnelsManager.tunnelInOperation()

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

1 files changed, 2 insertions, 17 deletions

diff --git a/WireGuard/WireGuard/UI/macOS/TunnelsTracker.swift b/WireGuard/WireGuard/UI/macOS/TunnelsTracker.swift
index 8df7789..69cc533 100644
--- a/WireGuard/WireGuard/UI/macOS/TunnelsTracker.swift
+++ b/WireGuard/WireGuard/UI/macOS/TunnelsTracker.swift
@@ -33,18 +33,7 @@ class TunnelsTracker {
 
     init(tunnelsManager: TunnelsManager) {
         self.tunnelsManager = tunnelsManager
-
-        if let waitingTunnel = tunnelsManager.waitingTunnel() {
-            currentTunnel = waitingTunnel
-        } else {
-            for index in 0 ..< tunnelsManager.numberOfTunnels() {
-                let tunnel = tunnelsManager.tunnel(at: index)
-                if tunnel.status != .inactive {
-                    currentTunnel = tunnel
-                    break
-                }
-            }
-        }
+        currentTunnel = tunnelsManager.tunnelInOperation()
 
         for index in 0 ..< tunnelsManager.numberOfTunnels() {
             let tunnel = tunnelsManager.tunnel(at: index)
@@ -61,11 +50,7 @@ class TunnelsTracker {
             guard let self = self else { return }
             if tunnel.status == .deactivating || tunnel.status == .inactive {
                 if self.currentTunnel == tunnel {
-                    if let waitingTunnel = self.tunnelsManager.waitingTunnel() {
-                        self.currentTunnel = waitingTunnel
-                    } else if tunnel.status == .inactive {
-                        self.currentTunnel = nil
-                    }
+                    self.currentTunnel = self.tunnelsManager.tunnelInOperation()
                 }
             } else {
                 self.currentTunnel = tunnel