feat: pass socket to onRes for ws

Author: ronag

README.md

@@ -109,9 +109,9 @@ See [`upgrade`](https://nodejs.org/api/http.html#http_event_upgrade)
   - `onReq(req, options)`: called before proxy request
     - `req`: [`http.IncomingMessage`](https://nodejs.org/api/http.html#http_class_http_incomingmessage) or [`http2.Http2ServerRequest`](https://nodejs.org/api/http2.html#http2_class_http2_http2serverrequest)
     - `options`: options passed to [`http.request(options)`](https://nodejs.org/api/http.html#http_http_request_options_callback)
-  - `onRes(req, res)`: called before proxy response
+  - `onRes(req, resOrSocket)`: called before proxy response
     - `req`: [`http.IncomingMessage`](https://nodejs.org/api/http.html#http_class_http_incomingmessage) or [`http2.Http2ServerRequest`](https://nodejs.org/api/http2.html#http2_class_http2_http2serverrequest)
-    - `res`: [`http.ServerResponse`](https://nodejs.org/api/http.html#http_http_request_options_callback) or [`http2.Http2ServerResponse`](https://nodejs.org/api/http2.html#http2_class_http2_http2serverresponse)
+    - `resOrSocket`: For `web` [`http.ServerResponse`](https://nodejs.org/api/http.html#http_http_request_options_callback) or [`http2.Http2ServerResponse`](https://nodejs.org/api/http2.html#http2_class_http2_http2serverresponse) and for `ws` [`net.Socket`](https://nodejs.org/api/net.html#net_class_net_socket)
 
 ### License
 

index.js

@@ -179,6 +179,10 @@ function proxy (req, resOrSocket, options, onRes, onError) {
         proxyRes.on('aborted', () => callback(createError('socket hang up', 'ECONNRESET', 502)))
 
         if (resOrSocket instanceof net.Socket) {
+          if (onRes) {
+            onRes(req, resOrSocket)
+          }
+
           if (!proxyRes.upgrade) {
             resOrSocket.end()
           }