load from toml first

This commit is contained in:
rxdn 2020-07-26 17:11:10 +01:00
parent b78aec6eef
commit 8e95edf3d7

View File

@ -1,6 +1,7 @@
package config package config
import ( import (
"github.com/BurntSushi/toml"
"github.com/TicketsBot/common/sentry" "github.com/TicketsBot/common/sentry"
"os" "os"
"strconv" "strconv"
@ -73,6 +74,20 @@ var (
) )
func LoadConfig() { func LoadConfig() {
if _, err := os.Stat("config.toml"); err == nil {
fromToml()
} else {
fromEnvvar()
}
}
func fromToml() {
if _, err := toml.DecodeFile("config.toml", &Conf); err != nil {
panic(err)
}
}
func fromEnvvar() {
var admins []uint64 var admins []uint64
for _, id := range strings.Split(os.Getenv("ADMINS"), ",") { for _, id := range strings.Split(os.Getenv("ADMINS"), ",") {
if parsed, err := strconv.ParseUint(id, 10, 64); err == nil { if parsed, err := strconv.ParseUint(id, 10, 64); err == nil {