Make hidebehinds retaliate when attacked while unable to attack you due to light, and make them immortal to attacks during this time. Also make hidebehinds despawn during the daytime (Fixes #13)

Author: itsmeow

src/main/java/dev/itsmeow/whisperwoods/entity/EntityHidebehind.java

@@ -13,6 +13,7 @@
 import net.minecraft.entity.ai.RandomPositionGenerator;
 import net.minecraft.entity.ai.attributes.Attributes;
 import net.minecraft.entity.ai.goal.Goal;
+import net.minecraft.entity.ai.goal.HurtByTargetGoal;
 import net.minecraft.entity.ai.goal.SwimGoal;
 import net.minecraft.entity.player.PlayerEntity;
 import net.minecraft.item.BlockItem;
@@ -22,6 +23,8 @@
 import net.minecraft.network.datasync.DataSerializers;
 import net.minecraft.network.datasync.EntityDataManager;
 import net.minecraft.pathfinding.*;
+import net.minecraft.potion.EffectInstance;
+import net.minecraft.potion.Effects;
 import net.minecraft.tags.BlockTags;
 import net.minecraft.util.*;
 import net.minecraft.util.math.BlockPos;
@@ -72,6 +75,22 @@ public boolean attackEntityFrom(DamageSource source, float amount) {
         if(source.getTrueSource() == this.getAttackTarget() && this.attackSequenceTicks() > 0) {
             this.setAttackSequenceTicks(0);
         }
+        if (!world.isRemote()) {
+            boolean isImmediate = source.getImmediateSource() instanceof PlayerEntity;
+            PlayerEntity player = isImmediate ? (PlayerEntity) source.getImmediateSource() : (source.getTrueSource() instanceof PlayerEntity ? (PlayerEntity) source.getTrueSource() : null);
+            if (player != null) {
+                if (!this.isEntityAttackable(player)) {
+                    // retaliate attacks if you can't chase due to light
+                    if (!player.isCreative()) {
+                        player.addPotionEffect(new EffectInstance(Effects.BLINDNESS, 15 * 20, 1));
+                        if (player.getDistance(this) < 3)
+                            player.attackEntityFrom(HIDEBEHIND, 1F);
+                    }
+                    HideFromTargetGoal.doTreeTick(this);
+                    return false;
+                }
+            }
+        }
         return super.attackEntityFrom(source, amount);
     }
 
@@ -330,7 +349,11 @@ public void startExecuting() {
 
         @Override
         public void tick() {
-            this.hidebehind.getNavigator().clearPath();
+            doTreeTick(hidebehind);
+        }
+
+        public static void doTreeTick(EntityHidebehind hidebehind) {
+            hidebehind.getNavigator().clearPath();
             boolean nearTree = false;
             for(Direction dir : Direction.values()) {
                 if(!nearTree) {
@@ -385,6 +408,11 @@ protected void registerData() {
         this.dataManager.register(ATTACK_SEQUENCE_TICKS, 0);
     }
 
+    @Override
+    public boolean canDespawn(double range) {
+        return world.isDaytime() && super.canDespawn(range);
+    }
+
     @Override
     protected PathNavigator createNavigator(World world) {
         return new HidebehindGroundNavigator(this, world);

src/main/java/dev/itsmeow/whisperwoods/init/ModEntities.java

@@ -53,6 +53,7 @@ public static void subscribe(IEventBus modBus) {
     .defaultPlacement((t, w, e, p, r) -> w.getDifficulty() != Difficulty.PEACEFUL && MobEntity.canSpawnOn(t, w, e, p, r))
     .egg(0x473123, 0xfff494)
     .size(1F, 5.2F)
+    .despawn()
     .variants(
     new HidebehindVariant("black"),
     new HidebehindVariant("coniferous"),