Merge f5dab9a into 5a85512

fastify · Feb 4, 2022 · 6b11a56 · 6b11a56
2 parents 5a85512 + f5dab9a
commit 6b11a56
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 5 deletions.
diff --git a/README.md b/README.md
@@ -254,7 +254,7 @@ The default behavior is `reply.send(error)`, which will be disabled if the
 option is specified.
 It must reply the error.
 
-#### `rewriteHeaders(headers)`
+#### `rewriteHeaders(headers, req)`
 
 Called to rewrite the headers of the response, before them being copied
 over to the outer response.

diff --git a/index.d.ts b/index.d.ts
@@ -43,7 +43,8 @@ export interface FastifyReplyFromHooks {
   ) => void;
   body?: unknown;
   rewriteHeaders?: (
-    headers: Http2IncomingHttpHeaders | IncomingHttpHeaders
+    headers: Http2IncomingHttpHeaders | IncomingHttpHeaders,
+    req?: Http2ServerRequest | IncomingMessage
   ) => Http2IncomingHttpHeaders | IncomingHttpHeaders;
   rewriteRequestHeaders?: (
     req: Http2ServerRequest | IncomingMessage,

diff --git a/index.js b/index.js
@@ -142,11 +142,11 @@ module.exports = fp(function from (fastify, opts, next) {
       this.request.log.info('response received')
       if (sourceHttp2) {
         copyHeaders(
-          rewriteHeaders(stripHttp1ConnectionHeaders(res.headers)),
+          rewriteHeaders(stripHttp1ConnectionHeaders(res.headers), req),
           this
         )
       } else {
-        copyHeaders(rewriteHeaders(res.headers), this)
+        copyHeaders(rewriteHeaders(res.headers, req), this)
       }
       this.code(res.statusCode)
       if (onResponse) {
@@ -193,7 +193,7 @@ function getQueryString (search, reqUrl, opts) {
   return ''
 }
 
-function headersNoOp (headers) {
+function headersNoOp (headers, originalReq) {
   return headers
 }