Merge pull request #32 from EthianWong/feat/enhance-cookie-catch

feat: enhanced cookie capture capabilities
GrinZero · Nov 19, 2024 · 6349b40 · 6349b40
2 parents 6c0f3d2 + 437d197
commit 6349b40
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/packages/network-debugger/src/core/request.ts b/packages/network-debugger/src/core/request.ts
@@ -156,6 +156,21 @@ function proxyCallbackFactory(
   }
 }
 
+function proxySetCookie(
+  request: ClientRequest,
+  mainProcess: MainProcess,
+  requestDetail: RequestDetail
+) {
+  let originSetHeader = request.setHeader
+  request.setHeader = function (name, val) {
+    if (name.toLowerCase() === 'cookie') {
+      requestDetail.requestHeaders.cookie = val.toString()
+      mainProcess.updateRequest(requestDetail)
+    }
+    return originSetHeader.call(request, name, val)
+  }
+}
+
 export function requestProxyFactory(
   this: any,
   actualRequestHandler: any,
@@ -219,10 +234,12 @@ export function requestProxyFactory(
         options as RequestOptions,
         proxyCallback
       )
+      proxySetCookie(request, mainProcess, requestDetail)
       return proxyClientRequestFactory(request, requestDetail, mainProcess)
     } else {
       // Call actualRequestHandler with 2 parameters
       const request: ClientRequest = actualRequestHandler(options as RequestOptions, proxyCallback)
+      proxySetCookie(request, mainProcess, requestDetail)
       return proxyClientRequestFactory(request, requestDetail, mainProcess)
     }
   }