SpaceNinjaServer/src/controllers/api/contributeToVaultController.ts

import { GuildMember } from "@/src/models/guildModel";
import { addVaultMiscItems, getGuildForRequestEx } from "@/src/services/guildService";
import { addFusionTreasures, addMiscItems, addShipDecorations, getInventory } from "@/src/services/inventoryService";
import { getAccountIdForRequest } from "@/src/services/loginService";
import { IFusionTreasure, IMiscItem, ITypeCount } from "@/src/types/inventoryTypes/inventoryTypes";
import { RequestHandler } from "express";

export const contributeToVaultController: RequestHandler = async (req, res) => {
    const accountId = await getAccountIdForRequest(req);
    const inventory = await getInventory(accountId);
    const guild = await getGuildForRequestEx(req, inventory);
    const guildMember = (await GuildMember.findOne(
        { accountId, guildId: guild._id },
        "RegularCreditsContributed MiscItemsContributed ShipDecorationsContributed"
    ))!;
    const request = JSON.parse(String(req.body)) as IContributeToVaultRequest;

    if (request.RegularCredits) {
        guild.VaultRegularCredits ??= 0;
        guild.VaultRegularCredits += request.RegularCredits;

        guildMember.RegularCreditsContributed ??= 0;
        guildMember.RegularCreditsContributed += request.RegularCredits;
    }
    if (request.MiscItems.length) {
        addVaultMiscItems(guild, request.MiscItems);

        guildMember.MiscItemsContributed ??= [];
        for (const item of request.MiscItems) {
            const miscItemContribution = guildMember.MiscItemsContributed.find(x => x.ItemType == item.ItemType);
            if (miscItemContribution) {
                miscItemContribution.ItemCount += item.ItemCount;
            } else {
                guildMember.MiscItemsContributed.push(item);
            }

            addMiscItems(inventory, [{ ...item, ItemCount: item.ItemCount * -1 }]);
        }
    }
    if (request.ShipDecorations.length) {
        guild.VaultShipDecorations ??= [];
        guildMember.ShipDecorationsContributed ??= [];
        for (const item of request.ShipDecorations) {
            guild.VaultShipDecorations.push(item);
            guildMember.ShipDecorationsContributed.push(item);
            addShipDecorations(inventory, [{ ...item, ItemCount: item.ItemCount * -1 }]);
        }
    }
    if (request.FusionTreasures.length) {
        guild.VaultFusionTreasures ??= [];
        for (const item of request.FusionTreasures) {
            guild.VaultFusionTreasures.push(item);
            addFusionTreasures(inventory, [{ ...item, ItemCount: item.ItemCount * -1 }]);
        }
    }

    await guild.save();
    await inventory.save();
    await guildMember.save();
    res.end();
};

interface IContributeToVaultRequest {
    RegularCredits: number;
    MiscItems: IMiscItem[];
    ShipDecorations: ITypeCount[];
    FusionTreasures: IFusionTreasure[];
}