1 files changed, 18 insertions, 1 deletions
diff --git a/Hotline/Library/NetSocket/NetSocket.swift b/Hotline/Library/NetSocket/NetSocket.swift
index 5c7d185..e66ef3f 100644
--- a/Hotline/Library/NetSocket/NetSocket.swift
+++ b/Hotline/Library/NetSocket/NetSocket.swift
@@ -173,7 +173,24 @@ public actor NetSocket {
     guard let nwPort = NWEndpoint.Port(rawValue: port) else {
       throw NetSocketError.invalidPort
     }
-    return try await self.connect(host: .name(host, nil), port: nwPort, tls: tls, config: config)
+
+    // Parse the host string to create the appropriate NWEndpoint.Host
+    let nwHost: NWEndpoint.Host
+
+    // Try parsing as IPv6 without zone
+    if let ipv6Addr = IPv6Address(host) {
+      nwHost = .ipv6(ipv6Addr)
+    }
+    // Try parsing as IPv4
+    else if let ipv4Addr = IPv4Address(host) {
+      nwHost = .ipv4(ipv4Addr)
+    }
+    // Fall back to treating as hostname
+    else {
+      nwHost = .name(host, nil)
+    }
+
+    return try await self.connect(host: nwHost, port: nwPort, tls: tls, config: config)
   }
   
   // MARK: Close