@@ -156,6 +156,21 @@ function proxyCallbackFactory(
156156 }
157157}
158158
159+ function proxySetCookie (
160+ request : ClientRequest ,
161+ mainProcess : MainProcess ,
162+ requestDetail : RequestDetail
163+ ) {
164+ let originSetHeader = request . setHeader
165+ request . setHeader = function ( name , val ) {
166+ if ( name . toLowerCase ( ) === 'cookie' ) {
167+ requestDetail . requestHeaders . cookie = val . toString ( )
168+ mainProcess . updateRequest ( requestDetail )
169+ }
170+ return originSetHeader . call ( request , name , val )
171+ }
172+ }
173+
159174export function requestProxyFactory (
160175 this : any ,
161176 actualRequestHandler : any ,
@@ -219,10 +234,12 @@ export function requestProxyFactory(
219234 options as RequestOptions ,
220235 proxyCallback
221236 )
237+ proxySetCookie ( request , mainProcess , requestDetail )
222238 return proxyClientRequestFactory ( request , requestDetail , mainProcess )
223239 } else {
224240 // Call actualRequestHandler with 2 parameters
225241 const request : ClientRequest = actualRequestHandler ( options as RequestOptions , proxyCallback )
242+ proxySetCookie ( request , mainProcess , requestDetail )
226243 return proxyClientRequestFactory ( request , requestDetail , mainProcess )
227244 }
228245 }
0 commit comments