directory.go

package copy

import (
	"io/ioutil"
	"os"
	"path/filepath"

	"github.com/pkg/errors"
)

type directory struct {
	base
}

func newDirectory(path string, fi os.FileInfo) directory {
	return directory{base{path, fi}}
}

// copyTo recursively copies directories from d.path to dst
func (d directory) copyTo(dst string, linkOrCopy bool) error {
	// create new directory with source mode
	if err := os.MkdirAll(dst, d.info.Mode()); err != nil {
		return errors.Wrapf(err, "MkdirAll(%s,%s)", dst, d.info.Mode().String())
	}

	// get all children
	children, err := ioutil.ReadDir(d.path)
	if err != nil {
		return errors.Wrapf(err, "ReadDir(%s)", d.path)
	}

	// Make sure we *can* copy the children if any
	if len(children) > 0 && d.info.Mode()&0200 == 0 {
		if err := os.Chmod(dst, d.info.Mode()|0200); err != nil {
			return errors.Wrapf(err, "Chmod(%s,%s)", dst, d.info.Mode()|0200)
		}
	}

	// copy each child recursively
	for _, child := range children {
		childSrc := filepath.Join(d.path, child.Name())
		childDst := filepath.Join(dst, child.Name())

		obj, err := newObject(childSrc)
		if err != nil {
			return errors.Wrapf(err, "newObject(%s)", childSrc)
		}

		if err = obj.copyTo(childDst, linkOrCopy); err != nil {
			return errors.Wrapf(err, "copyTo(%s,%t)", childDst, linkOrCopy)
		}
	}

	// Restore the directories modes if we made it writeable
	if len(children) > 0 && d.info.Mode()&0200 == 0 {
		if err := os.Chmod(dst, d.info.Mode()); err != nil {
			return errors.Wrapf(err, "Chmod(%s,%s)", dst, d.info.Mode())
		}
	}

	// successful
	return nil
}

func (d directory) String() string {
	return "directory: " + d.path
}