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
|
package plaintext
import (
"encoding/json"
"log"
"slices"
"strings"
"github.com/Loyalsoldier/geoip/lib"
)
const (
TypeTextOut = "text"
DescTextOut = "Convert data to plaintext CIDR format"
)
func init() {
lib.RegisterOutputConfigCreator(TypeTextOut, func(action lib.Action, data json.RawMessage) (lib.OutputConverter, error) {
return NewTextOutFromBytes(TypeTextOut, DescTextOut, action, data)
})
lib.RegisterOutputConverter(TypeTextOut, &textOut{
Description: DescTextOut,
})
}
func (t *textOut) GetType() string {
return t.Type
}
func (t *textOut) GetAction() lib.Action {
return t.Action
}
func (t *textOut) GetDescription() string {
return t.Description
}
func (t *textOut) Output(container lib.Container) error {
for _, name := range t.filterAndSortList(container) {
entry, found := container.GetEntry(name)
if !found {
log.Printf("❌ entry %s not found\n", name)
continue
}
data, err := t.marshalBytes(entry)
if err != nil {
return err
}
filename := strings.ToLower(entry.GetName()) + t.OutputExt
if err := t.writeFile(filename, data); err != nil {
return err
}
}
return nil
}
func (t *textOut) filterAndSortList(container lib.Container) []string {
excludeMap := make(map[string]bool)
for _, exclude := range t.Exclude {
if exclude = strings.ToUpper(strings.TrimSpace(exclude)); exclude != "" {
excludeMap[exclude] = true
}
}
wantList := make([]string, 0, len(t.Want))
for _, want := range t.Want {
if want = strings.ToUpper(strings.TrimSpace(want)); want != "" && !excludeMap[want] {
wantList = append(wantList, want)
}
}
if len(wantList) > 0 {
// Sort the list
slices.Sort(wantList)
return wantList
}
list := make([]string, 0, 300)
for entry := range container.Loop() {
name := entry.GetName()
if excludeMap[name] {
continue
}
list = append(list, name)
}
// Sort the list
slices.Sort(list)
return list
}
|
