From 59b9a6e5d297a15459f766272608fee37921a10b Mon Sep 17 00:00:00 2001
From: Roopesh Chander <roop@roopc.net>
Date: Fri, 9 Nov 2018 19:19:32 +0530
Subject: [PATCH] TunnelsManager: Ability to refresh connection statuses

Signed-off-by: Roopesh Chander <roop@roopc.net>
---
 WireGuard/WireGuard/VPN/TunnelsManager.swift | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/WireGuard/WireGuard/VPN/TunnelsManager.swift b/WireGuard/WireGuard/VPN/TunnelsManager.swift
index ba4ef7f..682aa79 100644
--- a/WireGuard/WireGuard/VPN/TunnelsManager.swift
+++ b/WireGuard/WireGuard/VPN/TunnelsManager.swift
@@ -208,6 +208,12 @@ class TunnelsManager {
         }
         tunnel.startDeactivation()
     }
+
+    func refreshConnectionStatuses() {
+        for t in tunnels {
+            t.refreshConnectionStatus()
+        }
+    }
 }
 
 class TunnelContainer: NSObject {
@@ -232,6 +238,14 @@ class TunnelContainer: NSObject {
         return (tunnelProvider.protocolConfiguration as! NETunnelProviderProtocol).tunnelConfiguration()
     }
 
+    func refreshConnectionStatus() {
+        let status = TunnelStatus(from: self.tunnelProvider.connection.status)
+        self.status = status
+        if (status != .inactive) {
+            startObservingTunnelStatus()
+        }
+    }
+
     fileprivate func startActivation(completionHandler: @escaping (Error?) -> Void) {
         assert(status == .inactive || status == .restarting)