Add new invocation system to command structure.

jda-v5
Nekojimi 3 years ago
parent 916e73eebd
commit ca3ebd712b
  1. 32
      src/main/java/moe/nekojimi/chords/Main.java
  2. 60
      src/main/java/moe/nekojimi/chords/SongRequest.java
  3. 4
      src/main/java/moe/nekojimi/chords/commands/Command.java
  4. 6
      src/main/java/moe/nekojimi/chords/commands/HelpCommand.java
  5. 89
      src/main/java/moe/nekojimi/chords/commands/Invocation.java
  6. 5
      src/main/java/moe/nekojimi/chords/commands/JoinCommand.java
  7. 2
      src/main/java/moe/nekojimi/chords/commands/LeaveCommand.java
  8. 23
      src/main/java/moe/nekojimi/chords/commands/PlayCommand.java
  9. 5
      src/main/java/moe/nekojimi/chords/commands/QueueCommand.java
  10. 14
      src/main/java/moe/nekojimi/chords/commands/RemoveCommand.java
  11. 8
      src/main/java/moe/nekojimi/chords/commands/RestartCommand.java
  12. 8
      src/main/java/moe/nekojimi/chords/commands/SkipCommand.java

@ -110,7 +110,7 @@ public final 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
{ {
@ -124,12 +124,12 @@ public final 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());
} }
@ -216,7 +216,7 @@ public final 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
@ -224,7 +224,7 @@ public final 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("!"))
@ -232,17 +232,21 @@ public final 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)
{ {
@ -262,16 +266,16 @@ public final 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;
} }

@ -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()
{ {

@ -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()
{ {

@ -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);
} }
} }

@ -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;
}
}

@ -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())

@ -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)
{ {

@ -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();
} }

@ -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);
} }
} }

@ -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!");
} }
} }

@ -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!");
} }
} }

@ -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!");
} }
} }

Loading…
Cancel
Save