mirror of
https://github.com/taigrr/wtf
synced 2025-01-18 04:03:14 -08:00
b67339662c
* Merged all windows kludges into unified files * Removed window kludges * Improved UI * Fixed faulty Windows username detection * Fixed dodgy Window firewall detection * Fixed overlapping/overflowing DNS in UI * Added fine-grained Windows firewall status * Added correct and sanitized PowerShell Calls Affcted Files: modified: dns.go deleted: dns_windows.go modified: firewall.go modified: users.go deleted: users_windows.go renamed: widget_windows.go -> widget.go
74 lines
1.8 KiB
Go
74 lines
1.8 KiB
Go
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() {
|
|
|
|
if widget.Disabled() {
|
|
return
|
|
}
|
|
|
|
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]\n"
|
|
str = str + fmt.Sprintf(" %8s: %4s\n", "Status", data.FirewallEnabled)
|
|
str = str + fmt.Sprintf(" %8s: %4s\n", "Stealth", data.FirewallStealth)
|
|
str = str + "\n"
|
|
|
|
str = str + " [red]Users[white]\n"
|
|
str = str + fmt.Sprintf(" %s", strings.Join(data.LoggedInUsers, "\n "))
|
|
str = str + "\n"
|
|
|
|
str = str + " [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 + fmt.Sprintf(" %12s\n", data.DnsAt(0))
|
|
str = str + fmt.Sprintf(" %12s\n", data.DnsAt(1))
|
|
str = str + "\n"
|
|
|
|
return str
|
|
}
|
|
|
|
func (widget *Widget) labelColor(label string) string {
|
|
switch label {
|
|
case "on":
|
|
return "green"
|
|
case "off":
|
|
return "red"
|
|
default:
|
|
return "white"
|
|
}
|
|
}
|