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
103 lines
2.0 KiB
Go
103 lines
2.0 KiB
Go
package hibp
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/rivo/tview"
|
|
"github.com/wtfutil/wtf/view"
|
|
)
|
|
|
|
// Widget is the container for hibp data
|
|
type Widget struct {
|
|
view.TextWidget
|
|
|
|
settings *Settings
|
|
data []*Status
|
|
err error
|
|
}
|
|
|
|
// NewWidget creates a new instance of a widget
|
|
func NewWidget(app *tview.Application, settings *Settings) *Widget {
|
|
widget := &Widget{
|
|
TextWidget: view.NewTextWidget(app, settings.common, false),
|
|
|
|
settings: settings,
|
|
}
|
|
|
|
return widget
|
|
}
|
|
|
|
/* -------------------- Exported Functions -------------------- */
|
|
|
|
// Fetch retrieves HIBP data from the HIBP API
|
|
func (widget *Widget) Fetch(accounts []string) ([]*Status, error) {
|
|
data := []*Status{}
|
|
|
|
for _, account := range accounts {
|
|
stat, err := widget.fetchForAccount(account, widget.settings.since)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
data = append(data, stat)
|
|
}
|
|
|
|
return data, nil
|
|
}
|
|
|
|
// Refresh updates the data for this widget and displays it onscreen
|
|
func (widget *Widget) Refresh() {
|
|
data, err := widget.Fetch(widget.settings.accounts)
|
|
|
|
if err != nil {
|
|
widget.err = err
|
|
widget.data = nil
|
|
} else {
|
|
widget.err = nil
|
|
widget.data = data
|
|
}
|
|
|
|
widget.Redraw(widget.content)
|
|
}
|
|
|
|
/* -------------------- Unexported Functions -------------------- */
|
|
|
|
func (widget *Widget) content() (string, string, bool) {
|
|
title := widget.CommonSettings().Title
|
|
if widget.err != nil {
|
|
return title, widget.err.Error(), true
|
|
}
|
|
title = title + widget.sinceDateForTitle()
|
|
str := ""
|
|
|
|
for _, stat := range widget.data {
|
|
color := widget.settings.colors.ok
|
|
if stat.HasBeenCompromised() {
|
|
color = widget.settings.colors.pwned
|
|
}
|
|
|
|
str += fmt.Sprintf(" [%s]%s[white]\n", color, stat.Account)
|
|
}
|
|
|
|
return title, str, false
|
|
}
|
|
|
|
func (widget *Widget) sinceDateForTitle() string {
|
|
dateStr := ""
|
|
|
|
if widget.settings.HasSince() {
|
|
sinceStr := ""
|
|
|
|
dt, err := widget.settings.SinceDate()
|
|
if err != nil {
|
|
sinceStr = widget.settings.since
|
|
} else {
|
|
sinceStr = dt.Format("Jan _2, 2006")
|
|
}
|
|
|
|
dateStr = dateStr + " since " + sinceStr
|
|
}
|
|
|
|
return dateStr
|
|
}
|