package sets

import (
	"sort"
	"strconv"
	"strings"
)

// Uint32s uint32 set.
type Uint32s map[uint32]struct{}

// NewUint32s 新建字符串set.
func NewUint32s(elems ...uint32) Uint32s {
	return Uint32s{}.Add(elems...)
}

// Add 增加字符串.
func (set Uint32s) Add(elems ...uint32) Uint32s {
	for _, s := range elems {
		set[s] = None
	}

	return set
}

// AddUint32s 增加字符串set.
func (set Uint32s) AddUint32s(elems ...Uint32s) Uint32s {
	for _, elem := range elems {
		for i := range elem {
			set[i] = None
		}
	}

	return set
}

// Any 包含任意个.
func (set Uint32s) Any(elems ...uint32) bool {
	for _, i := range elems {
		if _, has := set[i]; has {
			return true
		}
	}

	return false
}

// All 包含全部.
func (set Uint32s) All(elems ...uint32) bool {
	for _, i := range elems {
		if _, has := set[i]; !has {
			return false
		}
	}

	return true
}

// Del 删除.
func (set Uint32s) Del(elems ...uint32) Uint32s {
	for _, i := range elems {
		delete(set, i)
	}

	return set
}

// DelUint32s 删除uint32 set.
func (set Uint32s) DelUint32s(elems ...Uint32s) Uint32s {
	for _, elem := range elems {
		for i := range elem {
			delete(set, i)
		}
	}

	return set
}

// Has 包含.
func (set Uint32s) Has(i uint32) bool {
	_, has := set[i]

	return has
}

// Join 集合连接.
func (set Uint32s) Join(sep string) string {
	elems := set.Slice()

	switch len(elems) {
	case 0:
		return ""
	case 1:
		return strconv.FormatUint(uint64(elems[0]), ten)
	}

	var builder strings.Builder

	builder.WriteString(strconv.FormatUint(uint64(elems[0]), ten))

	for _, i := range elems[1:] {
		builder.WriteString(sep)
		builder.WriteString(strconv.FormatUint(uint64(i), ten))
	}

	return builder.String()
}

// Slice 转换切片.
func (set Uint32s) Slice() []uint32 {
	ret := make([]uint32, len(set))
	i := -1

	for s := range set {
		i++

		ret[i] = s
	}

	sort.Slice(ret, func(i, f int) bool { return ret[i] < ret[f] })

	return ret
}

// String 字符串.
func (set Uint32s) String() string {
	return set.Join(", ")
}