change project structure

add across nodes ip limit
add user ip recorder
del config file watch

main.go

@@ -0,0 +1,100 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+	"github.com/Yuzuki616/V2bX/api"
+	"github.com/Yuzuki616/V2bX/conf"
+	"github.com/Yuzuki616/V2bX/node"
+	"github.com/Yuzuki616/V2bX/xray"
+	"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"
+)
+
+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 *xray.Xray) error {
+	for i, _ := range nodes {
+		var apiClient = api.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 := xray.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
+	}
+}