//nolint:forbidigo // wrappers for os.File functions go here
package file

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

	"github.com/Southclaws/fault"
	"github.com/Southclaws/fault/fmsg"
)

type Root struct {
	*os.Root
}

func CreateAndOpenRoot(name string) (*Root, error) {
	exists, err := exists(name)
	if err != nil {
		return nil, fault.Wrap(err, fmsg.Withf("failed to check data root existence %s", name))
	}
	if !exists {
		err := mkdirp(name)
		if err != nil {
			return nil, fault.Wrap(err, fmsg.Withf("failed to create data root %s", name))
		}
	}

	return OpenRoot(name)
}

func OpenRoot(name string) (*Root, error) {
	if !filepath.IsAbs(name) {
		wd, err := os.Getwd()
		if err != nil {
			return nil, fault.Wrap(err, fmsg.With("failed to get current working directory"))
		}

		name = filepath.Join(wd, name)
	}

	r, err := os.OpenRoot(name)
	if err != nil {
		return nil, fault.Wrap(err, fmsg.Withf("failed to open data root directory %s", name))
	}

	return &Root{
		Root: r,
	}, nil
}

func (r *Root) JoinPath(path string) string {
	return filepath.Join(r.Name(), path)
}

func (r *Root) StatIfExists(name string) (fs.FileInfo, error) {
	stat, err := r.Stat(name)

	return stat, needNotExist(err)
}

func (r *Root) Exists(file string) (bool, error) {
	stat, err := r.StatIfExists(file)

	return stat != nil, err
}

func (r *Root) ReadFile(name string) ([]byte, error) {
	b, err := fs.ReadFile(r.FS(), name)
	if err != nil {
		return nil, fault.Wrap(err, fmsg.Withf("failed to read file %s", name))
	}

	return b, nil
}

func (r *Root) WriteFile(name string, data []byte, perm fs.FileMode) error {
	f, err := r.OpenFile(name, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, perm)
	if err != nil {
		return fault.Wrap(err, fmsg.Withf("failed to write file %s", name))
	}
	defer f.Close()

	_, err = f.Write(data)
	if err != nil {
		return fault.Wrap(err, fmsg.Withf("failed to write file %s", name))
	}

	return nil
}

func (r *Root) RemoveAll() error {
	err := os.RemoveAll(r.Name())
	if err != nil {
		return fault.Wrap(err, fmsg.Withf("failed to remove data root %s", r.Name()))
	}

	return nil
}