diff --git a/patches/change-role/change-role.go b/patches/change-role/change-role.go index 7e7fe491..dc300227 100644 --- a/patches/change-role/change-role.go +++ b/patches/change-role/change-role.go @@ -1,13 +1,30 @@ package main import ( + "flag" + "github.com/animenotifier/arn" ) +// Shell parameters +var userID string +var role string + +// Shell flags +func init() { + flag.StringVar(&userID, "id", "", "ID of the user") + flag.StringVar(&role, "role", "", "The user's new role") + flag.Parse() +} + func main() { defer arn.Node.Close() - user, _ := arn.GetUser("Vy2Hk5yvx") - user.Role = "" + // Get user + user, err := arn.GetUser(userID) + arn.PanicOnError(err) + + // Save role + user.Role = role user.Save() }