Improved approval checks

BeanstalkFarms · Jun 25, 2023 · 0f42a0b · 0f42a0b
1 parent c2e20c9
commit 0f42a0b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/projects/ui/src/components/Field/Actions/Sow.tsx b/projects/ui/src/components/Field/Actions/Sow.tsx
@@ -149,6 +149,11 @@ const SowForm: FC<
     claimBeansState: values.claimableBeans,
   });
 
+  /// Checks
+  const shouldApprove = 
+    values.balanceFrom === BalanceFrom.EXTERNAL || 
+    values.balanceFrom === BalanceFrom.TOTAL && values.tokens[0].amount?.gt(balances[tokenIn.address].internal);
+
   const handleSetBalanceFrom = useCallback(
     (_balanceFrom: BalanceFrom) => {
       setFieldValue('balanceFrom', _balanceFrom);
@@ -354,7 +359,7 @@ const SowForm: FC<
           size="large"
           disabled={!isSubmittable || isSubmitting}
           contract={sdk.contracts.beanstalk}
-          tokens={values.tokens}
+          tokens={shouldApprove ? values.tokens : []}
           mode="auto"
         >
           Sow

diff --git a/projects/ui/src/components/Swap/Actions/Transfer.tsx b/projects/ui/src/components/Swap/Actions/Transfer.tsx
@@ -239,7 +239,7 @@ const TransferForm: FC<
   /// Checks
   const shouldApprove =
     fromMode === FarmFromMode.EXTERNAL ||
-    fromMode === FarmFromMode.INTERNAL_EXTERNAL;
+    fromMode === FarmFromMode.INTERNAL_EXTERNAL && amount?.gt(balances[tokenIn.address]?.internal);
 
   const amountsCheck = amount?.gt(0);
   const enoughBalanceCheck = amount