card-drop/src/helpers/CardSearchHelper.ts

import {ActionRowBuilder, AttachmentBuilder, ButtonBuilder, ButtonStyle, EmbedBuilder} from "discord.js";
import Fuse from "fuse.js";
import {CoreClient} from "../client/client.js";
import Inventory from "../database/entities/app/Inventory.js";
import {readFileSync} from "fs";
import path from "path";
import AppLogger from "../client/appLogger.js";
import GetCardsHelper from "./DropHelpers/GetCardsHelper.js";
import DropEmbedHelper from "./DropHelpers/DropEmbedHelper.js";

interface ReturnedPage {
    embed: EmbedBuilder,
    row: ActionRowBuilder<ButtonBuilder>,
    attachments: AttachmentBuilder[],
    results: string[],
}

export default class CardSearchHelper {
    public static async GenerateSearchQuery(query: string, userid: string, pages: number): Promise<ReturnedPage | undefined> {
        AppLogger.LogSilly("CardSearchHelper/GenerateSearchQuery", `Parameters: query=${query}, userid=${userid}, pages=${pages}`);

        const fzf = new Fuse(CoreClient.Cards.flatMap(x => x.cards), { keys: ["name"] });
        const entries = fzf.search(query)
            .splice(0, pages);

        const entry = entries[0];
        const results = entries
            .flatMap(x => x.item.id);

        if (!entry) {
            AppLogger.LogVerbose("CardSearchHelper/GenerateSearchQuery", `Unable to find entry: ${query}`);

            return undefined;
        }

        const card = GetCardsHelper.GetCardByCardNumber(entry.item.id);

        if (!card) return undefined;

        const attachments = [];
        let imageFileName = "";

        if (!(card.card.path.startsWith("http://") || card.card.path.startsWith("https://"))) {
            const image = readFileSync(path.join(process.env.DATA_DIR!, "cards", card.card.path));
            imageFileName = card.card.path.split("/").pop()!;

            const attachment = new AttachmentBuilder(image, { name: imageFileName });

            attachments.push(attachment);
        }

        const inventory = await Inventory.FetchOneByCardNumberAndUserId(userid, card.card.id);
        const quantityClaimed = inventory?.Quantity ?? 0;

        const embed = DropEmbedHelper.GenerateDropEmbed(card, quantityClaimed, imageFileName);

        const row = new ActionRowBuilder<ButtonBuilder>()
            .addComponents(
                new ButtonBuilder()
                    .setCustomId(`view 0 ${results.join(" ")}`)
                    .setLabel("Previous")
                    .setStyle(ButtonStyle.Primary)
                    .setDisabled(true),
                new ButtonBuilder()
                    .setCustomId(`view 2 ${results.join(" ")}`)
                    .setLabel("Next")
                    .setStyle(ButtonStyle.Primary)
                    .setDisabled(pages == 1));

        return { embed, row, attachments, results };
    }

    public static async GenerateSearchPageFromQuery(results: string[], userid: string, page: number): Promise<ReturnedPage | undefined> {
        const currentPageId = results[page - 1];

        const card = GetCardsHelper.GetCardByCardNumber(currentPageId);

        if (!card) {
            AppLogger.LogError("CardSearchHelper/GenerateSearchPageFromQuery", `Unable to find card by id: ${currentPageId}.`);

            return undefined;
        }

        const attachments = [];
        let imageFileName = "";

        if (!(card.card.path.startsWith("http://") || card.card.path.startsWith("https://"))) {
            const image = readFileSync(path.join(process.env.DATA_DIR!, "cards", card.card.path));
            imageFileName = card.card.path.split("/").pop()!;

            const attachment = new AttachmentBuilder(image, { name: imageFileName });

            attachments.push(attachment);
        }

        const inventory = await Inventory.FetchOneByCardNumberAndUserId(userid, card.card.id);
        const quantityClaimed = inventory?.Quantity ?? 0;

        const embed = DropEmbedHelper.GenerateDropEmbed(card, quantityClaimed, imageFileName);

        const row = new ActionRowBuilder<ButtonBuilder>()
            .addComponents(
                new ButtonBuilder()
                    .setCustomId(`view ${page - 1} ${results.join(" ")}`)
                    .setLabel("Previous")
                    .setStyle(ButtonStyle.Primary)
                    .setDisabled(page - 1 == 0),
                new ButtonBuilder()
                    .setCustomId(`view ${page + 1} ${results.join(" ")}`)
                    .setLabel("Next")
                    .setStyle(ButtonStyle.Primary)
                    .setDisabled(page == results.length));

        return { embed, row, attachments, results };
    }
}