From e6926dbd17343cd07c23e511e0048efd358ac704 Mon Sep 17 00:00:00 2001
From: Linkie Link <linkielink.dev@gmail.com>
Date: Wed, 9 Aug 2023 15:29:28 +0200
Subject: [PATCH] fix: fixed the UI not being disabled on transaction (#356)

* fix: fixed the UI not being disabled on transaction

* fix: borrow modal repay toggle
---
 src/components/Account/AccountFund.tsx        |  2 ++
 src/components/Modals/BorrowModal.tsx         | 34 +++++++++++--------
 .../Modals/FundWithdraw/FundAccount.tsx       |  6 ++--
 3 files changed, 25 insertions(+), 17 deletions(-)

diff --git a/src/components/Account/AccountFund.tsx b/src/components/Account/AccountFund.tsx
index 59eaf26b..d9cee9f6 100644
--- a/src/components/Account/AccountFund.tsx
+++ b/src/components/Account/AccountFund.tsx
@@ -130,6 +130,7 @@ export default function AccountFund() {
                 max={BN(balance)}
                 balances={walletBalances}
                 maxText='Max'
+                disabled={isFunding}
               />
             </div>
           )
@@ -142,6 +143,7 @@ export default function AccountFund() {
           rightIcon={<Plus />}
           iconClassName='w-3'
           onClick={handleSelectAssetsClick}
+          disabled={isFunding}
         />
         <SwitchAutoLend
           className='pt-4 mt-4 border border-transparent border-t-white/10'
diff --git a/src/components/Modals/BorrowModal.tsx b/src/components/Modals/BorrowModal.tsx
index c28deed3..b1cc4f0a 100644
--- a/src/components/Modals/BorrowModal.tsx
+++ b/src/components/Modals/BorrowModal.tsx
@@ -185,21 +185,25 @@ function BorrowModal(props: Props) {
               maxText='Max'
               disabled={isConfirming}
             />
-            <Divider className='my-6' />
-            <div className='flex flex-wrap flex-1'>
-              <Text className='w-full mb-1'>Receive funds to Wallet</Text>
-              <Text size='xs' className='text-white/50'>
-                Your borrowed funds will directly go to your wallet
-              </Text>
-            </div>
-            <div className='flex flex-wrap items-center justify-end'>
-              <Switch
-                name='borrow-to-wallet'
-                checked={borrowToWallet}
-                onChange={setBorrowToWallet}
-                disabled={isConfirming}
-              />
-            </div>
+            {!isRepay && (
+              <>
+                <Divider className='my-6' />
+                <div className='flex flex-wrap flex-1'>
+                  <Text className='w-full mb-1'>Receive funds to Wallet</Text>
+                  <Text size='xs' className='text-white/50'>
+                    Your borrowed funds will directly go to your wallet
+                  </Text>
+                </div>
+                <div className='flex flex-wrap items-center justify-end'>
+                  <Switch
+                    name='borrow-to-wallet'
+                    checked={borrowToWallet}
+                    onChange={setBorrowToWallet}
+                    disabled={isConfirming}
+                  />
+                </div>
+              </>
+            )}
           </div>
           <Button
             onClick={onConfirmClick}
diff --git a/src/components/Modals/FundWithdraw/FundAccount.tsx b/src/components/Modals/FundWithdraw/FundAccount.tsx
index 360e78e6..1853090c 100644
--- a/src/components/Modals/FundWithdraw/FundAccount.tsx
+++ b/src/components/Modals/FundWithdraw/FundAccount.tsx
@@ -13,8 +13,8 @@ import useStore from 'store'
 import { BNCoin } from 'types/classes/BNCoin'
 import { byDenom } from 'utils/array'
 import { getAssetByDenom, getBaseAsset } from 'utils/assets'
-import { BN } from 'utils/helpers'
 import { defaultFee } from 'utils/constants'
+import { BN } from 'utils/helpers'
 
 interface Props {
   account: Account
@@ -52,7 +52,7 @@ export default function FundAccount(props: Props) {
       coins: fundingAssets,
     })
     setIsFunding(false)
-    if (result) useStore.setState({ focusComponent: null, walletAssetsModal: null })
+    if (result) useStore.setState({ fundAndWithdrawModal: null, walletAssetsModal: null })
   }, [fundingAssets, accountId, setIsFunding, deposit])
 
   const handleSelectAssetsClick = useCallback(() => {
@@ -115,6 +115,7 @@ export default function FundAccount(props: Props) {
               max={BN(balance)}
               balances={walletBalances}
               maxText='Max'
+              disabled={isFunding}
               className='w-full mb-4'
             />
           )
@@ -126,6 +127,7 @@ export default function FundAccount(props: Props) {
           rightIcon={<Plus />}
           iconClassName='w-3'
           onClick={handleSelectAssetsClick}
+          disabled={isFunding}
         />
         <SwitchAutoLend
           className='pt-4 mt-4 border border-transparent border-t-white/10'