package slice

func ContainsAny[C comparable](haystack []C, needles ...C) bool {
	for _, h := range haystack {
		for _, n := range needles {
			if h == n {
				return true
			}
		}
	}

	return false
}

func ContainsAll[C comparable](haystack []C, needles ...C) bool {
	var matches = 0
	for _, h := range haystack {
		for _, n := range needles {
			if h == n {
				matches++
			}
		}
	}

	return len(needles) == matches
}

func IntersectStrings(a []string, b []string) []string {
	var (
		out = make([]string, 0, len(a)+len(b))
		ah  = ToStringBoolMap(a)
	)

	for i := 0; i < len(b); i++ {
		if ah[b[i]] {
			out = append(out, b[i])
		}
	}

	return out
}

func ToStringBoolMap(s []string) (h map[string]bool) {
	h = make(map[string]bool)
	for i := 0; i < len(s); i++ {
		h[s[i]] = true
	}

	return
}

func ToUint64BoolMap(u []uint64) (h map[uint64]bool) {
	h = make(map[uint64]bool)
	for i := 0; i < len(u); i++ {
		h[u[i]] = true
	}

	return
}

// @todo replace with ContainsAny
func HasString(ss []string, s string) bool {
	for i := range ss {
		if ss[i] == s {
			return true
		}
	}

	return false
}

// RemoveString removes one or more strings form the input slice
func PluckString(ss []string, ff ...string) (o []string) {
	if len(ff) == 0 {
		return ss
	}

	f := ToStringBoolMap(ff)
	o = make([]string, 0, len(ss))

	for _, s := range ss {
		if !f[s] {
			// remove from the list
			o = append(o, s)
		}
	}

	return o
}