Add string conversion for tunnel config.

Signed-off-by: Jason A. Donenfeld <Jason@zx2c4.com>

3 files changed, 52 insertions, 0 deletions

diff --git a/WireGuard/Models/Interface+Extension.swift b/WireGuard/Models/Interface+Extension.swift
index 060b4b7..3b6ef62 100644
--- a/WireGuard/Models/Interface+Extension.swift
+++ b/WireGuard/Models/Interface+Extension.swift
@@ -57,6 +57,20 @@ extension Interface {
         }
     }
 
+    func export() -> String {
+        var exportString = "[Interface]\n"
+        if let privateKey = privateKey {
+            exportString.append("PrivateKey=\(privateKey)")
+        }
+        if listenPort > 0 {
+            exportString.append("ListenPort=\(listenPort)")
+        }
+
+        exportString.append("\n")
+
+        return exportString
+    }
+
 }
 
 enum InterfaceValidationError: Error {
diff --git a/WireGuard/Models/Peer+Extension.swift b/WireGuard/Models/Peer+Extension.swift
index f83c038..3ffd941 100644
--- a/WireGuard/Models/Peer+Extension.swift
+++ b/WireGuard/Models/Peer+Extension.swift
@@ -63,6 +63,29 @@ extension Peer {
         }
     }
 
+    func export() -> String {
+        var exportString = "[Peer]\n"
+        if let publicKey = publicKey {
+            exportString.append("PublicKey=\(publicKey)\n")
+        }
+        if let presharedKey = presharedKey {
+            exportString.append("PresharedKey=\(presharedKey)\n")
+        }
+        if let allowedIPs = allowedIPs {
+            exportString.append("AllowedIPs=\(allowedIPs)\n")
+        }
+        if let endpoint = endpoint {
+            exportString.append("Endpoint=\(endpoint)\n")
+        }
+        if persistentKeepalive > 0 {
+            exportString.append("PersistentKeepalive=\(persistentKeepalive)\n")
+        }
+
+        exportString.append("\n")
+
+        return exportString
+    }
+
 }
 
 enum PeerValidationError: Error {
diff --git a/WireGuard/Models/Tunnel+Extension.swift b/WireGuard/Models/Tunnel+Extension.swift
index 62014f1..0cced3c 100644
--- a/WireGuard/Models/Tunnel+Extension.swift
+++ b/WireGuard/Models/Tunnel+Extension.swift
@@ -159,6 +159,21 @@ extension Tunnel {
         return tunnel
     }
 
+    func export() -> String {
+        var exportString = ""
+        if let interfaceExport = self.interface?.export() {
+            exportString.append(interfaceExport)
+        }
+
+        if let peers = peers?.array as? [Peer] {
+            peers.forEach {
+                exportString.append($0.export())
+            }
+        }
+
+        return exportString
+    }
+
 }
 
 private func base64KeyToHex(_ base64: String?) -> String? {