Allow setting pot to all supporting clients with one method call

Author: devoxin

common/src/main/java/dev/lavalink/youtube/YoutubeSource.java

@@ -1,9 +1,16 @@
 package dev.lavalink.youtube;
 
+import dev.lavalink.youtube.clients.Web;
+import dev.lavalink.youtube.clients.WebEmbedded;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
 import java.io.IOException;
 import java.io.InputStream;
 
 public class YoutubeSource {
+    private static final Logger log = LoggerFactory.getLogger(YoutubeSource.class);
+
     public static String VERSION = "Unknown";
 
     static {
@@ -22,4 +29,18 @@ public class YoutubeSource {
 
         }
     }
+
+    /**
+     * Sets the given PoToken and VisitorData pair on all POT-supporting clients.
+     * This is a convenience method to allow for setting this from one method call.
+     * @param poToken The poToken to use. This must be paired to the specified visitorData.
+     *                You may specify {@code null} to unset.
+     * @param visitorData The visitorData to use. This must be paired to the specified poToken.
+     *                    You may specify {@code null} to unset.
+     */
+    public static void setPoTokenAndVisitorData(String poToken, String visitorData) {
+        log.debug("Applying pot: {} vd: {} to WEB, WEBEMBEDDED", poToken, visitorData);
+        Web.setPoTokenAndVisitorData(poToken, visitorData);
+        WebEmbedded.setPoTokenAndVisitorData(poToken, visitorData);
+    }
 }

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

@@ -8,6 +8,7 @@
 import com.sedmelluq.lava.extensions.youtuberotator.tools.ip.Ipv6Block;
 import dev.arbjerg.lavalink.api.AudioPlayerManagerConfiguration;
 import dev.lavalink.youtube.YoutubeAudioSourceManager;
+import dev.lavalink.youtube.YoutubeSource;
 import dev.lavalink.youtube.clients.ClientOptions;
 import dev.lavalink.youtube.clients.Web;
 import dev.lavalink.youtube.clients.WebEmbedded;
@@ -171,8 +172,7 @@ public AudioPlayerManager configure(AudioPlayerManager audioPlayerManager) {
 
                     if (token != null && visitorData != null) {
                         log.debug("Applying poToken and visitorData to WEB & WEBEMBEDDED client (token: {}, vd: {})", token, visitorData);
-                        WebEmbedded.setPoTokenAndVisitorData(token, visitorData);
-                        Web.setPoTokenAndVisitorData(token, visitorData);
+                        YoutubeSource.setPoTokenAndVisitorData(token, visitorData);
                     } else if (token != null || visitorData != null) {
                         log.warn("Both pot.token and pot.visitorData must be specified and valid for pot to apply.");
                     }