multi-protocol client

Fixes #4

1 files changed, 7 insertions, 3 deletions

diff --git a/gemini/client.go b/gemini/client.go
index 34d5839..0a621dd 100644
--- a/gemini/client.go
+++ b/gemini/client.go
@@ -54,7 +54,7 @@ func (client Client) RoundTrip(request *types.Request) (*types.Response, error)
 	}
 
 	tlsConf := client.tlsConf
-	if (tlsConf == nil) {
+	if tlsConf == nil {
 		tlsConf = &tls.Config{InsecureSkipVerify: true}
 	}
 
@@ -102,12 +102,12 @@ func (c Client) Fetch(url string) (*types.Response, error) {
 		if err != nil {
 			return nil, err
 		}
-		if ResponseCategoryForStatus(response.Status) != ResponseCategoryRedirect {
+		if !c.IsRedirect(response) {
 			return response, nil
 		}
 
 		prev := u
-		u, err = neturl.Parse(url)
+		u, err = neturl.Parse(response.Meta.(string))
 		if err != nil {
 			return nil, err
 		}
@@ -116,3 +116,7 @@ func (c Client) Fetch(url string) (*types.Response, error) {
 
 	return nil, ExceededMaxRedirects
 }
+
+func (c Client) IsRedirect(response *types.Response) bool {
+	return ResponseCategoryForStatus(response.Status) == ResponseCategoryRedirect
+}