init

pkg/conf/conf.go

@@ -0,0 +1,156 @@
+package conf
+
+import (
+	"github.com/cloudreve/Cloudreve/v3/pkg/util"
+	"github.com/go-ini/ini"
+	"github.com/go-playground/validator/v10"
+)
+
+// database 数据库
+type database struct {
+	Type        string
+	User        string
+	Password    string
+	Host        string
+	Name        string
+	TablePrefix string
+	DBFile      string
+	Port        int
+	Charset     string
+	UnixSocket  bool
+}
+
+// system 系统通用配置
+type system struct {
+	Mode          string `validate:"eq=master|eq=slave"`
+	Listen        string `validate:"required"`
+	Debug         bool
+	SessionSecret string
+	HashIDSalt    string
+	GracePeriod   int    `validate:"gte=0"`
+	ProxyHeader   string `validate:"required_with=Listen"`
+}
+
+type ssl struct {
+	CertPath string `validate:"omitempty,required"`
+	KeyPath  string `validate:"omitempty,required"`
+	Listen   string `validate:"required"`
+}
+
+type unix struct {
+	Listen string
+	Perm   uint32
+}
+
+// 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
+	User     string
+	Password string
+	DB       string
+}
+
+// 跨域配置
+type cors struct {
+	AllowOrigins     []string
+	AllowMethods     []string
+	AllowHeaders     []string
+	AllowCredentials bool
+	ExposeHeaders    []string
+	SameSite         string
+	Secure           bool
+}
+
+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("Failed to create config file: %s", err)
+		}
+
+		// 写入配置文件
+		_, err = f.WriteString(confContent)
+		if err != nil {
+			util.Log().Panic("Failed to write config file: %s", err)
+		}
+
+		f.Close()
+	}
+
+	cfg, err = ini.Load(path)
+	if err != nil {
+		util.Log().Panic("Failed to parse config file %q: %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("Failed to parse config section %q: %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
+}

pkg/conf/defaults.go

@@ -0,0 +1,55 @@
+package conf
+
+// RedisConfig Redis服务器配置
+var RedisConfig = &redis{
+	Network:  "tcp",
+	Server:   "",
+	Password: "",
+	DB:       "0",
+}
+
+// DatabaseConfig 数据库配置
+var DatabaseConfig = &database{
+	Type:       "UNSET",
+	Charset:    "utf8",
+	DBFile:     "cloudreve.db",
+	Port:       3306,
+	UnixSocket: false,
+}
+
+// SystemConfig 系统公用配置
+var SystemConfig = &system{
+	Debug:       false,
+	Mode:        "master",
+	Listen:      ":5212",
+	ProxyHeader: "X-Forwarded-For",
+}
+
+// CORSConfig 跨域配置
+var CORSConfig = &cors{
+	AllowOrigins:     []string{"UNSET"},
+	AllowMethods:     []string{"PUT", "POST", "GET", "OPTIONS"},
+	AllowHeaders:     []string{"Cookie", "X-Cr-Policy", "Authorization", "Content-Length", "Content-Type", "X-Cr-Path", "X-Cr-FileName"},
+	AllowCredentials: false,
+	ExposeHeaders:    nil,
+	SameSite:         "Default",
+	Secure:           false,
+}
+
+// SlaveConfig 从机配置
+var SlaveConfig = &slave{
+	CallbackTimeout: 20,
+	SignatureTTL:    60,
+}
+
+var SSLConfig = &ssl{
+	Listen:   ":443",
+	CertPath: "",
+	KeyPath:  "",
+}
+
+var UnixConfig = &unix{
+	Listen: "",
+}
+
+var OptionOverwrite = map[string]interface{}{}

pkg/conf/version.go

@@ -0,0 +1,22 @@
+package conf
+
+// plusVersion 增强版版本号
+const plusVersion = "+1.1"
+
+// BackendVersion 当前后端版本号
+const BackendVersion = "3.8.3" + plusVersion
+
+// KeyVersion 授权版本号
+const KeyVersion = "3.3.1"
+
+// RequiredDBVersion 与当前版本匹配的数据库版本
+const RequiredDBVersion = "3.8.1+1.0-plus"
+
+// RequiredStaticVersion 与当前版本匹配的静态资源版本
+const RequiredStaticVersion = "3.8.3" + plusVersion
+
+// IsPlus 是否为Plus版本
+const IsPlus = "true"
+
+// LastCommit 最后commit id
+const LastCommit = "88409cc"