src/main/java/io/github/moulberry/notenoughupdates/miscfeatures/entityviewer/WitherModifier.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
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;
    }
}