require('dotenv').config() const BN = require('bignumber.js') const Web3 = require('web3') const logger = require('./logger')('getBalances') const { BRIDGE_MODES } = require('./utils/bridgeMode') const Web3Utils = Web3.utils const { HOME_RPC_URL, FOREIGN_RPC_URL, HOME_BRIDGE_ADDRESS, FOREIGN_BRIDGE_ADDRESS } = process.env const homeProvider = new Web3.providers.HttpProvider(HOME_RPC_URL) const web3Home = new Web3(homeProvider) const foreignProvider = new Web3.providers.HttpProvider(FOREIGN_RPC_URL) const web3Foreign = new Web3(foreignProvider) const ERC20_ABI = require('../contracts/build/contracts/ERC20').abi const ERC677_ABI = require('../contracts/build/contracts/ERC677').abi const HOME_ERC_TO_ERC_ABI = require('../contracts/build/contracts/HomeBridgeErcToErc').abi const HOME_ERC_TO_NATIVE_ABI = require('../contracts/build/contracts/HomeBridgeErcToNative').abi const BLOCK_REWARD_ABI = require('../contracts/build/contracts/IBlockReward').abi const FOREIGN_ERC_TO_ERC_ABI = require('../contracts/build/contracts/ForeignBridgeErcToErc').abi const FOREIGN_ERC_TO_NATIVE_ABI = require('../contracts/build/contracts/ForeignBridgeErcToNative') .abi const FOREIGN_NATIVE_TO_ERC_ABI = require('../contracts/build/contracts/ForeignBridgeNativeToErc') .abi async function main(bridgeMode) { if (bridgeMode === BRIDGE_MODES.ERC_TO_ERC) { const foreignBridge = new web3Foreign.eth.Contract( FOREIGN_ERC_TO_ERC_ABI, FOREIGN_BRIDGE_ADDRESS ) const erc20Address = await foreignBridge.methods.erc20token().call() const erc20Contract = new web3Foreign.eth.Contract(ERC20_ABI, erc20Address) logger.debug('calling erc20Contract.methods.balanceOf') const foreignErc20Balance = await erc20Contract.methods.balanceOf(FOREIGN_BRIDGE_ADDRESS).call() const homeBridge = new web3Home.eth.Contract(HOME_ERC_TO_ERC_ABI, HOME_BRIDGE_ADDRESS) logger.debug('calling homeBridge.methods.erc677token') const tokenAddress = await homeBridge.methods.erc677token().call() const tokenContract = new web3Home.eth.Contract(ERC677_ABI, tokenAddress) logger.debug('calling tokenContract.methods.totalSupply()') const totalSupply = await tokenContract.methods.totalSupply().call() const foreignBalanceBN = new BN(foreignErc20Balance) const foreignTotalSupplyBN = new BN(totalSupply) const diff = foreignBalanceBN.minus(foreignTotalSupplyBN).toString(10) logger.debug('Done') return { home: { totalSupply: Web3Utils.fromWei(totalSupply) }, foreign: { erc20Balance: Web3Utils.fromWei(foreignErc20Balance) }, balanceDiff: Number(Web3Utils.fromWei(diff)), lastChecked: Math.floor(Date.now() / 1000) } } else if ( bridgeMode === BRIDGE_MODES.NATIVE_TO_ERC || bridgeMode === BRIDGE_MODES.NATIVE_TO_ERC_V1 ) { logger.debug('calling web3Home.eth.getBalance') const foreignBridge = new web3Foreign.eth.Contract( FOREIGN_NATIVE_TO_ERC_ABI, FOREIGN_BRIDGE_ADDRESS ) const erc20Address = await foreignBridge.methods.erc677token().call() const homeBalance = await web3Home.eth.getBalance(HOME_BRIDGE_ADDRESS) const tokenContract = new web3Foreign.eth.Contract(ERC20_ABI, erc20Address) logger.debug('calling tokenContract.methods.totalSupply()') const totalSupply = await tokenContract.methods.totalSupply().call() const homeBalanceBN = new BN(homeBalance) const foreignTotalSupplyBN = new BN(totalSupply) const diff = homeBalanceBN.minus(foreignTotalSupplyBN).toString(10) logger.debug('Done') return { home: { balance: Web3Utils.fromWei(homeBalance) }, foreign: { totalSupply: Web3Utils.fromWei(totalSupply) }, balanceDiff: Number(Web3Utils.fromWei(diff)), lastChecked: Math.floor(Date.now() / 1000) } } else if (bridgeMode === BRIDGE_MODES.ERC_TO_NATIVE) { const foreignBridge = new web3Foreign.eth.Contract( FOREIGN_ERC_TO_NATIVE_ABI, FOREIGN_BRIDGE_ADDRESS ) const erc20Address = await foreignBridge.methods.erc20token().call() const erc20Contract = new web3Foreign.eth.Contract(ERC20_ABI, erc20Address) logger.debug('calling erc20Contract.methods.balanceOf') const foreignErc20Balance = await erc20Contract.methods.balanceOf(FOREIGN_BRIDGE_ADDRESS).call() const homeBridge = new web3Home.eth.Contract(HOME_ERC_TO_NATIVE_ABI, HOME_BRIDGE_ADDRESS) logger.debug('calling homeBridge.methods.blockRewardContract') const blockRewardAddress = await homeBridge.methods.blockRewardContract().call() const blockRewardContract = new web3Home.eth.Contract(BLOCK_REWARD_ABI, blockRewardAddress) logger.debug('calling blockReward.methods.mintedTotally') const mintedCoins = await blockRewardContract.methods.mintedTotally().call() logger.debug('calling homeBridge.methods.totalBurntCoins') const burntCoins = await homeBridge.methods.totalBurntCoins().call() const mintedCoinsBN = new BN(mintedCoins) const burntCoinsBN = new BN(burntCoins) const totalSupplyBN = mintedCoinsBN.minus(burntCoinsBN) const foreignErc20BalanceBN = new BN(foreignErc20Balance) const diff = foreignErc20BalanceBN.minus(totalSupplyBN).toFixed() logger.debug('Done') return { home: { totalSupply: Web3Utils.fromWei(totalSupplyBN.toFixed()) }, foreign: { erc20Balance: Web3Utils.fromWei(foreignErc20Balance) }, balanceDiff: Number(Web3Utils.fromWei(diff)), lastChecked: Math.floor(Date.now() / 1000) } } else { throw new Error(`Unrecognized bridge mode: '${bridgeMode}'`) } } module.exports = main