better logic for initial balance check (#562)

Author: SamueleA

packages/checkout/src/views/Checkout/PaymentMethodSelect/PayWithCrypto/useInitialBalanceCheck.tsx

@@ -30,7 +30,11 @@ export const useInitialBalanceCheck = ({
 
   const isInitialBalanceChecked = (navigation.params as PaymentMethodSelectionParams).isInitialBalanceChecked
 
-  const { data: swapRoutes = [], isLoading: swapRoutesIsLoading } = useGetSwapRoutes(
+  const {
+    data: swapRoutes = [],
+    isLoading: swapRoutesIsLoading,
+    isError: isErrorSwapRoutes
+  } = useGetSwapRoutes(
     {
       walletAddress: userAddress ?? '',
       toTokenAddress: buyCurrencyAddress,
@@ -42,7 +46,11 @@ export const useInitialBalanceCheck = ({
     }
   )
 
-  const { data: swapRoutesTokenBalancesData, isLoading: swapRoutesTokenBalancesIsLoading } = useGetTokenBalancesSummary(
+  const {
+    data: swapRoutesTokenBalancesData,
+    isLoading: swapRoutesTokenBalancesIsLoading,
+    isError: isErrorSwapRoutesTokenBalances
+  } = useGetTokenBalancesSummary(
     {
       chainIds: [chainId],
       filter: {
@@ -80,6 +88,18 @@ export const useInitialBalanceCheck = ({
       }
     }
 
+    if (!validSwapRoute) {
+      setNavigation({
+        location: 'payment-method-selection',
+        params: {
+          ...navigation.params,
+          isInitialBalanceChecked: true
+        }
+      })
+
+      return
+    }
+
     setNavigation({
       location: 'payment-method-selection',
       params: {
@@ -95,7 +115,15 @@ export const useInitialBalanceCheck = ({
 
   useEffect(() => {
     if (!isInitialBalanceChecked && !tokenBalancesIsLoading && !swapRoutesIsLoading && !swapRoutesTokenBalancesIsLoading) {
-      if (isInsufficientBalance) {
+      if (isErrorSwapRoutes || isErrorSwapRoutesTokenBalances) {
+        setNavigation({
+          location: 'payment-method-selection',
+          params: {
+            ...navigation.params,
+            isInitialBalanceChecked: true
+          }
+        })
+      } else if (isInsufficientBalance) {
         findSwapQuote()
       } else {
         setNavigation({
@@ -112,6 +140,8 @@ export const useInitialBalanceCheck = ({
     isInsufficientBalance,
     tokenBalancesIsLoading,
     swapRoutesIsLoading,
-    swapRoutesTokenBalancesIsLoading
+    swapRoutesTokenBalancesIsLoading,
+    isErrorSwapRoutes,
+    isErrorSwapRoutesTokenBalances
   ])
 }