reflux/rfx.go

package reflux

import (
	"fmt"
	"reflect"
	"strconv"
	"strings"

	"github.com/spf13/cast"
)

// rfx 是 R 接口的具体实现
// 通过封装 reflect.Value 提供了对结构体、切片、映射等类型的统一访问方式
type rfx struct {
	// value 存储底层的反射值
	// 该值应该是指针类型,以支持修改操作
	value reflect.Value
}

// Get 通过路径获取嵌套字段的值
func (r *rfx) Get(p ...string) R {
	return &rfx{value: r.getValueByPath(p...)}
}

// Scope 创建一个指定路径的作用域视图(深度克隆)
func (r *rfx) Scope(p ...string) R {
	v := r.getValueByPath(p...)
	if !v.IsValid() {
		return &rfx{value: reflect.Value{}}
	}

	// 深度克隆值
	cloned := DeepClone(v)
	return &rfx{value: cloned}
}

// getValueByPath 通过路径获取值的辅助方法
// 支持两种路径格式:
// 1. 多个参数: Get("Address", "City")
// 2. 点号分割: Get("Address.City") 或混合使用 Get("Address.City", "ZipCode")
func (r *rfx) getValueByPath(p ...string) reflect.Value {
	v := r.value
	for v.Kind() == reflect.Ptr || v.Kind() == reflect.Interface {
		if v.IsNil() {
			return reflect.Value{}
		}
		v = v.Elem()
	}

	// 展开所有路径片段,支持点号分割
	keys := expandPath(p...)

	for _, key := range keys {
		if !v.IsValid() {
			return reflect.Value{}
		}

		switch v.Kind() {
		case reflect.Struct:
			v = tryStructField(v, key)
		case reflect.Map:
			v = v.MapIndex(reflect.ValueOf(key))
		case reflect.Slice, reflect.Array:
			// 尝试将 key 转换为索引
			idx, err := strconv.Atoi(key)
			if err == nil && idx >= 0 && idx < v.Len() {
				v = v.Index(idx)
			} else {
				return reflect.Value{}
			}
		default:
			return reflect.Value{}
		}

		// 解引用指针和接口
		for v.IsValid() && (v.Kind() == reflect.Ptr || v.Kind() == reflect.Interface) {
			if v.IsNil() {
				return reflect.Value{}
			}
			v = v.Elem()
		}
	}

	return v
}

// Set 设置指定路径的值,支持链式调用
// 如果路径不存在或设置失败,会 panic
func (r *rfx) Set(key string, v any) R {
	// 展开路径
	keys := expandPath(key)
	if len(keys) == 0 {
		panic("rfx: empty path")
	}

	// 如果只有一个键,直接设置
	if len(keys) == 1 {
		target := r.getParentValue()
		if !target.IsValid() {
			panic(fmt.Sprintf("rfx: invalid value for path '%s'", key))
		}
		if !target.CanSet() {
			panic(fmt.Sprintf("rfx: cannot set value at path '%s'", key))
		}
		if !r.setFieldValue(target, keys[0], v) {
			panic(fmt.Sprintf("rfx: failed to set value at path '%s'", key))
		}
		return r
	}

	// 多个键的情况,需要特殊处理 map 中的 struct
	if !r.setNestedValue(r.value, keys, v) {
		panic(fmt.Sprintf("rfx: failed to set value at path '%s'", key))
	}
	return r
}

// setNestedValue 递归设置嵌套值,特殊处理 map 中的 struct
func (r *rfx) setNestedValue(current reflect.Value, keys []string, v any) bool {
	// 解引用指针和接口
	for current.Kind() == reflect.Ptr || current.Kind() == reflect.Interface {
		if current.IsNil() {
			return false
		}
		current = current.Elem()
	}

	if !current.IsValid() {
		return false
	}

	// 如果只剩一个键,直接设置
	if len(keys) == 1 {
		if !current.CanSet() {
			return false
		}
		return r.setFieldValue(current, keys[0], v)
	}

	// 多个键的情况
	firstKey := keys[0]
	remainingKeys := keys[1:]

	switch current.Kind() {
	case reflect.Struct:
		field := tryStructField(current, firstKey)
		if !field.IsValid() {
			return false
		}
		return r.setNestedValue(field, remainingKeys, v)

	case reflect.Map:
		// Map 的特殊处理
		mapKey := reflect.ValueOf(firstKey)
		mapValue := current.MapIndex(mapKey)

		if !mapValue.IsValid() {
			return false
		}

		// 创建 map 值的副本以便修改
		valueCopy := reflect.New(mapValue.Type()).Elem()
		valueCopy.Set(mapValue)

		// 在副本上递归设置值
		if !r.setNestedValue(valueCopy, remainingKeys, v) {
			return false
		}

		// 将修改后的值设置回 map
		current.SetMapIndex(mapKey, valueCopy)
		return true

	case reflect.Slice, reflect.Array:
		idx, err := strconv.Atoi(firstKey)
		if err != nil || idx < 0 || idx >= current.Len() {
			return false
		}
		elem := current.Index(idx)
		if !elem.IsValid() {
			return false
		}
		return r.setNestedValue(elem, remainingKeys, v)
	}

	return false
}

// getParentValue 获取父级值的辅助方法
func (r *rfx) getParentValue(p ...string) reflect.Value {
	if len(p) == 0 {
		v := r.value
		for v.Kind() == reflect.Ptr || v.Kind() == reflect.Interface {
			if v.IsNil() {
				return reflect.Value{}
			}
			v = v.Elem()
		}
		return v
	}
	return r.getValueByPath(p...)
}

// setFieldValue 设置字段值的辅助方法
func (r *rfx) setFieldValue(target reflect.Value, key string, v any) bool {
	for target.Kind() == reflect.Ptr || target.Kind() == reflect.Interface {
		if target.IsNil() {
			return false
		}
		target = target.Elem()
	}

	switch target.Kind() {
	case reflect.Struct:
		field := tryStructField(target, key)
		if !field.IsValid() || !field.CanSet() {
			return false
		}
		return r.setValue(field, v)
	case reflect.Map:
		if target.IsNil() {
			target.Set(reflect.MakeMap(target.Type()))
		}
		val := reflect.ValueOf(v)
		if !val.Type().AssignableTo(target.Type().Elem()) {
			// 尝试转换
			if val.Type().ConvertibleTo(target.Type().Elem()) {
				val = val.Convert(target.Type().Elem())
			} else {
				return false
			}
		}
		target.SetMapIndex(reflect.ValueOf(key), val)
		return true
	case reflect.Slice, reflect.Array:
		idx, err := strconv.Atoi(key)
		if err != nil || idx < 0 || idx >= target.Len() {
			return false
		}
		elem := target.Index(idx)
		if !elem.CanSet() {
			return false
		}
		return r.setValue(elem, v)
	}
	return false
}

// setValue 设置值的辅助方法
// 使用 cast 库进行智能类型转换,支持更多的转换场景
func (r *rfx) setValue(field reflect.Value, v any) bool {
	val := reflect.ValueOf(v)

	// 尝试直接赋值(类型完全匹配)
	if val.Type().AssignableTo(field.Type()) {
		field.Set(val)
		return true
	}

	// 处理指针类型
	if field.Type().Kind() == reflect.Ptr {
		// 如果传入的值已经是指针类型,尝试直接赋值
		if val.Type().AssignableTo(field.Type()) {
			field.Set(val)
			return true
		}

		// 如果传入的值不是指针,创建新指针并设置值
		elemType := field.Type().Elem()
		newPtr := reflect.New(elemType)

		// 递归设置指针指向的值
		if !r.setValue(newPtr.Elem(), v) {
			return false
		}

		field.Set(newPtr)
		return true
	}

	// 优先使用 cast 进行智能类型转换
	// 这样可以处理 string <-> number, number <-> bool 等常见转换
	targetType := field.Type()
	var converted any
	var err error

	switch targetType.Kind() {
	case reflect.Bool:
		converted, err = cast.ToBoolE(v)
	case reflect.Int:
		converted, err = cast.ToIntE(v)
	case reflect.Int8:
		converted, err = cast.ToInt8E(v)
	case reflect.Int16:
		converted, err = cast.ToInt16E(v)
	case reflect.Int32:
		converted, err = cast.ToInt32E(v)
	case reflect.Int64:
		converted, err = cast.ToInt64E(v)
	case reflect.Uint:
		converted, err = cast.ToUintE(v)
	case reflect.Uint8:
		converted, err = cast.ToUint8E(v)
	case reflect.Uint16:
		converted, err = cast.ToUint16E(v)
	case reflect.Uint32:
		converted, err = cast.ToUint32E(v)
	case reflect.Uint64:
		converted, err = cast.ToUint64E(v)
	case reflect.Float32:
		converted, err = cast.ToFloat32E(v)
	case reflect.Float64:
		converted, err = cast.ToFloat64E(v)
	case reflect.String:
		converted, err = cast.ToStringE(v)
	default:
		// 对于 cast 不支持的类型,尝试标准的反射类型转换
		if val.Type().ConvertibleTo(field.Type()) {
			field.Set(val.Convert(field.Type()))
			return true
		}
		return false
	}

	if err != nil {
		// 如果 cast 失败,尝试标准的反射类型转换作为后备
		if val.Type().ConvertibleTo(field.Type()) {
			field.Set(val.Convert(field.Type()))
			return true
		}
		return false
	}

	field.Set(reflect.ValueOf(converted))
	return true
}

// Delete 删除指定路径的值,支持链式调用
// 如果删除失败会 panic
func (r *rfx) Delete(p ...string) R {
	if len(p) == 0 {
		panic("rfx: empty path")
	}

	// 展开路径
	keys := expandPath(p...)
	if len(keys) == 0 {
		panic("rfx: empty path")
	}

	target := r.getParentValue(keys[:len(keys)-1]...)
	if !target.IsValid() {
		panic(fmt.Sprintf("rfx: invalid path '%s'", strings.Join(keys[:len(keys)-1], ".")))
	}

	lastKey := keys[len(keys)-1]
	for target.Kind() == reflect.Ptr || target.Kind() == reflect.Interface {
		if target.IsNil() {
			panic(fmt.Sprintf("rfx: nil value at path '%s'", strings.Join(keys[:len(keys)-1], ".")))
		}
		target = target.Elem()
	}

	switch target.Kind() {
	case reflect.Map:
		target.SetMapIndex(reflect.ValueOf(lastKey), reflect.Value{})
		return r
	case reflect.Slice:
		idx, err := strconv.Atoi(lastKey)
		if err != nil {
			panic(fmt.Sprintf("rfx: invalid slice index '%s'", lastKey))
		}
		if idx < 0 || idx >= target.Len() {
			panic(fmt.Sprintf("rfx: slice index out of range: %d", idx))
		}
		// 删除切片元素
		newSlice := reflect.AppendSlice(
			target.Slice(0, idx),
			target.Slice(idx+1, target.Len()),
		)
		target.Set(newSlice)
		return r
	default:
		panic(fmt.Sprintf("rfx: cannot delete from type %s", target.Kind()))
	}
}

// Exists 检查指定路径的值是否存在
func (r *rfx) Exists(p ...string) bool {
	v := r.getValueByPath(p...)
	return v.IsValid()
}

// Array 将当前值转换为 R 切片
func (r *rfx) Array() []R {
	v := r.value
	for v.Kind() == reflect.Ptr || v.Kind() == reflect.Interface {
		if v.IsNil() {
			return nil
		}
		v = v.Elem()
	}

	if v.Kind() != reflect.Slice && v.Kind() != reflect.Array {
		return nil
	}

	result := make([]R, v.Len())
	for i := 0; i < v.Len(); i++ {
		result[i] = &rfx{value: v.Index(i)}
	}
	return result
}

// keys 返回当前映射或结构体的所有键名
func (r *rfx) Keys() []string {
	v := r.value
	for v.Kind() == reflect.Ptr || v.Kind() == reflect.Interface {
		if v.IsNil() {
			return nil
		}
		v = v.Elem()
	}

	switch v.Kind() {
	case reflect.Map:
		keys := v.MapKeys()
		result := make([]string, len(keys))
		for i, k := range keys {
			result[i] = fmt.Sprint(k.Interface())
		}
		return result
	case reflect.Struct:
		t := v.Type()
		result := make([]string, t.NumField())
		for i := 0; i < t.NumField(); i++ {
			result[i] = t.Field(i).Name
		}
		return result
	}
	return nil
}

// Value 返回底层的 reflect.Value
func (r *rfx) Value() reflect.Value {
	return r.value
}

// Ptr 返回指向当前值的指针
func (r *rfx) Ptr() any {
	v := r.value
	if v.Kind() == reflect.Ptr {
		return v.Interface()
	}
	if v.CanAddr() {
		return v.Addr().Interface()
	}
	return nil
}

// Any 将当前值转换为 any 类型
func (r *rfx) Any() any {
	v := r.value
	for v.Kind() == reflect.Ptr || v.Kind() == reflect.Interface {
		if v.IsNil() {
			return nil
		}
		v = v.Elem()
	}
	if !v.IsValid() {
		return nil
	}
	return v.Interface()
}

// Bool 将当前值转换为 bool 类型
func (r *rfx) Bool() bool {
	result, err := cast.ToBoolE(r.Any())
	if err != nil {
		panic(fmt.Sprintf("rfx: failed to convert to bool: %v", err))
	}
	return result
}

// Float64 将当前值转换为 float64 类型
func (r *rfx) Float64() float64 {
	result, err := cast.ToFloat64E(r.Any())
	if err != nil {
		panic(fmt.Sprintf("rfx: failed to convert to float64: %v", err))
	}
	return result
}

// Float32 将当前值转换为 float32 类型
func (r *rfx) Float32() float32 {
	result, err := cast.ToFloat32E(r.Any())
	if err != nil {
		panic(fmt.Sprintf("rfx: failed to convert to float32: %v", err))
	}
	return result
}

// Int64 将当前值转换为 int64 类型
func (r *rfx) Int64() int64 {
	result, err := cast.ToInt64E(r.Any())
	if err != nil {
		panic(fmt.Sprintf("rfx: failed to convert to int64: %v", err))
	}
	return result
}

// Int32 将当前值转换为 int32 类型
func (r *rfx) Int32() int32 {
	result, err := cast.ToInt32E(r.Any())
	if err != nil {
		panic(fmt.Sprintf("rfx: failed to convert to int32: %v", err))
	}
	return result
}

// Int16 将当前值转换为 int16 类型
func (r *rfx) Int16() int16 {
	result, err := cast.ToInt16E(r.Any())
	if err != nil {
		panic(fmt.Sprintf("rfx: failed to convert to int16: %v", err))
	}
	return result
}

// Int8 将当前值转换为 int8 类型
func (r *rfx) Int8() int8 {
	result, err := cast.ToInt8E(r.Any())
	if err != nil {
		panic(fmt.Sprintf("rfx: failed to convert to int8: %v", err))
	}
	return result
}

// Int 将当前值转换为 int 类型
func (r *rfx) Int() int {
	result, err := cast.ToIntE(r.Any())
	if err != nil {
		panic(fmt.Sprintf("rfx: failed to convert to int: %v", err))
	}
	return result
}

// Uint 将当前值转换为 uint 类型
func (r *rfx) Uint() uint {
	result, err := cast.ToUintE(r.Any())
	if err != nil {
		panic(fmt.Sprintf("rfx: failed to convert to uint: %v", err))
	}
	return result
}

// Uint64 将当前值转换为 uint64 类型
func (r *rfx) Uint64() uint64 {
	result, err := cast.ToUint64E(r.Any())
	if err != nil {
		panic(fmt.Sprintf("rfx: failed to convert to uint64: %v", err))
	}
	return result
}

// Uint32 将当前值转换为 uint32 类型
func (r *rfx) Uint32() uint32 {
	result, err := cast.ToUint32E(r.Any())
	if err != nil {
		panic(fmt.Sprintf("rfx: failed to convert to uint32: %v", err))
	}
	return result
}

// Uint16 将当前值转换为 uint16 类型
func (r *rfx) Uint16() uint16 {
	result, err := cast.ToUint16E(r.Any())
	if err != nil {
		panic(fmt.Sprintf("rfx: failed to convert to uint16: %v", err))
	}
	return result
}

// Uint8 将当前值转换为 uint8 类型
func (r *rfx) Uint8() uint8 {
	result, err := cast.ToUint8E(r.Any())
	if err != nil {
		panic(fmt.Sprintf("rfx: failed to convert to uint8: %v", err))
	}
	return result
}

// String 将当前值转换为 string 类型
func (r *rfx) String() string {
	result, err := cast.ToStringE(r.Any())
	if err != nil {
		panic(fmt.Sprintf("rfx: failed to convert to string: %v", err))
	}
	return result
}

// StringMapString 将当前值转换为 map[string]string 类型
func (r *rfx) StringMapString() map[string]string {
	result, err := cast.ToStringMapStringE(r.Any())
	if err != nil {
		panic(fmt.Sprintf("rfx: failed to convert to map[string]string: %v", err))
	}
	return result
}

// StringMapStringSlice 将当前值转换为 map[string][]string 类型
func (r *rfx) StringMapStringSlice() map[string][]string {
	result, err := cast.ToStringMapStringSliceE(r.Any())
	if err != nil {
		panic(fmt.Sprintf("rfx: failed to convert to map[string][]string: %v", err))
	}
	return result
}

// StringMapBool 将当前值转换为 map[string]bool 类型
func (r *rfx) StringMapBool() map[string]bool {
	result, err := cast.ToStringMapBoolE(r.Any())
	if err != nil {
		panic(fmt.Sprintf("rfx: failed to convert to map[string]bool: %v", err))
	}
	return result
}

// StringMapInt 将当前值转换为 map[string]int 类型
func (r *rfx) StringMapInt() map[string]int {
	result, err := cast.ToStringMapIntE(r.Any())
	if err != nil {
		panic(fmt.Sprintf("rfx: failed to convert to map[string]int: %v", err))
	}
	return result
}

// StringMapInt64 将当前值转换为 map[string]int64 类型
func (r *rfx) StringMapInt64() map[string]int64 {
	result, err := cast.ToStringMapInt64E(r.Any())
	if err != nil {
		panic(fmt.Sprintf("rfx: failed to convert to map[string]int64: %v", err))
	}
	return result
}

// StringMap 将当前值转换为 map[string]any 类型
func (r *rfx) StringMap() map[string]any {
	result, err := cast.ToStringMapE(r.Any())
	if err != nil {
		panic(fmt.Sprintf("rfx: failed to convert to map[string]any: %v", err))
	}
	return result
}

// Slice 将当前值转换为 []any 切片
func (r *rfx) Slice() []any {
	v := r.value
	for v.Kind() == reflect.Ptr || v.Kind() == reflect.Interface {
		if v.IsNil() {
			return nil
		}
		v = v.Elem()
	}
	if !v.IsValid() || (v.Kind() != reflect.Slice && v.Kind() != reflect.Array) {
		return nil
	}

	result := make([]any, v.Len())
	for i := 0; i < v.Len(); i++ {
		elem := v.Index(i)
		for elem.Kind() == reflect.Ptr || elem.Kind() == reflect.Interface {
			if elem.IsNil() {
				break
			}
			elem = elem.Elem()
		}
		if elem.IsValid() {
			result[i] = elem.Interface()
		}
	}
	return result
}

// BoolSlice 将当前值转换为 []bool 切片
func (r *rfx) BoolSlice() []bool {
	result, err := cast.ToBoolSliceE(r.Any())
	if err != nil {
		panic(fmt.Sprintf("rfx: failed to convert to []bool: %v", err))
	}
	return result
}

// StringSlice 将当前值转换为 []string 切片
func (r *rfx) StringSlice() []string {
	result, err := cast.ToStringSliceE(r.Any())
	if err != nil {
		panic(fmt.Sprintf("rfx: failed to convert to []string: %v", err))
	}
	return result
}

// IntSlice 将当前值转换为 []int 切片
func (r *rfx) IntSlice() []int {
	result, err := cast.ToIntSliceE(r.Any())
	if err != nil {
		panic(fmt.Sprintf("rfx: failed to convert to []int: %v", err))
	}
	return result
}