mirror of
https://github.com/nxtrace/NTrace-core.git
synced 2025-08-12 06:26:39 +00:00
117 lines
2.5 KiB
Go
117 lines
2.5 KiB
Go
package printer
|
||
|
||
import (
|
||
"fmt"
|
||
"strings"
|
||
|
||
"github.com/nxtrace/NTrace-core/trace"
|
||
|
||
"github.com/nxtrace/NTrace-core/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)
|
||
// }
|
||
// }
|
||
// }
|
||
|
||
//此文件目前仅供classic_printer使用
|
||
|
||
const (
|
||
RED_PREFIX = "\033[1;31m"
|
||
GREEN_PREFIX = "\033[1;32m"
|
||
YELLOW_PREFIX = "\033[1;33m"
|
||
BLUE_PREFIX = "\033[1;34m"
|
||
CYAN_PREFIX = "\033[1;36m"
|
||
RESET_PREFIX = "\033[0m"
|
||
)
|
||
|
||
func HopPrinter(h trace.Hop, info HopInfo) {
|
||
if h.Address == nil {
|
||
fmt.Println("\t*")
|
||
} else {
|
||
applyLangSetting(&h) // 应用语言设置
|
||
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)
|
||
}
|
||
for _, v := range h.MPLS {
|
||
txt += " " + v
|
||
}
|
||
switch info {
|
||
case IXP:
|
||
fmt.Print(CYAN_PREFIX)
|
||
case PoP:
|
||
fmt.Print(CYAN_PREFIX)
|
||
case Peer:
|
||
fmt.Print(YELLOW_PREFIX)
|
||
case Aboard:
|
||
fmt.Print(GREEN_PREFIX)
|
||
}
|
||
|
||
fmt.Println(txt)
|
||
|
||
if info != General {
|
||
fmt.Print(RESET_PREFIX)
|
||
}
|
||
}
|
||
}
|
||
|
||
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, "LAN Address")
|
||
//} else if strings.HasPrefix(ip, "11.") {
|
||
// res = append(res, "LAN Address")
|
||
//} else if data.Country == "" {
|
||
// res = append(res, "LAN Address")
|
||
if false {
|
||
} else {
|
||
// 有些IP的归属信息为空，这个时候将ISP的信息填入
|
||
if data.Owner == "" {
|
||
data.Owner = data.Isp
|
||
}
|
||
if data.Prov == "" && data.City == "" {
|
||
// anyCast或是骨干网数据不应该有国家信息
|
||
data.Owner = data.Owner + ", " + data.Owner
|
||
} else {
|
||
// 非骨干网正常填入IP的国家信息数据
|
||
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, ", ")
|
||
}
|