From 6178c260f86129286c2e35fe071fbb7a932a8afb Mon Sep 17 00:00:00 2001
From: Alexander Evchenko <alexivchenko9@gmail.com>
Date: Tue, 12 Sep 2023 13:13:31 +0400
Subject: [PATCH] feat: handle insuffient funds

---
 src/lib/i18n.ts                               |  1 +
 .../TransactionSummaryComponent.tsx           |  2 +-
 .../ReviewTransactionContainer.tsx            | 30 ++++++++++++++++++-
 3 files changed, 31 insertions(+), 2 deletions(-)

diff --git a/src/lib/i18n.ts b/src/lib/i18n.ts
index 9495eb393..32a3e6be8 100644
--- a/src/lib/i18n.ts
+++ b/src/lib/i18n.ts
@@ -205,6 +205,7 @@ const resources = {
       transaction_summary_function_type: 'type',
       transaction_summary_plus_fees: '+ fees',
       transaction_summary_plus_fees_capitalcase: '+ Fees',
+      transaction_summary_insufficient_funds: 'Insufficient funds',
       profile_screen_title: 'Profile',
       profile_contact_details_subtitle: 'Contact Details',
       profile_phone_label: 'Phone Number',
diff --git a/src/screens/transactionSummary/TransactionSummaryComponent.tsx b/src/screens/transactionSummary/TransactionSummaryComponent.tsx
index 29ebaae8d..a3164c9cb 100644
--- a/src/screens/transactionSummary/TransactionSummaryComponent.tsx
+++ b/src/screens/transactionSummary/TransactionSummaryComponent.tsx
@@ -33,8 +33,8 @@ import {
 import { TransactionSummaryScreenProps } from '.'
 
 interface Props {
-  goBack?: () => void
   wallet: RIFWallet
+  goBack?: () => void
 }
 
 type TransactionSummaryComponentProps = Omit<
diff --git a/src/ux/requestsModal/ReviewRelayTransaction/ReviewTransactionContainer.tsx b/src/ux/requestsModal/ReviewRelayTransaction/ReviewTransactionContainer.tsx
index ac9fe6119..3ef8965cb 100644
--- a/src/ux/requestsModal/ReviewRelayTransaction/ReviewTransactionContainer.tsx
+++ b/src/ux/requestsModal/ReviewRelayTransaction/ReviewTransactionContainer.tsx
@@ -5,8 +5,9 @@ import {
 import { BigNumber, constants } from 'ethers'
 import { useCallback, useEffect, useMemo, useState } from 'react'
 import { useTranslation } from 'react-i18next'
-import { StyleSheet, View } from 'react-native'
+import { Alert, StyleSheet, View } from 'react-native'
 import { useSafeAreaInsets } from 'react-native-safe-area-context'
+import { RIF_TOKEN_ADDRESS_TESTNET } from '@rsksmart/rif-relay-light-sdk'
 
 import { balanceToDisplay, convertTokenToUSD } from 'lib/utils'
 
@@ -23,9 +24,16 @@ import { ChainTypeEnum } from 'store/slices/settingsSlice/types'
 import { selectUsdPrices } from 'store/slices/usdPricesSlice'
 import { useAppDispatch, useAppSelector } from 'store/storeUtils'
 import { addRecentContact } from 'store/slices/contactsSlice'
+import { selectBalances } from 'store/slices/balancesSlice'
 
 import useEnhancedWithGas from '../useEnhancedWithGas'
 
+const tokenToBoolMap = new Map([
+  [TokenSymbol.RIF, true],
+  [TokenSymbol.TRIF, true],
+  [undefined, false],
+])
+
 interface Props {
   request: SendTransactionRequest
   onConfirm: () => void
@@ -42,6 +50,7 @@ export const ReviewTransactionContainer = ({
   const tokenPrices = useAppSelector(selectUsdPrices)
   // enhance the transaction to understand what it is:
   const { wallet, chainId } = useAppSelector(selectWalletState)
+  const balances = useAppSelector(selectBalances)
   const [txCostInRif, setTxCostInRif] = useState<BigNumber>()
   const { t } = useTranslation()
 
@@ -136,6 +145,23 @@ export const ReviewTransactionContainer = ({
     const feeValue = txCostInRif
       ? `${balanceToDisplay(txCostInRif, 18, 0)}`
       : '0'
+
+    let insufficientFunds = false
+
+    if (tokenToBoolMap.get(symbol as TokenSymbol)) {
+      insufficientFunds =
+        Number(value) + Number(feeValue) >
+        Number(balances[RIF_TOKEN_ADDRESS_TESTNET].balance)
+    } else {
+      insufficientFunds =
+        Number(feeValue) > Number(balances[RIF_TOKEN_ADDRESS_TESTNET].balance)
+    }
+
+    if (insufficientFunds) {
+      Alert.alert(t('transaction_summary_insufficient_funds'))
+    }
+
+    // get usd values
     const tokenUsd = convertToUSD(Number(value), tokenQuote)
     const feeUsd = convertToUSD(Number(feeValue), feeQuote)
 
@@ -166,6 +192,7 @@ export const ReviewTransactionContainer = ({
           color: sharedColors.white,
           textColor: sharedColors.black,
           accessibilityLabel: 'Confirm',
+          disabled: insufficientFunds,
         },
         {
           style: { marginTop: 10 },
@@ -178,6 +205,7 @@ export const ReviewTransactionContainer = ({
       functionName,
     }
   }, [
+    balances,
     txCostInRif,
     value,
     tokenQuote,