mirror of
https://github.com/taigrr/wtf
synced 2025-01-18 04:03:14 -08:00
14e7619075
Have all instances take a function Update the remaining modules to take this into account Numerous smaller refactors to make some widgets work more or less the same
107 lines
2.5 KiB
Go
107 lines
2.5 KiB
Go
package datadog
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/rivo/tview"
|
|
"github.com/wtfutil/wtf/utils"
|
|
"github.com/wtfutil/wtf/view"
|
|
datadog "github.com/zorkian/go-datadog-api"
|
|
)
|
|
|
|
type Widget struct {
|
|
view.KeyboardWidget
|
|
view.ScrollableWidget
|
|
|
|
monitors []datadog.Monitor
|
|
settings *Settings
|
|
}
|
|
|
|
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() {
|
|
monitors, monitorErr := widget.Monitors()
|
|
|
|
if monitorErr != nil {
|
|
widget.monitors = nil
|
|
widget.SetItemCount(0)
|
|
widget.Redraw(func() (string, string, bool) { return widget.CommonSettings().Title, monitorErr.Error(), true })
|
|
return
|
|
}
|
|
triggeredMonitors := []datadog.Monitor{}
|
|
|
|
for _, monitor := range monitors {
|
|
state := *monitor.OverallState
|
|
if state == "Alert" {
|
|
triggeredMonitors = append(triggeredMonitors, monitor)
|
|
}
|
|
}
|
|
widget.monitors = triggeredMonitors
|
|
widget.SetItemCount(len(widget.monitors))
|
|
|
|
widget.Render()
|
|
}
|
|
|
|
func (widget *Widget) Render() {
|
|
widget.Redraw(widget.content)
|
|
}
|
|
|
|
func (widget *Widget) HelpText() string {
|
|
return widget.KeyboardWidget.HelpText()
|
|
}
|
|
|
|
/* -------------------- Unexported Functions -------------------- */
|
|
|
|
func (widget *Widget) content() (string, string, bool) {
|
|
triggeredMonitors := widget.monitors
|
|
var str string
|
|
|
|
if len(triggeredMonitors) > 0 {
|
|
str += fmt.Sprintf(
|
|
" %s\n",
|
|
"[red]Triggered Monitors[white]",
|
|
)
|
|
for idx, triggeredMonitor := range triggeredMonitors {
|
|
row := fmt.Sprintf(`[%s][red] %s[%s]`,
|
|
widget.RowColor(idx),
|
|
*triggeredMonitor.Name,
|
|
widget.RowColor(idx),
|
|
)
|
|
str += utils.HighlightableHelper(widget.View, row, idx, len(*triggeredMonitor.Name))
|
|
}
|
|
} else {
|
|
str += fmt.Sprintf(
|
|
" %s\n",
|
|
"[green]No Triggered Monitors[white]",
|
|
)
|
|
}
|
|
|
|
return widget.CommonSettings().Title, str, false
|
|
}
|
|
|
|
func (widget *Widget) openItem() {
|
|
|
|
sel := widget.GetSelected()
|
|
if sel >= 0 && widget.monitors != nil && sel < len(widget.monitors) {
|
|
item := &widget.monitors[sel]
|
|
utils.OpenFile(fmt.Sprintf("https://app.datadoghq.com/monitors/%d?q=*", *item.Id))
|
|
}
|
|
}
|