Extract addHeaders to enginespec

Author: nuclearace

Source/SocketIO/Engine/SocketEngine.swift

@@ -269,17 +269,7 @@ public final class SocketEngine : NSObject, URLSessionDelegate, SocketEnginePoll
 
         var reqPolling = URLRequest(url: urlPolling, cachePolicy: .reloadIgnoringLocalCacheData, timeoutInterval: 60.0)
 
-        if cookies != nil {
-            let headers = HTTPCookie.requestHeaderFields(with: cookies!)
-            reqPolling.allHTTPHeaderFields = headers
-        }
-
-        if let extraHeaders = extraHeaders {
-            for (headerName, value) in extraHeaders {
-                reqPolling.setValue(value, forHTTPHeaderField: headerName)
-            }
-        }
-
+        addHeaders(to: &reqPolling)
         doLongPoll(for: reqPolling)
     }
 
@@ -303,9 +293,9 @@ public final class SocketEngine : NSObject, URLSessionDelegate, SocketEnginePoll
             urlWebSocket.scheme = "ws"
         }
 
-        if connectParams != nil {
-            for (key, value) in connectParams! {
-                let keyEsc   = key.urlEncode()!
+        if let connectParams = self.connectParams {
+            for (key, value) in connectParams {
+                let keyEsc = key.urlEncode()!
                 let valueEsc = "\(value)".urlEncode()!
 
                 queryString += "&\(keyEsc)=\(valueEsc)"

Source/SocketIO/Engine/SocketEnginePollable.swift

@@ -71,19 +71,6 @@ public protocol SocketEnginePollable : SocketEngineSpec {
 
 // Default polling methods
 extension SocketEnginePollable {
-    private func addHeaders(to req: inout URLRequest) {
-        if cookies != nil {
-            let headers = HTTPCookie.requestHeaderFields(with: cookies!)
-            req.allHTTPHeaderFields = headers
-        }
-
-        if extraHeaders != nil {
-            for (headerName, value) in extraHeaders! {
-                req.setValue(value, forHTTPHeaderField: headerName)
-            }
-        }
-    }
-
     func createRequestForPostWithPostWait() -> URLRequest {
         defer { postWait.removeAll(keepingCapacity: true) }
 

Source/SocketIO/Engine/SocketEngineSpec.swift

@@ -147,6 +147,18 @@ extension SocketEngineSpec {
         return com.url!
     }
 
+    func addHeaders(to req: inout URLRequest) {
+        if let cookies = cookies {
+            req.allHTTPHeaderFields = HTTPCookie.requestHeaderFields(with: cookies)
+        }
+
+        if let extraHeaders = extraHeaders {
+            for (headerName, value) in extraHeaders {
+                req.setValue(value, forHTTPHeaderField: headerName)
+            }
+        }
+    }
+
     func createBinaryDataForSend(using data: Data) -> Either<Data, String> {
         if websocket {
             var byteArray = [UInt8](repeating: 0x4, count: 1)