Update UserAssetsQuery to use new QueryConfigWithSelect type

rainbow-me · Oct 26, 2023 · 2e84de1 · 2e84de1
1 parent 6391e7d
commit 2e84de1
Showing 1 changed file with 15 additions and 5 deletions.
diff --git a/src/resources/assets/UserAssetsQuery.ts b/src/resources/assets/UserAssetsQuery.ts
@@ -9,7 +9,7 @@ import { rainbowFetch } from '@/rainbow-fetch';
 import {
   createQueryKey,
   queryClient,
-  QueryConfig,
+  QueryConfigWithSelect,
   QueryFunctionArgs,
   QueryFunctionResult,
 } from '@/react-query';
@@ -227,9 +227,14 @@ function parseUserAssetsByChain(message: AddysAccountAssetsResponse) {
 // ///////////////////////////////////////////////
 // Query Fetcher (Optional)
 
-export async function fetchUserAssets(
+export async function fetchUserAssets<TSelectResult = UserAssetsResult>(
   { address, currency, connectedToHardhat }: UserAssetsArgs,
-  config: QueryConfig<UserAssetsResult, Error, UserAssetsQueryKey> = {}
+  config: QueryConfigWithSelect<
+    UserAssetsResult,
+    Error,
+    TSelectResult,
+    UserAssetsQueryKey
+  > = {}
 ) {
   return await queryClient.fetchQuery(
     userAssetsQueryKey({ address, currency, connectedToHardhat }),
@@ -241,9 +246,14 @@ export async function fetchUserAssets(
 // ///////////////////////////////////////////////
 // Query Hook
 
-export function useUserAssets(
+export function useUserAssets<TSelectResult = UserAssetsResult>(
   { address, currency, connectedToHardhat }: UserAssetsArgs,
-  config: QueryConfig<UserAssetsResult, Error, UserAssetsQueryKey> = {}
+  config: QueryConfigWithSelect<
+    UserAssetsResult,
+    Error,
+    TSelectResult,
+    UserAssetsQueryKey
+  > = {}
 ) {
   return useQuery(
     userAssetsQueryKey({ address, currency, connectedToHardhat }),