cast/cast.go

package cast

import (
	"bytes"
	"encoding/json"
	"fmt"
	"reflect"
	"slices"
	"strconv"
	"strings"
	"time"

	"gopkg.in/yaml.v3"
)

// If 泛型三元表达式
func If[T any](condition bool, trueVal, falseVal T) T {
	if condition {
		return trueVal
	}
	return falseVal
}

// In 泛型包含判断
func In[T comparable](arr []T, val T) bool {
	return slices.Contains(arr, val)
}

func RealValue(v reflect.Value) reflect.Value {
	for v.Kind() == reflect.Pointer || v.Kind() == reflect.Interface {
		if v.IsNil() {
			return v
		}
		v = v.Elem()
	}
	return v
}

// --- Core Cast Logic ---

func parseInt(s string) int64 {
	i, err := strconv.ParseInt(s, 10, 64)
	if err == nil {
		return i
	}
	if f, err := strconv.ParseFloat(s, 64); err == nil {
		return int64(f)
	}
	return 0
}

func parseUint(s string) uint64 {
	i, err := strconv.ParseUint(s, 10, 64)
	if err == nil {
		return i
	}
	if f, err := strconv.ParseFloat(s, 64); err == nil {
		return uint64(f)
	}
	return 0
}

func Int(value any) int { return int(Int64(value)) }

// Helper for integer coercion to avoid repetition
func toInt64(value any) (int64, bool) {
	switch v := value.(type) {
	case int: return int64(v), true
	case int8: return int64(v), true
	case int16: return int64(v), true
	case int32: return int64(v), true
	case int64: return v, true
	case uint: return int64(v), true
	case uint8: return int64(v), true
	case uint16: return int64(v), true
	case uint32: return int64(v), true
	case uint64: return int64(v), true
	case float32: return int64(v), true
	case float64: return int64(v), true
	case bool: return If(v, int64(1), int64(0)), true
	}
	return 0, false
}

func Int64(value any) int64 {
	if value == nil { return 0 }
	if i, ok := toInt64(value); ok { return i }
	
	switch realValue := value.(type) {
	case []byte: return parseInt(string(realValue))
	case string: return parseInt(realValue)
	}
	
	rv := reflect.ValueOf(value)
	if rv.Kind() == reflect.Pointer || rv.Kind() == reflect.Interface {
		if rv = RealValue(rv); rv.IsValid() && rv.CanInterface() && rv.Kind() != reflect.Pointer {
			return Int64(rv.Interface())
		}
	}
	return 0
}

func Uint(value any) uint { return uint(Uint64(value)) }

func Uint64(value any) uint64 {
	if value == nil {
		return 0
	}
	switch realValue := value.(type) {
	case int:
		return uint64(realValue)
	case int8:
		return uint64(realValue)
	case int16:
		return uint64(realValue)
	case int32:
		return uint64(realValue)
	case int64:
		return uint64(realValue)
	case uint:
		return uint64(realValue)
	case uint8:
		return uint64(realValue)
	case uint16:
		return uint64(realValue)
	case uint32:
		return uint64(realValue)
	case uint64:
		return realValue
	case float32:
		return uint64(realValue)
	case float64:
		return uint64(realValue)
	case bool:
		return If(realValue, uint64(1), uint64(0))
	case []byte:
		return parseUint(string(realValue))
	case string:
		return parseUint(realValue)
	}
	rv := reflect.ValueOf(value)
	if rv.Kind() == reflect.Pointer || rv.Kind() == reflect.Interface {
		if rv = RealValue(rv); rv.IsValid() && rv.CanInterface() && rv.Kind() != reflect.Pointer {
			return Uint64(rv.Interface())
		}
	}
	return 0
}

func Float(value any) float32 { return float32(Float64(value)) }

func Float64(value any) float64 {
	if value == nil {
		return 0
	}
	switch realValue := value.(type) {
	case int, int8, int16, int32, int64:
		return float64(Int64(realValue))
	case uint, uint8, uint16, uint32, uint64:
		return float64(Uint64(realValue))
	case float32:
		return float64(realValue)
	case float64:
		return realValue
	case bool:
		return If(realValue, 1.0, 0.0)
	case []byte:
		i, err := strconv.ParseFloat(string(realValue), 64)
		if err == nil {
			return i
		}
	case string:
		i, err := strconv.ParseFloat(realValue, 64)
		if err == nil {
			return i
		}
	}
	rv := reflect.ValueOf(value)
	if rv.Kind() == reflect.Pointer || rv.Kind() == reflect.Interface {
		if rv = RealValue(rv); rv.IsValid() && rv.CanInterface() && rv.Kind() != reflect.Pointer {
			return Float64(rv.Interface())
		}
	}
	return 0
}

func String(value any) string {
	if value == nil {
		return ""
	}
	switch realValue := value.(type) {
	case int, int8, int16, int32, int64:
		return strconv.FormatInt(Int64(realValue), 10)
	case uint, uint8, uint16, uint32, uint64:
		return strconv.FormatUint(Uint64(realValue), 10)
	case float32:
		return strconv.FormatFloat(float64(realValue), 'f', -1, 32)
	case float64:
		return strconv.FormatFloat(realValue, 'f', -1, 64)
	case bool:
		return If(realValue, "true", "false")
	case string:
		return realValue
	case []byte:
		return string(realValue)
	}
	rv := reflect.ValueOf(value)
	if rv.Kind() == reflect.Pointer || rv.Kind() == reflect.Interface {
		if rv = RealValue(rv); rv.IsValid() && rv.CanInterface() && rv.Kind() != reflect.Pointer {
			return String(rv.Interface())
		}
	}
	return fmt.Sprint(value)
}

func Bool(value any) bool {
	switch realValue := value.(type) {
	case int, int8, int16, int32, int64, uint, uint8, uint16, uint32, uint64, float32, float64:
		return Uint64(realValue) != 0
	case bool:
		return realValue
	case []byte:
		s := strings.ToLower(string(realValue))
		return s == "1" || s == "t" || s == "true"
	case string:
		s := strings.ToLower(realValue)
		return s == "1" || s == "t" || s == "true"
	}
	rv := reflect.ValueOf(value)
	if rv.Kind() == reflect.Pointer || rv.Kind() == reflect.Interface {
		if rv = RealValue(rv); rv.IsValid() && rv.CanInterface() && rv.Kind() != reflect.Pointer {
			return Bool(rv.Interface())
		}
	}
	return false
}

func Ints(value any) []int64 {
	switch realValue := value.(type) {
	case []any:
		result := make([]int64, len(realValue))
		for i, v := range realValue {
			result[i] = Int64(v)
		}
		return result
	case string:
		if strings.HasPrefix(realValue, "[") {
			result := make([]int64, 0)
			UnmarshalJSON(realValue, &result)
			return result
		}
		return []int64{Int64(value)}
	}
	rv := reflect.ValueOf(value)
	if rv.Kind() == reflect.Pointer || rv.Kind() == reflect.Interface {
		if rv = RealValue(rv); rv.IsValid() && rv.CanInterface() && rv.Kind() != reflect.Pointer {
			return Ints(rv.Interface())
		}
	}
	return []int64{Int64(value)}
}

func Strings(value any) []string {
	switch realValue := value.(type) {
	case []any:
		result := make([]string, len(realValue))
		for i, v := range realValue {
			result[i] = String(v)
		}
		return result
	case string:
		if strings.HasPrefix(realValue, "[") {
			result := make([]string, 0)
			UnmarshalJSON(realValue, &result)
			return result
		}
		return []string{String(value)}
	}

	rv := reflect.ValueOf(value)
	if rv.Kind() == reflect.Pointer || rv.Kind() == reflect.Interface {
		if rv = RealValue(rv); rv.IsValid() && rv.CanInterface() && rv.Kind() != reflect.Pointer {
			return Strings(rv.Interface())
		}
	}
	return []string{String(value)}
}

func Duration(value any) time.Duration {
	if value == nil {
		return 0
	}
	switch realValue := value.(type) {
	case int, int8, int16, int32, int64, uint, uint8, uint16, uint32, uint64:
		return time.Duration(Int64(realValue))
	case float32, float64:
		return time.Duration(Float64(realValue) * float64(time.Second))
	case []byte, string:
		if result, err := time.ParseDuration(String(realValue)); err == nil {
			return result
		}
	}
	rv := reflect.ValueOf(value)
	if rv.Kind() == reflect.Pointer || rv.Kind() == reflect.Interface {
		if rv = RealValue(rv); rv.IsValid() && rv.CanInterface() && rv.Kind() != reflect.Pointer {
			return Duration(rv.Interface())
		}
	}
	return 0
}

func ToJSONBytes(value any) ([]byte, error) {
	return fastToJSONBytes(value)
}

func ToJSONDesensitize(value any, keys []string) (string, error) {
	b, err := fastToJSONBytes(value, keys...)
	if err != nil {
		return "", err
	}
	return string(b), nil
}

func ToJSONDesensitizeBytes(value any, keys []string) ([]byte, error) {
	return fastToJSONBytes(value, keys...)
}

func MustJSONBytes(value any) []byte {
	j, err := ToJSONBytes(value)
	if err != nil {
		return []byte{}
	}
	return j
}

func ToJSON(value any) (string, error) {
	j, err := ToJSONBytes(value)
	if err != nil {
		return "", err
	}
	return string(j), nil
}
func MustToJSON(value any) string { return string(MustJSONBytes(value)) }

func PrettyToJSONBytes(value any) []byte {
	j := MustJSONBytes(value)
	r := &bytes.Buffer{}
	if err := json.Indent(r, j, "", "  "); err == nil {
		return r.Bytes()
	}
	return j
}

func PrettyToJSON(value any) string { return string(PrettyToJSONBytes(value)) }

func UnmarshalJSONBytes(data []byte, value any) (any, error) {
	if value == nil {
		var v any
		value = &v
	}
	err := fastUnmarshalJSONBytes(data, value)
	return value, err
}

func MustUnmarshalJSONBytes(data []byte, value any) any {
	v, err := UnmarshalJSONBytes(data, value)
	if err != nil {
		return nil
	}
	return v
}

func UnmarshalJSON(str string, value any) (any, error) { return UnmarshalJSONBytes([]byte(str), value) }

func MustUnmarshalJSON(str string, value any) any { return MustUnmarshalJSONBytes([]byte(str), value) }

func ToYAML(value any) (string, error) {
	j, err := YAMLBytes(value)
	if err != nil {
		return "", err
	}
	return string(j), nil
}

func MustToYAML(value any) string { return string(MustYAMLBytes(value)) }

func UnmarshalYAML(data string, value any) (any, error) {
	return UnmarshalYAMLBytes([]byte(data), value)
}

func MustUnmarshalYAML(data string, value any) any { return MustUnmarshalYAMLBytes([]byte(data), value) }

func YAMLBytes(value any) ([]byte, error) {
	j, err := yaml.Marshal(value)
	if err == nil {
		return j, nil
	}
	return []byte(fmt.Sprint(value)), err
}

func MustYAMLBytes(value any) []byte {
	j, err := YAMLBytes(value)
	if err != nil {
		return []byte{}
	}
	return j
}

func UnmarshalYAMLBytes(data []byte, value any) (any, error) {
	err := yaml.Unmarshal(data, value)
	return value, err
}

func MustUnmarshalYAMLBytes(data []byte, value any) any {
	v, err := UnmarshalYAMLBytes(data, value)
	if err != nil {
		return nil
	}
	return v
}

// --- Others (Keep logic but clean style) ---

func Split(s, sep string) []string {
	ss := strings.Split(s, sep)
	out := make([]string, 0, len(ss))
	for _, s1 := range ss {
		if s2 := strings.TrimSpace(s1); s2 != "" {
			out = append(out, s2)
		}
	}
	return out
}

func SplitArgs(s string) []string {
	var res []string
	var builder strings.Builder
	inQuote := false
	escaped := false

	chars := []rune(s)
	for i := 0; i < len(chars); i++ {
		c := chars[i]

		if escaped {
			builder.WriteRune(c)
			escaped = false
			continue
		}

		if c == '\\' {
			escaped = true
			continue
		}

		if c == '"' {
			inQuote = !inQuote
			continue
		}

		if c == ' ' && !inQuote {
			if builder.Len() > 0 {
				res = append(res, builder.String())
				builder.Reset()
			}
			continue
		}

		builder.WriteRune(c)
	}

	if builder.Len() > 0 {
		res = append(res, builder.String())
	}
	return res
}

func JoinArgs(arr []string, sep string) string {
	var builder strings.Builder
	for i, s := range arr {
		if i > 0 {
			builder.WriteString(sep)
		}
		// 如果包含空格或引号，则需要包裹引号并转义内部引号
		if strings.ContainsRune(s, ' ') || strings.ContainsRune(s, '"') {
			builder.WriteByte('"')
			builder.WriteString(strings.ReplaceAll(s, "\"", "\\\""))
			builder.WriteByte('"')
		} else {
			builder.WriteString(s)
		}
	}
	return builder.String()
}

func UniqueAppend(to []string, from ...any) []string {
	exists := make(map[string]struct{}, len(to))
	for _, s := range to {
		exists[s] = struct{}{}
	}
	for _, a := range from {
		s := String(a)
		if _, ok := exists[s]; !ok {
			to = append(to, s)
			exists[s] = struct{}{}
		}
	}
	return to
}

func ArrayToBoolMap[T comparable](arr []T) map[T]bool {
	r := map[T]bool{}
	for _, s := range arr {
		r[s] = true
	}
	return r
}

// 补充缺失的 Key 转换工具
func GetLowerName(s string) string {
	if s == "" {
		return ""
	}
	return strings.ToLower(s[:1]) + s[1:]
}

func GetUpperName(s string) string {
	if s == "" {
		return ""
	}
	return strings.ToUpper(s[:1]) + s[1:]
}

// 指针工具
func Ptr[T any](v T) *T { return &v }

// FixUpperCase (保留以支持历史复杂的 Key 转换需求)
func FixUpperCase(data []byte, excludesKeys []string) {
	// 原有逻辑保持
	n := len(data)
	types, keys, tpos := make([]bool, 0), make([]string, 0), -1
	for i := 0; i < n-1; i++ {
		if tpos+1 >= len(types) {
			types = append(types, false)
			keys = append(keys, "")
		}
		switch data[i] {
		case '{':
			tpos++
			types[tpos] = true
			keys[tpos] = ""
		case '}':
			tpos--
		case '[':
			tpos++
			types[tpos] = false
			keys[tpos] = ""
		case ']':
			tpos--
		case '"':
			keyPos := -1
			if i > 0 && (data[i-1] == '{' || (data[i-1] == ',' && tpos >= 0 && types[tpos])) {
				keyPos = i + 1
			}
			i++
			for ; i < n-1; i++ {
				if data[i] == '\\' {
					i++
					continue
				}
				if data[i] == '"' {
					if keyPos >= 0 && len(excludesKeys) > 0 {
						keys[tpos] = string(data[keyPos:i])
					}
					break
				}
			}
			if keyPos >= 0 && (data[keyPos] >= 'A' && data[keyPos] <= 'Z') {
				excluded := false
				if len(excludesKeys) > 0 {
					checkStr := strings.Join(keys[0:tpos+1], ".")
					for _, ek := range excludesKeys {
						if strings.HasSuffix(ek, ".") {
							excluded = strings.HasPrefix(checkStr, ek)
						} else {
							excluded = checkStr == ek
						}
						if excluded {
							break
						}
					}
				}
				if !excluded {
					hasLower := false
					for _, b := range data[keyPos:i] {
						if b >= 'a' && b <= 'z' {
							hasLower = true
							break
						}
					}
					if hasLower {
						data[keyPos] |= 0x20
					}
				}
			}
		}
	}
}