mirror of
https://github.com/wyx2685/V2bX.git
synced 2025-01-22 09:58:14 -05:00
101 lines
2.3 KiB
Go
101 lines
2.3 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"github.com/Yuzuki616/V2bX/api/panel"
|
|
"github.com/Yuzuki616/V2bX/conf"
|
|
"github.com/Yuzuki616/V2bX/core"
|
|
"github.com/Yuzuki616/V2bX/node"
|
|
"github.com/spf13/viper"
|
|
"log"
|
|
"os"
|
|
"os/signal"
|
|
"path"
|
|
"runtime"
|
|
"strings"
|
|
"syscall"
|
|
)
|
|
|
|
var (
|
|
configFile = flag.String("config", "", "Config file for XrayR.")
|
|
printVersion = flag.Bool("version", false, "show version")
|
|
)
|
|
|
|
var (
|
|
version = "v0.0.4"
|
|
codename = "V2bX"
|
|
intro = "A V2board backend based on Xray-core"
|
|
)
|
|
|
|
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)
|
|
}
|
|
return config
|
|
}
|
|
|
|
func startNodes(nodes []*conf.NodeConfig, core *core.Core) error {
|
|
for i, _ := range nodes {
|
|
var apiClient = panel.New(nodes[i].ApiConfig)
|
|
// Register controller service
|
|
err := node.New(core, apiClient, nodes[i].ControllerConfig).Start()
|
|
if err != nil {
|
|
return fmt.Errorf("start node controller error: %v", err)
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
showVersion()
|
|
if *printVersion {
|
|
return
|
|
}
|
|
config := getConfig()
|
|
c := conf.New()
|
|
err := config.Unmarshal(c)
|
|
if err != nil {
|
|
log.Panicf("can't unmarshal config file: %s \n", err)
|
|
}
|
|
x := core.New(c)
|
|
x.Start()
|
|
defer x.Close()
|
|
err = startNodes(c.NodesConfig, x)
|
|
if err != nil {
|
|
log.Panicf("run nodes error: %v", err)
|
|
}
|
|
//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
|
|
}
|
|
}
|