FR-15071 - Rewrite cookies on failure request to support Microsoft SS…

…O activation process (#313)
frontegg · Jan 29, 2024 · 6088150 · 6088150
1 parent 8016655
commit 6088150
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/packages/nextjs/src/middleware/ProxyResponseCallback.ts b/packages/nextjs/src/middleware/ProxyResponseCallback.ts
@@ -50,9 +50,8 @@ const ProxyResponseCallback: ProxyResCallback<IncomingMessage, NextApiResponse>
         return;
       }
 
+      const cookies = CookieManager.modifySetCookie(proxyRes.headers['set-cookie'], isSecured) ?? [];
       if (isSuccess) {
-        const cookies = CookieManager.modifySetCookie(proxyRes.headers['set-cookie'], isSecured) ?? [];
-
         try {
           if (bodyStr && bodyStr.length > 0) {
             const body = JSON.parse(bodyStr);
@@ -101,6 +100,7 @@ const ProxyResponseCallback: ProxyResCallback<IncomingMessage, NextApiResponse>
           .forEach((header) => {
             res.setHeader(header, `${proxyRes.headers[header]}`);
           });
+        res.setHeader('set-cookie', cookies);
         res.status(statusCode).end(bodyStr);
       }
     } catch (e: any) {