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 pastVerb string

var aliases []string

var short string

switch sdkStatus {

case codersdk.UserStatusActive:

verb = "activate"

pastVerb = "activated"

aliases = []string{"active"}

short = "Update a user's status to 'active'. Active users can fully interact with the platform"

case codersdk.UserStatusSuspended:

verb = "suspend"

pastVerb = "suspended"

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: formatExamples(

example{

Command: 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

table, err := cliui.DisplayTable([]codersdk.User{user}, "", columns)

if err != nil {

return xerrors.Errorf("render user table: %w", err)

}

_, _ = fmt.Fprintln(cmd.OutOrStdout(), table)

// 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: cliui.ConfirmYes,

})

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)

}

_, _ = fmt.Fprintf(cmd.OutOrStdout(), "\nUser %s has been %s!\n", cliui.Styles.Keyword.Render(user.Username), pastVerb)

return nil

},

}

cmd.Flags().StringArrayVarP(&columns, "column", "c", []string{"username", "email", "created_at", "status"},

"Specify a column to filter in the table.")

return cmd

}