package service

import (
	"api/api/network/model"
	"fmt"
	"github.com/lionsoul2014/ip2region/binding/golang/xdb"
	"log"
	"strings"
)

var (
	searcher *xdb.Searcher
)

type IpService struct{}

func init() {
	var dbPath = "config/google.xdb"
	// 1、从 dbPath 加载整个 xdb 到内存
	cBuff, err := xdb.LoadContentFromFile(dbPath)
	if err != nil {
		fmt.Printf("failed to load content from `%s`: %s\n", dbPath, err)
		if err != nil {
			log.Panic("dbPath初始化失败: ", err)
		}
	}

	// 2、用全局的 cBuff 创建完全基于内存的查询对象。
	searcher, err = xdb.NewWithBuffer(cBuff)
	if err != nil {
		fmt.Printf("failed to create searcher with content: %s\n", err)
		if err != nil {
			log.Panic("searcher初始化失败: ", err)
		}
	}
}

// GetIpInfo 获取IP信息
func (*IpService) GetIpInfo(ip string) (model.IpInfo, error) {
	region, err := searcher.SearchByStr(ip)
	if err != nil {
		return model.IpInfo{}, err
	}
	info := strings.Split(region, "|")
	var ipInfo = model.IpInfo{
		IP:       ip,
		Country:  info[0],
		Region:   info[1],
		Province: info[2],
		City:     info[3],
		ISP:      info[4],
	}
	return ipInfo, nil
}