1 files changed, 29 insertions, 0 deletions
diff --git a/src/main/java/io/github/moulberry/notenoughupdates/miscfeatures/entityviewer/WitherModifier.java b/src/main/java/io/github/moulberry/notenoughupdates/miscfeatures/entityviewer/WitherModifier.java
new file mode 100644
index 00000000..c5580f17
--- /dev/null
+++ b/src/main/java/io/github/moulberry/notenoughupdates/miscfeatures/entityviewer/WitherModifier.java
@@ -0,0 +1,29 @@
+package io.github.moulberry.notenoughupdates.miscfeatures.entityviewer;
+
+import com.google.gson.JsonObject;
+import net.minecraft.entity.EntityLivingBase;
+import net.minecraft.entity.boss.EntityWither;
+
+public class WitherModifier extends EntityViewerModifier {
+    @Override
+    public EntityLivingBase applyModifier(EntityLivingBase base, JsonObject info) {
+        if (!(base instanceof EntityWither))
+            return null;
+        EntityWither wither = (EntityWither) base;
+        if (info.has("tiny")) {
+            if (info.get("tiny").getAsBoolean()) {
+                wither.setInvulTime(800);
+            } else {
+                wither.setInvulTime(0);
+            }
+        }
+        if (info.has("armored")) {
+            if (info.get("armored").getAsBoolean()) {
+                wither.setHealth(1);
+            } else {
+                wither.setHealth(wither.getMaxHealth());
+            }
+        }
+        return base;
+    }
+}