trade.go

package uexchange

import (
	"encoding/json"
	"fmt"
	"net/url"
	"strconv"
)

func (c *Client) sendTradeTask(
	orderType string,
	pairSymbol string,
	amount,
	price float64,
) (int64, error) {
	reqFields, err := url.ParseQuery(fmt.Sprintf(
		"amount=%v&price=%v",
		amount, price,
	))
	if err != nil {
		return 0, fmt.Errorf("parse query: %w", err)
	}
	reqFields.Add("pair", pairSymbol)

	body, err := c.sendRequest(c.getAPIURL("market/"+orderType), requestTypePOST, reqFields)
	if err != nil {
		return 0, err
	}

	// decode response
	var response APITradeResponse
	err = json.Unmarshal(body, &response)
	if err != nil {
		return 0, fmt.Errorf("decode response: %w", err)
	}

	if !response.Success {
		return 0, fmt.Errorf("send trade task: %s", response.Error)
	}
	return response.Result.OrderID, nil
}

// Buy currency. returns order ID
func (c *Client) Buy(pairSymbol string, amount, price float64) (int64, error) {
	return c.sendTradeTask("buy", pairSymbol, amount, price)
}

// Sell currency. returns order ID
func (c *Client) Sell(pairSymbol string, amount, price float64) (int64, error) {
	return c.sendTradeTask("sell", pairSymbol, amount, price)
}

// Hold or Unhold order
func (c *Client) Hold(orderID int64) error {
	reqFields := url.Values{}
	reqFields.Add("order_id", strconv.FormatInt(orderID, 10))
	body, err := c.sendRequest(c.getAPIURL("market/hold"), requestTypePOST, reqFields)
	if err != nil {
		return err
	}

	// decode response
	var response APITradeResponse
	err = json.Unmarshal(body, &response)
	if err != nil {
		return fmt.Errorf("decode response: %w", err)
	}

	if !response.Success {
		return fmt.Errorf("send trade task: %s", response.Error)
	}
	return nil
}

// Cancel the specified order
func (c *Client) Cancel(orderID int64) error {
	reqFields := url.Values{}
	reqFields.Add("order_id", strconv.FormatInt(orderID, 10))
	body, err := c.sendRequest(c.getAPIURL("market/cancel"), "POST", reqFields)
	if err != nil {
		return err
	}

	// decode response
	var response APITradeResponse
	err = json.Unmarshal(body, &response)
	if err != nil {
		return fmt.Errorf("decode response: %w", err)
	}

	if !response.Success {
		return fmt.Errorf("send trade task: %s", response.Error)
	}
	return nil
}

// GetPairs - get trading pairs list
func (c *Client) GetPairs() ([]PairsDataContainer, error) {
	body, err := c.sendRequest(c.getAPIURL("market/pairs"), requestTypeGET, url.Values{})
	if err != nil {
		return nil, err
	}

	// decode response
	var response APIPairsResponse
	err = json.Unmarshal(body, &response)
	if err != nil {
		return nil, fmt.Errorf("decode response: %w", err)
	}

	if !response.Success {
		return nil, fmt.Errorf("send trade task: %s", response.Error)
	}
	return response.Result.Pairs, nil
}

// GetOrderBook by trade pair
func (c *Client) GetOrderBook(pairSymbol string) (*BookValueDataContainer, error) {
	reqFields := url.Values{}
	reqFields.Add("pair", pairSymbol)
	body, err := c.sendRequest(c.getAPIURL("market/panel"), requestTypePOST, reqFields)
	if err != nil {
		return nil, err
	}

	// decode response
	var response APIBookValueResponse
	err = json.Unmarshal(body, &response)
	if err != nil {
		return nil, fmt.Errorf("decode response: %w", err)
	}

	if !response.Success {
		return nil, fmt.Errorf("send trade task: %s", response.Error)
	}
	return &response.Result, nil
}

// GetMarketCurrenciesList - get exchange currencies list
func (c *Client) GetMarketCurrenciesList(pairSymbol string) (*CurrenciesListData, error) {
	reqFields := url.Values{}
	reqFields.Add("pair", pairSymbol)
	body, err := c.sendRequest(c.getAPIURL("market/curlist"), "GET", reqFields)
	if err != nil {
		return nil, err
	}

	// decode response
	var response APICurrenciesListResponse
	err = json.Unmarshal(body, &response)
	if err != nil {
		return nil, fmt.Errorf("decode response: %w", err)
	}

	if !response.Success {
		return nil, fmt.Errorf("send trade task: %s", response.Error)
	}
	return &response.Result, nil
}

func (c *Client) GetPairPrice(pairCode string) (PairPriceData, error) {
	result := PairPriceData{
		PairCode: pairCode,
	}
	pairs, err := c.GetPairs()
	if err != nil {
		return result, fmt.Errorf("get pairs: %w", err)
	}

	for i := 0; i < len(pairs); i++ {
		pairData := pairs[i]
		if pairData.Pair.PairCode == pairCode {
			bookData, err := c.GetOrderBook(pairData.Pair.PairCode)
			if err != nil {
				return result, fmt.Errorf("get order book: %w", err)
			}

			if len(bookData.Buy) > 0 {
				result.BestAskPrice = bookData.Buy[0].Price
			}
			if len(bookData.Sell) > 0 {
				result.BestBidPrice = bookData.Sell[0].Price
			}
			return result, nil
		} else {
			continue
		}
	}

	return result, fmt.Errorf("pair %s not found", pairCode)
}