From 53d5e7c3f0ea0c1228ea95f1bbe47306bc05bf41 Mon Sep 17 00:00:00 2001
From: Sainan <sainan@calamity.inc>
Date: Sat, 11 Jan 2025 23:08:17 +0100
Subject: [PATCH] fix: make crew member slots optional (#766)

---
 src/models/inventoryModels/inventoryModel.ts | 26 ++++++++++----------
 src/types/inventoryTypes/inventoryTypes.ts   | 16 ++++++++----
 2 files changed, 24 insertions(+), 18 deletions(-)

diff --git a/src/models/inventoryModels/inventoryModel.ts b/src/models/inventoryModels/inventoryModel.ts
index 964c89eb..65a5d3d2 100644
--- a/src/models/inventoryModels/inventoryModel.ts
+++ b/src/models/inventoryModels/inventoryModel.ts
@@ -42,11 +42,12 @@ import {
     ICrewShipPortGuns,
     ICrewShipCustomization,
     ICrewShipWeapon,
-    ICrewShipMembers,
+    ICrewShipMembersClient,
     ICrewShip,
     ICrewShipPilotWeapon,
     IShipExterior,
-    IHelminthFoodRecord
+    IHelminthFoodRecord,
+    ICrewShipMembersDatabase
 } from "../../types/inventoryTypes/inventoryTypes";
 import { IOid } from "../../types/commonTypes";
 import {
@@ -670,23 +671,22 @@ const crewShipCustomizationSchema = new Schema<ICrewShipCustomization>(
     { _id: false }
 );
 
-const crewShipMembersSchema = new Schema<ICrewShipMembers>(
+const crewShipMembersSchema = new Schema<ICrewShipMembersDatabase>(
     {
-        SLOT_A: Schema.Types.ObjectId,
-        SLOT_B: Schema.Types.ObjectId,
-        SLOT_C: Schema.Types.ObjectId
+        SLOT_A: { type: Schema.Types.ObjectId, required: false },
+        SLOT_B: { type: Schema.Types.ObjectId, required: false },
+        SLOT_C: { type: Schema.Types.ObjectId, required: false }
     },
     { _id: false }
 );
 crewShipMembersSchema.set("toJSON", {
     virtuals: true,
-    transform(_doc, ret) {
-        // eslint-disable-next-line @typescript-eslint/no-unsafe-argument
-        ret.SLOT_A = { ItemId: toOid(ret.SLOT_A) };
-        // eslint-disable-next-line @typescript-eslint/no-unsafe-argument
-        ret.SLOT_B = { ItemId: toOid(ret.SLOT_B) };
-        // eslint-disable-next-line @typescript-eslint/no-unsafe-argument
-        ret.SLOT_C = { ItemId: toOid(ret.SLOT_C) };
+    transform(_doc, obj) {
+        const db = obj as ICrewShipMembersDatabase;
+        const client = obj as ICrewShipMembersClient;
+        client.SLOT_A = db.SLOT_A ? { ItemId: toOid(db.SLOT_A) } : undefined;
+        client.SLOT_B = db.SLOT_B ? { ItemId: toOid(db.SLOT_B) } : undefined;
+        client.SLOT_C = db.SLOT_C ? { ItemId: toOid(db.SLOT_C) } : undefined;
     }
 });
 
diff --git a/src/types/inventoryTypes/inventoryTypes.ts b/src/types/inventoryTypes/inventoryTypes.ts
index dc9e6ef3..6463a156 100644
--- a/src/types/inventoryTypes/inventoryTypes.ts
+++ b/src/types/inventoryTypes/inventoryTypes.ts
@@ -426,21 +426,27 @@ export interface ICrewShip {
     Customization?: ICrewShipCustomization;
     ItemName: string;
     RailjackImage?: IFlavourItem;
-    CrewMembers?: ICrewShipMembers;
+    CrewMembers?: ICrewShipMembersClient;
     ItemId: IOid;
     _id: Types.ObjectId;
 }
 
-export interface ICrewShipMembers {
-    SLOT_A: ISlot;
-    SLOT_B: ISlot;
-    SLOT_C: ISlot;
+export interface ICrewShipMembersClient {
+    SLOT_A?: ISlot;
+    SLOT_B?: ISlot;
+    SLOT_C?: ISlot;
 }
 
 export interface ISlot {
     ItemId: IOid;
 }
 
+export interface ICrewShipMembersDatabase {
+    SLOT_A?: Types.ObjectId;
+    SLOT_B?: Types.ObjectId;
+    SLOT_C?: Types.ObjectId;
+}
+
 export interface ICrewShipCustomization {
     CrewshipInterior: IShipExterior;
 }