assetxContracts/contracts/ytLp/core/YTPriceFeed.sol

// SPDX-License-Identifier: MIT
pragma solidity ^0.8.0;

import "@openzeppelin/contracts-upgradeable/proxy/utils/Initializable.sol";
import "@openzeppelin/contracts-upgradeable/proxy/utils/UUPSUpgradeable.sol";
import "../../interfaces/IYTToken.sol";

/**
 * @title YTPriceFeed
 * @notice 价格读取器，直接从YT合约读取价格变量（带保护机制和价差）
 * @dev UUPS可升级合约
 */
contract YTPriceFeed is Initializable, UUPSUpgradeable {

    error Forbidden();
    error MaxChangeTooHigh();
    error PriceChangeTooLarge();
    error SpreadTooHigh();
    error InvalidAddress();
    
    address public gov;
    
    uint256 public constant PRICE_PRECISION = 10 ** 30;
    uint256 public constant BASIS_POINTS_DIVISOR = 10000;
    uint256 public constant MAX_SPREAD_BASIS_POINTS = 200; // 最大2%价差
    
    // WUSD固定价格
    address public wusdAddress;
    
    // WUSD价格来源
    address public wusdPriceSource;
    
    // 价格保护参数
    uint256 public maxPriceChangeBps; // 5% 最大价格变动
    
    // 价差配置（每个代币可以有不同的价差）
    mapping(address => uint256) public spreadBasisPoints;
    
    // 价格历史记录
    mapping(address => uint256) public lastPrice;
    
    // 价格更新权限
    mapping(address => bool) public isKeeper;
    
    event PriceUpdate(address indexed token, uint256 oldPrice, uint256 newPrice, uint256 timestamp);
    event SpreadUpdate(address indexed token, uint256 spreadBps);
    event KeeperSet(address indexed keeper, bool isActive);
    
    modifier onlyGov() {
        if (msg.sender != gov) revert Forbidden();
        _;
    }
    
    modifier onlyKeeper() {
        if (!isKeeper[msg.sender] && msg.sender != gov) revert Forbidden();
        _;
    }
    
    /**
     * @notice 初始化合约
     */
    function initialize(address _wusdAddress) external initializer {
        __UUPSUpgradeable_init();
        if (_wusdAddress == address(0)) revert InvalidAddress();
        wusdAddress = _wusdAddress;
        gov = msg.sender;
        maxPriceChangeBps = 500; // 5% 最大价格变动
    }
    
    /**
     * @notice 授权升级（仅gov可调用）
     * @param newImplementation 新实现合约地址
     */
    function _authorizeUpgrade(address newImplementation) internal override onlyGov {}
    
    /**
     * @notice 设置WUSD价格来源（YTAssetVault地址）
     * @param _wusdPriceSource YTAssetVault合约地址
     */
    function setWusdPriceSource(address _wusdPriceSource) external onlyGov {
        wusdPriceSource = _wusdPriceSource;
    }
    
    /**
     * @notice 设置keeper权限
     * @param _keeper keeper地址
     * @param _isActive 是否激活
     */
    function setKeeper(address _keeper, bool _isActive) external onlyGov {
        isKeeper[_keeper] = _isActive;
        emit KeeperSet(_keeper, _isActive);
    }
    
    /**
     * @notice 设置最大价格变动百分比
     * @param _maxPriceChangeBps 最大变动（基点）
     */
    function setMaxPriceChangeBps(uint256 _maxPriceChangeBps) external onlyGov {
        if (_maxPriceChangeBps > 2000) revert MaxChangeTooHigh(); // 最大20%
        maxPriceChangeBps = _maxPriceChangeBps;
    }
    
    /**
     * @notice 设置代币价差
     * @param _token 代币地址
     * @param _spreadBasisPoints 价差（基点）例如：10 = 0.1%, 100 = 1%
     */
    function setSpreadBasisPoints(address _token, uint256 _spreadBasisPoints) external onlyGov {
        if (_spreadBasisPoints > MAX_SPREAD_BASIS_POINTS) revert SpreadTooHigh();
        spreadBasisPoints[_token] = _spreadBasisPoints;
        emit SpreadUpdate(_token, _spreadBasisPoints);
    }
    
    /**
     * @notice 批量设置代币价差
     * @param _tokens 代币地址数组
     * @param _spreadBasisPoints 价差数组
     */
    function setSpreadBasisPointsForMultiple(
        address[] calldata _tokens,
        uint256[] calldata _spreadBasisPoints
    ) external onlyGov {
        require(_tokens.length == _spreadBasisPoints.length, "length mismatch");
        for (uint256 i = 0; i < _tokens.length; i++) {
            if (_spreadBasisPoints[i] > MAX_SPREAD_BASIS_POINTS) revert SpreadTooHigh();
            spreadBasisPoints[_tokens[i]] = _spreadBasisPoints[i];
            emit SpreadUpdate(_tokens[i], _spreadBasisPoints[i]);
        }
    }
    
    /**
     * @notice 强制更新价格（紧急情况）
     * @param _token 代币地址
     * @param _price 新价格
     */
    function forceUpdatePrice(address _token, uint256 _price) external onlyGov {
        uint256 oldPrice = lastPrice[_token];
        lastPrice[_token] = _price;
        emit PriceUpdate(_token, oldPrice, _price, block.timestamp);
    }
    
    /**
     * @notice 获取YT代币价格（带波动保护和价差）
     * @param _token 代币地址
     * @param _maximise true=最大价格（上浮价差，对协议有利）, false=最小价格（下压价差，对协议有利）
     * @return 价格（30位精度）
     * 
     * 使用场景：
     * - 添加流动性时AUM计算：_maximise=true（高估AUM，用户获得较少LP）
     * - 移除流动性时AUM计算：_maximise=false（低估AUM，用户获得较少代币）
     * - buyUSDY时（用户卖代币）：_maximise=false（低估用户代币价值）
     * - sellUSDY时（用户买代币）：_maximise=true（高估需支付的代币价值）
     * - swap时tokenIn：_maximise=false（低估输入）
     * - swap时tokenOut：_maximise=true（高估输出）
     */
    function getPrice(address _token, bool _maximise) external view returns (uint256) {
        if (_token == wusdAddress) {
            return _getWUSDPrice();
        }
        
        uint256 basePrice = _getRawPrice(_token);
        
        // 价格波动检查
        _validatePriceChange(_token, basePrice);
        
        // 应用价差
        return _applySpread(_token, basePrice, _maximise);
    }
    
    /**
     * @notice 更新价格并返回（由keeper调用）
     * @param _token 代币地址
     * @return 新价格
     */
    function updatePrice(address _token) external onlyKeeper returns (uint256) {
        if (_token == wusdAddress) {
            return _getWUSDPrice();
        }
        
        uint256 oldPrice = lastPrice[_token];
        uint256 newPrice = _getRawPrice(_token);
        
        // 价格波动检查
        _validatePriceChange(_token, newPrice);
        
        lastPrice[_token] = newPrice;
        
        emit PriceUpdate(_token, oldPrice, newPrice, block.timestamp);
        
        return newPrice;
    }
    
    /**
     * @notice 直接读取YT代币的ytPrice变量
     */
    function _getRawPrice(address _token) private view returns (uint256) {
        return IYTToken(_token).ytPrice();
    }

    /**
     * @notice 从配置的YTAssetVault读取wusdPrice
     * @dev 如果未设置wusdPriceSource，返回固定价格1.0
     */
    function _getWUSDPrice() private view returns (uint256) {
        if (wusdPriceSource == address(0)) {
            return PRICE_PRECISION; // 默认1.0
        }
        return IYTToken(wusdPriceSource).wusdPrice();
    }
    
    /**
     * @notice 应用价差
     * @param _token 代币地址
     * @param _basePrice 基础价格
     * @param _maximise true=上浮价格，false=下压价格
     * @return 应用价差后的价格
     */
    function _applySpread(
        address _token,
        uint256 _basePrice,
        bool _maximise
    ) private view returns (uint256) {
        uint256 spread = spreadBasisPoints[_token];
        
        // 如果没有设置价差，直接返回基础价格
        if (spread == 0) {
            return _basePrice;
        }
        
        if (_maximise) {
            // 上浮价格：basePrice * (1 + spread%)
            return _basePrice * (BASIS_POINTS_DIVISOR + spread) / BASIS_POINTS_DIVISOR;
        } else {
            // 下压价格：basePrice * (1 - spread%)
            return _basePrice * (BASIS_POINTS_DIVISOR - spread) / BASIS_POINTS_DIVISOR;
        }
    }
    
    /**
     * @notice 验证价格变动是否在允许范围内
     */
    function _validatePriceChange(address _token, uint256 _newPrice) private view {
        uint256 oldPrice = lastPrice[_token];
        
        // 首次设置价格，跳过检查
        if (oldPrice == 0) {
            return;
        }
        
        // 计算价格变动百分比
        uint256 priceDiff = _newPrice > oldPrice ? _newPrice - oldPrice : oldPrice - _newPrice;
        uint256 maxDiff = oldPrice * maxPriceChangeBps / BASIS_POINTS_DIVISOR;
        
        if (priceDiff > maxDiff) revert PriceChangeTooLarge();
    }
    
    /**
     * @notice 获取价格详细信息
     */
    function getPriceInfo(address _token) external view returns (
        uint256 currentPrice,
        uint256 cachedPrice,
        uint256 maxPrice,
        uint256 minPrice,
        uint256 spread
    ) {
        if (_token == wusdAddress) {
            uint256 wusdPrice = _getWUSDPrice();
            currentPrice = wusdPrice;
            cachedPrice = wusdPrice;
            maxPrice = wusdPrice;
            minPrice = wusdPrice;
            spread = 0;
        } else {
            currentPrice = _getRawPrice(_token);
            cachedPrice = lastPrice[_token];
            spread = spreadBasisPoints[_token];
            maxPrice = _applySpread(_token, currentPrice, true);
            minPrice = _applySpread(_token, currentPrice, false);
        }
    }
    
    /**
     * @notice 获取最大价格（上浮价差）
     */
    function getMaxPrice(address _token) external view returns (uint256) {
        if (_token == wusdAddress) {
            // WUSD通常不需要价差，直接返回原价格
            return _getWUSDPrice();
        }
        uint256 basePrice = _getRawPrice(_token);
        _validatePriceChange(_token, basePrice);
        return _applySpread(_token, basePrice, true);
    }
    
    /**
     * @notice 获取最小价格（下压价差）
     */
    function getMinPrice(address _token) external view returns (uint256) {
        if (_token == wusdAddress) {
            // WUSD通常不需要价差，直接返回原价格
            return _getWUSDPrice();
        }
        uint256 basePrice = _getRawPrice(_token);
        _validatePriceChange(_token, basePrice);
        return _applySpread(_token, basePrice, false);
    }
    
    /**
     * @dev 预留存储空间，用于未来升级时添加新的状态变量
     * 50个slot = 50 * 32 bytes = 1600 bytes
     */
    uint256[50] private __gap;
}