fix: Re-add getUserCurrentDebtMemory()

aave · Dec 17, 2021 · 7742985 · 7742985
1 parent 78cfee5
commit 7742985
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/contracts/protocol/libraries/helpers/Helpers.sol b/contracts/protocol/libraries/helpers/Helpers.sol
@@ -28,6 +28,24 @@ library Helpers {
     );
   }
 
+  /**
+   * @notice Fetches the user current stable and variable debt balances
+   * @param user The user address
+   * @param reserve The reserve data object
+   * @return The stable debt balance
+   * @return The variable debt balance
+   **/
+  function getUserCurrentDebtMemory(address user, DataTypes.ReserveData memory reserve)
+    internal
+    view
+    returns (uint256, uint256)
+  {
+    return (
+      IERC20(reserve.stableDebtTokenAddress).balanceOf(user),
+      IERC20(reserve.variableDebtTokenAddress).balanceOf(user)
+    );
+  }
+
   function castUint128(uint256 input) internal pure returns (uint128) {
     require(input <= type(uint128).max, Errors.HLP_UINT128_OVERFLOW);
     return uint128(input);