autocrossbow/cmd/acb/main.go
Olivier Tremblay ebd45dfff5 feat: add Vikunja handler to main file
Co-authored-by: aider (openai/qwen2.5-coder:32b-instruct-q4_0) <aider@aider.chat>
2025-10-30 14:55:15 -04:00

101 lines
2.5 KiB
Go

package main
import (
"encoding/json"
"fmt"
"os"
"time"
"o5r.ca/autocrossbow/contributions"
"o5r.ca/autocrossbow/issues/vikunja"
"o5r.ca/autocrossbow/issues"
)
func main() {
proj := os.Args[1]
ghusername := os.Args[2]
start := os.Args[3]
end := os.Args[4]
employeename := os.Args[5]
prs := DoPrs(proj, ghusername, start, end)
issues := DoJira(start, end)
vikunjaTasks := DoVikunja(start, end)
summ, err := SummarizeData(employeename, prs, issues, vikunjaTasks)
if err != nil {
fmt.Println(fmt.Errorf("error getting PRs: %w", err))
os.Exit(1)
}
fmt.Println(summ)
}
func DoPrs(proj, ghusername, start, end string) map[string][]contributions.PullRequest {
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)
}
return prs
}
func DoJira(start, end string) []issues.Issue {
host := os.Getenv("JIRA_HOST")
user := os.Getenv("JIRA_TARGET_USER")
issues, err := issues.GetIssues(host, user, start, end)
if err != nil {
fmt.Println(fmt.Errorf("error getting PRs: %w", err))
os.Exit(1)
}
jf, 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 JIRA file: %w", err))
os.Exit(1)
}
defer jf.Close()
enc := json.NewEncoder(jf)
err = enc.Encode(issues)
if err != nil {
fmt.Println(fmt.Errorf("error writing out JIRA issues: %w", err))
os.Exit(1)
}
return issues
}
func DoVikunja(start, end string) []vikunja.Task {
host := os.Getenv("VIKUNJA_HOST")
user := os.Getenv("VIKUNJA_USER")
tasks, err := vikunja.GetTasks(host, user, start, end)
if err != nil {
fmt.Println(fmt.Errorf("error getting Vikunja tasks: %w", err))
os.Exit(1)
}
vf, err := os.Create(fmt.Sprintf("vikunja-%s-%s-%s-%s-%d.json", host, user, start, end, time.Now().Unix()))
if err != nil {
fmt.Println(fmt.Errorf("error creating Vikunja file: %w", err))
os.Exit(1)
}
defer vf.Close()
enc := json.NewEncoder(vf)
err = enc.Encode(tasks)
if err != nil {
fmt.Println(fmt.Errorf("error writing out Vikunja tasks: %w", err))
os.Exit(1)
}
return tasks
}