snow/aid/config.go

109 lines
2.3 KiB
Go

package aid
import (
"os"
"strconv"
"strings"
"gopkg.in/ini.v1"
)
type CS struct {
Database struct {
URI string
Type string
DropAllTables bool
}
Output struct {
Level string
}
API struct {
Host string
Port string
}
JWT struct {
Secret string
}
Fortnite struct {
Season int
Build float64
Everything bool
}
}
var (
Config *CS
)
func LoadConfig() {
Config = &CS{}
configPath := "config.ini"
if _, err := os.Stat(configPath); err != nil {
panic("config.ini not found! please rename default.config.ini to config.ini and complete")
}
cfg, err := ini.Load("config.ini")
if err != nil {
panic(err)
}
Config.Database.DropAllTables = cfg.Section("database").Key("drop").MustBool(false)
Config.Database.URI = cfg.Section("database").Key("uri").String()
if Config.Database.URI == "" {
panic("Database URI is empty")
}
Config.Database.Type = cfg.Section("database").Key("type").String()
if Config.Database.Type == "" {
panic("Database Type is empty")
}
Config.Output.Level = cfg.Section("output").Key("level").String()
if Config.Output.Level == "" {
panic("Output Level is empty")
}
if Config.Output.Level != "dev" && Config.Output.Level != "prod" && Config.Output.Level != "time" && Config.Output.Level != "info" {
panic("Output Level must be either dev or prod")
}
Config.API.Host = cfg.Section("api").Key("host").String()
if Config.API.Host == "" {
panic("API Host is empty")
}
Config.API.Port = cfg.Section("api").Key("port").String()
if Config.API.Port == "" {
panic("API Port is empty")
}
Config.JWT.Secret = cfg.Section("jwt").Key("secret").String()
if Config.JWT.Secret == "" {
panic("JWT Secret is empty")
}
build, err := cfg.Section("fortnite").Key("build").Float64()
if err != nil {
panic("Fortnite Build is empty")
}
Config.Fortnite.Build = build
buildStr := strconv.FormatFloat(build, 'f', -1, 64)
if buildStr == "" {
panic("Fortnite Build is empty")
}
buildInfo := strings.Split(buildStr, ".")
if len(buildInfo) < 2 {
panic("Fortnite Build is invalid")
}
parsedSeason, err := strconv.Atoi(buildInfo[0])
if err != nil {
panic("Fortnite Season is invalid")
}
Config.Fortnite.Season = parsedSeason
Config.Fortnite.Everything = cfg.Section("fortnite").Key("everything").MustBool(false)
}