oat/GMusicBot
1
0
Fork 0
forked from remote/MusicBot
Code Pull requests Actions Activity

Apply requested changes

Browse source
This commit is contained in:
MichailiK 2019-09-12 14:44:13 +02:00
parent 75abab2832
commit bc85fb9597
1 changed files with 46 additions and 32 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

78
src/main/java/com/jagrosh/jmusicbot/commands/dj/ForceRemoveCmd.java
View file

@ -1,60 +1,74 @@
package com.jagrosh.jmusicbot.commands.dj;
import com.jagrosh.jdautilities.command.CommandEvent;
import com.jagrosh.jdautilities.commons.utils.FinderUtil;
import com.jagrosh.jmusicbot.Bot;
import com.jagrosh.jmusicbot.audio.AudioHandler;
import com.jagrosh.jmusicbot.commands.MusicCommand;
import com.jagrosh.jmusicbot.commands.DJCommand;
import net.dv8tion.jda.core.entities.Member;
import net.dv8tion.jda.core.entities.User;
import java.util.List;
public class ForceRemoveCmd extends MusicCommand {
public ForceRemoveCmd(Bot bot) {
public class ForceRemoveCmd extends DJCommand
{
public ForceRemoveCmd(Bot bot)
{
super(bot);
this.name = "forceremove";
this.help = "removes all entries by the mentioned user from the queue";
this.arguments = "<@user|id>";
this.aliases = new String[]{"forcedelete"};
this.beListening = true;
this.help = "removes all entries by a user from the queue";
this.arguments = "<user>";
this.aliases = new String[]{"forcedelete", "modremove", "moddelete"};
this.beListening = false;
this.bePlaying = true;
}
@Override
public void doCommand(CommandEvent event) {
if (event.getArgs().isEmpty()) {
public void doCommand(CommandEvent event)
{
if (event.getArgs().isEmpty())
{
event.replyError("You need to mention a user!");
return;
}
AudioHandler handler = (AudioHandler) event.getGuild().getAudioManager().getSendingHandler();
if (handler.getQueue().isEmpty()) {
if (handler.getQueue().isEmpty())
{
event.replyError("There is nothing in the queue!");
return;
}
long target = -1;
String args = event.getArgs();
if (args.startsWith("<@") && args.endsWith(">")) {
try {
target = Long.parseLong(args.substring(2, args.length() - 1));
} catch (NumberFormatException ignored) {}
} else if (args.startsWith("<@!") && args.endsWith(">")) {
try {
target = Long.parseLong(args.substring(3, args.length() - 1));
} catch (NumberFormatException ignored) {}
} else {
try {
target = Long.parseLong(args);
} catch (NumberFormatException ignored) {}
}
if (target <= 0) {
event.replyError("You need to mention a user!");
User target;
List<Member> found = FinderUtil.findMembers(event.getArgs(), event.getGuild());
if(found.isEmpty())
{
event.replyError("Unable to find the user!");
return;
}
else if(found.size()>1)
{
event.replyWarning("Found multiple users!");
return;
}
int count = handler.getQueue().removeAll(target);
if (count == 0)
event.replyWarning("This user doesn't have any songs in the queue!");
else
event.replySuccess("Successfully removed their " + count + " entries.");
{
target = found.get(0).getUser();
}
int count = handler.getQueue().removeAll(target.getIdLong());
if (count == 0)
{
event.replyWarning(target.getName() + " doesn't have any songs in the queue!");
}
else
{
event.replySuccess("Successfully removed `" + count + "` entries from **" + target.getName() + "**#" + target.getDiscriminator() + ".");
}
}
}