mirror of
https://github.com/coder/coder.git
synced 2026-09-24 15:04:27 +08:00
feat: Add suspend/active user to cli (#1422)
* feat: Add suspend/active user to cli * UserID is now a string and allows for username too
This commit is contained in:
@@ -0,0 +1,85 @@
|
||||
package cli
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
|
||||
"github.com/spf13/cobra"
|
||||
"golang.org/x/xerrors"
|
||||
|
||||
"github.com/coder/coder/cli/cliui"
|
||||
"github.com/coder/coder/codersdk"
|
||||
)
|
||||
|
||||
// createUserStatusCommand sets a user status.
|
||||
func createUserStatusCommand(sdkStatus codersdk.UserStatus) *cobra.Command {
|
||||
var verb string
|
||||
var aliases []string
|
||||
var short string
|
||||
switch sdkStatus {
|
||||
case codersdk.UserStatusActive:
|
||||
verb = "activate"
|
||||
aliases = []string{"active"}
|
||||
short = "Update a user's status to 'active'. Active users can fully interact with the platform"
|
||||
case codersdk.UserStatusSuspended:
|
||||
verb = "suspend"
|
||||
aliases = []string{"rm", "delete"}
|
||||
short = "Update a user's status to 'suspended'. A suspended user cannot log into the platform"
|
||||
default:
|
||||
panic(fmt.Sprintf("%s is not supported", sdkStatus))
|
||||
}
|
||||
|
||||
var (
|
||||
columns []string
|
||||
)
|
||||
cmd := &cobra.Command{
|
||||
Use: fmt.Sprintf("%s <username|user_id>", verb),
|
||||
Short: short,
|
||||
Args: cobra.ExactArgs(1),
|
||||
Aliases: aliases,
|
||||
Example: fmt.Sprintf("coder users %s example_user", verb),
|
||||
RunE: func(cmd *cobra.Command, args []string) error {
|
||||
client, err := createClient(cmd)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
identifier := args[0]
|
||||
if identifier == "" {
|
||||
return xerrors.Errorf("user identifier cannot be an empty string")
|
||||
}
|
||||
|
||||
user, err := client.User(cmd.Context(), identifier)
|
||||
if err != nil {
|
||||
return xerrors.Errorf("fetch user: %w", err)
|
||||
}
|
||||
|
||||
// Display the user
|
||||
_, _ = fmt.Fprintln(cmd.OutOrStdout(), displayUsers(columns, user))
|
||||
|
||||
// User status is already set to this
|
||||
if user.Status == sdkStatus {
|
||||
_, _ = fmt.Fprintf(cmd.OutOrStdout(), "User status is already %q\n", sdkStatus)
|
||||
return nil
|
||||
}
|
||||
|
||||
// Prompt to confirm the action
|
||||
_, err = cliui.Prompt(cmd, cliui.PromptOptions{
|
||||
Text: fmt.Sprintf("Are you sure you want to %s this user?", verb),
|
||||
IsConfirm: true,
|
||||
Default: "yes",
|
||||
})
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
_, err = client.UpdateUserStatus(cmd.Context(), user.ID.String(), sdkStatus)
|
||||
if err != nil {
|
||||
return xerrors.Errorf("%s user: %w", verb, err)
|
||||
}
|
||||
return nil
|
||||
},
|
||||
}
|
||||
cmd.Flags().StringArrayVarP(&columns, "column", "c", []string{"username", "email", "created_at", "status"},
|
||||
"Specify a column to filter in the table.")
|
||||
return cmd
|
||||
}
|
||||
Reference in New Issue
Block a user