Refetch upvoters after tx

celo-org · Mar 8, 2024 · 79fc95c · 79fc95c
1 parent 4688544
commit 79fc95c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/src/features/governance/UpvoteForm.tsx b/src/features/governance/UpvoteForm.tsx
@@ -2,6 +2,7 @@ import { Form, Formik, FormikErrors } from 'formik';
 import { FormSubmitButton } from 'src/components/buttons/FormSubmitButton';
 import { ProposalFormDetails } from 'src/features/governance/components/ProposalFormDetails';
 import { useProposalQueue } from 'src/features/governance/hooks/useProposalQueue';
+import { useProposalUpvoters } from 'src/features/governance/hooks/useProposalUpvoters';
 import {
   useGovernanceVotingPower,
   useIsGovernanceUpVoting,
@@ -29,9 +30,11 @@ export function UpvoteForm({
   const { queue } = useProposalQueue();
   const { isUpvoting } = useIsGovernanceUpVoting(address);
   const { votingPower } = useGovernanceVotingPower(address);
+  const { refetch: refetchUpvoters } = useProposalUpvoters();
 
   const { getNextTx, onTxSuccess } = useTransactionPlan<UpvoteFormValues>({
     createTxPlan: (v) => getUpvoteTxPlan(v, queue || [], votingPower || 0n),
+    onStepSuccess: () => refetchUpvoters(),
     onPlanSuccess: (v, r) =>
       onConfirmed({
         message: 'Upvote successful',

diff --git a/src/features/governance/hooks/useProposalUpvoters.ts b/src/features/governance/hooks/useProposalUpvoters.ts
@@ -10,7 +10,7 @@ import { objFilter } from 'src/utils/objects';
 import { decodeEventLog, encodeEventTopics } from 'viem';
 
 export function useProposalUpvoters(id?: number) {
-  const { isLoading, isError, error, data } = useQuery({
+  const { isLoading, isError, error, data, refetch } = useQuery({
     queryKey: ['useProposalUpvoters', id],
     queryFn: () => {
       if (!id) return null;
@@ -27,6 +27,7 @@ export function useProposalUpvoters(id?: number) {
     isLoading,
     isError,
     upvoters: data || undefined,
+    refetch,
   };
 }