From ce2459113b3b52e490f4f7113e90f7c65b550ee8 Mon Sep 17 00:00:00 2001
From: Sainan <sainan@calamity.inc>
Date: Sun, 19 Jan 2025 17:21:02 +0100
Subject: [PATCH] import Upgrades

---
 src/services/importService.ts | 21 ++++++++++++++++++---
 1 file changed, 18 insertions(+), 3 deletions(-)

diff --git a/src/services/importService.ts b/src/services/importService.ts
index 2afa5313..87ba9231 100644
--- a/src/services/importService.ts
+++ b/src/services/importService.ts
@@ -1,11 +1,17 @@
-/* eslint-disable @typescript-eslint/no-unused-vars */
-
 import { Types } from "mongoose";
 import { IEquipmentClient, IEquipmentDatabase } from "../types/inventoryTypes/commonInventoryTypes";
 import { IMongoDate } from "../types/commonTypes";
-import { IInventoryClient, IWeaponSkinClient, IWeaponSkinDatabase } from "../types/inventoryTypes/inventoryTypes";
+import {
+    IInventoryClient,
+    IUpgradeClient,
+    IUpgradeDatabase,
+    IWeaponSkinClient,
+    IWeaponSkinDatabase
+} from "../types/inventoryTypes/inventoryTypes";
 import { TInventoryDatabaseDocument } from "../models/inventoryModels/inventoryModel";
 
+/* eslint-disable @typescript-eslint/no-unused-vars */
+
 const convertDate = (value: IMongoDate): Date => {
     return new Date(parseInt(value.$date.$numberLong));
 };
@@ -33,6 +39,14 @@ const convertWeaponSkin = (client: IWeaponSkinClient): IWeaponSkinDatabase => {
     };
 };
 
+const convertUpgrade = (client: IUpgradeClient): IUpgradeDatabase => {
+    const { ItemId, ...rest } = client;
+    return {
+        ...rest,
+        _id: new Types.ObjectId(client.ItemId.$oid)
+    };
+};
+
 const replaceArray = <T>(arr: T[], replacement: T[]): void => {
     arr.splice(0, arr.length);
     replacement.forEach(x => {
@@ -43,4 +57,5 @@ const replaceArray = <T>(arr: T[], replacement: T[]): void => {
 export const importInventory = (db: TInventoryDatabaseDocument, client: IInventoryClient): void => {
     replaceArray<IEquipmentDatabase>(db.Suits, client.Suits.map(convertEquipment));
     replaceArray<IWeaponSkinDatabase>(db.WeaponSkins, client.WeaponSkins.map(convertWeaponSkin));
+    replaceArray<IUpgradeDatabase>(db.Upgrades, client.Upgrades.map(convertUpgrade));
 };