package trade

import (
	"bytes"
	"encoding/json"
	"fmt"
	"math/rand"
	"os"
	"strconv"
	"strings"
	"time"
)

func GetSide(in string) string {
	in = strings.TrimSpace(in)

	if in == "buy" || in == "sell" {
		return in
	}

	switch strings.ToUpper(in) {
	case "LONG":
		return "buy"
	case "SHORT":
		return "sell"
	}

	return ""
}



// 根据持仓状态执行相应的方法
func GetPositionStats(LongExists, ShortExists bool) PositionStatus {
	switch {
	case !LongExists && !ShortExists:
		return NoPositions
	case LongExists && !ShortExists:
		return LongOnly
	case !LongExists && ShortExists:
		return ShortOnly
	case LongExists && ShortExists:
		return BothPositions
	}

	return -1
}

func FmtNumber(in float64, place int) string {
	if place == 0 {
		return fmt.Sprintf("%.0f", in)
	}
	if place == 1 {
		return fmt.Sprintf("%.1f", in)
	}
	if place == 2 {
		return fmt.Sprintf("%.2f", in)
	}
	if place == 3 {
		return fmt.Sprintf("%.3f", in)
	}
	if place == 4 {
		return fmt.Sprintf("%.4f", in)
	}
	if place == 5 {
		return fmt.Sprintf("%.5f", in)
	}

	return fmt.Sprintf("%.6f", in)
}

func FloatRound(in float64, place int) float64 {
	// 限制 place 范围在合理区间
	if place < 0 {
		place = 0
	}

	// 使用 strconv.FormatFloat 直接格式化，避免多次条件判断
	str := strconv.FormatFloat(in, 'f', place, 64)
	num, _ := strconv.ParseFloat(str, 64)
	return num
}

func RateToFloat64(s string) float64 {
	// 去掉百分号
	s = strings.TrimSuffix(s, "%")
	// 转换为 float64
	f, err := strconv.ParseFloat(s, 64)
	if err != nil {
		return 0
	}
	return f
}
func Output(v any) {
	jsonBy, _ := json.Marshal(v)
	var out bytes.Buffer
	json.Indent(&out, jsonBy, "", "\t")
	out.WriteTo(os.Stdout)
	fmt.Printf("\n")
}

func GenClientId() string {
	cTime := time.Now().Format("20060102150405")

	rand.Seed(time.Now().UnixNano())
	random := fmt.Sprintf("%04d", rand.Intn(10000))
	return cTime + random
}