From 4749216320da6d8adac202f468c759db18a8b735 Mon Sep 17 00:00:00 2001
From: "Yukai (Donald) Shan" <xxpinapelzxx@gmail.com>
Date: Wed, 7 Dec 2022 10:03:23 -0800
Subject: Add files via upload

---
 audio/TrackScheduler.java | 59 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 59 insertions(+)
 create mode 100644 audio/TrackScheduler.java

(limited to 'audio/TrackScheduler.java')

diff --git a/audio/TrackScheduler.java b/audio/TrackScheduler.java
new file mode 100644
index 0000000..8377e7a
--- /dev/null
+++ b/audio/TrackScheduler.java
@@ -0,0 +1,59 @@
+package audio;
+
+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.List;
+import java.util.concurrent.BlockingQueue;
+import java.util.concurrent.LinkedBlockingQueue;
+
+/**
+ * This class schedules tracks for the audio player. It contains the queue of tracks.
+ */
+public class TrackScheduler extends AudioEventAdapter {
+    public final AudioPlayer player;
+    public BlockingQueue<AudioTrack> queue;
+
+    /**
+     * @param player The audio player this scheduler uses
+     */
+    public TrackScheduler(AudioPlayer player) {
+        this.player = player;
+        this.queue = new LinkedBlockingQueue<>();
+    }
+
+    /**
+     * 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) {
+        if (!player.startTrack(track, true)) {
+            queue.offer(track);
+        }
+    }
+
+    /**
+     * Start the next track, stopping the current one if it is playing.
+     */
+    public void nextTrack() {
+
+        player.startTrack(queue.poll(), false);
+
+    }
+
+    @Override
+    public void onTrackEnd(AudioPlayer player, AudioTrack track, AudioTrackEndReason endReason) {
+        // Only start the next track if the end reason is suitable for it (FINISHED or LOAD_FAILED)
+        if (endReason.mayStartNext) {
+            nextTrack();
+        }
+    }
+    public void shuffle()
+    {
+        Collections.shuffle((List<?>) queue);
+    }
+}
\ No newline at end of file
-- 
cgit v1.2.3