Get partial headers new option (#145)

* Get partial headers new option * allow non promise too
soundxyz · Feb 8, 2024 · 58c7ef4 · 58c7ef4
1 parent f38de10
commit 58c7ef4
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/.changeset/old-mayflies-fry.md b/.changeset/old-mayflies-fry.md
@@ -0,0 +1,5 @@
+---
+'@soundxyz/graphql-react-query': patch
+---
+
+Get partial headers new option
diff --git a/packages/graphql-react-query/src/client.ts b/packages/graphql-react-query/src/client.ts
@@ -91,6 +91,8 @@ export function GraphQLReactQueryClient<
   fetcher: fetcherConfig,
 
   skipAbort,
+
+  getPartialHeaders,
 }: {
   clientConfig?: QueryClientConfig;
   endpoint: string;
@@ -100,6 +102,8 @@ export function GraphQLReactQueryClient<
   fetcher?: Fetcher;
 
   skipAbort?: [StringDocumentNode, ...StringDocumentNode[]] | boolean;
+
+  getPartialHeaders?(): Promise<Partial<Record<string, string>>> | Partial<Record<string, string>>;
 }) {
   const skipAbortSet = skipAbort ? (skipAbort === true ? true : new Set<string>(skipAbort)) : null;
 
@@ -201,6 +205,7 @@ export function GraphQLReactQueryClient<
           ...headersGlobal,
           ...fetchOptions?.headers,
           ...extraFetchOptions?.headers,
+          ...(getPartialHeaders ? await getPartialHeaders() : {}),
         }),
       },
     });