| package getter |
| |
| import ( |
| "context" |
| "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(ctx context.Context, 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 := Copy(ctx, dstF, srcF); err != nil { |
| return err |
| } |
| |
| // Chmod it |
| return os.Chmod(dstPath, info.Mode()) |
| } |
| |
| return filepath.Walk(src, walkFn) |
| } |