1
0
mirror of https://github.com/taigrr/wtf synced 2025-01-18 04:03:14 -08:00
wtf/modules/jenkins/widget.go
2019-08-05 10:50:12 -07:00

112 lines
2.3 KiB
Go

package jenkins
import (
"fmt"
"regexp"
"github.com/rivo/tview"
"github.com/wtfutil/wtf/utils"
"github.com/wtfutil/wtf/view"
)
type Widget struct {
view.KeyboardWidget
view.ScrollableWidget
settings *Settings
view *View
}
func NewWidget(app *tview.Application, pages *tview.Pages, settings *Settings) *Widget {
widget := Widget{
KeyboardWidget: view.NewKeyboardWidget(app, pages, settings.common),
ScrollableWidget: view.NewScrollableWidget(app, settings.common, true),
settings: settings,
}
widget.SetRenderFunction(widget.Render)
widget.initializeKeyboardControls()
widget.View.SetInputCapture(widget.InputCapture)
widget.KeyboardWidget.SetView(widget.View)
return &widget
}
/* -------------------- Exported Functions -------------------- */
func (widget *Widget) Refresh() {
if widget.Disabled() {
return
}
view, err := widget.Create(
widget.settings.url,
widget.settings.user,
widget.settings.apiKey,
)
widget.view = view
if err != nil {
widget.Redraw(widget.CommonSettings().Title, err.Error(), true)
return
}
widget.SetItemCount(len(widget.view.Jobs))
widget.Render()
}
/* -------------------- Unexported Functions -------------------- */
func (widget *Widget) Render() {
if widget.view == nil {
return
}
title := fmt.Sprintf("%s: [red]%s", widget.CommonSettings().Title, widget.view.Name)
widget.Redraw(title, widget.contentFrom(widget.view), false)
}
func (widget *Widget) contentFrom(view *View) string {
var str string
for idx, job := range view.Jobs {
var validID = regexp.MustCompile(widget.settings.jobNameRegex)
if validID.MatchString(job.Name) {
row := fmt.Sprintf(
`[%s] [%s]%-6s[white]`,
widget.RowColor(idx),
widget.jobColor(&job),
job.Name,
)
str += utils.HighlightableHelper(widget.View, row, idx, len(job.Name))
}
}
return str
}
func (widget *Widget) jobColor(job *Job) string {
switch job.Color {
case "blue":
// Override color if successBallColor boolean param provided in config
return widget.settings.successBallColor
case "red":
return "red"
default:
return "white"
}
}
func (widget *Widget) openJob() {
sel := widget.GetSelected()
if sel >= 0 && widget.view != nil && sel < len(widget.view.Jobs) {
job := &widget.view.Jobs[sel]
utils.OpenFile(job.Url)
}
}