pkg

type CanOrder struct {
	Symbol   string
	Exchange string
}

type Channel string

type Config struct {
	TgBotApiToken string
	TgChatid      int64
	WeiboUid      []int
	MergeMessage  bool
	Interval      int
	SavePicLocal  bool
	SendLivePics  bool
}

type Depth string

type Dispatcher func(e interface{})

type EndpointCreator func(ctx context.Context) (string, error)

type Kline struct {
	E      string `json:"e"`
	E1     int    `json:"E"`
	Symbol string `json:"s"`
	Kline  struct {
		T               int    `json:"t"`
		T1              int    `json:"T"`
		S               string `json:"s"`
		I               string `json:"i"`
		F               int    `json:"f"`
		L               int    `json:"L"`
		Open            string `json:"o"`
		Close           string `json:"c"`
		High            string `json:"h"`
		Low             string `json:"l"`
		Volume          string `json:"v"`
		Number          int    `json:"n"`
		X               bool   `json:"x"`
		Quantity        string `json:"q"`
		TakeBuyVolume   string `json:"V"`
		TakeBuyQuantity string `json:"Q"`
		B               string `json:"B"`
	} `json:"k"`
}

type KlineMessage struct {
	Data Kline `json:"data"`
}

type OrderInfo struct {
	Symbol   string
	Exchange string
}

type Parser func(message []byte) (interface{}, error)

type RuelEngine struct {
	Keys []string
	RuleEngine
	CanOrder chan *CanOrder
}

type RuleEngine struct {
	Rules  map[string]func(data map[string]interface{}) bool
	Rows   [][]interface{}
	Data   dataframe.DataFrame
	Titles []string
	Time   []string
	Source []string
	Url    []string
}

type Speed string

type StandardStream struct {

	// Conn is the websocket connection
	Conn *websocket.Conn

	// ConnCtx is the context of the current websocket connection
	ConnCtx context.Context

	// ConnCancel is the cancel funcion of the current websocket connection
	ConnCancel context.CancelFunc

	// ConnLock is used for locking Conn, ConnCtx and ConnCancel fields.
	// When changing these field values, be sure to call ConnLock
	ConnLock sync.Mutex

	PublicOnly bool

	// ReconnectC is a signal channel for reconnecting
	ReconnectC chan struct{}

	// CloseC is a signal channel for closing stream
	CloseC chan struct{}

	Subscriptions []Subscription
	// contains filtered or unexported fields
}

type StandardStreamEmitter interface {
	Stream
	EmitStart()
	EmitConnect()
	EmitDisconnect()
}

type StandardStreamEventHub interface {
	OnStart(cb func())

	OnConnect(cb func())

	OnDisconnect(cb func())

	OnTradeUpdate(cb func())
}

type Stream interface {
	StandardStreamEventHub

	Subscribe(channel Channel, symbol string, options SubscribeOptions)
	GetSubscriptions() []Subscription
	SetPublicOnly()
	GetPublicOnly() bool
	Connect(ctx context.Context) error
	Close() error
}

type SubscribeOptions struct {
	Depth Depth `json:"depth,omitempty"`
	Speed Speed `json:"speed,omitempty"`
}

type Subscription struct {
	Symbol  string           `json:"symbol"`
	Channel Channel          `json:"channel"`
	Options SubscribeOptions `json:"options"`
}

type Symbol struct {
	Symbol string `json:"symbol"`
}

type Ticker struct {
	E            string `json:"e"`
	E1           int    `json:"E"`
	Symbol       string `json:"s"`
	P            string `json:"p"`
	P1           string `json:"P"`
	W            string `json:"w"`
	CurrentPrice string `json:"c"`
	Quantity     string `json:"Q"`
	O            string `json:"o"`
	H            string `json:"h"`
	L            string `json:"l"`
	V            string `json:"v"`
	Q1           string `json:"q"`
	O1           int    `json:"O"`
	C1           int    `json:"C"`
	F            int    `json:"F"`
	L1           int    `json:"L"`
	N            int    `json:"n"`
}

type TickerMessage struct {
	Data Ticker `json:"data"`
}

type TradeBot struct {
	NewsStream    *news.Stream
	RuelEngine    *RuelEngine
	BinanceFuture *exchange.BinanceFuture
	CanOrder      chan *CanOrder
}