add redirect logic

hasura · Oct 24, 2024 · 8870ee8 · 8870ee8
1 parent 96e8d88
commit 8870ee8
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/registry-automation/cmd/utils.go b/registry-automation/cmd/utils.go
@@ -18,7 +18,15 @@ func generateGCPObjectName(namespace, connectorName, version string) string {
 
 func downloadFile(sourceURL, destination string, headers map[string]string) error {
 	// Create a new HTTP client
-	client := &http.Client{}
+	client := &http.Client{
+		CheckRedirect: func(req *http.Request, via []*http.Request) error {
+			// Copy headers to redirected request
+			for key, values := range headers {
+				req.Header.Set(key, values)
+			}
+			return nil
+		},
+	}
 
 	// Create a new GET request
 	req, err := http.NewRequest("GET", sourceURL, nil)