88 lines
2.9 KiB
Swift
88 lines
2.9 KiB
Swift
//
|
|
// GeminiConnection.swift
|
|
// Gemini
|
|
//
|
|
// Created by Shadowfacts on 7/12/20.
|
|
//
|
|
|
|
import Foundation
|
|
import Network
|
|
|
|
public protocol GeminiConnectionDelegate: class {
|
|
func connectionReady(_ connection: GeminiConnection)
|
|
func connection(_ connection: GeminiConnection, receivedData data: Data?, header: GeminiResponseHeader)
|
|
func connection(_ connection: GeminiConnection, handleError error: GeminiConnection.Error)
|
|
func connectionCompleted(_ connection: GeminiConnection)
|
|
}
|
|
|
|
public class GeminiConnection {
|
|
|
|
public typealias Error = NWError
|
|
|
|
public weak var delegate: GeminiConnectionDelegate?
|
|
|
|
private var connection: NWConnection?
|
|
|
|
public init(endpoint: NWEndpoint, delegate: GeminiConnectionDelegate? = nil) {
|
|
self.connection = NWConnection(to: endpoint, using: .gemini)
|
|
self.delegate = delegate
|
|
|
|
startConnection()
|
|
}
|
|
|
|
private func startConnection() {
|
|
guard let connection = connection else { return }
|
|
|
|
connection.stateUpdateHandler = { (newState) in
|
|
switch newState {
|
|
case .ready:
|
|
print("\(connection) established")
|
|
self.delegate?.connectionReady(self)
|
|
case let .failed(error):
|
|
print("\(connection) failed: \(error)")
|
|
connection.cancel()
|
|
default:
|
|
break
|
|
}
|
|
}
|
|
|
|
// todo: should this be on a background queue?
|
|
connection.start(queue: .main)
|
|
}
|
|
|
|
var report: NWConnection.PendingDataTransferReport!
|
|
|
|
public func sendRequest(_ request: GeminiRequest) {
|
|
guard let connection = connection else { return }
|
|
|
|
let message = NWProtocolFramer.Message(geminiRequest: request)
|
|
let context = NWConnection.ContentContext(identifier: "GeminiRequest", metadata: [message])
|
|
// todo: should this really always be idempotent?
|
|
connection.send(content: nil, contentContext: context, isComplete: true, completion: .contentProcessed({ (_) in }))
|
|
receiveNextMessage()
|
|
report = connection.startDataTransferReport()
|
|
}
|
|
|
|
private func receiveNextMessage() {
|
|
guard let connection = connection else { return }
|
|
|
|
connection.receiveMessage { (data, context, isComplete, error) in
|
|
if let error = error {
|
|
self.delegate?.connection(self, handleError: error)
|
|
} else if let message = context?.protocolMetadata(definition: GeminiProtocol.definition) as? NWProtocolFramer.Message,
|
|
let header = message.geminiResponseHeader,
|
|
let delegate = self.delegate {
|
|
delegate.connection(self, receivedData: data, header: header)
|
|
}
|
|
|
|
if isComplete {
|
|
self.delegate?.connectionCompleted(self)
|
|
connection.cancel()
|
|
} else {
|
|
self.receiveNextMessage()
|
|
}
|
|
}
|
|
}
|
|
|
|
}
|