Add proxy support

davecheney · Sep 28, 2016 · 004e1ba · 004e1ba
1 parent 2bb6516
commit 004e1ba
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/main.go b/main.go
@@ -217,7 +217,14 @@ func visit(url *url.URL) {
 
 	req = req.WithContext(httptrace.WithClientTrace(context.Background(), trace))
 
-	tr := &http.Transport{}
+	proxyURL, err := http.ProxyFromEnvironment(req)
+	if err != nil {
+		log.Fatalf("error retrieving proxy from environment: %v", err)
+	}
+
+	tr := &http.Transport{
+		Proxy: http.ProxyURL(proxyURL),
+	}
 	if scheme == "https" {
 		tr.TLSClientConfig = &tls.Config{
 			ServerName:         host,