1 files changed, 48 insertions, 0 deletions
diff --git a/src/main/java/soundbot/AudioPlayerSendHandler.java b/src/main/java/soundbot/AudioPlayerSendHandler.java
new file mode 100644
index 0000000..6499aad
--- /dev/null
+++ b/src/main/java/soundbot/AudioPlayerSendHandler.java
@@ -0,0 +1,48 @@
+package soundbot;
+
+import com.sedmelluq.discord.lavaplayer.player.AudioPlayer;
+import com.sedmelluq.discord.lavaplayer.track.playback.AudioFrame;
+import net.dv8tion.jda.core.audio.AudioSendHandler;
+
+/**
+ * This is a wrapper around AudioPlayer which makes it behave as an AudioSendHandler for JDA. As JDA calls canProvide
+ * before every call to provide20MsAudio(), we pull the frame in canProvide() and use the frame we already pulled in
+ * provide20MsAudio().
+ */
+public class AudioPlayerSendHandler implements AudioSendHandler {
+  private final AudioPlayer audioPlayer;
+  private AudioFrame lastFrame;
+
+  /**
+   * @param audioPlayer Audio player to wrap.
+   */
+  public AudioPlayerSendHandler(AudioPlayer audioPlayer) {
+    this.audioPlayer = audioPlayer;
+  }
+
+  @Override
+  public boolean canProvide() {
+    if (lastFrame == null) {
+      lastFrame = audioPlayer.provide();
+    }
+
+    return lastFrame != null;
+  }
+
+  @Override
+  public byte[] provide20MsAudio() {
+    if (lastFrame == null) {
+      lastFrame = audioPlayer.provide();
+    }
+
+    byte[] data = lastFrame != null ? lastFrame.data : null;
+    lastFrame = null;
+
+    return data;
+  }
+
+  @Override
+  public boolean isOpus() {
+    return true;
+  }
+}