feat: add configurable peeringdb-url for local cache

Author: natesales

docs/docs/configuration.md

@@ -101,6 +101,14 @@ Global BIRD configuration
 |------|---------|------------|
 | string   |       |          |
 
+### `peeringdb-url`
+
+PeeringDB API URL, can be set to a local PeeringDB cache server
+
+| Type | Default | Validation |
+|------|---------|------------|
+| string   | https://peeringdb.com/api/      |          |
+
 ### `blocklist`
 
 List of ASNs, prefixes, and IP addresses to block

pkg/config/config.go

@@ -332,6 +332,7 @@ type Config struct {
 	WebUIFile             string `yaml:"web-ui-file" description:"File to write web UI to (disabled if empty)" default:""`
 	LogFile               string `yaml:"log-file" description:"Log file location" default:"syslog"`
 	GlobalConfig          string `yaml:"global-config" description:"Global BIRD configuration" default:""`
+	PeeringDBURL          string `yaml:"peeringdb-url" description:"PeeringDB API URL, can be set to a local PeeringDB cache server" default:"https://peeringdb.com/api/"`
 
 	Blocklist      []string `yaml:"blocklist" description:"List of ASNs, prefixes, and IP addresses to block" default:""`
 	BlocklistURLs  []string `yaml:"blocklist-urls" description:"List of URLs to fetch blocklists from" default:""`

pkg/peeringdb/peeringdb.go

@@ -25,6 +25,9 @@ func endpoint() string {
 	}
 }
 
+// Endpoint is a public value to allow overwriting to a cache server
+var Endpoint = endpoint()
+
 type IxLanResponse struct {
 	Data []IxLanData `json:"data"`
 }
@@ -69,7 +72,7 @@ var (
 // networkInfo returns PeeringDB for an ASN
 func networkInfo(asn uint32, queryTimeout uint, apiKey string) (*Data, error) {
 	httpClient := http.Client{Timeout: time.Second * time.Duration(queryTimeout)}
-	req, err := http.NewRequest(http.MethodGet, fmt.Sprintf(endpoint()+"/net?asn=%d", asn), nil)
+	req, err := http.NewRequest(http.MethodGet, fmt.Sprintf(Endpoint+"/net?asn=%d", asn), nil)
 
 	if apiKey != "" {
 		req.Header.Add("AUTHORIZATION", "Api-Key "+apiKey)
@@ -176,7 +179,7 @@ func Update(peerData *config.Peer, queryTimeout uint, apiKey string, useCache bo
 // NeverViaRouteServers gets a list of networks that report should never be reachable via route servers
 func NeverViaRouteServers(queryTimeout uint, apiKey string) ([]uint32, error) {
 	httpClient := http.Client{Timeout: time.Second * time.Duration(queryTimeout)}
-	req, err := http.NewRequest(http.MethodGet, endpoint()+"/net?info_never_via_route_servers=1", nil)
+	req, err := http.NewRequest(http.MethodGet, Endpoint+"/net?info_never_via_route_servers=1", nil)
 	if err != nil {
 		return nil, fmt.Errorf("PeeringDB GET: %s", err)
 	}
@@ -222,7 +225,7 @@ func NeverViaRouteServers(queryTimeout uint, apiKey string) ([]uint32, error) {
 // IXLANs gets PeeringDB IX LANs for an ASN
 func IXLANs(asn uint32, peeringDbQueryTimeout uint, apiKey string) ([]IxLanData, error) {
 	httpClient := http.Client{Timeout: time.Second * time.Duration(peeringDbQueryTimeout)}
-	req, err := http.NewRequest(http.MethodGet, fmt.Sprintf(endpoint()+"/netixlan?asn=%d", asn), nil)
+	req, err := http.NewRequest(http.MethodGet, fmt.Sprintf(Endpoint+"/netixlan?asn=%d", asn), nil)
 	if err != nil {
 		return nil, fmt.Errorf("PeeringDB GET (This peer might not have a PeeringDB page): %s", err)
 	}

pkg/process/process.go

@@ -123,6 +123,9 @@ func Load(configBlob []byte) (*config.Config, error) {
 		}
 	}
 
+	// Set PeeringDB URL
+	peeringdb.Endpoint = c.PeeringDBURL
+
 	// Set hostname if empty
 	if c.Hostname == "" {
 		hostname, err := os.Hostname()