From db7260965078e1053e63a6d1d3b7895d380c3487 Mon Sep 17 00:00:00 2001 From: Eduard Urbach Date: Fri, 19 Apr 2019 16:44:37 +0900 Subject: [PATCH] Made change-role patch accept CLI arguments --- patches/change-role/change-role.go | 21 +++++++++++++++++++-- 1 file changed, 19 insertions(+), 2 deletions(-) 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() }