accounts.go

package tasty

import (
	"fmt"
	"net/http"
)

// Get the accounts for the authenticated client.
func (c *Client) GetMyAccounts() ([]Account, *http.Response, error) {
	path := "/customers/me/accounts"

	type accountResponse struct {
		Data struct {
			Items []struct {
				Account Account `json:"account"`
			} `json:"items"`
		} `json:"data"`
	}

	accountsRes := new(accountResponse)

	resp, err := c.request(http.MethodGet, path, nil, nil, accountsRes)
	if err != nil {
		return []Account{}, resp, err
	}

	var accounts []Account

	for _, acct := range accountsRes.Data.Items {
		accounts = append(accounts, acct.Account)
	}

	return accounts, resp, nil
}

// Returns current trading status for an account.
func (c *Client) GetAccountTradingStatus(accountNumber string) (AccountTradingStatus, *http.Response, error) {
	path := fmt.Sprintf("/accounts/%s/trading-status", accountNumber)

	type tradingStatusRes struct {
		AccountTradingStatus AccountTradingStatus `json:"data"`
	}
	accountsRes := new(tradingStatusRes)

	resp, err := c.request(http.MethodGet, path, nil, nil, accountsRes)
	if err != nil {
		return AccountTradingStatus{}, resp, err
	}

	return accountsRes.AccountTradingStatus, resp, nil
}

// Returns the current balance values for an account.
func (c *Client) GetAccountBalances(accountNumber string) (AccountBalance, *http.Response, error) {
	path := fmt.Sprintf("/accounts/%s/balances", accountNumber)

	type accountBalanceRes struct {
		AccountBalance AccountBalance `json:"data"`
	}
	accountsRes := new(accountBalanceRes)

	resp, err := c.request(http.MethodGet, path, nil, nil, accountsRes)
	if err != nil {
		return AccountBalance{}, resp, err
	}

	return accountsRes.AccountBalance, resp, nil
}

// Returns a list of the account's positions.
// Can be filtered by symbol, underlying_symbol.
func (c *Client) GetAccountPositions(accountNumber string, query AccountPositionQuery) ([]AccountPosition, *http.Response, error) {
	path := fmt.Sprintf("/accounts/%s/positions", accountNumber)

	type accountResponse struct {
		Data struct {
			AccountPositions []AccountPosition `json:"items"`
		} `json:"data"`
	}

	accountsRes := new(accountResponse)

	resp, err := c.request(http.MethodGet, path, query, nil, accountsRes)
	if err != nil {
		return []AccountPosition{}, resp, err
	}

	return accountsRes.Data.AccountPositions, resp, nil
}

// Returns most recent snapshot and current balance for an account.
func (c *Client) GetAccountBalanceSnapshots(accountNumber string, query AccountBalanceSnapshotsQuery) ([]AccountBalanceSnapshots, *http.Response, error) {
	// Default to EOD
	if query.TimeOfDay == "" {
		query.TimeOfDay = EndOfDay
	}

	path := fmt.Sprintf("/accounts/%s/balance-snapshots", accountNumber)

	type accountResponse struct {
		Data struct {
			AccountBalanceSnapshots []AccountBalanceSnapshots `json:"items"`
		} `json:"data"`
	}

	accountsRes := new(accountResponse)

	resp, err := c.request(http.MethodGet, path, query, nil, accountsRes)
	if err != nil {
		return []AccountBalanceSnapshots{}, resp, err
	}

	return accountsRes.Data.AccountBalanceSnapshots, resp, nil
}

// Returns a list of account net liquidating value snapshots.
func (c *Client) GetAccountNetLiqHistory(accountNumber string, query HistoricLiquidityQuery) ([]NetLiqOHLC, *http.Response, error) {
	path := fmt.Sprintf("/accounts/%s/net-liq/history", accountNumber)

	type accountResponse struct {
		Data struct {
			HistoricLiquidity []NetLiqOHLC `json:"items"`
		} `json:"data"`
	}

	accountsRes := new(accountResponse)

	resp, err := c.request(http.MethodGet, path, query, nil, accountsRes)
	if err != nil {
		return []NetLiqOHLC{}, resp, err
	}

	return accountsRes.Data.HistoricLiquidity, resp, nil
}

// Get the position limit.
func (c *Client) GetAccountPositionLimit(accountNumber string) (PositionLimit, *http.Response, error) {
	path := fmt.Sprintf("/accounts/%s/position-limit", accountNumber)

	type accountResponse struct {
		PositionLimit PositionLimit `json:"data"`
	}

	accountsRes := new(accountResponse)

	resp, err := c.request(http.MethodGet, path, nil, nil, accountsRes)
	if err != nil {
		return PositionLimit{}, resp, err
	}

	return accountsRes.PositionLimit, resp, nil
}