-
Notifications
You must be signed in to change notification settings - Fork 18
/
security_group_list.go
88 lines (68 loc) · 2.36 KB
/
security_group_list.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
package cmd
import (
"fmt"
"strings"
"github.com/spf13/cobra"
"github.com/exoscale/cli/pkg/account"
"github.com/exoscale/cli/pkg/globalstate"
"github.com/exoscale/cli/pkg/output"
exoapi "github.com/exoscale/egoscale/v2/api"
"github.com/exoscale/egoscale/v2/oapi"
)
type securityGroupListItemOutput struct {
ID string `json:"id"`
Name string `json:"name"`
Visibility string `json:"visibility"`
}
type securityGroupListOutput []securityGroupListItemOutput
func (o *securityGroupListOutput) ToJSON() { output.JSON(o) }
func (o *securityGroupListOutput) ToText() { output.Text(o) }
func (o *securityGroupListOutput) ToTable() { output.Table(o) }
type securityGroupListCmd struct {
cliCommandSettings `cli-cmd:"-"`
_ bool `cli-cmd:"list"`
Visibility string `cli-usage:"Security Group visibility: private (default) or public"`
}
func (c *securityGroupListCmd) cmdAliases() []string { return gListAlias }
func (c *securityGroupListCmd) cmdShort() string { return "List Security Groups" }
func (c *securityGroupListCmd) cmdLong() string {
return fmt.Sprintf(`This command lists Compute instance Security Groups.
Supported output template annotations: %s`,
strings.Join(output.TemplateAnnotations(&securityGroupListItemOutput{}), ", "))
}
func (c *securityGroupListCmd) cmdPreRun(cmd *cobra.Command, args []string) error {
return cliCommandDefaultPreRun(c, cmd, args)
}
func (c *securityGroupListCmd) cmdRun(_ *cobra.Command, _ []string) error {
ctx := exoapi.WithEndpoint(
gContext,
exoapi.NewReqEndpoint(account.CurrentAccount.Environment, account.CurrentAccount.DefaultZone),
)
params := &oapi.ListSecurityGroupsParams{}
if len(c.Visibility) > 0 {
params = &oapi.ListSecurityGroupsParams{
Visibility: (*oapi.ListSecurityGroupsParamsVisibility)(&c.Visibility),
}
}
securityGroups, err := globalstate.EgoscaleClient.FindSecurityGroups(ctx, account.CurrentAccount.DefaultZone, params)
if err != nil {
return err
}
out := make(securityGroupListOutput, 0)
for _, t := range securityGroups {
sg := securityGroupListItemOutput{Name: *t.Name}
if t.ID != nil {
sg.ID = *t.ID
sg.Visibility = "private"
} else {
sg.Visibility = "public"
}
out = append(out, sg)
}
return c.outputFunc(&out, nil)
}
func init() {
cobra.CheckErr(registerCLICommand(securityGroupCmd, &securityGroupListCmd{
cliCommandSettings: defaultCLICmdSettings(),
}))
}