V2bX/main.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"
	"log"
	"os"
	"os/signal"
	"runtime"
	"syscall"
)

var (
	configFile   = flag.String("config", "/etc/V2bX/config.yml", "Config file for V2bX.")
	printVersion = flag.Bool("version", false, "show version")
)

var (
	version  = "v0.0.7_beta"
	codename = "V2bX"
	intro    = "A V2board backend based on Xray-core"
)

func showVersion() {
	fmt.Printf("%s %s (%s) \n", codename, version, intro)
}

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 err
		}
	}
	return nil
}

func main() {
	flag.Parse()
	showVersion()
	if *printVersion {
		return
	}
	config := conf.New()
	err := config.LoadFromPath(*configFile)
	if err != nil {
		log.Panicf("can't unmarshal config file: %s \n", err)
	}
	log.Println("Start V2bX...")
	x := core.New(config)
	err = x.Start()
	if err != nil {
		log.Panicf("Failed to start core: %s", err)
	}
	defer x.Close()
	err = startNodes(config.NodesConfig, x)
	if err != nil {
		log.Panicf("run nodes error: %s", err)
	}
	//Explicitly triggering GC to remove garbage from config loading.
	runtime.GC()
	// Running backend
	{
		osSignals := make(chan os.Signal, 1)
		signal.Notify(osSignals, syscall.SIGINT, syscall.SIGKILL, syscall.SIGTERM)
		<-osSignals
	}
}