-
Notifications
You must be signed in to change notification settings - Fork 0
/
walker.go
73 lines (61 loc) · 1.43 KB
/
walker.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
package main
import (
"github.com/schollz/progressbar/v3"
"io/ioutil"
"path/filepath"
"regexp"
)
func walkerProgressBar(ch <-chan CounterType) {
fileBar := progressbar.Default(-1)
dirBar := progressbar.Default(-1)
for cType := range ch {
switch cType {
case File:
fileBar.Add(1)
case Dir:
dirBar.Add(1)
}
}
dirBar.Finish()
fileBar.Finish()
}
func Walker(dir string, fileTypesRegex *regexp.Regexp, recursive bool) ([]string, error) {
ch := make(chan CounterType)
defer close(ch)
go walkerProgressBar(ch)
files, dirs, err := walker(dir, fileTypesRegex, recursive, ch)
if err != nil {
return nil, err
}
for _, dir := range dirs {
newFiles, newDirs, err := walker(dir, fileTypesRegex, recursive, ch)
if err != nil {
return nil, err
}
files = append(files, newFiles...)
dirs = append(dirs, newDirs...)
}
return files, nil
}
func walker(dir string, fileTypesRegex *regexp.Regexp, recursive bool, counterCh chan CounterType) ([]string, []string, error) {
var files []string
var dirs []string
allFiles, err := ioutil.ReadDir(dir)
if err != nil {
return nil, nil, err
}
for _, file := range allFiles {
if file.IsDir() {
if recursive {
dirs = append(dirs, filepath.Join(dir, file.Name()))
counterCh <- Dir
}
} else {
if fileTypesRegex.MatchString(file.Name()) {
files = append(files, filepath.Join(dir, file.Name()))
counterCh <- File
}
}
}
return files, dirs, nil
}