1
0
mirror of https://github.com/taigrr/wtf synced 2025-01-18 04:03:14 -08:00
wtf/modules/pihole/settings.go
Chris Cummer d6a0797bf2 Simplify the inclusion of the Common config settings into each module
Signed-off-by: Chris Cummer <chriscummer@me.com>
2020-11-26 23:18:46 -08:00

44 lines
1.1 KiB
Go

package pihole
import (
"github.com/olebedev/config"
"github.com/wtfutil/wtf/cfg"
)
const (
defaultFocusable = true
defaultTitle = "Pi-hole"
)
type Settings struct {
*cfg.Common
wrapText bool
apiUrl string
token string
showTopItems int
showTopClients int
maxClientWidth int
maxDomainWidth int
showSummary bool
}
func NewSettingsFromYAML(name string, ymlConfig *config.Config, globalConfig *config.Config) *Settings {
settings := Settings{
Common: cfg.NewCommonSettingsFromModule(name, defaultTitle, defaultFocusable, ymlConfig, globalConfig),
apiUrl: ymlConfig.UString("apiUrl"),
token: ymlConfig.UString("token"),
showSummary: ymlConfig.UBool("showSummary", true),
showTopItems: ymlConfig.UInt("showTopItems", 5),
showTopClients: ymlConfig.UInt("showTopClients", 5),
maxClientWidth: ymlConfig.UInt("maxClientWidth", 20),
maxDomainWidth: ymlConfig.UInt("maxDomainWidth", 20),
}
cfg.ModuleSecret(name, globalConfig, &settings.token).
Service(settings.apiUrl).Load()
return &settings
}