package extypes

import (
	"fmt"
	"reflect"
	"strings"
)

type Set[T any] []T

func NewSetFrom[T any](slice []T) Set[T] {
	s := make(Set[T], 0)
	for _, v := range slice {
		if s.Index(v) >= 0 {
			continue
		}
		s = append(s, v)
	}
	return s
}
func (s Set[T]) Len() int        { return len(s) }
func (s Set[T]) Cap() int        { return cap(s) }
func (s Set[T]) Get(index int) T { return s[index] }
func (s Set[T]) Last() T         { return s.Get(s.Len() - 1) }
func (s Set[T]) First() T        { return s.Get(0) }

func (s Set[T]) Index(el T) int {
	for i := range s {
		if reflect.DeepEqual(s[i], el) {
			return i
		}
	}
	return -1
}
func (s Set[T]) Swap(i, j int) {
	s[i], s[j] = s[j], s[i]
}
func (s Set[T]) Add(v T) Set[T] {
	index := s.Index(v)
	if index >= 0 {
		return s
	}
	return append(s, v)
}
func (s Set[T]) Pop(index int) Set[T] {
	if index == 0 {
		return s[1:]
	}
	out := make(Set[T], s.Len()-1)
	copy(out, s[:index])
	copy(out[index:], s[index+1:])
	return out
}
func (s Set[T]) Remove(el T) Set[T] {
	index := s.Index(el)
	if index == -1 {
		return s
	}
	return s.Pop(index)
}
func (s Set[T]) Equal(s2 Set[T]) bool {
	if s.Len() != s2.Len() {
		return false
	}
	if s.Cap() != s2.Cap() {
		return false
	}
	for i := range s {
		if !reflect.DeepEqual(s[i], s2[i]) {
			return false
		}
	}
	return true
}

func (s Set[T]) ToSlice() Slice[T] { return NewSliceFrom(s) }
func (s Set[T]) ToTuple() Tuple[T] { return NewTupleFrom(s) }
func (s Set[T]) ToArray() []T {
	out := make([]T, len(s))
	copy(out, s)
	return out
}
func (s Set[T]) ToAnyArray() []any {
	out := make([]any, len(s))
	for i, v := range s {
		out[i] = v
	}
	return out
}
func (s Set[T]) Filter(f func(e T) bool) Set[T] {
	out := make(Set[T], 0)
	for _, v := range s {
		if f(v) {
			out = append(out, v)
		}
	}
	return out
}
func (s Set[T]) ForEach(f func(int, T)) Set[T] {
	for i, v := range s {
		f(i, v)
	}
	return s
}

// Map is mutable func
func (s Set[T]) Map(f func(e T) T) Set[T] {
	for i, v := range s {
		s[i] = f(v)
	}
	return s
}

func (s Set[T]) Join(sep string) string {
	st := make([]string, len(s))
	for i, v := range s {
		st[i] = fmt.Sprintf("%v", v)
	}
	return strings.Join(st, sep)
}