1 files changed, 20 insertions, 0 deletions
diff --git a/src/main/kotlin/gui/entity/ModifyWither.kt b/src/main/kotlin/gui/entity/ModifyWither.kt
new file mode 100644
index 0000000..6083d88
--- /dev/null
+++ b/src/main/kotlin/gui/entity/ModifyWither.kt
@@ -0,0 +1,20 @@
+
+package moe.nea.firmament.gui.entity
+
+import com.google.gson.JsonObject
+import net.minecraft.entity.LivingEntity
+import net.minecraft.entity.boss.WitherEntity
+
+object ModifyWither : EntityModifier {
+    override fun apply(entity: LivingEntity, info: JsonObject): LivingEntity {
+        require(entity is WitherEntity)
+        info["tiny"]?.let {
+            entity.setInvulTimer(if (it.asBoolean) 800 else 0)
+        }
+        info["armored"]?.let {
+            entity.health = if (it.asBoolean) 1F else entity.maxHealth
+        }
+        return entity
+    }
+
+}