Use getSetCookie to allow multiple set-cookie

vtex · Dec 4, 2023 · 5e1aa79 · 5e1aa79
1 parent 38577e6
commit 5e1aa79
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 5 deletions.
diff --git a/packages/api/src/platforms/vtex/index.ts b/packages/api/src/platforms/vtex/index.ts
@@ -57,7 +57,7 @@ export interface Context {
     locale: string
     flags: FeatureFlags
     searchArgs?: Omit<SearchArgs, 'type'>
-    cookies?: string | null
+    cookies?: string[] | null
   }
   headers: Record<string, string>
 }

diff --git a/packages/api/src/platforms/vtex/utils/cookies.ts b/packages/api/src/platforms/vtex/utils/cookies.ts
@@ -1,5 +1,5 @@
 import type { Context } from '../index'
 
 export const setCookie = (headers: Headers, ctx: Context) => {
-  ctx.storage.cookies = headers?.get('set-cookie') ?? ''
-}
+  ctx.storage.cookies = headers?.getSetCookie() ?? []
+}
diff --git a/packages/core/src/pages/api/graphql.ts b/packages/core/src/pages/api/graphql.ts
@@ -60,7 +60,9 @@ const handler: NextApiHandler = async (request, response) => {
         : 'no-cache, no-store'
 
     if (extensions.cookies && !hasErrors) {
-      response.setHeader('set-cookie', extensions.cookies)
+      extensions.cookies.forEach((cookie) =>
+        response.setHeader('set-cookie', cookie)
+      )
     }
 
     response.setHeader('cache-control', cacheControl)

diff --git a/packages/core/src/server/index.ts b/packages/core/src/server/index.ts
@@ -84,7 +84,7 @@ export const execute = async <V extends Maybe<{ [key: string]: unknown }>, D>(
 ): Promise<{
   data: D
   errors: unknown[]
-  extensions: { cacheControl?: CacheControl; cookies: string | null }
+  extensions: { cacheControl?: CacheControl; cookies: string[] | null }
 }> => {
   const { operationName, variables, query: maybeQuery } = options
   const query = maybeQuery ?? persistedQueries.get(operationName)