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
}