posthog/livestream/geoip.go

package main

import (
	"errors"
	"net"

	"github.com/oschwald/maxminddb-golang"
)

type GeoLocator struct {
	db *maxminddb.Reader
}

func NewGeoLocator(dbPath string) (*GeoLocator, error) {
	db, err := maxminddb.Open(dbPath)
	if err != nil {
		return nil, err
	}

	return &GeoLocator{
		db: db,
	}, nil
}

func (g *GeoLocator) Lookup(ipString string) (float64, float64, error) {
	ip := net.ParseIP(ipString)
	if ip == nil {
		return 0, 0, errors.New("invalid IP address")
	}

	var record struct {
		Location struct {
			Latitude  float64 `maxminddb:"latitude"`
			Longitude float64 `maxminddb:"longitude"`
		} `maxminddb:"location"`
	}

	err := g.db.Lookup(ip, &record)
	if err != nil {
		return 0, 0, err
	}
	return record.Location.Latitude, record.Location.Longitude, nil
}