aboutsummaryrefslogtreecommitdiff
path: root/Sources/swiftGopherClient/gopherClient.swift
blob: c7542eebae6e4089d9d1b10d61f858c09bca4c87 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
//
//  gopherClient.swift
//
//
//  Created by Navan Chauhan on 12/12/23.
//

import Foundation
import GopherHelpers
import NIO
import NIOTransportServices

/// `GopherClient` is a class for handling network connections and requests to Gopher servers.
///
/// It utilizes `NIOTSEventLoopGroup` on iOS/macOS (Not sure why you would run this on watchOS/tvOS but it supports that as well) for network operations, falling back to `MultiThreadedEventLoopGroup` otherwise.
public class GopherClient {
  private let group: EventLoopGroup

  /// Initializes a new instance of `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, *) {
      self.group = NIOTSEventLoopGroup()
    } else {
      self.group = MultiThreadedEventLoopGroup(numberOfThreads: System.coreCount)
    }
  }

  deinit {
    try? group.syncShutdownGracefully()
  }

  /// Sends a request to a Gopher server.
  ///
  /// - Parameters:
  ///   - host: The host address of the Gopher server.
  ///   - port: The port of the Gopher server. Defaults to 70.
  ///   - message: The message to be sent to the server.
  ///   - completion: A closure that handles the result of the request.
  ///
  /// The method asynchronously establishes a connection, sends the request, and calls the completion handler with the result.
  public func sendRequest(
    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, *) {
      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))
        }
      }
    } 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))
        }
      }
    }

  }

  /// Shuts down the event loop group, releasing any resources.
  ///
  /// This method is called during deinitialization to ensure clean shutdown of network resources.
  private func shutdownEventLoopGroup() {
    do {
      try group.syncShutdownGracefully()
    } catch {
      print("Error shutting down event loop group: \(error)")
    }
  }
}