sqan/scanner.go

package sqan

import (
	"iter"
	"os"
	"path"
	"slices"
	"strings"

	"github.com/gobwas/glob"
)

type Scanner struct {
	rootDir        string
	extensions     []string
	ignorePatterns []string

	compiledPatterns []*glob.Pattern

	onError func(error)
}

func (s *Scanner) OnError(onError func(error)) *Scanner {
	s.onError = onError
	return s
}

func (s *Scanner) Scan() iter.Seq[*File] {
	if err := s.compile(); err != nil {
		s.onError(err)
	}

	return func(yield func(*File) bool) {
		s.scan(s.rootDir, s.rootDir, yield)
	}
}

func (s *Scanner) compile() error {
	compiledPatterns := []*glob.Pattern{}
	for _, text := range s.ignorePatterns {
		pattern, err := glob.Compile(text)
		if err != nil {
			return err
		}

		compiledPatterns = append(compiledPatterns, pattern)
	}

	s.compiledPatterns = compiledPatterns
	return nil
}

func (s *Scanner) scan(rootDir, dir string, yield func(*File) bool) {
	entries, err := os.ReadDir(dir)
	if err != nil {
		s.onError(err)
		return
	}

	for _, entry := range entries {
		absPath := path.Join(dir, entry.Name())
		ignore := false
		for _, pattern := range s.compiledPatterns {
			if pattern.Match(absPath) {
				ignore = true
				break
			}
		}
		if ignore {
			continue
		}

		if entry.IsDir() {
			s.scan(rootDir, absPath, yield)
			continue
		}

		f := NewFile(rootDir, absPath)
		if len(s.extensions) > 0 && !slices.Contains(s.extensions, f.Extension) {
			continue
		}

		info, err := entry.Info()
		if err != nil {
			s.onError(err)
			continue
		}

		f.Size = info.Size()
		f.Modified = info.ModTime()
		if !yield(f) {
			return
		}
	}
}

func (s *Scanner) WithExtensions(exts ...string) *Scanner {
	lower := []string{}
	for _, ext := range exts {
		lower = append(lower, strings.ToLower(ext))
	}

	s.extensions = append(s.extensions, lower...)

	return s
}

func (s *Scanner) WithIgnorePatterns(patterns ...string) *Scanner {
	s.ignorePatterns = append(s.ignorePatterns, patterns...)

	return s
}

func NewScanner(rootDir string) *Scanner {
	if rootDir[0] != '/' {
		cwd, _ := os.Getwd()
		rootDir = path.Join(cwd, rootDir)
	}

	s := &Scanner{
		rootDir: rootDir,

		onError: func(err error) {},
	}

	return s
}