package sets

import (
	"sort"
	"strings"
)

// Strings 字符串set.
type Strings map[string]struct{}

// NewStrings 新建字符串set.
func NewStrings(elems ...string) Strings {
	return Strings{}.Add(elems...)
}

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

	return set
}

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

	return set
}

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

	return false
}

// AnyStrings 包含任意.
func (set Strings) AnyStrings(elems ...Strings) bool {
	for _, elem := range elems {
		for s := range elem {
			if set.Has(s) {
				return true
			}
		}
	}

	return false
}

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

	return true
}

// AllStrings 全部包含.
func (set Strings) AllStrings(elems ...Strings) bool {
	for _, elem := range elems {
		for s := range elem {
			if !set.Has(s) {
				return false
			}
		}
	}

	return true
}

// Del 删除.
func (set Strings) Del(elems ...string) Strings {
	for _, s := range elems {
		delete(set, s)
	}

	return set
}

// DelStrings 删除字符串set.
func (set Strings) DelStrings(elems ...Strings) Strings {
	for _, elem := range elems {
		for str := range elem {
			delete(set, str)
		}
	}

	return set
}

// Has 包含.
func (set Strings) Has(s string) bool {
	_, has := set[s]

	return has
}

// Join 集合连接.
func (set Strings) Join(sep string) string {
	return strings.Join(set.Slice(), sep)
}

// Slice 转换切片.
func (set Strings) Slice() []string {
	strs := make(sort.StringSlice, len(set))
	i := -1

	for s := range set {
		i++

		strs[i] = s
	}

	sort.Sort(strs)

	return strs
}

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