PMail/server/config/config.go

package config

import (
	"embed"
	"encoding/json"
	"io/fs"
	"os"
	"strings"
)

var IsInit bool

type Config struct {
	Domain               string            `json:"domain"`
	WebDomain            string            `json:"webDomain"`
	DkimPrivateKeyPath   string            `json:"dkimPrivateKeyPath"`
	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"`
	Tables               map[string]string `json:"-"`
	TablesInitData       map[string]string `json:"-"`
}

//go:embed tables/*
var tableConfig embed.FS

const Version = "1.1.0"

const DBTypeMySQL = "mysql"
const DBTypeSQLite = "sqlite"

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
	}

}