2424
2525import Dispatch
2626import Foundation
27- import StarscreamSocketIO
27+ import Starscream
2828
2929/// The class that handles the engine.io protocol and transports.
3030/// See `SocketEnginePollable` and `SocketEngineWebsocket` for transport specific methods.
@@ -134,7 +134,7 @@ public final class SocketEngine : NSObject, URLSessionDelegate, SocketEnginePoll
134134 private var pongsMissedMax = 0
135135 private var probeWait = ProbeWaitQueue ( )
136136 private var secure = false
137- private var security : SSLSecurity ?
137+ private var security : SocketIO . SSLSecurity ?
138138 private var selfSigned = false
139139
140140 // MARK: Initializers
@@ -310,27 +310,27 @@ public final class SocketEngine : NSObject, URLSessionDelegate, SocketEnginePoll
310310
311311 private func createWebSocketAndConnect( ) {
312312 ws? . delegate = nil // TODO this seems a bit defensive, is this really needed?
313- var request = URLRequest ( url: urlWebSocketWithSid)
313+ ws = WebSocket ( url: urlWebSocketWithSid)
314314
315315 if cookies != nil {
316316 let headers = HTTPCookie . requestHeaderFields ( with: cookies!)
317317 for (key, value) in headers {
318- request . setValue ( value , forHTTPHeaderField : key )
318+ ws ? . headers [ key ] = value
319319 }
320320 }
321321
322322 if extraHeaders != nil {
323323 for (headerName, value) in extraHeaders! {
324- request . setValue ( value , forHTTPHeaderField : headerName )
324+ ws ? . headers [ headerName ] = value
325325 }
326326 }
327327
328- ws = WebSocket ( request : request )
328+
329329 ws? . callbackQueue = engineQueue
330330 ws? . enableCompression = compress
331331 ws? . delegate = self
332332 ws? . disableSSLCertValidation = selfSigned
333- ws? . security = security
333+ ws? . security = security? . security
334334
335335 ws? . connect ( )
336336 }
@@ -605,7 +605,7 @@ public final class SocketEngine : NSObject, URLSessionDelegate, SocketEnginePoll
605605 // MARK: Starscream delegate conformance
606606
607607 /// Delegate method for connection.
608- public func websocketDidConnect( socket: WebSocketClient ) {
608+ public func websocketDidConnect( socket: WebSocket ) {
609609 if !forceWebsockets {
610610 probing = true
611611 probeWebSocket ( )
@@ -617,7 +617,7 @@ public final class SocketEngine : NSObject, URLSessionDelegate, SocketEnginePoll
617617 }
618618
619619 /// Delegate method for disconnection.
620- public func websocketDidDisconnect( socket: WebSocketClient , error: Error ? ) {
620+ public func websocketDidDisconnect( socket: WebSocket , error: NSError ? ) {
621621 probing = false
622622
623623 if closed {
0 commit comments