bbgo/pkg/types/side.go

package types

import (
	"encoding/json"
	"strings"

	"github.com/pkg/errors"

	"git.qtrade.icu/lychiyu/bbgo/pkg/style"
)

// SideType define side type of order
type SideType string

const (
	SideTypeBuy  = SideType("BUY")
	SideTypeSell = SideType("SELL")
	SideTypeSelf = SideType("SELF")

	// SideTypeBoth is only used for the configuration context
	SideTypeBoth = SideType("BOTH")
)

var ErrInvalidSideType = errors.New("invalid side type")

func StrToSideType(s string) (side SideType, err error) {
	switch strings.ToLower(s) {
	case "buy":
		side = SideTypeBuy

	case "sell":
		side = SideTypeSell

	case "both":
		side = SideTypeBoth

	default:
		err = ErrInvalidSideType
		return side, err

	}

	return side, err
}

func (side *SideType) UnmarshalJSON(data []byte) error {
	var s string
	if err := json.Unmarshal(data, &s); err != nil {
		return err
	}

	ss, err := StrToSideType(s)
	if err != nil {
		return err
	}

	*side = ss
	return nil
}

func (side SideType) Reverse() SideType {
	switch side {
	case SideTypeBuy:
		return SideTypeSell

	case SideTypeSell:
		return SideTypeBuy
	}

	return side
}

func (side SideType) String() string {
	return string(side)
}

func (side SideType) Color() string {
	if side == SideTypeBuy {
		return style.GreenColor
	}

	if side == SideTypeSell {
		return style.RedColor
	}

	return style.GrayColor
}

func SideToColorName(side SideType) string {
	return side.Color()
}