consume potency mods when they add to antivirus gain

src/controllers/api/nemesisController.ts

@@ -1,5 +1,12 @@
-import { encodeNemesisGuess, getInfNodes, getNemesisPasscode } from "@/src/helpers/nemesisHelpers";
+import {
+    consumeModCharge,
+    encodeNemesisGuess,
+    getInfNodes,
+    getNemesisPasscode,
+    IKnifeResponse
+} from "@/src/helpers/nemesisHelpers";
 import { getJSONfromString } from "@/src/helpers/stringHelpers";
+import { Loadout } from "@/src/models/inventoryModels/loadoutModel";
 import { freeUpSlot, getInventory } from "@/src/services/inventoryService";
 import { getAccountIdForRequest } from "@/src/services/loginService";
 import { SRng } from "@/src/services/rngService";
@@ -10,6 +17,7 @@ import {
     InventorySlot,
     IUpgradeClient,
     IWeaponSkinClient,
+    LoadoutIndex,
     TEquipmentKey
 } from "@/src/types/inventoryTypes/inventoryTypes";
 import { logger } from "@/src/utils/logger";
@@ -74,7 +82,10 @@ export const nemesisController: RequestHandler = async (req, res) => {
         }
         res.json({ GuessResult: guessResult });
     } else if (req.query.mode == "r") {
-        const inventory = await getInventory(accountId, "Nemesis");
+        const inventory = await getInventory(
+            accountId,
+            "Nemesis LoadOutPresets CurrentLoadOutIds DataKnives Upgrades RawUpgrades"
+        );
         const body = getJSONfromString<INemesisRequiemRequest>(String(req.body));
         if (inventory.Nemesis!.Faction == "FC_INFESTATION") {
             const guess: number[] = [body.guess & 0xf, (body.guess >> 4) & 0xf, (body.guess >> 8) & 0xf];
@@ -90,22 +101,32 @@ export const nemesisController: RequestHandler = async (req, res) => {
 
             // Increase antivirus
             let antivirusGain = 5;
+            const loadout = (await Loadout.findById(inventory.LoadOutPresets, "DATAKNIFE"))!;
+            const dataknifeLoadout = loadout.DATAKNIFE.id(inventory.CurrentLoadOutIds[LoadoutIndex.DATAKNIFE].$oid);
+            const dataknifeConfigIndex = dataknifeLoadout?.s?.mod ?? 0;
+            const dataknifeUpgrades = inventory.DataKnives[0].Configs[dataknifeConfigIndex].Upgrades!;
+            const response: IKnifeResponse = {};
             for (const upgrade of body.knife!.AttachedUpgrades) {
                 switch (upgrade.ItemType) {
                     case "/Lotus/Upgrades/Mods/DataSpike/Potency/GainAntivirusAndSpeedOnUseMod":
                         antivirusGain += 10;
+                        consumeModCharge(response, inventory, upgrade, dataknifeUpgrades);
                         break;
                     case "/Lotus/Upgrades/Mods/DataSpike/Potency/GainAntivirusAndWeaponDamageOnUseMod":
                         antivirusGain += 10;
+                        consumeModCharge(response, inventory, upgrade, dataknifeUpgrades);
                         break;
                     case "/Lotus/Upgrades/Mods/DataSpike/Potency/GainAntivirusLargeOnSingleUseMod": // Instant Secure
                         antivirusGain += 15;
+                        consumeModCharge(response, inventory, upgrade, dataknifeUpgrades);
                         break;
                     case "/Lotus/Upgrades/Mods/DataSpike/Potency/GainAntivirusOnUseMod": // Immuno Shield
                         antivirusGain += 15;
+                        consumeModCharge(response, inventory, upgrade, dataknifeUpgrades);
                         break;
                     case "/Lotus/Upgrades/Mods/DataSpike/Potency/GainAntivirusSmallOnSingleUseMod":
                         antivirusGain += 10;
+                        consumeModCharge(response, inventory, upgrade, dataknifeUpgrades);
                         break;
                 }
             }
@@ -122,10 +143,8 @@ export const nemesisController: RequestHandler = async (req, res) => {
                 inventory.Nemesis!.InfNodes = getInfNodes("FC_INFESTATION", 0);
             }
 
-            // TODO: Consume mod charges?
-
             await inventory.save();
-            res.end();
+            res.json(response);
         } else {
             const passcode = getNemesisPasscode(inventory.Nemesis!);
             if (passcode[body.position] != body.guess) {

src/helpers/nemesisHelpers.ts

@@ -1,6 +1,11 @@
 import { ExportRegions } from "warframe-public-export-plus";
 import { IInfNode } from "@/src/types/inventoryTypes/inventoryTypes";
 import { SRng } from "@/src/services/rngService";
+import { TInventoryDatabaseDocument } from "../models/inventoryModels/inventoryModel";
+import { logger } from "../utils/logger";
+import { IOid } from "../types/commonTypes";
+import { Types } from "mongoose";
+import { addMods } from "../services/inventoryService";
 
 export const getInfNodes = (faction: string, rank: number): IInfNode[] => {
     const infNodes = [];
@@ -64,3 +69,62 @@ export const encodeNemesisGuess = (
 export const decodeNemesisGuess = (val: number): number[] => {
     return [val & 0xf, (val >> 12) & 3, (val & 0xff) >> 4, (val & 0xffff) >> 14, (val >> 8) & 0xf, (val >> 16) & 3];
 };
+
+export interface IKnifeResponse {
+    UpgradeIds?: string[];
+    UpgradeTypes?: string[];
+    UpgradeFingerprints?: { lvl: number }[];
+    UpgradeNew?: boolean[];
+    HasKnife?: boolean;
+}
+
+export const consumeModCharge = (
+    response: IKnifeResponse,
+    inventory: TInventoryDatabaseDocument,
+    upgrade: { ItemId: IOid; ItemType: string },
+    dataknifeUpgrades: string[]
+): void => {
+    response.UpgradeIds ??= [];
+    response.UpgradeTypes ??= [];
+    response.UpgradeFingerprints ??= [];
+    response.UpgradeNew ??= [];
+    response.HasKnife ??= true;
+
+    if (upgrade.ItemId.$oid != "000000000000000000000000") {
+        const dbUpgrade = inventory.Upgrades.id(upgrade.ItemId.$oid)!;
+        const fingerprint = JSON.parse(dbUpgrade.UpgradeFingerprint!) as { lvl: number };
+        fingerprint.lvl += 1;
+        dbUpgrade.UpgradeFingerprint = JSON.stringify(fingerprint);
+
+        response.UpgradeIds.push(upgrade.ItemId.$oid);
+        response.UpgradeTypes.push(upgrade.ItemType);
+        response.UpgradeFingerprints.push(fingerprint);
+        response.UpgradeNew.push(false);
+    } else {
+        const id = new Types.ObjectId();
+        inventory.Upgrades.push({
+            _id: id,
+            ItemType: upgrade.ItemType,
+            UpgradeFingerprint: `{"lvl":1}`
+        });
+
+        addMods(inventory, [
+            {
+                ItemType: upgrade.ItemType,
+                ItemCount: -1
+            }
+        ]);
+
+        const dataknifeRawUpgradeIndex = dataknifeUpgrades.indexOf(upgrade.ItemType);
+        if (dataknifeRawUpgradeIndex != -1) {
+            dataknifeUpgrades[dataknifeRawUpgradeIndex] = id.toString();
+        } else {
+            logger.warn(`${upgrade.ItemType} not found in dataknife config`);
+        }
+
+        response.UpgradeIds.push(id.toString());
+        response.UpgradeTypes.push(upgrade.ItemType);
+        response.UpgradeFingerprints.push({ lvl: 1 });
+        response.UpgradeNew.push(true);
+    }
+};

src/types/inventoryTypes/inventoryTypes.ts

@@ -157,6 +157,11 @@ export type TSolarMapRegion =
 
 //TODO: perhaps split response and database into their own files
 
+export enum LoadoutIndex {
+    NORMAL = 0,
+    DATAKNIFE = 7
+}
+
 export interface IDailyAffiliations {
     DailyAffiliation: number;
     DailyAffiliationPvp: number;
@@ -220,7 +225,7 @@ export interface IInventoryClient extends IDailyAffiliations, InventoryClientEqu
     ActiveQuest: string;
     FlavourItems: IFlavourItem[];
     LoadOutPresets: ILoadOutPresets;
-    CurrentLoadOutIds: IOid[]; //TODO: we store it in the database using this representation as well :/
+    CurrentLoadOutIds: IOid[]; // we store it in the database using this representation as well :/
     Missions: IMission[];
     RandomUpgradesIdentified?: number;
     LastRegionPlayed: TSolarMapRegion;