wtf/modules/datadog/widget.go

package datadog

import (
	"fmt"

	"github.com/rivo/tview"
	"github.com/wtfutil/wtf/wtf"
	datadog "github.com/zorkian/go-datadog-api"
)

type Widget struct {
	wtf.TextWidget

	app      *tview.Application
	settings *Settings
}

func NewWidget(app *tview.Application, settings *Settings) *Widget {
	widget := Widget{
		TextWidget: wtf.NewTextWidget(app, settings.common, false),

		app:      app,
		settings: settings,
	}

	return &widget
}

/* -------------------- Exported Functions -------------------- */

func (widget *Widget) Refresh() {
	monitors, monitorErr := widget.Monitors()

	var content string
	if monitorErr != nil {
		widget.View.SetWrap(true)
		content = monitorErr.Error()
	} else {
		widget.View.SetWrap(false)
		content = widget.contentFrom(monitors)
	}
	widget.app.QueueUpdateDraw(func() {
		widget.View.SetTitle(widget.ContextualTitle(fmt.Sprintf("%s", widget.CommonSettings.Title)))
		widget.View.Clear()
		widget.View.SetText(content)
	})
}

/* -------------------- Unexported Functions -------------------- */

func (widget *Widget) contentFrom(monitors []datadog.Monitor) string {
	var str string

	triggeredMonitors := []datadog.Monitor{}

	for _, monitor := range monitors {
		state := *monitor.OverallState
		switch state {
		case "Alert":
			triggeredMonitors = append(triggeredMonitors, monitor)
		}
	}
	if len(triggeredMonitors) > 0 {
		str = str + fmt.Sprintf(
			" %s\n",
			"[red]Triggered Monitors[white]",
		)
		for _, triggeredMonitor := range triggeredMonitors {
			str = str + fmt.Sprintf("[red] %s\n", *triggeredMonitor.Name)
		}
	} else {
		str = str + fmt.Sprintf(
			" %s\n",
			"[green]No Triggered Monitors[white]",
		)
	}

	return str
}