package mmdb

import (
	"fmt"
	"net"

	"github.com/IncSW/geoip2"
)

type IpInfo struct {
	Ip             string `json:"ip"`
	CountryIsoCode string `json:"country_iso_code"`
	CountryName    string `json:"country_name"`
	City           string `json:"city"`
	TimeZone       string `json:"time_zone"`
}

type Reader struct {
	reader *geoip2.CityReader
}

func New(filePath string) (*Reader, error) {
	if filePath == "" {
		return nil, fmt.Errorf("mmdb file path is empty")
	}
	r, err := geoip2.NewCityReaderFromFile(filePath)
	if err != nil {
		return nil, fmt.Errorf("open mmdb: %w", err)
	}
	return &Reader{reader: r}, nil
}

func (r *Reader) QueryIP(ip string) (IpInfo, error) {
	if r == nil || r.reader == nil {
		return IpInfo{}, fmt.Errorf("mmdb reader not initialized")
	}
	record, err := r.reader.Lookup(net.ParseIP(ip))
	if err != nil {
		return IpInfo{}, fmt.Errorf("lookup ip: %w", err)
	}
	info := IpInfo{
		Ip:             ip,
		CountryIsoCode: record.Country.ISOCode,
		CountryName:    record.Country.Names["en"],
		City:           record.City.Names["en"],
		TimeZone:       record.Location.TimeZone,
	}
	return info, nil
}