Compare commits

..

2 Commits

16 changed files with 282 additions and 123 deletions

View File

@ -5,22 +5,27 @@
*/ */
package moe.nekojimi.chords; package moe.nekojimi.chords;
import com.amihaiemil.eoyaml.Yaml;
import com.amihaiemil.eoyaml.YamlMapping;
import java.io.File; import java.io.File;
import java.io.FilenameFilter;
import java.io.IOException;
import java.net.MalformedURLException; import java.net.MalformedURLException;
import java.net.URL; import java.net.URL;
import java.nio.file.Files;
import java.time.LocalDateTime; import java.time.LocalDateTime;
import java.time.format.DateTimeFormatter; import java.time.format.DateTimeFormatter;
import java.util.*; import java.util.*;
import java.util.function.BiConsumer; import java.util.function.BiConsumer;
import java.util.function.Consumer; import java.util.function.Consumer;
import java.util.logging.Level;
import java.util.logging.Logger;
import javax.security.auth.login.LoginException; import javax.security.auth.login.LoginException;
import moe.nekojimi.chords.commands.*; import moe.nekojimi.chords.commands.*;
import moe.nekojimi.musicsearcher.Result;
import moe.nekojimi.musicsearcher.providers.MetaSearcher; import moe.nekojimi.musicsearcher.providers.MetaSearcher;
import moe.nekojimi.musicsearcher.providers.Searcher; import moe.nekojimi.musicsearcher.providers.Searcher;
import net.dv8tion.jda.api.JDA; import net.dv8tion.jda.api.JDA;
import net.dv8tion.jda.api.JDABuilder; import net.dv8tion.jda.api.JDABuilder;
import net.dv8tion.jda.api.MessageBuilder;
import net.dv8tion.jda.api.entities.*; import net.dv8tion.jda.api.entities.*;
import net.dv8tion.jda.api.events.guild.voice.GuildVoiceLeaveEvent; import net.dv8tion.jda.api.events.guild.voice.GuildVoiceLeaveEvent;
import net.dv8tion.jda.api.events.message.guild.GuildMessageReceivedEvent; import net.dv8tion.jda.api.events.message.guild.GuildMessageReceivedEvent;
@ -34,9 +39,12 @@ import net.dv8tion.jda.api.utils.cache.CacheFlag;
* *
* @author jimj316 * @author jimj316
*/ */
public class Main extends ListenerAdapter public final class Main extends ListenerAdapter
{ {
private final File dataDirectory;
private final File playlistsDirectory;
private MusicHandler musicHandler; private MusicHandler musicHandler;
private final Downloader downloader; private final Downloader downloader;
private final Searcher searcher; private final Searcher searcher;
@ -51,10 +59,12 @@ public class Main extends ListenerAdapter
private int trackNumber = 1; private int trackNumber = 1;
private final Map<String, Playlist> playlists = new HashMap<>();
/** /**
* @param args the command line arguments * @param args the command line arguments
*/ */
public static void main(String[] args) throws LoginException public static void main(String[] args) throws LoginException, IOException
{ {
// We only need 2 gateway intents enabled for this example: // We only need 2 gateway intents enabled for this example:
EnumSet<GatewayIntent> intents = EnumSet.of( EnumSet<GatewayIntent> intents = EnumSet.of(
@ -100,7 +110,7 @@ public class Main extends ListenerAdapter
if (ex == null) if (ex == null)
if (song.getLocation() != null) if (song.getLocation() != null)
{ {
request.respond("Finished downloading " + song + ", added to queue!"); request.getInvocation().respond("Finished downloading " + song + ", added to queue!");
log("DOWN", "Downloaded " + song); log("DOWN", "Downloaded " + song);
} else } else
{ {
@ -114,22 +124,30 @@ public class Main extends ListenerAdapter
String bitFmt = (bitrate <= 0 ? "??" : bitrate) + "k"; String bitFmt = (bitrate <= 0 ? "??" : bitrate) + "k";
formatDetails = " (" + bitFmt + ", " + sizeFmt + ")"; formatDetails = " (" + bitFmt + ", " + sizeFmt + ")";
} }
request.respond("Now downloading " + song + formatDetails + " ..."); request.getInvocation().respond("Now downloading " + song + formatDetails + " ...");
log("DOWN", "Downloading " + song + "..."); log("DOWN", "Downloading " + song + "...");
} }
else else
{ {
request.respond("Failed to download " + song + "! Reason: " + ex.getMessage()); request.getInvocation().respond("Failed to download " + song + "! Reason: " + ex.getMessage());
log("DOWN", "Failed to download " + song + "! Reason: " + ex.getMessage()); log("DOWN", "Failed to download " + song + "! Reason: " + ex.getMessage());
} }
}; };
public Main() public Main() throws IOException
{ {
log("INFO", "Starting up..."); log("INFO", "Starting up...");
// init dirs
dataDirectory = new File(System.getProperty("user.dir"));
playlistsDirectory = initDirectory(dataDirectory, "playlists");
// init downloader
downloader = new Downloader(); downloader = new Downloader();
downloader.setMessageHandler(downloaderMessageHandler); downloader.setMessageHandler(downloaderMessageHandler);
// init searcher
searcher = MetaSearcher.loadYAML(new File("searchproviders.yml")); searcher = MetaSearcher.loadYAML(new File("searchproviders.yml"));
// init queue manager // init queue manager
@ -147,6 +165,9 @@ public class Main extends ListenerAdapter
helpCommand = new HelpCommand(this); helpCommand = new HelpCommand(this);
addCommand(helpCommand); addCommand(helpCommand);
// load playlists
loadPlaylists();
log("INFO", "Started OK!"); log("INFO", "Started OK!");
} }
@ -195,7 +216,7 @@ public class Main extends ListenerAdapter
try try
{ {
URL parseURL = new URL(content.trim()); URL parseURL = new URL(content.trim());
playCommand.call(event, List.of(parseURL.toExternalForm())); playCommand.call(null);
} catch (MalformedURLException ex) } catch (MalformedURLException ex)
{ {
// not a URL, then // not a URL, then
@ -203,7 +224,7 @@ public class Main extends ListenerAdapter
try try
{ {
String[] split = content.split("\\s+", 2); String[] split = content.split("\\s+");
String cmd = split[0].toLowerCase(); String cmd = split[0].toLowerCase();
if (!cmd.startsWith("!")) if (!cmd.startsWith("!"))
@ -211,17 +232,21 @@ public class Main extends ListenerAdapter
cmd = cmd.substring(1); // strip prefix char cmd = cmd.substring(1); // strip prefix char
String arg = ""; // String arg = "";
if (split.length > 1) List<String> args = new ArrayList<>();
arg = split[1]; Collections.addAll(args, split);
args.remove(0);
Invocation invocation = new Invocation(event, args);
invocation.setRequestMessage(message);
if (commands.containsKey(cmd)) if (commands.containsKey(cmd))
{ {
Command command = commands.get(cmd); Command command = commands.get(cmd);
command.call(event, List.of(arg)); command.call(invocation);
} else } else
{ {
helpCommand.call(event, List.of(arg)); helpCommand.call(invocation);
} }
} catch (Exception ex) } catch (Exception ex)
{ {
@ -241,16 +266,16 @@ public class Main extends ListenerAdapter
// interpret search result // interpret search result
throw new UnsupportedOperationException("Not supported yet."); throw new UnsupportedOperationException("Not supported yet.");
} }
if (request.getRequestMessage() != null) if (request.getInvocation().getRequestMessage() != null)
{ {
song.setRequestedBy(request.getRequestMessage().getAuthor().getName()); song.setRequestedBy(request.getInvocation().getRequestMessage().getAuthor().getName());
song.setRequestedIn(request.getRequestMessage().getChannel().getId()); song.setRequestedIn(request.getInvocation().getRequestMessage().getChannel().getId());
} }
song.setNumber(trackNumber); song.setNumber(trackNumber);
trackNumber++; trackNumber++;
request.setSong(song); request.setSong(song);
downloader.accept(new Downloader.DownloadTask(request, queueManager)); downloader.accept(new Downloader.DownloadTask(request, queueManager));
request.respond("Request pending..."); request.getInvocation().respond("Request pending...");
return song; return song;
} }
@ -329,6 +354,35 @@ public class Main extends ListenerAdapter
System.out.println(type + " " + LocalDateTime.now().format(DateTimeFormatter.ISO_DATE_TIME) + "\t" + message); System.out.println(type + " " + LocalDateTime.now().format(DateTimeFormatter.ISO_DATE_TIME) + "\t" + message);
} }
public File initDirectory(File parent, String name) throws IOException
{
File ret = new File(parent, name);
if (!ret.exists())
Files.createDirectories(ret.toPath());
if (!ret.canRead())
throw new RuntimeException("Cannot read directory " + ret.getAbsolutePath() + "!");
if (!ret.canWrite())
throw new RuntimeException("Cannot write to directory " + ret.getAbsolutePath() + "!");
return ret;
}
private void loadPlaylists()
{
File[] files = playlistsDirectory.listFiles((File file, String name) -> name.endsWith(".yaml"));
for (File file : files)
{
try
{
YamlMapping map = Yaml.createYamlInput(file).readYamlMapping();
Playlist playlist = Playlist.fromYaml(map);
playlists.put(playlist.getName(), playlist);
} catch (IOException ex)
{
Logger.getLogger(Main.class.getName()).log(Level.SEVERE, null, ex);
}
}
}
public MusicHandler getMusicHandler() public MusicHandler getMusicHandler()
{ {
return musicHandler; return musicHandler;
@ -363,4 +417,5 @@ public class Main extends ListenerAdapter
return trackNumber; return trackNumber;
} }
} }

View File

@ -16,19 +16,28 @@
*/ */
package moe.nekojimi.chords; package moe.nekojimi.chords;
import com.amihaiemil.eoyaml.Yaml;
import com.amihaiemil.eoyaml.YamlMapping; import com.amihaiemil.eoyaml.YamlMapping;
import java.util.ArrayList; import com.amihaiemil.eoyaml.YamlSequence;
import java.util.List; import com.amihaiemil.eoyaml.YamlSequenceBuilder;
import java.net.MalformedURLException;
import java.util.*;
import java.util.function.Consumer;
import java.util.logging.Level;
import java.util.logging.Logger;
/** /**
* *
* @author jimj316 * @author jimj316
*/ */
public class Playlist public class Playlist implements Consumer<Song>
{ {
private static final int SHUFFLE_DONT_REPEAT_LAST = 3;
private final String name; private final String name;
private final List<Song> songs = new ArrayList<>(); private final List<Song> songs = new ArrayList<>();
private final LinkedList<Song> playHistory = new LinkedList<>();
public Playlist(String name) public Playlist(String name)
{ {
@ -37,12 +46,31 @@ public class Playlist
public YamlMapping toYaml() public YamlMapping toYaml()
{ {
throw new UnsupportedOperationException("Not supported yet."); YamlSequenceBuilder songList = Yaml.createYamlSequenceBuilder();
for (Song song : songs)
songList = songList.add(song.toYaml());
return Yaml.createYamlMappingBuilder()
.add("name", name)
.add("songs", songList.build())
.build();
} }
public static Playlist fromYaml(YamlMapping yaml) public static Playlist fromYaml(YamlMapping yaml)
{ {
throw new UnsupportedOperationException("Not supported yet."); Playlist ret = new Playlist(yaml.string("name"));
YamlSequence songList = yaml.value("songs").asSequence();
for (int i = 0; i < songList.size(); i++)
{
try
{
ret.addSong(Song.fromYaml(songList.yamlMapping(i)));
} catch (MalformedURLException ex)
{
Logger.getLogger(Playlist.class.getName()).log(Level.SEVERE, null, ex);
}
}
return ret;
} }
public void addSong(Song song) public void addSong(Song song)
@ -61,9 +89,34 @@ public class Playlist
return songs; return songs;
} }
Song getNextSong() public Song getNextSong()
{ {
throw new UnsupportedOperationException("Not supported yet."); Song ret;
// copy the song list
List<Song> toShuffle = new LinkedList<>(songs);
// remove play history from candidates, latest first, unless we'd have less than 2 options
for (int i = playHistory.size() - 1; i >= 0; i--)
{
if (toShuffle.size() <= 2)
break;
toShuffle.remove(playHistory.get(i));
}
Collections.shuffle(toShuffle);
ret = toShuffle.get(0);
playHistory.add(ret);
if (playHistory.size() > SHUFFLE_DONT_REPEAT_LAST)
playHistory.remove();
return ret;
}
@Override
public void accept(Song t)
{
addSong(t);
} }
} }

View File

@ -26,8 +26,6 @@ import java.util.function.Consumer;
*/ */
public class QueueManager implements Consumer<Song> public class QueueManager implements Consumer<Song>
{ {
private Mode mode;
private final Queue<Song> jukeboxQueue; private final Queue<Song> jukeboxQueue;
private Playlist playlist; private Playlist playlist;
private MusicHandler handler; private MusicHandler handler;
@ -118,24 +116,8 @@ public class QueueManager implements Consumer<Song>
this.playlist = playlist; this.playlist = playlist;
} }
public Mode getMode()
{
return mode;
}
public void setMode(Mode mode)
{
this.mode = mode;
}
public boolean restartSong() public boolean restartSong()
{ {
throw new UnsupportedOperationException("Not supported yet."); throw new UnsupportedOperationException("Not supported yet.");
} }
public enum Mode
{
JUKEBOX,
PLAYLIST;
}
} }

View File

@ -60,6 +60,7 @@ public class Song
{ {
Song song = new Song(new URL(map.string("url"))); Song song = new Song(new URL(map.string("url")));
song.setArtist(map.string("artist")); song.setArtist(map.string("artist"));
song.setTitle(map.string("title"));
song.setLocation(new File(map.string("location"))); song.setLocation(new File(map.string("location")));
song.setNumber(map.integer("num")); song.setNumber(map.integer("num"));
song.setKept(Boolean.parseBoolean(map.string("kept"))); song.setKept(Boolean.parseBoolean(map.string("kept")));

View File

@ -18,11 +18,8 @@ package moe.nekojimi.chords;
import java.net.URL; import java.net.URL;
import java.util.List; import java.util.List;
import moe.nekojimi.chords.commands.Invocation;
import moe.nekojimi.musicsearcher.Result; import moe.nekojimi.musicsearcher.Result;
import net.dv8tion.jda.api.MessageBuilder;
import net.dv8tion.jda.api.entities.Message;
import net.dv8tion.jda.api.requests.restaction.MessageAction;
import net.dv8tion.jda.internal.entities.DataMessage;
/** /**
* *
@ -30,9 +27,7 @@ import net.dv8tion.jda.internal.entities.DataMessage;
*/ */
public class SongRequest public class SongRequest
{ {
private Invocation invocation;
private Message requestMessage;
private Message responseMessage;
private String query; private String query;
private URL url; private URL url;
@ -43,21 +38,6 @@ public class SongRequest
private Song song; private Song song;
@SuppressWarnings("null")
public void respond(String text)
{
MessageAction action = null;
if (responseMessage == null)
{
action = requestMessage.reply(text);
} else
{
action = responseMessage.editMessage(text);
}
responseMessage = action.complete();
}
public List<Result> getSearchResults() public List<Result> getSearchResults()
@ -80,25 +60,35 @@ public class SongRequest
this.result = result; this.result = result;
} }
public Message getRequestMessage() public Invocation getInvocation()
{ {
return requestMessage; return invocation;
} }
public void setRequestMessage(Message requestMessage) public void setInvocation(Invocation invocation)
{ {
this.requestMessage = requestMessage; this.invocation = invocation;
} }
public Message getResponseMessage() // public Message getRequestMessage()
{ // {
return responseMessage; // return requestMessage;
} // }
//
public void setResponseMessage(Message responseMessage) // public void setRequestMessage(Message requestMessage)
{ // {
this.responseMessage = responseMessage; // this.requestMessage = requestMessage;
} // }
//
// public Message getResponseMessage()
// {
// return responseMessage;
// }
//
// public void setResponseMessage(Message responseMessage)
// {
// this.responseMessage = responseMessage;
// }
public String getQuery() public String getQuery()
{ {

View File

@ -132,7 +132,7 @@ public class TrackPlayer implements Closeable
public void close() throws IOException public void close() throws IOException
{ {
input.close(); input.close(); //q
} }

View File

@ -5,9 +5,7 @@
*/ */
package moe.nekojimi.chords.commands; package moe.nekojimi.chords.commands;
import java.util.List;
import moe.nekojimi.chords.Main; import moe.nekojimi.chords.Main;
import net.dv8tion.jda.api.events.message.guild.GuildMessageReceivedEvent;
/** /**
* *
@ -26,7 +24,7 @@ public abstract class Command
this.keyword = keyword; this.keyword = keyword;
} }
public abstract void call(GuildMessageReceivedEvent event, List<String> arg); public abstract void call(Invocation invocation);
public String getKeyword() public String getKeyword()
{ {

View File

@ -16,9 +16,7 @@
*/ */
package moe.nekojimi.chords.commands; package moe.nekojimi.chords.commands;
import java.util.List;
import moe.nekojimi.chords.Main; import moe.nekojimi.chords.Main;
import net.dv8tion.jda.api.events.message.guild.GuildMessageReceivedEvent;
public class HelpCommand extends Command public class HelpCommand extends Command
{ {
@ -29,7 +27,7 @@ public class HelpCommand extends Command
} }
@Override @Override
public void call(GuildMessageReceivedEvent event, List<String> arg) public void call(Invocation invocation)
{ {
String help = "Commands available:\n" String help = "Commands available:\n"
+ "!join <Channel> - Joins a voice channel\n" + "!join <Channel> - Joins a voice channel\n"
@ -44,7 +42,7 @@ public class HelpCommand extends Command
// { // {
// help += "!" + key + ":" // help += "!" + key + ":"
// } // }
event.getChannel().sendMessage(help).queue(); invocation.respond(help);
} }
} }

View File

@ -0,0 +1,89 @@
/*
* Copyright (C) 2022 jimj316
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
package moe.nekojimi.chords.commands;
import java.util.List;
import net.dv8tion.jda.api.entities.Message;
import net.dv8tion.jda.api.events.message.guild.GuildMessageReceivedEvent;
import net.dv8tion.jda.api.requests.restaction.MessageAction;
/**
*
* @author jimj316
*/
public class Invocation
{
private Message requestMessage;
private Message responseMessage;
private final GuildMessageReceivedEvent event;
private final List<String> args;
public Invocation(GuildMessageReceivedEvent event, List<String> args)
{
this.event = event;
this.args = args;
}
public GuildMessageReceivedEvent getEvent()
{
return event;
}
public List<String> getArgs()
{
return args;
}
@SuppressWarnings("null")
public void respond(String text)
{
MessageAction action = null;
if (responseMessage == null)
{
action = requestMessage.reply(text);
} else
{
action = responseMessage.editMessage(text);
}
responseMessage = action.complete();
}
public Message getRequestMessage()
{
return requestMessage;
}
public void setRequestMessage(Message requestMessage)
{
this.requestMessage = requestMessage;
}
public Message getResponseMessage()
{
return responseMessage;
}
public void setResponseMessage(Message responseMessage)
{
this.responseMessage = responseMessage;
}
}

View File

@ -6,7 +6,6 @@
package moe.nekojimi.chords.commands; package moe.nekojimi.chords.commands;
import java.util.List; import java.util.List;
import java.util.Optional;
import moe.nekojimi.chords.Main; import moe.nekojimi.chords.Main;
import net.dv8tion.jda.api.entities.*; import net.dv8tion.jda.api.entities.*;
import net.dv8tion.jda.api.events.message.guild.GuildMessageReceivedEvent; import net.dv8tion.jda.api.events.message.guild.GuildMessageReceivedEvent;
@ -20,8 +19,10 @@ public class JoinCommand extends Command
} }
@Override @Override
public void call(GuildMessageReceivedEvent event, List<String> args) public void call(Invocation invocation)
{ {
GuildMessageReceivedEvent event = invocation.getEvent();
List<String> args = invocation.getArgs();
TextChannel textChannel = event.getChannel(); TextChannel textChannel = event.getChannel();
VoiceChannel channel = null; VoiceChannel channel = null;
if (args.isEmpty() || args.get(0).isBlank()) if (args.isEmpty() || args.get(0).isBlank())

View File

@ -18,7 +18,7 @@ public class LeaveCommand extends Command
} }
@Override @Override
public void call(GuildMessageReceivedEvent event, List<String> arg) public void call(Invocation invocation)
{ {
if (bot.getCurrentVoiceChannel() != null) if (bot.getCurrentVoiceChannel() != null)
{ {

View File

@ -25,7 +25,6 @@ import moe.nekojimi.chords.Main;
import moe.nekojimi.chords.SongRequest; import moe.nekojimi.chords.SongRequest;
import moe.nekojimi.musicsearcher.Query; import moe.nekojimi.musicsearcher.Query;
import moe.nekojimi.musicsearcher.Result; import moe.nekojimi.musicsearcher.Result;
import net.dv8tion.jda.api.events.message.guild.GuildMessageReceivedEvent;
/** /**
* *
@ -44,13 +43,13 @@ public class PlayCommand extends Command
} }
@Override @Override
public void call(GuildMessageReceivedEvent event, List<String> arg) public void call(Invocation invocation)
{ {
SongRequest request = new SongRequest(); SongRequest request = new SongRequest();
request.setRequestMessage(event.getMessage()); request.setInvocation(invocation);
try try
{ {
final URL url = new URL(arg.get(0)); final URL url = new URL(invocation.getArgs().get(0));
request.setUrl(url); request.setUrl(url);
bot.queueDownload(request); bot.queueDownload(request);
@ -61,7 +60,7 @@ public class PlayCommand extends Command
{ {
try try
{ {
int index = Integer.parseInt(arg.get(0)); int index = Integer.parseInt(invocation.getArgs().get(0));
int size = request.getSearchResults().size(); int size = request.getSearchResults().size();
if (index >= 1 && index <= size) if (index >= 1 && index <= size)
{ {
@ -70,9 +69,9 @@ public class PlayCommand extends Command
bot.queueDownload(request); bot.queueDownload(request);
// event.getChannel().sendMessage("Song removed.").queue(); // event.getChannel().sendMessage("Song removed.").queue();
} else if (size > 1) } else if (size > 1)
event.getChannel().sendMessage("That's not a number between 1 and " + size + "!").queue(); invocation.respond("That's not a number between 1 and " + size + "!");
else if (size == 1) else if (size == 1)
event.getChannel().sendMessage("There's only one song and that's not one of them!").queue(); invocation.respond("There's only one song and that's not one of them!");
return; return;
} catch (NumberFormatException nfx) } catch (NumberFormatException nfx)
@ -82,13 +81,13 @@ public class PlayCommand extends Command
} }
// otherwise, try searching // otherwise, try searching
CompletableFuture<List<Result>> search = bot.getSearcher().search(Query.fullText(arg.stream().reduce((t, u) -> t + " " + u).get())); CompletableFuture<List<Result>> search = bot.getSearcher().search(Query.fullText(invocation.getArgs().stream().reduce((t, u) -> t + " " + u).get()));
event.getChannel().sendMessage("Searching for \"" + arg + "\" ...").queue(); invocation.respond("Searching for \"" + invocation.getArgs() + "\" ...");
search.orTimeout(30, TimeUnit.SECONDS).whenCompleteAsync((List<Result> results, Throwable exec) -> search.orTimeout(30, TimeUnit.SECONDS).whenCompleteAsync((List<Result> results, Throwable exec) ->
{ {
if (exec != null) if (exec != null)
{ {
event.getChannel().sendMessage("Failed to search! Reason: " + exec.getMessage()).queue(); invocation.respond("Failed to search! Reason: " + exec.getMessage());
return; return;
} }
@ -97,7 +96,7 @@ public class PlayCommand extends Command
if (results.isEmpty()) if (results.isEmpty())
{ {
event.getChannel().sendMessage("Found nothing! :(").queue(); invocation.respond("Found nothing! :(");
return; return;
} }
@ -125,7 +124,7 @@ public class PlayCommand extends Command
i++; i++;
} }
resultString += "Type eg. `!play 1` to select"; resultString += "Type eg. `!play 1` to select";
event.getChannel().sendMessage(resultString).queue(); invocation.respond(resultString);
}); });
// event.getChannel().sendMessage("That's not a valid URL you idiot! " + ex.getMessage()).queue(); // event.getChannel().sendMessage("That's not a valid URL you idiot! " + ex.getMessage()).queue();
} }

View File

@ -21,7 +21,6 @@ import java.util.Queue;
import moe.nekojimi.chords.Downloader; import moe.nekojimi.chords.Downloader;
import moe.nekojimi.chords.Main; import moe.nekojimi.chords.Main;
import moe.nekojimi.chords.Song; import moe.nekojimi.chords.Song;
import net.dv8tion.jda.api.events.message.guild.GuildMessageReceivedEvent;
public class QueueCommand extends Command public class QueueCommand extends Command
{ {
@ -32,7 +31,7 @@ public class QueueCommand extends Command
} }
@Override @Override
public void call(GuildMessageReceivedEvent event, List<String> arg) public void call(Invocation invocation)
{ {
String message = ">>> "; String message = ">>> ";
int i = 1; int i = 1;
@ -68,7 +67,7 @@ public class QueueCommand extends Command
if (downloadQueue.isEmpty() && songQueue.isEmpty()) if (downloadQueue.isEmpty() && songQueue.isEmpty())
message += ":mailbox_with_no_mail: The track queue is empty."; message += ":mailbox_with_no_mail: The track queue is empty.";
// :inbox_tray: // :inbox_tray:
event.getChannel().sendMessage(message).queue(); invocation.respond(message);
} }
} }

View File

@ -16,9 +16,7 @@
*/ */
package moe.nekojimi.chords.commands; package moe.nekojimi.chords.commands;
import java.util.List;
import moe.nekojimi.chords.Main; import moe.nekojimi.chords.Main;
import net.dv8tion.jda.api.events.message.guild.GuildMessageReceivedEvent;
public class RemoveCommand extends Command public class RemoveCommand extends Command
{ {
@ -29,23 +27,23 @@ public class RemoveCommand extends Command
} }
@Override @Override
public void call(GuildMessageReceivedEvent event, List<String> arg) public void call(Invocation invocation)
{ {
try try
{ {
int i = Integer.parseInt(arg.get(0)); int i = Integer.parseInt(invocation.getArgs().get(0));
boolean removed = bot.getQueueManager().removeSong(i - 1); boolean removed = bot.getQueueManager().removeSong(i - 1);
final int size = bot.getQueueManager().getJukeboxQueue().size(); final int size = bot.getQueueManager().getJukeboxQueue().size();
if (removed) if (removed)
event.getChannel().sendMessage("Song removed.").queue(); invocation.respond("Song removed.");
else if (size > 1) else if (size > 1)
event.getChannel().sendMessage("That's not a number between 1 and " + size + "!").queue(); invocation.respond("That's not a number between 1 and " + size + "!");
else if (size == 1) else if (size == 1)
event.getChannel().sendMessage("There's only one song to remove and that's not one of them!").queue(); invocation.respond("There's only one song to remove and that's not one of them!");
} catch (NumberFormatException ex) } catch (NumberFormatException ex)
{ {
event.getChannel().sendMessage(arg + " isn't a number!").queue(); invocation.respond(invocation.getArgs().get(0) + " isn't a number!");
} }
} }

View File

@ -16,9 +16,7 @@
*/ */
package moe.nekojimi.chords.commands; package moe.nekojimi.chords.commands;
import java.util.List;
import moe.nekojimi.chords.Main; import moe.nekojimi.chords.Main;
import net.dv8tion.jda.api.events.message.guild.GuildMessageReceivedEvent;
public class RestartCommand extends Command public class RestartCommand extends Command
{ {
@ -29,14 +27,14 @@ public class RestartCommand extends Command
} }
@Override @Override
public void call(GuildMessageReceivedEvent event, List<String> arg) public void call(Invocation invocation)
{ {
// TODO: this needs to clear the current data queue // TODO: this needs to clear the current data queue
boolean ok = bot.getQueueManager().restartSong(); boolean ok = bot.getQueueManager().restartSong();
if (ok) if (ok)
event.getChannel().sendMessage("Restarted current song!").queue(); invocation.respond("Restarted current song!");
else else
event.getChannel().sendMessage("Cannot restart!").queue(); invocation.respond("Cannot restart!");
} }
} }

View File

@ -16,9 +16,7 @@
*/ */
package moe.nekojimi.chords.commands; package moe.nekojimi.chords.commands;
import java.util.List;
import moe.nekojimi.chords.Main; import moe.nekojimi.chords.Main;
import net.dv8tion.jda.api.events.message.guild.GuildMessageReceivedEvent;
public class SkipCommand extends Command public class SkipCommand extends Command
{ {
@ -29,13 +27,13 @@ public class SkipCommand extends Command
} }
@Override @Override
public void call(GuildMessageReceivedEvent event, List<String> arg) public void call(Invocation invocation)
{ {
boolean ok = bot.getMusicHandler().nextSong(true); boolean ok = bot.getMusicHandler().nextSong(true);
if (ok) if (ok)
event.getChannel().sendMessage("Skipped to next song!").queue(); invocation.respond("Skipped to next song!");
else else
event.getChannel().sendMessage("There's no more songs!").queue(); invocation.respond("There's no more songs!");
} }
} }