From d0082ab64464d0f0539a47215750a40eab3cb317 Mon Sep 17 00:00:00 2001
From: IllianiCBT <danielfarbrace@icloud.com>
Date: Thu, 26 Sep 2024 16:11:37 -0500
Subject: [PATCH] Added <50.01 Compatibility BayType Handlers

Added a compatibility handler to convert legacy bay type names "MECH" and "PROTOMECH" to their current equivalents "MEK" and "PROTOMEK".
---
 MekHQ/src/mekhq/campaign/parts/Cubicle.java   | 26 ++++++++++++++-----
 .../mekhq/campaign/parts/MissingCubicle.java  | 26 ++++++++++++++-----
 2 files changed, 38 insertions(+), 14 deletions(-)

diff --git a/MekHQ/src/mekhq/campaign/parts/Cubicle.java b/MekHQ/src/mekhq/campaign/parts/Cubicle.java
index 3a5bae7425..d318939754 100644
--- a/MekHQ/src/mekhq/campaign/parts/Cubicle.java
+++ b/MekHQ/src/mekhq/campaign/parts/Cubicle.java
@@ -18,11 +18,6 @@
  */
 package mekhq.campaign.parts;
 
-import java.io.PrintWriter;
-
-import org.w3c.dom.Node;
-import org.w3c.dom.NodeList;
-
 import megamek.common.BayType;
 import megamek.common.Entity;
 import megamek.common.ITechnology;
@@ -31,6 +26,11 @@
 import mekhq.campaign.Campaign;
 import mekhq.campaign.finances.Money;
 import mekhq.utilities.MHQXMLUtility;
+import org.w3c.dom.Node;
+import org.w3c.dom.NodeList;
+
+import java.io.PrintWriter;
+import java.util.Objects;
 
 /**
  * A transport bay cubicle for a Mek, ProtoMek, vehicle, fighter, or small
@@ -164,9 +164,21 @@ protected void loadFieldsFromXmlNode(Node wn) {
         for (int x = 0; x < nl.getLength(); x++) {
             Node wn2 = nl.item(x);
             if (wn2.getNodeName().equalsIgnoreCase("bayType")) {
-                bayType = BayType.parse(wn2.getTextContent());
+                // <50.01 compatibility handler
+                String bayRawValue = wn2.getTextContent();
+
+                if (Objects.equals(bayRawValue, "MECH")) {
+                    bayRawValue = "MEK";
+                }
+
+                if (Objects.equals(bayRawValue, "PROTOMECH")) {
+                    bayRawValue = "PROTOMEK";
+                }
+
+                bayType = BayType.parse(bayRawValue);
                 if (null == bayType) {
-                    logger.error("Could not parse bay type " + wn2.getTextContent());
+                    logger.error(String.format("Could not parse bay type %s treating as BayType.Mek",
+                        wn2.getTextContent()));
                     bayType = BayType.MEK;
                 }
                 name = bayType.getDisplayName() + " Cubicle";
diff --git a/MekHQ/src/mekhq/campaign/parts/MissingCubicle.java b/MekHQ/src/mekhq/campaign/parts/MissingCubicle.java
index 8ff46aafe6..3ea1906f89 100644
--- a/MekHQ/src/mekhq/campaign/parts/MissingCubicle.java
+++ b/MekHQ/src/mekhq/campaign/parts/MissingCubicle.java
@@ -18,11 +18,6 @@
  */
 package mekhq.campaign.parts;
 
-import java.io.PrintWriter;
-
-import org.w3c.dom.Node;
-import org.w3c.dom.NodeList;
-
 import megamek.common.BayType;
 import megamek.common.Entity;
 import megamek.common.ITechnology;
@@ -30,6 +25,11 @@
 import megamek.logging.MMLogger;
 import mekhq.campaign.Campaign;
 import mekhq.utilities.MHQXMLUtility;
+import org.w3c.dom.Node;
+import org.w3c.dom.NodeList;
+
+import java.io.PrintWriter;
+import java.util.Objects;
 
 /**
  * @author Neoancient
@@ -123,9 +123,21 @@ protected void loadFieldsFromXmlNode(Node wn) {
         for (int x = 0; x < nl.getLength(); x++) {
             Node wn2 = nl.item(x);
             if (wn2.getNodeName().equalsIgnoreCase("bayType")) {
-                bayType = BayType.parse(wn2.getTextContent());
+                // <50.01 compatibility handler
+                String bayRawValue = wn2.getTextContent();
+
+                if (Objects.equals(bayRawValue, "MECH")) {
+                    bayRawValue = "MEK";
+                }
+
+                if (Objects.equals(bayRawValue, "PROTOMECH")) {
+                    bayRawValue = "PROTOMEK";
+                }
+
+                bayType = BayType.parse(bayRawValue);
                 if (null == bayType) {
-                    logger.error("Could not parse bay type " + wn2.getTextContent());
+                    logger.error(String.format("Could not parse bay type %s treating as BayType.Mek",
+                        wn2.getTextContent()));
                     bayType = BayType.MEK;
                 }
                 name = bayType.getDisplayName() + " Cubicle";