aboutsummaryrefslogtreecommitdiff
path: root/Sources/swiftGopherClient/gopherRequestResponseHandler.swift
blob: 8349c6e660ee31049a698b784b75a7761881714d (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
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
//
//  gopherRequestResponseHandler.swift
//
//
//  Created by Navan Chauhan on 12/12/23.
//

import Foundation
import GopherHelpers
import NIO

final class GopherRequestResponseHandler: ChannelInboundHandler {
  typealias InboundIn = ByteBuffer
  typealias OutboundOut = ByteBuffer

  private var accumulatedData: ByteBuffer
  private let message: String
  private let completion: (Result<[gopherItem], Error>) -> Void

  init(message: String, completion: @escaping (Result<[gopherItem], Error>) -> Void) {
    self.message = message
    self.completion = completion
    self.accumulatedData = ByteBuffer()
  }

  func channelActive(context: ChannelHandlerContext) {
    var buffer = context.channel.allocator.buffer(capacity: message.utf8.count)
    buffer.writeString(message)
    context.writeAndFlush(self.wrapOutboundOut(buffer), promise: nil)
  }

  func channelRead(context: ChannelHandlerContext, data: NIOAny) {
    var buffer = unwrapInboundIn(data)
    accumulatedData.writeBuffer(&buffer)
  }

  func channelInactive(context: ChannelHandlerContext) {
    if let dataCopy = accumulatedData.getSlice(at: 0, length: accumulatedData.readableBytes) {
      parseGopherServerResponse(
        response: accumulatedData.readString(length: accumulatedData.readableBytes) ?? "",
        originalBytes: dataCopy)
    }
  }

  func errorCaught(context: ChannelHandlerContext, error: Error) {
    print("Error: ", error)
    context.close(promise: nil)
  }

  func createGopherItem(rawLine: String, itemType: gopherItemType = .info, rawData: ByteBuffer)
    -> gopherItem
  {
    var item = gopherItem(rawLine: rawLine)
    item.parsedItemType = itemType
    item.rawData = rawData

    if rawLine.isEmpty {
      item.valid = false
    } else {
      let components = rawLine.components(separatedBy: "\t")

      // Handle cases where rawLine does not have any itemType in the first character
      item.message = String(components[0].dropFirst())

      if components.indices.contains(1) {
        item.selector = String(components[1])
      }

      if components.indices.contains(2) {
        item.host = String(components[2])
      }

      if components.indices.contains(3) {
        item.port = Int(String(components[3])) ?? 70
      }
    }

    return item
  }

  func parseGopherServerResponse(response: String, originalBytes: ByteBuffer) {
    var gopherServerResponse: [gopherItem] = []

    print("parsing")
    let carriageReturnCount = response.filter({ $0 == "\r" }).count
    let newlineCarriageReturnCount = response.filter({ $0 == "\r\n" }).count
    print(
      "Carriage Returns: \(carriageReturnCount), Newline + Carriage Returns: \(newlineCarriageReturnCount)"
    )

    if carriageReturnCount == 0 {
      for line in response.split(separator: "\n") {
        let lineItemType = getGopherFileType(item: "\(line.first ?? " ")")
        let item = createGopherItem(
          rawLine: String(line), itemType: lineItemType, rawData: originalBytes)
        gopherServerResponse.append(item)

      }
    } else {
      for line in response.split(separator: "\r\n") {
        let lineItemType = getGopherFileType(item: "\(line.first ?? " ")")
        let item = createGopherItem(
          rawLine: String(line), itemType: lineItemType, rawData: originalBytes)
        gopherServerResponse.append(item)

      }
    }

    print("done parsing")

    completion(.success(gopherServerResponse))
  }
}