mirror of
https://github.com/wyx2685/V2bX.git
synced 2025-01-22 18:08:14 -05:00
104 lines
2.2 KiB
Go
104 lines
2.2 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"os/signal"
|
|
"path"
|
|
"runtime"
|
|
"strings"
|
|
"syscall"
|
|
"time"
|
|
|
|
"github.com/Yuzuki616/V2bX/panel"
|
|
"github.com/fsnotify/fsnotify"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
var (
|
|
configFile = flag.String("config", "", "Config file for XrayR.")
|
|
printVersion = flag.Bool("version", false, "show version")
|
|
)
|
|
|
|
var (
|
|
version = "v0.0.2"
|
|
codename = "V2bX"
|
|
intro = "A V2board backend based on Xray"
|
|
)
|
|
|
|
func showVersion() {
|
|
fmt.Printf("%s %s (%s) \n", codename, version, intro)
|
|
}
|
|
|
|
func getConfig() *viper.Viper {
|
|
config := viper.New()
|
|
|
|
// Set custom path and name
|
|
if *configFile != "" {
|
|
configName := path.Base(*configFile)
|
|
configFileExt := path.Ext(*configFile)
|
|
configNameOnly := strings.TrimSuffix(configName, configFileExt)
|
|
configPath := path.Dir(*configFile)
|
|
config.SetConfigName(configNameOnly)
|
|
config.SetConfigType(strings.TrimPrefix(configFileExt, "."))
|
|
config.AddConfigPath(configPath)
|
|
// Set ASSET Path and Config Path for XrayR
|
|
os.Setenv("XRAY_LOCATION_ASSET", configPath)
|
|
os.Setenv("XRAY_LOCATION_CONFIG", configPath)
|
|
} else {
|
|
// Set default config path
|
|
config.SetConfigName("config")
|
|
config.SetConfigType("yml")
|
|
config.AddConfigPath(".")
|
|
|
|
}
|
|
|
|
if err := config.ReadInConfig(); err != nil {
|
|
log.Panicf("Fatal error config file: %s \n", err)
|
|
}
|
|
|
|
config.WatchConfig() // Watch the config
|
|
|
|
return config
|
|
}
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
showVersion()
|
|
if *printVersion {
|
|
return
|
|
}
|
|
|
|
config := getConfig()
|
|
panelConfig := &panel.Config{}
|
|
config.Unmarshal(panelConfig)
|
|
p := panel.New(panelConfig)
|
|
lastTime := time.Now()
|
|
config.OnConfigChange(func(e fsnotify.Event) {
|
|
// Discarding event received within a short period of time after receiving an event.
|
|
if time.Now().After(lastTime.Add(3 * time.Second)) {
|
|
// Hot reload function
|
|
fmt.Println("Config file changed:", e.Name)
|
|
p.Close()
|
|
// Delete old instance and trigger GC
|
|
runtime.GC()
|
|
config.Unmarshal(panelConfig)
|
|
p.Start()
|
|
lastTime = time.Now()
|
|
}
|
|
})
|
|
p.Start()
|
|
defer p.Close()
|
|
|
|
//Explicitly triggering GC to remove garbage from config loading.
|
|
runtime.GC()
|
|
// Running backend
|
|
{
|
|
osSignals := make(chan os.Signal, 1)
|
|
signal.Notify(osSignals, os.Interrupt, os.Kill, syscall.SIGTERM)
|
|
<-osSignals
|
|
}
|
|
}
|