From da272735cf0c6fd0cda1573de21ed0145f394536 Mon Sep 17 00:00:00 2001
From: Sainan <sainan@calamity.gg>
Date: Sat, 4 May 2024 20:37:58 +0200
Subject: [PATCH] fix: unequipping weapons doesn't save also improved the types
 and models

---
 src/models/inventoryModels/loadoutModel.ts |  3 ++-
 src/services/saveLoadoutService.ts         |  3 +--
 src/types/saveLoadoutTypes.ts              | 22 ++++++++++++----------
 3 files changed, 15 insertions(+), 13 deletions(-)

diff --git a/src/models/inventoryModels/loadoutModel.ts b/src/models/inventoryModels/loadoutModel.ts
index 87871138..2ff08774 100644
--- a/src/models/inventoryModels/loadoutModel.ts
+++ b/src/models/inventoryModels/loadoutModel.ts
@@ -19,7 +19,8 @@ const EquipmentSelectionSchema = new Schema<IEquipmentSelection>(
             default: { $oid: "000000000000000000000000" }
         },
         mod: Number,
-        cus: Number
+        cus: Number,
+        hide: Boolean
     },
     {
         _id: false
diff --git a/src/services/saveLoadoutService.ts b/src/services/saveLoadoutService.ts
index 0b71f22f..9319035a 100644
--- a/src/services/saveLoadoutService.ts
+++ b/src/services/saveLoadoutService.ts
@@ -106,8 +106,7 @@ export const handleInventoryItemConfigChange = async (
                             throw new Error("loadout index not found");
                         }
 
-                        //perhaps .overwrite() is better
-                        loadout[loadoutSlot][loadoutIndex].set(loadoutConfig);
+                        loadout[loadoutSlot][loadoutIndex].overwrite(loadoutConfig);
                     }
                 }
                 await loadout.save();
diff --git a/src/types/saveLoadoutTypes.ts b/src/types/saveLoadoutTypes.ts
index 61020029..8022abc5 100644
--- a/src/types/saveLoadoutTypes.ts
+++ b/src/types/saveLoadoutTypes.ts
@@ -76,17 +76,19 @@ export interface ILoadoutConfigDatabase extends Omit<ILoadoutConfigClient, "Item
 // for request and response from and to client
 export interface ILoadoutConfigClient {
     ItemId: IOid;
-    n: string;
-    PresetIcon: string;
-    Favorite: boolean;
-    s: IEquipmentSelection;
-    p: IEquipmentSelection;
-    l: IEquipmentSelection;
-    m: IEquipmentSelection;
+    n?: string;
+    PresetIcon?: string;
+    Favorite?: boolean;
+    s?: IEquipmentSelection;
+    p?: IEquipmentSelection;
+    l?: IEquipmentSelection;
+    m?: IEquipmentSelection;
+    Remove?: boolean;
 }
 
 export interface IEquipmentSelection {
-    ItemId: IOid;
-    mod: number;
-    cus: number;
+    ItemId?: IOid;
+    mod?: number;
+    cus?: number;
+    hide?: boolean;
 }