feat(package/react) Added extensions support in other query types

packages/react/src/query/useLazyQuery.ts

@@ -21,6 +21,8 @@ export interface UseLazyQueryOptions<TData> {
   retry?: RetryOptions;
   suspense?: boolean;
   operationName?: string;
+  /** Custom GraphQL extensions to be exposed to the query fetcher. */
+  extensions?: Record<string, unknown>;
 }
 
 export interface UseLazyQueryState<TData> {
@@ -97,18 +99,18 @@ export interface UseLazyQuery<GeneratedSchema extends BaseGeneratedSchema> {
               args?: TArgs;
               fetchPolicy?: LazyFetchPolicy;
               operationName?: string;
-            }?
+            }?,
           ]
         : [
             {
               fn?: (query: GeneratedSchema['query'], args: TArgs) => TData;
               args: TArgs;
               fetchPolicy?: LazyFetchPolicy;
               opertionName?: string;
-            }
+            },
           ]
     ) => Promise<TData>,
-    UseLazyQueryState<TData>
+    UseLazyQueryState<TData>,
   ];
 }
 
@@ -131,6 +133,7 @@ export function createUseLazyQuery<TSchema extends BaseGeneratedSchema>(
       retry = defaultRetry,
       suspense = defaultSuspense,
       operationName: defaultOperationName,
+      extensions,
     } = {}
   ) => {
     type TCallback = typeof fn;
@@ -149,7 +152,7 @@ export function createUseLazyQuery<TSchema extends BaseGeneratedSchema>(
       InitUseLazyQueryReducer
     ) as [
       UseLazyQueryState<TData>,
-      React.Dispatch<UseLazyQueryReducerAction<TData>>
+      React.Dispatch<UseLazyQueryReducerAction<TData>>,
     ];
 
     if (suspense) {
@@ -177,6 +180,7 @@ export function createUseLazyQuery<TSchema extends BaseGeneratedSchema>(
               },
               retryPolicy: retry,
               operationName,
+              extensions,
             }
           ).then((data) => {
             const typedData = data as TData;
@@ -207,7 +211,7 @@ export function createUseLazyQuery<TSchema extends BaseGeneratedSchema>(
       };
 
       return Object.freeze([fetchQuery, state]);
-    }, [fn, onCompleted, onError, retry]);
+    }, [fn, onCompleted, onError, retry, extensions]);
   };
 
   return useLazyQuery;

packages/react/src/query/usePaginatedQuery.ts

@@ -58,6 +58,8 @@ export interface UsePaginatedQueryOptions<TData, TArgs> {
    * Activate suspense on first call
    */
   suspense?: boolean;
+  /** Custom GraphQL extensions to be exposed to the query fetcher. */
+  extensions?: Record<string, unknown>;
 }
 
 export interface UsePaginatedQueryData<TData, TArgs> {
@@ -135,6 +137,7 @@ export const createUsePaginatedQuery =
       skip = false,
       suspense = defaultSuspense,
       operationName,
+      extensions,
     }
   ) => {
     type TCallback = typeof fn;
@@ -151,6 +154,7 @@ export const createUsePaginatedQuery =
           cachePolicy: translateFetchPolicy(hookFetchPolicy),
           operationName,
           retryPolicy: retry,
+          extensions,
         }),
       [hookFetchPolicy, operationName, retry]
     );
@@ -195,6 +199,7 @@ export const createUsePaginatedQuery =
           onSelect(selection, cache) {
             context.select(selection, cache);
           },
+          extensions,
         }) as Promise<TData>;
 
         if (!context.shouldFetch) {
@@ -226,6 +231,7 @@ export const createUsePaginatedQuery =
         operationName,
         query,
         retry,
+        extensions,
       ]
     );
 

packages/react/src/query/useRefetch.ts

@@ -21,10 +21,14 @@ export interface UseRefetchOptions {
   retry?: RetryOptions;
   startWatching?: boolean;
   suspense?: boolean;
+  /** Custom GraphQL extensions to be exposed to the query fetcher. */
+  extensions?: Record<string, unknown>;
 }
 
 export interface UseRefetch<TSchema extends BaseGeneratedSchema> {
-  (refetchOptions?: UseRefetchOptions): (<T = void>(
+  (
+    refetchOptions?: UseRefetchOptions
+  ): (<T = void>(
     refetchArg?: T | ((query: TSchema['query']) => T)
   ) => Promise<T | undefined>) &
     UseRefetchState;
@@ -40,6 +44,7 @@ export const createUseRefetch = <TSchema extends BaseGeneratedSchema>(
     startWatching = true,
     retry = defaultRetry,
     suspense = false,
+    extensions,
   } = {}) => {
     const [state, setState] = React.useState<{
       error?: GQtyError;
@@ -78,6 +83,7 @@ export const createUseRefetch = <TSchema extends BaseGeneratedSchema>(
           const { context, resolve } = client.createResolver({
             retryPolicy: retry,
             operationName,
+            extensions,
           });
 
           selections.forEach((selection) => {
@@ -97,7 +103,7 @@ export const createUseRefetch = <TSchema extends BaseGeneratedSchema>(
           throw theError;
         }
       },
-      [notifyOnNetworkStatusChange, operationName, retry]
+      [notifyOnNetworkStatusChange, operationName, retry, extensions]
     );
 
     return React.useMemo(

packages/react/src/query/useTransactionQuery.ts

@@ -41,6 +41,8 @@ export type UseTransactionQueryOptions<TData, TVariables> = {
   retry?: RetryOptions;
   suspense?: boolean;
   operationName?: string;
+  /** Custom GraphQL extensions to be exposed to the query fetcher. */
+  extensions?: Record<string, unknown>;
 };
 
 export interface UseTransactionQuery<TSchema extends BaseGeneratedSchema> {
@@ -75,6 +77,7 @@ export function createUseTransactionQuery<TSchema extends BaseGeneratedSchema>(
       skip = false,
       suspense = defaultSuspense,
       variables,
+      extensions,
     } = {}
   ) => {
     const query = useQuery({
@@ -89,6 +92,7 @@ export function createUseTransactionQuery<TSchema extends BaseGeneratedSchema>(
       refetchOnWindowVisible: false,
       retry,
       suspense,
+      extensions,
     });
 
     useUpdateEffect(() => {