mirror of
https://github.com/taigrr/wtf
synced 2025-01-18 04:03:14 -08:00
fd794707cd
* WTF-1031 Rename WtfApp.app to WtfApp.tviewApp Signed-off-by: Chris Cummer <chriscummer@me.com> * WTF-1031 Add scaffolding for main to support multiple WtfApp instances Signed-off-by: Chris Cummer <chriscummer@me.com> * WTF-1031 WIP Signed-off-by: Chris Cummer <chriscummer@me.com> * Remove common functionality from KeyboardWidget and into Base Signed-off-by: Chris Cummer <chriscummer@me.com> * Augment with some descriptive comments Signed-off-by: Chris Cummer <chriscummer@me.com> * Add full support for multiple app instances via the AppManager. Still to do: * Config support for multiple apps/multiple config files * The ability to switch between apps Signed-off-by: Chris Cummer <chriscummer@me.com> * Move SetTerminal out of main and into its own file Signed-off-by: Chris Cummer <chriscummer@me.com>
80 lines
1.6 KiB
Go
80 lines
1.6 KiB
Go
package pihole
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/rivo/tview"
|
|
"github.com/wtfutil/wtf/view"
|
|
)
|
|
|
|
type Widget struct {
|
|
view.MultiSourceWidget
|
|
view.TextWidget
|
|
|
|
settings *Settings
|
|
}
|
|
|
|
// NewWidget creates a new instance of a widget
|
|
func NewWidget(tviewApp *tview.Application, pages *tview.Pages, settings *Settings) *Widget {
|
|
widget := Widget{
|
|
TextWidget: view.NewTextWidget(tviewApp, nil, settings.Common),
|
|
settings: settings,
|
|
}
|
|
|
|
widget.settings.RefreshInterval = 30
|
|
widget.initializeKeyboardControls()
|
|
widget.SetDisplayFunction(widget.Refresh)
|
|
widget.View.SetWordWrap(true)
|
|
widget.View.SetWrap(settings.wrapText)
|
|
|
|
return &widget
|
|
}
|
|
|
|
/* -------------------- Exported Functions -------------------- */
|
|
|
|
func (widget *Widget) Refresh() {
|
|
if widget.Disabled() {
|
|
return
|
|
}
|
|
|
|
widget.Redraw(widget.content)
|
|
}
|
|
|
|
/* -------------------- Unexported Functions -------------------- */
|
|
|
|
func (widget *Widget) content() (string, string, bool) {
|
|
title := widget.CommonSettings().Title
|
|
|
|
c := getClient()
|
|
|
|
if err := checkServer(c, widget.settings.apiUrl); err != nil {
|
|
return title, err.Error(), widget.settings.wrapText
|
|
}
|
|
|
|
var sb strings.Builder
|
|
|
|
if widget.settings.showSummary {
|
|
sb.WriteString(getSummaryView(c, widget.settings))
|
|
}
|
|
|
|
if widget.settings.showTopItems > 0 {
|
|
sb.WriteString(getTopItemsView(c, widget.settings))
|
|
}
|
|
|
|
if widget.settings.showTopClients > 0 {
|
|
sb.WriteString(getTopClientsView(c, widget.settings))
|
|
}
|
|
|
|
output := sb.String()
|
|
|
|
return title, output, widget.settings.wrapText
|
|
}
|
|
|
|
func (widget *Widget) disable() {
|
|
widget.adblockSwitch("disable")
|
|
}
|
|
|
|
func (widget *Widget) enable() {
|
|
widget.adblockSwitch("enable")
|
|
}
|