initial commit
This commit is contained in:
commit
c215536745
5 changed files with 298 additions and 0 deletions
62
cmd/acb/main.go
Normal file
62
cmd/acb/main.go
Normal file
|
|
@ -0,0 +1,62 @@
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"os"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"o5r.ca/autocrossbow/contributions"
|
||||||
|
"o5r.ca/autocrossbow/issues"
|
||||||
|
)
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
proj := os.Args[1]
|
||||||
|
ghusername := os.Args[2]
|
||||||
|
start := os.Args[3]
|
||||||
|
end := os.Args[4]
|
||||||
|
DoPrs(proj, ghusername, start, end)
|
||||||
|
DoJira(start, end)
|
||||||
|
}
|
||||||
|
|
||||||
|
func DoPrs(proj, ghusername, start, end string) {
|
||||||
|
|
||||||
|
prs, err := contributions.GetPRs(proj, ghusername, start, end)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println(fmt.Errorf("error getting PRs: %w", err))
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
|
||||||
|
ghf, err := os.Create(fmt.Sprintf("gh-%s-%s-%s-%s-%d.json", proj, ghusername, start, end, time.Now().Unix()))
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println(fmt.Errorf("error creating PR file: %w", err))
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
enc := json.NewEncoder(ghf)
|
||||||
|
err = enc.Encode(prs)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println(fmt.Errorf("error writing out PRs: %w", err))
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
func DoJira(start, end string) {
|
||||||
|
host := os.Getenv("JIRA_HOST")
|
||||||
|
user := os.Getenv("JIRA_TARGET_USER")
|
||||||
|
prs, err := issues.GetIssues(host, user, start, end)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println(fmt.Errorf("error getting PRs: %w", err))
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
|
||||||
|
ghf, err := os.Create(fmt.Sprintf("jira-%s-%s-%s-%s-%d.json", host, user, start, end, time.Now().Unix()))
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println(fmt.Errorf("error creating PR file: %w", err))
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
enc := json.NewEncoder(ghf)
|
||||||
|
err = enc.Encode(prs)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println(fmt.Errorf("error writing out PRs: %w", err))
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
}
|
||||||
66
cmd/acb/summarize.go
Normal file
66
cmd/acb/summarize.go
Normal file
|
|
@ -0,0 +1,66 @@
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"net/http"
|
||||||
|
"os"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
// SummarizeData takes GitHub PRs and Jira issues data and sends it to an OpenAI-compatible endpoint for summarization.
|
||||||
|
func SummarizeData(prs map[string][]PullRequest, issues []Issue) (string, error) {
|
||||||
|
// Build a prompt string
|
||||||
|
prompt := "Summarize the following GitHub PRs and Jira issues:\n\n"
|
||||||
|
for repo, prList := range prs {
|
||||||
|
prompt += fmt.Sprintf("Repository: %s\n", repo)
|
||||||
|
for _, pr := range prList {
|
||||||
|
prompt += fmt.Sprintf("- Title: %s\n", pr.Title)
|
||||||
|
prompt += fmt.Sprintf(" Body: %s\n", pr.Body)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
for _, issue := range issues {
|
||||||
|
prompt += fmt.Sprintf("\nJira Issue: %s\n", issue.Key)
|
||||||
|
prompt += fmt.Sprintf("Summary: %s\n", issue.Summary)
|
||||||
|
prompt += fmt.Sprintf("Description: %s\n", issue.Description)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Get OpenAI endpoint and token from environment variables
|
||||||
|
openaiEndpoint := os.Getenv("OPENAI_ENDPOINT")
|
||||||
|
openaiToken := os.Getenv("OPENAI_TOKEN")
|
||||||
|
|
||||||
|
if openaiEndpoint == "" || openaiToken == "" {
|
||||||
|
return "", fmt.Errorf("OPENAI_ENDPOINT and OPENAI_TOKEN must be set in environment variables")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Create a JSON payload for the OpenAI API
|
||||||
|
payload := map[string]string{"prompt": prompt}
|
||||||
|
jsonPayload, err := json.Marshal(payload)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
|
||||||
|
// Create a POST request to the OpenAI endpoint with JSON body
|
||||||
|
req, err := http.NewRequest("POST", openaiEndpoint, bytes.NewBuffer(jsonPayload))
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
req.Header.Set("Content-Type", "application/json")
|
||||||
|
req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", openaiToken))
|
||||||
|
|
||||||
|
client := &http.Client{}
|
||||||
|
resp, err := client.Do(req)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
defer resp.Body.Close()
|
||||||
|
|
||||||
|
body, err := io.ReadAll(resp.Body)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
|
||||||
|
return string(body), nil
|
||||||
|
}
|
||||||
80
contributions/gh.go
Normal file
80
contributions/gh.go
Normal file
|
|
@ -0,0 +1,80 @@
|
||||||
|
package contributions
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"net/http"
|
||||||
|
"os"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
var ghc = http.Client{}
|
||||||
|
var gh_token = os.Getenv("GH_TOKEN")
|
||||||
|
|
||||||
|
type SearchResponse struct {
|
||||||
|
Items []PRResp
|
||||||
|
}
|
||||||
|
|
||||||
|
type PRResp struct {
|
||||||
|
Body string
|
||||||
|
Title string
|
||||||
|
HtmlUrl string `json:"html_url"`
|
||||||
|
ClosedAt string `json:"closed_at"`
|
||||||
|
RepositoryUrl string `json:"repository_url"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type PullRequest struct {
|
||||||
|
Body string
|
||||||
|
Title string
|
||||||
|
ClosedAt string
|
||||||
|
URL string
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetPRs(org, username string, from string, to string) (map[string][]PullRequest, error) {
|
||||||
|
req, err := http.NewRequest(
|
||||||
|
http.MethodGet,
|
||||||
|
fmt.Sprintf(
|
||||||
|
"https://api.github.com/search/issues?q=type:pr+org:%s+author:%s+is:closed+merged:%s..%s",
|
||||||
|
org,
|
||||||
|
username,
|
||||||
|
from,
|
||||||
|
to,
|
||||||
|
),
|
||||||
|
nil)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("error making request: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
req.Header.Add("Authorization", fmt.Sprintf("Bearer %s", gh_token))
|
||||||
|
resp, err := ghc.Do(req)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("error talking to github: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if resp.StatusCode > 400 {
|
||||||
|
b := bytes.NewBuffer(nil)
|
||||||
|
io.Copy(b, resp.Body)
|
||||||
|
return nil, fmt.Errorf("error talking to github: %s", b.String())
|
||||||
|
}
|
||||||
|
r := &SearchResponse{}
|
||||||
|
dec := json.NewDecoder(resp.Body)
|
||||||
|
err = dec.Decode(&r)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("error decoding response: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
allthings := map[string][]PullRequest{}
|
||||||
|
|
||||||
|
for _, pr := range r.Items {
|
||||||
|
reponameparts := strings.Split(pr.RepositoryUrl, "/")
|
||||||
|
reponame := reponameparts[len(reponameparts)-1]
|
||||||
|
if prs, ok := allthings[reponame]; !ok {
|
||||||
|
allthings[reponame] = []PullRequest{{pr.Body, pr.Title, pr.ClosedAt, pr.HtmlUrl}}
|
||||||
|
} else {
|
||||||
|
allthings[reponame] = append(prs, PullRequest{pr.Body, pr.Title, pr.ClosedAt, pr.HtmlUrl})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return allthings, nil
|
||||||
|
}
|
||||||
3
go.mod
Normal file
3
go.mod
Normal file
|
|
@ -0,0 +1,3 @@
|
||||||
|
module o5r.ca/autocrossbow
|
||||||
|
|
||||||
|
go 1.25.0
|
||||||
87
issues/jira.go
Normal file
87
issues/jira.go
Normal file
|
|
@ -0,0 +1,87 @@
|
||||||
|
package issues
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"net/http"
|
||||||
|
"os"
|
||||||
|
)
|
||||||
|
|
||||||
|
//curl -u olivier@circleci.com:"$(cat ~/jiratoken)" 'https://circleci.atlassian.net/rest/api/2/search?jql=sprint+in+openSprints()+and+project=CIAM'
|
||||||
|
|
||||||
|
var jcl = http.Client{}
|
||||||
|
|
||||||
|
type Issue struct {
|
||||||
|
Summary string
|
||||||
|
Description string
|
||||||
|
Comments []Comment
|
||||||
|
}
|
||||||
|
|
||||||
|
type Comment struct {
|
||||||
|
Author Author
|
||||||
|
Body string
|
||||||
|
}
|
||||||
|
|
||||||
|
type Author struct {
|
||||||
|
DisplayName string
|
||||||
|
EmailAddress string
|
||||||
|
}
|
||||||
|
|
||||||
|
type JiraSearchResp struct {
|
||||||
|
Issues []RespIssue
|
||||||
|
}
|
||||||
|
|
||||||
|
type RespIssue struct {
|
||||||
|
Key string
|
||||||
|
Fields map[string]any
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetIssues(instance, user, from, to string) ([]Issue, error) {
|
||||||
|
req, err := http.NewRequest(http.MethodGet, fmt.Sprintf("https://%s/rest/api/2/search", instance), nil)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("error building jira search request: %w", err)
|
||||||
|
}
|
||||||
|
q := req.URL.Query()
|
||||||
|
q.Add("jql", fmt.Sprintf("assignee was '%s' and resolved >= %s and resolved <= %s", user, from, to))
|
||||||
|
req.URL.RawQuery = q.Encode()
|
||||||
|
req.SetBasicAuth(os.Getenv("JIRA_USER"), os.Getenv("JIRA_TOKEN"))
|
||||||
|
resp, err := jcl.Do(req)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("error executing jira search request: %w", err)
|
||||||
|
}
|
||||||
|
if resp.StatusCode >= 400 {
|
||||||
|
b := bytes.NewBuffer(nil)
|
||||||
|
io.Copy(b, resp.Body)
|
||||||
|
return nil, fmt.Errorf("error talking to jira: %s", b.String())
|
||||||
|
}
|
||||||
|
|
||||||
|
dec := json.NewDecoder(resp.Body)
|
||||||
|
var jsr *JiraSearchResp
|
||||||
|
dec.Decode(&jsr)
|
||||||
|
|
||||||
|
out := []Issue{}
|
||||||
|
|
||||||
|
for _, i := range jsr.Issues {
|
||||||
|
iss := Issue{}
|
||||||
|
if s, ok := i.Fields["summary"]; ok {
|
||||||
|
iss.Summary, _ = s.(string)
|
||||||
|
}
|
||||||
|
if d, ok := i.Fields["description"]; ok {
|
||||||
|
iss.Description, _ = d.(string)
|
||||||
|
}
|
||||||
|
if comms, ok := i.Fields["comment"].(map[string]any); ok {
|
||||||
|
if c := comms["comments"].([]map[string]any); len(c) > 0 {
|
||||||
|
iss.Comments = make([]Comment, 0, len(c))
|
||||||
|
for _, c2 := range c {
|
||||||
|
iss.Comments = append(iss.Comments, Comment{Author{DisplayName: c2["displayName"].(string), EmailAddress: c2["emailAddress"].(string)}, c2["body"].(string)})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
out = append(out, iss)
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
return out, nil
|
||||||
|
}
|
||||||
Loading…
Add table
Add a link
Reference in a new issue