122 lines
3.4 KiB
Go
122 lines
3.4 KiB
Go
package global
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/jmoiron/sqlx"
|
|
"github.com/pelletier/go-toml/v2"
|
|
)
|
|
|
|
type (
|
|
dbConfig struct {
|
|
Host string `toml:"host"`
|
|
Name string `toml:"name"`
|
|
User string `toml:"user"`
|
|
Pass string `toml:"pass"`
|
|
}
|
|
|
|
discordConfig struct {
|
|
AdminID string `toml:"admin_id" comment:"NOTE: admin_id to be deprecated in favour of local accounts and SSO."`
|
|
ClientID string `toml:"client_id"`
|
|
Secret string `toml:"secret"`
|
|
}
|
|
|
|
config struct {
|
|
BaseUrl string `toml:"base_url" comment:"Used for OAuth redirects."`
|
|
Port int64 `toml:"port"`
|
|
DataDirectory string `toml:"data_dir"`
|
|
DB dbConfig `toml:"db"`
|
|
Discord discordConfig `toml:"discord"`
|
|
}
|
|
)
|
|
|
|
var Config = func() config {
|
|
configFile := os.Getenv("ARIMELODY_CONFIG")
|
|
if configFile == "" {
|
|
configFile = "config.toml"
|
|
}
|
|
|
|
config := config{
|
|
BaseUrl: "https://arimelody.me",
|
|
Port: 8080,
|
|
}
|
|
|
|
data, err := os.ReadFile(configFile)
|
|
if err != nil {
|
|
configOut, _ := toml.Marshal(&config)
|
|
os.WriteFile(configFile, configOut, os.ModePerm)
|
|
fmt.Printf(
|
|
"A default config.toml has been created. " +
|
|
"Please configure before running again!\n")
|
|
os.Exit(0)
|
|
}
|
|
|
|
err = toml.Unmarshal([]byte(data), &config)
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "FATAL: Failed to parse configuration file: %s\n", err.Error())
|
|
os.Exit(1)
|
|
}
|
|
|
|
err = handleConfigOverrides(&config)
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "FATAL: Failed to parse environment variable %s\n", err.Error())
|
|
os.Exit(1)
|
|
}
|
|
|
|
return config
|
|
}()
|
|
|
|
func handleConfigOverrides(config *config) error {
|
|
var err error
|
|
|
|
if env, has := os.LookupEnv("ARIMELODY_BASE_URL"); has { config.BaseUrl = env }
|
|
if env, has := os.LookupEnv("ARIMELODY_PORT"); has {
|
|
config.Port, err = strconv.ParseInt(env, 10, 0)
|
|
if err != nil { return errors.New("ARIMELODY_PORT: " + err.Error()) }
|
|
}
|
|
if env, has := os.LookupEnv("ARIMELODY_DATA_DIR"); has { config.DataDirectory = env }
|
|
|
|
if env, has := os.LookupEnv("ARIMELODY_DB_HOST"); has { config.DB.Host = env }
|
|
if env, has := os.LookupEnv("ARIMELODY_DB_NAME"); has { config.DB.Name = env }
|
|
if env, has := os.LookupEnv("ARIMELODY_DB_USER"); has { config.DB.User = env }
|
|
if env, has := os.LookupEnv("ARIMELODY_DB_PASS"); has { config.DB.Pass = env }
|
|
|
|
if env, has := os.LookupEnv("ARIMELODY_DISCORD_ADMIN_ID"); has { config.Discord.AdminID = env }
|
|
if env, has := os.LookupEnv("ARIMELODY_DISCORD_CLIENT_ID"); has { config.Discord.ClientID = env }
|
|
if env, has := os.LookupEnv("ARIMELODY_DISCORD_SECRET"); has { config.Discord.Secret = env }
|
|
|
|
return nil
|
|
}
|
|
|
|
var Args = func() map[string]string {
|
|
args := map[string]string{}
|
|
|
|
index := 0
|
|
for index < len(os.Args[1:]) {
|
|
arg := os.Args[index + 1]
|
|
if !strings.HasPrefix(arg, "-") {
|
|
fmt.Printf("FATAL: Parameters must follow an argument (%s).\n", arg)
|
|
os.Exit(1)
|
|
}
|
|
|
|
if index + 3 > len(os.Args) || strings.HasPrefix(os.Args[index + 2], "-") {
|
|
args[arg[1:]] = "true"
|
|
index += 1
|
|
continue
|
|
}
|
|
|
|
val := os.Args[index + 2]
|
|
args[arg[1:]] = val
|
|
// fmt.Printf("%s: %s\n", arg[1:], val)
|
|
index += 2
|
|
}
|
|
|
|
return args
|
|
}()
|
|
|
|
var DB *sqlx.DB
|