From 7af5cd98116490910d6d7119ac8078c11fd5dcad Mon Sep 17 00:00:00 2001
From: Sainan <sainan@calamity.inc>
Date: Sun, 19 Jan 2025 01:58:09 +0100
Subject: [PATCH] fix: add slots when adding items via WebUI (#812)

---
 src/controllers/custom/addItemsController.ts | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/controllers/custom/addItemsController.ts b/src/controllers/custom/addItemsController.ts
index bc033174b..cce65a3cf 100644
--- a/src/controllers/custom/addItemsController.ts
+++ b/src/controllers/custom/addItemsController.ts
@@ -1,7 +1,8 @@
 import { getAccountIdForRequest } from "@/src/services/loginService";
 import { getWeaponType } from "@/src/services/itemDataService";
-import { addPowerSuit, addEquipment, getInventory } from "@/src/services/inventoryService";
+import { addPowerSuit, addEquipment, getInventory, updateSlots } from "@/src/services/inventoryService";
 import { RequestHandler } from "express";
+import { InventorySlot } from "@/src/types/inventoryTypes/inventoryTypes";
 
 export const addItemsController: RequestHandler = async (req, res) => {
     const accountId = await getAccountIdForRequest(req);
@@ -10,10 +11,12 @@ export const addItemsController: RequestHandler = async (req, res) => {
     for (const request of requests) {
         switch (request.type) {
             case ItemType.Powersuit:
+                updateSlots(inventory, InventorySlot.SUITS, 0, 1);
                 addPowerSuit(inventory, request.internalName);
                 break;
 
             case ItemType.Weapon:
+                updateSlots(inventory, InventorySlot.WEAPONS, 0, 1);
                 addEquipment(inventory, getWeaponType(request.internalName), request.internalName);
                 break;
         }