mirror of
https://github.com/nxtrace/NTrace-core.git
synced 2025-08-12 06:26:39 +00:00
115 lines
2.6 KiB
Go
115 lines
2.6 KiB
Go
package printer
|
||
|
||
import (
|
||
"fmt"
|
||
"net"
|
||
|
||
"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) {
|
||
// 初始化表格
|
||
tbl := New()
|
||
for hi := uint16(1); hi < 30; hi++ {
|
||
for _, v := range res[hi] {
|
||
data := tableDataGenerator(v)
|
||
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) *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()
|
||
|
||
// 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)
|
||
}
|
||
|
||
ptr, err_LookupAddr := net.LookupAddr(ipStr)
|
||
|
||
lantency = fmt.Sprintf("%.2fms", v2.RTT.Seconds()*1000)
|
||
|
||
if err_LookupAddr != nil {
|
||
IP = fmt.Sprint(ipStr)
|
||
} else {
|
||
IP = fmt.Sprint(ptr[0], " (", ipStr, ") ")
|
||
}
|
||
|
||
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,
|
||
}
|
||
}
|
||
}
|
||
}
|