rewrite set cookie

taichunmin · Sep 13, 2024 · 251d58d · 251d58d
1 parent abd8c53
commit 251d58d
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/index.js b/index.js
@@ -118,6 +118,14 @@ const middlewares = [
     }
     _.each(OMIT_RES_HEADERS, header => { ctx.res.removeHeader(header) })
     const axiosRes = await axios(axiosConfig)
+
+    // Rewrite Set-Cookie header
+    _.update(res, 'headers.set-cookie', cookies => {
+      const rewriteCookie = cookie => cookie.replaceAll(/; (Secure|Partitioned|SameSite=[^;]+)/imsg, '') + '; SameSite=None; Secure; Partitioned'
+      return _.isArray(cookies) ? _.map(cookies, rewriteCookie) : rewriteCookie(cookies)
+    })
+    for (const cookie of res.headers['set-cookie'] ?? []) {}
+
     ctx.res.status(axiosRes.status).header(axiosRes.headers)
     if (NODE_ENV !== 'production') {
       exports.log({