From 4404bb2b7d61f01d6d01c45fa6bb2ec20673ed02 Mon Sep 17 00:00:00 2001
From: Roopesh Chander <roop@roopc.net>
Date: Thu, 8 Nov 2018 16:26:17 +0530
Subject: Model: Endpoint.hostname()

Signed-off-by: Roopesh Chander <roop@roopc.net>
---
 WireGuard/Shared/Model/Endpoint.swift | 11 +++++++++++
 1 file changed, 11 insertions(+)

(limited to 'WireGuard/Shared/Model')

diff --git a/WireGuard/Shared/Model/Endpoint.swift b/WireGuard/Shared/Model/Endpoint.swift
index a46d6b7..ab357ba 100644
--- 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
+        }
+    }
 }
-- 
cgit v1.2.3-59-g8ed1b