package users

import (
	"github.com/databricks/cli/libs/cmdio"
	"github.com/databricks/databricks-sdk-go/service/iam"
	"github.com/spf13/cobra"
)

func listOverride(listCmd *cobra.Command, listReq *iam.ListAccountUsersRequest) {
	listReq.Attributes = "id,userName,groups,active"
	listCmd.Annotations["template"] = cmdio.Heredoc(`
	{{range .}}{{.Id|green}}	{{.UserName}}	{{range .Groups}}{{.Display}} {{end}}	{{if .Active}}{{"ACTIVE"|green}}{{else}}DISABLED{{end}}
	{{end}}`)
}

func init() {
	listOverrides = append(listOverrides, listOverride)
}