mirror of
https://gitlab.com/psuapp/psu.git
synced 2024-08-30 18:12:34 +00:00
82 lines
2.3 KiB
Go
82 lines
2.3 KiB
Go
package cmd
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"os"
|
|
"text/template"
|
|
|
|
portainer "github.com/portainer/portainer/api"
|
|
|
|
"github.com/greenled/portainer-stack-utils/common"
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/spf13/viper"
|
|
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
// endpointGroupListCmd represents the endpoint group list command
|
|
var endpointGroupListCmd = &cobra.Command{
|
|
Use: "list",
|
|
Short: "List endpoint groups",
|
|
Aliases: []string{"ls"},
|
|
Example: ` Print endpoint groups in a table format:
|
|
psu endpoint group ls
|
|
|
|
Print endpoint groups name and description:
|
|
psu endpoint group ls --format "{{ .Name }}: {{ .Description }}"`,
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
client, err := common.GetClient()
|
|
common.CheckError(err)
|
|
|
|
logrus.Debug("Getting endpoint groups")
|
|
endpointGroups, err := client.GetEndpointGroups()
|
|
common.CheckError(err)
|
|
|
|
switch viper.GetString("endpoint.group.list.format") {
|
|
case "table":
|
|
// Print endpoint groups in a table format
|
|
writer, err := common.NewTabWriter([]string{
|
|
"ID",
|
|
"NAME",
|
|
"DESCRIPTION",
|
|
})
|
|
common.CheckError(err)
|
|
for _, g := range endpointGroups {
|
|
_, err := fmt.Fprintln(writer, fmt.Sprintf(
|
|
"%v\t%s\t%s",
|
|
g.ID,
|
|
g.Name,
|
|
g.Description,
|
|
))
|
|
common.CheckError(err)
|
|
}
|
|
flushErr := writer.Flush()
|
|
common.CheckError(flushErr)
|
|
case "json":
|
|
// Print endpoint groups in a json format
|
|
statusJsonBytes, err := json.Marshal(endpointGroups)
|
|
common.CheckError(err)
|
|
fmt.Println(string(statusJsonBytes))
|
|
default:
|
|
// Print endpoint groups in a custom format
|
|
template, templateParsingErr := template.New("endpointGroupTpl").Parse(viper.GetString("endpoint.group.list.format"))
|
|
common.CheckError(templateParsingErr)
|
|
for _, g := range endpointGroups {
|
|
templateExecutionErr := template.Execute(os.Stdout, g)
|
|
common.CheckError(templateExecutionErr)
|
|
fmt.Println()
|
|
}
|
|
}
|
|
},
|
|
}
|
|
|
|
func init() {
|
|
endpointGroupCmd.AddCommand(endpointGroupListCmd)
|
|
|
|
endpointGroupListCmd.Flags().String("format", "table", `Output format. Can be "table", "json" or a Go template.`)
|
|
viper.BindPFlag("endpoint.group.list.format", endpointGroupListCmd.Flags().Lookup("format"))
|
|
|
|
endpointGroupListCmd.SetUsageTemplate(endpointGroupListCmd.UsageTemplate() + common.GetFormatHelp(portainer.EndpointGroup{}))
|
|
}
|