mirror of
https://github.com/taigrr/wtf
synced 2025-01-18 04:03:14 -08:00
Add gerrit widget
This commit is contained in:
Anand Sudhir Prayaga
73
gerrit/display.go
Normal file
73
gerrit/display.go
Normal file
@@ -0,0 +1,73 @@
|
||||
package gerrit
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
|
||||
"github.com/senorprogrammer/wtf/wtf"
|
||||
)
|
||||
|
||||
func (widget *Widget) display() {
|
||||
|
||||
project := widget.currentGerritProject()
|
||||
if project == nil {
|
||||
fmt.Fprintf(widget.View, "%s", " Gerrit project data is unavailable (1)")
|
||||
return
|
||||
}
|
||||
|
||||
widget.View.SetTitle(fmt.Sprintf("%s- %s", widget.Name, widget.title(project)))
|
||||
|
||||
str := wtf.SigilStr(len(widget.GerritProjects), widget.Idx, widget.View) + "\n"
|
||||
str = str + " [red]Stats[white]\n"
|
||||
str = str + widget.displayStats(project)
|
||||
str = str + "\n"
|
||||
str = str + " [red]Open Incoming Reviews[white]\n"
|
||||
str = str + widget.displayMyIncomingReviews(project, wtf.Config.UString("wtf.mods.gerrit.username"))
|
||||
str = str + "\n"
|
||||
str = str + " [red]My Outgoing Reviews[white]\n"
|
||||
str = str + widget.displayMyOutgoingReviews(project, wtf.Config.UString("wtf.mods.gerrit.username"))
|
||||
|
||||
widget.View.SetText(str)
|
||||
}
|
||||
|
||||
func (widget *Widget) displayMyOutgoingReviews(project *GerritProject, username string) string {
|
||||
ors := project.myOutgoingReviews(username)
|
||||
|
||||
if len(ors) == 0 {
|
||||
return " [grey]none[white]\n"
|
||||
}
|
||||
|
||||
str := ""
|
||||
for _, r := range ors {
|
||||
str = str + fmt.Sprintf(" [green]%4s[white] %s\n", r.ChangeID, r.Subject)
|
||||
}
|
||||
|
||||
return str
|
||||
}
|
||||
|
||||
func (widget *Widget) displayMyIncomingReviews(project *GerritProject, username string) string {
|
||||
irs := project.myIncomingReviews(username)
|
||||
|
||||
if len(irs) == 0 {
|
||||
return " [grey]none[white]\n"
|
||||
}
|
||||
|
||||
str := ""
|
||||
for _, r := range irs {
|
||||
str = str + fmt.Sprintf(" [green]%4s[white] %s\n", r.ChangeID, r.Subject)
|
||||
}
|
||||
|
||||
return str
|
||||
}
|
||||
|
||||
func (widget *Widget) displayStats(project *GerritProject) string {
|
||||
str := fmt.Sprintf(
|
||||
" Reviews: %d\n",
|
||||
project.ReviewCount(),
|
||||
)
|
||||
|
||||
return str
|
||||
}
|
||||
|
||||
func (widget *Widget) title(project *GerritProject) string {
|
||||
return fmt.Sprintf("[green]%s [white]", project.Path)
|
||||
}
|
||||
100
gerrit/gerrit_repo.go
Normal file
100
gerrit/gerrit_repo.go
Normal file
@@ -0,0 +1,100 @@
|
||||
package gerrit
|
||||
|
||||
import (
|
||||
glb "github.com/andygrunwald/go-gerrit"
|
||||
)
|
||||
|
||||
type GerritProject struct {
|
||||
gerrit *glb.Client
|
||||
Path string
|
||||
|
||||
Changes *[]glb.ChangeInfo
|
||||
}
|
||||
|
||||
func NewGerritProject(path string, gerrit *glb.Client) *GerritProject {
|
||||
project := GerritProject{
|
||||
gerrit: gerrit,
|
||||
Path: path,
|
||||
}
|
||||
|
||||
return &project
|
||||
}
|
||||
|
||||
// Refresh reloads the gerrit data via the Gerrit API
|
||||
func (project *GerritProject) Refresh() {
|
||||
project.Changes, _ = project.loadChanges()
|
||||
}
|
||||
|
||||
/* -------------------- Counts -------------------- */
|
||||
|
||||
func (project *GerritProject) IssueCount() int {
|
||||
if project.Changes == nil {
|
||||
return 0
|
||||
}
|
||||
|
||||
return len(*project.Changes)
|
||||
}
|
||||
|
||||
func (project *GerritProject) ReviewCount() int {
|
||||
if project.Changes == nil {
|
||||
return 0
|
||||
}
|
||||
|
||||
return len(*project.Changes)
|
||||
}
|
||||
|
||||
/* -------------------- Unexported Functions -------------------- */
|
||||
|
||||
// myOutgoingReviews returns a list of my outgoing reviews created by username on this project
|
||||
func (project *GerritProject) myOutgoingReviews(username string) []glb.ChangeInfo {
|
||||
changes := []glb.ChangeInfo{}
|
||||
|
||||
if project.Changes == nil {
|
||||
return changes
|
||||
}
|
||||
|
||||
for _, change := range *project.Changes {
|
||||
user := change.Owner
|
||||
|
||||
if user.Username == username {
|
||||
changes = append(changes, change)
|
||||
}
|
||||
}
|
||||
|
||||
return changes
|
||||
}
|
||||
|
||||
// myIncomingReviews returns a list of merge requests for which username has been requested to ChangeInfo
|
||||
func (project *GerritProject) myIncomingReviews(username string) []glb.ChangeInfo {
|
||||
changes := []glb.ChangeInfo{}
|
||||
|
||||
if project.Changes == nil {
|
||||
return changes
|
||||
}
|
||||
|
||||
for _, change := range *project.Changes {
|
||||
reviewers := change.Reviewers
|
||||
|
||||
for _, reviewer := range reviewers["REVIEWER"] {
|
||||
if reviewer.Username == username {
|
||||
changes = append(changes, change)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return changes
|
||||
}
|
||||
|
||||
func (project *GerritProject) loadChanges() (*[]glb.ChangeInfo, error) {
|
||||
opt := &glb.QueryChangeOptions{}
|
||||
opt.Query = []string{"(projects:" + project.Path + "+ is:open + owner:self) " + " OR " +
|
||||
"(projects:" + project.Path + " + is:open + ((reviewer:self + -owner:self + -star:ignore) + OR + assignee:self))"}
|
||||
opt.AdditionalFields = []string{"DETAILED_LABELS", "DETAILED_ACCOUNTS"}
|
||||
changes, _, err := project.gerrit.Changes.QueryChanges(opt)
|
||||
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return changes, err
|
||||
}
|
||||
180
gerrit/widget.go
Normal file
180
gerrit/widget.go
Normal file
@@ -0,0 +1,180 @@
|
||||
package gerrit
|
||||
|
||||
import (
|
||||
"crypto/tls"
|
||||
"fmt"
|
||||
glb "github.com/andygrunwald/go-gerrit"
|
||||
"github.com/gdamore/tcell"
|
||||
"github.com/rivo/tview"
|
||||
"github.com/senorprogrammer/wtf/wtf"
|
||||
"net/http"
|
||||
"os"
|
||||
"regexp"
|
||||
)
|
||||
|
||||
const HelpText = `
|
||||
Keyboard commands for Gerrit:
|
||||
|
||||
/: Show/hide this help window
|
||||
h: Previous project
|
||||
l: Next project
|
||||
r: Refresh the data
|
||||
|
||||
arrow left: Previous project
|
||||
arrow right: Next project
|
||||
`
|
||||
|
||||
type Widget struct {
|
||||
wtf.TextWidget
|
||||
|
||||
app *tview.Application
|
||||
pages *tview.Pages
|
||||
|
||||
gerrit *glb.Client
|
||||
|
||||
GerritProjects []*GerritProject
|
||||
Idx int
|
||||
}
|
||||
|
||||
var (
|
||||
GerritURLPattern = regexp.MustCompile(`^(http|https)://(.*)$`)
|
||||
)
|
||||
|
||||
func NewWidget(app *tview.Application, pages *tview.Pages) *Widget {
|
||||
baseURL := wtf.Config.UString("wtf.mods.gerrit.domain")
|
||||
username := wtf.Config.UString("wtf.mods.gerrit.username")
|
||||
password := os.Getenv("WTF_GERRIT_PASSWORD")
|
||||
verifyServerCertificate := wtf.Config.UBool("wtf.mods.gerrit.verifyServerCertificate", true)
|
||||
|
||||
httpClient := &http.Client{Transport: &http.Transport{
|
||||
TLSClientConfig: &tls.Config{
|
||||
InsecureSkipVerify: !verifyServerCertificate,
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
gerritUrl := baseURL
|
||||
submatches := GerritURLPattern.FindAllStringSubmatch(baseURL, -1)
|
||||
|
||||
if len(submatches) > 0 && len(submatches[0]) > 2 {
|
||||
submatch := submatches[0]
|
||||
gerritUrl = fmt.Sprintf(
|
||||
"%s://%s:%s@%s", submatch[1], username, password, submatch[2])
|
||||
}
|
||||
|
||||
gerrit, err := glb.NewClient(gerritUrl, httpClient)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
widget := Widget{
|
||||
TextWidget: wtf.NewTextWidget(" Gerrit ", "gerrit", true),
|
||||
|
||||
app: app,
|
||||
pages: pages,
|
||||
|
||||
gerrit: gerrit,
|
||||
|
||||
Idx: 0,
|
||||
}
|
||||
|
||||
widget.GerritProjects = widget.buildProjectCollection(wtf.Config.UList("wtf.mods.gerrit.projects"))
|
||||
|
||||
widget.View.SetInputCapture(widget.keyboardIntercept)
|
||||
|
||||
return &widget
|
||||
}
|
||||
|
||||
/* -------------------- Exported Functions -------------------- */
|
||||
|
||||
func (widget *Widget) Refresh() {
|
||||
for _, project := range widget.GerritProjects {
|
||||
project.Refresh()
|
||||
}
|
||||
|
||||
widget.UpdateRefreshedAt()
|
||||
widget.display()
|
||||
}
|
||||
|
||||
func (widget *Widget) Next() {
|
||||
widget.Idx = widget.Idx + 1
|
||||
if widget.Idx == len(widget.GerritProjects) {
|
||||
widget.Idx = 0
|
||||
}
|
||||
|
||||
widget.display()
|
||||
}
|
||||
|
||||
func (widget *Widget) Prev() {
|
||||
widget.Idx = widget.Idx - 1
|
||||
if widget.Idx < 0 {
|
||||
widget.Idx = len(widget.GerritProjects) - 1
|
||||
}
|
||||
|
||||
widget.display()
|
||||
}
|
||||
|
||||
/* -------------------- Unexported Functions -------------------- */
|
||||
|
||||
func (widget *Widget) buildProjectCollection(projectData []interface{}) []*GerritProject {
|
||||
gerritProjects := []*GerritProject{}
|
||||
|
||||
for _, name := range projectData {
|
||||
project := NewGerritProject(name.(string), widget.gerrit)
|
||||
gerritProjects = append(gerritProjects, project)
|
||||
}
|
||||
|
||||
return gerritProjects
|
||||
}
|
||||
|
||||
func (widget *Widget) currentGerritProject() *GerritProject {
|
||||
if len(widget.GerritProjects) == 0 {
|
||||
return nil
|
||||
}
|
||||
|
||||
if widget.Idx < 0 || widget.Idx >= len(widget.GerritProjects) {
|
||||
return nil
|
||||
}
|
||||
|
||||
return widget.GerritProjects[widget.Idx]
|
||||
}
|
||||
|
||||
func (widget *Widget) keyboardIntercept(event *tcell.EventKey) *tcell.EventKey {
|
||||
switch string(event.Rune()) {
|
||||
case "/":
|
||||
widget.showHelp()
|
||||
return nil
|
||||
case "h":
|
||||
widget.Prev()
|
||||
return nil
|
||||
case "l":
|
||||
widget.Next()
|
||||
return nil
|
||||
case "r":
|
||||
widget.Refresh()
|
||||
return nil
|
||||
}
|
||||
|
||||
switch event.Key() {
|
||||
case tcell.KeyLeft:
|
||||
widget.Prev()
|
||||
return nil
|
||||
case tcell.KeyRight:
|
||||
widget.Next()
|
||||
return nil
|
||||
default:
|
||||
return event
|
||||
}
|
||||
}
|
||||
|
||||
func (widget *Widget) showHelp() {
|
||||
closeFunc := func() {
|
||||
widget.pages.RemovePage("help")
|
||||
widget.app.SetFocus(widget.View)
|
||||
}
|
||||
|
||||
modal := wtf.NewBillboardModal(HelpText, closeFunc)
|
||||
|
||||
widget.pages.AddPage("help", modal, false, true)
|
||||
widget.app.SetFocus(modal)
|
||||
}
|
||||
Reference in New Issue
Block a user