From 6a69961c2da5d8b26ba0ce2170dfc77da927c9a0 Mon Sep 17 00:00:00 2001
From: VampireKitten <dynamightkobold@gmail.com>
Date: Tue, 15 Oct 2024 21:13:03 +0200
Subject: [PATCH] Added FocusType enum

---
 src/services/inventoryService.ts | 25 +++++++++++++++++++------
 1 file changed, 19 insertions(+), 6 deletions(-)

diff --git a/src/services/inventoryService.ts b/src/services/inventoryService.ts
index 4bc7d73d..85c3cbe1 100644
--- a/src/services/inventoryService.ts
+++ b/src/services/inventoryService.ts
@@ -722,13 +722,26 @@ const addMissionComplete = (inventory: IInventoryDatabaseDocument, { Tag, Comple
 };
 
 const addFocusXpIncreases = (inventory: IInventoryDatabaseDocument, focusXpPlus: number[] | undefined) => {
+    enum FocusType {
+        AP_UNIVERSAL,
+        AP_ATTACK,
+        AP_DEFENSE,
+        AP_TACTIC,
+        AP_POWER,
+        AP_PRECEPT,
+        AP_FUSION,
+        AP_WARD,
+        AP_UMBRA,
+        AP_ANY
+    };
+
     if (focusXpPlus) {
-        inventory.FocusXP ??= { AP_POWER: 0, AP_TACTIC: 0, AP_DEFENSE: 0, AP_ATTACK: 0, AP_WARD: 0 };
-        inventory.FocusXP.AP_ATTACK += focusXpPlus[1];
-        inventory.FocusXP.AP_DEFENSE += focusXpPlus[2];
-        inventory.FocusXP.AP_TACTIC += focusXpPlus[3];
-        inventory.FocusXP.AP_POWER += focusXpPlus[4];
-        inventory.FocusXP.AP_WARD += focusXpPlus[7];
+        inventory.FocusXP ??= { AP_ATTACK: 0, AP_DEFENSE: 0, AP_TACTIC: 0, AP_POWER: 0, AP_WARD: 0 };
+        inventory.FocusXP.AP_ATTACK += focusXpPlus[FocusType.AP_ATTACK];
+        inventory.FocusXP.AP_DEFENSE += focusXpPlus[FocusType.AP_DEFENSE];
+        inventory.FocusXP.AP_TACTIC += focusXpPlus[FocusType.AP_TACTIC];
+        inventory.FocusXP.AP_POWER += focusXpPlus[FocusType.AP_POWER];
+        inventory.FocusXP.AP_WARD += focusXpPlus[FocusType.AP_WARD];
     }
 };