EdgeNode/internal/utils/get.go

package utils

import (
	"reflect"
	"regexp"

	"github.com/iwind/TeaGo/types"
)

var RegexpDigitNumber = regexp.MustCompile(`^\d+$`)

func Get(object interface{}, keys []string) interface{} {
	if len(keys) == 0 {
		return object
	}

	if object == nil {
		return nil
	}

	firstKey := keys[0]
	keys = keys[1:]

	value := reflect.ValueOf(object)

	if !value.IsValid() {
		return nil
	}

	if value.Kind() == reflect.Ptr {
		value = value.Elem()
	}

	if value.Kind() == reflect.Struct {
		field := value.FieldByName(firstKey)
		if !field.IsValid() {
			return nil
		}

		if len(keys) == 0 {
			return field.Interface()
		}

		return Get(field.Interface(), keys)
	}

	if value.Kind() == reflect.Map {
		mapKey := reflect.ValueOf(firstKey)
		mapValue := value.MapIndex(mapKey)
		if !mapValue.IsValid() {
			return nil
		}

		if len(keys) == 0 {
			return mapValue.Interface()
		}

		return Get(mapValue.Interface(), keys)
	}

	if value.Kind() == reflect.Slice {
		if RegexpDigitNumber.MatchString(firstKey) {
			firstKeyInt := types.Int(firstKey)
			if value.Len() > firstKeyInt {
				result := value.Index(firstKeyInt).Interface()
				if len(keys) == 0 {
					return result
				}

				return Get(result, keys)
			}
		}

		return nil
	}

	return nil
}