package main

import (
	"flag"
	"fmt"
	"net/http"
	"os"

	"github.com/oneclickvirt/gostun/model"
	"github.com/oneclickvirt/gostun/stuncheck"
	"github.com/pion/logging"
)

func main() {
	var showVersion bool
	flag.BoolVar(&showVersion, "v", false, "show version")
	flag.IntVar(&model.Verbose, "verbose", 0, "the verbosity level")
	flag.IntVar(&model.Timeout, "timeout", 3, "the number of seconds to wait for STUN server's response")
	flag.StringVar(&model.AddrStr, "server", "stun.voipgate.com:3478", "STUN server address")
	flag.Parse()
	go func() {
		http.Get("https://hits.seeyoufarm.com/api/count/incr/badge.svg?url=https%3A%2F%2Fgithub.com%2Foneclickvirt%2Fgostun&count_bg=%2379C83D&title_bg=%23555555&icon=&icon_color=%23E7E7E7&title=hits&edge_flat=false")
	}()
	fmt.Println("项目地址:", "https://github.com/oneclickvirt/gostun")
	if showVersion {
		fmt.Println(model.GoStunVersion)
		return
	}
	var logLevel logging.LogLevel
	switch model.Verbose {
	case 0:
		logLevel = logging.LogLevelWarn // default // my changes
	case 1:
		logLevel = logging.LogLevelInfo
	case 2:
		logLevel = logging.LogLevelDebug
	case 3:
		logLevel = logging.LogLevelTrace
	}
	model.Log = logging.NewDefaultLeveledLoggerForScope("", logLevel, os.Stdout)
	if model.AddrStr != "stun.voipgate.com:3478" {
		if err := stuncheck.MappingTests(model.AddrStr); err != nil {
			model.NatMappingBehavior = "inconclusive" // my changes
			model.Log.Warn("NAT mapping behavior: inconclusive")
		}
		if err := stuncheck.FilteringTests(model.AddrStr); err != nil {
			model.NatFilteringBehavior = "inconclusive" // my changes
			model.Log.Warn("NAT filtering behavior: inconclusive")
		}
	} else {
		addrStrPtrList := []string{
			"stun.voipgate.com:3478",
			"stun.miwifi.com:3478",
			"stunserver.stunprotocol.org:3478",
		}
		checkStatus := true
		for _, addrStr := range addrStrPtrList {
			err1 := stuncheck.MappingTests(addrStr)
			if err1 != nil {
				model.NatMappingBehavior = "inconclusive"
				model.Log.Warn("NAT mapping behavior: inconclusive")
				checkStatus = false
			}
			err2 := stuncheck.FilteringTests(addrStr)
			if err2 != nil {
				model.NatFilteringBehavior = "inconclusive"
				model.Log.Warn("NAT filtering behavior: inconclusive")
				checkStatus = false
			}
			if model.NatMappingBehavior == "inconclusive" || model.NatFilteringBehavior == "inconclusive" {
				checkStatus = false
			} else if model.NatMappingBehavior != "inconclusive" && model.NatFilteringBehavior != "inconclusive" {
				checkStatus = true
			}
			if checkStatus {
				break
			}
		}
	}
	res := stuncheck.CheckType()
	fmt.Printf("NAT Type: %s", res)
}