mirror of
https://github.com/nxtrace/NTrace-core.git
synced 2025-08-12 06:26:39 +00:00
80 lines
1.7 KiB
Go
80 lines
1.7 KiB
Go
package printer
|
||
|
||
import (
|
||
"fmt"
|
||
"github.com/xgadget-lab/nexttrace/trace"
|
||
"strings"
|
||
|
||
"github.com/xgadget-lab/nexttrace/ipgeo"
|
||
)
|
||
|
||
var dataOrigin string
|
||
|
||
func TraceroutePrinter(res *trace.Result) {
|
||
for i, hop := range res.Hops {
|
||
fmt.Print(i + 1)
|
||
for _, h := range hop {
|
||
hopPrinter(h)
|
||
}
|
||
}
|
||
}
|
||
|
||
func hopPrinter(h trace.Hop) {
|
||
if h.Address == nil {
|
||
fmt.Println("\t*")
|
||
} else {
|
||
txt := "\t"
|
||
|
||
if h.Hostname == "" {
|
||
txt += fmt.Sprint(h.Address, " ", fmt.Sprintf("%.2f", h.RTT.Seconds()*1000), "ms")
|
||
} else {
|
||
txt += fmt.Sprint(h.Hostname, " (", h.Address, ") ", fmt.Sprintf("%.2f", h.RTT.Seconds()*1000), "ms")
|
||
}
|
||
|
||
if h.Geo != nil {
|
||
txt += " " + formatIpGeoData(h.Address.String(), h.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: 判断阿里云和腾讯云内网，数据不足，有待进一步完善
|
||
// TODO: 移动IDC判断到Hop.fetchIPData函数，减少API调用
|
||
if strings.HasPrefix(ip, "9.") {
|
||
res = append(res, "局域网", "腾讯云")
|
||
} else if strings.HasPrefix(ip, "11.") {
|
||
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, ", ")
|
||
}
|