init

config/config.go

@@ -0,0 +1,110 @@
+package config
+
+import (
+	"os"
+
+	"github.com/BurntSushi/toml"
+)
+
+type Config struct {
+	Server ServerConfig
+	Tmpl   TmplConfig
+	DB     DatabaseConfig `toml:"database"`
+}
+
+/*
+[server]
+host = "0.0.0.0"
+port = 8080
+debug = false
+caddyDir = "./"
+*/
+type ServerConfig struct {
+	Port     int    `toml:"port"`
+	Host     string `toml:"host"`
+	Debug    bool   `toml:"debug"`
+	CaddyDir string `toml:"caddyDir"`
+}
+
+/*
+[tmpl]
+path = "./tmpl"
+*/
+type TmplConfig struct {
+	Path string `toml:"path"`
+}
+
+/*
+[database]
+filepath = "sqlite.db"
+*/
+type DatabaseConfig struct {
+	Filepath string `toml:"filepath"`
+}
+
+// LoadConfig 从 TOML 配置文件加载配置
+func LoadConfig(filePath string) (*Config, error) {
+	if !FileExists(filePath) {
+		// 楔入配置文件
+		err := DefaultConfig().WriteConfig(filePath)
+		if err != nil {
+			return nil, err
+		}
+		return DefaultConfig(), nil
+	}
+
+	var config Config
+	if _, err := toml.DecodeFile(filePath, &config); err != nil {
+		return nil, err
+	}
+	return &config, nil
+}
+
+// 写入配置文件
+func (c *Config) WriteConfig(filePath string) error {
+	file, err := os.Create(filePath)
+	if err != nil {
+		return err
+	}
+	defer file.Close()
+
+	encoder := toml.NewEncoder(file)
+	return encoder.Encode(c)
+}
+
+// 检测文件是否存在
+func FileExists(filename string) bool {
+	_, err := os.Stat(filename)
+	return !os.IsNotExist(err)
+}
+
+// 默认配置结构体
+func DefaultConfig() *Config {
+	/*
+		[server]
+		host = "0.0.0.0"
+		port = 81
+		debug = false
+		caddyDir = "./"
+
+		[tmpl]
+		path = "./tmpl"
+
+		[database]
+		filepath = "caddydash.db"
+	*/
+	return &Config{
+		Server: ServerConfig{
+			Host:     "0.0.0.0",
+			Port:     81,
+			Debug:    false,
+			CaddyDir: "./",
+		},
+		Tmpl: TmplConfig{
+			Path: "./tmpl",
+		},
+		DB: DatabaseConfig{
+			Filepath: "caddydash.db",
+		},
+	}
+}

config/config.toml

@@ -0,0 +1,11 @@
+[server]
+host = "0.0.0.0" 
+port = 8080 
+debug = false
+caddyDir = "./"
+
+[tmpl]
+path = "/data/github/WJQSERVER/caddydash/tmpl"
+
+[database]
+filepath = "test.db"