dba861d262
1. we want optionally to log the body (we don't want to log the body when we're fetching psiphon secrets or tor targets) 2. we want body logging to _also_ happen on error since this is quite useful to debug possible errors when accessing the API This diff adds the above functionality, which were previously described in https://github.com/ooni/probe/issues/1951. This diff also adds comprehensive testing.
35 lines
763 B
Go
35 lines
763 B
Go
package geolocate
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
|
|
"github.com/ooni/probe-cli/v3/internal/engine/httpheader"
|
|
"github.com/ooni/probe-cli/v3/internal/engine/httpx"
|
|
"github.com/ooni/probe-cli/v3/internal/model"
|
|
)
|
|
|
|
type ipInfoResponse struct {
|
|
IP string `json:"ip"`
|
|
}
|
|
|
|
func ipInfoIPLookup(
|
|
ctx context.Context,
|
|
httpClient *http.Client,
|
|
logger model.Logger,
|
|
userAgent string,
|
|
) (string, error) {
|
|
var v ipInfoResponse
|
|
err := (&httpx.APIClientTemplate{
|
|
Accept: "application/json",
|
|
BaseURL: "https://ipinfo.io",
|
|
HTTPClient: httpClient,
|
|
Logger: logger,
|
|
UserAgent: httpheader.CLIUserAgent(), // we must be a CLI client
|
|
}).WithBodyLogging().Build().GetJSON(ctx, "/", &v)
|
|
if err != nil {
|
|
return DefaultProbeIP, err
|
|
}
|
|
return v.IP, nil
|
|
}
|