pub

internal/config/config.go

@@ -0,0 +1,77 @@
+package config
+
+import (
+	"encoding/json"
+	"fmt"
+	"os"
+	"path/filepath"
+
+	"gopkg.in/yaml.v3"
+)
+
+// Config represents the root configuration of the proxy.
+type Config struct {
+	HTTP HTTPConfig `yaml:"http" json:"http"`
+	TCP  TCPConfig  `yaml:"tcp" json:"tcp"`
+	UDP  UDPConfig  `yaml:"udp" json:"udp"`
+}
+
+type UDPConfig struct {
+	Port    int    `yaml:"port" json:"port"`
+	Backend string `yaml:"backend" json:"backend"`
+}
+
+type HTTPConfig struct {
+	Port    int           `yaml:"port" json:"port"`
+	TLS     TLSConfig     `yaml:"tls" json:"tls"`
+	Routers []RouterRoute `yaml:"routers" json:"routers"`
+}
+
+type TLSConfig struct {
+	Enabled  bool   `yaml:"enabled" json:"enabled"`
+	Port     int    `yaml:"port" json:"port"`
+	CertFile string `yaml:"cert_file" json:"cert_file"`
+	KeyFile  string `yaml:"key_file" json:"key_file"`
+}
+
+type TCPConfig struct {
+	Port    int    `yaml:"port" json:"port"`
+	Backend string `yaml:"backend" json:"backend"`
+}
+
+type RouterRoute struct {
+	Path        string   `yaml:"path" json:"path"`
+	Host        string   `yaml:"host" json:"host"`
+	Backends    []string `yaml:"backends" json:"backends"`       // Used for Load Balancing
+	BackendURL  string   `yaml:"backend_url" json:"backend_url"` // Legacy / Single backend
+	StripPrefix bool     `yaml:"strip_prefix" json:"strip_prefix"`
+}
+
+// LoadConfig reads a configuration file (YAML or JSON) and parses it into the Config struct.
+func LoadConfig(path string) (*Config, error) {
+	file, err := os.Open(path)
+	if err != nil {
+		return nil, fmt.Errorf("failed to open config file: %w", err)
+	}
+	defer file.Close()
+
+	var cfg Config
+	ext := filepath.Ext(path)
+	
+	switch ext {
+	case ".yaml", ".yml":
+		decoder := yaml.NewDecoder(file)
+		if err := decoder.Decode(&cfg); err != nil {
+			return nil, fmt.Errorf("failed to parse yaml config: %w", err)
+		}
+	case ".json":
+		decoder := json.NewDecoder(file)
+		if err := decoder.Decode(&cfg); err != nil {
+			return nil, fmt.Errorf("failed to parse json config: %w", err)
+		}
+	default:
+		return nil, fmt.Errorf("unsupported config format: %s", ext)
+	}
+
+	return &cfg, nil
+}

internal/config/watcher.go

@@ -0,0 +1,55 @@
+package config
+
+import (
+	"log"
+	"path/filepath"
+
+	"github.com/fsnotify/fsnotify"
+)
+
+// WatchConfig watches the config file for changes and triggers the callback
+func WatchConfig(path string, onChange func(*Config)) error {
+	watcher, err := fsnotify.NewWatcher()
+	if err != nil {
+		return err
+	}
+
+	absPath, err := filepath.Abs(path)
+	if err != nil {
+		return err
+	}
+	configDir := filepath.Dir(absPath)
+
+	go func() {
+		defer watcher.Close()
+		for {
+			select {
+			case event, ok := <-watcher.Events:
+				if !ok {
+					return
+				}
+				// Many editors trigger Write or Rename when saving
+				if event.Has(fsnotify.Write) || event.Has(fsnotify.Create) {
+					if filepath.Base(event.Name) == filepath.Base(absPath) {
+						log.Printf("Config file changed: %s", event.Name)
+						newCfg, err := LoadConfig(absPath)
+						if err != nil {
+							log.Printf("Failed to reload config: %v", err)
+							continue
+						}
+						onChange(newCfg)
+					}
+				}
+			case err, ok := <-watcher.Errors:
+				if !ok {
+					return
+				}
+				log.Printf("Watcher error: %v", err)
+			}
+		}
+	}()
+
+	// Watch the directory instead of the file to handle cases where 
+	// editors use atomic saves (create new file, delete old, rename new)
+	return watcher.Add(configDir)
+}