// Package ip2region 基于 ip2region xdb 的 IP 归属地查询
//
// 供登录日志与操作日志的 location 字段使用。
// 数据文件为 manifest/data/ip2region.xdb（IPv4），路径可用配置 ip2region.path 覆盖。
package ip2region

import (
	"context"
	"strings"
	"sync"

	"github.com/gogf/gf/v2/frame/g"
	"github.com/lionsoul2014/ip2region/binding/golang/xdb"
)

var (
	once     sync.Once     // 保证查询器仅初始化一次
	searcher *xdb.Searcher // 归属地查询器
	mu       sync.RWMutex  // Searcher 底层共享 ReadSeeker，读操作加读锁
	loadErr  error         // 初始化错误
)

// initSearcher 懒加载查询器
//
// 失败时仅告警一次，后续查询返回空
func initSearcher() {
	once.Do(func() {
		path := "manifest/data/ip2region.xdb"
		if v, err := g.Cfg().Get(context.Background(), "ip2region.path"); err == nil && v.String() != "" {
			path = v.String()
		}
		searcher, loadErr = xdb.NewWithFileOnly(xdb.IPv4, path)
		if loadErr != nil {
			g.Log().Warningf(context.Background(), "ip2region 加载失败(%s): %v（归属地将显示为 -）", path, loadErr)
		}
	})
}

// City 根据 IP 返回城市信息
//
// 优先返回市，其次返回省，失败或内网地址返回空串。
// 区域串段位兼容两种格式，各段可能为空：
//
//	新版：国家|省|市|ISP|国家码
//	经典：国家|区域|省|市|ISP
func City(ip string) string {
	ip = strings.TrimSpace(ip)
	if ip == "" || ip == "-" {
		return ""
	}
	// 内网/回环直接归空（xdb 返回 Reserved/LAN 等无意义值）
	if isPrivateIP(ip) {
		return ""
	}
	initSearcher()
	if searcher == nil {
		return ""
	}
	mu.RLock()
	region, err := searcher.Search(ip)
	mu.RUnlock()
	if err != nil || region == "" {
		return ""
	}
	// 典型内网占位值
	for _, kw := range []string{"Reserved", "LAN", "None", "广播"} {
		if strings.Contains(region, kw) {
			return ""
		}
	}
	parts := strings.Split(region, "|")
	var prov, city string
	if len(parts) >= 4 {
		p1 := strings.TrimSpace(parts[1])
		if isISP(parts[3]) || hasAdminSuffix(p1) {
			prov, city = p1, strings.TrimSpace(parts[2]) // 新版：第二段即省份
		} else {
			prov, city = strings.TrimSpace(parts[2]), strings.TrimSpace(parts[3]) // 经典：第三段起为省/市
		}
		switch {
		case city != "" && city != "0" && prov != "" && prov != "0":
			return prov + city
		case city != "" && city != "0":
			return city
		case prov != "" && prov != "0":
			return prov
		}
	}
	for _, p := range parts {
		p = strings.TrimSpace(p)
		if p != "" && p != "0" && !isISP(p) {
			return p
		}
	}
	return ""
}

// Location 返回归属地字段值
//
// 供登录日志与操作日志使用，无归属地时返回 "-"
func Location(ip string) string {
	if c := City(ip); c != "" {
		return c
	}
	return "-"
}

// isPrivateIP 判断是否为内网或回环地址
func isPrivateIP(ip string) bool {
	return strings.HasPrefix(ip, "127.") ||
		strings.HasPrefix(ip, "10.") ||
		strings.HasPrefix(ip, "192.168.") ||
		strings.HasPrefix(ip, "172.16.") ||
		strings.HasPrefix(ip, "172.17.") || strings.HasPrefix(ip, "172.18.") ||
		strings.HasPrefix(ip, "172.19.") || strings.HasPrefix(ip, "172.2") ||
		strings.HasPrefix(ip, "172.30.") || strings.HasPrefix(ip, "172.31.") ||
		ip == "::1" || ip == "0.0.0.0"
}

// isISP 粗略识别 ISP 文本
//
// 避免将运营商名称误判为城市
func isISP(s string) bool {
	for _, kw := range []string{"电信", "联通", "移动", "铁通", "教育", "科研", "广电"} {
		if strings.Contains(s, kw) {
			return true
		}
	}
	return false
}

// hasAdminSuffix 判断是否为行政区划名称
//
// 识别省、市、自治区、特别行政区等后缀，用于识别新版段位的省份段
func hasAdminSuffix(s string) bool {
	for _, kw := range []string{"省", "市", "自治区", "特别行政区"} {
		if strings.HasSuffix(s, kw) {
			return true
		}
	}
	return false
}