增加配置功能

config/config.go

@@ -0,0 +1,64 @@
+package config
+
+import (
+	"os"
+	"path/filepath"
+
+	"github.com/goccy/go-yaml"
+)
+
+type WebConfig struct {
+	Port int    `yaml:"port"`
+	Sock string `yaml:"sock"`
+}
+
+type Config struct {
+	Web WebConfig `yaml:"web"`
+}
+
+func defaultConfig() *Config {
+	return &Config{
+		Web: WebConfig{
+			Port: 8080,
+			Sock: "/run/lmvpnweb.sock",
+		},
+	}
+}
+
+func Load(path string) (*Config, error) {
+	dir := filepath.Dir(path)
+	if err := os.MkdirAll(dir, 0755); err != nil {
+		return nil, err
+	}
+
+	cfg := defaultConfig()
+
+	data, err := os.ReadFile(path)
+	if err != nil {
+		if !os.IsNotExist(err) {
+			return nil, err
+		}
+
+		if err := saveConfig(path, cfg); err != nil {
+			return nil, err
+		}
+		return cfg, nil
+	}
+
+	if err := yaml.Unmarshal(data, cfg); err != nil {
+		return nil, err
+	}
+
+	if err := saveConfig(path, cfg); err != nil {
+		return nil, err
+	}
+	return cfg, nil
+}
+
+func saveConfig(path string, cfg *Config) error {
+	data, err := yaml.Marshal(cfg)
+	if err != nil {
+		return err
+	}
+	return os.WriteFile(path, data, 0644)
+}