ytLending supports USDC as the base token
This commit is contained in:
@@ -1,39 +1,59 @@
|
||||
// SPDX-License-Identifier: MIT
|
||||
pragma solidity ^0.8.0;
|
||||
|
||||
import "../interfaces/IYTVault.sol";
|
||||
import "@openzeppelin/contracts/access/Ownable.sol";
|
||||
import "../interfaces/IYTAssetVault.sol";
|
||||
import "@openzeppelin/contracts-upgradeable/access/OwnableUpgradeable.sol";
|
||||
import "@openzeppelin/contracts-upgradeable/proxy/utils/UUPSUpgradeable.sol";
|
||||
import {AggregatorV3Interface} from "@chainlink/contracts/src/v0.8/shared/interfaces/AggregatorV3Interface.sol";
|
||||
|
||||
contract LendingPriceFeed is Ownable {
|
||||
address public ytVault;
|
||||
address public wusdAddress;
|
||||
contract LendingPriceFeed is OwnableUpgradeable, UUPSUpgradeable {
|
||||
address public usdcAddress;
|
||||
AggregatorV3Interface internal usdcPriceFeed;
|
||||
|
||||
error InvalidYTVaultAddress();
|
||||
error InvalidWUSDAddress();
|
||||
|
||||
constructor(address _ytVault, address _wusdAddress) Ownable(msg.sender) {
|
||||
if (_ytVault == address(0)) revert InvalidYTVaultAddress();
|
||||
if (_wusdAddress == address(0)) revert InvalidWUSDAddress();
|
||||
ytVault = _ytVault;
|
||||
wusdAddress = _wusdAddress;
|
||||
error InvalidUsdcAddress();
|
||||
error InvalidUsdcPriceFeedAddress();
|
||||
error InvalidChainlinkPrice();
|
||||
|
||||
/// @custom:oz-upgrades-unsafe-allow constructor
|
||||
constructor() {
|
||||
_disableInitializers();
|
||||
}
|
||||
|
||||
function setYTVault(address _ytVault) external onlyOwner {
|
||||
if (_ytVault == address(0)) revert InvalidYTVaultAddress();
|
||||
ytVault = _ytVault;
|
||||
function initialize(address _usdcAddress, address _usdcPriceFeed) external initializer {
|
||||
__UUPSUpgradeable_init();
|
||||
__Ownable_init(msg.sender);
|
||||
if (_usdcAddress == address(0)) revert InvalidUsdcAddress();
|
||||
if (_usdcPriceFeed == address(0)) revert InvalidUsdcPriceFeedAddress();
|
||||
usdcAddress = _usdcAddress;
|
||||
usdcPriceFeed = AggregatorV3Interface(_usdcPriceFeed);
|
||||
}
|
||||
|
||||
function setWUSDAddress(address _wusdAddress) external onlyOwner {
|
||||
if (_wusdAddress == address(0)) revert InvalidWUSDAddress();
|
||||
wusdAddress = _wusdAddress;
|
||||
function _authorizeUpgrade(address newImplementation) internal override onlyOwner {}
|
||||
|
||||
function setUsdcAddress(address _usdcAddress) external onlyOwner {
|
||||
if (_usdcAddress == address(0)) revert InvalidUsdcAddress();
|
||||
usdcAddress = _usdcAddress;
|
||||
}
|
||||
|
||||
function getPrice(address _token) external view returns (uint256) {
|
||||
if (_token == wusdAddress) {
|
||||
return IYTVault(ytVault).wusdPrice();
|
||||
} else {
|
||||
return IYTVault(_token).ytPrice();
|
||||
if (_token == usdcAddress) {
|
||||
return _getUSDCPrice();
|
||||
}
|
||||
return IYTAssetVault(_token).ytPrice();
|
||||
}
|
||||
|
||||
function _getUSDCPrice() internal view returns (uint256) {
|
||||
(
|
||||
/* uint80 roundId */,
|
||||
int256 price,
|
||||
/* uint256 startedAt */,
|
||||
/* uint256 updatedAt */,
|
||||
/* uint80 answeredInRound */
|
||||
) = usdcPriceFeed.latestRoundData();
|
||||
|
||||
if (price <= 0) revert InvalidChainlinkPrice();
|
||||
|
||||
return uint256(price) * 1e22; // 1e22 = 10^(30-8)
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user