86 lines
2.0 KiB
Go
86 lines
2.0 KiB
Go
|
package conf
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"encoding/json"
|
|||
|
|
"github.com/BurntSushi/toml"
|
|||
|
|
"github.com/mcuadros/go-defaults"
|
|||
|
|
"mc-client-updater-server/pkg/log"
|
|||
|
|
"mc-client-updater-server/pkg/util"
|
|||
|
|
"os"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type Config struct {
|
|||
|
|
Common Common `toml:"common"`
|
|||
|
|
Database Database `toml:"database"`
|
|||
|
|
Security Security `toml:"security"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type Common struct {
|
|||
|
|
Host string `toml:"host" default:"0.0.0.0"`
|
|||
|
|
Port uint16 `toml:"port" default:"8080"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type Database struct {
|
|||
|
|
Host string `toml:"host" default:"localhost"`
|
|||
|
|
Port uint16 `toml:"port" default:"5432"`
|
|||
|
|
User string `toml:"user" default:"user"`
|
|||
|
|
Password string `toml:"password" default:"password"`
|
|||
|
|
DB string `toml:"db" default:"mc_client_updater"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type Security struct {
|
|||
|
|
PasswordEncoder string `toml:"password_encoder" default:"argon2id"`
|
|||
|
|
JWTSecret string `toml:"jwt_secret"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var (
|
|||
|
|
Conf *Config
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// 释放默认配置文件
|
|||
|
|
func releaseConfig(file string, config *Config) {
|
|||
|
|
f, err := util.CreateNestedFile(file)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Logger.Fatalf("创建默认配置文件失败: %s", err.Error())
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if config.Security.JWTSecret == "" {
|
|||
|
|
config.Security.JWTSecret = util.RandStr(32)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
encoder := toml.NewEncoder(f)
|
|||
|
|
err = encoder.Encode(config)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Logger.Fatalf("创建默认配置文件失败: %s", err.Error())
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func InitConfig(file string) {
|
|||
|
|
Conf = &Config{}
|
|||
|
|
defaults.SetDefaults(Conf)
|
|||
|
|
|
|||
|
|
if util.NotExists(file) {
|
|||
|
|
// 默认配置文件不存在则创建
|
|||
|
|
if file == "config.toml" {
|
|||
|
|
releaseConfig(file, Conf)
|
|||
|
|
log.Logger.Infof("已创建默认配置文件: %s，请修改配置文件内容后重新启动", file)
|
|||
|
|
os.Exit(0)
|
|||
|
|
} else {
|
|||
|
|
log.Logger.Fatalf("配置文件不存在: %s", file)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 解析配置文件
|
|||
|
|
_, err := toml.DecodeFile(file, &Conf)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Logger.Fatalf("配置文件解析出错: %s", err.Error())
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 配置预检查
|
|||
|
|
|
|||
|
|
jsonConfig, err := json.Marshal(Conf)
|
|||
|
|
if err == nil {
|
|||
|
|
log.Logger.Debugf("配置文件已加载: %s", string(jsonConfig))
|
|||
|
|
}
|
|||
|
|
}
|