AltitudeBot/src/main/java/com/alttd/modalManager/modals/ModalEvidence.java

package com.alttd.modalManager.modals;

import com.alttd.database.queries.commandOutputChannels.CommandOutputChannels;
import com.alttd.database.queries.commandOutputChannels.OutputType;
import com.alttd.modalManager.DiscordModal;
import com.alttd.util.Util;
import net.dv8tion.jda.api.EmbedBuilder;
import net.dv8tion.jda.api.entities.Guild;
import net.dv8tion.jda.api.entities.Member;
import net.dv8tion.jda.api.entities.MessageEmbed;
import net.dv8tion.jda.api.entities.channel.middleman.GuildChannel;
import net.dv8tion.jda.api.entities.channel.middleman.GuildMessageChannel;
import net.dv8tion.jda.api.events.interaction.ModalInteractionEvent;
import net.dv8tion.jda.api.interactions.components.ActionRow;
import net.dv8tion.jda.api.interactions.components.Modal;
import net.dv8tion.jda.api.interactions.components.text.TextInput;
import net.dv8tion.jda.api.interactions.components.text.TextInputStyle;
import net.dv8tion.jda.api.interactions.modals.ModalMapping;
import net.dv8tion.jda.api.requests.RestAction;
import net.dv8tion.jda.api.requests.restaction.interactions.ReplyCallbackAction;

import java.util.List;

public class ModalEvidence extends DiscordModal {

    @Override
    public String getModalId() {
        return "evidence";
    }

    @Override
    public void execute(ModalInteractionEvent event) {
        List<ModalMapping> modalMappings = event.getValues();
        if (modalMappings.size() != 4) {
            event.replyEmbeds(Util.genericErrorEmbed("Error", "Found the wrong number of fields in your form input"))
                    .setEphemeral(true).queue(RestAction.getDefaultSuccess(), Util::handleFailure);
            return;
        }

        String user = modalMappings.get(0).getAsString();
        String punishmentType = modalMappings.get(1).getAsString();
        String reason = modalMappings.get(2).getAsString();
        String evidence = modalMappings.get(3).getAsString();

        Guild guild = event.getGuild();
        if (guild == null) {
            event.replyEmbeds(Util.genericErrorEmbed("Error", "Couldn't find this guild"))
                    .setEphemeral(true).queue(RestAction.getDefaultSuccess(), Util::handleFailure);
            return;
        }

        GuildChannel outputChannel = CommandOutputChannels.getOutputChannel(guild, OutputType.EVIDENCE);
        if (outputChannel == null) {
            event.replyEmbeds(Util.genericErrorEmbed("Error", "This guild does not have an evidence channel or it's not the right channel type"))
                    .setEphemeral(true).queue(RestAction.getDefaultSuccess(), Util::handleFailure);
            return;
        }

        if (!(outputChannel instanceof GuildMessageChannel channel)) {
            event.replyEmbeds(Util.genericErrorEmbed("Error", "Invalid Evidence channel type: " + outputChannel.getType()))
                    .setEphemeral(true).queue(RestAction.getDefaultSuccess(), Util::handleFailure);
            return;
        }

        Member member = event.getMember();
        if (member == null) {
            event.replyEmbeds(Util.genericErrorEmbed("Error", "This command can only be executed from a guild"))
                    .setEphemeral(true).queue(RestAction.getDefaultSuccess(), Util::handleFailure);
            return;
        }

        MessageEmbed evidenceEmbed = new EmbedBuilder()
                .setDescription("Evidence by " + member.getAsMention())
                .addField("`" + user + "`", "", false)
                .addField(punishmentType, reason, false)
                .addField("", evidence, false)
                .setFooter(member.getIdLong() + "")
                .build();

        ReplyCallbackAction replyCallbackAction = event.deferReply(true);
        channel.sendMessageEmbeds(evidenceEmbed)
                .queue(success -> replyCallbackAction.setEmbeds(Util.genericSuccessEmbed("Success", "Your evidence was submitted to the evidence channel!"), evidenceEmbed)
                                .setEphemeral(true).queue(RestAction.getDefaultSuccess(), Util::handleFailure),
                        Util::handleFailure);
    }

    @Override
    public Modal getModal() {
        TextInput user = TextInput.create("user", "User", TextInputStyle.SHORT)
                .setPlaceholder("username/id")
                .setRequiredRange(1, 256)
                .setRequired(true)
                .build();

        TextInput punishmentType = TextInput.create("punishment-type", "Punishment Type", TextInputStyle.SHORT)
                .setPlaceholder("punishment type")
                .setRequiredRange(3, 256)
                .setRequired(true)
                .build();

        TextInput reason = TextInput.create("reason", "Reason", TextInputStyle.SHORT)
                .setPlaceholder("punishment reason")
                .setRequiredRange(10, 256)
                .setRequired(true)
                .build();

        TextInput evidence = TextInput.create("evidence", "Evidence", TextInputStyle.PARAGRAPH)
                .setPlaceholder("evidence")
                .setRequiredRange(10, 1024)
                .setRequired(true)
                .build();

        return Modal.create(getModalId(), "Evidence")
                .addActionRows(ActionRow.of(user), ActionRow.of(punishmentType), ActionRow.of(reason), ActionRow.of(evidence))
                .build();
    }
}