handlers: add headers as queryparam for pull sources

livepeer · Feb 1, 2024 · 7ffcd02 · 7ffcd02
1 parent 316bac6
commit 7ffcd02
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/handlers/geolocation/geolocation.go b/handlers/geolocation/geolocation.go
@@ -169,7 +169,17 @@ func (c *GeolocationHandlersCollection) getStreamPull(playbackID string) (string
 		return "", nil
 	}
 
-	return stream.Pull.Source, nil
+	if len(stream.Pull.Headers) == 0 {
+		return stream.Pull.Source, nil
+	}
+
+	var params []string
+	for k, v := range stream.Pull.Headers {
+		param := "addheader=" + url.QueryEscape(k+":"+v)
+		params = append(params, param)
+	}
+	finalPullURL := stream.Pull.Source + "?" + strings.Join(params, "&")
+	return finalPullURL, nil
 }
 
 func parsePlus(plusString string) (string, string) {