123 lines
2.3 KiB
Go
123 lines
2.3 KiB
Go
package rice
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"sort"
|
|
"strings"
|
|
)
|
|
|
|
// Walk is like filepath.Walk()
|
|
// Visit http://golang.org/pkg/path/filepath/#Walk for more information
|
|
func (b *Box) Walk(path string, walkFn filepath.WalkFunc) error {
|
|
|
|
pathFile, err := b.Open(path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer pathFile.Close()
|
|
|
|
pathInfo, err := pathFile.Stat()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if b.IsAppended() || b.IsEmbedded() {
|
|
return b.walk(path, pathInfo, walkFn)
|
|
}
|
|
|
|
// We don't have any embedded or appended box so use live filesystem mode
|
|
return filepath.Walk(b.absolutePath+string(os.PathSeparator)+path, func(path string, info os.FileInfo, err error) error {
|
|
|
|
// Strip out the box name from the returned paths
|
|
path = strings.TrimPrefix(path, b.absolutePath+string(os.PathSeparator))
|
|
return walkFn(path, info, err)
|
|
|
|
})
|
|
|
|
}
|
|
|
|
// walk recursively descends path.
|
|
// See walk() in $GOROOT/src/pkg/path/filepath/path.go
|
|
func (b *Box) walk(path string, info os.FileInfo, walkFn filepath.WalkFunc) error {
|
|
|
|
err := walkFn(path, info, nil)
|
|
if err != nil {
|
|
if info.IsDir() && err == filepath.SkipDir {
|
|
return nil
|
|
}
|
|
return err
|
|
}
|
|
|
|
if !info.IsDir() {
|
|
return nil
|
|
}
|
|
|
|
names, err := b.readDirNames(path)
|
|
if err != nil {
|
|
return walkFn(path, info, err)
|
|
}
|
|
|
|
for _, name := range names {
|
|
|
|
filename := filepath.Join(path, name)
|
|
fileObject, err := b.Open(filename)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer fileObject.Close()
|
|
|
|
fileInfo, err := fileObject.Stat()
|
|
if err != nil {
|
|
if err := walkFn(filename, fileInfo, err); err != nil && err != filepath.SkipDir {
|
|
return err
|
|
}
|
|
} else {
|
|
err = b.walk(filename, fileInfo, walkFn)
|
|
if err != nil {
|
|
if !fileInfo.IsDir() || err != filepath.SkipDir {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
// readDirNames reads the directory named by path and returns a sorted list of directory entries.
|
|
// See readDirNames() in $GOROOT/pkg/path/filepath/path.go
|
|
func (b *Box) readDirNames(path string) ([]string, error) {
|
|
|
|
f, err := b.Open(path)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer f.Close()
|
|
|
|
stat, err := f.Stat()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if !stat.IsDir() {
|
|
return nil, nil
|
|
}
|
|
|
|
infos, err := f.Readdir(0)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var names []string
|
|
|
|
for _, info := range infos {
|
|
names = append(names, info.Name())
|
|
}
|
|
|
|
sort.Strings(names)
|
|
return names, nil
|
|
|
|
}
|