Made player created Stone mobs attack any undead mob

TeamMetallurgy · Jun 9, 2019 · 5c113c2 · 5c113c2
1 parent 4285bb1
commit 5c113c2
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/main/java/com/teammetallurgy/atum/entity/stone/EntityStoneBase.java b/src/main/java/com/teammetallurgy/atum/entity/stone/EntityStoneBase.java
@@ -6,6 +6,7 @@
 import com.teammetallurgy.atum.init.AtumItems;
 import net.minecraft.block.Block;
 import net.minecraft.entity.Entity;
+import net.minecraft.entity.EnumCreatureAttribute;
 import net.minecraft.entity.IEntityLivingData;
 import net.minecraft.entity.SharedMonsterAttributes;
 import net.minecraft.entity.ai.*;
@@ -52,11 +53,12 @@ protected void initEntityAI() {
     }
 
     private void applyEntityAI() {
+        this.targetTasks.addTask(1, new EntityAIHurtByTarget(this, false));
         this.targetTasks.addTask(1, new EntityAINearestAttackableTarget<>(this, EntityPlayer.class, 10, true, false, input -> !isPlayerCreated()));
         this.targetTasks.addTask(1, new EntityAINearestAttackableTarget<>(this, EntityStoneBase.class, 10, true, false, input -> input != null && (!input.isPlayerCreated() && isPlayerCreated() || input.isPlayerCreated() && !isPlayerCreated())));
         this.targetTasks.addTask(1, new EntityAINearestAttackableTarget<>(this, EntityUndeadBase.class, true));
         this.targetTasks.addTask(1, new EntityAINearestAttackableTarget<>(this, EntityBanditBase.class, true));
-        this.targetTasks.addTask(1, new EntityAIHurtByTarget(this, false));
+        this.targetTasks.addTask(2, new EntityAINearestAttackableTarget<>(this, EntityMob.class, 10, true, false, input -> input != null && this.isPlayerCreated() && !(input instanceof EntityStoneBase) && input.getCreatureAttribute() == EnumCreatureAttribute.UNDEAD));
     }
 
     @Override