GoDoxy/internal/docker/label.go

package docker

import (
	"reflect"
	"strings"

	E "github.com/yusing/go-proxy/internal/error"
	U "github.com/yusing/go-proxy/internal/utils"
	F "github.com/yusing/go-proxy/internal/utils/functional"
)

/*
Formats:
  - namespace.attribute
  - namespace.target.attribute
  - namespace.target.attribute.namespace2.attribute
*/
type (
	Label struct {
		Namespace string
		Target    string
		Attribute string
		Value     any
	}
	NestedLabelMap map[string]U.SerializedObject
	ValueParser    func(string) (any, E.NestedError)
	ValueParserMap map[string]ValueParser
)

func (l *Label) String() string {
	if l.Attribute == "" {
		return l.Namespace + "." + l.Target
	}
	return l.Namespace + "." + l.Target + "." + l.Attribute
}

// Apply applies the value of a Label to the corresponding field in the given object.
//
// Parameters:
//   - obj: a pointer to the object to which the Label value will be applied.
//   - l: a pointer to the Label containing the attribute and value to be applied.
//
// Returns:
//   - error: an error if the field does not exist.
func ApplyLabel[T any](obj *T, l *Label) E.NestedError {
	if obj == nil {
		return E.Invalid("nil object", l)
	}
	switch nestedLabel := l.Value.(type) {
	case *Label:
		var field reflect.Value
		objType := reflect.TypeFor[T]()
		for i := 0; i < reflect.TypeFor[T]().NumField(); i++ {
			if objType.Field(i).Tag.Get("yaml") == l.Attribute {
				field = reflect.ValueOf(obj).Elem().Field(i)
				break
			}
		}
		if !field.IsValid() {
			return E.NotExist("field", l.Attribute)
		}
		dst, ok := field.Interface().(NestedLabelMap)
		if !ok {
			return E.Invalid("type", field.Type())
		}
		if dst == nil {
			field.Set(reflect.MakeMap(reflect.TypeFor[NestedLabelMap]()))
			dst = field.Interface().(NestedLabelMap)
		}
		if dst[nestedLabel.Namespace] == nil {
			dst[nestedLabel.Namespace] = make(U.SerializedObject)
		}
		dst[nestedLabel.Namespace][nestedLabel.Attribute] = nestedLabel.Value
		return nil
	default:
		return U.Deserialize(U.SerializedObject{l.Attribute: l.Value}, obj)
	}
}

func ParseLabel(label string, value string) (*Label, E.NestedError) {
	parts := strings.Split(label, ".")

	if len(parts) < 2 {
		return &Label{
			Namespace: label,
			Value:     value,
		}, nil
	}

	l := &Label{
		Namespace: parts[0],
		Target:    parts[1],
		Value:     value,
	}

	switch len(parts) {
	case 2:
		l.Attribute = l.Target
	case 3:
		l.Attribute = parts[2]
	default:
		l.Attribute = parts[2]
		nestedLabel, err := ParseLabel(strings.Join(parts[3:], "."), value)
		if err.HasError() {
			return nil, err
		}
		l.Value = nestedLabel
	}

	// find if namespace has value parser
	pm, ok := valueParserMap.Load(U.ToLowerNoSnake(l.Namespace))
	if !ok {
		return l, nil
	}
	// find if attribute has value parser
	p, ok := pm[U.ToLowerNoSnake(l.Attribute)]
	if !ok {
		return l, nil
	}
	// try to parse value
	v, err := p(value)
	if err.HasError() {
		return nil, err.Subject(label)
	}
	l.Value = v
	return l, nil
}

func RegisterNamespace(namespace string, pm ValueParserMap) {
	pmCleaned := make(ValueParserMap, len(pm))
	for k, v := range pm {
		pmCleaned[U.ToLowerNoSnake(k)] = v
	}
	valueParserMap.Store(U.ToLowerNoSnake(namespace), pmCleaned)
}

func GetRegisteredNamespaces() map[string][]string {
	r := make(map[string][]string)

	valueParserMap.RangeAll(func(ns string, vpm ValueParserMap) {
		r[ns] = make([]string, 0, len(vpm))
		for attr := range vpm {
			r[ns] = append(r[ns], attr)
		}
	})

	return r
}

// namespace:target.attribute -> func(string) (any, error)
var valueParserMap = F.NewMapOf[string, ValueParserMap]()