This repository has been archived on 2024-05-03. You can view files and clone it, but cannot push or open issues or pull requests.
treefmt/walk/walker.go
2024-05-02 08:58:02 +01:00

55 lines
967 B
Go

package walk
import (
"context"
"fmt"
"io/fs"
)
type Type string
const (
Git Type = "git"
Auto Type = "auto"
Filesystem Type = "filesystem"
)
type File struct {
Path string
RelPath string
Info fs.FileInfo
}
func (f File) String() string {
return f.Path
}
type WalkFunc func(file *File, err error) error
type Walker interface {
Root() string
Walk(ctx context.Context, fn WalkFunc) error
}
func New(walkerType Type, root string, paths []string) (Walker, error) {
switch walkerType {
case Git:
return NewGit(root, paths)
case Auto:
return Detect(root, paths)
case Filesystem:
return NewFilesystem(root, paths)
default:
return nil, fmt.Errorf("unknown walker type: %v", walkerType)
}
}
func Detect(root string, paths []string) (Walker, error) {
// for now, we keep it simple and try git first, filesystem second
w, err := NewGit(root, paths)
if err == nil {
return w, err
}
return NewFilesystem(root, paths)
}