@cityofzion/neon-api
Version:
Neon-API module: High level API for neon-js
33 lines • 1.45 kB
JavaScript
import { sc, u } from "@cityofzion/neon-core";
const CHUNK_SIZE = 2;
export async function getTokenBalances(address, contracts, client) {
const script = contracts
.map((scriptHash) => scriptHash instanceof sc.Nep17Contract
? scriptHash
: new sc.Nep17Contract(scriptHash))
.map((contract) => [contract.decimals(), contract.balanceOf(address)])
.reduce((sb, contractCalls) => {
contractCalls.forEach((cc) => sb.emitContractCall(cc));
return sb;
}, new sc.ScriptBuilder())
.build();
const response = await client.invokeScript(u.HexString.fromHex(script));
if (response.state === "FAULT") {
throw new Error(response.exception
? `Invoke exception: ${response.exception}}`
: "No exception message returned.");
}
const expectedStackLength = contracts.length * CHUNK_SIZE;
if (response.stack.length !== expectedStackLength) {
throw new Error(`Received unexpected results. Expected ${expectedStackLength} but got ${response.stack.length} instead.`);
}
const results = [];
for (let i = 0; i < response.stack.length; i += CHUNK_SIZE) {
results.push(response.stack.slice(i, i + 3));
}
return results.map((result) => {
const decimals = parseInt(result[0].value);
return u.BigInteger.fromNumber(result[1].value).toDecimal(decimals);
});
}
//# sourceMappingURL=getTokenBalances.js.map