79 lines
1.5 KiB
Go
79 lines
1.5 KiB
Go
![]() |
package getter
|
||
|
|
||
|
import (
|
||
|
"io"
|
||
|
"os"
|
||
|
"path/filepath"
|
||
|
"strings"
|
||
|
)
|
||
|
|
||
|
// copyDir copies the src directory contents into dst. Both directories
|
||
|
// should already exist.
|
||
|
//
|
||
|
// If ignoreDot is set to true, then dot-prefixed files/folders are ignored.
|
||
|
func copyDir(dst string, src string, ignoreDot bool) error {
|
||
|
src, err := filepath.EvalSymlinks(src)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
walkFn := func(path string, info os.FileInfo, err error) error {
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
if path == src {
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
if ignoreDot && strings.HasPrefix(filepath.Base(path), ".") {
|
||
|
// Skip any dot files
|
||
|
if info.IsDir() {
|
||
|
return filepath.SkipDir
|
||
|
} else {
|
||
|
return nil
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// The "path" has the src prefixed to it. We need to join our
|
||
|
// destination with the path without the src on it.
|
||
|
dstPath := filepath.Join(dst, path[len(src):])
|
||
|
|
||
|
// If we have a directory, make that subdirectory, then continue
|
||
|
// the walk.
|
||
|
if info.IsDir() {
|
||
|
if path == filepath.Join(src, dst) {
|
||
|
// dst is in src; don't walk it.
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
if err := os.MkdirAll(dstPath, 0755); err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
// If we have a file, copy the contents.
|
||
|
srcF, err := os.Open(path)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
defer srcF.Close()
|
||
|
|
||
|
dstF, err := os.Create(dstPath)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
defer dstF.Close()
|
||
|
|
||
|
if _, err := io.Copy(dstF, srcF); err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
// Chmod it
|
||
|
return os.Chmod(dstPath, info.Mode())
|
||
|
}
|
||
|
|
||
|
return filepath.Walk(src, walkFn)
|
||
|
}
|