Basic config package to parse NTP Pool system config
This commit is contained in:
90
config/config.go
Normal file
90
config/config.go
Normal file
@@ -0,0 +1,90 @@
|
||||
// Package config provides NTP Pool specific
|
||||
// configuration tools.
|
||||
package config
|
||||
|
||||
import (
|
||||
"net/url"
|
||||
"os"
|
||||
"strconv"
|
||||
"strings"
|
||||
|
||||
"go.ntppool.org/common/logger"
|
||||
)
|
||||
|
||||
//go:generate accessory -type Config
|
||||
|
||||
type Config struct {
|
||||
deploymentMode string `accessor:"getter"`
|
||||
|
||||
manageHostname string `accessor:"getter"`
|
||||
manageTLS bool
|
||||
|
||||
webHostname string `accessor:"getter"`
|
||||
webHostnames []string
|
||||
webTLS bool
|
||||
|
||||
valid bool `accessor:"getter"`
|
||||
}
|
||||
|
||||
func New() *Config {
|
||||
c := Config{}
|
||||
c.deploymentMode = os.Getenv("deployment_mode")
|
||||
c.manageHostname = os.Getenv("manage_hostname")
|
||||
|
||||
c.webHostnames = strings.Split(os.Getenv("web_hostname"), ",")
|
||||
for i, h := range c.webHostnames {
|
||||
c.webHostnames[i] = strings.TrimSpace(h)
|
||||
}
|
||||
if len(c.webHostnames) > 0 {
|
||||
c.webHostname = c.webHostnames[0]
|
||||
c.valid = true
|
||||
}
|
||||
|
||||
c.manageTLS = parseBool(os.Getenv("manage_tls"))
|
||||
c.webTLS = parseBool(os.Getenv("web_tls"))
|
||||
|
||||
return &c
|
||||
}
|
||||
|
||||
func (c *Config) WebURL(path string, query *url.Values) string {
|
||||
return baseURL(c.webHostname, c.webTLS, path, query)
|
||||
}
|
||||
|
||||
func baseURL(host string, tls bool, path string, query *url.Values) string {
|
||||
uri := url.URL{}
|
||||
uri.Host = host
|
||||
if tls {
|
||||
uri.Scheme = "https"
|
||||
} else {
|
||||
uri.Scheme = "http"
|
||||
}
|
||||
uri.Path = path
|
||||
if query != nil {
|
||||
uri.RawQuery = query.Encode()
|
||||
}
|
||||
|
||||
return uri.String()
|
||||
}
|
||||
|
||||
func parseBool(s string) bool {
|
||||
switch strings.ToLower(s) {
|
||||
case "yes":
|
||||
return true
|
||||
case "y":
|
||||
return true
|
||||
case "no":
|
||||
return false
|
||||
case "n":
|
||||
return false
|
||||
case "":
|
||||
return false
|
||||
}
|
||||
|
||||
t, err := strconv.ParseBool(s)
|
||||
if err != nil {
|
||||
logger.Setup().Error("could not parse bool", "string", s, "err", err)
|
||||
return false
|
||||
}
|
||||
|
||||
return t
|
||||
}
|
Reference in New Issue
Block a user