59 lines
1.7 KiB
TypeScript
59 lines
1.7 KiB
TypeScript
|
import BN from 'bn.js'
|
||
|
|
import useWalletStore from '../stores/useWalletStore'
|
||
|
|
import { ProgramAccount, TokenAccount } from '../utils/tokens'
|
||
|
|
|
||
|
|
function fixedPointToNumber(value: BN, decimals: number) {
|
||
|
|
const divisor = new BN(10).pow(new BN(decimals))
|
||
|
|
const quotient = value.div(divisor)
|
||
|
|
const remainder = value.mod(divisor)
|
||
|
|
return quotient.toNumber() + remainder.toNumber() / divisor.toNumber()
|
||
|
|
}
|
||
|
|
|
||
|
|
function calculateBalance(mints, account: TokenAccount): number {
|
||
|
|
const mint = mints[account.mint.toBase58()]
|
||
|
|
return mint ? fixedPointToNumber(account.amount, mint.decimals) : 0
|
||
|
|
}
|
||
|
|
|
||
|
|
export function findLargestBalanceAccountForMint(
|
||
|
|
mints,
|
||
|
|
tokenAccounts: ProgramAccount<TokenAccount>[],
|
||
|
|
mintPk
|
||
|
|
) {
|
||
|
|
const accounts = tokenAccounts.filter((a) => a.account.mint.equals(mintPk))
|
||
|
|
if (!accounts.length) return undefined
|
||
|
|
|
||
|
|
const balances = accounts.map((a) => calculateBalance(mints, a.account))
|
||
|
|
const maxBalanceAccountIndex = balances.reduce(
|
||
|
|
(iMax, bal, iBal) => (bal > balances[iMax] ? iBal : iMax),
|
||
|
|
0
|
||
|
|
)
|
||
|
|
const account = accounts[maxBalanceAccountIndex]
|
||
|
|
const balance = balances[maxBalanceAccountIndex]
|
||
|
|
|
||
|
|
console.log(
|
||
|
|
'findLargestBalanceAccountForMint',
|
||
|
|
maxBalanceAccountIndex,
|
||
|
|
account,
|
||
|
|
balance
|
||
|
|
)
|
||
|
|
|
||
|
|
return { account, balance }
|
||
|
|
}
|
||
|
|
|
||
|
|
export default function useLargestAccounts() {
|
||
|
|
const { pool, tokenAccounts, mints, usdcVault } = useWalletStore(
|
||
|
|
(state) => state
|
||
|
|
)
|
||
|
|
const usdc = usdcVault
|
||
|
|
? findLargestBalanceAccountForMint(mints, tokenAccounts, usdcVault.mint)
|
||
|
|
: undefined
|
||
|
|
const redeemable = pool
|
||
|
|
? findLargestBalanceAccountForMint(
|
||
|
|
mints,
|
||
|
|
tokenAccounts,
|
||
|
|
pool.redeemableMint
|
||
|
|
)
|
||
|
|
: undefined
|
||
|
|
return { usdc, redeemable }
|
||
|
|
}
|