remove.go

package wrfs

import (
	"path"
)

// RemoveFS is a file system that supports the Remove function.
type RemoveFS interface {
	FS

	// Remove removes the named file or (empty) directory.
	Remove(name string) error
}

// Remove removes the named file or (empty) directory.
func Remove(fsys FS, name string) error {
	if fsys, ok := fsys.(RemoveFS); ok {
		return fsys.Remove(name)
	}
	return &PathError{Op: "remove", Path: name, Err: ErrUnsupported}
}

// RemoveAllFS is a file system that supports the RemoveAll function.
type RemoveAllFS interface {
	FS

	// RemoveAll removes path and any children it contains.
	RemoveAll(path string) error
}

// RemoveAll removes path and any children it contains.
func RemoveAll(fsys FS, removePath string) error {
	if fsys, ok := fsys.(RemoveAllFS); ok {
		return fsys.RemoveAll(removePath)
	}

	fi, err := Stat(fsys, removePath)
	if err != nil {
		return err
	}

	// Check if we are removing a file or a directory.
	if !fi.IsDir() {
		return Remove(fsys, removePath)
	}

	files, err := ReadDir(fsys, removePath)
	if err != nil {
		return err
	}

	for _, fi := range files {
		if fi.IsDir() {
			if err = RemoveAll(fsys, path.Join(removePath, fi.Name())); err != nil {
				return err
			}
		} else if err = Remove(fsys, path.Join(removePath, fi.Name())); err != nil {
			return err
		}
	}

	return Remove(fsys, removePath)
}