@reown/appkit-core-react-native 0.0.0-develop-20250728141602 → 0.0.0-develop-20251008155354

package/lib/module/controllers/SwapController.js

@@ -1,20 +1,20 @@
+"use strict";
+
 import { subscribeKey as subKey } from 'valtio/utils';
 import { proxy, subscribe as sub } from 'valtio';
 import { NumberUtil } from '@reown/appkit-common-react-native';
 import { ConstantsUtil } from '../utils/ConstantsUtil';
 import { SwapApiUtil } from '../utils/SwapApiUtil';
-import { NetworkController } from './NetworkController';
 import { BlockchainApiController } from './BlockchainApiController';
 import { OptionsController } from './OptionsController';
 import { SwapCalculationUtil } from '../utils/SwapCalculationUtil';
 import { SnackController } from './SnackController';
+import { LogController } from './LogController';
 import { RouterController } from './RouterController';
-import { ConnectorController } from './ConnectorController';
-import { AccountController } from './AccountController';
 import { CoreHelperUtil } from '../utils/CoreHelperUtil';
-import { ConnectionController } from './ConnectionController';
 import { TransactionsController } from './TransactionsController';
 import { EventsController } from './EventsController';
+import { ConnectionsController } from './ConnectionsController';
 
 // -- Constants ---------------------------------------- //
 export const INITIAL_GAS_LIMIT = 150000;
@@ -32,8 +32,7 @@ class TransactionError extends Error {
 // -- State --------------------------------------------- //
 const initialState = {
   // Loading states
-  initializing: false,
-  initialized: false,
+  loadingTokens: false,
   loadingPrices: false,
   loadingQuote: false,
   loadingApprovalTransaction: false,
@@ -53,7 +52,6 @@ const initialState = {
   toTokenAmount: '',
   toTokenPriceInUSD: 0,
   networkPrice: '0',
-  networkBalanceInUSD: '0',
   networkTokenSymbol: '',
   inputError: undefined,
   // Request values
@@ -72,7 +70,9 @@ const initialState = {
   maxSlippage: undefined,
   providerFee: undefined
 };
-const state = proxy(initialState);
+const state = proxy({
+  ...initialState
+});
 
 // -- Controller ---------------------------------------- //
 export const SwapController = {
@@ -84,10 +84,17 @@ export const SwapController = {
     return subKey(state, key, callback);
   },
   getParams() {
-    const caipAddress = AccountController.state.caipAddress;
-    const address = CoreHelperUtil.getPlainAddress(caipAddress);
-    const networkAddress = NetworkController.getActiveNetworkTokenAddress();
-    const type = ConnectorController.state.connectedConnector;
+    const {
+      activeAddress,
+      activeNamespace,
+      activeNetwork,
+      connection
+    } = ConnectionsController.state;
+    const address = CoreHelperUtil.getPlainAddress(activeAddress);
+    if (!activeNamespace || !activeNetwork) {
+      throw new Error('No active namespace or network found to swap the tokens from.');
+    }
+    const networkAddress = `${activeNetwork.caipNetworkId}:${ConstantsUtil.NATIVE_TOKEN_ADDRESS[activeNamespace]}`;
     if (!address) {
       throw new Error('No address found to swap the tokens from.');
     }
@@ -96,8 +103,9 @@ export const SwapController = {
     const invalidSourceTokenAmount = !state.sourceTokenAmount;
     return {
       networkAddress,
+      network: activeNetwork,
       fromAddress: address,
-      fromCaipAddress: caipAddress,
+      fromCaipAddress: activeAddress,
       sourceTokenAddress: state.sourceToken?.address,
       toTokenAddress: state.toToken?.address,
       toTokenAmount: state.toTokenAmount,
@@ -107,12 +115,12 @@ export const SwapController = {
       invalidToToken,
       invalidSourceToken,
       invalidSourceTokenAmount,
-      availableToSwap: caipAddress && !invalidToToken && !invalidSourceToken && !invalidSourceTokenAmount,
-      isAuthConnector: type === 'AUTH'
+      availableToSwap: activeAddress && !invalidToToken && !invalidSourceToken && !invalidSourceTokenAmount,
+      isAuthConnector: !!connection?.properties?.provider
     };
   },
   switchTokens() {
-    if (state.initializing || !state.initialized) {
+    if (state.loadingTokens) {
       return;
     }
     let newSourceToken = state.toToken ? {
@@ -132,35 +140,33 @@ export const SwapController = {
     this.setToTokenAmount('');
     this.swapTokens();
   },
-  resetState() {
-    state.myTokensWithBalance = initialState.myTokensWithBalance;
-    state.tokensPriceMap = initialState.tokensPriceMap;
-    state.initialized = initialState.initialized;
-    state.sourceToken = initialState.sourceToken;
-    state.sourceTokenAmount = initialState.sourceTokenAmount;
-    state.sourceTokenPriceInUSD = initialState.sourceTokenPriceInUSD;
-    state.toToken = initialState.toToken;
-    state.toTokenAmount = initialState.toTokenAmount;
-    state.toTokenPriceInUSD = initialState.toTokenPriceInUSD;
-    state.networkPrice = initialState.networkPrice;
-    state.networkTokenSymbol = initialState.networkTokenSymbol;
-    state.networkBalanceInUSD = initialState.networkBalanceInUSD;
-    state.inputError = initialState.inputError;
-  },
   async fetchTokens() {
-    const {
-      networkAddress
-    } = this.getParams();
-    await this.getTokenList();
-    await this.getNetworkTokenPrice();
-    await this.getMyTokensWithBalance();
-    const networkToken = state.tokens?.find(token => token.address === networkAddress);
-    if (networkToken) {
-      state.networkTokenSymbol = networkToken.symbol;
+    try {
+      const {
+        networkAddress
+      } = this.getParams();
+      state.loadingTokens = true;
+      await this.getTokenList();
+      await this.getNetworkTokenPrice();
+      await this.getMyTokensWithBalance();
+      const networkToken = state.tokens?.find(token => token.address === networkAddress);
+      if (networkToken) {
+        state.networkTokenSymbol = networkToken.symbol;
+      }
+
+      // Set default source token if not set
+      if (!state.sourceToken && state.myTokensWithBalance?.length) {
+        const sourceToken = state.myTokensWithBalance?.find(token => token.address.startsWith(networkAddress)) || state.myTokensWithBalance?.[0];
+        this.setSourceToken(sourceToken);
+        this.setSourceTokenAmount('1');
+      }
+    } catch (error) {
+      LogController.sendError(error, 'SwapController.ts', 'initializeState');
+      SnackController.showError('Failed to initialize swap');
+      RouterController.goBack();
+    } finally {
+      state.loadingTokens = false;
     }
-    const sourceToken = state.myTokensWithBalance?.find(token => token.address.startsWith(networkAddress)) || state.myTokensWithBalance?.[0];
-    this.setSourceToken(sourceToken);
-    this.setSourceTokenAmount('1');
   },
   async getTokenList() {
     const tokens = await SwapApiUtil.getTokenList();
@@ -182,15 +188,19 @@ export const SwapController = {
     }, {});
   },
   async getMyTokensWithBalance(forceUpdate) {
-    const balances = await SwapApiUtil.getMyTokensWithBalance(forceUpdate);
-    if (!balances) {
+    await ConnectionsController.fetchBalance(forceUpdate);
+    const swapBalances = SwapApiUtil.mapBalancesToSwapTokens(ConnectionsController.state.balances);
+    if (!swapBalances) {
       return;
     }
     await this.getInitialGasPrice();
-    this.setBalances(balances);
+    this.setBalances(swapBalances);
   },
-  getFilteredPopularTokens() {
-    return state.popularTokens?.filter(token => !state.myTokensWithBalance?.some(t => t.address === token.address));
+  getFilteredPopularTokens(balances) {
+    if (!balances) {
+      return state.popularTokens;
+    }
+    return state.popularTokens?.filter(token => !balances.some(t => t.address === token.address));
   },
   setSourceToken(sourceToken) {
     if (!sourceToken) {
@@ -208,31 +218,17 @@ export const SwapController = {
       state.toTokenAmount = '';
     }
   },
-  async initializeState() {
-    if (state.initializing) {
-      return;
-    }
-    state.initializing = true;
-    if (!state.initialized) {
-      try {
-        await this.fetchTokens();
-        state.initialized = true;
-      } catch (error) {
-        state.initialized = false;
-        SnackController.showError('Failed to initialize swap');
-        RouterController.goBack();
-      }
-    }
-    state.initializing = false;
-  },
   async getAddressPrice(address) {
+    const [namespace, chain] = address.split(':');
+    const networkId = `${namespace}:${chain}`;
     const existPrice = state.tokensPriceMap[address];
     if (existPrice) {
       return existPrice;
     }
     const response = await BlockchainApiController.fetchTokenPrice({
       projectId: OptionsController.state.projectId,
-      addresses: [address]
+      addresses: [address],
+      caipNetworkId: networkId
     });
     const fungibles = response?.fungibles || [];
     const allTokens = [...(state.tokens || []), ...(state.myTokensWithBalance || [])];
@@ -246,9 +242,12 @@ export const SwapController = {
     const {
       networkAddress
     } = this.getParams();
+    const [namespace, chain] = networkAddress.split(':');
+    const networkId = `${namespace}:${chain}`;
     const response = await BlockchainApiController.fetchTokenPrice({
       projectId: OptionsController.state.projectId,
-      addresses: [networkAddress]
+      addresses: [networkAddress],
+      caipNetworkId: networkId
     });
     const token = response?.fungibles?.[0];
     const price = token?.price.toString() || '0';
@@ -279,19 +278,10 @@ export const SwapController = {
     return SwapCalculationUtil.getProviderFeePrice(state.sourceTokenAmount, state.sourceTokenPriceInUSD);
   },
   setBalances(balances) {
-    const {
-      networkAddress
-    } = this.getParams();
-    const caipNetwork = NetworkController.state.caipNetwork;
-    if (!caipNetwork) {
-      return;
-    }
-    const networkToken = balances.find(token => token.address === networkAddress);
+    state.myTokensWithBalance = balances;
     balances.forEach(token => {
       state.tokensPriceMap[token.address] = token.price || 0;
     });
-    state.myTokensWithBalance = balances.filter(token => token.address?.startsWith(caipNetwork.id));
-    state.networkBalanceInUSD = networkToken ? NumberUtil.multiply(networkToken.quantity.numeric, networkToken.price).toString() : '0';
   },
   setToToken(toToken) {
     if (!toToken) {
@@ -326,36 +316,43 @@ export const SwapController = {
   },
   // -- Swap ---------------------------------------------- //
   async swapTokens() {
-    const address = AccountController.state.address;
+    const address = ConnectionsController.state.activeAddress;
     const sourceToken = state.sourceToken;
     const toToken = state.toToken;
     const haveSourceTokenAmount = NumberUtil.bigNumber(state.sourceTokenAmount).isGreaterThan(0);
-    if (!toToken || !sourceToken || state.loadingPrices || !haveSourceTokenAmount) {
+    if (!toToken || !sourceToken || state.loadingPrices || !haveSourceTokenAmount || !address) {
       return;
     }
     state.loadingQuote = true;
     const amountDecimal = NumberUtil.bigNumber(state.sourceTokenAmount).multipliedBy(10 ** sourceToken.decimals).integerValue();
-    const quoteResponse = await BlockchainApiController.fetchSwapQuote({
-      userAddress: address,
-      projectId: OptionsController.state.projectId,
-      from: sourceToken.address,
-      to: toToken.address,
-      gasPrice: state.gasFee,
-      amount: amountDecimal.toString()
-    });
-    state.loadingQuote = false;
-    const quoteToAmount = quoteResponse?.quotes?.[0]?.toAmount;
-    if (!quoteToAmount) {
-      return;
-    }
-    const toTokenAmount = NumberUtil.bigNumber(quoteToAmount).dividedBy(10 ** toToken.decimals).toString();
-    this.setToTokenAmount(toTokenAmount);
-    const isInsufficientToken = this.hasInsufficientToken(state.sourceTokenAmount, sourceToken.address);
-    if (isInsufficientToken) {
-      state.inputError = 'Insufficient balance';
-    } else {
-      state.inputError = undefined;
-      this.setTransactionDetails();
+    try {
+      const quoteResponse = await BlockchainApiController.fetchSwapQuote({
+        userAddress: address,
+        projectId: OptionsController.state.projectId,
+        from: sourceToken.address,
+        to: toToken.address,
+        gasPrice: state.gasFee,
+        amount: amountDecimal.toString()
+      });
+      const quoteToAmount = quoteResponse?.quotes?.[0]?.toAmount;
+      if (!quoteToAmount) {
+        state.loadingQuote = false;
+        return;
+      }
+      const toTokenAmount = NumberUtil.bigNumber(quoteToAmount).dividedBy(10 ** toToken.decimals).toString();
+      this.setToTokenAmount(toTokenAmount);
+      const isInsufficientToken = this.hasInsufficientToken(state.sourceTokenAmount, sourceToken.address);
+      if (isInsufficientToken) {
+        state.inputError = 'Insufficient balance';
+      } else {
+        state.inputError = undefined;
+        this.setTransactionDetails();
+      }
+    } catch (error) {
+      LogController.sendError(error, 'SwapController.ts', 'getQuote');
+      SnackController.showError('Failed to get swap quote');
+    } finally {
+      state.loadingQuote = false;
     }
   },
   // -- Create Transactions -------------------------------------- //
@@ -419,7 +416,7 @@ export const SwapController = {
       if (!response) {
         throw new Error('createAllowanceTransaction - No response from generateApproveCalldata');
       }
-      const gasLimit = await ConnectionController.estimateGas({
+      const gasLimit = await ConnectionsController.estimateGas({
         address: fromAddress,
         to: CoreHelperUtil.getPlainAddress(response.tx.to),
         data: response.tx.data
@@ -469,7 +466,7 @@ export const SwapController = {
     if (!fromCaipAddress || !sourceTokenAmount || !sourceToken || !toToken) {
       return undefined;
     }
-    const amount = ConnectionController.parseUnits(sourceTokenAmount, sourceToken.decimals)?.toString();
+    const amount = ConnectionsController.parseUnits(sourceTokenAmount, sourceToken.decimals)?.toString();
     try {
       const response = await BlockchainApiController.generateSwapCalldata({
         projectId: OptionsController.state.projectId,
@@ -508,39 +505,29 @@ export const SwapController = {
   async sendTransactionForApproval(data) {
     const {
       fromAddress,
-      isAuthConnector
+      network
     } = this.getParams();
     state.loadingApprovalTransaction = true;
-    const approveLimitMessage = `Approve limit increase in your wallet`;
-    if (isAuthConnector) {
-      RouterController.pushTransactionStack({
-        view: null,
-        goBack: true,
-        onSuccess() {
-          SnackController.showLoading(approveLimitMessage);
-        }
-      });
-    } else {
-      SnackController.showLoading(approveLimitMessage);
-    }
+    SnackController.showLoading('Approve limit increase in your wallet');
     try {
-      await ConnectionController.sendTransaction({
+      await ConnectionsController.sendTransaction({
         address: fromAddress,
         to: data.to,
         data: data.data,
         value: BigInt(data.value),
         gasPrice: BigInt(data.gasPrice),
-        chainNamespace: 'eip155'
+        chainNamespace: ConnectionsController.state.activeNamespace,
+        network
       });
       await this.swapTokens();
       await this.getTransaction();
       state.approvalTransaction = undefined;
-      state.loadingApprovalTransaction = false;
     } catch (err) {
       const error = err;
       state.transactionError = error?.shortMessage;
-      state.loadingApprovalTransaction = false;
       SnackController.showError(error?.shortMessage ?? 'Transaction error');
+    } finally {
+      state.loadingApprovalTransaction = false;
     }
   },
   async sendTransactionForSwap(data) {
@@ -549,61 +536,54 @@ export const SwapController = {
     }
     const {
       fromAddress,
-      toTokenAmount,
-      isAuthConnector
+      isAuthConnector,
+      network
     } = this.getParams();
     state.loadingTransaction = true;
-    const snackbarPendingMessage = `Swapping ${state.sourceToken?.symbol} to ${NumberUtil.formatNumberToLocalString(toTokenAmount, 3)} ${state.toToken?.symbol}`;
-    const snackbarSuccessMessage = `Swapped ${state.sourceToken?.symbol} to ${NumberUtil.formatNumberToLocalString(toTokenAmount, 3)} ${state.toToken?.symbol}`;
-    if (isAuthConnector) {
-      RouterController.pushTransactionStack({
-        view: 'Account',
-        goBack: false,
-        onSuccess() {
-          SnackController.showLoading(snackbarPendingMessage);
-          SwapController.resetState();
-        }
-      });
-    } else {
-      SnackController.showLoading('Confirm transaction in your wallet');
-    }
+    const snackbarSuccessMessage = `Swapped ${state.sourceToken?.symbol} to ${state.toToken?.symbol}`;
+    SnackController.showLoading('Confirm transaction in your wallet');
     try {
-      const forceUpdateAddresses = [state.sourceToken?.address, state.toToken?.address].join(',');
-      const transactionHash = await ConnectionController.sendTransaction({
+      const forceUpdateAddresses = [state.sourceToken?.address, state.toToken?.address].filter(Boolean);
+      const transactionHash = await ConnectionsController.sendTransaction({
         address: fromAddress,
         to: data.to,
         data: data.data,
         gas: data.gas,
         gasPrice: BigInt(data.gasPrice),
         value: data.value,
-        chainNamespace: 'eip155'
+        chainNamespace: ConnectionsController.state.activeNamespace,
+        network
       });
       state.loadingTransaction = false;
-      SnackController.showSuccess(snackbarSuccessMessage);
+      SnackController.showSuccess(snackbarSuccessMessage, true);
       EventsController.sendEvent({
         type: 'track',
         event: 'SWAP_SUCCESS',
         properties: {
-          network: NetworkController.state.caipNetwork?.id || '',
+          network: ConnectionsController.state.activeNetwork?.caipNetworkId,
           swapFromToken: this.state.sourceToken?.symbol || '',
           swapToToken: this.state.toToken?.symbol || '',
           swapFromAmount: this.state.sourceTokenAmount || '',
           swapToAmount: this.state.toTokenAmount || '',
-          isSmartAccount: AccountController.state.preferredAccountType === 'smartAccount'
+          isSmartAccount: ConnectionsController.state.accountType === 'smartAccount'
         }
       });
-      SwapController.resetState();
-      if (!isAuthConnector) {
-        RouterController.replace('AccountDefault');
-      }
+      RouterController.replace(isAuthConnector ? 'Account' : 'AccountDefault');
+      SwapController.clearTokens();
       SwapController.getMyTokensWithBalance(forceUpdateAddresses);
-      AccountController.fetchTokenBalance();
       setTimeout(() => {
-        TransactionsController.fetchTransactions(AccountController.state.address, true);
+        if (ConnectionsController.state.activeAddress) {
+          TransactionsController.fetchTransactions(ConnectionsController.state.activeAddress, true);
+        }
       }, 5000);
       return transactionHash;
     } catch (err) {
       const error = err;
+      LogController.sendError(error, 'SwapController.ts', 'sendTransactionForSwap', {
+        sourceToken: this.state.sourceToken?.symbol,
+        toToken: this.state.toToken?.symbol,
+        amount: this.state.sourceTokenAmount
+      });
       state.transactionError = error?.shortMessage;
       state.loadingTransaction = false;
       SnackController.showError(error?.shortMessage ?? 'Transaction error');
@@ -612,26 +592,59 @@ export const SwapController = {
         event: 'SWAP_ERROR',
         properties: {
           message: error?.shortMessage ?? error?.message ?? 'Unknown',
-          network: NetworkController.state.caipNetwork?.id || '',
+          network: ConnectionsController.state.activeNetwork?.caipNetworkId,
           swapFromToken: this.state.sourceToken?.symbol || '',
           swapToToken: this.state.toToken?.symbol || '',
           swapFromAmount: this.state.sourceTokenAmount || '',
           swapToAmount: this.state.toTokenAmount || '',
-          isSmartAccount: AccountController.state.preferredAccountType === 'smartAccount'
+          isSmartAccount: ConnectionsController.state.accountType === 'smartAccount'
         }
       });
       return undefined;
     }
   },
+  clearTransactionLoaders() {
+    state.loadingApprovalTransaction = false;
+    state.loadingBuildTransaction = false;
+    state.loadingTransaction = false;
+  },
+  clearTokens() {
+    state.sourceToken = initialState.sourceToken;
+    state.sourceTokenAmount = initialState.sourceTokenAmount;
+    state.sourceTokenPriceInUSD = initialState.sourceTokenPriceInUSD;
+    state.toToken = initialState.toToken;
+    state.toTokenAmount = initialState.toTokenAmount;
+    state.toTokenPriceInUSD = initialState.toTokenPriceInUSD;
+    state.inputError = initialState.inputError;
+    state.loadingApprovalTransaction = initialState.loadingApprovalTransaction;
+    state.loadingBuildTransaction = initialState.loadingBuildTransaction;
+    state.loadingTransaction = initialState.loadingTransaction;
+    state.fetchError = initialState.fetchError;
+    state.transactionError = initialState.transactionError;
+    state.swapTransaction = initialState.swapTransaction;
+    state.approvalTransaction = initialState.approvalTransaction;
+  },
+  resetState() {
+    this.clearTokens();
+    state.myTokensWithBalance = initialState.myTokensWithBalance;
+    state.tokensPriceMap = initialState.tokensPriceMap;
+    state.networkPrice = initialState.networkPrice;
+    state.networkTokenSymbol = initialState.networkTokenSymbol;
+  },
   // -- Checks -------------------------------------------- //
   hasInsufficientToken(sourceTokenAmount, sourceTokenAddress) {
+    const {
+      balances
+    } = ConnectionsController.state;
+    const networkToken = balances?.find(t => t.address === undefined);
+    const networkBalanceInUSD = networkToken ? NumberUtil.multiply(networkToken.quantity?.numeric ?? '0', networkToken.price).toString() : '0';
     const isInsufficientSourceTokenForSwap = SwapCalculationUtil.isInsufficientSourceTokenForSwap(sourceTokenAmount, sourceTokenAddress, state.myTokensWithBalance);
     let insufficientNetworkTokenForGas = true;
-    if (AccountController.state.preferredAccountType === 'smartAccount') {
+    if (ConnectionsController.state.accountType === 'smartAccount') {
       // Smart Accounts may pay gas in any ERC20 token
       insufficientNetworkTokenForGas = false;
     } else {
-      insufficientNetworkTokenForGas = SwapCalculationUtil.isInsufficientNetworkTokenForGas(state.networkBalanceInUSD, state.gasPriceInUSD);
+      insufficientNetworkTokenForGas = SwapCalculationUtil.isInsufficientNetworkTokenForGas(networkBalanceInUSD, state.gasPriceInUSD);
     }
     return insufficientNetworkTokenForGas || isInsufficientSourceTokenForSwap;
   },