package net.dv8tion.discord.music; import com.sedmelluq.discord.lavaplayer.player.AudioPlayer; import com.sedmelluq.discord.lavaplayer.player.event.AudioEventAdapter; import com.sedmelluq.discord.lavaplayer.track.AudioTrack; import com.sedmelluq.discord.lavaplayer.track.AudioTrackEndReason; import java.util.Collections; import java.util.LinkedList; import java.util.List; import java.util.Queue; /** * This class schedules tracks for the audio player. It contains the queue of tracks. */ public class TrackScheduler extends AudioEventAdapter { private boolean repeating = false; final AudioPlayer player; final Queue<AudioTrack> queue; AudioTrack lastTrack; /** * @param player The audio player this scheduler uses */ public TrackScheduler(AudioPlayer player) { this.player = player; this.queue = new LinkedList<>(); } /** * Add the next track to queue or play right away if nothing is in the queue. * * @param track The track to play or add to queue. */ public void queue(AudioTrack track) { // Calling startTrack with the noInterrupt set to true will start the track only if nothing is currently playing. If // something is playing, it returns false and does nothing. In that case the player was already playing so this // track goes to the queue instead. if (!player.startTrack(track, true)) { queue.offer(track); } } /** * Start the next track, stopping the current one if it is playing. */ public void nextTrack() { // Start the next track, regardless of if something is already playing or not. In case queue was empty, we are // giving null to startTrack, which is a valid argument and will simply stop the player. player.startTrack(queue.poll(), false); } @Override public void onTrackEnd(AudioPlayer player, AudioTrack track, AudioTrackEndReason endReason) { this.lastTrack = track; // Only start the next track if the end reason is suitable for it (FINISHED or LOAD_FAILED) if (endReason.mayStartNext) { if (repeating) player.startTrack(lastTrack.makeClone(), false); else nextTrack(); } } public boolean isRepeating() { return repeating; } public void setRepeating(boolean repeating) { this.repeating = repeating; } public void shuffle() { Collections.shuffle((List<?>) queue); } }