diff --git a/contracts/misc/UiIncentiveDataProvider.sol b/contracts/misc/UiIncentiveDataProvider.sol index e92111d6..4b8372b2 100644 --- a/contracts/misc/UiIncentiveDataProvider.sol +++ b/contracts/misc/UiIncentiveDataProvider.sol @@ -70,6 +70,7 @@ contract UiIncentiveDataProvider is IUiIncentiveDataProvider { aTokenIncentiveController.DISTRIBUTION_END(), baseData.aTokenAddress, aRewardToken, + address(aTokenIncentiveController), IERC20Detailed(aRewardToken).decimals() ); } @@ -93,6 +94,7 @@ contract UiIncentiveDataProvider is IUiIncentiveDataProvider { sTokenIncentiveController.DISTRIBUTION_END(), baseData.stableDebtTokenAddress, sRewardToken, + address(sTokenIncentiveController), IERC20Detailed(sRewardToken).decimals() ); } @@ -116,6 +118,7 @@ contract UiIncentiveDataProvider is IUiIncentiveDataProvider { vTokenIncentiveController.DISTRIBUTION_END(), baseData.variableDebtTokenAddress, vRewardToken, + address(vTokenIncentiveController), IERC20Detailed(vRewardToken).decimals() ); } @@ -165,6 +168,7 @@ contract UiIncentiveDataProvider is IUiIncentiveDataProvider { ); aUserIncentiveData.tokenAddress = baseData.aTokenAddress; aUserIncentiveData.rewardTokenAddress = aRewardToken; + aUserIncentiveData.incentiveControllerAddress = address(aTokenIncentiveController); aUserIncentiveData.rewardTokenDecimals = IERC20Detailed(aRewardToken).decimals(); } @@ -185,6 +189,7 @@ contract UiIncentiveDataProvider is IUiIncentiveDataProvider { ); vUserIncentiveData.tokenAddress = baseData.variableDebtTokenAddress; vUserIncentiveData.rewardTokenAddress = vRewardToken; + vUserIncentiveData.incentiveControllerAddress = address(vTokenIncentiveController); vUserIncentiveData.rewardTokenDecimals = IERC20Detailed(vRewardToken).decimals(); } @@ -205,6 +210,7 @@ contract UiIncentiveDataProvider is IUiIncentiveDataProvider { ); sUserIncentiveData.tokenAddress = baseData.stableDebtTokenAddress; sUserIncentiveData.rewardTokenAddress = sRewardToken; + sUserIncentiveData.incentiveControllerAddress = address(sTokenIncentiveController); sUserIncentiveData.rewardTokenDecimals = IERC20Detailed(sRewardToken).decimals(); } diff --git a/contracts/misc/interfaces/IUiIncentiveDataProvider.sol b/contracts/misc/interfaces/IUiIncentiveDataProvider.sol index ed062bf5..c9d7f5d0 100644 --- a/contracts/misc/interfaces/IUiIncentiveDataProvider.sol +++ b/contracts/misc/interfaces/IUiIncentiveDataProvider.sol @@ -19,6 +19,7 @@ interface IUiIncentiveDataProvider { uint256 emissionEndTimestamp; address tokenAddress; address rewardTokenAddress; + address incentiveControllerAddress; uint8 rewardTokenDecimals; } @@ -34,6 +35,7 @@ interface IUiIncentiveDataProvider { uint256 userUnclaimedRewards; address tokenAddress; address rewardTokenAddress; + address incentiveControllerAddress; uint8 rewardTokenDecimals; }