mirror of
https://github.com/therootcompany/golib.git
synced 2026-11-03 22:44:58 +00:00
AJ ONeal
2f33b5e430
Add a Go package for reading MaxMind GeoLite2 City and ASN databases.
- ParseConf: parses geoipupdate-style config files (AccountID,
LicenseKey, EditionIDs, DatabaseDirectory)
- Databases: holds open City + ASN readers; Open() loads both
editions from a directory, extracting .mmdb from tar.gz in memory
without writing to disk
- FindTarGz: resolves cache paths, preferring *_LATEST.tar.gz
- Lookup: returns city, region, country, ASN, and AS org for an IP
- DownloadBase constant, TarGzName helper, DefaultConfPaths and
DefaultCacheDir conveniences
Integration tests verify download and conditional-get freshness.
60 lines
1.6 KiB
Go
60 lines
1.6 KiB
Go
package geoip
|
|
|
|
import (
|
|
"bufio"
|
|
"errors"
|
|
"strings"
|
|
)
|
|
|
|
// Conf holds the fields parsed from a geoipupdate-style config file.
|
|
//
|
|
// GeoLite2 is free: sign up at https://www.maxmind.com/en/geolite2/signup
|
|
// to get an AccountID and generate a LicenseKey, then write them to
|
|
// GeoIP.conf alongside the desired EditionIDs:
|
|
//
|
|
// AccountID 123456
|
|
// LicenseKey xxxxxxxxxxxxxxxx
|
|
// EditionIDs GeoLite2-City GeoLite2-ASN
|
|
type Conf struct {
|
|
AccountID string
|
|
LicenseKey string
|
|
EditionIDs []string
|
|
DatabaseDirectory string
|
|
}
|
|
|
|
// ErrMissingCredentials is returned by ParseConf when AccountID or LicenseKey
|
|
// is absent from the input.
|
|
var ErrMissingCredentials = errors.New("AccountID and LicenseKey are required")
|
|
|
|
// ParseConf parses a geoipupdate-style config (whitespace-separated key/value
|
|
// pairs, # comments). Compatible with GeoIP.conf files used by the official
|
|
// geoipupdate tool.
|
|
func ParseConf(s string) (*Conf, error) {
|
|
kv := make(map[string]string)
|
|
scanner := bufio.NewScanner(strings.NewReader(s))
|
|
for scanner.Scan() {
|
|
line := strings.TrimSpace(scanner.Text())
|
|
if line == "" || strings.HasPrefix(line, "#") {
|
|
continue
|
|
}
|
|
key, value, _ := strings.Cut(line, " ")
|
|
kv[strings.TrimSpace(key)] = strings.TrimSpace(value)
|
|
}
|
|
if err := scanner.Err(); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
c := &Conf{
|
|
AccountID: kv["AccountID"],
|
|
LicenseKey: kv["LicenseKey"],
|
|
DatabaseDirectory: kv["DatabaseDirectory"],
|
|
}
|
|
if c.AccountID == "" || c.LicenseKey == "" {
|
|
return nil, ErrMissingCredentials
|
|
}
|
|
if ids := kv["EditionIDs"]; ids != "" {
|
|
c.EditionIDs = strings.Fields(ids)
|
|
}
|
|
return c, nil
|
|
}
|