package config import ( "embed" "encoding/json" "io/fs" "os" "strings" ) var IsInit bool type Config struct { LogLevel string `json:"logLevel"` // 日志级别 Domain string `json:"domain"` WebDomain string `json:"webDomain"` DkimPrivateKeyPath string `json:"dkimPrivateKeyPath"` SSLType string `json:"sslType"` // 0表示自动生成证书,1表示用户上传证书 SSLPrivateKeyPath string `json:"SSLPrivateKeyPath"` SSLPublicKeyPath string `json:"SSLPublicKeyPath"` DbDSN string `json:"dbDSN"` DbType string `json:"dbType"` WeChatPushAppId string `json:"weChatPushAppId"` WeChatPushSecret string `json:"weChatPushSecret"` WeChatPushTemplateId string `json:"weChatPushTemplateId"` WeChatPushUserId string `json:"weChatPushUserId"` IsInit bool `json:"isInit"` HttpsEnabled int `json:"httpsEnabled"` //后台页面是否启用https,0默认(启用),1启用,2不启用 Tables map[string]string `json:"-"` TablesInitData map[string]string `json:"-"` } //go:embed tables/* var tableConfig embed.FS const Version = "2.1.0" const DBTypeMySQL = "mysql" const DBTypeSQLite = "sqlite" const SSLTypeAuto = "0" //自动生成证书 const SSLTypeUser = "1" //用户上传证书 var DBTypes []string = []string{DBTypeMySQL, DBTypeSQLite} var Instance *Config func Init() { var cfgData []byte var err error args := os.Args if len(args) >= 2 && args[len(args)-1] == "dev" { cfgData, err = os.ReadFile("./config/config.dev.json") if err != nil { return } } else { cfgData, err = os.ReadFile("./config/config.json") if err != nil { return } } err = json.Unmarshal(cfgData, &Instance) if err != nil { return } // 读取表设置 Instance.Tables = map[string]string{} Instance.TablesInitData = map[string]string{} root := "tables/mysql" if Instance.DbType == DBTypeSQLite { root = "tables/sqlite" } err = fs.WalkDir(tableConfig, root, func(path string, info fs.DirEntry, err error) error { if !info.IsDir() && strings.HasSuffix(info.Name(), ".sql") { tableName := strings.ReplaceAll(info.Name(), ".sql", "") i, e := tableConfig.ReadFile(path) if e != nil { panic(e) } if strings.Contains(path, "data") { Instance.TablesInitData[tableName] = string(i) } else { Instance.Tables[tableName] = string(i) } } return nil }) if err != nil { panic(err) } if Instance.Domain != "" && Instance.IsInit { IsInit = true } }