package printer

import (
	"fmt"
	"net"
	"strings"

	"github.com/xgadget-lab/nexttrace/ipgeo"
	"github.com/xgadget-lab/nexttrace/methods"
)

var dataOrigin string

func TraceroutePrinter(ip net.IP, res map[uint16][]methods.TracerouteHop, dataOrigin string) {
	for hi := uint16(1); hi < 30; hi++ {
		fmt.Print(hi)
		for _, v := range res[hi] {
			hopPrinter(v)
			if v.Address != nil && ip.String() == v.Address.String() {
				hi = 31
			}
		}
	}
}

func hopPrinter(v2 methods.TracerouteHop) {
	if v2.Address == nil {
		fmt.Println("\t*")
	} else {
		var iPGeoData *ipgeo.IPGeoData
		var err error

		ipStr := v2.Address.String()

		// TODO: 判断 err 返回，并且在CLI终端提示错误
		if dataOrigin == "LeoMoeAPI" {
			iPGeoData, err = ipgeo.LeoIP(ipStr)
		} else if dataOrigin == "IP.SB" {
			iPGeoData, err = ipgeo.IPSB(ipStr)
		} else if dataOrigin == "IPInfo" {
			iPGeoData, err = ipgeo.IPInfo(ipStr)
		} else if dataOrigin == "IPInsight" {
			iPGeoData, err = ipgeo.IPInSight(ipStr)
		} else {
			iPGeoData, err = ipgeo.LeoIP(ipStr)
		}

		geo := ""
		if err != nil {
			geo = fmt.Sprint("Error: ", err)
		} else {
			geo = formatIpGeoData(ipStr, iPGeoData)
		}

		ptr, err := net.LookupAddr(ipStr)

		txt := "\t"
		if err != nil {
			txt += fmt.Sprint(ipStr, " ", fmt.Sprintf("%.2f", v2.RTT.Seconds()*1000), "ms ", geo)
		} else {
			txt += fmt.Sprint(ptr[0], " (", ipStr, ") ", fmt.Sprintf("%.2f", v2.RTT.Seconds()*1000), "ms ", geo)
		}
		fmt.Println(txt)
	}
}

func formatIpGeoData(ip string, data *ipgeo.IPGeoData) string {
	var res = make([]string, 0, 10)

	if data.Asnumber == "" {
		res = append(res, "*")
	} else {
		res = append(res, "AS"+data.Asnumber)
	}

	// TODO: 判断阿里云和腾讯云内网，数据不足，有待进一步完善
	if strings.HasPrefix(ip, "9.31.") || strings.HasPrefix(ip, "11.72.") {
		res = append(res, "局域网", "腾讯云")
	} else if strings.HasPrefix(ip, "11.13.") {
		res = append(res, "局域网", "阿里云")
	} else if data.Country == "" {
		res = append(res, "局域网")
	} else {
		if data.Owner == "" {
			data.Owner = data.Isp
		}
		if data.District != "" {
			data.City = data.City + ", " + data.District
		}
		res = append(res, data.Country)
		if data.Prov != "" {
			res = append(res, data.Prov)
		}
		if data.City != "" {
			res = append(res, data.City)
		}
		if data.Owner != "" {
			res = append(res, data.Owner)
		}
	}

	return strings.Join(res, ", ")
}