// SPDX-License-Identifier: MIT

pragma solidity 0.8.6;

import "./interfaces/ILaunchEvent.sol";
import "./interfaces/IRocketJoeFactory.sol";

/// @title Launch Event Lens
/// @author Trader Joe
/// @notice Helper contract to fetch launch event data
contract LaunchEventLens {
    struct LaunchEventData {
        uint256 auctionStart;
        uint256 avaxAllocated;
        uint256 avaxReserve;
        uint256 floorPrice;
        uint256 incentives;
        uint256 issuerTimelock;
        uint256 maxAllocation;
        uint256 maxWithdrawPenalty;
        uint256 pairBalance;
        uint256 penalty;
        uint256 phaseOneDuration;
        uint256 phaseOneNoFeeDuration;
        uint256 phaseTwoDuration;
        uint256 rJoePerAvax;
        uint256 tokenAllocated;
        uint256 tokenDecimals;
        uint256 tokenIncentivesPercent;
        uint256 tokenReserve;
        uint256 userTimelock;
        address id;
        address token;
        address pair;
        ILaunchEvent.UserInfo userInfo;
    }

    IRocketJoeFactory public rocketJoeFactory;

    /// @notice Create a new instance with required parameters
    /// @param _rocketJoeFactory Address of the RocketJoeFactory
    constructor(address _rocketJoeFactory) {
        rocketJoeFactory = IRocketJoeFactory(_rocketJoeFactory);
    }

    /// @notice Get all launch event datas
    /// @param _offset Index to start at when looking up launch events
    /// @param _limit Maximum number of launch event datas to return
    /// @return Array of all launch event datas
    function getAllLaunchEvents(uint256 _offset, uint256 _limit)
        external
        view
        returns (LaunchEventData[] memory)
    {
        LaunchEventData[] memory launchEventDatas;
        uint256 numLaunchEvents = rocketJoeFactory.numLaunchEvents();

        if (_offset >= numLaunchEvents || _limit == 0) {
            return launchEventDatas;
        }

        uint256 end = _offset + _limit > numLaunchEvents
            ? numLaunchEvents
            : _offset + _limit;
        launchEventDatas = new LaunchEventData[](end - _offset);

        for (uint256 i = _offset; i < end; i++) {
            address launchEventAddr = rocketJoeFactory.allRJLaunchEvents(i);
            ILaunchEvent launchEvent = ILaunchEvent(launchEventAddr);
            launchEventDatas[i] = getLaunchEventData(launchEvent);
        }

        return launchEventDatas;
    }

    /// @notice Get all launch event datas with a given `_user`
    /// @param _offset Index to start at when looking up launch events
    /// @param _limit Maximum number of launch event datas to return
    /// @param _user User to lookup
    /// @return Array of all launch event datas with user info
    function getAllLaunchEventsWithUser(
        uint256 _offset,
        uint256 _limit,
        address _user
    ) external view returns (LaunchEventData[] memory) {
        LaunchEventData[] memory launchEventDatas;
        uint256 numLaunchEvents = rocketJoeFactory.numLaunchEvents();

        if (_offset >= numLaunchEvents || _limit == 0) {
            return launchEventDatas;
        }

        uint256 end = _offset + _limit > numLaunchEvents
            ? numLaunchEvents
            : _offset + _limit;
        launchEventDatas = new LaunchEventData[](end - _offset);

        for (uint256 i = _offset; i < end; i++) {
            address launchEventAddr = rocketJoeFactory.allRJLaunchEvents(i);
            ILaunchEvent launchEvent = ILaunchEvent(launchEventAddr);
            launchEventDatas[i] = getUserLaunchEventData(launchEvent, _user);
        }

        return launchEventDatas;
    }

    /// @notice Get launch event data for a given launch event and user
    /// @param _launchEvent Launch event to lookup
    /// @param _user User to look up
    /// @return Launch event data for the given `_launchEvent` and `_user`
    function getUserLaunchEventData(ILaunchEvent _launchEvent, address _user)
        public
        view
        returns (LaunchEventData memory)
    {
        LaunchEventData memory launchEventData = getLaunchEventData(
            _launchEvent
        );
        launchEventData.incentives = _launchEvent.getIncentives(_user);
        launchEventData.pairBalance = _launchEvent.pairBalance(_user);
        launchEventData.userInfo = _launchEvent.getUserInfo(_user);
        return launchEventData;
    }

    /// @notice Get launch event data for a given launch event
    /// @param _launchEvent Launch event to lookup
    /// @return Launch event data for the given `_launchEvent`
    function getLaunchEventData(ILaunchEvent _launchEvent)
        public
        view
        returns (LaunchEventData memory)
    {
        (uint256 avaxReserve, uint256 tokenReserve) = _launchEvent
            .getReserves();
        IERC20Metadata token = _launchEvent.token();

        return
            LaunchEventData({
                auctionStart: _launchEvent.auctionStart(),
                avaxAllocated: _launchEvent.avaxAllocated(),
                avaxReserve: avaxReserve,
                floorPrice: _launchEvent.floorPrice(),
                incentives: 0,
                issuerTimelock: _launchEvent.issuerTimelock(),
                maxAllocation: _launchEvent.maxAllocation(),
                maxWithdrawPenalty: _launchEvent.maxWithdrawPenalty(),
                penalty: _launchEvent.getPenalty(),
                pairBalance: 0,
                phaseOneDuration: _launchEvent.phaseOneDuration(),
                phaseOneNoFeeDuration: _launchEvent.phaseOneNoFeeDuration(),
                phaseTwoDuration: _launchEvent.phaseTwoDuration(),
                rJoePerAvax: _launchEvent.rJoePerAvax(),
                tokenAllocated: _launchEvent.tokenAllocated(),
                tokenDecimals: token.decimals(),
                tokenIncentivesPercent: _launchEvent.tokenIncentivesPercent(),
                tokenReserve: tokenReserve,
                userTimelock: _launchEvent.userTimelock(),
                id: address(_launchEvent),
                token: address(token),
                pair: address(_launchEvent.pair()),
                userInfo: ILaunchEvent.UserInfo({
                    allocation: 0,
                    balance: 0,
                    hasWithdrawnPair: false,
                    hasWithdrawnIncentives: false
                })
            });
    }
}