package main

import (
	"flag"
	"log"
	"time"

	"github.com/ginuerzh/gost"
)

var (
	laddr, faddr string
	quiet        bool
)

func init() {
	log.SetFlags(log.LstdFlags | log.Lshortfile)

	flag.StringVar(&laddr, "L", ":18080", "listen address")
	flag.StringVar(&faddr, "F", ":8080", "forward address")
	flag.BoolVar(&quiet, "q", false, "quiet mode")
	flag.BoolVar(&gost.Debug, "d", false, "debug mode")
	flag.Parse()

	if quiet {
		gost.SetLogger(&gost.NopLogger{})
	}
}
func main() {
	udpRemoteForwardServer()
}

func udpRemoteForwardServer() {
	s := &gost.Server{}
	ln, err := gost.UDPRemoteForwardListener(
		laddr,
		/*
			gost.NewChain(gost.Node{
				Protocol:  "socks5",
				Transport: "tcp",
				Addr:      ":11080",
				User:      url.UserPassword("admin", "123456"),
				Client: &gost.Client{
					Connector: gost.SOCKS5Connector(
						url.UserPassword("admin", "123456"),
					),
					Transporter: gost.TCPTransporter(),
				},
			}),
		*/
		nil,
		time.Second*30)
	if err != nil {
		log.Fatal(err)
	}
	h := gost.UDPRemoteForwardHandler(
		faddr,
	)
	log.Fatal(s.Serve(ln, h))
}
