Add cow variant attribute

Author: OliverSchlueter

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

@@ -36,6 +36,7 @@ public static List<NpcAttribute> getAllAttributes() {
         attributes.addAll(WolfAttributes.getAllAttributes());
         attributes.addAll(SlimeAttributes.getAllAttributes());
         attributes.addAll(PigAttributes.getAllAttributes());
+        attributes.addAll(CowAttributes.getAllAttributes());
         attributes.addAll(ArmorStandAttributes.getAllAttributes());
         attributes.addAll(BeeAttributes.getAllAttributes());
         attributes.addAll(VexAttributes.getAllAttributes());

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

@@ -0,0 +1,57 @@
+package de.oliver.fancynpcs.v1_21_5.attributes;
+
+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.animal.Cow;
+import net.minecraft.world.entity.animal.CowVariant;
+import org.bukkit.entity.EntityType;
+
+import java.util.ArrayList;
+import java.util.List;
+
+public class CowAttributes {
+
+    public static List<NpcAttribute> getAllAttributes() {
+        List<NpcAttribute> attributes = new ArrayList<>();
+
+        attributes.add(new NpcAttribute(
+                "variant",
+                getCowVariantRegistry()
+                        .listElementIds()
+                        .map(id -> id.location().getPath())
+                        .toList(),
+                List.of(EntityType.COW),
+                CowAttributes::setVariant
+        ));
+
+        return attributes;
+    }
+
+    private static void setVariant(Npc npc, String value) {
+        final Cow cow = ReflectionHelper.getEntity(npc);
+
+        Holder<CowVariant> variant = getCowVariantRegistry()
+                .get(ResourceKey.create(
+                        Registries.COW_VARIANT,
+                        ResourceLocation.withDefaultNamespace(value.toLowerCase())
+                ))
+                .orElseThrow();
+
+        cow.setVariant(variant);
+    }
+
+    private static HolderLookup.RegistryLookup<CowVariant> getCowVariantRegistry() {
+        return VanillaRegistries
+                .createLookup()
+                .lookup(Registries.COW_VARIANT)
+                .orElseThrow();
+    }
+
+}

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

@@ -29,7 +29,7 @@ public static List<NpcAttribute> getAllAttributes() {
                         .listElementIds()
                         .map(id -> id.location().getPath())
                         .toList(),
-                List.of(EntityType.CAT),
+                List.of(EntityType.PIG),
                 PigAttributes::setVariant
         ));