Skip to content

Commit

Permalink
add possibility to define plugin to run after a creation of user home…
Browse files Browse the repository at this point in the history
… folder
  • Loading branch information
gmgigi96 committed Sep 13, 2022
1 parent 10caf12 commit dc94be4
Show file tree
Hide file tree
Showing 2 changed files with 29 additions and 10 deletions.
5 changes: 5 additions & 0 deletions pkg/storage/utils/eosfs/config.go
Original file line number Diff line number Diff line change
Expand Up @@ -168,4 +168,9 @@ type Config struct {
// TokenExpiry stores in seconds the time after which generated tokens will expire
// Default is 3600
TokenExpiry int

// Path of the script to run after an user home folder has been created
OnPostCreateHomeHook string `mapstructure:"on_post_create_home_hook"`
// Whether to enable the post create home hook
EnablePostCreateHomeHook bool `mapstructure:"enable_post_create_home_hook"`
}
34 changes: 24 additions & 10 deletions pkg/storage/utils/eosfs/eosfs.go
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,7 @@ import (
"io"
"net/url"
"os"
"os/exec"
"path"
"regexp"
"strconv"
Expand Down Expand Up @@ -1430,36 +1431,36 @@ func (fs *eosfs) createShadowHome(ctx context.Context) error {
return nil
}

func (fs *eosfs) createNominalHome(ctx context.Context) error {
func (fs *eosfs) createNominalHome(ctx context.Context) (string, error) {
home := fs.wrap(ctx, "/")

u, err := getUser(ctx)
if err != nil {
return errors.Wrap(err, "eosfs: no user in ctx")
return "", errors.Wrap(err, "eosfs: no user in ctx")
}
auth, err := fs.getUserAuth(ctx, u, "")
if err != nil {
return err
return "", err
}

rootAuth, err := fs.getRootAuth(ctx)
if err != nil {
return nil
return "", nil
}

_, err = fs.c.GetFileInfoByPath(ctx, rootAuth, home)
if err == nil { // home already exists
return nil
return "", nil
}

if _, ok := err.(errtypes.IsNotFound); !ok {
return errors.Wrap(err, "eosfs: error verifying if user home directory exists")
return "", errors.Wrap(err, "eosfs: error verifying if user home directory exists")
}

err = fs.createUserDir(ctx, u, home, false)
if err != nil {
err := errors.Wrap(err, "eosfs: error creating user dir")
return err
return "", err
}

// set quota for user
Expand All @@ -1475,28 +1476,41 @@ func (fs *eosfs) createNominalHome(ctx context.Context) error {
err = fs.c.SetQuota(ctx, rootAuth, quotaInfo)
if err != nil {
err := errors.Wrap(err, "eosfs: error setting quota")
return err
return "", err
}

return err
return home, nil
}

func (fs *eosfs) CreateHome(ctx context.Context) error {
if !fs.conf.EnableHome {
return errtypes.NotSupported("eosfs: create home not supported")
}

if err := fs.createNominalHome(ctx); err != nil {
var err error
var home string
if home, err = fs.createNominalHome(ctx); err != nil {
return errors.Wrap(err, "eosfs: error creating nominal home")
}

if err := fs.createShadowHome(ctx); err != nil {
return errors.Wrap(err, "eosfs: error creating shadow home")
}

if fs.conf.EnablePostCreateHomeHook {
if err := fs.runPostCreateHomeHook(ctx, home); err != nil {
return errors.Wrap(err, "eosfs: error running post create home hook")
}
}

return nil
}

func (fs *eosfs) runPostCreateHomeHook(ctx context.Context, home string) error {
user := ctxpkg.ContextMustGetUser(ctx)
return exec.Command(fs.conf.OnPostCreateHomeHook, user.Username, home).Run()
}

func (fs *eosfs) createUserDir(ctx context.Context, u *userpb.User, path string, recursiveAttr bool) error {
rootAuth, err := fs.getRootAuth(ctx)
if err != nil {
Expand Down

0 comments on commit dc94be4

Please sign in to comment.