Repository navigation
Expand file tree
/
Copy pathfs.go
More file actions
96 lines (79 loc) · 2.17 KB
/
Copy pathfs.go
File metadata and controls
96 lines (79 loc) · 2.17 KB
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
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
// Package fs provides functions for filesystem operations.
package fs
import (
"os"
"path/filepath"
"strings"
"github.com/spf13/afero"
)
var (
// MarkdownOnly is a filter that only lets pass Markdown files.
MarkdownOnly = func(file string) bool {
return filepath.Ext(file) == ".md"
}
// NoUnderscores is a predefined filter that doesn't let pass
// files starting with an underscore.
NoUnderscores = func(file string) bool {
filename := filepath.Base(file)
return !strings.HasPrefix(filename, "_")
}
// ErrStreaming is returned from StreamFiles.
ErrStreaming error = nil
)
// StreamFiles sends all relative file paths inside a given path that
// match the given filters through the files channel.
func StreamFiles(path string, files chan<- string, filters ...func(file string) bool) error {
if _, err := os.Stat(path); os.IsNotExist(err) {
close(files)
return nil
}
// Convert to absolute path so that it does not make a difference if the paths are in different formats.
// e.g. one "example/" and the other one "./example"
path, err := filepath.Abs(path)
if err != nil {
return err
}
ErrStreaming = filepath.Walk(path, func(file string, info os.FileInfo, err error) error {
if err != nil {
return err
}
if info.IsDir() {
return nil
}
for _, filter := range filters {
if !filter(file) {
return nil
}
}
// Convert to absolute path so that it does not make a difference if the paths are in different formats.
// e.g. one "example/" and the other one "./example"
file, err = filepath.Abs(file)
if err != nil {
return err
}
files <- file[len(path):]
return nil
})
close(files)
return ErrStreaming
}
// MkdirAll creates one or more directories inside the given path.
func MkdirAll(path string, dirs ...string) error {
for _, dir := range dirs {
if err := os.MkdirAll(filepath.Join(path, dir), 0755); err != nil {
return err
}
}
return nil
}
// Rmdir removes an entire directory along with its contents. If the
// directory does not exist, nothing happens.
func Rmdir(fs afero.Fs, path string) error {
_, err := fs.Stat(path)
if err != nil {
if !os.IsNotExist(err) {
return err
}
}
return fs.RemoveAll(path)
}