internal/file/root.go (view raw)
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 | //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) {
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
}
|