add relay proxy

cmd/gost/main.go

@@ -1,9 +1,12 @@
 package main
 
 import (
-	stdlog "log"
+	"flag"
+	"fmt"
 	"net/http"
 	_ "net/http/pprof"
+	"os"
+	"runtime"
 
 	"github.com/go-gost/gost/pkg/config"
 	"github.com/go-gost/gost/pkg/logger"
@@ -11,14 +14,63 @@ import (
 
 var (
 	log = logger.NewLogger()
+
+	cfgFile       string
+	outputCfgFile string
+	services      stringList
+	nodes         stringList
+	debug         bool
 )
 
+func init() {
+	var printVersion bool
+
+	flag.Var(&services, "L", "service list")
+	flag.Var(&nodes, "F", "chain node list")
+	flag.StringVar(&cfgFile, "C", "", "configure file")
+	flag.BoolVar(&printVersion, "V", false, "print version")
+	flag.BoolVar(&debug, "D", false, "debug mode")
+	flag.StringVar(&outputCfgFile, "O", "", "write config to FILE")
+	flag.Parse()
+
+	if printVersion {
+		fmt.Fprintf(os.Stdout, "gost %s (%s %s/%s)\n",
+			version, runtime.Version(), runtime.GOOS, runtime.GOARCH)
+		os.Exit(0)
+	}
+}
+
 func main() {
-	stdlog.SetFlags(stdlog.LstdFlags | stdlog.Lshortfile)
 	cfg := &config.Config{}
-	if err := cfg.Load(); err != nil {
+	var err error
+	if len(services) > 0 {
+		cfg, err = buildConfigFromCmd(services, nodes)
+		if debug && cfg != nil {
+			if cfg.Log == nil {
+				cfg.Log = &config.LogConfig{}
+			}
+			cfg.Log.Level = string(logger.DebugLevel)
+		}
+	} else {
+		if cfgFile != "" {
+			err = cfg.ReadFile(cfgFile)
+		} else {
+			err = cfg.Load()
+		}
+	}
+	if err != nil {
 		log.Fatal(err)
 	}
+
+	normConfig(cfg)
+
+	if outputCfgFile != "" {
+		if err := cfg.WriteFile(outputCfgFile); err != nil {
+			log.Fatal(err)
+		}
+		os.Exit(0)
+	}
+
 	log = logFromConfig(cfg.Log)
 
 	if cfg.Profiling != nil && cfg.Profiling.Enabled {