|
|
|
@ -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) |
|
|
|
|
} |
|
|
|
|