Instadapp/aave-protocol-v2
1
0
Fork 0
mirror of https://github.com/Instadapp/aave-protocol-v2.git synced 2024-07-29 21:47:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

add eth to the wallet balance provider

Browse Source
This commit is contained in:
andyk 2020-11-06 15:51:08 +03:00
parent 0f93c549eb
commit e32b62fb6f
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
contracts/misc/WalletBalanceProvider.sol
View File

@ -25,6 +25,7 @@ contract WalletBalanceProvider {
using ReserveConfiguration for ReserveConfiguration.Map; using ReserveConfiguration for ReserveConfiguration.Map;
LendingPoolAddressesProvider internal immutable _provider; LendingPoolAddressesProvider internal immutable _provider;
address constant MOCK_ETH_ADDRESS = '0xEeeeeEeeeEeEeeEeEeEeeEEEeeeeEeeeeeeeEEeE';
constructor(LendingPoolAddressesProvider provider) public { constructor(LendingPoolAddressesProvider provider) public {
_provider = provider; _provider = provider;
@ -91,11 +92,16 @@ contract WalletBalanceProvider {
ILendingPool pool = ILendingPool(_provider.getLendingPool()); ILendingPool pool = ILendingPool(_provider.getLendingPool());
address[] memory reserves = pool.getReservesList(); address[] memory reserves = pool.getReservesList();
address[] memory reservesWithEth = new address[](reserves.length + 1);
for (uint256 i = 0; i < reserves.length; i++) {
reservesWithEth[i] = reserves[i];
}
reservesWithEth[reserves.length] = MOCK_ETH_ADDRESS;
uint256[] memory balances = new uint256[](reserves.length); uint256[] memory balances = new uint256[](reservesWithEth.length);
for (uint256 j = 0; j < reserves.length; j++) { for (uint256 j = 0; j < reservesWithEth.length; j++) {
ReserveConfiguration.Map memory configuration = pool.getConfiguration(reserves[j]); ReserveConfiguration.Map memory configuration = pool.getConfiguration(reservesWithEth[j]);
(bool isActive, , , ) = configuration.getFlagsMemory(); (bool isActive, , , ) = configuration.getFlagsMemory();