Don't apply token/UA/visitor ID to googlevideo URLs

lavalink-devs · Jul 12, 2024 · 1b215bb · 1b215bb
1 parent 0fca093
commit 1b215bb
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/common/src/main/java/dev/lavalink/youtube/http/YoutubeHttpContextFilter.java b/common/src/main/java/dev/lavalink/youtube/http/YoutubeHttpContextFilter.java
@@ -67,13 +67,15 @@ public void onRequest(HttpClientContext context,
 
     String userAgent = context.getAttribute(ATTRIBUTE_USER_AGENT_SPECIFIED, String.class);
 
-    if (userAgent != null) {
-      request.setHeader("User-Agent", userAgent);
-      request.setHeader("X-Goog-Visitor-Id", tokenTracker.getVisitorId());
-      context.removeAttribute(ATTRIBUTE_USER_AGENT_SPECIFIED);
-    }
+    if (!request.getURI().getHost().contains("googlevideo")) {
+      if (userAgent != null) {
+        request.setHeader("User-Agent", userAgent);
+        request.setHeader("X-Goog-Visitor-Id", tokenTracker.getVisitorId());
+        context.removeAttribute(ATTRIBUTE_USER_AGENT_SPECIFIED);
+      }
 
-    oauth2Handler.applyToken(request);
+      oauth2Handler.applyToken(request);
+    }
 
 //    try {
 //      URI uri = new URIBuilder(request.getURI())