cli/cmd/sudo/superuser/remove.go - main - dimamir/rill

cli/cmd/sudo/superuser/remove.go

37 строк744 B

Benjamin Egelund-Müller

CLI: Util functions cleanup (#4141)

26 фев 2024, 15:20

26 фев 2024, 15:209f946f9

package superuser

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 {
	removeCmd := &cobra.Command{
		Use:   "remove <email>",
		Args:  cobra.ExactArgs(1),
		Short: "Remove a superuser",
		RunE: func(cmd *cobra.Command, args []string) error {
			ctx := cmd.Context()

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

			_, err = client.SetSuperuser(ctx, &adminv1.SetSuperuserRequest{
				Email:     args[0],
				Superuser: false,
			})
			if err != nil {
				return err
			}

			ch.PrintfSuccess("Removed superuser from %q\n", args[0])

			return nil
		},
	}

	return removeCmd
}