cli/cmd/user/remove.go - main - dimamir/rill

cli/cmd/user/remove.go
75 строк2 KB
Benjamin Egelund-Müller
CLI: Clarify user and usergroup commands (#5552)
29 авг 2024, 14:18
29 авг 2024, 14:1802bfc53
package user

import (
	"github.com/rilldata/rill/cli/pkg/cmdutil"
	adminv1 "github.com/rilldata/rill/proto/gen/rill/admin/v1"
	"github.com/spf13/cobra"
)

func RemoveCmd(ch *cmdutil.Helper) *cobra.Command {
	var projectName string
	var groupName string
	var email string
	var keepProjectRoles bool

	removeCmd := &cobra.Command{
		Use:   "remove",
		Short: "Remove a user",
		RunE: func(cmd *cobra.Command, args []string) error {
			err := cmdutil.StringPromptIfEmpty(&email, "Enter email")
			if err != nil {
				return err
			}

			client, err := ch.Client()
			if err != nil {
				return err
			}

			if groupName != "" {
				_, err = client.RemoveUsergroupMemberUser(cmd.Context(), &adminv1.RemoveUsergroupMemberUserRequest{
					Organization: ch.Org,
					Usergroup:    groupName,
					Email:        email,
				})
				if err != nil {
					return err
				}

				ch.PrintfSuccess("Removed user %q from user group \"%s/%s\"\n", email, ch.Org, groupName)
				return nil
			} else if projectName != "" {
				_, err = client.RemoveProjectMemberUser(cmd.Context(), &adminv1.RemoveProjectMemberUserRequest{
					Organization: ch.Org,
					Project:      projectName,
					Email:        email,
				})
				if err != nil {
					return err
				}

				ch.PrintfSuccess("Removed user %q from project \"%s/%s\"\n", email, ch.Org, projectName)
			} else {
				_, err = client.RemoveOrganizationMemberUser(cmd.Context(), &adminv1.RemoveOrganizationMemberUserRequest{
					Organization:     ch.Org,
					Email:            email,
					KeepProjectRoles: keepProjectRoles,
				})
				if err != nil {
					return err
				}
				ch.PrintfSuccess("Removed user %q from organization %q\n", email, ch.Org)
			}

			return nil
		},
	}

	removeCmd.Flags().StringVar(&ch.Org, "org", ch.Org, "Organization")
	removeCmd.Flags().StringVar(&projectName, "project", "", "Project")
	removeCmd.Flags().StringVar(&groupName, "group", "", "User group")
	removeCmd.Flags().StringVar(&email, "email", "", "Email of the user")
	removeCmd.Flags().BoolVar(&keepProjectRoles, "keep-project-roles", false, "Keep roles granted directly on projects in the org")

	return removeCmd
}