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

Added getIncentivesController() getter to AToken and Variable Debt Token

Browse Source
This commit is contained in:
David Racero 2021-04-12 14:17:57 +02:00
parent de22637c31
commit 3ac461315f
4 changed files with 29 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
contracts/interfaces/IAToken.sol
View File

@ -3,6 +3,7 @@ pragma solidity 0.6.12;
import {IERC20} from '../dependencies/openzeppelin/contracts/IERC20.sol'; import {IERC20} from '../dependencies/openzeppelin/contracts/IERC20.sol';
import {IScaledBalanceToken} from './IScaledBalanceToken.sol'; import {IScaledBalanceToken} from './IScaledBalanceToken.sol';
import {IAaveIncentivesController} from './IAaveIncentivesController.sol';
interface IAToken is IERC20, IScaledBalanceToken { interface IAToken is IERC20, IScaledBalanceToken {
/** /**
@ -85,4 +86,9 @@ interface IAToken is IERC20, IScaledBalanceToken {
* @return The amount transferred * @return The amount transferred
**/ **/
function transferUnderlyingTo(address user, uint256 amount) external returns (uint256); function transferUnderlyingTo(address user, uint256 amount) external returns (uint256);
/**
* @dev Returns the address of the incentives controller contract
**/
function getIncentivesController() external view returns (IAaveIncentivesController);
} }

6
contracts/interfaces/IVariableDebtToken.sol
View File

@ -2,6 +2,7 @@
pragma solidity 0.6.12; pragma solidity 0.6.12;
import {IScaledBalanceToken} from './IScaledBalanceToken.sol'; import {IScaledBalanceToken} from './IScaledBalanceToken.sol';
import {IAaveIncentivesController} from './IAaveIncentivesController.sol';
/** /**
* @title IVariableDebtToken * @title IVariableDebtToken
@ -52,4 +53,9 @@ interface IVariableDebtToken is IScaledBalanceToken {
uint256 amount, uint256 amount,
uint256 index uint256 index
) external; ) external;
/**
* @dev Returns the address of the incentives controller contract
**/
function getIncentivesController() external view returns (IAaveIncentivesController);
} }

8
contracts/protocol/tokenization/AToken.sol
View File

@ -9,6 +9,7 @@ import {WadRayMath} from '../libraries/math/WadRayMath.sol';
import {Errors} from '../libraries/helpers/Errors.sol'; import {Errors} from '../libraries/helpers/Errors.sol';
import {VersionedInitializable} from '../libraries/aave-upgradeability/VersionedInitializable.sol'; import {VersionedInitializable} from '../libraries/aave-upgradeability/VersionedInitializable.sol';
import {IncentivizedERC20} from './IncentivizedERC20.sol'; import {IncentivizedERC20} from './IncentivizedERC20.sol';
import {IAaveIncentivesController} from '../../interfaces/IAaveIncentivesController.sol';
/** /**
* @title Aave ERC20 AToken * @title Aave ERC20 AToken
@ -229,6 +230,13 @@ contract AToken is VersionedInitializable, IncentivizedERC20, IAToken {
return currentSupplyScaled.rayMul(POOL.getReserveNormalizedIncome(UNDERLYING_ASSET_ADDRESS)); return currentSupplyScaled.rayMul(POOL.getReserveNormalizedIncome(UNDERLYING_ASSET_ADDRESS));
} }
/**
* @dev Returns the address of the incentives controller contract
**/
function getIncentivesController() external view override returns (IAaveIncentivesController) {
return _incentivesController;
}
/** /**
* @dev Returns the scaled total supply of the variable debt token. Represents sum(debt/index) * @dev Returns the scaled total supply of the variable debt token. Represents sum(debt/index)
* @return the scaled total supply * @return the scaled total supply

9
contracts/protocol/tokenization/VariableDebtToken.sol
View File

@ -5,6 +5,7 @@ import {IVariableDebtToken} from '../../interfaces/IVariableDebtToken.sol';
import {WadRayMath} from '../libraries/math/WadRayMath.sol'; import {WadRayMath} from '../libraries/math/WadRayMath.sol';
import {Errors} from '../libraries/helpers/Errors.sol'; import {Errors} from '../libraries/helpers/Errors.sol';
import {DebtTokenBase} from './base/DebtTokenBase.sol'; import {DebtTokenBase} from './base/DebtTokenBase.sol';
import {IAaveIncentivesController} from '../../interfaces/IAaveIncentivesController.sol';
/** /**
* @title VariableDebtToken * @title VariableDebtToken
@ -139,4 +140,12 @@ contract VariableDebtToken is DebtTokenBase, IVariableDebtToken {
{ {
return (super.balanceOf(user), super.totalSupply()); return (super.balanceOf(user), super.totalSupply());
} }
/**
* @dev Returns the address of the incentives controller contract
* @return incentives address
**/
function getIncentivesController() external view override returns (IAaveIncentivesController) {
return _incentivesController;
}
} }