code

package fr.noalegeek.pepite_dor_bot.commands.moderation;

import com.jagrosh.jdautilities.command.Command;
import com.jagrosh.jdautilities.command.CommandEvent;
import fr.noalegeek.pepite_dor_bot.Main;
import fr.noalegeek.pepite_dor_bot.enums.CommandCategories;
import fr.noalegeek.pepite_dor_bot.utils.helpers.MessageHelper;
import net.dv8tion.jda.api.Permission;
import net.dv8tion.jda.api.entities.*;

public class MuteCommand extends Command {
    public MuteCommand() {
        this.category = CommandCategories.STAFF.category;
        this.aliases = new String[]{"m", "mu", "mut"};
        this.name = "mute";
        this.arguments = "<identifiant/mention du membre> [raison]";
        this.help = "Mute définitivement un membre avec une raison ou non. Unmute si la personne est déjà mute.";
        this.cooldown = 5;
        this.example = "@NoaLeGeek spam";
        this.userPermissions = new Permission[]{Permission.MESSAGE_MANAGE};
        this.botPermissions = new Permission[]{Permission.MESSAGE_MANAGE};
        this.guildOnly = true;
    }

    @Override
    protected void execute(CommandEvent event) {
        User author = event.getAuthor();
        if (author.isBot()) return;
        String[] args = event.getArgs().split("\\s+");
        if (args.length > 2) {
            event.replyError(MessageHelper.syntaxError(author, this) + "Mettre une raison n'est pas obligatoire.");
            return;
        }
        String targetID = args[0].replaceAll("\\D+", "");
        Guild guild = event.getGuild();
        String mutedRoleID = Main.getServerConfig().mutedRole.get(guild.getId());
        Main.getJda().retrieveUserById(targetID).queue(user -> {
            guild.retrieveMember(user).queue(member -> {
                if (!event.getSelfMember().canInteract(member)) {
                    event.replyError(MessageHelper.formattedMention(author) + "Le bot n'a pas les permissions de faire cela.");
                    return;
                }
                if (!event.getMember().canInteract(member)) {
                    event.replyError(MessageHelper.formattedMention(author) + "Vous n'avez pas la permission de mute ce membre.");
                    return;
                }
                if (Main.getServerConfig().mutedRole.containsValue(event.getGuild().getId()) && event.getGuild().getRoleById(Main.getServerConfig().mutedRole.get(event.getGuild().getId())) != null) {
                if (args.length == 1) {
                    mute(event, member, null);
                } else {
                    mute(event, member, args[1]);
                }
            } else {
                RestAction.allOf(
                                event.getGuild().createRole().setName("Muted Role").setColor(0x010101).map(mutedRole -> Main.getServerConfig().mutedRole.put(event.getGuild().getId(), mutedRole.getId())))
                        .queue(unused -> {
                            List<RestAction<?>> restActionList = new ArrayList<>();
                            for (GuildChannel guildChannel : event.getGuild().getChannels()) {
                                restActionList.add(guildChannel.putPermissionOverride(event.getGuild().getRoleById(Main.getServerConfig().mutedRole.get(event.getGuild().getId()))).setDeny(Permission.MESSAGE_WRITE));
                            }
                            RestAction.allOf(restActionList).queue(unused1 -> {
                                        if (args.length == 1) {
                                            mute(event, member, null);
                                        } else {
                                            mute(event, member, args[1]);
                                        }
                                    }
                            );
                        });
                try {
                    Listener.saveConfigs();
                } catch (IOException e) {
                    e.printStackTrace();
                }
                event.replyWarning(MessageHelper.formattedMention(event.getAuthor()) + ":warning: Le rôle configuré par défaut n'est pas présent donc j'ai créé un nouveau rôle nommé \"Muted Role\".");
            }

            }, memberNull -> event.replyError(MessageHelper.formattedMention(author) + "Vous devez spécifié une personne présente sur le serveur."));
        }, userNull -> event.replyError(MessageHelper.formattedMention(author) + "Vous devez spécifié une personne existante."));
    }

    public static void mute(CommandEvent event, Member targetMember, String reason) {
        Guild guild = event.getGuild();
        User author = event.getAuthor();
        String mutedRoleID = Main.getServerConfig().mutedRole.get(guild.getId());
        if (targetMember.getRoles().contains(guild.getRoleById(mutedRoleID))) { // Unmute the target
            targetMember.getRoles().remove(guild.getRoleById(mutedRoleID));
            if (reason != null) {
                event.replySuccess(MessageHelper.formattedMention(author) + targetMember.getEffectiveName() + " a bien été démuter pour la raison " + reason + ".");
                return;
            }
            event.replySuccess(MessageHelper.formattedMention(author) + targetMember.getEffectiveName() + " a bien été démuter.");
        } else { // Mute the target
            guild.addRoleToMember(targetMember, guild.getRoleById(mutedRoleID)).queue();
            if (reason != null) {
                event.replySuccess(MessageHelper.formattedMention(author) + targetMember.getEffectiveName() + " a bien été muter pour la raison " + reason + ".");
                return;
            }
            event.replySuccess(MessageHelper.formattedMention(author) + targetMember.getEffectiveName() + " a bien été muter.");
        }
    }
}