private.go

package ws

import (
	"encoding/json"
	"github.com/drinkthere/okx"
	"github.com/drinkthere/okx/events"
	"github.com/drinkthere/okx/events/private"
	requests "github.com/drinkthere/okx/requests/ws/private"
)

// Private
//
// https://www.okx.com/docs-v5/en/#websocket-api-private-channel
type Private struct {
	*ClientWs
	aCh   chan *private.Account
	pCh   chan *private.Position
	bnpCh chan *private.BalanceAndPosition
	oCh   chan *private.Order
	aoCh  chan *private.AlgoOrder
	aaoCh chan *private.AlgoOrder
}

// NewPrivate returns a pointer to a fresh Private
func NewPrivate(c *ClientWs) *Private {
	return &Private{ClientWs: c}
}

// Account
// Retrieve account information. Data will be pushed when triggered by events such as placing/canceling order, and will also be pushed in regular interval according to subscription granularity.
//
// https://www.okx.com/docs-v5/en/#websocket-api-private-channel-account-channel
func (c *Private) Account(req requests.Account, ch ...chan *private.Account) error {
	m := okx.S2M(req)
	if len(ch) > 0 {
		c.aCh = ch[0]
	}
	return c.Subscribe(true, []okx.ChannelName{"account"}, m)
}

// UAccount
//
// https://www.okx.com/docs-v5/en/#websocket-api-private-channel-account-channel
func (c *Private) UAccount(req requests.Account, rCh ...bool) error {
	m := okx.S2M(req)
	if len(rCh) > 0 && rCh[0] {
		c.aCh = nil
	}
	return c.Unsubscribe(true, []okx.ChannelName{"account"}, m)
}

// Position
// Retrieve position information. Initial snapshot will be pushed according to subscription granularity. Data will be pushed when triggered by events such as placing/canceling order, and will also be pushed in regular interval according to subscription granularity.
//
// https://www.okx.com/docs-v5/en/#websocket-api-private-channel-positions-channel
func (c *Private) Position(req requests.Position, ch ...chan *private.Position) error {
	m := okx.S2M(req)
	if len(ch) > 0 {
		c.pCh = ch[0]
	}
	return c.Subscribe(true, []okx.ChannelName{"positions"}, m)
}

// UPosition
//
// https://www.okx.com/docs-v5/en/#websocket-api-private-channel-positions-channel
func (c *Private) UPosition(req requests.Position, rCh ...bool) error {
	m := okx.S2M(req)
	if len(rCh) > 0 && rCh[0] {
		c.pCh = nil
	}
	return c.Unsubscribe(true, []okx.ChannelName{"positions"}, m)
}

// BalanceAndPosition
// Retrieve account balance and position information. Data will be pushed when triggered by events such as filled order, funding transfer.
//
// https://www.okx.com/docs-v5/en/#websocket-api-private-channel-balance-and-position-channel
func (c *Private) BalanceAndPosition(ch ...chan *private.BalanceAndPosition) error {
	m := make(map[string]string)
	if len(ch) > 0 {
		c.bnpCh = ch[0]
	}
	return c.Subscribe(true, []okx.ChannelName{"balance_and_position"}, m)
}

// UBalanceAndPosition unsubscribes a position channel
//
// https://www.okx.com/docs-v5/en/#websocket-api-private-channel-balance-and-position-channel
func (c *Private) UBalanceAndPosition(rCh ...bool) error {
	m := make(map[string]string)
	if len(rCh) > 0 && rCh[0] {
		c.bnpCh = nil
	}
	return c.Unsubscribe(true, []okx.ChannelName{"balance_and_position"}, m)
}

// Order
// Retrieve position information. Initial snapshot will be pushed according to subscription granularity. Data will be pushed when triggered by events such as placing/canceling order, and will also be pushed in regular interval according to subscription granularity.
//
// https://www.okx.com/docs-v5/en/#websocket-api-private-channel-order-channel
func (c *Private) Order(req requests.Order, ch ...chan *private.Order) error {
	m := okx.S2M(req)
	if len(ch) > 0 {
		c.oCh = ch[0]
	}
	return c.Subscribe(true, []okx.ChannelName{"orders"}, m)
}

// UOrder
//
// https://www.okx.com/docs-v5/en/#websocket-api-private-channel-order-channel
func (c *Private) UOrder(req requests.Order, rCh ...bool) error {
	m := okx.S2M(req)
	if len(rCh) > 0 && rCh[0] {
		c.oCh = nil
	}
	return c.Unsubscribe(true, []okx.ChannelName{"orders"}, m)
}

// Algo Order
// Retrieve position information. Initial snapshot will be pushed according to subscription granularity. Data will be pushed when triggered by events such as placing/canceling algo order, and will also be pushed in regular interval according to subscription granularity.
//
// https://www.okx.com/docs-v5/en/#websocket-api-private-channel-algo-orders-channel
func (c *Private) AlgoOrder(req requests.AlgoOrder, ch ...chan *private.AlgoOrder) error {
	m := okx.S2M(req)
	if len(ch) > 0 {
		c.aoCh = ch[0]
	}
	return c.Subscribe(true, []okx.ChannelName{"orders-algo"}, m)
}

// UOrder
//
// https://www.okx.com/docs-v5/en/#websocket-api-private-channel-order-channel
func (c *Private) UAlgoOrder(req requests.AlgoOrder, rCh ...bool) error {
	m := okx.S2M(req)
	if len(rCh) > 0 && rCh[0] {
		c.aoCh = nil
	}
	return c.Unsubscribe(true, []okx.ChannelName{"orders-algo"}, m)
}

// Algo Order
// Retrieve position information. Initial snapshot will be pushed according to subscription granularity. Data will be pushed when triggered by events such as placing/canceling advanced algo order, and will also be pushed in regular interval according to subscription granularity.
//
// https://www.okx.com/docs-v5/en/#websocket-api-private-channel-algo-orders-channel
func (c *Private) AdvancedAlgoOrder(req requests.AlgoOrder, ch ...chan *private.AlgoOrder) error {
	m := okx.S2M(req)
	if len(ch) > 0 {
		c.aaoCh = ch[0]
	}
	return c.Subscribe(true, []okx.ChannelName{"algo-advance"}, m)
}

// UOrder
//
// https://www.okx.com/docs-v5/en/#websocket-api-private-channel-order-channel
func (c *Private) UAdvancedAlgoOrder(req requests.AlgoOrder, rCh ...bool) error {
	m := okx.S2M(req)
	if len(rCh) > 0 && rCh[0] {
		c.aaoCh = nil
	}
	return c.Unsubscribe(true, []okx.ChannelName{"algo-advance"}, m)
}

func (c *Private) Process(data []byte, e *events.Basic) bool {
	if e.Event == "" && e.Arg != nil && e.Data != nil && len(e.Data) > 0 {
		ch, ok := e.Arg.Get("channel")
		if !ok {
			return false
		}
		switch ch {
		case "account":
			e := private.Account{}
			err := json.Unmarshal(data, &e)
			if err != nil {
				return false
			}
			go func() {
				if c.aCh != nil {
					c.aCh <- &e
				}
			}()
			return true
		case "positions":
			e := private.Position{}
			err := json.Unmarshal(data, &e)
			if err != nil {
				return false
			}
			go func() {
				if c.pCh != nil {
					c.pCh <- &e
				}
			}()
			return true
		case "balance_and_position":
			e := private.BalanceAndPosition{}
			err := json.Unmarshal(data, &e)
			if err != nil {
				return false
			}
			go func() {
				if c.bnpCh != nil {
					c.bnpCh <- &e
				}
			}()
			return true
		case "orders":
			e := private.Order{}
			err := json.Unmarshal(data, &e)
			if err != nil {
				return false
			}
			go func() {
				if c.oCh != nil {
					c.oCh <- &e
				}
			}()
			return true
		case "orders-algo":
			e := private.AlgoOrder{}
			err := json.Unmarshal(data, &e)
			if err != nil {
				return false
			}
			go func() {
				if c.aoCh != nil {
					c.aoCh <- &e
				}
			}()
			return true
		case "algo-advance":
			e := private.AlgoOrder{}
			err := json.Unmarshal(data, &e)
			if err != nil {
				return false
			}
			go func() {
				if c.aaoCh != nil {
					c.aaoCh <- &e
				}
			}()
			return true
		}
	}
	return false
}