Model: Endpoint.hostname()

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

diff --git a/WireGuard/Shared/Model/Endpoint.swift b/WireGuard/Shared/Model/Endpoint.swift
index a46d6b763d61cf667e3bfe7dc0874e02fbc8de5d..ab357ba219ca81023134d85b5a64a66a84fac9c6 100644 (file)
--- a/WireGuard/Shared/Model/Endpoint.swift
+++ b/WireGuard/Shared/Model/Endpoint.swift
@@ -87,4 +87,15 @@ extension Endpoint {
             return true
         }
     }
+
+    func hostname() -> String? {
+        switch (host) {
+        case .name(let hostname, _):
+            return hostname
+        case .ipv4(_):
+            return nil
+        case .ipv6(_):
+            return nil
+        }
+    }
 }