Model: Endpoint.hasHostAsIPAddress()

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

diff --git a/WireGuard/WireGuard/Model/Endpoint.swift b/WireGuard/WireGuard/Model/Endpoint.swift
index 483615b90639789ef6cf95c795bcc9b57743721d..a46d6b763d61cf667e3bfe7dc0874e02fbc8de5d 100644 (file)
--- a/WireGuard/WireGuard/Model/Endpoint.swift
+++ b/WireGuard/WireGuard/Model/Endpoint.swift
@@ -75,3 +75,16 @@ extension Endpoint: Codable {
         case invalidData
     }
 }
+
+extension Endpoint {
+    func hasHostAsIPAddress() -> Bool {
+        switch (host) {
+        case .name(_, _):
+            return false
+        case .ipv4(_):
+            return true
+        case .ipv6(_):
+            return true
+        }
+    }
+}