107 lines
1.9 KiB
Go
107 lines
1.9 KiB
Go
package fileutils
|
|
|
|
import (
|
|
"errors"
|
|
"path/filepath"
|
|
|
|
"github.com/spf13/afero"
|
|
)
|
|
|
|
// CopyDir copies a directory from source to dest and all
|
|
// of its sub-directories. It doesn't stop if it finds an error
|
|
// during the copy. Returns an error if any.
|
|
func CopyDir(fs afero.Fs, source, dest string) error {
|
|
// Get properties of source.
|
|
srcinfo, err := fs.Stat(source)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Create the destination directory.
|
|
err = fs.MkdirAll(dest, srcinfo.Mode())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
dir, _ := fs.Open(source)
|
|
obs, err := dir.Readdir(-1)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var errs []error
|
|
|
|
for _, obj := range obs {
|
|
fsource := source + "/" + obj.Name()
|
|
fdest := dest + "/" + obj.Name()
|
|
|
|
if obj.IsDir() {
|
|
// Create sub-directories, recursively.
|
|
err = CopyDir(fs, fsource, fdest)
|
|
if err != nil {
|
|
errs = append(errs, err)
|
|
}
|
|
} else {
|
|
// Perform the file copy.
|
|
err = CopyFile(fs, fsource, fdest)
|
|
if err != nil {
|
|
errs = append(errs, err)
|
|
}
|
|
}
|
|
}
|
|
|
|
var errString string
|
|
for _, err := range errs {
|
|
errString += err.Error() + "\n"
|
|
}
|
|
|
|
if errString != "" {
|
|
return errors.New(errString)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func DiskUsage(fs afero.Fs, path string, maxDepth int) (size, inodes int64, err error) {
|
|
info, err := fs.Stat(path)
|
|
if err != nil {
|
|
return 0, 0, err
|
|
}
|
|
|
|
size = info.Size()
|
|
inodes = int64(1)
|
|
|
|
if !info.IsDir() {
|
|
return size, inodes, err
|
|
}
|
|
|
|
if maxDepth < 1 {
|
|
return size, inodes, err
|
|
}
|
|
|
|
dir, err := fs.Open(path)
|
|
if err != nil {
|
|
return size, inodes, err
|
|
}
|
|
defer dir.Close()
|
|
|
|
fis, err := dir.Readdir(-1)
|
|
if err != nil {
|
|
return size, inodes, err
|
|
}
|
|
|
|
for _, fi := range fis {
|
|
if fi.Name() == "." || fi.Name() == ".." {
|
|
continue
|
|
}
|
|
s, i, e := DiskUsage(fs, filepath.Join(path, fi.Name()), maxDepth-1)
|
|
if e != nil {
|
|
return size, inodes, e
|
|
}
|
|
size += s
|
|
inodes += i
|
|
}
|
|
|
|
return size, inodes, err
|
|
}
|