pkg.re/essentialkaos/ek@v12.36.0+incompatible/fsutil/helpers.go

// +build !windows

package fsutil

// ////////////////////////////////////////////////////////////////////////////////// //
//                                                                                    //
//                         Copyright (c) 2021 ESSENTIAL KAOS                          //
//      Apache License, Version 2.0 <https://www.apache.org/licenses/LICENSE-2.0>     //
//                                                                                    //
// ////////////////////////////////////////////////////////////////////////////////// //

import (
	"bufio"
	"errors"
	"io"
	"os"
	"path"
)

// ////////////////////////////////////////////////////////////////////////////////// //

var _disableCopyFileChecks bool // Flag for testing purposes only
var _disableMoveFileChecks bool // Flag for testing purposes only
var _disableCopyDirChecks bool  // Flag for testing purposes only

// ////////////////////////////////////////////////////////////////////////////////// //

// CopyFile copies file using bufio
func CopyFile(from, to string, perms ...os.FileMode) error {
	var targetExist = IsExist(to)

	if !_disableCopyFileChecks {
		dir := path.Dir(to)

		switch {
		case from == "":
			return errors.New("Source file can't be blank")
		case to == "":
			return errors.New("Target file can't be blank")

		case !IsExist(from):
			return errors.New("File " + from + " does not exists")
		case !IsRegular(from):
			return errors.New("File " + from + " is not a regular file")
		case !IsReadable(from):
			return errors.New("File " + from + " is not readable")

		case !targetExist && !IsExist(dir):
			return errors.New("Directory " + dir + " does not exists")
		case !targetExist && !IsWritable(dir):
			return errors.New("Directory " + dir + " is not writable")

		case targetExist && !IsWritable(to):
			return errors.New("Target file " + to + " is not writable")
		case targetExist && !IsRegular(to):
			return errors.New("Target is not a file")
		}
	}

	return copyFile(from, to, perms)
}

// MoveFile moves file
func MoveFile(from, to string, perms ...os.FileMode) error {
	if !_disableMoveFileChecks {
		targetExist := IsExist(to)
		dir := path.Dir(to)

		switch {
		case from == "":
			return errors.New("Source file can't be blank")
		case to == "":
			return errors.New("Target file can't be blank")

		case !IsExist(from):
			return errors.New("File " + from + " does not exists")
		case !IsRegular(from):
			return errors.New("File " + from + " is not a regular file")
		case !IsReadable(from):
			return errors.New("File " + from + " is not readable")

		case !targetExist && !IsExist(dir):
			return errors.New("Directory " + dir + " does not exists")
		case !targetExist && !IsWritable(dir):
			return errors.New("Directory " + dir + " is not writable")
		}
	}

	return moveFile(from, to, perms)
}

// CopyDir copies directory content recursively to target directory
func CopyDir(from, to string) error {
	if !_disableCopyDirChecks {
		switch {
		case from == "":
			return errors.New("Source directory can't be blank")
		case to == "":
			return errors.New("Target directory can't be blank")

		case !IsExist(from):
			return errors.New("Directory " + from + " does not exists")
		case !IsDir(from):
			return errors.New("Target " + from + " is not a directory")
		case !IsReadable(from):
			return errors.New("Directory " + from + " is not readable")
		case IsExist(to) && !IsDir(to):
			return errors.New("Target " + to + " is not a directory")
		case IsExist(to) && !IsWritable(to):
			return errors.New("Directory " + to + " is not writable")
		}
	}

	if !IsExist(to) {
		err := os.Mkdir(to, GetMode(from))

		if err != nil {
			return err
		}
	}

	return copyDir(from, to)
}

// TouchFile creates empty file
func TouchFile(path string, perm os.FileMode) error {
	fd, err := os.OpenFile(path, os.O_CREATE, perm)

	if err != nil {
		return err
	}

	return fd.Close()
}

// ////////////////////////////////////////////////////////////////////////////////// //

func copyFile(from, to string, perms []os.FileMode) error {
	var targetExist bool
	var perm os.FileMode

	if IsExist(to) {
		targetExist = true
	}

	if len(perms) == 0 {
		perm = GetMode(from)
	} else {
		perm = perms[0]
	}

	ffd, err := os.OpenFile(from, os.O_RDONLY, 0)

	if err != nil {
		return err
	}

	defer ffd.Close()

	tfd, err := os.OpenFile(to, os.O_CREATE|os.O_WRONLY|os.O_TRUNC, perm)

	if err != nil {
		return err
	}

	defer tfd.Close()

	reader := bufio.NewReader(ffd)

	_, err = io.Copy(tfd, reader)

	if err != nil {
		return err
	}

	if targetExist {
		return os.Chmod(to, perm)
	}

	return nil
}

func moveFile(from, to string, perms []os.FileMode) error {
	err := os.Rename(from, to)

	if err != nil {
		return err
	}

	if len(perms) == 0 {
		return nil
	}

	return os.Chmod(to, perms[0])
}

func copyDir(from, to string) error {
	var err error

	for _, target := range List(from, false) {
		fp := from + "/" + target
		tp := to + "/" + target

		if IsDir(fp) {
			err = os.Mkdir(tp, GetMode(fp))

			if err != nil {
				return err
			}

			err = copyDir(fp, tp)

			if err != nil {
				return err
			}
		} else {
			err = CopyFile(fp, tp)

			if err != nil {
				return err
			}
		}
	}

	return nil
}