|
1 | 1 | package printer
|
2 | 2 |
|
3 | 3 | import (
|
| 4 | + "fmt" |
| 5 | + |
4 | 6 | "github.com/vultr/govultr/v2"
|
5 | 7 | )
|
6 | 8 |
|
7 | 9 | 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"} |
9 | 11 | display(col)
|
10 | 12 | 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}) |
12 | 14 | }
|
13 | 15 |
|
14 | 16 | Meta(meta)
|
15 | 17 | flush()
|
16 | 18 | }
|
17 | 19 |
|
18 | 20 | 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"} |
20 | 22 | 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}) |
23 | 24 | flush()
|
24 | 25 | }
|
| 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