From 6a21acca8a294df4bf113d7c11863733ff84336c Mon Sep 17 00:00:00 2001
From: Navan Chauhan <navanchauhan@gmail.com>
Date: Tue, 19 Dec 2023 13:13:46 -0700
Subject: Platform check for linux

---
 Sources/swiftGopherClient/gopherClient.swift | 64 +++++++++++++++++++---------
 1 file changed, 44 insertions(+), 20 deletions(-)

(limited to 'Sources/swiftGopherClient/gopherClient.swift')

diff --git a/Sources/swiftGopherClient/gopherClient.swift b/Sources/swiftGopherClient/gopherClient.swift
index e44bfcf..802d646 100644
--- a/Sources/swiftGopherClient/gopherClient.swift
+++ b/Sources/swiftGopherClient/gopherClient.swift
@@ -20,11 +20,15 @@ public class GopherClient {
   ///
   /// It automatically chooses the appropriate `EventLoopGroup` based on the running platform.
   public init() {
-      if #available(macOS 10.14, iOS 12.0, tvOS 12.0, watchOS 6.0, visionOS 1.0, *) {
-      self.group = NIOTSEventLoopGroup()
-    } else {
+    #if os(Linux)
       self.group = MultiThreadedEventLoopGroup(numberOfThreads: System.coreCount)
-    }
+    #else
+      if #available(macOS 10.14, iOS 12.0, tvOS 12.0, watchOS 6.0, visionOS 1.0, *) {
+        self.group = NIOTSEventLoopGroup()
+      } else {
+        self.group = MultiThreadedEventLoopGroup(numberOfThreads: System.coreCount)
+      }
+    #endif
   }
 
   deinit {
@@ -44,8 +48,8 @@ public class GopherClient {
     to host: String, port: Int = 70, message: String,
     completion: @escaping (Result<[gopherItem], Error>) -> Void
   ) {
-      if #available(macOS 10.14, iOS 12.0, tvOS 12.0, watchOS 6.0, visionOS 1.0, *) {
-      let bootstrap = NIOTSConnectionBootstrap(group: group)
+    #if os(Linux)
+      let bootstrap = ClientBootstrap(group: group)
         .channelOption(ChannelOptions.socketOption(.so_reuseaddr), value: 1)
         .channelInitializer { channel in
           channel.pipeline.addHandler(
@@ -61,24 +65,44 @@ public class GopherClient {
           completion(.failure(error))
         }
       }
-    } else {
-      let bootstrap = ClientBootstrap(group: group)
-        .channelOption(ChannelOptions.socketOption(.so_reuseaddr), value: 1)
-        .channelInitializer { channel in
-          channel.pipeline.addHandler(
-            GopherRequestResponseHandler(message: message, completion: completion))
+    #else
+
+      if #available(macOS 10.14, iOS 12.0, tvOS 12.0, watchOS 6.0, visionOS 1.0, *) {
+        let bootstrap = NIOTSConnectionBootstrap(group: group)
+          .channelOption(ChannelOptions.socketOption(.so_reuseaddr), value: 1)
+          .channelInitializer { channel in
+            channel.pipeline.addHandler(
+              GopherRequestResponseHandler(message: message, completion: completion))
+          }
+        bootstrap.connect(host: host, port: port).whenComplete { result in
+          switch result {
+          case .success(let channel):
+            channel.closeFuture.whenComplete { _ in
+              print("Connection closed")
+            }
+          case .failure(let error):
+            completion(.failure(error))
+          }
         }
-      bootstrap.connect(host: host, port: port).whenComplete { result in
-        switch result {
-        case .success(let channel):
-          channel.closeFuture.whenComplete { _ in
-            print("Connection closed")
+      } else {
+        let bootstrap = ClientBootstrap(group: group)
+          .channelOption(ChannelOptions.socketOption(.so_reuseaddr), value: 1)
+          .channelInitializer { channel in
+            channel.pipeline.addHandler(
+              GopherRequestResponseHandler(message: message, completion: completion))
+          }
+        bootstrap.connect(host: host, port: port).whenComplete { result in
+          switch result {
+          case .success(let channel):
+            channel.closeFuture.whenComplete { _ in
+              print("Connection closed")
+            }
+          case .failure(let error):
+            completion(.failure(error))
           }
-        case .failure(let error):
-          completion(.failure(error))
         }
       }
-    }
+    #endif
 
   }
 
-- 
cgit v1.2.3