48 lines
814 B
Go
48 lines
814 B
Go
package diyhrt
|
|
|
|
import (
|
|
"slices"
|
|
)
|
|
|
|
type StoreFilter struct{
|
|
Limit int
|
|
|
|
IncludeIds []int
|
|
ExcludeIds []int
|
|
|
|
ShipsTo []string
|
|
}
|
|
|
|
|
|
func (f StoreFilter) Filter (stores []Store) []Store {
|
|
result := make([]Store, 0)
|
|
|
|
if len(f.IncludeIds) > 0 {
|
|
for _, s := range stores {
|
|
if f.Limit > 0 && len(result) >= f.Limit {
|
|
break
|
|
}
|
|
|
|
if slices.Contains(f.IncludeIds, s.Id) {
|
|
result = append(result, s)
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
for _, s := range stores {
|
|
if f.Limit > 0 && len(result) >= f.Limit {
|
|
break
|
|
}
|
|
|
|
|
|
if slices.Contains(f.ExcludeIds, s.Id) || slices.Contains(f.IncludeIds, s.Id) {
|
|
continue
|
|
}
|
|
|
|
result = append(result, s)
|
|
}
|
|
|
|
return result
|
|
}
|