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

Ledger Tron Coin integration

"use strict";
var __importDefault = (this && this.__importDefault) || function (mod) {
    return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.getBalance = getBalance;
exports.computeBalance = computeBalance;
exports.computeBalanceBridge = computeBalanceBridge;
const bignumber_js_1 = __importDefault(require("bignumber.js"));
const network_1 = require("../network");
const utils_1 = require("./utils");
const bigIntOrZero = (val) => BigInt(val?.toString() ?? 0);
async function getBalance(address) {
    const accounts = await (0, network_1.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),
        };
    });
}
function computeBalance(account) {
    const tronResources = (0, utils_1.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;
}
function computeBalanceBridge(account) {
    const tronResources = (0, utils_1.getTronResources)(account);
    const spendableBalance = account.balance ? new bignumber_js_1.default(account.balance) : new bignumber_js_1.default(0);
    const balance = spendableBalance
        .plus(tronResources.frozen.bandwidth ? tronResources.frozen.bandwidth.amount : new bignumber_js_1.default(0))
        .plus(tronResources.frozen.energy ? tronResources.frozen.energy.amount : new bignumber_js_1.default(0))
        .plus(tronResources.delegatedFrozen.bandwidth
        ? tronResources.delegatedFrozen.bandwidth.amount
        : new bignumber_js_1.default(0))
        .plus(tronResources.delegatedFrozen.energy
        ? tronResources.delegatedFrozen.energy.amount
        : new bignumber_js_1.default(0))
        .plus(tronResources.unFrozen.energy
        ? tronResources.unFrozen.energy.reduce((accum, cur) => {
            return accum.plus(cur.amount);
        }, new bignumber_js_1.default(0))
        : new bignumber_js_1.default(0))
        .plus(tronResources.unFrozen.bandwidth
        ? tronResources.unFrozen.bandwidth.reduce((accum, cur) => {
            return accum.plus(cur.amount);
        }, new bignumber_js_1.default(0))
        : new bignumber_js_1.default(0))
        .plus(tronResources.legacyFrozen.bandwidth
        ? tronResources.legacyFrozen.bandwidth.amount
        : new bignumber_js_1.default(0))
        .plus(tronResources.legacyFrozen.energy
        ? tronResources.legacyFrozen.energy.amount
        : new bignumber_js_1.default(0));
    return balance;
}
//# sourceMappingURL=getBalance.js.map