feat(solana): load token balances

apps/cowswap-frontend/package.json

@@ -71,6 +71,7 @@
     "@reach/menu-button": "0.18.0",
     "@react-spring/web": "9.7.3",
     "@reduxjs/toolkit": "1.9.5",
+    "@reown/appkit-adapter-solana": "1.8.19",
     "@reown/appkit-adapter-wagmi": "1.8.19",
     "@reown/appkit-controllers": "1.8.19",
     "@reown/appkit": "1.8.19",

apps/cowswap-frontend/src/modules/combinedBalances/hooks/useTokensBalancesCombined.ts

@@ -1,9 +1,9 @@
 import { useAtomValue } from 'jotai'
 
+import type { BalancesState } from '@cowprotocol/balances-and-allowances'
+
 import { balancesCombinedAtom } from '../state/balanceCombinedAtom'
 
-// TODO: Add proper return type annotation
-// eslint-disable-next-line @typescript-eslint/explicit-function-return-type
-export function useTokensBalancesCombined() {
+export function useTokensBalancesCombined(): BalancesState {
   return useAtomValue(balancesCombinedAtom)
 }

apps/cowswap-frontend/src/modules/combinedBalances/updater/BalancesCombinedUpdater.tsx

@@ -12,9 +12,7 @@ import { useIsHooksTradeType } from 'modules/trade'
 
 import { balancesCombinedAtom } from '../state/balanceCombinedAtom'
 
-// TODO: Add proper return type annotation
-// eslint-disable-next-line @typescript-eslint/explicit-function-return-type
-export function BalancesCombinedUpdater() {
+export function BalancesCombinedUpdater(): null {
   const { account, chainId } = useWalletInfo()
   const setBalancesCombined = useSetAtom(balancesCombinedAtom)
   const preHooksBalancesDiff = usePreHookBalanceDiff()

apps/cowswap-frontend/src/modules/permit/hooks/usePermitInfo.ts

@@ -4,7 +4,7 @@ import { useEffect, useMemo } from 'react'
 import { useConfig, usePublicClient } from 'wagmi'
 
 import { getIsNativeToken, getWrappedToken, COW_PROTOCOL_VAULT_RELAYER_ADDRESS } from '@cowprotocol/common-utils'
-import { getAddressKey, mapSupportedNetworks, SupportedChainId } from '@cowprotocol/cow-sdk'
+import { getAddressKey, isNonEvmChain, mapSupportedNetworks, SupportedChainId } from '@cowprotocol/cow-sdk'
 import { Currency } from '@cowprotocol/currency'
 import { DEFAULT_MIN_GAS_LIMIT, getTokenPermitInfo, PermitInfo } from '@cowprotocol/permit-utils'
 import { useWalletInfo } from '@cowprotocol/wallet'
@@ -73,9 +73,11 @@ export function usePermitInfo(
 
   const spender = customSpender || COW_PROTOCOL_VAULT_RELAYER_ADDRESS[chainId]
 
+  // eslint-disable-next-line complexity
   useEffect(() => {
     if (
       !chainId ||
+      isNonEvmChain(chainId) ||
       !isPermitEnabled ||
       !lowerCaseAddress ||
       !config ||

apps/cowswap-frontend/src/modules/tokensList/pure/TokenListItem/index.tsx

@@ -2,7 +2,7 @@ import { MouseEventHandler, ReactNode } from 'react'
 
 import { TokenWithLogo } from '@cowprotocol/common-const'
 import { getCurrencyAddress } from '@cowprotocol/common-utils'
-import { areAddressesEqual, getAddressKey, getTokenId, isEvmChain } from '@cowprotocol/cow-sdk'
+import { areAddressesEqual, getAddressKey, getTokenId } from '@cowprotocol/cow-sdk'
 import { Currency, CurrencyAmount } from '@cowprotocol/currency'
 import { TokenListTags } from '@cowprotocol/tokens'
 import { FiatAmount, HoverTooltip, LoadingRows, LoadingRowSmall, TokenAmount } from '@cowprotocol/ui'
@@ -111,9 +111,7 @@ export function TokenListItem(props: TokenListItemProps): ReactNode {
   // Balances are only fetched for EVM chains (wagmi + BFF are EVM-only). Bridge-only
   // destinations like Solana or Bitcoin land here too — for them, skip the balance column
   // entirely instead of rendering an indefinite loading skeleton.
-  // todo remove it when FE supports Solana rpc connection
-  const canShowBalances = isEvmChain(token.chainId)
-  const shouldShowBalances = isWalletConnected && canShowBalances
+  const shouldShowBalances = isWalletConnected
   const shouldFormatBalances = shouldShowBalances && hasIntersected
   const balanceAmount =
     shouldFormatBalances && balance !== undefined ? CurrencyAmount.fromRawAmount(token, balance.toString()) : undefined

apps/cowswap-frontend/vite.config.mts

@@ -1,5 +1,6 @@
 /// <reference types="vitest" />
 import { lingui } from '@lingui/vite-plugin'
+
 import { sentryVitePlugin } from '@sentry/vite-plugin'
 import react from '@vitejs/plugin-react-swc'
 import { bundleStats } from 'rollup-plugin-bundle-stats'
@@ -289,7 +290,14 @@ export default defineConfig(({ mode, isPreview }) => {
       // in @reown/appkit-adapter-solana (#7709), pnpm resolves the appkit family to two
       // peer-instances; without deduping the controllers package, code in libs/wallet reads
       // an empty ConnectorController while the deduped appkit/adapter-wagmi populate the other.
-      dedupe: ['react-router', '@reown/appkit', '@reown/appkit-adapter-wagmi', '@reown/appkit-controllers', 'wagmi'],
+      dedupe: [
+        'react-router',
+        '@reown/appkit',
+        '@reown/appkit-adapter-wagmi',
+        '@reown/appkit-adapter-solana',
+        '@reown/appkit-controllers',
+        'wagmi',
+      ],
     },
 
     build: {

libs/balances-and-allowances/package.json

@@ -35,6 +35,10 @@
     "@cowprotocol/wallet": "workspace:*",
     "@cowprotocol/wallet-provider": "workspace:*",
     "@cowprotocol/currency": "workspace:*",
+    "@reown/appkit-adapter-solana": "1.8.19",
+    "@solana/spl-token": "0.4.14",
+    "@solana/web3.js": "1.98.4",
+    "@tanstack/react-query": "5.90.20",
     "jotai": "2.16.2",
     "ms.macro": "2.0.0",
     "react": "19.1.2",

libs/balances-and-allowances/src/hooks/usePersistBalancesViaWebCalls.test.tsx

@@ -17,6 +17,12 @@ jest.mock('wagmi', () => ({
   useReadContracts: jest.fn(),
 }))
 
+// The Solana path has its own dedicated test; stub it here so this suite stays focused on
+// EVM wagmi gating and avoids pulling in the reown/web3/react-query runtime.
+jest.mock('./usePersistSolanaBalancesViaWebCalls', () => ({
+  usePersistSolanaBalancesViaWebCalls: jest.fn(),
+}))
+
 const mockBalancesUpdate: PersistentStateByChain<Record<string, number | undefined>> = mapSupportedNetworks({})
 
 const wrapper = ({ children }: { children: ReactNode }): ReactNode => {

libs/balances-and-allowances/src/hooks/usePersistBalancesViaWebCalls.ts

@@ -8,6 +8,7 @@ import { getIsNativeToken } from '@cowprotocol/common-utils'
 import { isEvmChain, SupportedChainId } from '@cowprotocol/cow-sdk'
 
 import { useIsBlockNumberRelevant } from './useIsBlockNumberRelevant'
+import { usePersistSolanaBalancesViaWebCalls } from './usePersistSolanaBalancesViaWebCalls'
 
 import { balancesAtom, BalancesState, balancesUpdateAtom } from '../state/balancesAtom'
 
@@ -49,6 +50,9 @@ export function usePersistBalancesViaWebCalls(params: PersistBalancesAndAllowanc
   // wagmi + viem only support evm chains
   const isEvm = isEvmChain(chainId)
 
+  // Non-EVM chains (e.g. Solana) load balances via their own web calls
+  usePersistSolanaBalancesViaWebCalls(params)
+
   const {
     data: balances,
     isLoading: isBalancesLoading,
@@ -79,21 +83,21 @@ export function usePersistBalancesViaWebCalls(params: PersistBalancesAndAllowanc
 
   // Set balances loading state
   useEffect(() => {
-    if (!setLoadingState) return
+    if (!setLoadingState || !isEvm) return
 
     setBalances((state) => ({ ...state, isLoading: isBalancesLoading, chainId }))
-  }, [setBalances, isBalancesLoading, setLoadingState, chainId])
+  }, [setBalances, isBalancesLoading, setLoadingState, isEvm, chainId])
 
   // Set balances error state for full balances fetches only
   useEffect(() => {
-    if (!setLoadingState) return
+    if (!setLoadingState || !isEvm) return
 
     if (!error) return
 
     const message = error instanceof Error ? error.message : String(error)
 
     setBalances((state) => ({ ...state, error: message, isLoading: false }))
-  }, [setBalances, error, setLoadingState])
+  }, [setBalances, error, setLoadingState, isEvm])
 
   // Set balances to the store
   useEffect(() => {

libs/balances-and-allowances/src/hooks/usePersistSolanaBalancesViaWebCalls.test.tsx

@@ -0,0 +1,232 @@
+import { Provider, useAtomValue } from 'jotai'
+import { useHydrateAtoms } from 'jotai/utils'
+import React, { ReactNode } from 'react'
+
+import { QueryClient, QueryClientProvider } from '@tanstack/react-query'
+
+import { TOKEN_2022_TAG } from '@cowprotocol/common-const'
+import { getAddressKey, SupportedChainId, mapSupportedNetworks, solana } from '@cowprotocol/cow-sdk'
+import { PersistentStateByChain } from '@cowprotocol/types'
+
+import { PublicKey } from '@solana/web3.js'
+import { renderHook, waitFor } from '@testing-library/react'
+
+import { PersistBalancesAndAllowancesParams } from './usePersistBalancesViaWebCalls'
+import { usePersistSolanaBalancesViaWebCalls } from './usePersistSolanaBalancesViaWebCalls'
+
+import { balancesAtom, BalancesState, balancesUpdateAtom } from '../state/balancesAtom'
+
+// Valid base58 addresses so `new PublicKey(...)` inside the hook does not throw.
+const ACCOUNT = 'So11111111111111111111111111111111111111112'
+const MINT_A = 'EPjFWdd5AufqSSqeM2qN1xzybapC8G4wEGGkZwyTDt1v'
+const MINT_B = 'Es9vMFrzaCERmJfrF4H2FYD4KCoNkY11McCe8BenwNYB'
+const NATIVE_MINT = solana.nativeCurrency.address
+
+const CLASSIC_PROGRAM = 'TOKEN_PROGRAM_ID'
+const TOKEN_2022_PROGRAM = 'TOKEN_2022_PROGRAM_ID'
+
+// The ATA key encodes the selected program, so a test can assert which token program a mint was read under.
+function ataKey(mint: string, program: string = CLASSIC_PROGRAM): string {
+  return `ata:${program}:${mint}`
+}
+
+let mockConnection: MockConnection | undefined
+// getAddressKey(address) -> token metadata, mirroring the token list; its tags drive program selection.
+let mockTokensByAddress: Record<string, { tags: string[] } | undefined>
+
+jest.mock('@reown/appkit-adapter-solana/react', () => ({
+  useAppKitConnection: () => ({ connection: mockConnection }),
+}))
+
+jest.mock('@cowprotocol/tokens', () => ({
+  useTokensByAddressMapForChain: () => mockTokensByAddress,
+}))
+
+// The ATA-derivation math is not what we are testing; make it deterministic and echo the mint plus the
+// selected program back so the mocked RPC/`unpackAccount` can look accounts up by their ATA key. We only
+// assert what lands in the atom and which program each ATA was derived with.
+jest.mock('@solana/spl-token', () => ({
+  TOKEN_PROGRAM_ID: 'TOKEN_PROGRAM_ID',
+  TOKEN_2022_PROGRAM_ID: 'TOKEN_2022_PROGRAM_ID',
+  ASSOCIATED_TOKEN_PROGRAM_ID: 'ASSOCIATED_TOKEN_PROGRAM_ID',
+  getAssociatedTokenAddressSync: (
+    mint: { toBase58(): string },
+    _owner: unknown,
+    _allowOwnerOffCurve: boolean,
+    programId: string,
+  ) => ({ toBase58: () => `ata:${programId}:${mint.toBase58()}` }),
+  unpackAccount: (ata: { toBase58(): string }) => ({ amount: mockAmountByAta[ata.toBase58()] }),
+}))
+
+interface MockConnection {
+  rpcEndpoint: string
+  getMultipleAccountsInfo: jest.Mock
+}
+
+// ATA base58 -> token amount, read by the mocked `unpackAccount`.
+let mockAmountByAta: Record<string, bigint>
+// ATA base58 -> account info; an absent entry means "no account exists" (a zero balance).
+let mockInfoByAta: Record<string, { present: true } | undefined>
+
+function createConnection(): MockConnection {
+  return {
+    rpcEndpoint: 'https://solana.example/rpc',
+    getMultipleAccountsInfo: jest.fn((batch: Array<{ toBase58(): string }>) =>
+      Promise.resolve(batch.map((ata) => mockInfoByAta[ata.toBase58()] ?? null)),
+    ),
+  }
+}
+
+const mockBalancesUpdate: PersistentStateByChain<Record<string, number | undefined>> = mapSupportedNetworks({})
+
+function makeParams(overrides: Partial<PersistBalancesAndAllowancesParams> = {}): PersistBalancesAndAllowancesParams {
+  return {
+    account: ACCOUNT,
+    chainId: SupportedChainId.SOLANA,
+    tokenAddresses: [MINT_A, MINT_B],
+    setLoadingState: true,
+    ...overrides,
+  }
+}
+
+function renderWithBalances(params: PersistBalancesAndAllowancesParams): { result: { current: BalancesState } } {
+  return renderHook(
+    () => {
+      usePersistSolanaBalancesViaWebCalls(params)
+      return useAtomValue(balancesAtom)
+    },
+    { wrapper },
+  )
+}
+
+function wrapper({ children }: { children: ReactNode }): ReactNode {
+  const queryClient = new QueryClient({ defaultOptions: { queries: { retry: false } } })
+
+  const HydrateAtoms = ({ children }: { children: ReactNode }): ReactNode => {
+    useHydrateAtoms([
+      [
+        balancesAtom,
+        {
+          isLoading: false,
+          chainId: SupportedChainId.SOLANA,
+          values: {},
+          fromCache: false,
+          hasFirstLoad: false,
+          error: null,
+        } as BalancesState,
+      ],
+      [balancesUpdateAtom, mockBalancesUpdate],
+    ])
+    return <>{children}</>
+  }
+
+  return (
+    <QueryClientProvider client={queryClient}>
+      <Provider>
+        <HydrateAtoms>{children}</HydrateAtoms>
+      </Provider>
+    </QueryClientProvider>
+  )
+}
+
+describe('usePersistSolanaBalancesViaWebCalls', () => {
+  beforeEach(() => {
+    mockTokensByAddress = {}
+    mockAmountByAta = { [ataKey(MINT_A)]: 100n, [ataKey(MINT_B)]: 250n }
+    mockInfoByAta = { [ataKey(MINT_A)]: { present: true }, [ataKey(MINT_B)]: { present: true } }
+    mockConnection = createConnection()
+  })
+
+  it('reads SPL balances via a single batched RPC call and persists them to the atom', async () => {
+    const { result } = renderWithBalances(makeParams())
+
+    await waitFor(() => expect(result.current.hasFirstLoad).toBe(true))
+
+    expect(mockConnection?.getMultipleAccountsInfo).toHaveBeenCalledTimes(1)
+    expect(result.current.values[getAddressKey(MINT_A)]).toBe(100n)
+    expect(result.current.values[getAddressKey(MINT_B)]).toBe(250n)
+  })
+
+  it('clears the loading state once balances are loaded', async () => {
+    const { result } = renderWithBalances(makeParams())
+
+    await waitFor(() => expect(result.current.hasFirstLoad).toBe(true))
+
+    expect(result.current.isLoading).toBe(false)
+  })
+
+  it('treats a missing token account as a zero balance rather than an error', async () => {
+    delete mockInfoByAta[ataKey(MINT_B)]
+
+    const { result } = renderWithBalances(makeParams())
+
+    await waitFor(() => expect(result.current.hasFirstLoad).toBe(true))
+
+    expect(result.current.values[getAddressKey(MINT_A)]).toBe(100n)
+    expect(result.current.values[getAddressKey(MINT_B)]).toBe(0n)
+    expect(result.current.error).toBeNull()
+  })
+
+  it('batches ATAs into chunks of 100 so large token lists do not exceed the RPC limit', async () => {
+    // 250 mints -> ceil(250 / 100) = 3 `getMultipleAccountsInfo` calls, none over the 100-account limit.
+    const mints = Array.from({ length: 250 }, (_, i) =>
+      new PublicKey(Uint8Array.from({ length: 32 }, (_, j) => (i + j + 1) % 256)).toBase58(),
+    )
+    mints.forEach((mint) => {
+      mockAmountByAta[ataKey(mint)] = 7n
+      mockInfoByAta[ataKey(mint)] = { present: true }
+    })
+
+    const { result } = renderWithBalances(makeParams({ tokenAddresses: mints }))
+
+    await waitFor(() => expect(result.current.hasFirstLoad).toBe(true))
+
+    expect(mockConnection?.getMultipleAccountsInfo).toHaveBeenCalledTimes(3)
+    mockConnection?.getMultipleAccountsInfo.mock.calls.forEach(([batch]) => {
+      expect(batch.length).toBeLessThanOrEqual(100)
+    })
+    expect(Object.keys(result.current.values)).toHaveLength(250)
+    expect(result.current.values[getAddressKey(mints[0])]).toBe(7n)
+    expect(result.current.values[getAddressKey(mints[249])]).toBe(7n)
+  })
+
+  it('excludes the native SOL address from the SPL batch (it has no ATA)', async () => {
+    renderWithBalances(makeParams({ tokenAddresses: [NATIVE_MINT, MINT_A] }))
+
+    await waitFor(() => expect(mockConnection?.getMultipleAccountsInfo).toHaveBeenCalled())
+
+    const [atas] = mockConnection!.getMultipleAccountsInfo.mock.calls[0]
+    expect(atas).toHaveLength(1)
+    expect(atas[0].toBase58()).toBe(ataKey(MINT_A))
+  })
+
+  it('keys the update timestamp by the case-sensitive Solana account, not a lowercased alias', async () => {
+    const { result } = renderHook(
+      () => {
+        usePersistSolanaBalancesViaWebCalls(makeParams())
+        return useAtomValue(balancesUpdateAtom)
+      },
+      { wrapper },
+    )
+
+    await waitFor(() => expect(result.current[SupportedChainId.SOLANA]?.[getAddressKey(ACCOUNT)]).toBeDefined())
+
+    // getAddressKey preserves case for Solana pubkeys; a lowercased key would alias distinct owners.
+    expect(ACCOUNT).not.toBe(ACCOUNT.toLowerCase())
+    expect(result.current[SupportedChainId.SOLANA]?.[ACCOUNT.toLowerCase()]).toBeUndefined()
+  })
+
+  it('derives a Token-2022 ATA for mints tagged as Token-2022 in the token list', async () => {
+    mockTokensByAddress = { [getAddressKey(MINT_A)]: { tags: [TOKEN_2022_TAG] } }
+    mockAmountByAta = { [ataKey(MINT_A, TOKEN_2022_PROGRAM)]: 999n }
+    mockInfoByAta = { [ataKey(MINT_A, TOKEN_2022_PROGRAM)]: { present: true } }
+
+    const { result } = renderWithBalances(makeParams({ tokenAddresses: [MINT_A] }))
+
+    await waitFor(() => expect(result.current.hasFirstLoad).toBe(true))
+
+    const [atas] = mockConnection!.getMultipleAccountsInfo.mock.calls[0]
+    expect(atas[0].toBase58()).toBe(ataKey(MINT_A, TOKEN_2022_PROGRAM))
+    expect(result.current.values[getAddressKey(MINT_A)]).toBe(999n)
+  })
+})