V2bX/conf/conf.go

package conf

import (
	"fmt"
	"github.com/fsnotify/fsnotify"
	"gopkg.in/yaml.v3"
	"log"
	"os"
	"path"
)

type Conf struct {
	LogConfig          *LogConfig       `yaml:"Log"`
	DnsConfigPath      string           `yaml:"DnsConfigPath"`
	InboundConfigPath  string           `yaml:"InboundConfigPath"`
	OutboundConfigPath string           `yaml:"OutboundConfigPath"`
	RouteConfigPath    string           `yaml:"RouteConfigPath"`
	ConnectionConfig   *ConnetionConfig `yaml:"ConnectionConfig"`
	NodesConfig        []*NodeConfig    `yaml:"Nodes"`
}

func New() *Conf {
	return &Conf{
		LogConfig:          NewLogConfig(),
		DnsConfigPath:      "",
		InboundConfigPath:  "",
		OutboundConfigPath: "",
		RouteConfigPath:    "",
		ConnectionConfig:   NewConnetionConfig(),
		NodesConfig:        []*NodeConfig{},
	}
}

func (p *Conf) LoadFromPath(filePath string) error {
	confPath := path.Dir(filePath)
	os.Setenv("XRAY_LOCATION_ASSET", confPath)
	os.Setenv("XRAY_LOCATION_CONFIG", confPath)
	f, err := os.Open(filePath)
	if err != nil {
		return fmt.Errorf("open config file error: %s", err)
	}
	err = yaml.NewDecoder(f).Decode(p)
	if err != nil {
		return fmt.Errorf("decode config error: %s", err)
	}
	return nil
}

func (p *Conf) Watch(filePath string, reload func()) error {
	watcher, err := fsnotify.NewWatcher()
	if err != nil {
		return fmt.Errorf("new watcher error: %s", err)
	}
	go func() {
		defer watcher.Close()
		for {
			select {
			case event := <-watcher.Events:
				if event.Op&fsnotify.Chmod == fsnotify.Chmod {
					if event.Name == filePath {
						log.Println("config file changed, reloading...")
						err := p.LoadFromPath(filePath)
						if err != nil {
							log.Printf("reload config error: %s", err)
						}
						log.Println("reload config success")
						reload()
					}
				}
			case err := <-watcher.Errors:
				if err != nil {
					log.Panicf("watcher error: %s", err)
				}
			}
		}
	}()
	err = watcher.Add(path.Dir(filePath))
	if err != nil {
		return fmt.Errorf("watch file error: %s", err)
	}
	return nil
}