mirror of
https://github.com/qdm12/gluetun.git
synced 2026-05-07 04:20:12 +02:00
Quentin McGaw
4a78989d9d
- main reason being it's a burden to always define sentinel errors at global scope, wrap them with `%w` instead of using a string directly - only use sentinel errors when it has to be checked using `errors.Is` - replace all usage of these sentinel errors in `fmt.Errorf` with direct strings that were in the sentinel error - exclude the sentinel error definition requirement from .golangci.yml - update unit tests to use ContainersError instead of ErrorIs so it stays as a "not a change detector test" without requiring a sentinel error
121 lines
2.9 KiB
Go
121 lines
2.9 KiB
Go
package api
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"net/http"
|
|
"net/netip"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/qdm12/gluetun/internal/constants"
|
|
"github.com/qdm12/gluetun/internal/models"
|
|
)
|
|
|
|
type ipInfo struct {
|
|
client *http.Client
|
|
token string
|
|
}
|
|
|
|
func newIPInfo(client *http.Client, token string) *ipInfo {
|
|
return &ipInfo{
|
|
client: client,
|
|
token: token,
|
|
}
|
|
}
|
|
|
|
func (i *ipInfo) String() string {
|
|
return string(IPInfo)
|
|
}
|
|
|
|
func (i *ipInfo) CanFetchAnyIP() bool {
|
|
return true
|
|
}
|
|
|
|
func (i *ipInfo) Token() string {
|
|
return i.token
|
|
}
|
|
|
|
// FetchInfo obtains information on the ip address provided
|
|
// using the ipinfo.io API. If the ip is the zero value, the public IP address
|
|
// of the machine is used as the IP.
|
|
func (i *ipInfo) FetchInfo(ctx context.Context, ip netip.Addr) (
|
|
result models.PublicIP, err error,
|
|
) {
|
|
// Define a timeout since the default client has a large timeout and we don't
|
|
// want to wait too long.
|
|
const timeout = 15 * time.Second
|
|
ctx, cancel := context.WithTimeout(ctx, timeout)
|
|
defer cancel()
|
|
|
|
url := "https://ipinfo.io/"
|
|
switch {
|
|
case ip.Is6():
|
|
url = "https://v6.ipinfo.io/" + ip.String()
|
|
case ip.Is4():
|
|
url = "https://ipinfo.io/" + ip.String()
|
|
}
|
|
|
|
request, err := http.NewRequestWithContext(ctx, http.MethodGet, url, nil)
|
|
if err != nil {
|
|
return result, err
|
|
}
|
|
request.Header.Set("Authorization", "Bearer "+i.token)
|
|
|
|
response, err := i.client.Do(request)
|
|
if err != nil {
|
|
return result, err
|
|
}
|
|
defer response.Body.Close()
|
|
|
|
if i.token != "" && response.StatusCode == http.StatusUnauthorized {
|
|
return result, fmt.Errorf("token is not valid: %s", response.Status)
|
|
}
|
|
|
|
switch response.StatusCode {
|
|
case http.StatusOK:
|
|
case http.StatusTooManyRequests, http.StatusForbidden:
|
|
return result, fmt.Errorf("%w from %s: %d %s",
|
|
ErrTooManyRequests, url, response.StatusCode, response.Status)
|
|
default:
|
|
return result, fmt.Errorf("bad HTTP status received from %s: %d %s",
|
|
url, response.StatusCode, response.Status)
|
|
}
|
|
|
|
decoder := json.NewDecoder(response.Body)
|
|
var data struct {
|
|
IP netip.Addr `json:"ip,omitempty"`
|
|
Region string `json:"region,omitempty"`
|
|
Country string `json:"country,omitempty"`
|
|
City string `json:"city,omitempty"`
|
|
Hostname string `json:"hostname,omitempty"`
|
|
Loc string `json:"loc,omitempty"`
|
|
Org string `json:"org,omitempty"`
|
|
Postal string `json:"postal,omitempty"`
|
|
Timezone string `json:"timezone,omitempty"`
|
|
}
|
|
if err := decoder.Decode(&data); err != nil {
|
|
return result, fmt.Errorf("decoding response: %w", err)
|
|
}
|
|
|
|
countryCode := strings.ToLower(data.Country)
|
|
country, ok := constants.CountryCodes()[countryCode]
|
|
if ok {
|
|
data.Country = country
|
|
}
|
|
|
|
result = models.PublicIP{
|
|
IP: data.IP,
|
|
Region: data.Region,
|
|
Country: data.Country,
|
|
City: data.City,
|
|
Hostname: data.Hostname,
|
|
Location: data.Loc,
|
|
Organization: data.Org,
|
|
PostalCode: data.Postal,
|
|
Timezone: data.Timezone,
|
|
}
|
|
return result, nil
|
|
}
|