From cc0ea648c22bf12e8f88a8ef64c873d6c83306aa Mon Sep 17 00:00:00 2001
From: Jun Tang <tangjun@shou.tv>
Date: Thu, 25 Oct 2018 15:02:52 +0800
Subject: [PATCH] Add webSocketWillOpen

---
 SocketRocket/SRWebSocket.h | 2 ++
 SocketRocket/SRWebSocket.m | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/SocketRocket/SRWebSocket.h b/SocketRocket/SRWebSocket.h
index a3806f1a0..91ce55ec8 100644
--- a/SocketRocket/SRWebSocket.h
+++ b/SocketRocket/SRWebSocket.h
@@ -368,6 +368,8 @@ extern NSString *const SRHTTPResponseErrorKey;
  */
 - (void)webSocketDidOpen:(SRWebSocket *)webSocket;
 
+- (NSURLRequest *)webSocketWillOpen:(NSURLRequest *)request;
+
 /**
  Called when a given web socket encountered an error.
 
diff --git a/SocketRocket/SRWebSocket.m b/SocketRocket/SRWebSocket.m
index 4e30aef54..76d627e71 100644
--- a/SocketRocket/SRWebSocket.m
+++ b/SocketRocket/SRWebSocket.m
@@ -449,6 +449,8 @@ - (void)didConnect;
     _secKey = SRBase64EncodedStringFromData(SRRandomData(16));
     assert([_secKey length] == 24);
 
+    _urlRequest = [self.delegate webSocketWillOpen:_urlRequest];
+
     CFHTTPMessageRef message = SRHTTPConnectMessageCreate(_urlRequest,
                                                           _secKey,
                                                           SRWebSocketProtocolVersion,