Skip to content

Commit

Permalink
feat: implemente amo interface in Aave AMO to ensure amo standard
Browse files Browse the repository at this point in the history
  • Loading branch information
zugdev committed Sep 19, 2024
1 parent 94f9c4f commit 02bbe8f
Showing 1 changed file with 6 additions and 3 deletions.
9 changes: 6 additions & 3 deletions packages/contracts/src/dollar/amo/AaveAmo.sol
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@
pragma solidity ^0.8.19;

import {UbiquityAmoMinter} from "../core/UbiquityAmoMinter.sol";
import {IAmo} from "../interfaces/IAmo.sol";
import {ERC20} from "@openzeppelin/contracts/token/ERC20/ERC20.sol";
import {SafeERC20} from "@openzeppelin/contracts/token/ERC20/utils/SafeERC20.sol";
import {Ownable} from "@openzeppelin/contracts/access/Ownable.sol";
Expand All @@ -14,7 +15,7 @@ import {IRewardsController} from "@aavev3-periphery/contracts/rewards/interfaces
* @notice AMO to interact with Aave V3 and manage rewards and borrowing mechanisms.
* @notice Can receive collateral from UbiquityAmoMinter and interact with Aave's V3 pool.
*/
contract AaveAmo is Ownable {
contract AaveAmo is IAmo, Ownable {
using SafeERC20 for ERC20;

/// @notice UbiquityAmoMinter instance
Expand Down Expand Up @@ -185,7 +186,7 @@ contract AaveAmo is Ownable {
*/
function returnCollateralToMinter(
uint256 collateralAmount
) public onlyOwner {
) public override onlyOwner {
ERC20 collateralToken = amoMinter.collateralToken();

if (collateralAmount == 0) {
Expand All @@ -203,7 +204,9 @@ contract AaveAmo is Ownable {
* @notice Sets the AMO minter address
* @param _amoMinterAddress New address of the AMO minter
*/
function setAmoMinter(address _amoMinterAddress) external onlyOwner {
function setAmoMinter(
address _amoMinterAddress
) external override onlyOwner {
amoMinter = UbiquityAmoMinter(_amoMinterAddress);

emit AmoMinterSet(_amoMinterAddress);
Expand Down

0 comments on commit 02bbe8f

Please sign in to comment.