Allow adding new MediaContainerProbes via plugins (#631)

* Allow adding new MediaContainerProbes via plugins

* Document custom MediaContainerProbe bean support

* Changes plugin decoumentation wording

Co-authored-by: Freya Arbjerg <[email protected]>

Author: TheEssem

LavalinkServer/src/main/java/lavalink/server/config/AudioPlayerConfiguration.kt

@@ -1,5 +1,7 @@
 package lavalink.server.config
 
+import com.sedmelluq.discord.lavaplayer.container.MediaContainerProbe
+import com.sedmelluq.discord.lavaplayer.container.MediaContainerRegistry
 import com.sedmelluq.discord.lavaplayer.player.AudioPlayerManager
 import com.sedmelluq.discord.lavaplayer.player.DefaultAudioPlayerManager
 import com.sedmelluq.discord.lavaplayer.source.AudioSourceManager
@@ -42,7 +44,8 @@ class AudioPlayerConfiguration {
         serverConfig: ServerConfig,
         routePlanner: AbstractRoutePlanner?,
         audioSourceManagers: Collection<AudioSourceManager>,
-        audioPlayerManagerConfigurations: Collection<AudioPlayerManagerConfiguration>
+        audioPlayerManagerConfigurations: Collection<AudioPlayerManagerConfiguration>,
+        mediaContainerProbes: Collection<MediaContainerProbe>
     ): AudioPlayerManager {
         val audioPlayerManager = DefaultAudioPlayerManager()
 
@@ -61,6 +64,8 @@ class AudioPlayerConfiguration {
             audioPlayerManager.frameBufferDuration = bufferDuration
         }
 
+        val mcr: MediaContainerRegistry = MediaContainerRegistry.extended(*mediaContainerProbes.toTypedArray())
+
         if (sources.isYoutube) {
             val youtube = YoutubeAudioSourceManager(serverConfig.isYoutubeSearchEnabled)
             if (routePlanner != null) {
@@ -109,7 +114,7 @@ class AudioPlayerConfiguration {
         if (sources.isTwitch) audioPlayerManager.registerSourceManager(TwitchStreamAudioSourceManager())
         if (sources.isVimeo) audioPlayerManager.registerSourceManager(VimeoAudioSourceManager())
         if (sources.isMixer) audioPlayerManager.registerSourceManager(BeamAudioSourceManager())
-        if (sources.isLocal) audioPlayerManager.registerSourceManager(LocalAudioSourceManager())
+        if (sources.isLocal) audioPlayerManager.registerSourceManager(LocalAudioSourceManager(mcr))
 
         audioSourceManagers.forEach {
             audioPlayerManager.registerSourceManager(it)
@@ -126,7 +131,7 @@ class AudioPlayerConfiguration {
 
         // This must be loaded last
         if (sources.isHttp) {
-            val httpAudioSourceManager = HttpAudioSourceManager()
+            val httpAudioSourceManager = HttpAudioSourceManager(mcr)
 
             serverConfig.httpConfig?.let { httpConfig ->
                 httpAudioSourceManager.configureBuilder {

PLUGINS.md

@@ -59,3 +59,14 @@ class MyAudioSourceManager implements AudioSourceManager {
     // ...
 } 
 ```
+
+Likewise, you can also add new MediaContainerProbe this way, to be used with the HTTP and local sources:
+
+```java
+import org.springframework.stereotype.Service;
+
+@Service
+class MyMediaContainerProbe implements MediaContainerProbe {
+    // ...
+} 
+```