Server/project/src/controllers/BuildController.ts

import { inject, injectable } from "tsyringe";

import { ItemHelper } from "@spt-aki/helpers/ItemHelper";
import { ProfileHelper } from "@spt-aki/helpers/ProfileHelper";
import { ISetMagazineRequest } from "@spt-aki/models/eft/builds/ISetMagazineRequest";
import { IPresetBuildActionRequestData } from "@spt-aki/models/eft/presetBuild/IPresetBuildActionRequestData";
import { IRemoveBuildRequestData } from "@spt-aki/models/eft/presetBuild/IRemoveBuildRequestData";
import { IEquipmentBuild, IMagazineBuild, IUserBuilds, IWeaponBuild } from "@spt-aki/models/eft/profile/IAkiProfile";
import { EquipmentBuildType } from "@spt-aki/models/enums/EquipmentBuildType";
import { ILogger } from "@spt-aki/models/spt/utils/ILogger";
import { EventOutputHolder } from "@spt-aki/routers/EventOutputHolder";
import { DatabaseServer } from "@spt-aki/servers/DatabaseServer";
import { SaveServer } from "@spt-aki/servers/SaveServer";
import { HashUtil } from "@spt-aki/utils/HashUtil";
import { JsonUtil } from "@spt-aki/utils/JsonUtil";

@injectable()
export class BuildController
{
    constructor(
        @inject("WinstonLogger") protected logger: ILogger,
        @inject("HashUtil") protected hashUtil: HashUtil,
        @inject("EventOutputHolder") protected eventOutputHolder: EventOutputHolder,
        @inject("JsonUtil") protected jsonUtil: JsonUtil,
        @inject("DatabaseServer") protected databaseServer: DatabaseServer,
        @inject("ProfileHelper") protected profileHelper: ProfileHelper,
        @inject("ItemHelper") protected itemHelper: ItemHelper,
        @inject("SaveServer") protected saveServer: SaveServer,
    )
    {}

    /** Handle client/handbook/builds/my/list */
    public getUserBuilds(sessionID: string): IUserBuilds
    {
        const secureContainerSlotId = "SecuredContainer";
        const profile = this.saveServer.getProfile(sessionID);
        if (!profile.userbuilds)
        {
            profile.userbuilds = { equipmentBuilds: [], weaponBuilds: [], magazineBuilds: [] };
        }

        // Ensure the secure container in the default presets match what the player has equipped
        const defaultEquipmentPresets = this.jsonUtil.clone(
            this.databaseServer.getTables().templates.defaultEquipmentPresets,
        );
        const playerSecureContainer = profile.characters.pmc.Inventory.items?.find((x) =>
            x.slotId === secureContainerSlotId
        );
        const firstDefaultItemsSecureContainer = defaultEquipmentPresets[0]?.Items?.find((x) =>
            x.slotId === secureContainerSlotId
        );
        if (playerSecureContainer && playerSecureContainer?._tpl !== firstDefaultItemsSecureContainer?._tpl)
        {
            // Default equipment presets' secure container tpl doesn't match players secure container tpl
            for (const defaultPreset of defaultEquipmentPresets)
            {
                // Find presets secure container
                const secureContainer = defaultPreset.Items.find((item) => item.slotId === secureContainerSlotId);
                if (secureContainer)
                {
                    secureContainer._tpl = playerSecureContainer._tpl;
                }
            }
        }

        // Clone player build data from profile and append the above defaults onto end
        const result = this.jsonUtil.clone(profile.userbuilds);
        result.equipmentBuilds.push(...defaultEquipmentPresets);

        return result;
    }

    /** Handle client/builds/weapon/save */
    public saveWeaponBuild(sessionId: string, body: IPresetBuildActionRequestData): void
    {
        const pmcData = this.profileHelper.getPmcProfile(sessionId);

        // Replace duplicate Id's. The first item is the base item.
        // The root ID and the base item ID need to match.
        body.Items = this.itemHelper.replaceIDs(pmcData, body.Items);
        body.Root = body.Items[0]._id;

        // Create new object ready to save into profile userbuilds.weaponBuilds
        const newBuild: IWeaponBuild = { Id: body.Id, Name: body.Name, Root: body.Root, Items: body.Items };

        const savedWeaponBuilds = this.saveServer.getProfile(sessionId).userbuilds.weaponBuilds;
        const existingBuild = savedWeaponBuilds.find((x) => x.Id === body.Id);
        if (existingBuild)
        {
            // exists, replace
            this.saveServer.getProfile(sessionId).userbuilds.weaponBuilds.splice(
                savedWeaponBuilds.indexOf(existingBuild),
                1,
                newBuild,
            );
        }
        else
        {
            // Add fresh
            this.saveServer.getProfile(sessionId).userbuilds.weaponBuilds.push(newBuild);
        }
    }

    /** Handle client/builds/equipment/save event */
    public saveEquipmentBuild(sessionID: string, request: IPresetBuildActionRequestData): void
    {
        const buildType = "equipmentBuilds";
        const pmcData = this.profileHelper.getPmcProfile(sessionID);

        const existingSavedEquipmentBuilds: IEquipmentBuild[] =
            this.saveServer.getProfile(sessionID).userbuilds[buildType];

        // Replace duplicate ID's. The first item is the base item.
        // Root ID and the base item ID need to match.
        request.Items = this.itemHelper.replaceIDs(pmcData, request.Items);

        const newBuild: IEquipmentBuild = {
            Id: request.Id,
            Name: request.Name,
            BuildType: EquipmentBuildType.CUSTOM,
            Root: request.Items[0]._id,
            Items: request.Items,
        };

        const existingBuild = existingSavedEquipmentBuilds.find((build) =>
            build.Name === request.Name || build.Id === request.Id
        );
        if (existingBuild)
        {
            // Already exists, replace
            this.saveServer.getProfile(sessionID).userbuilds[buildType].splice(
                existingSavedEquipmentBuilds.indexOf(existingBuild),
                1,
                newBuild,
            );
        }
        else
        {
            // Fresh, add new
            this.saveServer.getProfile(sessionID).userbuilds[buildType].push(newBuild);
        }
    }

    /** Handle client/builds/delete*/
    public removeBuild(sessionID: string, request: IRemoveBuildRequestData): void
    {
        this.removePlayerBuild(request.id, sessionID);
    }

    protected removePlayerBuild(id: string, sessionID: string): void
    {
        const profile = this.saveServer.getProfile(sessionID);
        const weaponBuilds = profile.userbuilds.weaponBuilds;
        const equipmentBuilds = profile.userbuilds.equipmentBuilds;
        const magazineBuilds = profile.userbuilds.magazineBuilds;

        // Check for id in weapon array first
        const matchingWeaponBuild = weaponBuilds.find((x) => x.Id === id);
        if (matchingWeaponBuild)
        {
            weaponBuilds.splice(weaponBuilds.indexOf(matchingWeaponBuild), 1);

            return;
        }

        // Id not found in weapons, try equipment
        const matchingEquipmentBuild = equipmentBuilds.find((x) => x.Id === id);
        if (matchingEquipmentBuild)
        {
            equipmentBuilds.splice(equipmentBuilds.indexOf(matchingEquipmentBuild), 1);

            return;
        }

        // Id not found in weapons/equipment, try mags
        const matchingMagazineBuild = magazineBuilds.find((x) => x.Id === id);
        if (matchingMagazineBuild)
        {
            magazineBuilds.splice(magazineBuilds.indexOf(matchingMagazineBuild), 1);

            return;
        }

        // Not found in weapons,equipment or magazines, not good
        this.logger.error(`Unable to delete preset, cannot find ${id} in weapon, equipment or magazine presets`);
    }

    /**
     * Handle client/builds/magazine/save
     */
    public createMagazineTemplate(sessionId: string, request: ISetMagazineRequest): void
    {
        const result: IMagazineBuild = {
            Id: request.Id,
            Name: request.Name,
            Caliber: request.Caliber,
            TopCount: request.TopCount,
            BottomCount: request.BottomCount,
            Items: request.Items,
        };

        const profile = this.profileHelper.getFullProfile(sessionId);

        if (!profile.userbuilds.magazineBuilds)
        {
            profile.userbuilds.magazineBuilds = [];
        }

        const existingArrayId = profile.userbuilds.magazineBuilds.findIndex((item) => item.Name === request.Name);

        if (existingArrayId === -1)
        {
            profile.userbuilds.magazineBuilds.push(result);
        }
        else
        {
            profile.userbuilds.magazineBuilds.splice(existingArrayId, 1, result);
        }
    }
}