From e1732202254d9da1a8b093ebbf490a24e39d9f0c Mon Sep 17 00:00:00 2001
From: Sainan <sainan@calamity.gg>
Date: Thu, 20 Jun 2024 04:13:59 +0200
Subject: [PATCH] improve: add EquipmentFeatures

---
 src/controllers/api/upgradesController.ts        | 8 ++++----
 src/types/inventoryTypes/commonInventoryTypes.ts | 6 ++++++
 2 files changed, 10 insertions(+), 4 deletions(-)

diff --git a/src/controllers/api/upgradesController.ts b/src/controllers/api/upgradesController.ts
index 25b548e6..da3f013c 100644
--- a/src/controllers/api/upgradesController.ts
+++ b/src/controllers/api/upgradesController.ts
@@ -1,6 +1,6 @@
 import { RequestHandler } from "express";
 import { IUpgradesRequest } from "@/src/types/requestTypes";
-import { FocusSchool, IEquipmentDatabase } from "@/src/types/inventoryTypes/commonInventoryTypes";
+import { FocusSchool, IEquipmentDatabase, EquipmentFeatures } from "@/src/types/inventoryTypes/commonInventoryTypes";
 import { IMiscItem, TEquipmentKey } from "@/src/types/inventoryTypes/inventoryTypes";
 import { getAccountIdForRequest } from "@/src/services/loginService";
 import { addMiscItems, getInventory, updateCurrency } from "@/src/services/inventoryService";
@@ -31,7 +31,7 @@ export const upgradesController: RequestHandler = async (req, res) => {
                 for (const item of inventory[payload.ItemCategory as TEquipmentKey] as IEquipmentDatabase[]) {
                     if (item._id.toString() == payload.ItemId.$oid) {
                         item.Features ??= 0;
-                        item.Features |= 1;
+                        item.Features |= EquipmentFeatures.DOUBLE_CAPACITY;
                         break;
                     }
                 }
@@ -41,7 +41,7 @@ export const upgradesController: RequestHandler = async (req, res) => {
                 for (const item of inventory[payload.ItemCategory as TEquipmentKey] as IEquipmentDatabase[]) {
                     if (item._id.toString() == payload.ItemId.$oid) {
                         item.Features ??= 0;
-                        item.Features |= 2;
+                        item.Features |= EquipmentFeatures.UTILITY_SLOT;
                         break;
                     }
                 }
@@ -52,7 +52,7 @@ export const upgradesController: RequestHandler = async (req, res) => {
                 for (const item of inventory[payload.ItemCategory as TEquipmentKey] as IEquipmentDatabase[]) {
                     if (item._id.toString() == payload.ItemId.$oid) {
                         item.Features ??= 0;
-                        item.Features |= 32;
+                        item.Features |= EquipmentFeatures.ARCANE_SLOT;
                         break;
                     }
                 }
diff --git a/src/types/inventoryTypes/commonInventoryTypes.ts b/src/types/inventoryTypes/commonInventoryTypes.ts
index aadc1b62..9c7fbca8 100644
--- a/src/types/inventoryTypes/commonInventoryTypes.ts
+++ b/src/types/inventoryTypes/commonInventoryTypes.ts
@@ -82,6 +82,12 @@ export interface IEquipmentClient extends Omit<IEquipmentDatabase, "_id"> {
     ItemId: IOid;
 }
 
+export enum EquipmentFeatures {
+    DOUBLE_CAPACITY = 1,
+    UTILITY_SLOT = 2,
+    ARCANE_SLOT = 32,
+}
+
 export interface IEquipmentDatabase {
     ItemType: string;
     ItemName?: string;