mirror of
https://github.com/nxtrace/NTrace-core.git
synced 2025-08-12 06:26:39 +00:00
340 lines
7.3 KiB
Go
340 lines
7.3 KiB
Go
package ipgeo
|
|
|
|
import (
|
|
"github.com/nxtrace/NTrace-core/util"
|
|
"io"
|
|
"net/http"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/tidwall/gjson"
|
|
)
|
|
|
|
func IPInfo(ip string, timeout time.Duration, _ string, _ bool) (*IPGeoData, error) {
|
|
url := token.BaseOrDefault("http://ipinfo.io/") + ip + "?token=" + token.ipinfo
|
|
client := &http.Client{
|
|
// 2 秒超时
|
|
Timeout: timeout,
|
|
}
|
|
resp, err := client.Get(url)
|
|
//resp, err := http.Get("https://ipinfo.io/" + ip + "?token=" + token.ipinfo)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer resp.Body.Close()
|
|
body, err := io.ReadAll(resp.Body)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
res := gjson.ParseBytes(body)
|
|
|
|
// ISO-3166 转换
|
|
var countryMap = map[string]string{
|
|
"AF": "Afghanistan",
|
|
"AX": "Åland Islands",
|
|
"AL": "Albania",
|
|
"DZ": "Algeria",
|
|
"AS": "American Samoa",
|
|
"AD": "Andorra",
|
|
"AO": "Angola",
|
|
"AI": "Anguilla",
|
|
"AQ": "Antarctica",
|
|
"AG": "Antigua and Barbuda",
|
|
"AR": "Argentina",
|
|
"AM": "Armenia",
|
|
"AW": "Aruba",
|
|
"AU": "Australia",
|
|
"AT": "Austria",
|
|
"AZ": "Azerbaijan",
|
|
"BH": "Bahrain",
|
|
"BS": "Bahamas",
|
|
"BD": "Bangladesh",
|
|
"BB": "Barbados",
|
|
"BY": "Belarus",
|
|
"BE": "Belgium",
|
|
"BZ": "Belize",
|
|
"BJ": "Benin",
|
|
"BM": "Bermuda",
|
|
"BT": "Bhutan",
|
|
"BO": "Bolivia",
|
|
"BQ": "Bonaire",
|
|
"BA": "Bosnia and Herzegovina",
|
|
"BW": "Botswana",
|
|
"BV": "Bouvet Island",
|
|
"BR": "Brazil",
|
|
"IO": "British Indian Ocean Territory",
|
|
"BN": "Brunei Darussalam",
|
|
"BG": "Bulgaria",
|
|
"BF": "Burkina Faso",
|
|
"BI": "Burundi",
|
|
"KH": "Cambodia",
|
|
"CM": "Cameroon",
|
|
"CA": "Canada",
|
|
"CV": "Cape Verde",
|
|
"KY": "Cayman Islands",
|
|
"CF": "Central African Republic",
|
|
"TD": "Chad",
|
|
"CL": "Chile",
|
|
"CN": "China",
|
|
"CX": "Christmas Island",
|
|
"CC": "Cocos (Keeling) Islands",
|
|
"CO": "Colombia",
|
|
"KM": "Comoros",
|
|
"CG": "Congo",
|
|
"CD": "Congo",
|
|
"CK": "Cook Islands",
|
|
"CR": "Costa Rica",
|
|
"CI": "Côte d'Ivoire",
|
|
"HR": "Croatia",
|
|
"CU": "Cuba",
|
|
"CW": "Curaçao",
|
|
"CY": "Cyprus",
|
|
"CZ": "Czech Republic",
|
|
"DK": "Denmark",
|
|
"DJ": "Djibouti",
|
|
"DM": "Dominica",
|
|
"DO": "Dominican Republic",
|
|
"EC": "Ecuador",
|
|
"EG": "Egypt",
|
|
"SV": "El Salvador",
|
|
"GQ": "Equatorial Guinea",
|
|
"ER": "Eritrea",
|
|
"EE": "Estonia",
|
|
"ET": "Ethiopia",
|
|
"FK": "Falkland Islands (Malvinas)",
|
|
"FO": "Faroe Islands",
|
|
"FJ": "Fiji",
|
|
"FI": "Finland",
|
|
"FR": "France",
|
|
"GF": "French Guiana",
|
|
"PF": "French Polynesia",
|
|
"TF": "French Southern Territories",
|
|
"GA": "Gabon",
|
|
"GM": "Gambia",
|
|
"GE": "Georgia",
|
|
"DE": "Germany",
|
|
"GH": "Ghana",
|
|
"GI": "Gibraltar",
|
|
"GR": "Greece",
|
|
"GL": "Greenland",
|
|
"GD": "Grenada",
|
|
"GP": "Guadeloupe",
|
|
"GU": "Guam",
|
|
"GT": "Guatemala",
|
|
"GG": "Guernsey",
|
|
"GN": "Guinea",
|
|
"GW": "Guinea-Bissau",
|
|
"GY": "Guyana",
|
|
"HT": "Haiti",
|
|
"HM": "Heard Island and McDonald Islands",
|
|
"VA": "Holy See (Vatican City State)",
|
|
"HN": "Honduras",
|
|
"HK": "Hong Kong",
|
|
"HU": "Hungary",
|
|
"IS": "Iceland",
|
|
"IN": "India",
|
|
"ID": "Indonesia",
|
|
"IR": "Iran",
|
|
"IQ": "Iraq",
|
|
"IE": "Ireland",
|
|
"IM": "Isle of Man",
|
|
"IL": "Israel",
|
|
"IT": "Italy",
|
|
"JM": "Jamaica",
|
|
"JP": "Japan",
|
|
"JE": "Jersey",
|
|
"JO": "Jordan",
|
|
"KZ": "Kazakhstan",
|
|
"KE": "Kenya",
|
|
"KI": "Kiribati",
|
|
"KP": "Korea",
|
|
"KR": "Korea",
|
|
"KW": "Kuwait",
|
|
"KG": "Kyrgyzstan",
|
|
"LA": "Lao People's Democratic Republic",
|
|
"LV": "Latvia",
|
|
"LB": "Lebanon",
|
|
"LS": "Lesotho",
|
|
"LR": "Liberia",
|
|
"LY": "Libya",
|
|
"LI": "Liechtenstein",
|
|
"LT": "Lithuania",
|
|
"LU": "Luxembourg",
|
|
"MO": "Macao",
|
|
"MK": "Macedonia",
|
|
"MG": "Madagascar",
|
|
"MW": "Malawi",
|
|
"MY": "Malaysia",
|
|
"MV": "Maldives",
|
|
"ML": "Mali",
|
|
"MT": "Malta",
|
|
"MH": "Marshall Islands",
|
|
"MQ": "Martinique",
|
|
"MR": "Mauritania",
|
|
"MU": "Mauritius",
|
|
"YT": "Mayotte",
|
|
"MX": "Mexico",
|
|
"FM": "Micronesia",
|
|
"MD": "Moldova",
|
|
"MC": "Monaco",
|
|
"MN": "Mongolia",
|
|
"ME": "Montenegro",
|
|
"MS": "Montserrat",
|
|
"MA": "Morocco",
|
|
"MZ": "Mozambique",
|
|
"MM": "Myanmar",
|
|
"NA": "Namibia",
|
|
"NR": "Nauru",
|
|
"NP": "Nepal",
|
|
"NL": "Netherlands",
|
|
"NC": "New Caledonia",
|
|
"NZ": "New Zealand",
|
|
"NI": "Nicaragua",
|
|
"NE": "Niger",
|
|
"NG": "Nigeria",
|
|
"NU": "Niue",
|
|
"NF": "Norfolk Island",
|
|
"MP": "Northern Mariana Islands",
|
|
"NO": "Norway",
|
|
"OM": "Oman",
|
|
"PK": "Pakistan",
|
|
"PW": "Palau",
|
|
"PS": "Palestine",
|
|
"PA": "Panama",
|
|
"PG": "Papua New Guinea",
|
|
"PY": "Paraguay",
|
|
"PE": "Peru",
|
|
"PH": "Philippines",
|
|
"PN": "Pitcairn",
|
|
"PL": "Poland",
|
|
"PT": "Portugal",
|
|
"PR": "Puerto Rico",
|
|
"QA": "Qatar",
|
|
"RE": "Réunion",
|
|
"RO": "Romania",
|
|
"RU": "Russian Federation",
|
|
"RW": "Rwanda",
|
|
"BL": "Saint Barthélemy",
|
|
"SH": "Saint Helena",
|
|
"KN": "Saint Kitts and Nevis",
|
|
"LC": "Saint Lucia",
|
|
"MF": "Saint Martin (French part)",
|
|
"PM": "Saint Pierre and Miquelon",
|
|
"VC": "Saint Vincent and the Grenadines",
|
|
"WS": "Samoa",
|
|
"SM": "San Marino",
|
|
"ST": "Sao Tome and Principe",
|
|
"SA": "Saudi Arabia",
|
|
"SN": "Senegal",
|
|
"RS": "Serbia",
|
|
"SC": "Seychelles",
|
|
"SL": "Sierra Leone",
|
|
"SG": "Singapore",
|
|
"SX": "Sint Maarten (Dutch part)",
|
|
"SK": "Slovakia",
|
|
"SI": "Slovenia",
|
|
"SB": "Solomon Islands",
|
|
"SO": "Somalia",
|
|
"ZA": "South Africa",
|
|
"GS": "South Georgia and the South Sandwich Islands",
|
|
"SS": "South Sudan",
|
|
"ES": "Spain",
|
|
"LK": "Sri Lanka",
|
|
"SD": "Sudan",
|
|
"SR": "Suriname",
|
|
"SJ": "Svalbard and Jan Mayen",
|
|
"SZ": "Swaziland",
|
|
"SE": "Sweden",
|
|
"CH": "Switzerland",
|
|
"SY": "Syrian Arab Republic",
|
|
"TW": "Taiwan",
|
|
"TJ": "Tajikistan",
|
|
"TZ": "Tanzania",
|
|
"TH": "Thailand",
|
|
"TL": "Timor-Leste",
|
|
"TG": "Togo",
|
|
"TK": "Tokelau",
|
|
"TO": "Tonga",
|
|
"TT": "Trinidad and Tobago",
|
|
"TN": "Tunisia",
|
|
"TR": "Turkey",
|
|
"TM": "Turkmenistan",
|
|
"TC": "Turks and Caicos Islands",
|
|
"TV": "Tuvalu",
|
|
"UG": "Uganda",
|
|
"UA": "Ukraine",
|
|
"AE": "United Arab Emirates",
|
|
"GB": "United Kingdom",
|
|
"US": "United States of America",
|
|
"UM": "United States Minor Outlying Islands",
|
|
"UY": "Uruguay",
|
|
"UZ": "Uzbekistan",
|
|
"VU": "Vanuatu",
|
|
"VE": "Venezuela",
|
|
"VN": "Viet Nam",
|
|
"VG": "Virgin Islands",
|
|
"VI": "Virgin Islands",
|
|
"WF": "Wallis and Futuna",
|
|
"EH": "Western Sahara",
|
|
"YE": "Yemen",
|
|
"ZM": "Zambia",
|
|
"ZW": "Zimbabwe",
|
|
}
|
|
var country = res.Get("country").String()
|
|
var prov = res.Get("region").String()
|
|
var city = res.Get("city").String()
|
|
var district = ""
|
|
if util.StringInSlice(country, []string{"TW", "MO", "HK"}) {
|
|
district = prov + " " + city
|
|
city = countryMap[country]
|
|
prov = ""
|
|
country = "CN"
|
|
}
|
|
country = countryMap[country]
|
|
|
|
var anycast = false
|
|
if res.Get("anycast").String() == "true" {
|
|
country = "ANYCAST"
|
|
prov = "ANYCAST"
|
|
city = ""
|
|
anycast = true
|
|
}
|
|
|
|
i := strings.Index(res.Get("org").String(), " ")
|
|
var owner string
|
|
if i == -1 {
|
|
owner = ""
|
|
} else {
|
|
owner = res.Get("org").String()[i:]
|
|
}
|
|
|
|
var asnumber = ""
|
|
// 有时候不返回asn或其本身没有asn
|
|
if strings.HasPrefix(res.Get("org").String(), "AS") {
|
|
asnumber = strings.Fields(strings.TrimPrefix(res.Get("org").String(), "AS"))[0]
|
|
}
|
|
|
|
//"loc": "34.0522,-118.2437",
|
|
var lat, lng float64
|
|
if res.Get("loc").String() != "" {
|
|
lat, _ = strconv.ParseFloat(strings.Split(res.Get("loc").String(), ",")[0], 32)
|
|
lng, _ = strconv.ParseFloat(strings.Split(res.Get("loc").String(), ",")[1], 32)
|
|
}
|
|
if anycast {
|
|
lat, lng = 0, 0
|
|
}
|
|
|
|
return &IPGeoData{
|
|
Asnumber: asnumber,
|
|
Country: country,
|
|
City: city,
|
|
Prov: prov,
|
|
District: district,
|
|
Owner: owner,
|
|
Lat: lat,
|
|
Lng: lng,
|
|
}, nil
|
|
}
|