feat: cds modules

Signed-off-by: Yvonnick Esnault <yvonnick@esnau.lt>

modules/cds/status/display.go

@@ -0,0 +1,121 @@
+package cdsstatus
+
+import (
+	"fmt"
+	"strings"
+
+	"github.com/ovh/cds/sdk"
+)
+
+func (widget *Widget) display() {
+	widget.TextWidget.Redraw(widget.content)
+}
+
+func (widget *Widget) content() (string, string, bool) {
+	if len(widget.View.GetHighlights()) > 0 {
+		widget.View.ScrollToHighlight()
+	} else {
+		widget.View.ScrollToBeginning()
+	}
+
+	widget.Items = make([]sdk.MonitoringStatusLine, 0)
+	str := widget.displayStatus()
+	title := widget.CommonSettings().Title
+	return title, str, false
+}
+
+func (widget *Widget) displayStatus() string {
+	status, err := widget.client.MonStatus()
+
+	widget.SetItemCount(len(status.Lines))
+
+	if err != nil || len(status.Lines) == 0 {
+		return fmt.Sprintf(" [red]Error: %v[white]\n", err.Error())
+	}
+
+	var (
+		global     []string
+		globalWarn []string
+		globalRed  []string
+		ok         []string
+		warn       []string
+		red        []string
+	)
+
+	for _, line := range status.Lines {
+		if line.Status == sdk.MonitoringStatusWarn && strings.Contains(line.Component, "Global") {
+			globalWarn = append(globalWarn, line.String())
+		} else if line.Status != sdk.MonitoringStatusOK && strings.Contains(line.Component, "Global") {
+			globalRed = append(globalRed, line.String())
+		} else if strings.Contains(line.Component, "Global") {
+			global = append(global, line.String())
+		} else if line.Status == sdk.MonitoringStatusWarn {
+			warn = append(warn, line.String())
+		} else if line.Status == sdk.MonitoringStatusOK {
+			ok = append(ok, line.String())
+		} else {
+			red = append(red, line.String())
+		}
+	}
+	var idx int
+	var content string
+	for _, v := range globalRed {
+		content += fmt.Sprintf("[grey][\"%d\"][red]%s\n", idx, v)
+		idx++
+	}
+	for _, v := range globalWarn {
+		content += fmt.Sprintf("[grey][\"%d\"][yellow]%s\n", idx, v)
+		idx++
+	}
+	for _, v := range global {
+		content += fmt.Sprintf("[grey][\"%d\"][grey]%s\n", idx, v)
+		idx++
+	}
+	for _, v := range red {
+		content += fmt.Sprintf("[grey][\"%d\"][red]%s\n", idx, v)
+		idx++
+	}
+	for _, v := range warn {
+		content += fmt.Sprintf("[grey][\"%d\"][yellow]%s\n", idx, v)
+		idx++
+	}
+	for _, v := range ok {
+		content += fmt.Sprintf("[grey][\"%d\"][grey]%s\n", idx, v)
+		idx++
+	}
+	return content
+}
+
+func getStatusColor(status string) string {
+	switch status {
+	case sdk.StatusSuccess:
+		return "green"
+	case sdk.StatusBuilding, sdk.StatusWaiting:
+		return "blue"
+	case sdk.StatusFail:
+		return "red"
+	case sdk.StatusStopped:
+		return "red"
+	case sdk.StatusSkipped:
+		return "grey"
+	case sdk.StatusDisabled:
+		return "grey"
+	}
+	return "red"
+}
+
+func pad(t string, size int) string {
+	if len(t) > size {
+		return t[0:size-3] + "..."
+	}
+	return t + strings.Repeat(" ", size-len(t))
+}
+
+func getVarsInPbj(key string, ps []sdk.Parameter) string {
+	for _, p := range ps {
+		if p.Name == key {
+			return p.Value
+		}
+	}
+	return ""
+}

modules/cds/status/keyboard.go

@@ -0,0 +1,18 @@
+package cdsstatus
+
+import (
+	"github.com/gdamore/tcell"
+)
+
+func (widget *Widget) initializeKeyboardControls() {
+	widget.InitializeCommonControls(widget.Refresh)
+
+	widget.SetKeyboardChar("j", widget.Next, "Select next line")
+	widget.SetKeyboardChar("k", widget.Prev, "Select previous line")
+	widget.SetKeyboardChar("o", widget.openWorkflow, "Open status in browser")
+
+	widget.SetKeyboardKey(tcell.KeyDown, widget.Next, "Select next line")
+	widget.SetKeyboardKey(tcell.KeyUp, widget.Prev, "Select previous line")
+	widget.SetKeyboardKey(tcell.KeyEnter, widget.openWorkflow, "Open status in browser")
+	widget.SetKeyboardKey(tcell.KeyEsc, widget.Unselect, "Clear selection")
+}

modules/cds/status/settings.go

@@ -0,0 +1,31 @@
+package cdsstatus
+
+import (
+	"os"
+
+	"github.com/olebedev/config"
+	"github.com/wtfutil/wtf/cfg"
+)
+
+const (
+	defaultFocusable = true
+	defaultTitle     = "CDS Status"
+)
+
+// Settings defines the configuration properties for this module
+type Settings struct {
+	common *cfg.Common
+	token  string `help:"Your CDS API token."`
+	apiURL string `help:"Your CDS API URL."`
+	uiURL  string
+}
+
+// NewSettingsFromYAML creates a new settings instance from a YAML config block
+func NewSettingsFromYAML(name string, ymlConfig *config.Config, globalConfig *config.Config) *Settings {
+	settings := Settings{
+		common: cfg.NewCommonSettingsFromModule(name, defaultTitle, defaultFocusable, ymlConfig, globalConfig),
+		token:  ymlConfig.UString("token", ymlConfig.UString("token", os.Getenv("CDS_TOKEN"))),
+		apiURL: ymlConfig.UString("apiURL", os.Getenv("CDS_API_URL")),
+	}
+	return &settings
+}

modules/cds/status/widget.go

@@ -0,0 +1,141 @@
+package cdsstatus
+
+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.KeyboardWidget
+	view.TextWidget
+
+	jobs    []sdk.WorkflowNodeJobRun
+	filters []string
+
+	client cdsclient.Interface
+
+	settings *Settings
+	Selected int
+	maxItems int
+	Items    []sdk.MonitoringStatusLine
+}
+
+// NewWidget creates a new instance of the widget
+func NewWidget(app *tview.Application, pages *tview.Pages, settings *Settings) *Widget {
+	widget := Widget{
+		KeyboardWidget:    view.NewKeyboardWidget(app, pages, settings.common),
+		MultiSourceWidget: view.NewMultiSourceWidget(settings.common, "workflow", "workflows"),
+		TextWidget:        view.NewTextWidget(app, settings.common),
+
+		settings: settings,
+	}
+
+	widget.initializeKeyboardControls()
+	widget.View.SetRegions(true)
+	widget.View.SetInputCapture(widget.InputCapture)
+	widget.SetDisplayFunction(widget.display)
+
+	widget.Unselect()
+	widget.filters = []string{sdk.StatusWaiting, sdk.StatusBuilding}
+
+	widget.KeyboardWidget.SetView(widget.View)
+
+	widget.client = cdsclient.New(cdsclient.Config{
+		Host:                              settings.apiURL,
+		BuitinConsumerAuthenticationToken: settings.token,
+	})
+
+	config, _ := widget.client.ConfigUser()
+
+	if config.URLUI != "" {
+		widget.settings.uiURL = config.URLUI
+	}
+
+	return &widget
+}
+
+/* -------------------- Exported Functions -------------------- */
+
+// SetItemCount sets the amount of line throughout the widgets display creation
+func (widget *Widget) SetItemCount(items int) {
+	widget.maxItems = items
+}
+
+// GetItemCount returns the amount of line 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()
+}
+
+// HelpText displays the widgets controls
+func (widget *Widget) HelpText() string {
+	return widget.KeyboardWidget.HelpText()
+}
+
+/* -------------------- Unexported Functions -------------------- */
+
+func (widget *Widget) currentFilter() string {
+	if len(widget.filters) == 0 {
+		return sdk.StatusWaiting
+	}
+
+	if widget.Idx < 0 || widget.Idx >= len(widget.filters) {
+		widget.Idx = 0
+		return sdk.StatusWaiting
+	}
+
+	return widget.filters[widget.Idx]
+}
+
+func (widget *Widget) openWorkflow() {
+	currentSelection := widget.View.GetHighlights()
+	if widget.Selected >= 0 && currentSelection[0] != "" {
+		url := fmt.Sprintf("%s/admin/services", widget.settings.uiURL)
+		utils.OpenFile(url)
+	}
+}