@openocean.finance/widget-sdk/src/core/UTXO/getUTXOBalance.ts

OpenOcean Any-to-Any Cross-Chain-Swap SDK

import {
  ChainId,
  type Token,
  type TokenAmount,
} from '@openocean.finance/widget-types'
import { getUTXOPublicClient } from './getUTXOPublicClient.js'

export const getUTXOBalance = async (
  walletAddress: string,
  tokens: Token[]
): Promise<TokenAmount[]> => {
  if (tokens.length === 0) {
    return []
  }
  const { chainId } = tokens[0]
  for (const token of tokens) {
    if (token.chainId !== chainId) {
      console.warn('Requested tokens have to be on the same chain.')
    }
  }
  const client = await getUTXOPublicClient(ChainId.BTC)
  const [balance, blockCount] = await Promise.all([
    client.getBalance({ address: walletAddress }),
    client.getBlockCount(),
  ])

  return tokens.map((token) => ({
    ...token,
    amount: balance,
    blockNumber: BigInt(blockCount),
  }))
}