diff options
Diffstat (limited to 'client.go')
| -rw-r--r-- | client.go | 17 |
1 files changed, 11 insertions, 6 deletions
@@ -35,7 +35,10 @@ var ExceededMaxRedirects = errors.New("Client: exceeded MaxRedirects") // tlsConf may be nil, in which case gemini requests connections will not be made // with any client certificate. func NewClient(tlsConf *tls.Config) Client { - hc := httpClient{} + hc := httpClient{tp: http.DefaultTransport.(*http.Transport).Clone()} + if tlsConf != nil { + hc.tp.TLSClientConfig = tlsConf + } return Client{ protos: map[string]protocolClient{ "finger": finger.Client{}, @@ -95,12 +98,14 @@ func getRedirectLocation(proto string, meta any) string { case "gemini", "spartan": return meta.(string) case "http", "https": - return meta.(http.Header).Get("Location") + return meta.(*http.Request).Header.Get("Location") } return "" } -type httpClient struct{} +type httpClient struct{ + tp *http.Transport +} func (hc httpClient) RoundTrip(request *Request) (*Response, error) { hreq, err := http.NewRequest("GET", request.URL.String(), nil) @@ -108,18 +113,18 @@ func (hc httpClient) RoundTrip(request *Request) (*Response, error) { return nil, err } - hresp, err := http.DefaultTransport.RoundTrip(hreq) + hresp, err := hc.tp.RoundTrip(hreq) if err != nil { return nil, err } return &Response{ Status: Status(hresp.StatusCode), - Meta: hresp.Header, + Meta: hresp, Body: hresp.Body, }, nil } func (hc httpClient) IsRedirect(response *Response) bool { - return response.Meta.(http.Header).Get("Location") != "" + return response.Meta.(*http.Request).Header.Get("Location") != "" } |