1 files changed, 46 insertions, 0 deletions
diff --git a/src/main/java/io/github/moulberry/notenoughupdates/miscfeatures/entityviewer/SkinModifier.java b/src/main/java/io/github/moulberry/notenoughupdates/miscfeatures/entityviewer/SkinModifier.java
new file mode 100644
index 00000000..55e8a432
--- /dev/null
+++ b/src/main/java/io/github/moulberry/notenoughupdates/miscfeatures/entityviewer/SkinModifier.java
@@ -0,0 +1,46 @@
+package io.github.moulberry.notenoughupdates.miscfeatures.entityviewer;
+
+import com.google.gson.JsonElement;
+import com.google.gson.JsonObject;
+import net.minecraft.entity.EntityLivingBase;
+import net.minecraft.entity.player.EnumPlayerModelParts;
+import net.minecraft.util.ResourceLocation;
+
+import java.util.Map;
+
+public class SkinModifier extends EntityViewerModifier {
+    @Override
+    public EntityLivingBase applyModifier(EntityLivingBase base, JsonObject info) {
+        if (base instanceof GUIClientPlayer) {
+            GUIClientPlayer player = (GUIClientPlayer) base;
+            if (info.has("cape")) {
+                player.overrideCape = new ResourceLocation(info.get("cape").getAsString());
+            }
+            if (info.has("skin")) {
+                player.overrideSkin = new ResourceLocation(info.get("skin").getAsString());
+            }
+            if (info.has("slim")) {
+                player.overrideIsSlim = info.get("slim").getAsBoolean();
+            }
+            if (info.has("parts")) {
+                JsonElement parts = info.get("parts");
+                byte partBitField = player.getDataWatcher().getWatchableObjectByte(10);
+                if (parts.isJsonPrimitive() && parts.getAsJsonPrimitive().isBoolean()) {
+                    partBitField = parts.getAsBoolean() ? (byte) -1 : 0;
+                } else {
+                    JsonObject obj = parts.getAsJsonObject();
+                    for (Map.Entry<String, JsonElement> part : obj.entrySet()) {
+                        EnumPlayerModelParts modelPart = EnumPlayerModelParts.valueOf(part.getKey());
+                        if (part.getValue().getAsBoolean()) {
+                            partBitField |= modelPart.getPartMask();
+                        } else {
+                            partBitField &= ~modelPart.getPartMask();
+                        }
+                    }
+                }
+                player.getDataWatcher().updateObject(10, partBitField);
+            }
+        }
+        return base;
+    }
+}