lookup.go

package mods

type (
	lookupID string
	mod      interface {
		ID() ModID
		Kinds() Kinds
		Mod() *Mod
	}
	ModLookup[T mod] interface {
		All() []T
		Clear()
		Get(m T) (found T, ok bool)
		GetByID(modID ModID) (found T, ok bool)
		Has(m T) bool
		Len() int
		Remove(m T)
		RemoveConditionally(f func(t T) bool)
		Set(m T)
	}
	// ModLookupConc is a public for serialization purposes.
	ModLookupConc[T mod] struct {
		Lookup map[lookupID]T `json:"Lookup"`
	}
)

func NewModLookup[T mod]() ModLookup[T] {
	return &ModLookupConc[T]{
		Lookup: make(map[lookupID]T),
	}
}

func (l *ModLookupConc[T]) All() []T {
	s := make([]T, 0, len(l.Lookup))
	for _, m := range l.Lookup {
		s = append(s, m)
	}
	return s
}

func (l *ModLookupConc[T]) Clear() {
	l.Lookup = make(map[lookupID]T)
}

func (l *ModLookupConc[T]) Set(m T) {
	l.Lookup[l.newLookupID(m)] = m
}

func (l *ModLookupConc[T]) Has(m T) bool {
	_, ok := l.Lookup[l.newLookupID(m)]
	return ok
}

func (l *ModLookupConc[T]) Get(m T) (found T, ok bool) {
	found, ok = l.Lookup[l.newLookupID(m)]
	return
}

func (l *ModLookupConc[T]) GetByID(modID ModID) (found T, ok bool) {
	for _, m := range l.Lookup {
		if m.ID() == modID {
			found = m
			ok = true
			break
		}
	}
	return
}

func (l *ModLookupConc[T]) Remove(m T) {
	delete(l.Lookup, l.newLookupID(m))
}
func (l *ModLookupConc[T]) RemoveConditionally(f func(m T) bool) {
	var toRemove []lookupID
	for k, m := range l.Lookup {
		if f(m) {
			toRemove = append(toRemove, k)
		}
	}
	for _, k := range toRemove {
		delete(l.Lookup, k)
	}
}

func (l *ModLookupConc[T]) Len() int {
	return len(l.Lookup)
}

func (l *ModLookupConc[T]) newLookupID(m T) lookupID {
	return lookupID(m.ID())
}