package ws

import (
	"encoding/json"
	"errors"
	okex "gitee.com/zqs118/go-okx"
	"gitee.com/zqs118/go-okx/common"
	"github.com/gorilla/websocket"
)

type Handler interface {
	handle(message []byte) error
}

type WebsocketTrade struct {
	conf       *okex.Config
	conn       *websocket.Conn
	errHandler ErrHandler
	Handler    Handler
}

func NewWebsocketTrade(conf *okex.Config, errHandler ErrHandler) (w *WebsocketTrade) {
	return &WebsocketTrade{
		conf:       conf,
		errHandler: errHandler,
	}
}

func (w *WebsocketTrade) Dail() (doneC, stopC chan struct{}) {
	conn, err := dail(PRIVATE)
	if err != nil {
		w.errHandler(err)
	}
	w.conn = conn
	err = login(conn, w.conf)
	if err != nil {
		w.errHandler(err)
	}
	doneC = make(chan struct{})
	stopC = make(chan struct{})
	go func() {
		// 出现错误或者收到停止指令时关闭链接
		defer close(doneC)
		if KeepAlive {
			keepAlive(conn)
		}
		// 接收指令等待关闭连接
		silent := false
		go func() {
			select {
			case <-stopC:
				silent = true
			case <-doneC:
			}
			conn.Close()
		}()
		for {
			_, message, err := conn.ReadMessage()
			if err != nil {
				if !silent {
					w.errHandler(err)
				}
				return
			}
			err = w.Handler.handle(message)
			if err != nil {
				w.errHandler(err)
			}
		}
	}()
	return
}

// CreateOrder 创建订单
type CreateOrder struct {
	Id      string         `json:"id,omitempty"`
	Op      common.Operate `json:"op,omitempty"`
	ExpTime string         `json:"expTime,omitempty"`
	Args    []*OrderArgs   `json:"args,omitempty"`
}

func NewCreateOrder(op common.Operate) *CreateOrder {
	return &CreateOrder{
		Op: op,
	}
}

func NewOrder() *CreateOrder {
	return NewCreateOrder(common.OrderOp)
}
func NewBatchOrder() *CreateOrder {
	return NewCreateOrder(common.BatchOrderOp)
}

func (o *CreateOrder) SetId(id string) *CreateOrder {
	o.Id = id
	return o
}

func (o *CreateOrder) SetExpTime(expTime string) *CreateOrder {
	o.ExpTime = expTime
	return o
}

func (o *CreateOrder) SetArgs(args *OrderArgs) *CreateOrder {
	o.Args = append(o.Args, args)
	return o
}

type OrderArgs struct {
	Side       common.SideType  `json:"side,omitempty"`    // 订单方向
	PosSide    common.PosSide   `json:"posSide,omitempty"` // 持仓方向
	InstID     string           `json:"instId,omitempty"`
	TdMode     common.TdMode    `json:"tdMode,omitempty"`
	OrdType    common.OrderType `json:"ordType,omitempty"`
	Sz         string           `json:"sz,omitempty"`
	Ccy        string           `json:"ccy,omitempty"`
	ClOrdID    string           `json:"clOrdId,omitempty"`
	Tag        string           `json:"tag,omitempty"`
	Px         string           `json:"px,omitempty"`
	ReduceOnly string           `json:"reduceOnly,omitempty"`
	TgtCcy     string           `json:"tgtCcy,omitempty"`
	BanAmend   string           `json:"banAmend,omitempty"`
}

func NewOrderArgs() *OrderArgs {
	return new(OrderArgs)
}

func (o *OrderArgs) SetSide(side common.SideType) *OrderArgs {
	o.Side = side
	return o
}

func (o *OrderArgs) SetPosSide(posSide common.PosSide) *OrderArgs {
	o.PosSide = posSide
	return o
}

func (o *OrderArgs) SetInstID(instID string) *OrderArgs {
	o.InstID = instID
	return o
}

func (o *OrderArgs) SetTdMode(tdMode common.TdMode) *OrderArgs {
	o.TdMode = tdMode
	return o
}

func (o *OrderArgs) SetOrdType(ordType common.OrderType) *OrderArgs {
	o.OrdType = ordType
	return o
}

func (o *OrderArgs) SetSz(sz string) *OrderArgs {
	o.Sz = sz
	return o
}

func (o *OrderArgs) SetCcy(ccy string) *OrderArgs {
	o.Ccy = ccy
	return o
}

func (o *OrderArgs) SetClOrdID(clOrdID string) *OrderArgs {
	o.ClOrdID = clOrdID
	return o
}

func (o *OrderArgs) SetTag(tag string) *OrderArgs {
	o.Tag = tag
	return o
}

func (o *OrderArgs) SetPx(px string) *OrderArgs {
	o.Px = px
	return o
}

func (o *OrderArgs) SetReduceOnly(reduceOnly string) *OrderArgs {
	o.ReduceOnly = reduceOnly
	return o
}

func (o *OrderArgs) SetTgtCcy(tgtCcy string) *OrderArgs {
	o.TgtCcy = tgtCcy
	return o
}

func (o *OrderArgs) SetBanAmend(banAmend string) *OrderArgs {
	o.BanAmend = banAmend
	return o
}

type CreateOrderEvent struct {
	Done bool
	ID   string      `json:"id"`
	Op   string      `json:"op"`
	Data []OrderData `json:"data"`
	Code string      `json:"code"`
	Msg  string      `json:"msg"`
}

type OrderData struct {
	ClOrdID string `json:"clOrdId"`
	OrdID   string `json:"ordId"`
	Tag     string `json:"tag"`
	SCode   string `json:"sCode"`
	SMsg    string `json:"sMsg"`
}

func (c *CreateOrderEvent) handle(message []byte) (err error) {
	m, err := common.NewJSON(message)
	if err != nil {
		return err
	}
	if data, ok := m.CheckGet("event"); ok {
		if data.MustString() == "error" {
			c.Done = true
			return errors.New("\n  event: " + data.MustString() + "\n  code: " + m.Get("code").MustString() + "\n  msg: " + m.Get("msg").MustString())
		}
	} else {
		err = json.Unmarshal(message, &c)
		c.Done = true
		if err != nil {
			return
		}
	}
	return
}

func (w *WebsocketTrade) Order(param *CreateOrder) (event CreateOrderEvent, err error) {
	w.Handler = &event
	err = w.conn.WriteJSON(param)
	if err != nil {
		return
	}
	for {
		if event.Done {
			return
		}
	}
}