From 22be3cc70725b8458c49f0e0ce442d399fade3a3 Mon Sep 17 00:00:00 2001
From: Sainan <sainan@calamity.inc>
Date: Sun, 16 Mar 2025 18:22:10 +0100
Subject: [PATCH] let client know that it also acquired RawUpgrades for
 defaultUpgrades

---
 src/controllers/api/modularWeaponCraftingController.ts | 8 +++++---
 src/controllers/api/modularWeaponSaleController.ts     | 6 +++++-
 2 files changed, 10 insertions(+), 4 deletions(-)

diff --git a/src/controllers/api/modularWeaponCraftingController.ts b/src/controllers/api/modularWeaponCraftingController.ts
index c53de2af..2b116a8d 100644
--- a/src/controllers/api/modularWeaponCraftingController.ts
+++ b/src/controllers/api/modularWeaponCraftingController.ts
@@ -28,13 +28,15 @@ export const modularWeaponCraftingController: RequestHandler = async (req, res)
     const category = modularWeaponTypes[data.WeaponType];
     const inventory = await getInventory(accountId);
 
-    const configs = applyDefaultUpgrades(inventory, getDefaultUpgrades(data.Parts));
-
-    // Give weapon
+    const defaultUpgrades = getDefaultUpgrades(data.Parts);
+    const configs = applyDefaultUpgrades(inventory, defaultUpgrades);
     const inventoryChanges: IInventoryChanges = {
         ...addEquipment(inventory, category, data.WeaponType, data.Parts, {}, { Configs: configs }),
         ...occupySlot(inventory, productCategoryToInventoryBin(category)!, false)
     };
+    if (defaultUpgrades) {
+        inventoryChanges.RawUpgrades = defaultUpgrades.map(x => ({ ItemType: x.ItemType, ItemCount: 1 }));
+    }
 
     // Remove credits & parts
     const miscItemChanges = [];
diff --git a/src/controllers/api/modularWeaponSaleController.ts b/src/controllers/api/modularWeaponSaleController.ts
index 41297a25..46c4bec5 100644
--- a/src/controllers/api/modularWeaponSaleController.ts
+++ b/src/controllers/api/modularWeaponSaleController.ts
@@ -39,7 +39,8 @@ export const modularWeaponSaleController: RequestHandler = async (req, res) => {
         const weaponInfo = getSaleInfos(partTypeToParts, payload.Revision).find(x => x.Name == payload.SaleName)!
             .Weapons[payload.ItemIndex];
         const category = modularWeaponTypes[weaponInfo.ItemType];
-        const configs = applyDefaultUpgrades(inventory, getDefaultUpgrades(weaponInfo.ModularParts));
+        const defaultUpgrades = getDefaultUpgrades(weaponInfo.ModularParts);
+        const configs = applyDefaultUpgrades(inventory, defaultUpgrades);
         const inventoryChanges: IInventoryChanges = {
             ...addEquipment(
                 inventory,
@@ -62,6 +63,9 @@ export const modularWeaponSaleController: RequestHandler = async (req, res) => {
             ...occupySlot(inventory, productCategoryToInventoryBin(category)!, true),
             ...updateCurrency(inventory, weaponInfo.PremiumPrice, true)
         };
+        if (defaultUpgrades) {
+            inventoryChanges.RawUpgrades = defaultUpgrades.map(x => ({ ItemType: x.ItemType, ItemCount: 1 }));
+        }
         await inventory.save();
         res.json({
             InventoryChanges: inventoryChanges