@ledgerhq/coin-tron/lib-es/logic/getBalance.js

Ledger Tron Coin integration

import BigNumber from "bignumber.js";
import { fetchTronAccount } from "../network";
import { getTronResources } from "./utils";
const bigIntOrZero = (val) => BigInt(val?.toString() ?? 0);
export async function getBalance(address) {
    const accounts = await fetchTronAccount(address);
    const account = accounts[0];
    return {
        native: computeBalance(account),
        tokens: extractTrc10Balance(account).concat(extractTrc20Balance(account)),
    };
}
function extractTrc10Balance(account) {
    return (account.assetV2?.map(trc => {
        return {
            standard: "trc10",
            contractAddress: trc.key,
            balance: BigInt(trc.value),
        };
    }) ?? []);
}
function extractTrc20Balance(account) {
    return account.trc20.map(trc => {
        const [[contractAddress, balance]] = Object.entries(trc);
        return {
            standard: "trc20",
            contractAddress,
            balance: BigInt(balance),
        };
    });
}
export function computeBalance(account) {
    const tronResources = getTronResources(account);
    let balance = bigIntOrZero(account.balance ?? 0);
    balance += bigIntOrZero(tronResources.frozen.bandwidth?.amount);
    balance += bigIntOrZero(tronResources.frozen.energy?.amount);
    balance += bigIntOrZero(tronResources.delegatedFrozen.bandwidth?.amount);
    balance += bigIntOrZero(tronResources.delegatedFrozen.energy?.amount);
    balance += tronResources.unFrozen.energy
        ? tronResources.unFrozen.energy.reduce((accum, cur) => {
            return accum + BigInt(cur.amount.toString());
        }, BigInt(0))
        : BigInt(0);
    balance += tronResources.unFrozen.bandwidth
        ? tronResources.unFrozen.bandwidth.reduce((accum, cur) => {
            return accum + BigInt(cur.amount.toString());
        }, BigInt(0))
        : BigInt(0);
    balance += bigIntOrZero(tronResources.legacyFrozen.bandwidth?.amount);
    balance += bigIntOrZero(tronResources.legacyFrozen.energy?.amount);
    return balance;
}
export function computeBalanceBridge(account) {
    const tronResources = getTronResources(account);
    const spendableBalance = account.balance ? new BigNumber(account.balance) : new BigNumber(0);
    const balance = spendableBalance
        .plus(tronResources.frozen.bandwidth ? tronResources.frozen.bandwidth.amount : new BigNumber(0))
        .plus(tronResources.frozen.energy ? tronResources.frozen.energy.amount : new BigNumber(0))
        .plus(tronResources.delegatedFrozen.bandwidth
        ? tronResources.delegatedFrozen.bandwidth.amount
        : new BigNumber(0))
        .plus(tronResources.delegatedFrozen.energy
        ? tronResources.delegatedFrozen.energy.amount
        : new BigNumber(0))
        .plus(tronResources.unFrozen.energy
        ? tronResources.unFrozen.energy.reduce((accum, cur) => {
            return accum.plus(cur.amount);
        }, new BigNumber(0))
        : new BigNumber(0))
        .plus(tronResources.unFrozen.bandwidth
        ? tronResources.unFrozen.bandwidth.reduce((accum, cur) => {
            return accum.plus(cur.amount);
        }, new BigNumber(0))
        : new BigNumber(0))
        .plus(tronResources.legacyFrozen.bandwidth
        ? tronResources.legacyFrozen.bandwidth.amount
        : new BigNumber(0))
        .plus(tronResources.legacyFrozen.energy
        ? tronResources.legacyFrozen.energy.amount
        : new BigNumber(0));
    return balance;
}
//# sourceMappingURL=getBalance.js.map