-
Notifications
You must be signed in to change notification settings - Fork 0
/
mx_get.go
98 lines (83 loc) · 3.11 KB
/
mx_get.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
89
90
91
92
93
94
95
96
97
98
package main
import (
"fmt"
"strings"
. "github.com/dirtman/sitepkg"
)
// Implement the "get" command.
func getMX(invokedAs []string) error {
var input *UserInput
var states StatesMX = make(StatesMX)
var err, result error
var duo, ref bool
SetStringOpt("view", "V", true, "any", "Specify the the view to which the record belongs")
SetStringOpt("fields", "F", true, "", "Specify fields to be used in the search")
SetStringOpt("rFields", "R", true, "", "Specify additional fields to show in verbose mode")
SetStringOpt("filename", "f", true, "", "Specify an input file")
SetStringOpt("mx", "m", false, "", "Specify the MX of the record to fetch")
SetStringOpt("preference", "p", false, "", "Specify the preference of the record to fetch")
SetBoolOpt("ref", "r", true, false, "Show only the object \"reference\" of each fetched object")
if input, err = subCommandInit(invokedAs[1], invokedAs[2], duo); err != nil {
return Error("failure initializing program and getting user input: %v", err)
}
// The preference field is returned by default, so this is not really needed.
if inList, _ := InList(input.rFields, "preference"); !inList {
input.rFields = append(input.rFields, "preference")
}
if err = getStates(states, input.ndList, input.fields, input.rFields, false, false); err != nil {
return Error("failure getting states: %v", err)
} else if ref, err = GetBoolOpt("ref"); err != nil {
return Error("failure getting ref option: %v", err)
}
if errors := checkStateErrors(states, false, true); len(errors) > 0 {
return Error("Aborting process; no records fetched.")
}
// Loop through the user provided input (name/data) list.
space := input.maxNameLength + 8
var numNotFound uint
for _, nameData := range input.ndList {
records := states[nameData].records
request := strings.TrimLeft(nameData, nameDataSep)
request = strings.TrimRight(request, nameDataSep)
// I prefer to keep the output short, only showing the user-specified name/data
// fields. But if the user provided no name or data, let's show the fields.
if request == "" {
request = strings.Join(input.fields, ",")
}
if err := states[nameData].err; err != nil {
Print("%-*s FAILED: %v\n", space, "MX("+request+")", err)
result = Error("one or more errors occurred")
} else if len(records) == 0 {
if !Quiet {
Print("%-*s NOTFOUND\n", space, "MX("+request+")")
}
numNotFound++
}
for _, record := range records {
if err = states[nameData].err; err != nil {
Print("Failure getting %s: %v\n", request, err)
} else if ref {
Print("%-*s %s\n", space, "MX("+request+"): ", record.Ref)
} else {
data := fmt.Sprintf("%s %d", record.MX, record.Preference)
sep := " ("
end := ""
if input.view != "default" {
data += fmt.Sprintf("%s%s view", sep, record.View)
sep = ", "
end = ")"
}
if record.Disable {
data += fmt.Sprintf("%s%s", sep, "DISABLED")
end = ")"
}
data += end
Print("%-*s %s %s\n", space, "MX("+request+"): ", record.Fqdn, data)
}
}
}
if result == nil && numNotFound != 0 {
return Error("One or more records not found.")
}
return result
}