-
Notifications
You must be signed in to change notification settings - Fork 0
/
jq_tags_parser.go
79 lines (72 loc) · 1.96 KB
/
jq_tags_parser.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
package opslevel_jq_parser
import (
"encoding/json"
"strings"
"github.com/opslevel/opslevel-go/v2024"
)
type JQTagsParser struct {
creates []*JQFieldParser
assigns []*JQFieldParser
}
func NewJQTagsParser(cfg TagRegistrationConfig) *JQTagsParser {
creates := make([]*JQFieldParser, len(cfg.Create))
for i, expression := range cfg.Create {
creates[i] = NewJQFieldParser(expression)
}
assigns := make([]*JQFieldParser, len(cfg.Assign))
for i, expression := range cfg.Assign {
assigns[i] = NewJQFieldParser(expression)
}
return &JQTagsParser{
creates: creates,
assigns: assigns,
}
}
func (p *JQTagsParser) parse(programs []*JQFieldParser, data string) []opslevel.TagInput {
output := make([]opslevel.TagInput, 0, len(programs))
for _, program := range programs {
response, err := program.Run(data)
if err != nil {
// TODO: log error
continue
}
if response == "" {
continue
}
if strings.HasPrefix(response, "[") && strings.HasSuffix(response, "]") {
var tags []map[string]string
if err := json.Unmarshal([]byte(response), &tags); err != nil {
// TODO: log error
continue
}
for _, item := range tags {
for key, value := range item {
if key == "" || value == "" {
// TODO: log warning
continue
}
output = append(output, opslevel.TagInput{Key: key, Value: value})
}
}
}
if strings.HasPrefix(response, "{") && strings.HasSuffix(response, "}") {
var tags map[string]string
if err := json.Unmarshal([]byte(response), &tags); err != nil {
// TODO: log error
continue
}
for key, value := range tags {
if key == "" || value == "" {
// TODO: log warning
continue
}
output = append(output, opslevel.TagInput{Key: key, Value: value})
}
}
}
unique, _ := runJQUnique[opslevel.TagInput](output)
return unique
}
func (p *JQTagsParser) Run(data string) ([]opslevel.TagInput, []opslevel.TagInput, error) {
return p.parse(p.creates, data), p.parse(p.assigns, data), nil
}