107 lines
2.4 KiB
Go
107 lines
2.4 KiB
Go
package api
|
|
|
|
import (
|
|
"reflect"
|
|
"strings"
|
|
)
|
|
|
|
func (issue *Issue) ExportData(fields []string) *map[string]interface{} {
|
|
v := reflect.ValueOf(issue).Elem()
|
|
data := map[string]interface{}{}
|
|
|
|
for _, f := range fields {
|
|
switch f {
|
|
case "milestone":
|
|
if issue.Milestone.Title != "" {
|
|
data[f] = &issue.Milestone
|
|
} else {
|
|
data[f] = nil
|
|
}
|
|
case "comments":
|
|
data[f] = issue.Comments.Nodes
|
|
case "assignees":
|
|
data[f] = issue.Assignees.Nodes
|
|
case "labels":
|
|
data[f] = issue.Labels.Nodes
|
|
case "projectCards":
|
|
data[f] = issue.ProjectCards.Nodes
|
|
default:
|
|
sf := fieldByName(v, f)
|
|
data[f] = sf.Interface()
|
|
}
|
|
}
|
|
|
|
return &data
|
|
}
|
|
|
|
func (pr *PullRequest) ExportData(fields []string) *map[string]interface{} {
|
|
v := reflect.ValueOf(pr).Elem()
|
|
data := map[string]interface{}{}
|
|
|
|
for _, f := range fields {
|
|
switch f {
|
|
case "headRepository":
|
|
data[f] = map[string]string{"name": pr.HeadRepository.Name}
|
|
case "milestone":
|
|
if pr.Milestone.Title != "" {
|
|
data[f] = &pr.Milestone
|
|
} else {
|
|
data[f] = nil
|
|
}
|
|
case "statusCheckRollup":
|
|
if n := pr.Commits.Nodes; len(n) > 0 {
|
|
data[f] = n[0].Commit.StatusCheckRollup.Contexts.Nodes
|
|
} else {
|
|
data[f] = nil
|
|
}
|
|
case "comments":
|
|
data[f] = pr.Comments.Nodes
|
|
case "assignees":
|
|
data[f] = pr.Assignees.Nodes
|
|
case "labels":
|
|
data[f] = pr.Labels.Nodes
|
|
case "projectCards":
|
|
data[f] = pr.ProjectCards.Nodes
|
|
case "reviews":
|
|
data[f] = pr.Reviews.Nodes
|
|
case "files":
|
|
data[f] = pr.Files.Nodes
|
|
case "reviewRequests":
|
|
requests := make([]interface{}, 0, len(pr.ReviewRequests.Nodes))
|
|
for _, req := range pr.ReviewRequests.Nodes {
|
|
if req.RequestedReviewer.TypeName == "" {
|
|
continue
|
|
}
|
|
requests = append(requests, req.RequestedReviewer)
|
|
}
|
|
data[f] = &requests
|
|
default:
|
|
sf := fieldByName(v, f)
|
|
data[f] = sf.Interface()
|
|
}
|
|
}
|
|
|
|
return &data
|
|
}
|
|
|
|
func ExportIssues(issues []Issue, fields []string) *[]interface{} {
|
|
data := make([]interface{}, len(issues))
|
|
for i, issue := range issues {
|
|
data[i] = issue.ExportData(fields)
|
|
}
|
|
return &data
|
|
}
|
|
|
|
func ExportPRs(prs []PullRequest, fields []string) *[]interface{} {
|
|
data := make([]interface{}, len(prs))
|
|
for i, pr := range prs {
|
|
data[i] = pr.ExportData(fields)
|
|
}
|
|
return &data
|
|
}
|
|
|
|
func fieldByName(v reflect.Value, field string) reflect.Value {
|
|
return v.FieldByNameFunc(func(s string) bool {
|
|
return strings.EqualFold(field, s)
|
|
})
|
|
}
|