bitfinex-api-py/bfxapi/rest/endpoints/rest_auth_endpoints.py

from typing import Dict, List, Tuple, Union, Literal, Optional

from decimal import Decimal

from ..middleware import Middleware

from ..enums import Sort, OrderType, FundingOfferType

from ...types import Notification, \
    UserInfo, LoginHistory, BalanceAvailable, \
    Order, Position, Trade, \
    FundingTrade, OrderTrade, Ledger, \
    FundingOffer, FundingCredit, FundingLoan, \
    FundingAutoRenew, FundingInfo, Wallet, \
    Transfer, Withdrawal, DepositAddress, \
    LightningNetworkInvoice, Movement, SymbolMarginInfo, \
    BaseMarginInfo, PositionClaim, PositionIncreaseInfo, \
    PositionIncrease, PositionHistory, PositionSnapshot, \
    PositionAudit, DerivativePositionCollateral, DerivativePositionCollateralLimits

from ...types import serializers

from ...types.serializers import _Notification

from ...utils.json_encoder import JSON

class RestAuthEndpoints(Middleware):
    def get_user_info(self) -> UserInfo:
        return serializers.UserInfo \
            .parse(*self._post("auth/r/info/user"))

    def get_login_history(self) -> List[LoginHistory]:
        return [ serializers.LoginHistory.parse(*sub_data)
            for sub_data in self._post("auth/r/logins/hist") ]

    def get_balance_available_for_orders_or_offers(self,
                                                   symbol: str,
                                                   type: str,
                                                   *,
                                                   dir: Optional[int] = None,
                                                   rate: Optional[str] = None,
                                                   lev: Optional[str] = None) -> BalanceAvailable:
        body = {
            "symbol": symbol, "type": type, "dir": dir,
            "rate": rate, "lev": lev
        }

        return serializers.BalanceAvailable \
            .parse(*self._post("auth/calc/order/avail", body=body))

    def get_wallets(self) -> List[Wallet]:
        return [ serializers.Wallet.parse(*sub_data) \
            for sub_data in self._post("auth/r/wallets") ]

    def get_orders(self,
                   *,
                   symbol: Optional[str] = None,
                   ids: Optional[List[str]] = None) -> List[Order]:
        if symbol is None:
            endpoint = "auth/r/orders"
        else: endpoint = f"auth/r/orders/{symbol}"

        return [ serializers.Order.parse(*sub_data) \
            for sub_data in self._post(endpoint, body={ "id": ids }) ]

    def submit_order(self,
                     type: OrderType,
                     symbol: str,
                     amount: Union[Decimal, float, str],
                     *,
                     price: Optional[Union[Decimal, float, str]] = None,
                     lev: Optional[int] = None,
                     price_trailing: Optional[Union[Decimal, float, str]] = None,
                     price_aux_limit: Optional[Union[Decimal, float, str]] = None,
                     price_oco_stop: Optional[Union[Decimal, float, str]] = None,
                     gid: Optional[int] = None,
                     cid: Optional[int] = None,
                     flags: Optional[int] = 0,
                     tif: Optional[str] = None,
                     meta: Optional[JSON] = None) -> Notification[Order]:
        body = {
            "type": type, "symbol": symbol, "amount": amount,
            "price": price, "lev": lev, "price_trailing": price_trailing,
            "price_aux_limit": price_aux_limit, "price_oco_stop": price_oco_stop, "gid": gid,
            "cid": cid, "flags": flags, "tif": tif,
            "meta": meta
        }

        return _Notification[Order](serializers.Order) \
            .parse(*self._post("auth/w/order/submit", body=body))

    def update_order(self,
                     id: int,
                     *,
                     amount: Optional[Union[Decimal, float, str]] = None,
                     price: Optional[Union[Decimal, float, str]] = None,
                     cid: Optional[int] = None,
                     cid_date: Optional[str] = None,
                     gid: Optional[int] = None,
                     flags: Optional[int] = 0,
                     lev: Optional[int] = None,
                     delta: Optional[Union[Decimal, float, str]] = None,
                     price_aux_limit: Optional[Union[Decimal, float, str]] = None,
                     price_trailing: Optional[Union[Decimal, float, str]] = None,
                     tif: Optional[str] = None) -> Notification[Order]:
        body = {
            "id": id, "amount": amount, "price": price,
            "cid": cid, "cid_date": cid_date, "gid": gid,
            "flags": flags, "lev": lev, "delta": delta,
            "price_aux_limit": price_aux_limit, "price_trailing": price_trailing, "tif": tif
        }

        return _Notification[Order](serializers.Order) \
            .parse(*self._post("auth/w/order/update", body=body))

    def cancel_order(self,
                     *,
                     id: Optional[int] = None,
                     cid: Optional[int] = None,
                     cid_date: Optional[str] = None) -> Notification[Order]:
        return _Notification[Order](serializers.Order) \
            .parse(*self._post("auth/w/order/cancel", \
                body={ "id": id, "cid": cid, "cid_date": cid_date }))

    def cancel_order_multi(self,
                           *,
                           ids: Optional[List[int]] = None,
                           cids: Optional[List[Tuple[int, str]]] = None,
                           gids: Optional[List[int]] = None,
                           all: bool = False) -> Notification[List[Order]]:
        body = {
            "ids": ids, "cids": cids, "gids": gids, 
            "all": all
        }

        return _Notification[List[Order]](serializers.Order, is_iterable=True) \
            .parse(*self._post("auth/w/order/cancel/multi", body=body))

    def get_orders_history(self,
                           *,
                           symbol: Optional[str] = None,
                           ids: Optional[List[int]] = None,
                           start: Optional[str] = None,
                           end: Optional[str] = None,
                           limit: Optional[int] = None) -> List[Order]:
        if symbol is None:
            endpoint = "auth/r/orders/hist"
        else: endpoint = f"auth/r/orders/{symbol}/hist"

        body = {
            "id": ids, "start": start, "end": end, 
            "limit": limit 
        }

        return [ serializers.Order.parse(*sub_data) \
            for sub_data in self._post(endpoint, body=body) ]

    def get_order_trades(self,
                         symbol: str,
                         id: int) -> List[OrderTrade]:
        return [ serializers.OrderTrade.parse(*sub_data) \
            for sub_data in self._post(f"auth/r/order/{symbol}:{id}/trades") ]

    def get_trades_history(self,
                           *,
                           symbol: Optional[str] = None,
                           sort: Optional[Sort] = None,
                           start: Optional[str] = None,
                           end: Optional[str] = None,
                           limit: Optional[int] = None) -> List[Trade]:
        if symbol is None:
            endpoint = "auth/r/trades/hist"
        else: endpoint = f"auth/r/trades/{symbol}/hist"

        body = {
            "sort": sort, "start": start, "end": end,
            "limit": limit
        }

        return [ serializers.Trade.parse(*sub_data) \
            for sub_data in self._post(endpoint, body=body) ]

    def get_ledgers(self,
                    currency: str,
                    *,
                    category: Optional[int] = None,
                    start: Optional[str] = None,
                    end: Optional[str] = None,
                    limit: Optional[int] = None) -> List[Ledger]:
        body = {
            "category": category, "start": start, "end": end, 
            "limit": limit 
        }

        return [ serializers.Ledger.parse(*sub_data) \
            for sub_data in self._post(f"auth/r/ledgers/{currency}/hist", body=body) ]

    def get_base_margin_info(self) -> BaseMarginInfo:
        return serializers.BaseMarginInfo \
            .parse(*(self._post("auth/r/info/margin/base")[1]))

    def get_symbol_margin_info(self, symbol: str) -> SymbolMarginInfo:
        return serializers.SymbolMarginInfo \
            .parse(*self._post(f"auth/r/info/margin/{symbol}"))

    def get_all_symbols_margin_info(self) -> List[SymbolMarginInfo]:
        return [ serializers.SymbolMarginInfo.parse(*sub_data) \
            for sub_data in self._post("auth/r/info/margin/sym_all") ]

    def get_positions(self) -> List[Position]:
        return [ serializers.Position.parse(*sub_data) \
            for sub_data in self._post("auth/r/positions") ]

    def claim_position(self,
                       id: int,
                       *,
                       amount: Optional[Union[Decimal, float, str]] = None) -> Notification[PositionClaim]:
        return _Notification[PositionClaim](serializers.PositionClaim) \
            .parse(*self._post("auth/w/position/claim", \
                body={ "id": id, "amount": amount }))

    def increase_position(self,
                          symbol: str,
                          amount: Union[Decimal, float, str]) -> Notification[PositionIncrease]:
        return _Notification[PositionIncrease](serializers.PositionIncrease) \
            .parse(*self._post("auth/w/position/increase", \
                body={ "symbol": symbol, "amount": amount }))

    def get_increase_position_info(self,
                                   symbol: str,
                                   amount: Union[Decimal, float, str]) -> PositionIncreaseInfo:
        return serializers.PositionIncreaseInfo \
            .parse(*self._post("auth/r/position/increase/info", \
                body={ "symbol": symbol, "amount": amount }))

    def get_positions_history(self,
                              *,
                              start: Optional[str] = None,
                              end: Optional[str] = None,
                              limit: Optional[int] = None) -> List[PositionHistory]:
        return [ serializers.PositionHistory.parse(*sub_data) \
            for sub_data in self._post("auth/r/positions/hist", \
                body={ "start": start, "end": end, "limit": limit }) ]

    def get_positions_snapshot(self,
                               *,
                               start: Optional[str] = None,
                               end: Optional[str] = None,
                               limit: Optional[int] = None) -> List[PositionSnapshot]:
        return [ serializers.PositionSnapshot.parse(*sub_data) \
            for sub_data in self._post("auth/r/positions/snap", \
                body={ "start": start, "end": end, "limit": limit }) ]

    def get_positions_audit(self,
                            *,
                            ids: Optional[List[int]] = None,
                            start: Optional[str] = None,
                            end: Optional[str] = None,
                            limit: Optional[int] = None) -> List[PositionAudit]:
        body = {
            "ids": ids, "start": start, "end": end, 
            "limit": limit 
        }

        return [ serializers.PositionAudit.parse(*sub_data) \
            for sub_data in self._post("auth/r/positions/audit", body=body) ]

    def set_derivative_position_collateral(self,
                                           symbol: str,
                                           collateral: Union[Decimal, float, str]) -> DerivativePositionCollateral:
        return serializers.DerivativePositionCollateral \
            .parse(*(self._post("auth/w/deriv/collateral/set", \
                body={ "symbol": symbol, "collateral": collateral })[0]))

    def get_derivative_position_collateral_limits(self, symbol: str) -> DerivativePositionCollateralLimits:
        return serializers.DerivativePositionCollateralLimits \
            .parse(*self._post("auth/calc/deriv/collateral/limit", body={ "symbol": symbol }))

    def get_funding_offers(self, *, symbol: Optional[str] = None) -> List[FundingOffer]:
        if symbol is None:
            endpoint = "auth/r/funding/offers"
        else: endpoint = f"auth/r/funding/offers/{symbol}"

        return [ serializers.FundingOffer.parse(*sub_data) \
            for sub_data in self._post(endpoint) ]

    #pylint: disable-next=too-many-arguments
    def submit_funding_offer(self,
                             type: FundingOfferType,
                             symbol: str,
                             amount: Union[Decimal, float, str],
                             rate: Union[Decimal, float, str],
                             period: int,
                             *,
                             flags: Optional[int] = 0) -> Notification[FundingOffer]:
        body = {
            "type": type, "symbol": symbol, "amount": amount,
            "rate": rate, "period": period, "flags": flags
        }

        return _Notification[FundingOffer](serializers.FundingOffer) \
            .parse(*self._post("auth/w/funding/offer/submit", body=body))

    def cancel_funding_offer(self, id: int) -> Notification[FundingOffer]:
        return _Notification[FundingOffer](serializers.FundingOffer) \
            .parse(*self._post("auth/w/funding/offer/cancel", body={ "id": id }))

    def cancel_all_funding_offers(self, currency: str) -> Notification[Literal[None]]:
        return _Notification[Literal[None]](None) \
            .parse(*self._post("auth/w/funding/offer/cancel/all", body={ "currency": currency }))

    def submit_funding_close(self, id: int) -> Notification[Literal[None]]:
        return _Notification[Literal[None]](None) \
            .parse(*self._post("auth/w/funding/close", body={ "id": id }))

    def toggle_auto_renew(self,
                          status: bool,
                          currency: str,
                          *,
                          amount: Optional[str] = None,
                          rate: Optional[int] = None,
                          period: Optional[int] = None) -> Notification[FundingAutoRenew]:
        body = {
            "status": status, "currency": currency, "amount": amount,
            "rate": rate, "period": period
        }

        return _Notification[FundingAutoRenew](serializers.FundingAutoRenew) \
            .parse(*self._post("auth/w/funding/auto", body=body))

    def toggle_keep_funding(self,
                            type: Literal["credit", "loan"],
                            *,
                            ids: Optional[List[int]] = None,
                            changes: Optional[Dict[int, Literal[1, 2]]] = None) -> Notification[Literal[None]]:
        return _Notification[Literal[None]](None) \
            .parse(*self._post("auth/w/funding/keep", \
                body={ "type": type, "id": ids, "changes": changes }))

    def get_funding_offers_history(self,
                                   *,
                                   symbol: Optional[str] = None,
                                   start: Optional[str] = None,
                                   end: Optional[str] = None,
                                   limit: Optional[int] = None) -> List[FundingOffer]:
        if symbol is None:
            endpoint = "auth/r/funding/offers/hist"
        else: endpoint = f"auth/r/funding/offers/{symbol}/hist"

        return [ serializers.FundingOffer.parse(*sub_data) \
            for sub_data in self._post(endpoint, \
                body={ "start": start, "end": end, "limit": limit }) ]

    def get_funding_loans(self, *, symbol: Optional[str] = None) -> List[FundingLoan]:
        if symbol is None:
            endpoint = "auth/r/funding/loans"
        else: endpoint = f"auth/r/funding/loans/{symbol}"

        return [ serializers.FundingLoan.parse(*sub_data) \
            for sub_data in self._post(endpoint) ]

    def get_funding_loans_history(self,
                                  *,
                                  symbol: Optional[str] = None,
                                  start: Optional[str] = None,
                                  end: Optional[str] = None,
                                  limit: Optional[int] = None) -> List[FundingLoan]:
        if symbol is None:
            endpoint = "auth/r/funding/loans/hist"
        else: endpoint = f"auth/r/funding/loans/{symbol}/hist"

        return [ serializers.FundingLoan.parse(*sub_data) \
            for sub_data in self._post(endpoint, \
                body={ "start": start, "end": end, "limit": limit }) ]

    def get_funding_credits(self, *, symbol: Optional[str] = None) -> List[FundingCredit]:
        if symbol is None:
            endpoint = "auth/r/funding/credits"
        else: endpoint = f"auth/r/funding/credits/{symbol}"

        return [ serializers.FundingCredit.parse(*sub_data) \
            for sub_data in self._post(endpoint) ]

    def get_funding_credits_history(self,
                                    *,
                                    symbol: Optional[str] = None,
                                    start: Optional[str] = None,
                                    end: Optional[str] = None,
                                    limit: Optional[int] = None) -> List[FundingCredit]:
        if symbol is None:
            endpoint = "auth/r/funding/credits/hist"
        else: endpoint = f"auth/r/funding/credits/{symbol}/hist"

        return [ serializers.FundingCredit.parse(*sub_data) \
            for sub_data in self._post(endpoint, \
                body={ "start": start, "end": end, "limit": limit }) ]

    def get_funding_trades_history(self,
                                   *,
                                   symbol: Optional[str] = None,
                                   sort: Optional[Sort] = None,
                                   start: Optional[str] = None,
                                   end: Optional[str] = None,
                                   limit: Optional[int] = None) -> List[FundingTrade]:
        if symbol is None:
            endpoint = "auth/r/funding/trades/hist"
        else: endpoint = f"auth/r/funding/trades/{symbol}/hist"

        body = {
            "sort": sort, "start": start, "end": end, 
            "limit": limit }

        return [ serializers.FundingTrade.parse(*sub_data) \
            for sub_data in self._post(endpoint, body=body) ]

    def get_funding_info(self, key: str) -> FundingInfo:
        return serializers.FundingInfo \
            .parse(*(self._post(f"auth/r/info/funding/{key}")[2]))

    #pylint: disable-next=too-many-arguments
    def transfer_between_wallets(self,
                                 from_wallet: str,
                                 to_wallet: str,
                                 currency: str,
                                 currency_to: str,
                                 amount: Union[Decimal, float, str]) -> Notification[Transfer]:
        body = {
            "from": from_wallet, "to": to_wallet, "currency": currency, 
            "currency_to": currency_to, "amount": amount
        }

        return _Notification[Transfer](serializers.Transfer) \
            .parse(*self._post("auth/w/transfer", body=body))

    def submit_wallet_withdrawal(self,
                                 wallet: str,
                                 method: str,
                                 address: str,
                                 amount: Union[Decimal, float, str]) -> Notification[Withdrawal]:
        body = {
            "wallet": wallet, "method": method, "address": address, 
            "amount": amount
        }

        return _Notification[Withdrawal](serializers.Withdrawal) \
            .parse(*self._post("auth/w/withdraw", body=body))

    def get_deposit_address(self,
                            wallet: str,
                            method: str,
                            renew: bool = False) -> Notification[DepositAddress]:
        return _Notification[DepositAddress](serializers.DepositAddress) \
            .parse(*self._post("auth/w/deposit/address", \
                body={ "wallet": wallet, "method": method, "renew": renew }))

    def generate_deposit_invoice(self,
                                 wallet: str,
                                 currency: str,
                                 amount: Union[Decimal, float, str]) -> LightningNetworkInvoice:
        return serializers.LightningNetworkInvoice \
            .parse(*self._post("auth/w/deposit/invoice", \
                body={ "wallet": wallet, "currency": currency, "amount": amount }))

    def get_movements(self,
                      *,
                      currency: Optional[str] = None,
                      start: Optional[str] = None,
                      end: Optional[str] = None,
                      limit: Optional[int] = None) -> List[Movement]:
        if currency is None:
            endpoint = "auth/r/movements/hist"
        else: endpoint = f"auth/r/movements/{currency}/hist"

        return [ serializers.Movement.parse(*sub_data) \
            for sub_data in self._post(endpoint, \
                body={ "start": start, "end": end, "limit": limit }) ]