HidekoBot/src/main/java/wtf/beatrice/hidekobot/commands/slash/CoinFlipCommand.java

package wtf.beatrice.hidekobot.commands.slash;

import net.dv8tion.jda.api.entities.Message;
import net.dv8tion.jda.api.entities.User;
import net.dv8tion.jda.api.entities.emoji.Emoji;
import net.dv8tion.jda.api.events.interaction.command.SlashCommandInteractionEvent;
import net.dv8tion.jda.api.events.interaction.component.ButtonInteractionEvent;
import net.dv8tion.jda.api.interactions.components.ActionRow;
import net.dv8tion.jda.api.interactions.components.buttons.Button;
import org.jetbrains.annotations.NotNull;
import wtf.beatrice.hidekobot.Configuration;
import wtf.beatrice.hidekobot.utils.RandomUtil;

import java.util.List;

public class CoinFlipCommand
{

    private final Button reflipButton = Button.primary("coinflip_reflip", "Flip again")
            .withEmoji(Emoji.fromFormatted("\uD83E\uDE99"));

    public void runSlashCommand(@NotNull SlashCommandInteractionEvent event)
    {
        // perform coin flip
        event.reply(genRandom())
                .addActionRow(reflipButton)
                .queue((interaction) ->
                {
                    // set the command as expiring and restrict it to the user who ran it
                    interaction.retrieveOriginal().queue((message) ->
                    {
                        trackAndRestrict(message, event.getUser());
                    }, (error) -> {});
                }, (error) -> {});

    }

    public void buttonReFlip(ButtonInteractionEvent event)
    {
        // check if the user interacting is the same one who ran the command
        if(!(Configuration.getDatabaseManager().isUserTrackedFor(event.getUser().getId(), event.getMessageId())))
        {
            event.reply("❌ You did not run this command!").setEphemeral(true).queue();
            return;
        }

        // set old message's button as disabled
        List<ActionRow> actionRows = event.getMessage().getActionRows();
        actionRows.set(0, actionRows.get(0).asDisabled());
        event.editComponents(actionRows).queue();

        // perform coin flip
        event.getHook().sendMessage(genRandom())
                .addActionRow(reflipButton)
                .queue((message) ->
                {
                    // set the command as expiring and restrict it to the user who ran it
                    trackAndRestrict(message, event.getUser());
                }, (error) -> {});
    }

    private void trackAndRestrict(Message replyMessage, User user)
    {
        String replyMessageId = replyMessage.getId();
        String replyChannelId = replyMessage.getId();
        String replyGuildId = replyMessage.getGuild().getId();
        String userId = user.getId();

        Configuration.getDatabaseManager().queueDisabling(replyGuildId, replyChannelId, replyMessageId);
        Configuration.getDatabaseManager().trackRanCommandReply(replyGuildId, replyChannelId, replyMessageId, userId);
    }

    private String genRandom()
    {
        int rand = RandomUtil.getRandomNumber(0, 1);
        String msg;

        if(rand == 1)
        {
            msg = ":coin: It's **Heads**!";
        } else {
            msg = "It's **Tails**! :coin:";
        }

        return msg;
    }

}