ERC4626 compliance

reserve-protocol · May 7, 2024 · d8a38ed · d8a38ed
1 parent dacc928
commit d8a38ed
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/contracts/rewards/GenericMultiRewardsVault.sol b/contracts/rewards/GenericMultiRewardsVault.sol
@@ -24,6 +24,8 @@ import { RewardInfo, Errors, Events, SCALAR } from "./definitions.sol";
  * - {s} = Seconds
  */
 contract GenericMultiRewardsVault is ERC4626, Ownable {
+ uint256 private totalDeposited; // {qAsset}
+
  constructor(
  string memory _name,
  string memory _symbol,
@@ -34,6 +36,11 @@ contract GenericMultiRewardsVault is ERC4626, Ownable {
  /**
  * Core Vault Functionality
  */
+ function totalAssets() public view override returns (uint256) {
+ // {qAsset} = {qAsset} + {qAsset}
+ return totalDeposited;
+ }
+
  function _convertToShares(uint256 assets, Math.Rounding) internal pure override returns (uint256) {
  return assets;
  }
@@ -49,6 +56,7 @@ contract GenericMultiRewardsVault is ERC4626, Ownable {
  uint256 shares
  ) internal override accrueRewards(caller, receiver) {
  super._deposit(caller, receiver, assets, shares);
+ totalDeposited += assets;
  }
 
  function _withdraw(
@@ -59,6 +67,7 @@ contract GenericMultiRewardsVault is ERC4626, Ownable {
  uint256 shares
  ) internal override accrueRewards(owner_, receiver) {
  super._withdraw(caller, receiver, owner_, assets, shares);
+ totalDeposited -= assets;
  }
 
  /**