123niel/tsviewer
1
0
Fork 0
forked from Cliffbreak/tsviewer
Code Issues Pull requests Releases Wiki Activity

changed config to yaml; removed flag support

Browse source
This commit is contained in:
Hendrik Schlehlein 2019-04-22 02:51:02 +02:00
parent a454dbc947
commit 59bd3d7139
1 changed files with 44 additions and 121 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

151
config/config.go
View file

@ -1,161 +1,84 @@
package config package config
import ( import (
"encoding/json"
"flag"
"io/ioutil"
"log"
"os" "os"
"strconv" "strconv"
"github.com/spf13/viper"
) )
type Config struct { type Config struct {
User User `json:"user"` User struct {
ServerTS ServerTS `json:"serverTS"` Nickname string
ServerWeb ServerWeb `json:"serverWeb"` Name string
Password string
} }
ServerTS struct {
type User struct { IP string
Nickname string `json:"nickname"` PortServer uint16
Name string `json:"name"` PortQuery uint16
Password string `json:"password"`
} }
ServerWeb struct {
type ServerTS struct { Port uint16
IP string `json:"ip"`
PortServer uint16 `json:"portServer"`
PortQuery uint16 `json:"portQuery"`
} }
type ServerWeb struct {
Port uint16 `json:"port"`
} }
const FileName = "config.json" const FileName = "config.json"
func New() (*Config, error) { func New() (*Config, error) {
config := defaults() viper.SetConfigName("config")
viper.SetConfigFile("yaml")
configFile, err := os.Open(FileName) viper.AddConfigPath(".")
if err != nil { if err := viper.ReadInConfig(); err != nil {
if err := config.createFile(); err != nil {
return nil, err return nil, err
} }
log.Println(config) setDefaults()
return &config, nil
}
defer configFile.Close()
jsonParser := json.NewDecoder(configFile) var config Config
jsonParser.Decode(&config) err := viper.Unmarshal(&config)
config.overrideWithEnv() config.overrideWithEnvironmentVars()
config.overrideWithFlags()
return &config, nil return &config, err
} }
func (config Config) createFile() error { func setDefaults() {
configJSON, err := json.MarshalIndent(config, "", " ") viper.SetDefault("User.Nickname", "serveradmin")
if err != nil { viper.SetDefault("User.Name", "serveradmin")
return err viper.SetDefault("User.Password", "<changeMe>")
viper.SetDefault("ServerTS.IP", "127.0.0.1")
viper.SetDefault("ServerTS.PortServer", "9987")
viper.SetDefault("ServerTS.PortQuery", "10011")
viper.SetDefault("ServerWeb.Port", "80")
} }
return ioutil.WriteFile(FileName, configJSON, 0644) func (config *Config) overrideWithEnvironmentVars() {
} if nickname := os.Getenv("TS3_NICKNAME"); nickname != "" {
func defaults() Config {
return Config{
User: User{
Nickname: "serveradmin",
Name: "serveradmin",
Password: "",
},
ServerTS: ServerTS{
IP: "127.0.0.1",
PortServer: 9987,
PortQuery: 10011,
},
ServerWeb: ServerWeb{
Port: 80,
},
}
}
func (config *Config) overrideWithEnv() {
nickname := os.Getenv("TS3_NICKNAME")
username := os.Getenv("TS3_NAME")
password := os.Getenv("TS3_PW")
tsIP := os.Getenv("TS3_IP")
tsPort, _ := strconv.Atoi(os.Getenv("TS3_PORT"))
tsQuery, _ := strconv.Atoi(os.Getenv("TS3_QUERY"))
webPort, _ := strconv.Atoi(os.Getenv("WEB_PORT"))
if nickname != "" {
config.User.Nickname = nickname config.User.Nickname = nickname
} }
if username != "" { if username := os.Getenv("TS3_NAME"); username != "" {
config.User.Name = username config.User.Name = username
} }
if password != "" { if password := os.Getenv("TS3_PW"); password != "" {
config.User.Password = password config.User.Password = password
} }
if tsIP != "" { if tsIP := os.Getenv("TS3_IP"); tsIP != "" {
config.ServerTS.IP = tsIP config.ServerTS.IP = tsIP
} }
if tsPort <= 65535 && tsPort != 0 { if tsPort, err := strconv.Atoi(os.Getenv("TS3_PORT")); tsPort <= 65535 && tsPort >= 0 && err == nil {
config.ServerTS.PortServer = uint16(tsPort) config.ServerTS.PortServer = uint16(tsPort)
} }
if tsQuery <= 65535 && tsQuery != 0 { if tsQuery, err := strconv.Atoi(os.Getenv("TS3_QUERY")); tsQuery <= 65535 && tsQuery >= 0 && err == nil {
config.ServerTS.PortQuery = uint16(tsQuery) config.ServerTS.PortQuery = uint16(tsQuery)
} }
if webPort <= 65535 && webPort != 0 { if webPort, err := strconv.Atoi(os.Getenv("WEB_PORT")); webPort <= 65535 && webPort >= 0 && err == nil {
config.ServerWeb.Port = uint16(webPort) config.ServerWeb.Port = uint16(webPort)
} }
} }
func (config *Config) overrideWithFlags() {
nickname := flag.String("ts3_nickname", "", "TS3Query nickname")
username := flag.String("ts3_name", "", "TS3Query username")
password := flag.String("ts3_pw", "", "TS3Query user password")
tsIP := flag.String("ts3_ip", "", "TS3 IP address")
tsPort := flag.Uint64("ts3_port", 65536, "TS3 port")
tsQuery := flag.Uint64("ts3_query", 65536, "TS3 query port")
webPort := flag.Uint64("web_port", 65536, "Webserver port")
flag.Parse()
if *nickname != "" {
config.User.Nickname = *nickname
}
if *username != "" {
config.User.Name = *username
}
if *password != "" {
config.User.Password = *password
}
if *tsIP != "" {
config.ServerTS.IP = *tsIP
}
if *tsPort <= 65535 && *tsPort != 0 {
config.ServerTS.PortServer = uint16(*tsPort)
}
if *tsQuery <= 65535 && *tsQuery != 0 {
config.ServerTS.PortQuery = uint16(*tsQuery)
}
if *webPort <= 65535 && *webPort != 0 {
config.ServerWeb.Port = uint16(*webPort)
}
}