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.Has(fsnotify.Write) {
					log.Println("config dir 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.Printf("File watcher error: %s", err)
				}
			}
		}
	}()
	err = watcher.Add(path.Dir(filePath))
	if err != nil {
		return fmt.Errorf("watch file error: %s", err)
	}
	return nil
}