Merge pull request #174 from sonroyaalmerol/check-redirect

Immediately close response body on client disconnect

proxy/stream_handler.go

@@ -364,7 +364,6 @@ func Handler(w http.ResponseWriter, r *http.Request) {
 				cancel()
 			}
 
-			_, _ = io.Copy(io.Discard, resp.Body)
 			resp.Body.Close()
 		}
 	}

utils/http.go

@@ -11,7 +11,13 @@ func CustomHttpRequest(method string, url string) (*http.Response, error) {
 	userAgent := GetEnv("USER_AGENT")
 
 	// Create a new HTTP client with a custom User-Agent header
-	client := &http.Client{}
+	client := &http.Client{
+		CheckRedirect: func(req *http.Request, via []*http.Request) error {
+			// Follow redirects while preserving the custom User-Agent header
+			req.Header.Set("User-Agent", userAgent)
+			return nil
+		},
+	}
 
 	req, err := http.NewRequest(method, url, nil)
 	if err != nil {