Allow updating poToken and visitorData over REST

Author: devoxin

README.md

@@ -214,7 +214,7 @@ Currently, the following clients are available for use:
 ## Using OAuth Tokens
 You may notice that some requests are flagged by YouTube, causing an error message asking you to sign in to confirm you're not a bot.
 With OAuth integration, you can request that `youtube-source` use your account credentials to appear as a normal user, with varying degrees
-of efficacy. You can instruct `youtube-source` to use OAuth with the following:
+of efficacy. **You do _not_ need to use `poToken` with OAuth.**
 
 > [!WARNING]
 > Similar to the `poToken` method, this is NOT a silver bullet solution, and worst case could get your account terminated!
@@ -231,6 +231,8 @@ of efficacy. You can instruct `youtube-source` to use OAuth with the following:
 > successfully completing the OAuth flow at least **once**. If you do not see your token, you may need to configure your
 > logging (see above note).
 
+You can instruct `youtube-source` to use OAuth with the following:
+
 ### Lavaplayer
 ```java
 YoutubeAudioSourceManager source = new YoutubeAudioSourceManager();
@@ -276,6 +278,11 @@ To obtain a `poToken`, you can use https://github.com/iv-org/youtube-trusted-ses
 or the docker image. Both methods will print a `poToken` after a successful run, which you can supply to `youtube-source`
 to try and work around having automated requests blocked.
 
+
+> [!NOTE]
+> A `poToken` is not a silver bullet, and currently it only applies to requests made via the `WEB` client.
+> You do not need to specify a `poToken` if using OAuth, and vice versa.
+
 Specifying the token is as simple as doing:
 
 ### Lavaplayer
@@ -293,17 +300,22 @@ plugins:
       visitorData: "paste your visitor_data here"
 ```
 
-> [!NOTE]
-> A `poToken` is not a silver bullet, and currently it only applies to requests made via the `WEB` client.
-
 ## REST routes (`plugin` only)
 ### `POST` `/youtube`
 
 Body:
+
+> [!NOTE]
+> You do not need to provide everything as it is shown.
+> For example, you can specify just `refreshToken` and `skipInitialization`, or just `poToken` and `visitorData`.
+> You do **not** need to use `poToken` with OAuth and vice versa.
+
 ```json
 {
   "refreshToken": "your new refresh token",
-  "skipInitialization": true
+  "skipInitialization": true,
+  "poToken": "your po_token",
+  "visitorData": "your visitor_data"
 }
 ```
 

plugin/src/main/java/dev/lavalink/youtube/plugin/YoutubeRestHandler.java

@@ -1,7 +1,10 @@
 package dev.lavalink.youtube.plugin;
 
 import com.sedmelluq.discord.lavaplayer.player.AudioPlayerManager;
+import com.sedmelluq.discord.lavaplayer.tools.DataFormatTools;
 import dev.lavalink.youtube.YoutubeAudioSourceManager;
+import dev.lavalink.youtube.clients.Web;
+import dev.lavalink.youtube.plugin.rest.MinimalConfigRequest;
 import dev.lavalink.youtube.plugin.rest.MinimalConfigResponse;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
@@ -34,14 +37,26 @@ public MinimalConfigResponse getYoutubeConfig() {
 
     @PostMapping("/youtube")
     @ResponseStatus(HttpStatus.NO_CONTENT)
-    public void updateOauth(@RequestBody YoutubeOauthConfig config) {
+    public void updateYoutubeConfig(@RequestBody MinimalConfigRequest config) {
         YoutubeAudioSourceManager source = playerManager.source(YoutubeAudioSourceManager.class);
 
         if (source == null) {
             throw new ResponseStatusException(HttpStatus.BAD_REQUEST, "The YouTube source manager is not registered.");
         }
 
-        source.useOauth2(config.getRefreshToken(), config.getSkipInitialization());
-        log.debug("Updated YouTube OAuth2 refresh token to {}", config.getRefreshToken());
+        String refreshToken = config.getRefreshToken();
+
+        if (!"x".equals(refreshToken)) {
+            source.useOauth2(refreshToken, config.getSkipInitialization());
+            log.debug("Updated YouTube OAuth2 refresh token to \"{}\"", config.getRefreshToken());
+        }
+
+        String poToken = config.getPoToken();
+        String visitorData = config.getVisitorData();
+
+        if (poToken == null || visitorData == null || (!poToken.isEmpty() && !visitorData.isEmpty())) {
+            Web.setPoTokenAndVisitorData(poToken, visitorData);
+            log.debug("Updated poToken to \"{}\" and visitorData to \"{}\"", poToken, visitorData);
+        }
     }
 }

plugin/src/main/java/dev/lavalink/youtube/plugin/rest/MinimalConfigRequest.java

@@ -0,0 +1,40 @@
+package dev.lavalink.youtube.plugin.rest;
+
+public class MinimalConfigRequest {
+    private String refreshToken = "x"; // null is a valid value so we have a default placeholder.
+    private boolean skipInitialization = true;
+    private String poToken = null;
+    private String visitorData = null;
+
+    public String getRefreshToken() {
+        return this.refreshToken;
+    }
+
+    public boolean getSkipInitialization() {
+        return this.skipInitialization;
+    }
+
+    public String getPoToken() {
+        return this.poToken;
+    }
+
+    public String getVisitorData() {
+        return this.visitorData;
+    }
+
+    public void setRefreshToken(String refreshToken) {
+        this.refreshToken = refreshToken;
+    }
+
+    public void setSkipInitialization(boolean skipInitialization) {
+        this.skipInitialization = skipInitialization;
+    }
+
+    public void setPoToken(String poToken) {
+        this.poToken = poToken;
+    }
+
+    public void setVisitorData(String visitorData) {
+        this.visitorData = visitorData;
+    }
+}