Fixes MatrexsVigil#43

Catches errors if a mob is riding another with custom ai and its chunk unloads. Mostly noticeable in the case of Alex's mobs. It even targets the right branch this time!
Weedoof · Jul 15, 2022 · 1b89af4 · 1b89af4
1 parent eba4676
commit 1b89af4
Showing 1 changed file with 26 additions and 6 deletions.
diff --git a/src/main/java/pam/pamhc2crops/events/TemptationTask.java b/src/main/java/pam/pamhc2crops/events/TemptationTask.java
@@ -180,27 +180,47 @@ public void onEntitySpawn(EntityJoinWorldEvent event) {
 
 		if (entity instanceof ChickenEntity) {
 			ChickenEntity chicken = (ChickenEntity) entity;
-			chicken.goalSelector.addGoal(3, new MoreTemptation(chicken, 1.0D, false, Chicken));
+			try {
+            			chicken.goalSelector.addGoal(3, new MoreTemptation(chicken, 1.0D, false, Chicken));
+            		} catch(IllegalArgumentException e) {
+            			// whoops
+            		}
 		}
 
 		if (entity instanceof CowEntity) {
 			CowEntity cow = (CowEntity) entity;
-			cow.goalSelector.addGoal(3, new MoreTemptation(cow, 1.25D, false, Grain));
+			try {
+            			cow.goalSelector.addGoal(3, new MoreTemptation(cow, 1.25D, false, Grain));
+            		} catch(IllegalArgumentException e) {
+            			// whoops
+            		}
 		}
 
 		if (entity instanceof SheepEntity) {
 			SheepEntity sheep = (SheepEntity) entity;
-			sheep.goalSelector.addGoal(3, new MoreTemptation(sheep, 1.0D, false, Grain));
+			try {
+            			sheep.goalSelector.addGoal(3, new MoreTemptation(sheep, 1.0D, false, Grain));
+            		} catch(IllegalArgumentException e) {
+            			// whoops
+            		}
 		}
 
 		if (entity instanceof PigEntity) {
 			PigEntity pig = (PigEntity) entity;
-			pig.goalSelector.addGoal(4, new MoreTemptation(pig, 1.2D, false, Pig));
+			try {
+            			pig.goalSelector.addGoal(4, new MoreTemptation(pig, 1.2D, false, Pig));
+            		} catch(IllegalArgumentException e) {
+            			// whoops
+            		}
 		}
 
 		if (entity instanceof RabbitEntity) {
 			RabbitEntity rabbit = (RabbitEntity) entity;
-			rabbit.goalSelector.addGoal(4, new MoreTemptation(rabbit, 1.2D, false, Rabbit));
+			try {
+            			rabbit.goalSelector.addGoal(4, new MoreTemptation(rabbit, 1.2D, false, Rabbit));
+            		} catch(IllegalArgumentException e) {
+            			// whoops
+            		}
 		}
 	}
-}
+}