swap.py

from typing import List, Optional

from fastapi import APIRouter, Depends, Path, Query, HTTPException
from fastapi.security import HTTPBearer
from pydantic import conint

from meta_aggregation_api.config.auth import AuthJWT
from meta_aggregation_api.models.meta_agg_models import (
    MetaPriceModel,
    ProviderQuoteResponse,
)
from meta_aggregation_api.rest_api import dependencies
from meta_aggregation_api.utils.common import address_to_lower
from meta_aggregation_api.utils.errors import responses

PRICE_CACHE_TTL_SEC = 5
swap_route = APIRouter()


@swap_route.get('/{chain_id}/price', response_model=MetaPriceModel, responses=responses)
@swap_route.get(
    '/{chain_id}/price/', response_model=MetaPriceModel, include_in_schema=False
)
async def get_swap_price(
    buy_token: address_to_lower = Query(..., alias='buyToken'),
    sell_token: address_to_lower = Query(..., alias='sellToken'),
    sell_amount: conint(gt=0) = Query(..., alias='sellAmount'),
    chain_id: int = Path(..., description='Chain ID'),
    gas_price: Optional[int] = Query(
        None, description='Gas price', gt=0, alias='gasPrice'
    ),
    slippage_percentage: Optional[float] = Query(
        0.005, gte=0, alias='slippagePercentage'
    ),
    taker_address: Optional[address_to_lower] = Query(None, alias='takerAddress'),
    fee_recipient: Optional[address_to_lower] = Query(None, alias='feeRecipient'),
    buy_token_percentage_fee: Optional[float] = Query(
        None, alias='buyTokenPercentageFee'
    ),
    provider: Optional[str] = Query(None, alias='provider'),
    meta_aggregation_service: dependencies.MetaAggregationService = Depends(
        dependencies.meta_aggregation_service
    ),
) -> MetaPriceModel:
    """
    Price endpoints are used to get the best price for a swap. It does not return data for swap and therefore
    require any approvals. If you want to get data for swap, use /price_response endpoint.

    - **buy_token**: Address of the token to buy
    - **sell_token**: Address of the token to sell
    - **sell_amount**: Amount of the token to sell in base units (e.g. 1 ETH = 10**18)
    - **chain_id**: Chain ID. See /info for supported chains
    - **gas_price**: Gas price in wei (optional)
    - **slippage_percentage**: Slippage percentage  (0.01 = 1%) (default: 0.005)
    - **taker_address**: Address of the taker (optional)
    - **fee_recipient**: Address of the fee recipient (optional)
    - **buy_token_percentage_fee**: Percentage of the buy token fee (optional) (0.01 = 1%)
    - **provider**: Provider name from /info (optional). If not specified, the best price will be returned
    """
    params = {
        "buy_token": buy_token,
        "sell_token": sell_token,
        "sell_amount": sell_amount,
        "chain_id": chain_id,
        "gas_price": gas_price,
        "slippage_percentage": slippage_percentage,
        "taker_address": taker_address,
        "fee_recipient": fee_recipient,
        "buy_token_percentage_fee": buy_token_percentage_fee,
    }
    if provider:
        res = await meta_aggregation_service.get_provider_price(
            provider=provider, **params
        )
        if not res:
            raise HTTPException(
                status_code=404,
                detail='No prices found',
            )
        return res
    else:
        res = await meta_aggregation_service.get_swap_meta_price(**params)
    if not res:
        raise HTTPException(
            status_code=404,
            detail='No prices found',
        )
    return next((quote for quote in res if quote.is_best), None)


@swap_route.get(
    '/{chain_id}/price/all', response_model=List[MetaPriceModel], responses=responses
)
@swap_route.get(
    '/{chain_id}/price/all/',
    include_in_schema=False,
    response_model=List[MetaPriceModel],
)
async def get_all_swap_prices(
    buy_token: address_to_lower = Query(..., alias='buyToken'),
    sell_token: address_to_lower = Query(..., alias='sellToken'),
    sell_amount: conint(gt=0) = Query(..., alias='sellAmount'),
    chain_id: int = Path(..., description='Chain ID'),
    gas_price: Optional[int] = Query(
        None, description='Gas price', gt=0, alias='gasPrice'
    ),
    slippage_percentage: Optional[float] = Query(0.005, alias='slippagePercentage'),
    taker_address: Optional[address_to_lower] = Query(None, alias='takerAddress'),
    fee_recipient: Optional[address_to_lower] = Query(None, alias='feeRecipient'),
    buy_token_percentage_fee: Optional[float] = Query(
        None, alias='buyTokenPercentageFee'
    ),
    meta_aggregation_service: dependencies.MetaAggregationService = Depends(
        dependencies.meta_aggregation_service
    ),
) -> List[MetaPriceModel]:
    """
    Works the same as /price endpoint, but returns all prices from all supported providers.

    - **buy_token**: Address of the token to buy
    - **sell_token**: Address of the token to sell
    - **sell_amount**: Amount of the token to sell in base units (e.g. 1 ETH = 10**18)
    - **chain_id**: Chain ID. See /info for supported chains
    - **gas_price**: Gas price in wei (optional)
    - **slippage_percentage**: Slippage percentage  (0.01 = 1%) (default: 0.005)
    - **taker_address**: Address of the taker (optional)
    - **fee_recipient**: Address of the fee recipient (optional)
    - **buy_token_percentage_fee**: Percentage of the buy token fee (optional) (0.01 = 1%)
    """
    res = await meta_aggregation_service.get_swap_meta_price(
        buy_token=buy_token,
        sell_token=sell_token,
        sell_amount=sell_amount,
        chain_id=chain_id,
        gas_price=gas_price,
        slippage_percentage=slippage_percentage,
        taker_address=taker_address,
        fee_recipient=fee_recipient,
        buy_token_percentage_fee=buy_token_percentage_fee,
    )
    if not res:
        raise HTTPException(
            status_code=404,
            detail='No prices found',
        )
    return res


@swap_route.get(
    '/{chain_id}/quote',
    response_model=ProviderQuoteResponse,
    responses=responses,
    # dependencies=[Depends(HTTPBearer())],
)
@swap_route.get(
    '/{chain_id}/quote/',
    response_model=ProviderQuoteResponse,
    include_in_schema=False,
    # dependencies=[Depends(HTTPBearer())],
)
async def get_swap_quote(
    # authorize: AuthJWT = Depends(),
    buy_token: address_to_lower = Query(..., alias='buyToken'),
    sell_token: address_to_lower = Query(..., alias='sellToken'),
    sell_amount: conint(gt=0) = Query(..., alias='sellAmount'),
    chain_id: int = Path(..., description='Chain ID'),
    provider: str = Query(..., alias='provider'),
    taker_address: address_to_lower = Query(..., alias='takerAddress'),
    gas_price: Optional[int] = Query(
        None, description='Gas price', gt=0, alias='gasPrice'
    ),
    slippage_percentage: Optional[float] = Query(0.005, alias='slippagePercentage'),
    fee_recipient: Optional[address_to_lower] = Query(None, alias='feeRecipient'),
    buy_token_percentage_fee: Optional[float] = Query(
        None, alias='buyTokenPercentageFee'
    ),
    meta_aggregation_service: dependencies.MetaAggregationService = Depends(
        dependencies.meta_aggregation_service
    ),
) -> ProviderQuoteResponse:
    """
    Returns a data for swap from a specific provider.

    - **buy_token**:Address of the token to buy
    - **sell_token**:Address of the token to sell
    - **sell_amount**:Amount of the token to sell in base units (e.g. 1 ETH = 10**18)
    - **chain_id**: Chain ID. See /info for supported chains
    - **provider**: Provider name from /info
    - **gas_price**: Gas price in wei (optional)
    - **slippage_percentage**: Slippage percentage  (0.01 = 1%) (default: 0.005)
    - **taker_address**: Address of the taker (optional)
    - **fee_recipient**: Address of the fee recipient (optional)
    - **buy_token_percentage_fee**: Percentage of the buy token fee (optional) (0.01 = 1%)
    """
    # authorize.jwt_required()
    quote = await meta_aggregation_service.get_meta_swap_quote(
        buy_token=buy_token,
        sell_token=sell_token,
        sell_amount=sell_amount,
        chain_id=chain_id,
        provider=provider,
        gas_price=gas_price,
        slippage_percentage=slippage_percentage,
        taker_address=taker_address,
        fee_recipient=fee_recipient,
        buy_token_percentage_fee=buy_token_percentage_fee,
    )
    return quote