@openocean.finance/widget-sdk/src/_esm/core/Solana/getSolanaBalance.js

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

import { PublicKey } from '@solana/web3.js';
import { SolSystemProgram } from '../../constants.js';
import { withDedupe } from '../../utils/withDedupe.js';
import { callSolanaWithRetry } from './connection.js';
import { TokenProgramAddress } from './types.js';
export const getSolanaBalance = async (walletAddress, tokens) => {
    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.');
        }
    }
    return getSolanaBalanceDefault(chainId, tokens, walletAddress);
};
const getSolanaBalanceDefault = async (_chainId, tokens, walletAddress) => {
    const accountPublicKey = new PublicKey(walletAddress);
    const tokenProgramPublicKey = new PublicKey(TokenProgramAddress);
    const [slot, balance, tokenAccountsByOwner] = await Promise.allSettled([
        withDedupe(() => callSolanaWithRetry((connection) => connection.getSlot('confirmed')), { id: `${getSolanaBalanceDefault.name}.getSlot` }),
        withDedupe(() => callSolanaWithRetry((connection) => connection.getBalance(accountPublicKey, 'confirmed')), { id: `${getSolanaBalanceDefault.name}.getBalance` }),
        withDedupe(() => callSolanaWithRetry((connection) => connection.getParsedTokenAccountsByOwner(accountPublicKey, {
            programId: tokenProgramPublicKey,
        }, 'confirmed')), { id: `${getSolanaBalanceDefault.name}.getParsedTokenAccountsByOwner` }),
    ]);
    const blockNumber = slot.status === 'fulfilled' ? BigInt(slot.value) : 0n;
    const solBalance = balance.status === 'fulfilled' ? BigInt(balance.value) : 0n;
    const walletTokenAmounts = tokenAccountsByOwner.status === 'fulfilled'
        ? tokenAccountsByOwner.value.value.reduce((tokenAmounts, value) => {
            const amount = BigInt(value.account.data.parsed.info.tokenAmount.amount);
            if (amount > 0n) {
                tokenAmounts[value.account.data.parsed.info.mint] = amount;
            }
            return tokenAmounts;
        }, {})
        : {};
    walletTokenAmounts[SolSystemProgram] = solBalance;
    const tokenAmounts = tokens.map((token) => {
        if (walletTokenAmounts[token.address]) {
            return {
                ...token,
                amount: walletTokenAmounts[token.address],
                blockNumber,
            };
        }
        return {
            ...token,
            blockNumber,
        };
    });
    return tokenAmounts;
};
//# sourceMappingURL=getSolanaBalance.js.map