CP-12207: Add xp balance fetching in background

packages/core-mobile/app/new/common/contexts/useBalanceForAccount.ts

@@ -4,7 +4,7 @@ import { selectTokenVisibility } from 'store/portfolio'
 import {
   fetchBalanceForAccount,
   QueryStatus,
-  selectBalanceStatus,
+  selectAllBalanceStatus,
   selectBalanceTotalInCurrencyForAccount,
   selectIsBalanceLoadedForAccount
 } from 'store/balance'
@@ -19,8 +19,8 @@ export const useBalanceForAccount = (
   balance: number
 } => {
   const dispatch = useDispatch()
-  const balanceStatus = useSelector(selectBalanceStatus)
-  const isBalanceLoading = balanceStatus !== QueryStatus.IDLE
+  const allBalanceStatus = useSelector(selectAllBalanceStatus)
+  const isBalanceLoading = allBalanceStatus !== QueryStatus.IDLE
   const [isFetchingBalance, setIsFetchingBalance] = useState(true)
   const tokenVisibility = useSelector(selectTokenVisibility)
   const accountBalance = useSelector(

packages/core-mobile/app/new/features/portfolio/assets/components/AssetsScreen.tsx

@@ -20,6 +20,7 @@ import {
   selectIsAllBalancesInaccurate,
   selectIsBalanceLoadedForAccount,
   selectIsLoadingBalances,
+  selectIsPollingBalances,
   selectIsRefetchingBalances
 } from 'store/balance'
 import { selectEnabledNetworks } from 'store/network'
@@ -58,6 +59,7 @@ const AssetsScreen: FC<Props> = ({
   )
   const isAllBalancesError = useSelector(selectIsAllBalancesError)
   const isBalanceLoading = useSelector(selectIsLoadingBalances)
+  const isBalancePolling = useSelector(selectIsPollingBalances)
   const isRefetchingBalance = useSelector(selectIsRefetchingBalances)
   const tokenVisibility = useSelector(selectTokenVisibility)
   const balanceTotalInCurrency = useSelector(
@@ -80,11 +82,13 @@ const AssetsScreen: FC<Props> = ({
     [goToTokenManagement, view, onScrollResync]
   )
 
-  const isLoadingBalance = isRefetchingBalance || isBalanceLoading
+  const isLoadingBalance =
+    isRefetchingBalance || isBalanceLoading || isBalancePolling
+
   const isGridView = view.selected === AssetManageView.Grid
   const numColumns = isGridView ? 2 : 1
 
-  // Only show loading state for initial load, not during background polling
+  // Only show loading state for initial load
   const isInitialLoading = isLoadingBalance && !isBalanceLoaded
 
   const hasNoAssets =

packages/core-mobile/app/services/balance/BalanceService.ts

@@ -11,6 +11,7 @@ import ModuleManager from 'vmModule/ModuleManager'
 import { mapToVmNetwork } from 'vmModule/utils/mapToVmNetwork'
 import { coingeckoInMemoryCache } from 'utils/coingeckoInMemoryCache'
 import { NetworkVMType } from '@avalabs/core-chains-sdk'
+import { chunk } from 'lodash'
 
 export type BalancesForAccount = {
   accountId: string
@@ -20,6 +21,8 @@ export type BalancesForAccount = {
   error: Error | null
 }
 
+export type BalancesForXpAddress = Omit<BalancesForAccount, 'accountId'>
+
 export class BalanceService {
   async getBalancesForAccount({
     network,
@@ -68,6 +71,58 @@ export class BalanceService {
       error: null
     }
   }
+
+  /**
+   * Get balances for active addresses on XP chain
+   */
+  async getXPBalances({
+    currency,
+    network,
+    addresses
+  }: {
+    currency: string
+    network: Network & { vmName: NetworkVMType.AVM | NetworkVMType.PVM }
+    addresses: string[]
+  }): Promise<BalancesForXpAddress[]> {
+    const allBalances: BalancesForXpAddress[] = []
+
+    // avalancheModule.getBalances can only process up to 64 addresses at a time, so we need to split the addresses into chunks
+    const chunkSize = 64
+    const chunks = chunk(addresses, chunkSize)
+
+    await Promise.all(
+      chunks.map(async c => {
+        const balancesResponse =
+          await ModuleManager.avalancheModule.getBalances({
+            addresses: c,
+            currency,
+            network: mapToVmNetwork(network),
+            storage: coingeckoInMemoryCache,
+            tokenTypes: [TokenType.NATIVE]
+          })
+
+        for (const address in balancesResponse) {
+          const balances = balancesResponse[address] ?? {}
+          if ('error' in balances) {
+            allBalances.push({
+              accountAddress: address,
+              chainId: network.chainId,
+              tokens: [],
+              error: balances.error as Error
+            })
+          } else {
+            allBalances.push({
+              accountAddress: address,
+              chainId: network.chainId,
+              tokens: Object.values(balances),
+              error: null
+            })
+          }
+        }
+      })
+    )
+    return allBalances
+  }
 }
 
 export default new BalanceService()

packages/core-mobile/app/services/sentry/types.ts

@@ -15,6 +15,7 @@ export type SpanName =
  * Keeps track of all possible op names
  */
 export type OpName =
+  | 'svc.balance.get_for_xp_networks'
   | 'svc.balance.get_for_account'
   | 'svc.balance.get_for_address'
   | 'svc.balance.btc.get'

packages/core-mobile/app/store/account/listeners.ts

@@ -35,6 +35,7 @@ import {
   migrateRemainingActiveAccounts,
   shouldMigrateActiveAccounts
 } from './utils'
+
 const initAccounts = async (
   _action: AnyAction,
   listenerApi: AppListenerEffectAPI

packages/core-mobile/app/store/account/utils.ts

@@ -3,6 +3,7 @@ import { Account, AccountCollection } from 'store/account/types'
 import { Network, NetworkVMType } from '@avalabs/core-chains-sdk'
 import WalletFactory from 'services/wallet/WalletFactory'
 import { WalletType } from 'services/wallet/types'
+import { uniqWith } from 'lodash'
 import SeedlessWallet from 'seedless/services/wallet/SeedlessWallet'
 import { transactionSnackbar } from 'common/utils/toast'
 import Logger from 'utils/Logger'
@@ -15,7 +16,8 @@ import { commonStorage } from 'utils/mmkv'
 import { StorageKey } from 'resources/Constants'
 import { appendToStoredArray, loadArrayFromStorage } from 'utils/mmkv/storages'
 import { setIsMigratingActiveAccounts } from 'store/wallet/slice'
-import { setNonActiveAccounts } from './slice'
+import WalletService from 'services/wallet/WalletService'
+import { setAccounts, setNonActiveAccounts } from './slice'
 
 export function getAddressByVM(
   vm: VM,
@@ -120,7 +122,12 @@ export const migrateRemainingActiveAccounts = async ({
 
     const accountIds = Object.keys(accounts)
     if (accountIds.length > 0) {
-      listenerApi.dispatch(setNonActiveAccounts(accounts))
+      // set accounts for seedless wallet, which trigger balance update
+      // * seedless wallet fetches xp balances by iterating over xp addresses over all accounts
+      // * so we need to wait for all accounts to be fetched to update balances
+      walletType === WalletType.SEEDLESS
+        ? listenerApi.dispatch(setAccounts(accounts))
+        : listenerApi.dispatch(setNonActiveAccounts(accounts))
 
       recentAccountsStore.getState().addRecentAccounts(accountIds)
 
@@ -202,3 +209,46 @@ const markWalletAsMigrated = (walletId: string): void => {
     walletId
   )
 }
+export async function getAddressesForXP({
+  isDeveloperMode,
+  walletId,
+  walletType,
+  networkType,
+  onlyWithActivity
+}: {
+  isDeveloperMode: boolean
+  walletId: string | null
+  walletType: WalletType | undefined
+  networkType: NetworkVMType.AVM | NetworkVMType.PVM
+  onlyWithActivity: boolean
+}): Promise<string[]> {
+  if (!walletId) {
+    throw new Error('Wallet ID is required')
+  }
+  if (!walletType) {
+    throw new Error('Wallet type is unknown')
+  }
+  try {
+    const activeAddresses = await WalletService.getAddressesFromXpubXP({
+      walletId,
+      walletType,
+      networkType,
+      isTestnet: isDeveloperMode,
+      onlyWithActivity
+    })
+
+    const externalAddresses = activeAddresses.externalAddresses.map(
+      address => address.address
+    )
+    const internalAddresses = activeAddresses.internalAddresses.map(
+      address => address.address
+    )
+    return uniqWith(
+      [...externalAddresses, ...internalAddresses],
+      (a, b) => a === b
+    )
+  } catch (error) {
+    Logger.error('Failed to get addresses for XP', error)
+    throw new Error('Failed to get addresses for XP')
+  }
+}