taigrr/wtf
1
0
Fork 0
mirror of https://github.com/taigrr/wtf synced 2025-01-18 04:03:14 -08:00
Code Issues Projects Releases Wiki Activity

WTF-400 Security extracted to new config format

Browse Source
This commit is contained in:
Chris Cummer 2019-04-16 02:04:06 -07:00
parent 8df624f2c0
commit d3eef146cb
3 changed files with 29 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
main.go
View File

@ -270,8 +270,11 @@ func makeWidget(app *tview.Application, pages *tview.Pages, widgetName string) w
case "resourceusage":
settings := resourceusage.NewSettingsFromYAML(wtf.Config)
widget = resourceusage.NewWidget(app, settings)
case "rollbar":
widget = rollbar.NewWidget(app, pages)
case "security":
widget = security.NewWidget(app)
settings := security.NewSettingsFromYAML(wtf.Config)
widget = security.NewWidget(app, settings)
case "status":
widget = status.NewWidget(app)
case "system":
@ -289,8 +292,6 @@ func makeWidget(app *tview.Application, pages *tview.Pages, widgetName string) w
widget = todoist.NewWidget(app, pages)
case "travisci":
widget = travisci.NewWidget(app, pages)
case "rollbar":
widget = rollbar.NewWidget(app, pages)
case "trello":
widget = trello.NewWidget(app)
case "twitter":

18
modules/security/settings.go Normal file
View File

@ -0,0 +1,18 @@
package security
import (
"github.com/olebedev/config"
"github.com/wtfutil/wtf/cfg"
)
type Settings struct {
common *cfg.Common
}
func NewSettingsFromYAML(ymlConfig *config.Config) *Settings {
settings := Settings{
common: cfg.NewCommonSettingsFromYAML(ymlConfig),
}
return &settings
}

12
modules/security/widget.go
View File

@ -10,11 +10,15 @@ import (
type Widget struct {
wtf.TextWidget
settings *Settings
}
func NewWidget(app *tview.Application) *Widget {
func NewWidget(app *tview.Application, settings *Settings) *Widget {
widget := Widget{
TextWidget: wtf.NewTextWidget(app, "Security", "security", false),
settings: settings,
}
return &widget
@ -48,12 +52,10 @@ func (widget *Widget) contentFrom(data *SecurityData) string {
str = str + "\n"
str = str + " [red]Users[white]\n"
str = str + fmt.Sprintf(" %s", strings.Join(data.LoggedInUsers, "\n "))
str = str + "\n"
str = str + fmt.Sprintf(" %s", strings.Join(data.LoggedInUsers, "\n "))
str = str + "\n\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"