Support attaching oauth token in userdata on track (#110)

* Support injecting oauth token from userdata on track

* remove unused import

* adress PR comments, change context name, catch userdata json parting error

* Move userdata oauth token checks to the context filter

* use JsonBrowser to parse userdata, fix extra tab

* add userData oauth to Readme

Author: kikkia

README.md

@@ -256,6 +256,17 @@ plugins:
       # skipInitialization: true
 ```
 
+### Passing an oauth token from your client
+Another option to use oauth is by using oauth access tokens that are managed from your client. In this case your 
+bot/client provides LavaLink with the token to use when playing a track. To do this simply add the oauth access token 
+to a track's [userData](https://lavalink.dev/api/rest#track) field in a json format when updating the player to 
+play a track like:
+```json
+{
+  "oauth-token": "access token to use"
+}
+```
+
 ## Using a `poToken`
 A `poToken`, also known as a "Proof of Origin Token" is a way to identify what requests originate from.
 In YouTube's case, this is sent as a JavaScript challenge that browsers must evaluate, and send back the resolved

common/src/main/java/dev/lavalink/youtube/http/YoutubeHttpContextFilter.java

@@ -14,6 +14,7 @@
 import org.slf4j.LoggerFactory;
 
 import static com.sedmelluq.discord.lavaplayer.tools.FriendlyException.Severity.COMMON;
+import static dev.lavalink.youtube.http.YoutubeOauth2Handler.OAUTH_INJECT_CONTEXT_ATTRIBUTE;
 
 public class YoutubeHttpContextFilter extends BaseYoutubeHttpContextFilter {
   private static final Logger log = LoggerFactory.getLogger(YoutubeHttpContextFilter.class);
@@ -83,8 +84,14 @@ public void onRequest(HttpClientContext context,
       boolean isRequestFromOauthedClient = context.getAttribute(Client.OAUTH_CLIENT_ATTRIBUTE) == Boolean.TRUE;
 
       if (isRequestFromOauthedClient && Client.PLAYER_URL.equals(request.getURI().toString())) {
+        // Look at the userdata for any provided oauth-token
+        String oauthToken = context.getAttribute(OAUTH_INJECT_CONTEXT_ATTRIBUTE, String.class);
         // only apply the token to /player requests.
-        oauth2Handler.applyToken(request);
+        if (oauthToken != null && !oauthToken.isEmpty()) {
+          oauth2Handler.applyToken(request, oauthToken);
+        } else {
+          oauth2Handler.applyToken(request);
+        }
       }
     }
 

common/src/main/java/dev/lavalink/youtube/http/YoutubeOauth2Handler.java

@@ -35,6 +35,7 @@ public class YoutubeOauth2Handler {
     private static final String CLIENT_SECRET = "SboVhoG9s0rNafixCSGGKXAT";
     private static final String SCOPES = "http://gdata.youtube.com https://www.googleapis.com/auth/youtube";
     private static final String OAUTH_FETCH_CONTEXT_ATTRIBUTE = "yt-oauth";
+    public static final String OAUTH_INJECT_CONTEXT_ATTRIBUTE = "yt-oauth-token";
 
     private final HttpInterfaceManager httpInterfaceManager;
 
@@ -289,6 +290,10 @@ public void applyToken(HttpUriRequest request) {
         }
     }
 
+    public void applyToken(HttpUriRequest request, String token) {
+        request.setHeader("Authorization", String.format("%s %s", "Bearer", token));
+    }
+
     private HttpInterface getHttpInterface() {
         HttpInterface httpInterface = httpInterfaceManager.getInterface();
         httpInterface.getContext().setAttribute(OAUTH_FETCH_CONTEXT_ATTRIBUTE, true);

common/src/main/java/dev/lavalink/youtube/track/YoutubeAudioTrack.java

@@ -6,6 +6,7 @@
 import com.sedmelluq.discord.lavaplayer.tools.ExceptionTools;
 import com.sedmelluq.discord.lavaplayer.tools.FriendlyException;
 import com.sedmelluq.discord.lavaplayer.tools.FriendlyException.Severity;
+import com.sedmelluq.discord.lavaplayer.tools.JsonBrowser;
 import com.sedmelluq.discord.lavaplayer.tools.io.HttpInterface;
 import com.sedmelluq.discord.lavaplayer.track.AudioTrack;
 import com.sedmelluq.discord.lavaplayer.track.AudioTrackInfo;
@@ -24,6 +25,7 @@
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
+import java.io.IOException;
 import java.net.URI;
 import java.net.URISyntaxException;
 import java.util.Arrays;
@@ -32,6 +34,7 @@
 import static com.sedmelluq.discord.lavaplayer.container.Formats.MIME_AUDIO_WEBM;
 import static com.sedmelluq.discord.lavaplayer.tools.DataFormatTools.decodeUrlEncodedItems;
 import static com.sedmelluq.discord.lavaplayer.tools.Units.CONTENT_LENGTH_UNKNOWN;
+import static dev.lavalink.youtube.http.YoutubeOauth2Handler.OAUTH_INJECT_CONTEXT_ATTRIBUTE;
 
 /**
  * Audio track that handles processing Youtube videos as audio tracks.
@@ -66,6 +69,14 @@ public void process(LocalAudioTrackExecutor localExecutor) throws Exception {
     }
 
     try (HttpInterface httpInterface = sourceManager.getInterface()) {
+      try {
+        JsonBrowser userData = JsonBrowser.parse(getUserData().toString());
+        if (userData.get("oauth-token") != null) {
+          httpInterface.getContext().setAttribute(OAUTH_INJECT_CONTEXT_ATTRIBUTE, userData.get("oauth-token").text());
+        }
+      } catch (IOException e) {
+        log.debug("Failed to parse token from userData", e);
+      }
       Exception lastException = null;
 
       for (Client client : clients) {