NTrace-core/util/printer/tableprinter.go

package printer

import (
	"fmt"
	"net"
	"strings"

	"github.com/fatih/color"
	"github.com/rodaine/table"
	"github.com/xgadget-lab/nexttrace/ipgeo"
	"github.com/xgadget-lab/nexttrace/methods"
)

type rowData struct {
	Hop      int64
	IP       string
	Latency  string
	Asnumber string
	Country  string
	Prov     string
	City     string
	District string
	Owner    string
}

func TracerouteTablePrinter(ip net.IP, res map[uint16][]methods.TracerouteHop, dataOrigin string, rdnsenable bool) {
	// 初始化表格
	tbl := New()
	for hi := uint16(1); hi < 30; hi++ {
		for _, v := range res[hi] {
			data := tableDataGenerator(v, rdnsenable)
			tbl.AddRow(data.Hop, data.IP, data.Latency, data.Asnumber, data.Country, data.Prov, data.City, data.Owner)
			if v.Address != nil && ip.String() == v.Address.String() {
				hi = 31
			}
		}
	}
	// 打印表格
	tbl.Print()
}

func New() table.Table {
	// 初始化表格
	headerFmt := color.New(color.FgGreen, color.Underline).SprintfFunc()
	columnFmt := color.New(color.FgYellow).SprintfFunc()

	tbl := table.New("Hop", "IP", "Lantency", "ASN", "Country", "Province", "City", "Owner")
	tbl.WithHeaderFormatter(headerFmt).WithFirstColumnFormatter(columnFmt)
	return tbl
}

func tableDataGenerator(v2 methods.TracerouteHop, rdnsenable bool) *rowData {
	if v2.Address == nil {
		return &rowData{
			Hop: int64(v2.TTL),
			IP:  "*",
		}
	} else {
		// 初始化变量
		var iPGeoData *ipgeo.IPGeoData
		var err error
		var lantency, IP string

		ipStr := v2.Address.String()

		if strings.HasPrefix(ipStr, "9.") {
			return &rowData{
				Hop:     int64(v2.TTL),
				IP:      ipStr,
				Latency: lantency,
				Country: "局域网",
				Owner:   "腾讯云",
			}
		} else if strings.HasPrefix(ipStr, "11.") {
			return &rowData{
				Hop:     int64(v2.TTL),
				IP:      ipStr,
				Latency: lantency,
				Country: "局域网",
				Owner:   "阿里云",
			}
		}

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

		if rdnsenable {
			ptr, err_LookupAddr := net.LookupAddr(ipStr)
			if err_LookupAddr != nil {
				IP = fmt.Sprint(ipStr)
			} else {
				IP = fmt.Sprint(ptr[0], " (", ipStr, ") ")
			}
		} else {
			IP = fmt.Sprint(ipStr)
		}

		lantency = fmt.Sprintf("%.2fms", v2.RTT.Seconds()*1000)

		if iPGeoData.Owner == "" {
			iPGeoData.Owner = iPGeoData.Isp
		}

		if err != nil {
			fmt.Print("Error: ", err)
			return &rowData{
				Hop:     int64(v2.TTL),
				IP:      IP,
				Latency: lantency,
			}
		} else {

			return &rowData{
				Hop:      int64(v2.TTL),
				IP:       IP,
				Latency:  lantency,
				Asnumber: iPGeoData.Asnumber,
				Country:  iPGeoData.Country,
				Prov:     iPGeoData.Prov,
				City:     iPGeoData.City,
				District: iPGeoData.District,
				Owner:    iPGeoData.Owner,
			}
		}
	}
}