@lifi/sdk 1.1.1 → 1.1.4

package/dist/allowance/index.js

@@ -1,4 +1,3 @@
-"use strict";
 var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
     function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
     return new (P || (P = Promise))(function (resolve, reject) {
@@ -8,153 +7,62 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
         step((generator = generator.apply(thisArg, _arguments || [])).next());
     });
 };
-var __generator = (this && this.__generator) || function (thisArg, body) {
-    var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g;
-    return g = { next: verb(0), "throw": verb(1), "return": verb(2) }, typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g;
-    function verb(n) { return function (v) { return step([n, v]); }; }
-    function step(op) {
-        if (f) throw new TypeError("Generator is already executing.");
-        while (_) try {
-            if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
-            if (y = 0, t) op = [op[0] & 2, t.value];
-            switch (op[0]) {
-                case 0: case 1: t = op; break;
-                case 4: _.label++; return { value: op[1], done: false };
-                case 5: _.label++; y = op[1]; op = [0]; continue;
-                case 7: op = _.ops.pop(); _.trys.pop(); continue;
-                default:
-                    if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; }
-                    if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; }
-                    if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; }
-                    if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; }
-                    if (t[2]) _.ops.pop();
-                    _.trys.pop(); continue;
-            }
-            op = body.call(thisArg, _);
-        } catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; }
-        if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
+import BigNumber from 'bignumber.js';
+import { constants } from 'ethers';
+import { isSameToken } from '../helpers';
+import { isNativeTokenAddress } from '../utils/utils';
+import { getAllowanceViaMulticall, getApproved, groupByChain, setApproval, } from './utils';
+export const getTokenApproval = (signer, token, approvalAddress) => __awaiter(void 0, void 0, void 0, function* () {
+    // native token don't need approval
+    if (isNativeTokenAddress(token.address)) {
+        return;
     }
-};
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.revokeTokenApproval = exports.approveToken = exports.bulkGetTokenApproval = exports.getTokenApproval = void 0;
-var bignumber_js_1 = __importDefault(require("bignumber.js"));
-var ethers_1 = require("ethers");
-var helpers_1 = require("../helpers");
-var utils_1 = require("../utils/utils");
-var utils_2 = require("./utils");
-var getTokenApproval = function (signer, token, approvalAddress) { return __awaiter(void 0, void 0, void 0, function () {
-    var approved;
-    return __generator(this, function (_a) {
-        switch (_a.label) {
-            case 0:
-                // native token don't need approval
-                if ((0, utils_1.isNativeTokenAddress)(token.address)) {
-                    return [2 /*return*/];
-                }
-                return [4 /*yield*/, (0, utils_2.getApproved)(signer, token.address, approvalAddress)];
-            case 1:
-                approved = _a.sent();
-                return [2 /*return*/, approved.toString()];
-        }
-    });
-}); };
-exports.getTokenApproval = getTokenApproval;
-var bulkGetTokenApproval = function (signer, tokenData) { return __awaiter(void 0, void 0, void 0, function () {
-    var filteredTokenData, tokenDataByChain, approvalPromises, approvalsByChain, approvals;
-    return __generator(this, function (_a) {
-        switch (_a.label) {
-            case 0:
-                filteredTokenData = tokenData.filter(function (_a) {
-                    var token = _a.token;
-                    return !(0, utils_1.isNativeTokenAddress)(token.address);
-                });
-                tokenDataByChain = (0, utils_2.groupByChain)(filteredTokenData);
-                approvalPromises = Object.keys(tokenDataByChain).map(function (chainId) { return __awaiter(void 0, void 0, void 0, function () {
-                    var parsedChainId;
-                    return __generator(this, function (_a) {
-                        parsedChainId = Number.parseInt(chainId);
-                        // get allowances for current chain and token list
-                        return [2 /*return*/, (0, utils_2.getAllowanceViaMulticall)(signer, parsedChainId, tokenDataByChain[parsedChainId])];
-                    });
-                }); });
-                return [4 /*yield*/, Promise.all(approvalPromises)];
-            case 1:
-                approvalsByChain = _a.sent();
-                approvals = approvalsByChain.flat();
-                return [2 /*return*/, tokenData.map(function (_a) {
-                        var token = _a.token;
-                        // native token don't need approval
-                        if ((0, utils_1.isNativeTokenAddress)(token.address)) {
-                            return { token: token, approval: undefined };
-                        }
-                        var approved = approvals.find(function (approval) {
-                            return (0, helpers_1.isSameToken)(approval.token, token);
-                        });
-                        return { token: token, approval: approved === null || approved === void 0 ? void 0 : approved.approvedAmount.toString() };
-                    })];
+    const approved = yield getApproved(signer, token.address, approvalAddress);
+    return approved.toString();
+});
+export const bulkGetTokenApproval = (signer, tokenData) => __awaiter(void 0, void 0, void 0, function* () {
+    // filter out native tokens
+    const filteredTokenData = tokenData.filter(({ token }) => !isNativeTokenAddress(token.address));
+    // group by chain
+    const tokenDataByChain = groupByChain(filteredTokenData);
+    const approvalPromises = Object.keys(tokenDataByChain).map((chainId) => __awaiter(void 0, void 0, void 0, function* () {
+        const parsedChainId = Number.parseInt(chainId);
+        // get allowances for current chain and token list
+        return getAllowanceViaMulticall(signer, parsedChainId, tokenDataByChain[parsedChainId]);
+    }));
+    const approvalsByChain = yield Promise.all(approvalPromises);
+    const approvals = approvalsByChain.flat();
+    return tokenData.map(({ token }) => {
+        // native token don't need approval
+        if (isNativeTokenAddress(token.address)) {
+            return { token, approval: undefined };
         }
+        const approved = approvals.find((approval) => isSameToken(approval.token, token));
+        return { token, approval: approved === null || approved === void 0 ? void 0 : approved.approvedAmount.toString() };
     });
-}); };
-exports.bulkGetTokenApproval = bulkGetTokenApproval;
-var approveToken = function (_a) {
-    var signer = _a.signer, token = _a.token, approvalAddress = _a.approvalAddress, amount = _a.amount, _b = _a.infiniteApproval, infiniteApproval = _b === void 0 ? false : _b;
-    return __awaiter(void 0, void 0, void 0, function () {
-        var approvedAmount, approvalAmount, approveTx;
-        return __generator(this, function (_c) {
-            switch (_c.label) {
-                case 0:
-                    // native token don't need approval
-                    if ((0, utils_1.isNativeTokenAddress)(token.address)) {
-                        return [2 /*return*/];
-                    }
-                    return [4 /*yield*/, (0, utils_2.getApproved)(signer, token.address, approvalAddress)];
-                case 1:
-                    approvedAmount = _c.sent();
-                    if (!new bignumber_js_1.default(amount).gt(approvedAmount)) return [3 /*break*/, 4];
-                    approvalAmount = infiniteApproval
-                        ? ethers_1.constants.MaxUint256.toString()
-                        : amount;
-                    return [4 /*yield*/, (0, utils_2.setApproval)(signer, token.address, approvalAddress, approvalAmount)];
-                case 2:
-                    approveTx = _c.sent();
-                    return [4 /*yield*/, approveTx.wait()];
-                case 3:
-                    _c.sent();
-                    _c.label = 4;
-                case 4: return [2 /*return*/];
-            }
-        });
-    });
-};
-exports.approveToken = approveToken;
-var revokeTokenApproval = function (_a) {
-    var signer = _a.signer, token = _a.token, approvalAddress = _a.approvalAddress;
-    return __awaiter(void 0, void 0, void 0, function () {
-        var approvedAmount, approveTx;
-        return __generator(this, function (_b) {
-            switch (_b.label) {
-                case 0:
-                    // native token don't need approval
-                    if ((0, utils_1.isNativeTokenAddress)(token.address)) {
-                        return [2 /*return*/];
-                    }
-                    return [4 /*yield*/, (0, utils_2.getApproved)(signer, token.address, approvalAddress)];
-                case 1:
-                    approvedAmount = _b.sent();
-                    if (!!approvedAmount.isZero()) return [3 /*break*/, 4];
-                    return [4 /*yield*/, (0, utils_2.setApproval)(signer, token.address, approvalAddress, '0')];
-                case 2:
-                    approveTx = _b.sent();
-                    return [4 /*yield*/, approveTx.wait()];
-                case 3:
-                    _b.sent();
-                    _b.label = 4;
-                case 4: return [2 /*return*/];
-            }
-        });
-    });
-};
-exports.revokeTokenApproval = revokeTokenApproval;
+});
+export const approveToken = ({ signer, token, approvalAddress, amount, infiniteApproval = false, }) => __awaiter(void 0, void 0, void 0, function* () {
+    // native token don't need approval
+    if (isNativeTokenAddress(token.address)) {
+        return;
+    }
+    const approvedAmount = yield getApproved(signer, token.address, approvalAddress);
+    if (new BigNumber(amount).gt(approvedAmount)) {
+        const approvalAmount = infiniteApproval
+            ? constants.MaxUint256.toString()
+            : amount;
+        const approveTx = yield setApproval(signer, token.address, approvalAddress, approvalAmount);
+        yield approveTx.wait();
+    }
+});
+export const revokeTokenApproval = ({ signer, token, approvalAddress, }) => __awaiter(void 0, void 0, void 0, function* () {
+    // native token don't need approval
+    if (isNativeTokenAddress(token.address)) {
+        return;
+    }
+    const approvedAmount = yield getApproved(signer, token.address, approvalAddress);
+    if (!approvedAmount.isZero()) {
+        const approveTx = yield setApproval(signer, token.address, approvalAddress, '0');
+        yield approveTx.wait();
+    }
+});

package/dist/allowance/utils.js

@@ -1,4 +1,3 @@
-"use strict";
 var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
     function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
     return new (P || (P = Promise))(function (resolve, reject) {
@@ -8,125 +7,62 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
         step((generator = generator.apply(thisArg, _arguments || [])).next());
     });
 };
-var __generator = (this && this.__generator) || function (thisArg, body) {
-    var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g;
-    return g = { next: verb(0), "throw": verb(1), "return": verb(2) }, typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g;
-    function verb(n) { return function (v) { return step([n, v]); }; }
-    function step(op) {
-        if (f) throw new TypeError("Generator is already executing.");
-        while (_) try {
-            if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
-            if (y = 0, t) op = [op[0] & 2, t.value];
-            switch (op[0]) {
-                case 0: case 1: t = op; break;
-                case 4: _.label++; return { value: op[1], done: false };
-                case 5: _.label++; y = op[1]; op = [0]; continue;
-                case 7: op = _.ops.pop(); _.trys.pop(); continue;
-                default:
-                    if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; }
-                    if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; }
-                    if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; }
-                    if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; }
-                    if (t[2]) _.ops.pop();
-                    _.trys.pop(); continue;
-            }
-            op = body.call(thisArg, _);
-        } catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; }
-        if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
+import BigNumber from 'bignumber.js';
+import { Contract } from 'ethers';
+import ChainsService from '../services/ChainsService';
+import { ERC20_ABI } from '../types';
+import { ServerError } from '../utils/errors';
+import { fetchDataUsingMulticall } from '../utils/multicall';
+export const getApproved = (signer, tokenAddress, contractAddress) => __awaiter(void 0, void 0, void 0, function* () {
+    const signerAddress = yield signer.getAddress();
+    const erc20 = new Contract(tokenAddress, ERC20_ABI, signer);
+    try {
+        const approved = yield erc20.allowance(signerAddress, contractAddress);
+        return new BigNumber(approved.toString());
     }
-};
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.groupByChain = exports.getAllowanceViaMulticall = exports.setApproval = exports.getApproved = void 0;
-var bignumber_js_1 = __importDefault(require("bignumber.js"));
-var ethers_1 = require("ethers");
-var ChainsService_1 = __importDefault(require("../services/ChainsService"));
-var types_1 = require("../types");
-var errors_1 = require("../utils/errors");
-var multicall_1 = require("../utils/multicall");
-var getApproved = function (signer, tokenAddress, contractAddress) { return __awaiter(void 0, void 0, void 0, function () {
-    var signerAddress, erc20, approved, e_1;
-    return __generator(this, function (_a) {
-        switch (_a.label) {
-            case 0: return [4 /*yield*/, signer.getAddress()];
-            case 1:
-                signerAddress = _a.sent();
-                erc20 = new ethers_1.Contract(tokenAddress, types_1.ERC20_ABI, signer);
-                _a.label = 2;
-            case 2:
-                _a.trys.push([2, 4, , 5]);
-                return [4 /*yield*/, erc20.allowance(signerAddress, contractAddress)];
-            case 3:
-                approved = _a.sent();
-                return [2 /*return*/, new bignumber_js_1.default(approved.toString())];
-            case 4:
-                e_1 = _a.sent();
-                return [2 /*return*/, new bignumber_js_1.default(0)];
-            case 5: return [2 /*return*/];
-        }
-    });
-}); };
-exports.getApproved = getApproved;
-var setApproval = function (signer, tokenAddress, contractAddress, amount) {
-    var erc20 = new ethers_1.Contract(tokenAddress, types_1.ERC20_ABI, signer);
+    catch (e) {
+        return new BigNumber(0);
+    }
+});
+export const setApproval = (signer, tokenAddress, contractAddress, amount) => {
+    const erc20 = new Contract(tokenAddress, ERC20_ABI, signer);
     return erc20.approve(contractAddress, amount);
 };
-exports.setApproval = setApproval;
-var getAllowanceViaMulticall = function (signer, chainId, tokenData) { return __awaiter(void 0, void 0, void 0, function () {
-    var chainsService, chain, ownerAddress, calls, result, parsedResult;
-    return __generator(this, function (_a) {
-        switch (_a.label) {
-            case 0:
-                chainsService = ChainsService_1.default.getInstance();
-                return [4 /*yield*/, chainsService.getChainById(chainId)];
-            case 1:
-                chain = _a.sent();
-                if (!chain.multicallAddress) {
-                    throw new errors_1.ServerError("No multicall address configured for chainId ".concat(chainId, "."));
-                }
-                return [4 /*yield*/, signer.getAddress()];
-            case 2:
-                ownerAddress = _a.sent();
-                calls = [];
-                tokenData.map(function (_a) {
-                    var token = _a.token, approvalAddress = _a.approvalAddress;
-                    calls.push({
-                        address: token.address,
-                        name: 'allowance',
-                        params: [ownerAddress, approvalAddress],
-                    });
-                });
-                return [4 /*yield*/, (0, multicall_1.fetchDataUsingMulticall)(calls, types_1.ERC20_ABI, chainId, chain.multicallAddress)];
-            case 3:
-                result = _a.sent();
-                if (!result.length) {
-                    throw new errors_1.ServerError("Couldn't load allowance from chainId ".concat(chainId, " using multicall."));
-                }
-                parsedResult = result.map(function (_a) {
-                    var _b;
-                    var data = _a.data;
-                    return ({
-                        approvalAmount: (_b = data) !== null && _b !== void 0 ? _b : new bignumber_js_1.default(0),
-                    });
-                });
-                return [2 /*return*/, tokenData.map(function (_a, i) {
-                        var token = _a.token, approvalAddress = _a.approvalAddress;
-                        return ({
-                            token: token,
-                            approvalAddress: approvalAddress,
-                            approvedAmount: parsedResult[i].approvalAmount,
-                        });
-                    })];
-        }
+export const getAllowanceViaMulticall = (signer, chainId, tokenData) => __awaiter(void 0, void 0, void 0, function* () {
+    const chainsService = ChainsService.getInstance();
+    const chain = yield chainsService.getChainById(chainId);
+    if (!chain.multicallAddress) {
+        throw new ServerError(`No multicall address configured for chainId ${chainId}.`);
+    }
+    const ownerAddress = yield signer.getAddress();
+    const calls = [];
+    tokenData.map(({ token, approvalAddress }) => {
+        calls.push({
+            address: token.address,
+            name: 'allowance',
+            params: [ownerAddress, approvalAddress],
+        });
+    });
+    const result = yield fetchDataUsingMulticall(calls, ERC20_ABI, chainId, chain.multicallAddress);
+    if (!result.length) {
+        throw new ServerError(`Couldn't load allowance from chainId ${chainId} using multicall.`);
+    }
+    const parsedResult = result.map(({ data }) => {
+        var _a;
+        return ({
+            approvalAmount: (_a = data) !== null && _a !== void 0 ? _a : new BigNumber(0),
+        });
     });
-}); };
-exports.getAllowanceViaMulticall = getAllowanceViaMulticall;
-var groupByChain = function (tokenDataList) {
+    return tokenData.map(({ token, approvalAddress }, i) => ({
+        token,
+        approvalAddress,
+        approvedAmount: parsedResult[i].approvalAmount,
+    }));
+});
+export const groupByChain = (tokenDataList) => {
     // group by chain
-    var tokenDataByChain = {};
-    tokenDataList.forEach(function (tokenData) {
+    const tokenDataByChain = {};
+    tokenDataList.forEach((tokenData) => {
         if (!tokenDataByChain[tokenData.token.chainId]) {
             tokenDataByChain[tokenData.token.chainId] = [];
         }
@@ -134,4 +70,3 @@ var groupByChain = function (tokenDataList) {
     });
     return tokenDataByChain;
 };
-exports.groupByChain = groupByChain;

package/dist/balances/index.js

@@ -1,4 +1,3 @@
-"use strict";
 var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
     function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
     return new (P || (P = Promise))(function (resolve, reject) {
@@ -8,97 +7,35 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
         step((generator = generator.apply(thisArg, _arguments || [])).next());
     });
 };
-var __generator = (this && this.__generator) || function (thisArg, body) {
-    var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g;
-    return g = { next: verb(0), "throw": verb(1), "return": verb(2) }, typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g;
-    function verb(n) { return function (v) { return step([n, v]); }; }
-    function step(op) {
-        if (f) throw new TypeError("Generator is already executing.");
-        while (_) try {
-            if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
-            if (y = 0, t) op = [op[0] & 2, t.value];
-            switch (op[0]) {
-                case 0: case 1: t = op; break;
-                case 4: _.label++; return { value: op[1], done: false };
-                case 5: _.label++; y = op[1]; op = [0]; continue;
-                case 7: op = _.ops.pop(); _.trys.pop(); continue;
-                default:
-                    if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; }
-                    if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; }
-                    if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; }
-                    if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; }
-                    if (t[2]) _.ops.pop();
-                    _.trys.pop(); continue;
-            }
-            op = body.call(thisArg, _);
-        } catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; }
-        if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
-    }
-};
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-var utils_1 = __importDefault(require("./utils"));
-var getTokenBalance = function (walletAddress, token) { return __awaiter(void 0, void 0, void 0, function () {
-    var tokenAmounts;
-    return __generator(this, function (_a) {
-        switch (_a.label) {
-            case 0: return [4 /*yield*/, getTokenBalances(walletAddress, [token])];
-            case 1:
-                tokenAmounts = _a.sent();
-                return [2 /*return*/, tokenAmounts.length ? tokenAmounts[0] : null];
-        }
-    });
-}); };
-var getTokenBalances = function (walletAddress, tokens) { return __awaiter(void 0, void 0, void 0, function () {
-    var tokensByChain, tokenAmountsByChain;
-    return __generator(this, function (_a) {
-        switch (_a.label) {
-            case 0:
-                tokensByChain = {};
-                tokens.forEach(function (token) {
-                    if (!tokensByChain[token.chainId]) {
-                        tokensByChain[token.chainId] = [];
-                    }
-                    tokensByChain[token.chainId].push(token);
-                });
-                return [4 /*yield*/, getTokenBalancesForChains(walletAddress, tokensByChain)];
-            case 1:
-                tokenAmountsByChain = _a.sent();
-                return [2 /*return*/, Object.values(tokenAmountsByChain).flat()];
-        }
-    });
-}); };
-var getTokenBalancesForChains = function (walletAddress, tokensByChain) { return __awaiter(void 0, void 0, void 0, function () {
-    var tokenAmountsByChain, promises;
-    return __generator(this, function (_a) {
-        switch (_a.label) {
-            case 0:
-                tokenAmountsByChain = {};
-                promises = Object.keys(tokensByChain).map(function (chainIdStr) { return __awaiter(void 0, void 0, void 0, function () {
-                    var chainId, tokenAmounts;
-                    return __generator(this, function (_a) {
-                        switch (_a.label) {
-                            case 0:
-                                chainId = parseInt(chainIdStr);
-                                return [4 /*yield*/, utils_1.default.getBalances(walletAddress, tokensByChain[chainId])];
-                            case 1:
-                                tokenAmounts = _a.sent();
-                                tokenAmountsByChain[chainId] = tokenAmounts;
-                                return [2 /*return*/];
-                        }
-                    });
-                }); });
-                return [4 /*yield*/, Promise.allSettled(promises)];
-            case 1:
-                _a.sent();
-                return [2 /*return*/, tokenAmountsByChain];
+import utils from './utils';
+const getTokenBalance = (walletAddress, token) => __awaiter(void 0, void 0, void 0, function* () {
+    const tokenAmounts = yield getTokenBalances(walletAddress, [token]);
+    return tokenAmounts.length ? tokenAmounts[0] : null;
+});
+const getTokenBalances = (walletAddress, tokens) => __awaiter(void 0, void 0, void 0, function* () {
+    // split by chain
+    const tokensByChain = {};
+    tokens.forEach((token) => {
+        if (!tokensByChain[token.chainId]) {
+            tokensByChain[token.chainId] = [];
         }
+        tokensByChain[token.chainId].push(token);
     });
-}); };
-exports.default = {
-    getTokenBalance: getTokenBalance,
-    getTokenBalances: getTokenBalances,
-    getTokenBalancesForChains: getTokenBalancesForChains,
+    const tokenAmountsByChain = yield getTokenBalancesForChains(walletAddress, tokensByChain);
+    return Object.values(tokenAmountsByChain).flat();
+});
+const getTokenBalancesForChains = (walletAddress, tokensByChain) => __awaiter(void 0, void 0, void 0, function* () {
+    const tokenAmountsByChain = {};
+    const promises = Object.keys(tokensByChain).map((chainIdStr) => __awaiter(void 0, void 0, void 0, function* () {
+        const chainId = parseInt(chainIdStr);
+        const tokenAmounts = yield utils.getBalances(walletAddress, tokensByChain[chainId]);
+        tokenAmountsByChain[chainId] = tokenAmounts;
+    }));
+    yield Promise.allSettled(promises);
+    return tokenAmountsByChain;
+});
+export default {
+    getTokenBalance,
+    getTokenBalances,
+    getTokenBalancesForChains,
 };