-
-
Notifications
You must be signed in to change notification settings - Fork 41
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #34 from Ree6-Applications/feature/voice-recording
Voicechannel Recording.
- Loading branch information
Showing
63 changed files
with
598 additions
and
324 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
121 changes: 121 additions & 0 deletions
121
src/main/java/de/presti/ree6/audio/AudioPlayerReceiveHandler.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,121 @@ | ||
| package de.presti.ree6.audio; | ||
|
|
||
| import de.presti.ree6.utils.data.AudioUtil; | ||
| import net.dv8tion.jda.api.audio.AudioReceiveHandler; | ||
| import net.dv8tion.jda.api.audio.CombinedAudio; | ||
| import net.dv8tion.jda.api.entities.User; | ||
| import net.dv8tion.jda.api.entities.VoiceChannel; | ||
| import org.jetbrains.annotations.NotNull; | ||
|
|
||
| import java.nio.ByteBuffer; | ||
| import java.text.SimpleDateFormat; | ||
| import java.util.Queue; | ||
| import java.util.concurrent.ConcurrentLinkedQueue; | ||
|
|
||
| /* | ||
| All methods in this class are called by JDA threads when resources are available/ready for processing. | ||
| The receiver will be provided with the latest 20ms of PCM stereo audio | ||
| Note you can receive even while setting yourself to deafened | ||
| */ | ||
| public class AudioPlayerReceiveHandler implements AudioReceiveHandler { | ||
|
|
||
| /** | ||
| * Queue of audio to be sent afterwards. | ||
| */ | ||
| private final Queue<byte[]> queue = new ConcurrentLinkedQueue<>(); | ||
|
|
||
| /** | ||
| * Boolean used to indicated that handler finished his Job. | ||
| */ | ||
| private boolean finished = false; | ||
|
|
||
| /** | ||
| * The voice channel this handler is currently handling. | ||
| */ | ||
| private final VoiceChannel voiceChannel; | ||
|
|
||
| /** | ||
| * Constructor. | ||
| * | ||
| * @param voiceChannel The voice channel this handler should handle. | ||
| */ | ||
| public AudioPlayerReceiveHandler(VoiceChannel voiceChannel) { | ||
| this.voiceChannel = voiceChannel; | ||
| } | ||
|
|
||
| /** | ||
| * @see AudioReceiveHandler#canReceiveCombined() | ||
| */ | ||
| @Override // combine multiple user audio-streams into a single one | ||
| public boolean canReceiveCombined() { | ||
| /* one entry = 20ms of audio, which means 20 * 100 = 2000ms = 2s of audio, | ||
| * but since we want to allow up to 5 minute of audio we have to do | ||
| * 20 * 100 * 150 = 300.000ms = 5 minutes of audio. | ||
| * And since 100 entries are 2s we would need 15000 entries for 5 minutes of audio. | ||
| */ | ||
| return queue.size() < 15000; | ||
| } | ||
|
|
||
| /** | ||
| * @see AudioReceiveHandler#canReceiveUser() | ||
| */ | ||
| @Override | ||
| public boolean includeUserInCombinedAudio(@NotNull User user) { | ||
| return !user.isBot(); | ||
| } | ||
|
|
||
| /** | ||
| * @see AudioReceiveHandler#handleCombinedAudio(CombinedAudio) | ||
| */ | ||
| @Override | ||
| public void handleCombinedAudio(CombinedAudio combinedAudio) { | ||
| if (finished) { | ||
| return; | ||
| } | ||
|
|
||
| if (combinedAudio.getUsers().isEmpty()) { | ||
| if (voiceChannel.getMembers().size() == 1) { | ||
| endReceiving(); | ||
| } | ||
| return; | ||
| } | ||
|
|
||
| if (voiceChannel.getMembers().size() == 1) { | ||
| endReceiving(); | ||
| return; | ||
| } | ||
|
|
||
| byte[] data = combinedAudio.getAudioData(1.0f); | ||
| queue.add(data); | ||
|
|
||
| if (!canReceiveCombined()) { | ||
| endReceiving(); | ||
| } | ||
| } | ||
|
|
||
| /** | ||
| * Method called when the recording should stop. | ||
| */ | ||
| public void endReceiving() { | ||
| if (finished) { | ||
| return; | ||
| } | ||
|
|
||
| finished = true; | ||
|
|
||
| try { | ||
| int queueSize = queue.stream().mapToInt(data -> data.length).sum(); | ||
| ByteBuffer byteBuffer = ByteBuffer.wrap(new byte[queueSize]); | ||
| for (byte[] data : queue) { | ||
| byteBuffer.put(data); | ||
| } | ||
|
|
||
| voiceChannel.sendMessage("Here is your audio!").addFile(AudioUtil.convert(byteBuffer), new SimpleDateFormat("dd.MM.yyyy HH/mm").format(System.currentTimeMillis()) + "-" + voiceChannel.getId() + ".wav").queue(); | ||
| } catch (Exception ex) { | ||
| voiceChannel.sendMessage("Something went wrong while converting your audio!\nReason: " + ex.getMessage()).queue(); | ||
| } | ||
|
|
||
| voiceChannel.getGuild().getAudioManager().closeAudioConnection(); | ||
| queue.clear(); | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.