trivy/pkg/cloud/config_modify.go

package cloud

import (
	"reflect"
	"strings"

	"golang.org/x/xerrors"
	"gopkg.in/yaml.v3"
)

// Set sets a nested field in the Trivy Cloud config
func Set(attribute string, value any) error {
	config, err := Load()
	if err != nil {
		return xerrors.Errorf("failed to load Trivy Cloud config file: %w", err)
	}

	if err := setNestedField(reflect.ValueOf(config).Elem(), attribute, value); err != nil {
		return xerrors.Errorf("failed to set attribute %q: %w", attribute, err)
	}

	return config.Save()
}

// Unset sets a nested field in the Trivy Cloud config to its default value
func Unset(attribute string) error {
	config, err := Load()
	if err != nil {
		return xerrors.Errorf("failed to load Trivy Cloud config file: %w", err)
	}

	if err := unsetNestedField(reflect.ValueOf(config).Elem(), attribute); err != nil {
		return xerrors.Errorf("failed to unset attribute %q: %w", attribute, err)
	}

	return config.Save()
}

func unsetNestedField(value reflect.Value, attribute string) error {
	field, err := navigateToField(value, attribute)
	if err != nil {
		return err
	}

	defaultField, err := navigateToField(reflect.ValueOf(defaultConfig).Elem(), attribute)
	if err != nil {
		return err
	}

	field.Set(defaultField)
	return nil
}

// Get gets a nested field from the Trivy Cloud config
func Get(attribute string) (any, error) {
	return GetWithDefault[any](attribute, nil)
}

// GetWithDefault gets a nested field from the Trivy Cloud config with a default value
func GetWithDefault[T any](attribute string, defaultValue T) (T, error) {
	config, err := Load()
	if err != nil {
		return defaultValue, xerrors.Errorf("failed to load Trivy Cloud config file: %w", err)
	}

	field, err := navigateToField(reflect.ValueOf(config).Elem(), attribute)
	if err != nil {
		return defaultValue, xerrors.Errorf("failed to get attribute %q: %w", attribute, err)
	}

	return field.Interface().(T), nil
}

func setNestedField(v reflect.Value, path string, value any) error {
	field, err := navigateToField(v, path)
	if err != nil {
		return err
	}

	convertedValue, err := convertToType(value, field.Type())
	if err != nil {
		return xerrors.Errorf("failed to convert value: %w", err)
	}

	field.Set(convertedValue)
	return nil
}

func convertToType(value any, targetType reflect.Type) (reflect.Value, error) {
	val := reflect.ValueOf(value)
	if val.Type().AssignableTo(targetType) {
		return val, nil
	}
	targetPtr := reflect.New(targetType) // *T
	targetInterface := targetPtr.Interface()
	data, err := yaml.Marshal(value)
	if err != nil {
		return reflect.Value{}, xerrors.Errorf("failed to marshal value: %w", err)
	}
	if err := yaml.Unmarshal(data, targetInterface); err != nil {
		return reflect.Value{}, xerrors.Errorf("failed to decode into %v: %w", targetType, err)
	}
	return targetPtr.Elem(), nil
}

func navigateToField(v reflect.Value, path string) (reflect.Value, error) {
	parts := strings.Split(path, ".")
	if len(parts) == 0 {
		return reflect.Value{}, xerrors.New("empty attribute path")
	}

	for i, part := range parts {
		fieldName := yamlTagToFieldName(v, part)
		if fieldName == "" {
			return reflect.Value{}, xerrors.Errorf("field %q not found in config", part)
		}

		field := v.FieldByName(fieldName)
		if !field.IsValid() {
			return reflect.Value{}, xerrors.Errorf("field %q not found", fieldName)
		}
		if !field.CanSet() {
			return reflect.Value{}, xerrors.Errorf("field %q cannot be set", fieldName)
		}

		if i == len(parts)-1 {
			return field, nil
		}

		v = field
	}

	return reflect.Value{}, xerrors.New("unexpected end of path")
}

func yamlTagToFieldName(v reflect.Value, yamlTag string) string {
	t := v.Type()
	for i := 0; i < t.NumField(); i++ {
		field := t.Field(i)
		tag := field.Tag.Get("yaml")
		tagName := strings.Split(tag, ",")[0]
		if tagName == yamlTag {
			return field.Name
		}
	}
	return ""
}