mirror of
https://github.com/strongdm/comply
synced 2024-08-26 17:26:47 +00:00
180 lines
3.9 KiB
Go
180 lines
3.9 KiB
Go
package github
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"strconv"
|
|
"sync"
|
|
|
|
"github.com/google/go-github/github"
|
|
"github.com/pkg/errors"
|
|
"github.com/strongdm/comply/internal/model"
|
|
"golang.org/x/oauth2"
|
|
)
|
|
|
|
const (
|
|
cfgToken = "token"
|
|
cfgUsername = "username"
|
|
cfgRepo = "repo"
|
|
)
|
|
|
|
var prompts = map[string]string{
|
|
cfgToken: "GitHub Token",
|
|
cfgUsername: "GitHub Username",
|
|
cfgRepo: "GitHub Repository",
|
|
}
|
|
|
|
// Prompts are human-readable configuration element names
|
|
func (g *githubPlugin) Prompts() map[string]string {
|
|
return prompts
|
|
}
|
|
|
|
// Register causes the Github plugin to register itself
|
|
func Register() {
|
|
model.Register(model.Github, &githubPlugin{})
|
|
}
|
|
|
|
type githubPlugin struct {
|
|
token string
|
|
username string
|
|
reponame string
|
|
|
|
clientMu sync.Mutex
|
|
client *github.Client
|
|
}
|
|
|
|
func (g *githubPlugin) api() *github.Client {
|
|
g.clientMu.Lock()
|
|
defer g.clientMu.Unlock()
|
|
|
|
if g.client == nil {
|
|
ts := oauth2.StaticTokenSource(
|
|
&oauth2.Token{AccessToken: g.token},
|
|
)
|
|
|
|
// get go-github client
|
|
gh := github.NewClient(oauth2.NewClient(context.Background(), ts))
|
|
g.client = gh
|
|
}
|
|
return g.client
|
|
}
|
|
|
|
func (g *githubPlugin) Get(ID string) (*model.Ticket, error) {
|
|
return nil, nil
|
|
}
|
|
|
|
func (g *githubPlugin) Configure(cfg map[string]interface{}) error {
|
|
var err error
|
|
|
|
if g.token, err = getCfg(cfg, cfgToken); err != nil {
|
|
return err
|
|
}
|
|
if g.username, err = getCfg(cfg, cfgUsername); err != nil {
|
|
return err
|
|
}
|
|
if g.reponame, err = getCfg(cfg, cfgRepo); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func getCfg(cfg map[string]interface{}, k string) (string, error) {
|
|
v, ok := cfg[k]
|
|
if !ok {
|
|
return "", errors.New("Missing key: " + k)
|
|
}
|
|
|
|
vS, ok := v.(string)
|
|
if !ok {
|
|
return "", errors.New("Malformatted key: " + k)
|
|
}
|
|
return vS, nil
|
|
}
|
|
|
|
func (g *githubPlugin) FindOpen() ([]*model.Ticket, error) {
|
|
issues, _, err := g.api().Issues.ListByRepo(context.Background(), g.username, g.reponame, &github.IssueListByRepoOptions{
|
|
State: "open",
|
|
})
|
|
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "error during FindOpen")
|
|
}
|
|
|
|
return toTickets(issues), nil
|
|
}
|
|
|
|
func (g *githubPlugin) FindByTag(name, value string) ([]*model.Ticket, error) {
|
|
panic("not implemented")
|
|
}
|
|
|
|
func (g *githubPlugin) FindByTagName(name string) ([]*model.Ticket, error) {
|
|
issues, _, err := g.api().Issues.ListByRepo(context.Background(), g.username, g.reponame, &github.IssueListByRepoOptions{
|
|
State: "all",
|
|
Labels: []string{name},
|
|
})
|
|
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "error during FindOpen")
|
|
}
|
|
|
|
return toTickets(issues), nil
|
|
}
|
|
|
|
func (g *githubPlugin) LinkFor(t *model.Ticket) string {
|
|
return fmt.Sprintf("https://github.com/strongdm/comply/issues/%s", t.ID)
|
|
}
|
|
|
|
func (g *githubPlugin) Links() (string, string) {
|
|
return fmt.Sprintf("https://github.com/strongdm/comply/issues?q=is%3Aissue+is%3Aclosed+label%3Acomply", g.username, g.reponame),
|
|
fmt.Sprintf("https://github.com/%s/%s/issues?q=is%3Aissue+is%3Aopen+label%3Acomply", g.username, g.reponame)
|
|
}
|
|
|
|
func (g *githubPlugin) Create(ticket *model.Ticket, labels []string) error {
|
|
_, _, err := g.api().Issues.Create(context.Background(), g.username, g.reponame, &github.IssueRequest{
|
|
Title: &ticket.Name,
|
|
Body: &ticket.Body,
|
|
Labels: &labels,
|
|
})
|
|
return err
|
|
}
|
|
|
|
func toTickets(issues []*github.Issue) []*model.Ticket {
|
|
var tickets []*model.Ticket
|
|
for _, i := range issues {
|
|
tickets = append(tickets, toTicket(i))
|
|
}
|
|
return tickets
|
|
}
|
|
|
|
func toTicket(i *github.Issue) *model.Ticket {
|
|
t := &model.Ticket{Attributes: make(map[string]interface{})}
|
|
t.ID = strconv.Itoa(*i.Number)
|
|
t.Name = ss(i.Title)
|
|
t.Body = ss(i.Body)
|
|
t.CreatedAt = i.CreatedAt
|
|
t.State = toState(ss(i.State))
|
|
|
|
for _, l := range i.Labels {
|
|
if l.Name != nil {
|
|
t.SetBool(*l.Name)
|
|
}
|
|
}
|
|
return t
|
|
}
|
|
|
|
func toState(state string) model.TicketState {
|
|
switch state {
|
|
case "closed":
|
|
return model.Closed
|
|
}
|
|
return model.Open
|
|
}
|
|
|
|
func ss(s *string) string {
|
|
if s == nil {
|
|
return ""
|
|
}
|
|
return *s
|
|
}
|