From 45829daa856b376b1ab04d415917110b71b1dec5 Mon Sep 17 00:00:00 2001
From: Ruben Beltran del Rio <git@r.bdr.sh>
Date: Wed, 5 Feb 2025 22:27:17 +0100
Subject: Apply formatting

---
 Hotline/Shared/NetSocket.swift | 155 +++++++++++++++++++++--------------------
 1 file changed, 79 insertions(+), 76 deletions(-)

(limited to 'Hotline/Shared/NetSocket.swift')

diff --git a/Hotline/Shared/NetSocket.swift b/Hotline/Shared/NetSocket.swift
index 263487b..59c26af 100644
--- a/Hotline/Shared/NetSocket.swift
+++ b/Hotline/Shared/NetSocket.swift
@@ -1,4 +1,3 @@
-
 // NetSocket.swift
 // A simple delegate based buffered read/write TCP socket.
 // Created by Dustin Mierau
@@ -27,56 +26,57 @@ enum NetSocketStatus {
 
 final class NetSocket: NSObject, StreamDelegate {
   weak var delegate: NetSocketDelegate? = nil
-  
+
   private var output: OutputStream? = nil
   private var input: InputStream? = nil
-  
+
   private var outputBuffer: [UInt8] = []
   private var inputBuffer: [UInt8] = []
-  
+
   private var readBuffer: [UInt8] = Array(repeating: 0, count: 4 * 1024)
-  
+
   public func peek() -> [UInt8] { self.inputBuffer }
   public var available: Int { self.inputBuffer.count }
-  
+
   private var status: NetSocketStatus = .disconnected
-  
+
   @MainActor public func has(_ length: Int) -> Bool {
     return (self.available >= length)
   }
-  
+
   override init() {}
-  
+
   @MainActor public func connect(host: String, port: Int) {
     self.close()
-    
+
     var outputStream: OutputStream? = nil
     var inputStream: InputStream? = nil
-    
+
     self.status = .connecting
-    
-    Stream.getStreamsToHost(withName: host, port: port, inputStream: &inputStream, outputStream: &outputStream)
-    
+
+    Stream.getStreamsToHost(
+      withName: host, port: port, inputStream: &inputStream, outputStream: &outputStream)
+
     self.input = inputStream
     self.output = outputStream
-    
+
     inputStream?.delegate = self
     outputStream?.delegate = self
-    
+
     inputStream?.schedule(in: .current, forMode: .default)
     outputStream?.schedule(in: .current, forMode: .default)
-    
+
     inputStream?.open()
     outputStream?.open()
   }
-  
+
   @MainActor public func close(_ err: Error? = nil) {
     print("NetSocket: Closed")
-    
+
     let disconnected = (self.status != .disconnected)
-    
+
     self.status = .disconnected
-    
+
     self.input?.delegate = nil
     self.output?.delegate = nil
     self.input?.close()
@@ -87,122 +87,120 @@ final class NetSocket: NSObject, StreamDelegate {
     self.output = nil
     self.inputBuffer = []
     self.outputBuffer = []
-    
+
     if disconnected {
       self.delegate?.netsocketDisconnected(socket: self, error: err)
     }
   }
-  
+
   @MainActor public func write(_ data: Data) {
     guard let output = self.output else {
       return
     }
-    
+
     self.outputBuffer.append(contentsOf: data)
-    
+
     if output.hasSpaceAvailable {
       self.writeBufferToStream()
     }
   }
-  
+
   @MainActor public func write(_ data: [UInt8]) {
     guard let output = self.output else {
       return
     }
-    
+
     self.outputBuffer.append(contentsOf: data)
-    
+
     if output.hasSpaceAvailable {
       self.writeBufferToStream()
     }
   }
-  
+
   @MainActor public func read(count: Int) -> [UInt8] {
     guard self.inputBuffer.count > 0, count > 0 else {
       return []
     }
-    
+
     let amountToRead = min(count, self.inputBuffer.count)
     let dataRead: [UInt8] = Array(self.inputBuffer[0..<amountToRead])
     self.inputBuffer.removeFirst(amountToRead)
-    
+
     return dataRead
   }
-  
+
   @MainActor public func read(count: Int) -> Data {
     guard self.inputBuffer.count > 0, count > 0 else {
       return Data()
     }
-    
+
     let amountToRead = min(count, self.inputBuffer.count)
-    
+
     let dataRead: Data = Data(self.inputBuffer[0..<amountToRead])
     self.inputBuffer.removeFirst(amountToRead)
-    
+
     return dataRead
   }
-  
+
   @MainActor public func readAll() -> [UInt8] {
     guard self.inputBuffer.count > 0 else {
       return []
     }
-    
+
     let dataRead: [UInt8] = Array(self.inputBuffer)
     self.inputBuffer = []
-    
+
     return dataRead
   }
-  
+
   @MainActor public func readAll() -> Data {
     guard self.inputBuffer.count > 0 else {
       return Data()
     }
-    
+
     let dataRead: Data = Data(self.inputBuffer)
     self.inputBuffer = []
-    
+
     return dataRead
   }
-  
+
   @MainActor private func writeBufferToStream() {
     guard let output = self.output, self.outputBuffer.count > 0 else {
       return
     }
-    
+
     let bytesWritten = output.write(self.outputBuffer, maxLength: self.outputBuffer.count)
     print("NetSocket => \(bytesWritten) bytes")
     if bytesWritten > 0 {
       self.outputBuffer.removeFirst(bytesWritten)
       self.delegate?.netsocketSent(socket: self, count: bytesWritten)
-    }
-    else if bytesWritten == -1 {
+    } else if bytesWritten == -1 {
       self.close(output.streamError)
     }
   }
-  
+
   @MainActor private func readStreamToBuffer() {
     guard let input = self.input else {
       return
     }
-    
+
     let bytesRead = input.read(&self.readBuffer, maxLength: 4 * 1024)
     print("NetSocket <= \(bytesRead) bytes")
     if bytesRead > 0 {
       self.inputBuffer.append(contentsOf: self.readBuffer[0..<bytesRead])
       self.delegate?.netsocketReceived(socket: self, bytes: self.inputBuffer)
-    }
-    else if bytesRead == -1 {
+    } else if bytesRead == -1 {
       self.close(input.streamError)
     }
   }
-  
+
   // MARK: -
-    
+
   @MainActor func stream(_ aStream: Stream, handle eventCode: Stream.Event) {
     guard let input = self.input, let output = self.output else {
       return
     }
-    
+
     switch eventCode {
     case .openCompleted:
       if aStream == input {
@@ -226,44 +224,49 @@ final class NetSocket: NSObject, StreamDelegate {
       print("NetSocket: End encounterd")
       self.close()
     case .errorOccurred:
-      print("NetSocket: Error encounterd", input.streamError.debugDescription, output.streamError.debugDescription)
+      print(
+        "NetSocket: Error encounterd", input.streamError.debugDescription,
+        output.streamError.debugDescription)
       let err = input.streamError ?? output.streamError
       self.close(err)
     default:
       break
     }
   }
-  
+
   // MARK: -
-  
+
   private func setupStreamOptions() {
     if let input = self.input {
-      let socketData: Data = CFReadStreamCopyProperty(input as CFReadStream, CFStreamPropertyKey.socketNativeHandle) as! Data;
-      var socketHandle: CFSocketNativeHandle = 0;
-      (socketData as NSData).getBytes(&socketHandle, length: MemoryLayout.size(ofValue: socketHandle));
-      
-      var value: Int = 0;
-      let size = UInt32(MemoryLayout.size(ofValue: value));
-      
-      value = 1;
+      let socketData: Data =
+        CFReadStreamCopyProperty(input as CFReadStream, CFStreamPropertyKey.socketNativeHandle)
+        as! Data
+      var socketHandle: CFSocketNativeHandle = 0
+      (socketData as NSData).getBytes(
+        &socketHandle, length: MemoryLayout.size(ofValue: socketHandle))
+
+      var value: Int = 0
+      let size = UInt32(MemoryLayout.size(ofValue: value))
+
+      value = 1
       if setsockopt(socketHandle, IPPROTO_TCP, TCP_NODELAY, &value, size) != 0 {
-        print("NetSocket: failed to set TCP_NODELAY");
+        print("NetSocket: failed to set TCP_NODELAY")
       }
       // Enable keepalive
-      value = 1;
+      value = 1
       if setsockopt(socketHandle, SOL_SOCKET, SO_KEEPALIVE, &value, size) != 0 {
-        print("NetSocket: failed to set SO_KEEPALIVE");
+        print("NetSocket: failed to set SO_KEEPALIVE")
       }
-//      // Number of keepalives before close (including first keepalive packet)
-//      value = 5
-//      if setsockopt(socketHandle, IPPROTO_TCP, TCP_KEEPCNT, &value, size) != 0 {
-//        print("NetSocket: failed to set TCP_KEEPCNT");
-//      }
-//      // Idle time used when SO_KEEPALIVE is enabled. Sets how long connection must be idle before keepalive is sent.
-//      value = 60
-//      if setsockopt(socketHandle, IPPROTO_TCP, TCP_KEEPALIVE, &value, size) != 0 {
-//        print("NetSocket: failed to set TCP_KEEPALIVE")
-//      }
+      //      // Number of keepalives before close (including first keepalive packet)
+      //      value = 5
+      //      if setsockopt(socketHandle, IPPROTO_TCP, TCP_KEEPCNT, &value, size) != 0 {
+      //        print("NetSocket: failed to set TCP_KEEPCNT");
+      //      }
+      //      // Idle time used when SO_KEEPALIVE is enabled. Sets how long connection must be idle before keepalive is sent.
+      //      value = 60
+      //      if setsockopt(socketHandle, IPPROTO_TCP, TCP_KEEPALIVE, &value, size) != 0 {
+      //        print("NetSocket: failed to set TCP_KEEPALIVE")
+      //      }
     }
   }
 }
-- 
cgit