/*
    Copyright 2021 Set Labs Inc.

    Licensed under the Apache License, Version 2.0 (the "License");
    you may not use this file except in compliance with the License.
    You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

    Unless required by applicable law or agreed to in writing, software
    distributed under the License is distributed on an "AS IS" BASIS,
    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    See the License for the specific language governing permissions and
    limitations under the License.

    SPDX-License-Identifier: Apache License, Version 2.0
*/

pragma solidity 0.6.10;
pragma experimental ABIEncoderV2;

import { IERC20 } from "@openzeppelin/contracts/token/ERC20/IERC20.sol";
import { IClearingHouse } from "../../../../interfaces/external/perp-v2/IClearingHouse.sol";
import { IVault } from "../../../../interfaces/external/perp-v2/IVault.sol";
import { IQuoter } from "../../../../interfaces/external/perp-v2/IQuoter.sol";

import { PerpV2 } from "../../../../protocol/integration/lib/PerpV2.sol";
import { ISetToken } from "../../../../interfaces/ISetToken.sol";

/**
 * @title PerpV2Mock
 * @author Set Protocol
 *
 * Mock for PerpV2 Library contract. Used for testing PerpV2 Library contract, as the library
 * contract can't be tested directly using ethers.js
 */
contract PerpV2Mock {

    /* ============ External ============ */

    function testGetDepositCalldata(
        IVault _vault,
        IERC20 _asset,
        uint256 _amountNotional
    )
        public
        pure
        returns (address, uint256, bytes memory)
    {
        return PerpV2.getDepositCalldata(_vault, _asset, _amountNotional);
    }

    function testInvokeDeposit(
        ISetToken _setToken,
        IVault _vault,
        IERC20 _asset,
        uint256 _amountNotional
    )
        external
    {
        return PerpV2.invokeDeposit(_setToken, _vault, _asset, _amountNotional);
    }

    function testGetWithdrawCalldata(
        IVault _vault,
        IERC20 _asset,
        uint256 _amountNotional
    )
        public
        pure
        returns (address, uint256, bytes memory)
    {
        return PerpV2.getWithdrawCalldata(_vault, _asset, _amountNotional);
    }

    function testInvokeWithdraw(
        ISetToken _setToken,
        IVault _vault,
        IERC20 _asset,
        uint256 _amountNotional
    )
        external
    {
        return PerpV2.invokeWithdraw(_setToken, _vault, _asset, _amountNotional);
    }

    function testGetOpenPositionCalldata(
        IClearingHouse _clearingHouse,
        IClearingHouse.OpenPositionParams memory _params
    )
        public
        pure
        returns (address, uint256, bytes memory)
    {
        return PerpV2.getOpenPositionCalldata(_clearingHouse, _params);
    }

    function testInvokeOpenPosition(
        ISetToken _setToken,
        IClearingHouse _clearingHouse,
        IClearingHouse.OpenPositionParams memory _params
    )
        external
        returns (uint256 deltaBase, uint256 deltaQuote)
    {
        return PerpV2.invokeOpenPosition(_setToken, _clearingHouse, _params);
    }

    function testGetSwapCalldata(
        IQuoter _quoter,
        IQuoter.SwapParams memory _params
    )
        public
        pure
        returns (address, uint256, bytes memory)
    {
        return PerpV2.getSwapCalldata(_quoter, _params);
    }

    function testInvokeSwap(
        ISetToken _setToken,
        IQuoter _quoter,
        IQuoter.SwapParams memory _params
    )
        external
        returns (IQuoter.SwapResponse memory)
    {
        return PerpV2.invokeSwap(_setToken, _quoter, _params);
    }

    /* ============ Helper Functions ============ */

    function initializeModuleOnSet(ISetToken _setToken) external {
        _setToken.initializeModule();
    }
}