utils.go

package utils

import (
	"gopkg.in/yaml.v3"
)

func Uniq[T comparable](values []T) []T {
	var s = make(map[T]struct{}, len(values))
	var out = make([]T, 0, len(values))
	for _, v := range values {
		if _, ok := s[v]; !ok {
			s[v] = struct{}{}
			out = append(out, v)
		}
	}
	return out
}

type Set[T comparable] map[T]struct{}

func NewSet[T comparable](values ...T) Set[T] {
	var s = make(Set[T], len(values))
	for _, v := range values {
		s[v] = struct{}{}
	}
	return s
}

func (s Set[T]) Has(values ...T) bool {
	for _, v := range values {
		if _, ok := s[v]; !ok {
			return false
		}
	}
	return true
}

func (s Set[T]) Add(values ...T) {
	for _, v := range values {
		s[v] = struct{}{}
	}
}

func (s *Set[T]) UnmarshalYAML(value *yaml.Node) error {
	var data []T
	if err := value.Decode(&data); err != nil {
		return err
	}
	*s = NewSet(data...)
	return nil
}