fix: make swap reverse work

ava-labs · Dec 18, 2024 · f8858d4 · f8858d4
1 parent 3223b37
commit f8858d4
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 7 deletions.
diff --git a/src/components/common/BNInput.tsx b/src/components/common/BNInput.tsx
@@ -60,7 +60,10 @@ export function BNInput({
       return;
     }
 
-    if (value === 0n && Number(valStr) === 0) {
+    if (value === 0n) {
+      if (Number(valStr) !== 0) {
+        setValStr('');
+      }
       return;
     }
 

diff --git a/src/pages/Swap/hooks/useSwap.tsx b/src/pages/Swap/hooks/useSwap.tsx
@@ -163,6 +163,7 @@ export function useSwap() {
   return {
     setValuesDebouncedSubject,
     swapError,
+    setSwapError,
     isSwapLoading,
     optimalRate,
     swapGasLimit,

diff --git a/src/pages/Swap/hooks/useSwapStateFunctions.ts b/src/pages/Swap/hooks/useSwapStateFunctions.ts
@@ -48,6 +48,7 @@ export function useSwapStateFunctions() {
     swapGasLimit,
     destAmount,
     setDestAmount,
+    setSwapError,
   } = useSwap();
 
   const calculateTokenValueToInput = useCallback(
@@ -186,15 +187,24 @@ export function useSwapStateFunctions() {
       );
       return;
     }
+    setSwapError({ message: '' });
     setSelectedFromToken(toToken);
     setSelectedToToken(fromToken);
     setIsReversed((reversed) => reversed);
-    calculateSwapValue({
-      fromToken: toToken,
-      toToken: fromToken,
-      fromValue: undefined,
-      toValue: undefined,
-    });
+
+    if (!toToken || !fromToken || !destAmount) {
+      return;
+    }
+
+    const fromValue = {
+      amount: bigIntToString(BigInt(destAmount), toToken.decimals),
+      bigint: BigInt(destAmount),
+    };
+    setDestAmount('');
+    setFromTokenValue(fromValue);
+    setFromDefaultValue(fromValue.bigint);
+    setToTokenValue(undefined);
+    calculateTokenValueToInput(BigInt(destAmount), 'to', toToken, fromToken);
   };
 
   const onTokenChange = ({