Fix crashing by turtle unequip

Fix NullPointerException by upgrade.getCraftingItem()
PolymerPorts · Jun 2, 2024 · bfc0f6d · bfc0f6d
1 parent 5ed76d0
commit bfc0f6d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/main/java/eu/pb4/cctpatch/impl/poly/model/TurtleModel.java b/src/main/java/eu/pb4/cctpatch/impl/poly/model/TurtleModel.java
@@ -124,6 +124,9 @@ public void setUpgrades(TurtleBrain brain, ITurtleUpgrade left, ITurtleUpgrade r
     }
 
     private ItemStack getUpgradeModel(ITurtleUpgrade upgrade, TurtleBrain brain, TurtleSide turtleSide) {
+        if(upgrade == null){
+            return ItemStack.EMPTY;
+        }
         if (upgrade instanceof TurtleModemAccessor modem) {
             var type = modem.isAdvanced() ? ADVANCED_MODEM_MODEL : NORMAL_MODEM_MODEL;
             var x = brain.getUpgradeData(turtleSide).get(ModRegistry.DataComponents.ON.get());