github.com/1aal/kubeblocks@v0.0.0-20231107070852-e1c03e598921/pkg/cli/cmd/cluster/accounts.go

/*
Copyright (C) 2022-2023 ApeCloud Co., Ltd

This file is part of KubeBlocks project

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU Affero General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU Affero General Public License for more details.

You should have received a copy of the GNU Affero General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/

package cluster

import (
	"github.com/spf13/cobra"
	"k8s.io/cli-runtime/pkg/genericiooptions"
	cmdutil "k8s.io/kubectl/pkg/cmd/util"
	"k8s.io/kubectl/pkg/util/templates"

	"github.com/1aal/kubeblocks/pkg/cli/cmd/accounts"
	"github.com/1aal/kubeblocks/pkg/cli/types"
	"github.com/1aal/kubeblocks/pkg/cli/util"
)

var (
	createUserExamples = templates.Examples(`
		# create account with password
		kbcli cluster create-account CLUSTERNAME --component COMPNAME --name USERNAME --password PASSWD
		# create account without password
		kbcli cluster create-account CLUSTERNAME --component COMPNAME --name USERNAME
		# create account with default component
		kbcli cluster create-account CLUSTERNAME --name USERNAME
		# create account for instance
		kbcli cluster create-account  --instance INSTANCE --name USERNAME
 `)

	deleteUserExamples = templates.Examples(`
		# delete account by name
		kbcli cluster delete-account CLUSTERNAME --component COMPNAME --name USERNAME
		# delete account with default component
		kbcli cluster delete-account CLUSTERNAME --name USERNAME
		# delete account for instance
		kbcli cluster delete-account --instance INSTANCE --name USERNAME
 `)

	descUserExamples = templates.Examples(`
		# describe account and show role information
		kbcli cluster describe-account CLUSTERNAME --component COMPNAME --name USERNAME
		# describe account with default component
		kbcli cluster describe-account CLUSTERNAME --name USERNAME
		# describe account for instance
		kbcli cluster describe-account --instance INSTANCE --name USERNAME
 `)

	listUsersExample = templates.Examples(`
		# list all users for component
		kbcli cluster list-accounts CLUSTERNAME --component COMPNAME
		# list all users with default component
		kbcli cluster list-accounts CLUSTERNAME
		# list all users from instance
		kbcli cluster list-accounts --instance INSTANCE
	`)
	grantRoleExamples = templates.Examples(`
		# grant role to user
		kbcli cluster grant-role CLUSTERNAME --component COMPNAME --name USERNAME --role ROLENAME
		# grant role to user with default component
		kbcli cluster grant-role CLUSTERNAME --name USERNAME --role ROLENAME
		# grant role to user for instance
		kbcli cluster grant-role --instance INSTANCE --name USERNAME --role ROLENAME
	`)
	revokeRoleExamples = templates.Examples(`
		# revoke role from user
		kbcli cluster revoke-role CLUSTERNAME --component COMPNAME --name USERNAME --role ROLENAME
		# revoke role from user with default component
		kbcli cluster revoke-role CLUSTERNAME --name USERNAME --role ROLENAME
		# revoke role from user for instance
		kbcli cluster revoke-role --instance INSTANCE --name USERNAME --role ROLENAME
	`)
)

func NewCreateAccountCmd(f cmdutil.Factory, streams genericiooptions.IOStreams) *cobra.Command {
	o := accounts.NewCreateUserOptions(f, streams)
	cmd := &cobra.Command{
		Use:               "create-account",
		Short:             "Create account for a cluster",
		Example:           createUserExamples,
		ValidArgsFunction: util.ResourceNameCompletionFunc(f, types.ClusterGVR()),
		Run: func(cmd *cobra.Command, args []string) {
			cmdutil.CheckErr(o.Validate(args))
			cmdutil.CheckErr(o.Complete(f))
			cmdutil.CheckErr(o.Run(cmd, f, streams))
		},
	}
	o.AddFlags(cmd)
	return cmd
}

func NewDeleteAccountCmd(f cmdutil.Factory, streams genericiooptions.IOStreams) *cobra.Command {
	o := accounts.NewDeleteUserOptions(f, streams)
	cmd := &cobra.Command{
		Use:               "delete-account",
		Short:             "Delete account for a cluster",
		Example:           deleteUserExamples,
		ValidArgsFunction: util.ResourceNameCompletionFunc(f, types.ClusterGVR()),
		Run: func(cmd *cobra.Command, args []string) {
			cmdutil.CheckErr(o.Validate(args))
			cmdutil.CheckErr(o.Complete(f))
			cmdutil.CheckErr(o.Run(cmd, f, streams))
		},
	}
	o.AddFlags(cmd)
	cmd.Flags().BoolVar(&o.AutoApprove, "auto-approve", false, "Skip interactive approval before deleting account")
	return cmd
}

func NewDescAccountCmd(f cmdutil.Factory, streams genericiooptions.IOStreams) *cobra.Command {
	o := accounts.NewDescribeUserOptions(f, streams)
	cmd := &cobra.Command{
		Use:               "describe-account",
		Short:             "Describe account roles and related information",
		Example:           descUserExamples,
		ValidArgsFunction: util.ResourceNameCompletionFunc(f, types.ClusterGVR()),
		Run: func(cmd *cobra.Command, args []string) {
			cmdutil.CheckErr(o.Validate(args))
			cmdutil.CheckErr(o.Complete(f))
			cmdutil.CheckErr(o.Run(cmd, f, streams))
		},
	}
	o.AddFlags(cmd)
	return cmd
}

func NewListAccountsCmd(f cmdutil.Factory, streams genericiooptions.IOStreams) *cobra.Command {
	o := accounts.NewListUserOptions(f, streams)

	cmd := &cobra.Command{
		Use:               "list-accounts",
		Short:             "List accounts for a cluster",
		Aliases:           []string{"ls-accounts"},
		Example:           listUsersExample,
		ValidArgsFunction: util.ResourceNameCompletionFunc(f, types.ClusterGVR()),
		Run: func(cmd *cobra.Command, args []string) {
			cmdutil.CheckErr(o.Validate(args))
			cmdutil.CheckErr(o.Complete(f))
			cmdutil.CheckErr(o.Run(cmd, f, streams))
		},
	}
	o.AddFlags(cmd)
	return cmd
}

func NewGrantOptions(f cmdutil.Factory, streams genericiooptions.IOStreams) *cobra.Command {
	o := accounts.NewGrantOptions(f, streams)

	cmd := &cobra.Command{
		Use:               "grant-role",
		Short:             "Grant role to account",
		Aliases:           []string{"grant", "gr"},
		Example:           grantRoleExamples,
		ValidArgsFunction: util.ResourceNameCompletionFunc(f, types.ClusterGVR()),
		Run: func(cmd *cobra.Command, args []string) {
			cmdutil.CheckErr(o.Validate(args))
			cmdutil.CheckErr(o.Complete(f))
			cmdutil.CheckErr(o.Run(cmd, f, streams))
		},
	}
	o.AddFlags(cmd)
	return cmd
}

func NewRevokeOptions(f cmdutil.Factory, streams genericiooptions.IOStreams) *cobra.Command {
	o := accounts.NewRevokeOptions(f, streams)

	cmd := &cobra.Command{
		Use:               "revoke-role",
		Short:             "Revoke role from account",
		Aliases:           []string{"revoke", "rv"},
		Example:           revokeRoleExamples,
		ValidArgsFunction: util.ResourceNameCompletionFunc(f, types.ClusterGVR()),
		Run: func(cmd *cobra.Command, args []string) {
			cmdutil.CheckErr(o.Validate(args))
			cmdutil.CheckErr(o.Complete(f))
			cmdutil.CheckErr(o.Run(cmd, f, streams))
		},
	}
	o.AddFlags(cmd)
	return cmd
}