move fs and file ownership related funcs to fs package

This commit is contained in:
Pavel Sviderski
2024-11-27 14:30:08 +10:00
parent ca607ee82d
commit a6ccd6055a
3 changed files with 52 additions and 44 deletions
+2 -42
View File
@@ -6,9 +6,7 @@ import (
"github.com/BurntSushi/toml"
"net/netip"
"os"
"os/user"
"path/filepath"
"strconv"
"uncloud/internal/fs"
)
const (
@@ -54,46 +52,8 @@ func (c *Config) Write(path, owner string) error {
if err := os.WriteFile(path, data.Bytes(), 0600); err != nil {
return err
}
if err := Chown(path, owner); err != nil {
if err := fs.Chown(path, owner); err != nil {
return err
}
return nil
}
func MkDataDir(dir, owner string) error {
parent, _ := filepath.Split(dir)
// Use 0711 for parent directories to allow `owner` to access its nested data directory.
if err := os.MkdirAll(parent, 0711); err != nil {
return fmt.Errorf("create directory %q: %w", parent, err)
}
if err := os.Mkdir(dir, 0700); err != nil {
if !os.IsExist(err) {
return fmt.Errorf("create directory %q: %w", dir, err)
}
}
if err := Chown(dir, owner); err != nil {
return err
}
return nil
}
func Chown(path, owner string) error {
if owner != "" {
usr, err := user.Lookup(owner)
if err != nil {
return fmt.Errorf("lookup user %q: %w", owner, err)
}
uid, err := strconv.Atoi(usr.Uid)
if err != nil {
return fmt.Errorf("parse %q user ID (UID) %q: %w", owner, usr.Uid, err)
}
gid, err := strconv.Atoi(usr.Gid)
if err != nil {
return fmt.Errorf("parse %q user group ID (GID) %q: %w", owner, usr.Gid, err)
}
if err = os.Chown(path, uid, gid); err != nil {
return fmt.Errorf("chown %q: %w", path, err)
}
}
return nil
}