-
Notifications
You must be signed in to change notification settings - Fork 0
/
jq_repository_parser.go
86 lines (79 loc) · 2.52 KB
/
jq_repository_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
80
81
82
83
84
85
86
package opslevel_jq_parser
import (
"encoding/json"
"strings"
"github.com/opslevel/opslevel-go/v2024"
"github.com/rs/zerolog/log"
)
type RepositoryDTO struct {
Name string
Directory string
Repo string
}
func (r *RepositoryDTO) Convert() opslevel.ServiceRepositoryCreateInput {
return opslevel.ServiceRepositoryCreateInput{
Repository: *opslevel.NewIdentifier(r.Repo),
BaseDirectory: opslevel.RefOf(r.Directory),
DisplayName: opslevel.RefOf(r.Name),
}
}
type JQRepositoryParser struct {
programs []*JQFieldParser
}
func NewJQRepositoryParser(expressions []string) *JQRepositoryParser {
programs := make([]*JQFieldParser, len(expressions))
for i, expression := range expressions {
programs[i] = NewJQFieldParser(expression)
}
return &JQRepositoryParser{
programs: programs,
}
}
func (p *JQRepositoryParser) Run(data string) ([]opslevel.ServiceRepositoryCreateInput, error) {
output := make([]opslevel.ServiceRepositoryCreateInput, 0)
for _, program := range p.programs {
response, err := program.Run(data)
// log.Warn().Msgf("expression: %s\nresponse: %s", program.program.Program, response)
if err != nil {
log.Warn().Msgf("unable to parse alias from expression: %s", program.program.Program)
return nil, err
}
if response == "" {
continue
}
if strings.HasPrefix(response, "[") && strings.HasSuffix(response, "]") {
if response == "[]" {
continue
}
var repos []RepositoryDTO
if err := json.Unmarshal([]byte(response), &repos); err == nil {
for _, repo := range repos {
if repo.Repo != "" {
output = append(output, repo.Convert())
}
}
} else {
// Try as []string
var repoNames []string
if err := json.Unmarshal([]byte(response), &repoNames); err == nil {
for _, repoName := range repoNames {
output = append(output, opslevel.ServiceRepositoryCreateInput{Repository: *opslevel.NewIdentifier(repoName)})
}
}
}
} else if strings.HasPrefix(response, "{") && strings.HasSuffix(response, "}") {
var repo RepositoryDTO
if err := json.Unmarshal([]byte(response), &repo); err != nil {
// TODO: log error
log.Warn().Err(err).Msgf("unable to marshal repo expression: %s\n%s", program.program.Program, response)
continue
}
if repo.Repo != "" {
output = append(output, repo.Convert())
}
} else {
output = append(output, opslevel.ServiceRepositoryCreateInput{Repository: *opslevel.NewIdentifier(response)})
}
}
return runJQUnique[opslevel.ServiceRepositoryCreateInput](output, ".repository.alias")
}