toho-miku/commands/drop.js

const { SlashCommandBuilder, ActionRowBuilder, ButtonBuilder, ButtonStyle, ComponentType, AttachmentBuilder } = require("discord.js");
const { Card, User, Character, DropHistory, sequelize } = require("../models");
const { customAlphabet } = require("nanoid");
const { CardUtils, UserUtils, ReplyUtils, GeneralUtils, Rendering } = require("../util");
const { QUALITY } = require("../config/constants");
const card = require("../models/card");

module.exports = {
    data: new SlashCommandBuilder()
    .setName("drop")
    .setDescription("Drop a card"),
    
    async execute(interaction) {
        const user = await UserUtils.getUserByDiscordId(interaction.member.id);
        
        let permissionLevel = await UserUtils.getPermissionLevel(interaction.member);
        const cooldowns = await UserUtils.getCooldowns(user);
        if (cooldowns.dropCooldown > 0 && permissionLevel < 2) {
            interaction.reply({
                content: `You can drop more cards in ${Math.floor((cooldowns.dropCooldown % 3600000) / 60000)} minutes`,
                ephemeral: false
            });
            return;
        }
        
        //Generate 3 cards, each is persisted with an initial userId of NULL
        const cards = [];
        let characters = await Character.findAll({
            where: {
                enabled: true
            },
            order: sequelize.random(),
            limit: 3
        });

        for (let i = 0; i < 3; i++) {
            //generate quality based on drop rate
            let quality = undefined;
            let roll = Math.random() * 100;
            if (roll <= 45.0) {
                quality = QUALITY.BAD; //45%
            }
            if (roll > 45.0) {
                quality = QUALITY.OK; //25%
            }
            if (roll > 70.0) {
                quality = QUALITY.GOOD; //15%
            }
            if (roll > 85.0) {
                quality = QUALITY.GREAT; //10%
            }
            if (roll > 95.0) {
            quality = QUALITY.EXCELLENT; //5%
            }
            if (roll > 99.9) {
                quality = QUALITY.SHINY; //0.1%
            }

            //get number of characters in database
            const characterId = characters[i].id
            console.log(`characterId: ${characterId}`);
            //random number between 1 and 6
            let newCard = await Card.create({
                characterId: characterId,
                identifier: CardUtils.generateIdentifier(),
                quality: quality,
                printNr: await CardUtils.getNextPrintNumber(characterId),
                
            });
            cards.push(newCard);
        }

        cards.sort((a, b) => a.characterId - b.characterId);
        
        const row = new ActionRowBuilder();
        let deckImage = await Rendering.renderCardStack(cards);

        for (let i = 0; i < cards.length; i++) {
            //Add claim button for each card
            row.addComponents(
				new ButtonBuilder()
					.setCustomId(`claim-${i}-${cards[i].identifier}`)
					.setLabel(`Claim  ${i+1}`)
					.setStyle(ButtonStyle.Primary),
			);            
        }
        //add 10 experience to the user
        await user.addExperience(10);

        const file = new AttachmentBuilder(deckImage);
    
        const message = await interaction.reply({ content: 'asd', components: [row], files: [file], fetchReply: true });

		//const message = await interaction.reply({ content: reply, components: [row], fetchReply: true });
        //set users drop cooldown
        await UserUtils.setCooldown(user, "drop", await GeneralUtils.getBotProperty("dropTimeout"));

        const collector = message.createMessageComponentCollector({ componentType: ComponentType.Button, time: 25000 });
        
        let collectionReplies = [];
        collector.on('collect', async i => {
            let cardId = i.customId.split("-")[1];
            if (await cards[cardId].userId) { i.reply({ content: "This card has already been claimed!", ephemeral: true }); return; }

            let claimUser = await UserUtils.getUserByDiscordId(i.user.id);
            let cooldowns = await UserUtils.getCooldowns(claimUser);
            let permissionLevel = await UserUtils.getPermissionLevel(i.member);
            if (cooldowns.pullCooldown > 0 && permissionLevel < 2) {
                i.reply({
                    content: `You can claim more cards in ${Math.floor((cooldowns.pullCooldown % 3600000) / 60000)} minutes`,
                    ephemeral: false
                });
                return;
            }

            if (claimUser) {
                //Update card with the user id
                cards[cardId].userId = claimUser.id;
                await UserUtils.setCooldown(claimUser, "pull", await GeneralUtils.getBotProperty("pullTimeout"));
                await cards[cardId].save();
                await claimUser.addExperience(5);
                //fetch character name from database given the character id
                let character = await Character.findOne({
                    attributes: ["name"],
                    where: {
                        id: cards[cardId].characterId
                    }
                });
                let reply = await i.reply({ content: `${i.user} (${claimUser.id}) claimed a card!  [${cards[cardId].identifier}]`, ephemeral: false, fetchReply: true  });
                collectionReplies.push({ ref: reply, characterName: character.name, card: cards[cardId] });
                let newRow = ReplyUtils.recreateComponents(i.message.components);
                newRow.components[cardId].setLabel("Claimed");
                newRow.components[cardId].setStyle(ButtonStyle.Success);
                newRow.components[cardId].setDisabled(true);
                //let deckImage = await Rendering.renderCardStack(cards);
                message.edit({ components: [newRow] });
            }
        });
        
        collector.on('end', async collected => {
            let dropHistory = {};
            for (reply of collectionReplies) {
                //TODO: strings shouldn't be inlined. Needs refactoring
                let notableProps = [];
                if (reply.card.printNr == 1) notableProps.push("📦 First Print");
                if (reply.card.quality == 6) notableProps.push("<a:sparklu:1019505245572837428> Shiny");

                reply.ref.edit({ content: `${reply.ref.content.replace('a card', reply.characterName)} ${notableProps.join()}` });
            }
            console.log(`Collected ${collected.size} interactions.`);
            for (let card of cards) {
                if (!card.userId) {
                    card.userId = 1;
                    await card.save();
                }
                let historyEntry = {
                    cardData: JSON.stringify(card),
                    ogUserId: card.userId,
                };
                dropHistory[card.identifier] = historyEntry;
            }

            //create new drop history entry
            let history = await DropHistory.create({
                dropData: JSON.stringify(dropHistory),
            });


            
            let deckImage = await Rendering.renderCardStack(cards);
            message.edit({ components: [], files: [new AttachmentBuilder(deckImage)] });
        });
        
    }
}