pyth-network_pyth-crosschain/target_chains/ethereum/entropy_sdk/solidity/MockEntropy.sol

// SPDX-License-Identifier: Apache 2
pragma solidity ^0.8.0;

import "./IEntropyV2.sol";
import "./IEntropyConsumer.sol";
import "./EntropyStructsV2.sol";
import "./EntropyEventsV2.sol";

contract MockEntropy is IEntropyV2 {
    address public defaultProvider;

    mapping(address => EntropyStructsV2.ProviderInfo) public providers;
    mapping(address => mapping(uint64 => EntropyStructsV2.Request))
        public requests;

    constructor(address _defaultProvider) {
        require(_defaultProvider != address(0), "Invalid default provider");
        defaultProvider = _defaultProvider;

        providers[_defaultProvider].sequenceNumber = 1;
        providers[_defaultProvider].feeInWei = 1;
        providers[_defaultProvider].defaultGasLimit = 100000;
    }

    function requestV2()
        external
        payable
        override
        returns (uint64 assignedSequenceNumber)
    {
        return _requestV2(defaultProvider, bytes32(0), 0);
    }

    function requestV2(
        uint32 gasLimit
    ) external payable override returns (uint64 assignedSequenceNumber) {
        return _requestV2(defaultProvider, bytes32(0), gasLimit);
    }

    function requestV2(
        address provider,
        uint32 gasLimit
    ) external payable override returns (uint64 assignedSequenceNumber) {
        return _requestV2(provider, bytes32(0), gasLimit);
    }

    function requestV2(
        address provider,
        bytes32 userRandomNumber,
        uint32 gasLimit
    ) external payable override returns (uint64 assignedSequenceNumber) {
        return _requestV2(provider, userRandomNumber, gasLimit);
    }

    function _requestV2(
        address provider,
        bytes32 userRandomNumber,
        uint32 gasLimit
    ) internal returns (uint64 assignedSequenceNumber) {
        EntropyStructsV2.ProviderInfo storage providerInfo = providers[
            provider
        ];

        if (providerInfo.sequenceNumber == 0) {
            providerInfo.sequenceNumber = 1;
            providerInfo.feeInWei = 1;
            providerInfo.defaultGasLimit = 100000;
        }

        assignedSequenceNumber = providerInfo.sequenceNumber;
        providerInfo.sequenceNumber += 1;

        uint32 effectiveGasLimit = gasLimit == 0
            ? providerInfo.defaultGasLimit
            : gasLimit;

        EntropyStructsV2.Request storage req = requests[provider][
            assignedSequenceNumber
        ];
        req.provider = provider;
        req.sequenceNumber = assignedSequenceNumber;
        req.requester = msg.sender;
        req.blockNumber = uint64(block.number);
        req.useBlockhash = false;
        req.gasLimit10k = uint16(effectiveGasLimit / 10000);

        emit Requested(
            provider,
            msg.sender,
            assignedSequenceNumber,
            userRandomNumber,
            effectiveGasLimit,
            bytes("")
        );

        return assignedSequenceNumber;
    }

    function mockReveal(
        address provider,
        uint64 sequenceNumber,
        bytes32 randomNumber
    ) external {
        EntropyStructsV2.Request storage req = requests[provider][
            sequenceNumber
        ];
        require(req.requester != address(0), "Request not found");

        address requester = req.requester;

        emit Revealed(
            provider,
            requester,
            sequenceNumber,
            randomNumber,
            bytes32(0),
            bytes32(0),
            false,
            bytes(""),
            0,
            bytes("")
        );

        delete requests[provider][sequenceNumber];

        uint256 codeSize;
        assembly {
            codeSize := extcodesize(requester)
        }

        if (codeSize > 0) {
            IEntropyConsumer(requester)._entropyCallback(
                sequenceNumber,
                provider,
                randomNumber
            );
        }
    }

    function getProviderInfoV2(
        address provider
    ) external view override returns (EntropyStructsV2.ProviderInfo memory) {
        return providers[provider];
    }

    function getDefaultProvider() external view override returns (address) {
        return defaultProvider;
    }

    function getRequestV2(
        address provider,
        uint64 sequenceNumber
    ) external view override returns (EntropyStructsV2.Request memory) {
        return requests[provider][sequenceNumber];
    }

    function getFeeV2() external pure override returns (uint128) {
        return 0;
    }

    function getFeeV2(uint32) external pure override returns (uint128) {
        return 0;
    }

    function getFeeV2(
        address,
        uint32
    ) external pure override returns (uint128) {
        return 0;
    }
}