Skip to content

Commit 437d197

Browse files
author
WangYunjian
committed
feat: enhanced cookie capture capabilities
1 parent 9edf6e3 commit 437d197

File tree

1 file changed

+17
-0
lines changed

1 file changed

+17
-0
lines changed

packages/network-debugger/src/core/request.ts

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -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+
159174
export 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

Comments
 (0)