🚧 WalkFiles

convids/logic.go

@@ -0,0 +1,63 @@
+package convids
+
+import (
+	"os"
+	fp "path/filepath"
+	"regexp"
+	"strings"
+
+	"gopkg.in/yaml.v3"
+)
+
+func NewData(path string) (*Data, error) {
+	f, err := os.Open(path)
+	if err != nil {
+		return nil, err
+	}
+	ydec := yaml.NewDecoder(f)
+	var data Data
+	err = ydec.Decode(&data)
+	return &data, err
+}
+
+func ensureExtRe(c *Config) (err error) {
+	if c.extRe != nil {
+		return nil
+	}
+	c.extRe, err = regexp.Compile("(" + strings.Join(c.Extensions, "|") + ")$")
+	return
+}
+
+func WalkFiles(d *Data, stopOnError, silent bool, cb func(s *Show, path string) error) (err error) {
+	err = ensureExtRe(d.Config)
+	if err != nil {
+		return
+	}
+	var files []os.DirEntry
+	files, err = os.ReadDir(d.Config.Source)
+	if err != nil {
+		return
+	}
+	for s := range d.AllShows(silent) {
+		var found bool
+		for _, file := range files {
+			if file.IsDir() {
+				continue
+			}
+			if !d.Config.extRe.MatchString(file.Name()) {
+				continue
+			}
+			found, err = s.Match(file.Name())
+			if err != nil {
+				return
+			}
+			if found {
+				err = cb(s, fp.Join(d.Config.Source, file.Name()))
+				if err != nil && stopOnError {
+					return
+				}
+			}
+		}
+	}
+	return nil
+}

convids/methods.go

@@ -1,22 +1,61 @@
 package convids
 
 import (
+	"errors"
 	"fmt"
 	"iter"
+	"regexp"
 	"slices"
+	"strings"
 )
 
-func (s Shows) All() iter.Seq[Show] {
-	return slices.Values(s)
+var ErrNoName = errors.New("missing name")
+
+func (s *Show) Match(path string) (bool, error) {
+	if s.Pattern != "" {
+		return s.matchRegexp(path)
+	}
+	if s.Name == "" {
+		return false, ErrNoName
+	}
+	return strings.HasPrefix(path, s.Name), nil
 }
 
-func (d Data) AllShows(silent bool) iter.Seq[Show] {
-	return func(yield func(Show) bool) {
+func (s *Show) matchRegexp(path string) (f bool, err error) {
+	if s.re == nil {
+		p := s.Pattern
+		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] {
+	return slices.Values(*ss)
+}
+
+func (d *Data) AllShows(silent bool) iter.Seq[*Show] {
+	return func(yield func(*Show) bool) {
+		if d.Config == nil {
+			return
+		}
 		for _, show := range d.Config.Shows {
+			sh := (*d.Shows)[show]
+			if sh == nil || len(*sh) < 1 {
+				continue
+			}
 			if !silent {
 				fmt.Println("\nChecking", show, "shows\n")
 			}
-			for s := range d.Shows[show].All() {
+			for s := range sh.All() {
+				if s == nil {
+					continue
+				}
 				if !yield(s) {
 					return
 				}

convids/models.go

@@ -1,23 +1,29 @@
 package convids
 
+import "regexp"
+
 type Data struct {
-	Config Config
-	Shows  ShowMap
+	Config *Config
+	Shows  *ShowMap
 }
 
 type Config struct {
 	Source     string
 	Extensions []string
 	Shows      []string
+
+	extRe *regexp.Regexp
 }
 
-type ShowMap map[string]Shows
+type ShowMap map[string]*Shows
 
-type Shows []Show
+type Shows []*Show
 
 type Show struct {
 	Folder  string
 	Pattern string
 	Name    string
 	Anime   bool
+
+	re *regexp.Regexp
 }