[#1404] improve query data type from any

IntersectMBO · Jul 5, 2024 · 6a08db6 · 6a08db6
1 parent 76b0459
commit 6a08db6
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/govtool/frontend/src/context/pendingTransaction/utils.tsx b/govtool/frontend/src/context/pendingTransaction/utils.tsx
@@ -5,6 +5,7 @@ import {
   AutomatedVotingOptionCurrentDelegation,
   AutomatedVotingOptionDelegationId,
 } from "@/types/automatedVotingOptions";
+import { CurrentDelegation, VoterInfo } from "@/models";
 
 export const getDesiredResult = (
   type: TransactionType,
@@ -61,11 +62,9 @@ export const refetchData = async (
   if (queryKey === undefined) return;
 
   await queryClient.invalidateQueries(queryKey);
-  // TODO add better type for query data
-  // eslint-disable-next-line @typescript-eslint/no-explicit-any
-  const data = await queryClient.getQueryData<any>(queryKey);
 
   if (type === "delegate") {
+    const data = await queryClient.getQueryData<CurrentDelegation>(queryKey);
     if (
       resourceId === AutomatedVotingOptionDelegationId.no_confidence ||
       resourceId === AutomatedVotingOptionDelegationId.abstain
@@ -74,6 +73,9 @@ export const refetchData = async (
     }
     return data?.dRepHash;
   }
+
+  const data = await queryClient.getQueryData<VoterInfo>(queryKey);
+
   if (type === "registerAsDrep" || type === "retireAsDrep")
     return data?.isRegisteredAsDRep;
   if (type === "registerAsDirectVoter" || type === "retireAsDirectVoter")