mirror of
https://github.com/numtide/treefmt.git
synced 2024-10-05 21:17:37 +03:00
8b92dca9c6
Signed-off-by: Brian McGee <brian@bmcgee.ie>
49 lines
851 B
Go
49 lines
851 B
Go
package walk
|
|
|
|
import (
|
|
"context"
|
|
"io/fs"
|
|
"path/filepath"
|
|
)
|
|
|
|
type filesystemWalker struct {
|
|
root string
|
|
pathsCh chan string
|
|
}
|
|
|
|
func (f filesystemWalker) Root() string {
|
|
return f.root
|
|
}
|
|
|
|
func (f filesystemWalker) Walk(_ context.Context, fn WalkFunc) error {
|
|
relPathOffset := len(f.root) + 1
|
|
|
|
relPathFn := func(path string) (relPath string) {
|
|
if len(path) >= relPathOffset {
|
|
relPath = path[relPathOffset:]
|
|
}
|
|
return
|
|
}
|
|
|
|
walkFn := func(path string, info fs.FileInfo, err error) error {
|
|
file := File{
|
|
Path: path,
|
|
RelPath: relPathFn(path),
|
|
Info: info,
|
|
}
|
|
return fn(&file, err)
|
|
}
|
|
|
|
for path := range f.pathsCh {
|
|
if err := filepath.Walk(path, walkFn); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func NewFilesystem(root string, paths chan string) (Walker, error) {
|
|
return filesystemWalker{root, paths}, nil
|
|
}
|