Merge pull request #570 from vonox7/patch-1

Add onHttpResponseHeaders callback. Fixes #567
daltoniam · Oct 8, 2018 · 0f18ae5 · 0f18ae5
2 parents 82a0121 + 46d8cac
commit 0f18ae5
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/Sources/Starscream/WebSocket.swift b/Sources/Starscream/WebSocket.swift
@@ -404,6 +404,7 @@ open class WebSocket : NSObject, StreamDelegate, WebSocketClient, WSStreamDelega
     public var onText: ((String) -> Void)?
     public var onData: ((Data) -> Void)?
     public var onPong: ((Data?) -> Void)?
+    public var onHttpResponseHeaders: (([String: String]) -> Void)?
 
     public var disableSSLCertValidation = false
     public var overrideTrustHostname = false
@@ -876,6 +877,7 @@ open class WebSocket : NSObject, StreamDelegate, WebSocketClient, WSStreamDelega
             i += 1
         }
         advancedDelegate?.websocketHttpUpgrade(socket: self, response: str)
+        onHttpResponseHeaders?(headers)
         if code != httpSwitchProtocolCode {
             return code
         }