1 files changed, 30 insertions, 0 deletions
diff --git a/src/main/java/io/github/moulberry/notenoughupdates/miscfeatures/entityviewer/AgeModifier.java b/src/main/java/io/github/moulberry/notenoughupdates/miscfeatures/entityviewer/AgeModifier.java
new file mode 100644
index 00000000..5884512f
--- /dev/null
+++ b/src/main/java/io/github/moulberry/notenoughupdates/miscfeatures/entityviewer/AgeModifier.java
@@ -0,0 +1,30 @@
+package io.github.moulberry.notenoughupdates.miscfeatures.entityviewer;
+
+import com.google.gson.JsonObject;
+import io.github.moulberry.notenoughupdates.mixins.AccessorEntityAgeable;
+import io.github.moulberry.notenoughupdates.mixins.AccessorEntityArmorStand;
+import net.minecraft.entity.EntityAgeable;
+import net.minecraft.entity.EntityLivingBase;
+import net.minecraft.entity.item.EntityArmorStand;
+import net.minecraft.entity.monster.EntityZombie;
+
+public class AgeModifier extends EntityViewerModifier {
+    @Override
+    public EntityLivingBase applyModifier(EntityLivingBase base, JsonObject info) {
+        boolean baby = info.has("baby") && info.get("baby").getAsBoolean();
+        if (base instanceof EntityAgeable) {
+            ((AccessorEntityAgeable) base).setGrowingAgeDirect(baby ? -1 : 1);
+            return base;
+        }
+        if (base instanceof EntityZombie) {
+            ((EntityZombie) base).setChild(baby);
+            return base;
+        }
+        if (base instanceof EntityArmorStand) {
+            ((AccessorEntityArmorStand) base).setSmallDirect(baby);
+            return base;
+        }
+        System.out.println("Cannot apply age to a non ageable entity: " + base);
+        return null;
+    }
+}