Add pig variant attribute

Author: OliverSchlueter

implementation_1_21_5/src/main/java/de/oliver/fancynpcs/v1_21_5/attributes/PigAttributes.java

@@ -3,8 +3,15 @@
 import de.oliver.fancynpcs.api.Npc;
 import de.oliver.fancynpcs.api.NpcAttribute;
 import de.oliver.fancynpcs.v1_21_5.ReflectionHelper;
+import net.minecraft.core.Holder;
+import net.minecraft.core.HolderLookup;
+import net.minecraft.core.registries.Registries;
+import net.minecraft.data.registries.VanillaRegistries;
+import net.minecraft.resources.ResourceKey;
+import net.minecraft.resources.ResourceLocation;
 import net.minecraft.world.entity.EquipmentSlot;
 import net.minecraft.world.entity.animal.Pig;
+import net.minecraft.world.entity.animal.PigVariant;
 import net.minecraft.world.item.Items;
 import org.bukkit.entity.EntityType;
 
@@ -16,6 +23,16 @@ public class PigAttributes {
     public static List<NpcAttribute> getAllAttributes() {
         List<NpcAttribute> attributes = new ArrayList<>();
 
+        attributes.add(new NpcAttribute(
+                "variant",
+                getPigVariantRegistry()
+                        .listElementIds()
+                        .map(id -> id.location().getPath())
+                        .toList(),
+                List.of(EntityType.CAT),
+                PigAttributes::setVariant
+        ));
+
         attributes.add(new NpcAttribute(
                 "has_saddle",
                 List.of("true", "false"),
@@ -26,6 +43,19 @@ public static List<NpcAttribute> getAllAttributes() {
         return attributes;
     }
 
+    private static void setVariant(Npc npc, String value) {
+        final Pig pig = ReflectionHelper.getEntity(npc);
+
+        Holder<PigVariant> variant = getPigVariantRegistry()
+                .get(ResourceKey.create(
+                        Registries.PIG_VARIANT,
+                        ResourceLocation.withDefaultNamespace(value.toLowerCase())
+                ))
+                .orElseThrow();
+
+        pig.setVariant(variant);
+    }
+
     private static void setHasSaddle(Npc npc, String value) {
         Pig pig = ReflectionHelper.getEntity(npc);
 
@@ -36,4 +66,11 @@ private static void setHasSaddle(Npc npc, String value) {
         }
     }
 
+    private static HolderLookup.RegistryLookup<PigVariant> getPigVariantRegistry() {
+        return VanillaRegistries
+                .createLookup()
+                .lookup(Registries.PIG_VARIANT)
+                .orElseThrow();
+    }
+
 }