From: Roopesh Chander <roop@roopc.net>
Date: Thu, 24 Jan 2019 12:41:55 +0000 (+0530)
Subject: TunnelsTracker: Simplify using TunnelsManager.tunnelInOperation()
X-Git-Tag: 0.0.20190207-1~50
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=52eec55d36fa12abeafa8a416cb14e3043223df8;p=thirdparty%2Fwireguard-apple.git

TunnelsTracker: Simplify using TunnelsManager.tunnelInOperation()

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

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