2024-02-13 11:11:10 +00:00
|
|
|
package api
|
|
|
|
|
|
|
|
|
|
import (
|
|
|
|
|
"errors"
|
|
|
|
|
"fmt"
|
|
|
|
|
"net/http"
|
|
|
|
|
"strings"
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
type Provider string
|
|
|
|
|
|
|
|
|
|
const (
|
2024-10-06 06:30:33 -07:00
|
|
|
Cloudflare Provider = "cloudflare"
|
2024-10-08 19:03:10 +00:00
|
|
|
IfConfigCo Provider = "ifconfigco"
|
2024-02-14 07:35:39 +00:00
|
|
|
IPInfo Provider = "ipinfo"
|
|
|
|
|
IP2Location Provider = "ip2location"
|
2024-02-13 11:11:10 +00:00
|
|
|
)
|
|
|
|
|
|
2024-10-19 15:21:14 +02:00
|
|
|
type NameToken struct {
|
|
|
|
|
Name string
|
|
|
|
|
Token string
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func New(nameTokenPairs []NameToken, client *http.Client) (
|
|
|
|
|
fetchers []Fetcher, err error,
|
2024-10-11 19:20:48 +00:00
|
|
|
) {
|
2024-10-19 15:21:14 +02:00
|
|
|
fetchers = make([]Fetcher, len(nameTokenPairs))
|
|
|
|
|
for i, nameTokenPair := range nameTokenPairs {
|
|
|
|
|
provider, err := ParseProvider(nameTokenPair.Name)
|
|
|
|
|
if err != nil {
|
|
|
|
|
return nil, fmt.Errorf("parsing API name: %w", err)
|
|
|
|
|
}
|
|
|
|
|
switch provider {
|
|
|
|
|
case Cloudflare:
|
|
|
|
|
fetchers[i] = newCloudflare(client)
|
|
|
|
|
case IfConfigCo:
|
|
|
|
|
fetchers[i] = newIfConfigCo(client)
|
|
|
|
|
case IPInfo:
|
|
|
|
|
fetchers[i] = newIPInfo(client, nameTokenPair.Token)
|
|
|
|
|
case IP2Location:
|
|
|
|
|
fetchers[i] = newIP2Location(client, nameTokenPair.Token)
|
|
|
|
|
default:
|
|
|
|
|
panic("provider not valid: " + provider)
|
|
|
|
|
}
|
2024-02-13 11:11:10 +00:00
|
|
|
}
|
2024-10-19 15:21:14 +02:00
|
|
|
return fetchers, nil
|
2024-02-13 11:11:10 +00:00
|
|
|
}
|
|
|
|
|
|
2024-10-11 19:20:48 +00:00
|
|
|
var ErrProviderNotValid = errors.New("API name is not valid")
|
2024-02-13 11:11:10 +00:00
|
|
|
|
|
|
|
|
func ParseProvider(s string) (provider Provider, err error) {
|
|
|
|
|
switch strings.ToLower(s) {
|
2024-10-06 06:30:33 -07:00
|
|
|
case "cloudflare":
|
|
|
|
|
return Cloudflare, nil
|
2024-10-08 19:03:10 +00:00
|
|
|
case string(IfConfigCo):
|
|
|
|
|
return IfConfigCo, nil
|
2024-02-13 11:11:10 +00:00
|
|
|
case "ipinfo":
|
|
|
|
|
return IPInfo, nil
|
2024-02-14 07:35:39 +00:00
|
|
|
case "ip2location":
|
|
|
|
|
return IP2Location, nil
|
2024-02-13 11:11:10 +00:00
|
|
|
default:
|
2024-10-08 19:03:10 +00:00
|
|
|
return "", fmt.Errorf(`%w: %q can only be "cloudflare", "ifconfigco", "ip2location" or "ipinfo"`,
|
2024-02-13 11:11:10 +00:00
|
|
|
ErrProviderNotValid, s)
|
|
|
|
|
}
|
|
|
|
|
}
|