From 3799c7828ad8ef2968f39e4044db36885aec1ab4 Mon Sep 17 00:00:00 2001 From: Brandon Date: Sat, 7 Apr 2018 15:55:32 -0400 Subject: Initial commit --- src/main/java/soundbot/Main.java | 167 +++++++++++++++++++++++++++++++++++++++ 1 file changed, 167 insertions(+) create mode 100644 src/main/java/soundbot/Main.java (limited to 'src/main/java/soundbot/Main.java') diff --git a/src/main/java/soundbot/Main.java b/src/main/java/soundbot/Main.java new file mode 100644 index 0000000..2255c68 --- /dev/null +++ b/src/main/java/soundbot/Main.java @@ -0,0 +1,167 @@ +package soundbot; + +import com.sedmelluq.discord.lavaplayer.player.AudioLoadResultHandler; +import com.sedmelluq.discord.lavaplayer.player.AudioPlayerManager; +import com.sedmelluq.discord.lavaplayer.player.DefaultAudioPlayerManager; +import com.sedmelluq.discord.lavaplayer.source.AudioSourceManagers; +import com.sedmelluq.discord.lavaplayer.tools.FriendlyException; +import com.sedmelluq.discord.lavaplayer.track.AudioPlaylist; +import com.sedmelluq.discord.lavaplayer.track.AudioTrack; +import net.dv8tion.jda.core.AccountType; +import net.dv8tion.jda.core.JDA; +import net.dv8tion.jda.core.JDABuilder; +import net.dv8tion.jda.core.entities.Guild; +import net.dv8tion.jda.core.entities.TextChannel; +import net.dv8tion.jda.core.entities.VoiceChannel; +import net.dv8tion.jda.core.events.message.MessageReceivedEvent; +import net.dv8tion.jda.client.events.call.voice.CallVoiceJoinEvent; +import net.dv8tion.jda.core.hooks.ListenerAdapter; +import net.dv8tion.jda.core.managers.AudioManager; + +import java.util.HashMap; +import java.util.Map; + +public class Main extends ListenerAdapter { + public static void main(String[] args) throws Exception { + JDA jda = new JDABuilder(AccountType.BOT) + .setToken(System.getProperty("botToken")) + .buildBlocking(); + + jda.addEventListener(new Main()); + } + + private final AudioPlayerManager playerManager; + private final Map musicManagers; + + private Main() { + this.musicManagers = new HashMap<>(); + + this.playerManager = new DefaultAudioPlayerManager(); + AudioSourceManagers.registerRemoteSources(playerManager); + AudioSourceManagers.registerLocalSource(playerManager); + } + + private synchronized GuildMusicManager getGuildAudioPlayer(Guild guild) { + long guildId = Long.parseLong(guild.getId()); + GuildMusicManager musicManager = musicManagers.get(guildId); + + if (musicManager == null) { + musicManager = new GuildMusicManager(playerManager); + musicManagers.put(guildId, musicManager); + } + + guild.getAudioManager().setSendingHandler(musicManager.getSendHandler()); + + return musicManager; + } + + @Override + public void onCallVoiceJoin(CallVoiceJoinEvent event){ + + } + + @Override + public void onMessageReceived(MessageReceivedEvent event) { + String[] command = event.getMessage().getContentRaw().split(" ", 2); + Guild guild = event.getGuild(); + + if (guild != null) { + if ("~play".equals(command[0]) && command.length == 2) { + loadAndPlay(event.getTextChannel(), command[1]); + } else if ("~skip".equals(command[0])) { + skipTrack(event.getTextChannel()); + } else if ("~volume".equals(command[0]) && command.length == 2) { + changeVolume(event.getTextChannel(), command[1]); + } else if ("~pause".equals(command[0])) { + pauseTrack(event.getTextChannel()); + } else if ("~unpause".equals(command[0])) { + unpauseTrack(event.getTextChannel()); + } + } + + super.onMessageReceived(event); + } + + private void changeVolume(final TextChannel channel, final String volume) { + GuildMusicManager musicManager = getGuildAudioPlayer(channel.getGuild()); + musicManager.player.setVolume(Integer.parseInt(volume)); + channel.sendMessage("Volume now set to " + volume + "%").queue(); + } + + private void pauseTrack(final TextChannel channel){ + GuildMusicManager musicManager = getGuildAudioPlayer(channel.getGuild()); + musicManager.player.setPaused(true); + channel.sendMessage("Playback Paused.").queue(); + } + + private void unpauseTrack(final TextChannel channel){ + GuildMusicManager musicManager = getGuildAudioPlayer(channel.getGuild()); + musicManager.player.setPaused(false); + channel.sendMessage("Unpaused playback.").queue(); + } + + private void loadAndPlay(final TextChannel channel, final String trackUrl) { + GuildMusicManager musicManager = getGuildAudioPlayer(channel.getGuild()); + + playerManager.loadItemOrdered(musicManager, trackUrl, new AudioLoadResultHandler() { + @Override + public void trackLoaded(AudioTrack track) { + int timeStart = trackUrl.lastIndexOf('='); + if(timeStart != -1){ + String timeString = trackUrl.substring(timeStart); + //The format will be 1h2m53s, need to parse that into seconds and then call + //track.setPosition(long position) + + } + channel.sendMessage("Adding to queue " + track.getInfo().title).queue(); + + play(channel.getGuild(), musicManager, track); + } + + @Override + public void playlistLoaded(AudioPlaylist playlist) { + AudioTrack firstTrack = playlist.getSelectedTrack(); + + if (firstTrack == null) { + firstTrack = playlist.getTracks().get(0); + } + + channel.sendMessage("Adding to queue " + firstTrack.getInfo().title + " (first track of playlist " + playlist.getName() + ")").queue(); + + play(channel.getGuild(), musicManager, firstTrack); + } + + @Override + public void noMatches() { + channel.sendMessage("Nothing found by " + trackUrl).queue(); + } + + @Override + public void loadFailed(FriendlyException exception) { + channel.sendMessage("Could not play: " + exception.getMessage()).queue(); + } + }); + } + + private void play(Guild guild, GuildMusicManager musicManager, AudioTrack track) { + connectToFirstVoiceChannel(guild.getAudioManager()); + + musicManager.scheduler.queue(track); + } + + private void skipTrack(TextChannel channel) { + GuildMusicManager musicManager = getGuildAudioPlayer(channel.getGuild()); + musicManager.scheduler.nextTrack(); + + channel.sendMessage("Skipped to next track.").queue(); + } + + private static void connectToFirstVoiceChannel(AudioManager audioManager) { + if (!audioManager.isConnected() && !audioManager.isAttemptingToConnect()) { + for (VoiceChannel voiceChannel : audioManager.getGuild().getVoiceChannels()) { + audioManager.openAudioConnection(voiceChannel); + break; + } + } + } +} -- cgit v1.2.3