Added update-user-group subcommand to repos command (#36)

* Added add-user subcommand to repos

* Fixed review comments

api/repositories.go

@@ -2,6 +2,7 @@ package api
 
 import (
 	"fmt"
+	"strings"
 
 	"github.com/shurcooL/graphql"
 )
@@ -105,6 +106,53 @@ func (r *Repositories) Delete(name, reason string, allowDataDeletion bool) error
 	return nil
 }
 
+type DefaultGroupEnum string
+
+const (
+	DefaultGroupEnumMember     DefaultGroupEnum = "Member"
+	DefaultGroupEnumAdmin      DefaultGroupEnum = "Admin"
+	DefaultGroupEnumEliminator DefaultGroupEnum = "Eliminator"
+)
+
+func (e DefaultGroupEnum) String() string {
+	return string(e)
+}
+
+func (e *DefaultGroupEnum) ParseString(s string) bool {
+	switch strings.ToLower(s) {
+	case "member":
+		*e = DefaultGroupEnumMember
+		return true
+	case "admin":
+		*e = DefaultGroupEnumAdmin
+		return true
+	case "eliminator":
+		*e = DefaultGroupEnumEliminator
+		return true
+	default:
+		return false
+	}
+}
+
+func (r *Repositories) UpdateUserGroup(name, username string, groups ...DefaultGroupEnum) error {
+	if len(groups) == 0 {
+		return fmt.Errorf("at least one group must be defined")
+	}
+
+	var mutation struct {
+		UpdateDefaultGroupMembershipsMutation struct {
+			ClientMutationId string
+		} `graphql:"updateDefaultGroupMemberships(input: {viewName: $name, userName: $username, groups: $groups})"`
+	}
+	variables := map[string]interface{}{
+		"name":     graphql.String(name),
+		"username": graphql.String(username),
+		"groups":   groups,
+	}
+
+	return r.client.Mutate(&mutation, variables)
+}
+
 func (r *Repositories) UpdateTimeBasedRetention(name string, retentionInDays float64, allowDataDeletion bool) error {
 	existingRepo, err := r.Get(name)
 	if err != nil {

cmd/humioctl/repos.go

@@ -33,6 +33,7 @@ func newReposCmd() *cobra.Command {
 	cmd.AddCommand(newReposCreateCmd())
 	cmd.AddCommand(newReposUpdateCmd())
 	cmd.AddCommand(newReposDeleteCmd())
+	cmd.AddCommand(newReposUpdateUserGroupCmd())
 
 	return cmd
 }

cmd/humioctl/repos_update_user_group.go

@@ -0,0 +1,38 @@
+package main
+
+import (
+	"os"
+
+	"github.com/humio/cli/api"
+	"github.com/spf13/cobra"
+)
+
+func newReposUpdateUserGroupCmd() *cobra.Command {
+	var groups []string
+	cmd := cobra.Command{
+		Use:   "update-user-group [flags] <repo> <username>",
+		Short: "Updates the users permissions to a repository based on default groups",
+		Args:  cobra.ExactArgs(2),
+		Run: func(cmd *cobra.Command, args []string) {
+			repoName := args[0]
+			userName := args[1]
+
+			var defaultGroups []api.DefaultGroupEnum
+			for _, group := range groups {
+				var defaultGroup api.DefaultGroupEnum
+				if !defaultGroup.ParseString(group) {
+					cmd.Println("the group '%s' was not valid (must be either 'Member', 'Admin' or 'Eliminator')")
+					os.Exit(1)
+				}
+				defaultGroups = append(defaultGroups, defaultGroup)
+			}
+
+			client := NewApiClient(cmd)
+			apiErr := client.Repositories().UpdateUserGroup(repoName, userName, defaultGroups...)
+			exitOnError(cmd, apiErr, "error adding user")
+		},
+	}
+	cmd.Flags().StringSliceVarP(&groups, "groups", "g", []string{api.DefaultGroupEnumMember.String()}, "the groups that the user should be added in")
+
+	return &cmd
+}