From c089bcd774d76f52aba6ac29a4bc57aa15a7a025 Mon Sep 17 00:00:00 2001
From: fmtabbara <fmtabbara@hotmail.co.uk>
Date: Wed, 7 Dec 2022 10:59:52 +0000
Subject: [PATCH] refresh balance after sending tokens

---
 .../src/components/Send/SendSuccessModal.tsx  | 58 +++++++++++--------
 1 file changed, 34 insertions(+), 24 deletions(-)

diff --git a/nym-wallet/src/components/Send/SendSuccessModal.tsx b/nym-wallet/src/components/Send/SendSuccessModal.tsx
index 29f3ffa1e9..cee7c2fdff 100644
--- a/nym-wallet/src/components/Send/SendSuccessModal.tsx
+++ b/nym-wallet/src/components/Send/SendSuccessModal.tsx
@@ -1,8 +1,9 @@
-import React from 'react';
+import React, { useContext } from 'react';
 import { Stack, Typography, SxProps } from '@mui/material';
 import { Link } from '@nymproject/react/link/Link';
 import { TTransactionDetails } from './types';
 import { ConfirmationModal } from '../Modals/ConfirmationModal';
+import { AppContext } from 'src/context';
 
 export const SendSuccessModal = ({
   txDetails,
@@ -14,26 +15,35 @@ export const SendSuccessModal = ({
   onClose: () => void;
   sx?: SxProps;
   backdropProps?: object;
-}) => (
-  <ConfirmationModal
-    open
-    onConfirm={onClose}
-    onClose={onClose}
-    title=""
-    confirmButton="Done"
-    maxWidth="xs"
-    fullWidth
-    sx={sx}
-    backdropProps={backdropProps}
-  >
-    <Stack alignItems="center" spacing={2}>
-      <Typography>You sent</Typography>
-      {txDetails && (
-        <>
-          <Typography variant="h5">{txDetails.amount}</Typography>
-          <Link href={txDetails.txUrl} target="_blank" sx={{ ml: 1 }} text="View on blockchain" />
-        </>
-      )}
-    </Stack>
-  </ConfirmationModal>
-);
+}) => {
+  const { userBalance } = useContext(AppContext);
+
+  const handleClose = async () => {
+    await userBalance.refreshBalances();
+    onClose();
+  };
+
+  return (
+    <ConfirmationModal
+      open
+      onConfirm={handleClose}
+      onClose={handleClose}
+      title=""
+      confirmButton="Done"
+      maxWidth="xs"
+      fullWidth
+      sx={sx}
+      backdropProps={backdropProps}
+    >
+      <Stack alignItems="center" spacing={2}>
+        <Typography>You sent</Typography>
+        {txDetails && (
+          <>
+            <Typography variant="h5">{txDetails.amount}</Typography>
+            <Link href={txDetails.txUrl} target="_blank" sx={{ ml: 1 }} text="View on blockchain" />
+          </>
+        )}
+      </Stack>
+    </ConfirmationModal>
+  );
+};