89 lines
1.7 KiB
Go
89 lines
1.7 KiB
Go
package convids
|
|
|
|
import (
|
|
"errors"
|
|
"iter"
|
|
"regexp"
|
|
"slices"
|
|
"strings"
|
|
)
|
|
|
|
var ErrNoName = errors.New("missing name")
|
|
var ErrNilPointer = errors.New("nil pointer")
|
|
|
|
var animePattern = `^\[.+\] `
|
|
var animeRegexp = regexp.MustCompile(animePattern)
|
|
|
|
func (s *Show) Match(path string) (bool, error) {
|
|
if s == nil {
|
|
return false, ErrNilPointer
|
|
}
|
|
if s.Pattern != "" {
|
|
return s.matchRegexp(path)
|
|
}
|
|
if s.Name == "" {
|
|
return false, ErrNoName
|
|
}
|
|
if s.Anime {
|
|
if !animeRegexp.MatchString(path) {
|
|
return false, nil
|
|
}
|
|
path = animeRegexp.ReplaceAllString(path, "")
|
|
}
|
|
return strings.HasPrefix(path, s.Name), nil
|
|
}
|
|
|
|
func (s *Show) matchRegexp(path string) (f bool, err error) {
|
|
if s.re == nil {
|
|
p := s.Pattern
|
|
if s.Anime {
|
|
p = animePattern + strings.TrimPrefix(p, "^")
|
|
} else if !strings.HasPrefix(p, "^") {
|
|
p = "^" + p
|
|
}
|
|
s.re, err = regexp.Compile(p)
|
|
if err != nil {
|
|
return
|
|
}
|
|
}
|
|
return s.re.MatchString(path), nil
|
|
}
|
|
|
|
func (ss *Shows) All() iter.Seq[*Show] {
|
|
if ss == nil {
|
|
return func(func(*Show) bool) {}
|
|
}
|
|
return slices.Values(*ss)
|
|
}
|
|
|
|
func (d *Data) AllShows(outputGroup GroupPrinter) iter.Seq[*Show] {
|
|
return func(yield func(*Show) bool) {
|
|
if d == nil || d.Config == nil {
|
|
return
|
|
}
|
|
for _, groupName := range d.Config.Groups {
|
|
if !d.yieldGroup(yield, outputGroup, groupName) {
|
|
return
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func (d *Data) yieldGroup(yield func(*Show) bool, outputGroup GroupPrinter, groupName string) bool {
|
|
group := (*d.Shows)[groupName]
|
|
if group == nil || len(*group) < 1 {
|
|
return true
|
|
}
|
|
if outputGroup != nil {
|
|
outputGroup(groupName, *group)
|
|
}
|
|
for sh := range group.All() {
|
|
if sh == nil {
|
|
continue
|
|
}
|
|
if !yield(sh) {
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
}
|