From a89861128cbebbd8b218226e2224bcb68cc742bb Mon Sep 17 00:00:00 2001
From: Pax1601 <d.passoni92@gmail.com>
Date: Fri, 8 Nov 2024 20:42:31 +0100
Subject: [PATCH] Fixed missing category error

---
 frontend/react/src/unit/databases/unitdatabase.ts | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/frontend/react/src/unit/databases/unitdatabase.ts b/frontend/react/src/unit/databases/unitdatabase.ts
index 0d218033..aecd0383 100644
--- a/frontend/react/src/unit/databases/unitdatabase.ts
+++ b/frontend/react/src/unit/databases/unitdatabase.ts
@@ -8,7 +8,7 @@ export class UnitDatabase {
 
   constructor() {}
 
-  load(url: string) {
+  load(url: string, category?: string) {
     if (url !== "") {
       var xhr = new XMLHttpRequest();
       xhr.open("GET", url, true);
@@ -18,6 +18,12 @@ export class UnitDatabase {
         var status = xhr.status;
         if (status === 200) {
           const newBlueprints = xhr.response as { [key: string]: UnitBlueprint };
+          for (let unit in newBlueprints) {
+            if (!newBlueprints[unit].category) {
+              if (category)  newBlueprints[unit].category = category
+              else console.warn(`No category provided for ${unit}`)
+            } 
+          }
           this.blueprints = { ...this.blueprints, ...newBlueprints };
           UnitDatabaseLoadedEvent.dispatch();
         } else {