140 lines
3.6 KiB
Go
140 lines
3.6 KiB
Go
package cmd
|
|
|
|
import (
|
|
wisski_distillery "github.com/FAU-CDI/wisski-distillery"
|
|
"github.com/FAU-CDI/wisski-distillery/internal/cli"
|
|
"github.com/FAU-CDI/wisski-distillery/internal/dis/component/instances"
|
|
"github.com/FAU-CDI/wisski-distillery/internal/models"
|
|
"github.com/tkw1536/goprogram/exit"
|
|
)
|
|
|
|
// DisGrant is the 'dis_grant' command
|
|
var DisGrant wisski_distillery.Command = disGrant{}
|
|
|
|
type disGrant struct {
|
|
AddAll bool `short:"m" long:"add-all" description:"add grant to all WissKIs"`
|
|
AddUser bool `short:"a" long:"add" description:"add or update a user to a given wisski"`
|
|
RemoveUser bool `short:"r" long:"remove" description:"remove a user from a given wisski"`
|
|
|
|
DrupalAdmin bool `short:"A" long:"admin" description:"grant user the admin role"`
|
|
|
|
Positionals struct {
|
|
User string `positional-arg-name:"USER" required:"1-1" description:"distillery username"`
|
|
Slug string `positional-arg-name:"SLUG" description:"WissKI instance"`
|
|
DrupalUser string `positional-arg-name:"DRUPAL" description:"drupal username"`
|
|
} `positional-args:"true"`
|
|
}
|
|
|
|
func (disGrant) Description() wisski_distillery.Description {
|
|
return wisski_distillery.Description{
|
|
Requirements: cli.Requirements{
|
|
NeedsDistillery: true,
|
|
},
|
|
Command: "dis_grant",
|
|
Description: "grant distillery users access to specific WissKIs",
|
|
}
|
|
}
|
|
|
|
var errNoSlugSelect = exit.Error{
|
|
Message: "slug not provided",
|
|
ExitCode: exit.ExitCommandArguments,
|
|
}
|
|
|
|
func (dg disGrant) AfterParse() error {
|
|
var counter int
|
|
for _, action := range []bool{
|
|
dg.AddUser,
|
|
dg.RemoveUser,
|
|
dg.AddAll,
|
|
} {
|
|
if action {
|
|
counter++
|
|
}
|
|
}
|
|
|
|
if counter != 1 {
|
|
return errNoActionSelected
|
|
}
|
|
|
|
if !dg.AddAll && dg.Positionals.Slug == "" {
|
|
return errNoSlugSelect
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
var errFailedGrant = exit.Error{
|
|
Message: "unable to manage grants",
|
|
ExitCode: exit.ExitGeneric,
|
|
}
|
|
|
|
func (dg disGrant) Run(context wisski_distillery.Context) (err error) {
|
|
defer errFailedGrant.DeferWrap(&err)
|
|
|
|
switch {
|
|
case dg.AddUser:
|
|
return dg.runAddUser(context)
|
|
case dg.AddAll:
|
|
return dg.runAddAll(context)
|
|
case dg.RemoveUser:
|
|
return dg.runRemoveUser(context)
|
|
}
|
|
panic("never reached")
|
|
}
|
|
|
|
func (dg disGrant) checkHasSlug(context wisski_distillery.Context) error {
|
|
has, err := context.Environment.Instances().Has(context.Context, dg.Positionals.Slug)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if !has {
|
|
return instances.ErrWissKINotFound
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (dg disGrant) runAddUser(context wisski_distillery.Context) error {
|
|
if err := dg.checkHasSlug(context); err != nil {
|
|
return err
|
|
}
|
|
|
|
policy := context.Environment.Policy()
|
|
return policy.Set(context.Context, models.Grant{
|
|
User: dg.Positionals.User,
|
|
Slug: dg.Positionals.Slug,
|
|
DrupalUsername: dg.Positionals.DrupalUser,
|
|
DrupalAdminRole: dg.DrupalAdmin,
|
|
})
|
|
}
|
|
|
|
func (dg disGrant) runRemoveUser(context wisski_distillery.Context) error {
|
|
if err := dg.checkHasSlug(context); err != nil {
|
|
return err
|
|
}
|
|
|
|
policy := context.Environment.Policy()
|
|
return policy.Remove(context.Context, dg.Positionals.User, dg.Positionals.Slug)
|
|
}
|
|
|
|
func (dg disGrant) runAddAll(context wisski_distillery.Context) error {
|
|
policy := context.Environment.Policy()
|
|
|
|
instances, err := context.Environment.Instances().All(context.Context)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for _, instance := range instances {
|
|
context.Printf("Adding grant for user %s to %s\n", dg.Positionals.User, instance.Slug)
|
|
if err := policy.Set(context.Context, models.Grant{
|
|
User: dg.Positionals.User,
|
|
Slug: instance.Slug,
|
|
DrupalUsername: dg.Positionals.User,
|
|
DrupalAdminRole: dg.DrupalAdmin,
|
|
}); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|