mirror of
https://github.com/taigrr/wtf
synced 2025-01-18 04:03:14 -08:00
144 lines
3.5 KiB
Go
144 lines
3.5 KiB
Go
package cdsfavorites
|
|
|
|
import (
|
|
"fmt"
|
|
"strconv"
|
|
|
|
"github.com/ovh/cds/sdk"
|
|
"github.com/ovh/cds/sdk/cdsclient"
|
|
"github.com/rivo/tview"
|
|
"github.com/wtfutil/wtf/utils"
|
|
"github.com/wtfutil/wtf/view"
|
|
)
|
|
|
|
// Widget define wtf widget to register widget later
|
|
type Widget struct {
|
|
view.MultiSourceWidget
|
|
view.TextWidget
|
|
|
|
workflows []sdk.Workflow
|
|
|
|
client cdsclient.Interface
|
|
|
|
settings *Settings
|
|
Selected int
|
|
maxItems int
|
|
Items []int64
|
|
}
|
|
|
|
// NewWidget creates a new instance of the widget
|
|
func NewWidget(app *tview.Application, pages *tview.Pages, settings *Settings) *Widget {
|
|
widget := Widget{
|
|
MultiSourceWidget: view.NewMultiSourceWidget(settings.common, "workflow", "workflows"),
|
|
TextWidget: view.NewTextWidget(app, pages, settings.common),
|
|
|
|
settings: settings,
|
|
}
|
|
|
|
widget.initializeKeyboardControls()
|
|
widget.View.SetRegions(true)
|
|
widget.SetDisplayFunction(widget.display)
|
|
|
|
widget.Unselect()
|
|
|
|
widget.client = cdsclient.New(cdsclient.Config{
|
|
Host: settings.apiURL,
|
|
BuitinConsumerAuthenticationToken: settings.token,
|
|
})
|
|
|
|
config, _ := widget.client.ConfigUser()
|
|
|
|
if config.URLUI != "" {
|
|
widget.settings.uiURL = config.URLUI
|
|
}
|
|
|
|
widget.workflows = widget.buildWorkflowsCollection()
|
|
return &widget
|
|
}
|
|
|
|
/* -------------------- Exported Functions -------------------- */
|
|
|
|
// SetItemCount sets the amount of workflows throughout the widgets display creation
|
|
func (widget *Widget) SetItemCount(items int) {
|
|
widget.maxItems = items
|
|
}
|
|
|
|
// GetItemCount returns the amount of workflows calculated so far as an int
|
|
func (widget *Widget) GetItemCount() int {
|
|
return widget.maxItems
|
|
}
|
|
|
|
// GetSelected returns the index of the currently highlighted item as an int
|
|
func (widget *Widget) GetSelected() int {
|
|
if widget.Selected < 0 {
|
|
return 0
|
|
}
|
|
return widget.Selected
|
|
}
|
|
|
|
// Next cycles the currently highlighted text down
|
|
func (widget *Widget) Next() {
|
|
widget.Selected++
|
|
if widget.Selected >= widget.maxItems {
|
|
widget.Selected = 0
|
|
}
|
|
widget.View.Highlight(strconv.Itoa(widget.Selected)).ScrollToHighlight()
|
|
}
|
|
|
|
// Prev cycles the currently highlighted text up
|
|
func (widget *Widget) Prev() {
|
|
widget.Selected--
|
|
if widget.Selected < 0 {
|
|
widget.Selected = widget.maxItems - 1
|
|
}
|
|
widget.View.Highlight(strconv.Itoa(widget.Selected)).ScrollToHighlight()
|
|
}
|
|
|
|
// Unselect stops highlighting the text and jumps the scroll position to the top
|
|
func (widget *Widget) Unselect() {
|
|
widget.Selected = -1
|
|
widget.View.Highlight()
|
|
widget.View.ScrollToBeginning()
|
|
}
|
|
|
|
// Refresh reloads the data
|
|
func (widget *Widget) Refresh() {
|
|
widget.display()
|
|
}
|
|
|
|
/* -------------------- Unexported Functions -------------------- */
|
|
|
|
func (widget *Widget) buildWorkflowsCollection() []sdk.Workflow {
|
|
workflows := []sdk.Workflow{}
|
|
data, _ := widget.client.Navbar()
|
|
for _, v := range data {
|
|
if v.Favorite && v.WorkflowName != "" {
|
|
workflows = append(workflows, sdk.Workflow{ProjectKey: v.Key, Name: v.WorkflowName})
|
|
}
|
|
}
|
|
return workflows
|
|
}
|
|
|
|
func (widget *Widget) currentCDSWorkflow() *sdk.Workflow {
|
|
if len(widget.workflows) == 0 {
|
|
return nil
|
|
}
|
|
|
|
if widget.Idx < 0 || widget.Idx >= len(widget.workflows) {
|
|
widget.Idx = 0
|
|
}
|
|
|
|
p := widget.workflows[widget.Idx]
|
|
return &p
|
|
}
|
|
|
|
func (widget *Widget) openWorkflow() {
|
|
currentSelection := widget.View.GetHighlights()
|
|
if widget.Selected >= 0 && currentSelection[0] != "" {
|
|
wf := widget.currentCDSWorkflow()
|
|
url := fmt.Sprintf("%s/project/%s/workflow/%s/run/%d",
|
|
widget.settings.uiURL, wf.ProjectKey, wf.Name, widget.Items[widget.Selected])
|
|
utils.OpenFile(url)
|
|
}
|
|
}
|