You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
151 lines
3.2 KiB
151 lines
3.2 KiB
package conf
|
|
|
|
import (
|
|
"github.com/cloudreve/Cloudreve/v3/pkg/util"
|
|
"github.com/go-ini/ini"
|
|
"gopkg.in/go-playground/validator.v9"
|
|
)
|
|
|
|
// database 数据库
|
|
type database struct {
|
|
Type string
|
|
User string
|
|
Password string
|
|
Host string
|
|
Name string
|
|
TablePrefix string
|
|
DBFile string
|
|
Port int
|
|
Charset string
|
|
}
|
|
|
|
// system 系统通用配置
|
|
type system struct {
|
|
Mode string `validate:"eq=master|eq=slave"`
|
|
Listen string `validate:"required"`
|
|
Debug bool
|
|
SessionSecret string
|
|
HashIDSalt string
|
|
}
|
|
|
|
type ssl struct {
|
|
CertPath string `validate:"omitempty,required"`
|
|
KeyPath string `validate:"omitempty,required"`
|
|
Listen string `validate:"required"`
|
|
}
|
|
|
|
type unix struct {
|
|
Listen string
|
|
ProxyHeader string `validate:"required_with=Listen"`
|
|
}
|
|
|
|
// slave 作为slave存储端配置
|
|
type slave struct {
|
|
Secret string `validate:"omitempty,gte=64"`
|
|
CallbackTimeout int `validate:"omitempty,gte=1"`
|
|
SignatureTTL int `validate:"omitempty,gte=1"`
|
|
}
|
|
|
|
// redis 配置
|
|
type redis struct {
|
|
Network string
|
|
Server string
|
|
Password string
|
|
DB string
|
|
}
|
|
|
|
// 跨域配置
|
|
type cors struct {
|
|
AllowOrigins []string
|
|
AllowMethods []string
|
|
AllowHeaders []string
|
|
AllowCredentials bool
|
|
ExposeHeaders []string
|
|
}
|
|
|
|
var cfg *ini.File
|
|
|
|
const defaultConf = `[System]
|
|
Debug = false
|
|
Mode = master
|
|
Listen = :5212
|
|
SessionSecret = {SessionSecret}
|
|
HashIDSalt = {HashIDSalt}
|
|
`
|
|
|
|
// Init 初始化配置文件
|
|
func Init(path string) {
|
|
var err error
|
|
|
|
if path == "" || !util.Exists(path) {
|
|
// 创建初始配置文件
|
|
confContent := util.Replace(map[string]string{
|
|
"{SessionSecret}": util.RandStringRunes(64),
|
|
"{HashIDSalt}": util.RandStringRunes(64),
|
|
}, defaultConf)
|
|
f, err := util.CreatNestedFile(path)
|
|
if err != nil {
|
|
util.Log().Panic("无法创建配置文件, %s", err)
|
|
}
|
|
|
|
// 写入配置文件
|
|
_, err = f.WriteString(confContent)
|
|
if err != nil {
|
|
util.Log().Panic("无法写入配置文件, %s", err)
|
|
}
|
|
|
|
f.Close()
|
|
}
|
|
|
|
cfg, err = ini.Load(path)
|
|
if err != nil {
|
|
util.Log().Panic("无法解析配置文件 '%s': %s", path, err)
|
|
}
|
|
|
|
sections := map[string]interface{}{
|
|
"Database": DatabaseConfig,
|
|
"System": SystemConfig,
|
|
"SSL": SSLConfig,
|
|
"UnixSocket": UnixConfig,
|
|
"Redis": RedisConfig,
|
|
"CORS": CORSConfig,
|
|
"Slave": SlaveConfig,
|
|
}
|
|
for sectionName, sectionStruct := range sections {
|
|
err = mapSection(sectionName, sectionStruct)
|
|
if err != nil {
|
|
util.Log().Panic("配置文件 %s 分区解析失败: %s", sectionName, err)
|
|
}
|
|
}
|
|
|
|
// 映射数据库配置覆盖
|
|
for _, key := range cfg.Section("OptionOverwrite").Keys() {
|
|
OptionOverwrite[key.Name()] = key.Value()
|
|
}
|
|
|
|
// 重设log等级
|
|
if !SystemConfig.Debug {
|
|
util.Level = util.LevelInformational
|
|
util.GloablLogger = nil
|
|
util.Log()
|
|
}
|
|
|
|
}
|
|
|
|
// mapSection 将配置文件的 Section 映射到结构体上
|
|
func mapSection(section string, confStruct interface{}) error {
|
|
err := cfg.Section(section).MapTo(confStruct)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 验证合法性
|
|
validate := validator.New()
|
|
err = validate.Struct(confStruct)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|