mirror of
https://github.com/taigrr/wtf
synced 2025-01-18 04:03:14 -08:00
136 lines
3.2 KiB
Go
136 lines
3.2 KiB
Go
package github
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/google/go-github/v26/github"
|
|
)
|
|
|
|
func (widget *Widget) display() {
|
|
widget.TextWidget.Redraw(widget.content)
|
|
}
|
|
|
|
func (widget *Widget) content() (string, string, bool) {
|
|
repo := widget.currentGithubRepo()
|
|
|
|
title := fmt.Sprintf("%s - %s", widget.CommonSettings().Title, widget.title(repo))
|
|
if repo == nil {
|
|
return title, " GitHub repo data is unavailable ", false
|
|
}
|
|
|
|
_, _, width, _ := widget.View.GetRect()
|
|
str := widget.settings.common.SigilStr(len(widget.GithubRepos), widget.Idx, width) + "\n"
|
|
str += " [red]Stats[white]\n"
|
|
str += widget.displayStats(repo)
|
|
str += "\n [red]Open Review Requests[white]\n"
|
|
str += widget.displayMyReviewRequests(repo, widget.settings.username)
|
|
str += "\n [red]My Pull Requests[white]\n"
|
|
str += widget.displayMyPullRequests(repo, widget.settings.username)
|
|
for _, customQuery := range widget.settings.customQueries {
|
|
str += fmt.Sprintf("\n [red]%s[white]\n", customQuery.title)
|
|
str += widget.displayCustomQuery(repo, customQuery.filter, customQuery.perPage)
|
|
}
|
|
|
|
return title, str, false
|
|
}
|
|
|
|
func (widget *Widget) displayMyPullRequests(repo *GithubRepo, username string) string {
|
|
prs := repo.myPullRequests(username, widget.settings.enableStatus)
|
|
|
|
prLength := len(prs)
|
|
|
|
if prLength == 0 {
|
|
return " [grey]none[white]\n"
|
|
}
|
|
|
|
widget.SetItemCount(prLength)
|
|
|
|
str := ""
|
|
for idx, pr := range prs {
|
|
str += fmt.Sprintf(`[green]["%d"]%4d[""][white] %s`, idx, *pr.Number, *pr.Title)
|
|
str += "\n"
|
|
}
|
|
|
|
return str
|
|
}
|
|
|
|
func (widget *Widget) displayCustomQuery(repo *GithubRepo, filter string, perPage int) string {
|
|
res := repo.customIssueQuery(filter, perPage)
|
|
|
|
if res == nil {
|
|
return " [grey]Invalid Query[white]\n"
|
|
}
|
|
|
|
issuesLength := len(res.Issues)
|
|
|
|
if issuesLength == 0 {
|
|
return " [grey]none[white]\n"
|
|
}
|
|
|
|
maxItems := widget.GetItemCount()
|
|
|
|
str := ""
|
|
for idx, issue := range res.Issues {
|
|
str += fmt.Sprintf(`[green]["%d"]%4d[""][white] %s`, maxItems + idx , *issue.Number, *issue.Title)
|
|
str += "\n"
|
|
}
|
|
|
|
widget.SetItemCount(maxItems + issuesLength)
|
|
|
|
return str
|
|
}
|
|
|
|
func (widget *Widget) displayMyReviewRequests(repo *GithubRepo, username string) string {
|
|
prs := repo.myReviewRequests(username)
|
|
|
|
prLength := len(prs)
|
|
|
|
if prLength == 0 {
|
|
return " [grey]none[white]\n"
|
|
}
|
|
|
|
maxItems := widget.GetItemCount()
|
|
|
|
str := ""
|
|
for idx, pr := range prs {
|
|
str += fmt.Sprintf(`[green]["%d"]%4d[""][white] %s`, maxItems + idx, *pr.Number, *pr.Title)
|
|
str += "\n"
|
|
}
|
|
|
|
widget.SetItemCount(maxItems + prLength)
|
|
|
|
return str
|
|
}
|
|
|
|
func (widget *Widget) displayStats(repo *GithubRepo) string {
|
|
str := fmt.Sprintf(
|
|
" PRs: %d Issues: %d Stars: %d\n",
|
|
repo.PullRequestCount(),
|
|
repo.IssueCount(),
|
|
repo.StarCount(),
|
|
)
|
|
|
|
return str
|
|
}
|
|
|
|
func (widget *Widget) title(repo *GithubRepo) string {
|
|
return fmt.Sprintf("[green]%s - %s[white]", repo.Owner, repo.Name)
|
|
}
|
|
|
|
var mergeIcons = map[string]string{
|
|
"dirty": "[red]![white] ",
|
|
"clean": "[green]✔[white] ",
|
|
"unstable": "[red]✖[white] ",
|
|
"blocked": "[red]✖[white] ",
|
|
}
|
|
|
|
func (widget *Widget) mergeString(pr *github.PullRequest) string {
|
|
if !widget.settings.enableStatus {
|
|
return ""
|
|
}
|
|
if str, ok := mergeIcons[pr.GetMergeableState()]; ok {
|
|
return str
|
|
}
|
|
return "? "
|
|
}
|