Skip to content

Commit 3484f23

Browse files
authored
add ip type, source and subnet size to firewall rule printer (#234)
1 parent fdf7937 commit 3484f23

File tree

1 file changed

+17
-5
lines changed

1 file changed

+17
-5
lines changed

cmd/printer/firewallRule.go

Lines changed: 17 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,24 +1,36 @@
11
package printer
22

33
import (
4+
"fmt"
5+
46
"github.com/vultr/govultr/v2"
57
)
68

79
func FirewallRules(fwr []govultr.FirewallRule, meta *govultr.Meta) {
8-
col := columns{"RULE NUMBER", "ACTION", "PROTOCOL", "PORT", "NETWORK", "NOTES"}
10+
col := columns{"RULE NUMBER", "ACTION", "TYPE", "PROTOCOL", "PORT", "NETWORK", "SOURCE", "NOTES"}
911
display(col)
1012
for _, f := range fwr {
11-
display(columns{f.ID, f.Action, f.Protocol, f.Port, f.Subnet, f.Notes})
13+
display(columns{f.ID, f.Action, f.Type, f.Protocol, f.Port, getFirewallNetwork(f.Subnet, f.SubnetSize), getFirewallSource(f.Source), f.Notes})
1214
}
1315

1416
Meta(meta)
1517
flush()
1618
}
1719

1820
func FirewallRule(fwr *govultr.FirewallRule) {
19-
col := columns{"RULE NUMBER", "ACTION", "PROTOCOL", "PORT", "NETWORK", "NOTES"}
21+
col := columns{"RULE NUMBER", "ACTION", "TYPE", "PROTOCOL", "PORT", "NETWORK", "SOURCE", "NOTES"}
2022
display(col)
21-
22-
display(columns{fwr.ID, fwr.Action, fwr.Protocol, fwr.Port, fwr.Subnet, fwr.Notes})
23+
display(columns{fwr.ID, fwr.Action, fwr.Type, fwr.Protocol, fwr.Port, getFirewallNetwork(fwr.Subnet, fwr.SubnetSize), getFirewallSource(fwr.Source), fwr.Notes})
2324
flush()
2425
}
26+
27+
func getFirewallSource(source string) string {
28+
if source == "" {
29+
return "anywhere"
30+
}
31+
return source
32+
}
33+
34+
func getFirewallNetwork(subnet string, size int) string {
35+
return fmt.Sprintf("%s/%d", subnet, size)
36+
}

0 commit comments

Comments
 (0)