From a233a75a7fdea9f77c0281e259a2004e5d5df95f Mon Sep 17 00:00:00 2001
From: Matias Poblete <86752543+MattPoblete@users.noreply.github.com>
Date: Wed, 4 Sep 2024 17:45:26 -0400
Subject: [PATCH] =?UTF-8?q?=E2=9C=A8Updating=20balances=20after=20swap?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/components/Swap/SwapComponent.tsx | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/components/Swap/SwapComponent.tsx b/src/components/Swap/SwapComponent.tsx
index a87f4c92..e0216c35 100644
--- a/src/components/Swap/SwapComponent.tsx
+++ b/src/components/Swap/SwapComponent.tsx
@@ -14,7 +14,7 @@ import { requiresTrustline } from 'helpers/stellar';
 import { relevantTokensType } from 'hooks';
 import { useToken } from 'hooks/tokens/useToken';
 import useGetNativeTokenBalance from 'hooks/useGetNativeTokenBalance';
-import { SuccessfullSwapResponse, useSwapCallback } from 'hooks/useSwapCallback';
+import { useSwapCallback } from 'hooks/useSwapCallback';
 import useSwapMainButton from 'hooks/useSwapMainButton';
 import useSwapNetworkFees from 'hooks/useSwapNetworkFees';
 import { TokenType } from 'interfaces';
@@ -37,7 +37,7 @@ import { opacify } from 'themes/utils';
 import SwapCurrencyInputPanel from '../CurrencyInputPanel/SwapCurrencyInputPanel';
 import SwapHeader from './SwapHeader';
 import { ArrowWrapper, SwapWrapper } from './styleds';
-import * as StellarSdk from '@stellar/stellar-sdk';
+import useGetMyBalances from 'hooks/useGetMyBalances';
 
 export const SwapSection = styled('div')(({ theme }) => ({
   position: 'relative',
@@ -116,6 +116,7 @@ export function SwapComponent({
   handleDoSwap?: (setSwapState: (value: SetStateAction<SwapStateProps>) => void) => void;
 }) {
   const sorobanContext = useSorobanReact();
+  const { refetch } = useGetMyBalances()
   const { SnackbarContext } = useContext(AppContext);
   const [showPriceImpactModal, setShowPriceImpactModal] = useState<boolean>(false);
   const [txError, setTxError] = useState<boolean>(false);
@@ -300,6 +301,9 @@ export function SwapComponent({
           ...currentState,
           showConfirm: false,
         }));
+      }).finally(() => {
+        refetch()
+        nativeBalance.mutate()
       });
   };