Merge pull request #120 from JohnSundell/live-cli-improve-error-handling

Live CLI: Improve error handling

Author: JohnSundell

live/sources/SocketClient.swift

@@ -25,6 +25,7 @@ import Foundation
 class SocketClient {
     private let port: Int
     private var stream: OutputStream?
+    private lazy var asyncQueue: DispatchQueue = .init(label: "SocketClient.async")
 
     /**
      *  Initialize an instance of this class
@@ -64,16 +65,36 @@ class SocketClient {
             return
         }
 
+        switch stream.streamStatus {
+        case .notOpen:
+            printError(message: "Stream is not open")
+        case .opening, .writing:
+            self.asyncQueue.async {
+                self.send(data: data)
+            }
+            
+            return
+        case .open, .reading, .atEnd:
+            break
+        case .closed:
+            printError(message: "Stream has been closed")
+            return
+        case .error:
+            printError(message: "An error occured. Make sure your app is running in the iOS Simulator")
+            return
+        }
+        
         if !stream.hasSpaceAvailable {
-            printError()
+            printError(message: "No space available in stream")
+            return
         }
 
         let result = data.withUnsafeBytes { bytes in
             return stream.write(bytes, maxLength: data.count)
         }
 
         if result <= 0 {
-            printError()
+            printError(message: "No data could be written")
         }
     }
 
@@ -99,7 +120,7 @@ class SocketClient {
         }
     }
 
-    private func printError() {
-        "Could not send JSON data to application".print(withColor: .red)
+    private func printError(message: String) {
+        "Could not send JSON data to application. \(message).".print(withColor: .red)
     }
 }