mirror of
https://github.com/taigrr/wtf
synced 2025-01-18 04:03:14 -08:00
61 lines
1.5 KiB
Go
61 lines
1.5 KiB
Go
// +build !windows
|
|
|
|
package security
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/rivo/tview"
|
|
"github.com/wtfutil/wtf/wtf"
|
|
)
|
|
|
|
type Widget struct {
|
|
wtf.TextWidget
|
|
}
|
|
|
|
func NewWidget(app *tview.Application) *Widget {
|
|
widget := Widget{
|
|
TextWidget: wtf.NewTextWidget(app, "Security", "security", false),
|
|
}
|
|
|
|
return &widget
|
|
}
|
|
|
|
/* -------------------- Exported Functions -------------------- */
|
|
|
|
func (widget *Widget) Refresh() {
|
|
data := NewSecurityData()
|
|
data.Fetch()
|
|
|
|
widget.View.SetText(widget.contentFrom(data))
|
|
}
|
|
|
|
/* -------------------- Unexported Functions -------------------- */
|
|
|
|
func (widget *Widget) contentFrom(data *SecurityData) string {
|
|
str := " [red]WiFi[white]\n"
|
|
str = str + fmt.Sprintf(" %8s: %s\n", "Network", data.WifiName)
|
|
str = str + fmt.Sprintf(" %8s: %s\n", "Crypto", data.WifiEncryption)
|
|
str = str + "\n"
|
|
str = str + " [red]Firewall[white] [red]DNS[white]\n"
|
|
str = str + fmt.Sprintf(" %8s: [%s]%-3s[white] %-16s\n", "Enabled", widget.labelColor(data.FirewallEnabled), data.FirewallEnabled, data.DnsAt(0))
|
|
str = str + fmt.Sprintf(" %8s: [%s]%-3s[white] %-16s\n", "Stealth", widget.labelColor(data.FirewallStealth), data.FirewallStealth, data.DnsAt(1))
|
|
str = str + "\n"
|
|
str = str + " [red]Users[white]\n"
|
|
str = str + fmt.Sprintf(" %s", strings.Join(data.LoggedInUsers, ", "))
|
|
|
|
return str
|
|
}
|
|
|
|
func (widget *Widget) labelColor(label string) string {
|
|
switch label {
|
|
case "on":
|
|
return "green"
|
|
case "off":
|
|
return "red"
|
|
default:
|
|
return "white"
|
|
}
|
|
}
|