|
|
|
@ -5,8 +5,11 @@
|
|
|
|
|
package util
|
|
|
|
|
|
|
|
|
|
import (
|
|
|
|
|
"errors"
|
|
|
|
|
"os"
|
|
|
|
|
"path"
|
|
|
|
|
"path/filepath"
|
|
|
|
|
"strings"
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
// EnsureAbsolutePath ensure that a path is absolute, making it
|
|
|
|
@ -70,3 +73,80 @@ func IsExist(path string) (bool, error) {
|
|
|
|
|
}
|
|
|
|
|
return false, err
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func statDir(dirPath, recPath string, includeDir, isDirOnly, followSymlinks bool) ([]string, error) {
|
|
|
|
|
dir, err := os.Open(dirPath)
|
|
|
|
|
if err != nil {
|
|
|
|
|
return nil, err
|
|
|
|
|
}
|
|
|
|
|
defer dir.Close()
|
|
|
|
|
|
|
|
|
|
fis, err := dir.Readdir(0)
|
|
|
|
|
if err != nil {
|
|
|
|
|
return nil, err
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
statList := make([]string, 0)
|
|
|
|
|
for _, fi := range fis {
|
|
|
|
|
if strings.Contains(fi.Name(), ".DS_Store") {
|
|
|
|
|
continue
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
relPath := path.Join(recPath, fi.Name())
|
|
|
|
|
curPath := path.Join(dirPath, fi.Name())
|
|
|
|
|
if fi.IsDir() {
|
|
|
|
|
if includeDir {
|
|
|
|
|
statList = append(statList, relPath+"/")
|
|
|
|
|
}
|
|
|
|
|
s, err := statDir(curPath, relPath, includeDir, isDirOnly, followSymlinks)
|
|
|
|
|
if err != nil {
|
|
|
|
|
return nil, err
|
|
|
|
|
}
|
|
|
|
|
statList = append(statList, s...)
|
|
|
|
|
} else if !isDirOnly {
|
|
|
|
|
statList = append(statList, relPath)
|
|
|
|
|
} else if followSymlinks && fi.Mode()&os.ModeSymlink != 0 {
|
|
|
|
|
link, err := os.Readlink(curPath)
|
|
|
|
|
if err != nil {
|
|
|
|
|
return nil, err
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
isDir, err := IsDir(link)
|
|
|
|
|
if err != nil {
|
|
|
|
|
return nil, err
|
|
|
|
|
}
|
|
|
|
|
if isDir {
|
|
|
|
|
if includeDir {
|
|
|
|
|
statList = append(statList, relPath+"/")
|
|
|
|
|
}
|
|
|
|
|
s, err := statDir(curPath, relPath, includeDir, isDirOnly, followSymlinks)
|
|
|
|
|
if err != nil {
|
|
|
|
|
return nil, err
|
|
|
|
|
}
|
|
|
|
|
statList = append(statList, s...)
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
return statList, nil
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// StatDir gathers information of given directory by depth-first.
|
|
|
|
|
// It returns slice of file list and includes subdirectories if enabled;
|
|
|
|
|
// it returns error and nil slice when error occurs in underlying functions,
|
|
|
|
|
// or given path is not a directory or does not exist.
|
|
|
|
|
//
|
|
|
|
|
// Slice does not include given path itself.
|
|
|
|
|
// If subdirectories is enabled, they will have suffix '/'.
|
|
|
|
|
func StatDir(rootPath string, includeDir ...bool) ([]string, error) {
|
|
|
|
|
if isDir, err := IsDir(rootPath); err != nil {
|
|
|
|
|
return nil, err
|
|
|
|
|
} else if !isDir {
|
|
|
|
|
return nil, errors.New("not a directory or does not exist: " + rootPath)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
isIncludeDir := false
|
|
|
|
|
if len(includeDir) != 0 {
|
|
|
|
|
isIncludeDir = includeDir[0]
|
|
|
|
|
}
|
|
|
|
|
return statDir(rootPath, "", isIncludeDir, false, false)
|
|
|
|
|
}
|
|
|
|
|