Server/project/src/services/RagfairCategoriesService.ts

import { inject, injectable } from "tsyringe";

import { PaymentHelper } from "@spt-aki/helpers/PaymentHelper";
import { IRagfairOffer } from "@spt-aki/models/eft/ragfair/IRagfairOffer";
import { ISearchRequestData, OfferOwnerType } from "@spt-aki/models/eft/ragfair/ISearchRequestData";
import { MemberCategory } from "@spt-aki/models/enums/MemberCategory";
import { ILogger } from "@spt-aki/models/spt/utils/ILogger";

@injectable()
export class RagfairCategoriesService
{
    constructor(
        @inject("WinstonLogger") protected logger: ILogger,
        @inject("PaymentHelper") protected paymentHelper: PaymentHelper,
    )
    {}

    /**
     * Get a dictionary of each item the play can see in their flea menu, filtered by what is available for them to buy
     * @param offers All offers in flea
     * @param searchRequestData Search criteria requested
     * @param fleaUnlocked Can player see full flea yet (level 15 by default)
     * @returns KVP of item tpls + count of offers
     */
    public getCategoriesFromOffers(offers: IRagfairOffer[], searchRequestData: ISearchRequestData, fleaUnlocked: boolean): Record<string, number>
    {
        const validOffersForPlayerToSee = {};
        for (const offer of offers)
        {
            const isTraderOffer = offer.user.memberType === MemberCategory.TRADER;

            // Not level 15 and offer is from player, skip
            if (!fleaUnlocked && !isTraderOffer)
            {
                continue;
            }

            // Remove items not for money when `removeBartering` is enabled
            if (searchRequestData.removeBartering && (offer.requirements.length > 1 || !this.paymentHelper.isMoneyTpl(offer.requirements[0]._tpl)))
            {
                continue;
            }

            // Remove when filter set to players only + offer is from trader
            if (searchRequestData.offerOwnerType === OfferOwnerType.PLAYEROWNERTYPE && isTraderOffer) 
            {
                continue;
            }

            // Remove when filter set to traders only + offer is not from trader
            if (searchRequestData.offerOwnerType === OfferOwnerType.TRADEROWNERTYPE && !isTraderOffer) 
            {
                continue;
            }

            const itemTpl = offer.items[0]._tpl

            if (!validOffersForPlayerToSee[itemTpl])
            {
                validOffersForPlayerToSee[itemTpl] = 1;
            }
            else
            {
                validOffersForPlayerToSee[itemTpl]++;
            }
        }

        return validOffersForPlayerToSee;

    }
}