99import dev .arbjerg .lavalink .api .AudioPlayerManagerConfiguration ;
1010import dev .lavalink .youtube .YoutubeAudioSourceManager ;
1111import dev .lavalink .youtube .clients .skeleton .Client ;
12+ import lavalink .server .config .RateLimitConfig ;
13+ import lavalink .server .config .ServerConfig ;
1214import org .slf4j .Logger ;
1315import org .slf4j .LoggerFactory ;
1416import org .springframework .stereotype .Service ;
@@ -28,18 +30,18 @@ public class YoutubePluginLoader implements AudioPlayerManagerConfiguration {
2830 private static final Logger log = LoggerFactory .getLogger (YoutubePluginLoader .class );
2931
3032 private final YoutubeConfig youtubeConfig ;
31- private final RatelimitConfig ratelimitConfig ;
3233 private final ServerConfig serverConfig ;
34+ private final RateLimitConfig ratelimitConfig ;
3335 private final ClientProvider clientProvider ;
3436
3537 // This entire thing is a hack BTW. Designed to support Lavalink v3 and v4
3638 // with a single plugin. Totally worth it!
3739 public YoutubePluginLoader (final YoutubeConfig youtubeConfig ,
38- final RatelimitConfig ratelimitConfig ,
39- final ServerConfig serverConfig ) {
40+ final ServerConfig serverConfig ,
41+ final RateLimitConfig ratelimitConfig ) {
4042 this .youtubeConfig = youtubeConfig ;
41- this .ratelimitConfig = ratelimitConfig ;
4243 this .serverConfig = serverConfig ;
44+ this .ratelimitConfig = ratelimitConfig ;
4345
4446 final String providerName = isV4OrNewer ()
4547 ? "ClientProviderV4"
@@ -155,7 +157,12 @@ public AudioPlayerManager configure(AudioPlayerManager audioPlayerManager) {
155157 rotator .setup ();
156158 }
157159
158- source .setPlaylistPageCount (serverConfig .getYoutubePlaylistLoadLimit ());
160+ Integer playlistLoadLimit = serverConfig .getYoutubePlaylistLoadLimit ();
161+
162+ if (playlistLoadLimit != null ) {
163+ source .setPlaylistPageCount (playlistLoadLimit );
164+ }
165+
159166 audioPlayerManager .registerSourceManager (source );
160167 return audioPlayerManager ;
161168 }
0 commit comments