From 15ab3f76b1cafcc71cbbd0b499149037734e4422 Mon Sep 17 00:00:00 2001
From: holmityd <nbayramov100@gmail.com>
Date: Wed, 30 Aug 2023 07:51:56 +0400
Subject: [PATCH] mission inventory update - exp

---
 .../api/missionInventoryUpdateController.ts   | 15 +++--
 src/services/inventoryService.ts              | 63 ++++++++++---------
 2 files changed, 42 insertions(+), 36 deletions(-)

diff --git a/src/controllers/api/missionInventoryUpdateController.ts b/src/controllers/api/missionInventoryUpdateController.ts
index 8a255514..52e68106 100644
--- a/src/controllers/api/missionInventoryUpdateController.ts
+++ b/src/controllers/api/missionInventoryUpdateController.ts
@@ -1,14 +1,14 @@
 import { RequestHandler } from "express";
-import { addRawUpgrades, addMiscItems } from "@/src/services/inventoryService";
+import { missionInventoryUpdate } from "@/src/services/inventoryService";
 import fs from 'fs';
 /*
 - [ ]  crossPlaySetting
 - [ ]  rewardsMultiplier
 - [ ]  ActiveBoosters
-- [ ]  LongGuns
-- [ ]  Pistols
-- [ ]  Suits
-- [ ]  Melee
+- [x]  LongGuns
+- [x]  Pistols
+- [x]  Suits
+- [x]  Melee
 - [x]  RawUpgrades
 - [x]  MiscItems
 - [ ]  RegularCredits
@@ -47,10 +47,9 @@ const missionInventoryUpdateController: RequestHandler = async (req, res) => {
     });
     
     const [data, _secondIGuessIsSalt] = String(req.body).split("\n");
-    const {RawUpgrades, MiscItems} = JSON.parse(data);
     const id = req.query.accountId as string;
-    addRawUpgrades(RawUpgrades, id);
-    addMiscItems(MiscItems, id);
+    
+    await missionInventoryUpdate(JSON.parse(data), id);
 
     res.json({});
 };
diff --git a/src/services/inventoryService.ts b/src/services/inventoryService.ts
index 2e9e0b96..c9f2520b 100644
--- a/src/services/inventoryService.ts
+++ b/src/services/inventoryService.ts
@@ -2,10 +2,11 @@ import { Inventory } from "@/src/models/inventoryModel";
 import new_inventory from "@/static/fixed_responses/postTutorialInventory.json";
 import config from "@/config.json";
 import { Types } from "mongoose";
-import { ISuitResponse } from "@/src/types/inventoryTypes/SuitTypes";
+import { ISuitDatabase, ISuitResponse } from "@/src/types/inventoryTypes/SuitTypes";
 import { SlotType } from "@/src/types/purchaseTypes";
-import { IWeaponResponse } from "@/src/types/inventoryTypes/weaponTypes";
-import { FlavourItem, RawUpgrade, MiscItem } from "@/src/types/inventoryTypes/inventoryTypes";
+import { IWeaponDatabase, IWeaponResponse } from "@/src/types/inventoryTypes/weaponTypes";
+import { FlavourItem, RawUpgrade, MiscItem, IInventoryDatabase } from "@/src/types/inventoryTypes/inventoryTypes";
+import { items } from "@/static/data/items";
 
 const createInventory = async (accountOwnerId: Types.ObjectId) => {
     try {
@@ -106,35 +107,41 @@ export const addCustomization = async (customizatonName: string, accountId: stri
     return changedInventory.FlavourItems[flavourItemIndex].toJSON(); //mongoose bug forces as FlavourItem
 };
 
-export const addRawUpgrades = async (rawUpgrades: RawUpgrade[], accountId: string): Promise<void> => {
+export const missionInventoryUpdate = async (data: any, accountId: string): Promise<void> => {
+    const { RawUpgrades, MiscItems, Suits, Pistols, LongGuns, Melee } = data;
     const inventory = await getInventory(accountId);
-    rawUpgrades?.forEach(item => {
-        const { ItemCount, ItemType } = item;
-        const existingItemIndex = inventory.RawUpgrades.findIndex(i => i.ItemType === ItemType);
 
-        if (existingItemIndex !== -1) {
-            inventory.RawUpgrades[existingItemIndex].ItemCount += ItemCount;
-            inventory.markModified('RawUpgrades.' + existingItemIndex + '.ItemCount');
-        } else {
-            inventory.RawUpgrades.push({ ItemCount, ItemType });
-        }
-    });
-    await inventory.save();
-};
+    const addGearExpByCategory = (gearArray: (ISuitDatabase|IWeaponDatabase)[], category: 'Pistols'|'LongGuns'|'Melee'|'Suits') => {
+        gearArray.forEach(({ ItemId, XP }: any) => {
+            const itemIndex = inventory[category].findIndex(i => i._id?.equals(ItemId.$oid));
+            if (itemIndex !== -1) {
+                inventory[category][itemIndex].XP += XP;
+                inventory.markModified(`${category}.${itemIndex}.XP`);
+            }
+        });
+    };
 
-export const addMiscItems = async (miscItems: MiscItem[], accountId: string): Promise<void> => {
-    const inventory = await getInventory(accountId);
-    miscItems?.forEach(item => {
-        const { ItemCount, ItemType } = item;
-        const existingItemIndex = inventory.MiscItems.findIndex(i => i.ItemType === ItemType);
+    const addItemsByCategory = (itemsArray: (RawUpgrade | MiscItem)[], category: 'RawUpgrades'|'MiscItems') => {
+        itemsArray?.forEach(item => {
+            const { ItemCount, ItemType } = item;
+            const existingItemIndex = inventory[category].findIndex(i => i.ItemType === ItemType);
+
+            if (existingItemIndex !== -1) {
+                inventory[category][existingItemIndex].ItemCount += ItemCount;
+                inventory.markModified(category + '.' + existingItemIndex + '.ItemCount');
+            } else {
+                inventory[category].push({ ItemCount, ItemType });
+            }
+        });
+    };
+    
+    addGearExpByCategory(Pistols, 'Pistols');
+    addGearExpByCategory(LongGuns, 'LongGuns');
+    addGearExpByCategory(Melee, 'Melee');
+    addGearExpByCategory(Suits, 'Suits');
+    addItemsByCategory(RawUpgrades, 'RawUpgrades'); // TODO - check mods fusion level
+    addItemsByCategory(MiscItems, 'MiscItems');
 
-        if (existingItemIndex !== -1) {
-            inventory.MiscItems[existingItemIndex].ItemCount += ItemCount;
-            inventory.markModified('MiscItems.' + existingItemIndex + '.ItemCount');
-        } else {
-            inventory.MiscItems.push({ ItemCount, ItemType });
-        }
-    });
     await inventory.save();
 };