mirror of
https://github.com/zu1k/nali.git
synced 2025-01-22 21:29:02 +08:00
70 lines
1.2 KiB
Go
70 lines
1.2 KiB
Go
package geoip
|
||
|
||
import (
|
||
"errors"
|
||
"fmt"
|
||
"log"
|
||
"net"
|
||
"os"
|
||
|
||
"github.com/spf13/viper"
|
||
|
||
"github.com/oschwald/geoip2-golang"
|
||
)
|
||
|
||
// GeoIP2
|
||
type GeoIP struct {
|
||
db *geoip2.Reader
|
||
}
|
||
|
||
// new geoip from database file
|
||
func NewGeoIP(filePath string) (*GeoIP, error) {
|
||
// 判断文件是否存在
|
||
_, err := os.Stat(filePath)
|
||
if err != nil && os.IsNotExist(err) {
|
||
log.Println("文件不存在，请自行下载 Geoip2 City库，并保存在", filePath)
|
||
return nil, err
|
||
} else {
|
||
db, err := geoip2.Open(filePath)
|
||
if err != nil {
|
||
log.Fatal(err)
|
||
}
|
||
return &GeoIP{db: db}, nil
|
||
}
|
||
}
|
||
|
||
func (g GeoIP) Find(query string, params ...string) (result fmt.Stringer, err error) {
|
||
ip := net.ParseIP(query)
|
||
if ip == nil {
|
||
return nil, errors.New("Query should be valid IP")
|
||
}
|
||
record, err := g.db.City(ip)
|
||
if err != nil {
|
||
return
|
||
}
|
||
|
||
lang := viper.GetString("selected.lang")
|
||
if lang == "" {
|
||
lang = "zh-CN"
|
||
}
|
||
|
||
result = Result{
|
||
Country: record.Country.Names[lang],
|
||
City: record.City.Names[lang],
|
||
}
|
||
return
|
||
}
|
||
|
||
type Result struct {
|
||
Country string
|
||
City string
|
||
}
|
||
|
||
func (r Result) String() string {
|
||
if r.City == "" {
|
||
return r.Country
|
||
} else {
|
||
return fmt.Sprintf("%s %s", r.Country, r.City)
|
||
}
|
||
}
|