added growing health

Globox1997 · Mar 23, 2021 · ac99b6d · ac99b6d
1 parent c27b956
commit ac99b6d
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/main/java/net/adventurez/entity/DragonEntity.java b/src/main/java/net/adventurez/entity/DragonEntity.java
@@ -581,9 +581,11 @@ public void tick() {
          }
          if (this.dragonAge == 5 && this.getSize() == 1) {
             this.setSize(2);
+            this.healDragonIfGrowing();
          }
          if (this.dragonAge == 15 && this.getSize() == 2) {
             this.setSize(3);
+            this.healDragonIfGrowing();
          }
          if (this.dragonAgeFoodBonus > 5) {
             this.dragonAgeFoodBonus = 0;
@@ -593,6 +595,12 @@ public void tick() {
       }
    }
 
+   private void healDragonIfGrowing() {
+      if (this.getHealth() < this.getMaxHealth()) {
+         this.setHealth(this.getHealth() + 20F);
+      }
+   }
+
    @Override
    public boolean isPushable() {
       return !this.hasPassengers();