Add plugin module

Author: devoxin

.gitignore

@@ -27,4 +27,5 @@ replay_pid*
 .gradle/*
 .idea/*
 .DS_Store
-*/build/*
+*/build/*
+build

build.gradle.kts

@@ -3,16 +3,17 @@ plugins {
     `maven-publish`
 }
 
-group = "dev.lavalink"
+group = "dev.lavalink.youtube"
+version = "1.0.0"
 
 allprojects {
     group = rootProject.group
+    version = rootProject.version
 
     repositories {
         mavenLocal()
         mavenCentral()
         maven(url = "https://maven.lavalink.dev/releases")
-        maven(url = "https://m2.dv8tion.net/releases")
         maven(url = "https://jitpack.io")
     }
 

common/build.gradle.kts

@@ -4,23 +4,8 @@ plugins {
 }
 
 val moduleName = "common"
-version = "1.0.0"
 
 dependencies {
     compileOnly("dev.arbjerg:lavaplayer:1.5.3")
 }
 
-val sourcesJar by tasks.registering(Jar::class) {
-    archiveClassifier.set("sources")
-    from(sourceSets["common"].allSource)
-}
-
-publishing {
-    publications {
-        create<MavenPublication>("mavenJava") {
-            from(components["java"])
-            artifactId = moduleName
-            artifact(sourcesJar)
-        }
-    }
-}

lldevs/build.gradle.kts

@@ -4,24 +4,8 @@ plugins {
 }
 
 val moduleName = "lldevs"
-version = "1.0.0"
 
 dependencies {
     compileOnly(project(":common"))
     compileOnly("dev.arbjerg:lavaplayer:2.1.1")
 }
-
-val sourcesJar by tasks.registering(Jar::class) {
-    archiveClassifier.set("sources")
-    from(sourceSets["main"].allSource)
-}
-
-publishing {
-    publications {
-        create<MavenPublication>("mavenJava") {
-            from(components["java"])
-            artifactId = moduleName
-            artifact(sourcesJar)
-        }
-    }
-}

plugin/build.gradle.kts

@@ -0,0 +1,13 @@
+plugins {
+    `java-library`
+    `maven-publish`
+}
+
+val moduleName = "plugin"
+
+dependencies {
+    compileOnly("dev.arbjerg.lavalink:plugin-api:3.7.11")
+    compileOnly("dev.arbjerg:lavaplayer-ext-youtube-rotator:1.5.3")
+    compileOnly(project(":common"))
+    compileOnly(project(":lldevs"))
+}

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

@@ -0,0 +1,45 @@
+package dev.lavalink.youtube.plugin;
+
+import dev.lavalink.youtube.clients.skeleton.Client;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.List;
+
+public interface ClientProvider {
+    Logger log = LoggerFactory.getLogger(ClientProvider.class);
+
+    Client[] getClients(String[] clients);
+
+    default Client[] getClients(ClientReference[] clientValues, String[] clients) {
+        List<Client> resolved = new ArrayList<>();
+
+        for (String clientName : clients) {
+            Client client = getClientByName(clientValues, clientName);
+
+            if (client == null) {
+                log.warn("Failed to resolve {} into a Client", clientName);
+                continue;
+            }
+
+            resolved.add(client);
+        }
+
+        return resolved.toArray(new Client[0]);
+    }
+
+    interface ClientReference {
+        String getName();
+        Client getClient();
+    }
+
+    static Client getClientByName(ClientReference[] enumValues, String name) {
+        return Arrays.stream(enumValues)
+            .filter(it -> it.getName().equals(name))
+            .findFirst()
+            .map(ClientReference::getClient)
+            .orElse(null);
+    }
+}

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

@@ -0,0 +1,38 @@
+package dev.lavalink.youtube.plugin;
+
+import dev.lavalink.youtube.clients.*;
+import dev.lavalink.youtube.clients.skeleton.Client;
+
+import java.util.function.Supplier;
+
+public class ClientProviderV3 implements ClientProvider {
+    @Override
+    public Client[] getClients(String[] clients) {
+        return getClients(ClientMapping.values(), clients);
+    }
+
+    private enum ClientMapping implements ClientReference {
+        ANDROID(Android::new),
+        ANDROID_TESTSUITE(AndroidTestsuite::new),
+        IOS(Ios::new),
+        MUSIC(Music::new),
+        TVHTML5EMBEDDED(TvHtml5Embedded::new),
+        WEB(Web::new);
+
+        private final Supplier<Client> clientFactory;
+
+        ClientMapping(Supplier<Client> clientFactory) {
+            this.clientFactory = clientFactory;
+        }
+
+        @Override
+        public String getName() {
+            return name();
+        }
+
+        @Override
+        public Client getClient() {
+            return clientFactory.get();
+        }
+    }
+}

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

@@ -0,0 +1,38 @@
+package dev.lavalink.youtube.plugin;
+
+import dev.lavalink.youtube.clients.*;
+import dev.lavalink.youtube.clients.skeleton.Client;
+
+import java.util.function.Supplier;
+
+public class ClientProviderV4 implements ClientProvider {
+    @Override
+    public Client[] getClients(String[] clients) {
+        return getClients(ClientMapping.values(), clients);
+    }
+
+    private enum ClientMapping implements ClientReference {
+        ANDROID(AndroidWithThumbnail::new),
+        ANDROID_TESTSUITE(AndroidTestsuiteWithThumbnail::new),
+        IOS(IosWithThumbnail::new),
+        MUSIC(MusicWithThumbnail::new),
+        TVHTML5EMBEDDED(TvHtml5EmbeddedWithThumbnail::new),
+        WEB(WebWithThumbnail::new);
+
+        private final Supplier<Client> clientFactory;
+
+        ClientMapping(Supplier<Client> clientFactory) {
+            this.clientFactory = clientFactory;
+        }
+
+        @Override
+        public String getName() {
+            return name();
+        }
+
+        @Override
+        public Client getClient() {
+            return clientFactory.get();
+        }
+    }
+}

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

@@ -0,0 +1,57 @@
+package dev.lavalink.youtube.plugin;
+
+import org.springframework.boot.context.properties.ConfigurationProperties;
+import org.springframework.stereotype.Component;
+
+import java.util.Collections;
+import java.util.List;
+
+@ConfigurationProperties(prefix = "lavalink.server.ratelimit")
+@Component
+public class RatelimitConfig {
+    private List<String> ipBlocks = Collections.emptyList();
+    private List<String> excludedIps = Collections.emptyList();
+    private String strategy = "RotatingNanoSwitch";
+    private int retryLimit = -1;
+    private boolean searchTriggersFail = true;
+
+    public List<String> getIpBlocks() {
+        return this.ipBlocks;
+    }
+
+    public List<String> getExcludedIps() {
+        return this.excludedIps;
+    }
+
+    public String getStrategy() {
+        return this.strategy;
+    }
+
+    public int getRetryLimit() {
+        return this.retryLimit;
+    }
+
+    public boolean getSearchTriggersFail() {
+        return this.searchTriggersFail;
+    }
+
+    public void setIpBlocks(List<String> ipBlocks) {
+        this.ipBlocks = ipBlocks;
+    }
+
+    public void setExcludedIps(List<String> excludedIps) {
+        this.excludedIps = excludedIps;
+    }
+
+    public void setStrategy(String strategy) {
+        this.strategy = strategy;
+    }
+
+    public void setRetryLimit(int retryLimit) {
+        this.retryLimit = retryLimit;
+    }
+
+    public void setSearchTriggersFail(boolean searchTriggersFail) {
+        this.searchTriggersFail = searchTriggersFail;
+    }
+}

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

@@ -0,0 +1,18 @@
+package dev.lavalink.youtube.plugin;
+
+import org.springframework.boot.context.properties.ConfigurationProperties;
+import org.springframework.stereotype.Component;
+
+@ConfigurationProperties(prefix = "lavalink.server.ratelimit")
+@Component
+public class ServerConfig {
+    private int youtubePlaylistLoadLimit = 6;
+
+    public int getYoutubePlaylistLoadLimit() {
+        return this.youtubePlaylistLoadLimit;
+    }
+
+    public void setYoutubePlaylistLoadLimit(int youtubePlaylistLoadLimit) {
+        this.youtubePlaylistLoadLimit = youtubePlaylistLoadLimit;
+    }
+}