From 8f72ada8c9519db3292e67192941b5b90864220f Mon Sep 17 00:00:00 2001
From: Jake Potrebic <jake.m.potrebic@gmail.com>
Date: Fri, 20 Aug 2021 13:03:55 -0700
Subject: [PATCH] Get entity default attributes

---
 .../main/java/org/bukkit/UnsafeValues.java    | 17 +++++++++++++++++
 .../java/org/bukkit/entity/EntityType.java    | 19 +++++++++++++++++++
 2 files changed, 36 insertions(+)

diff --git a/paper-api/src/main/java/org/bukkit/UnsafeValues.java b/paper-api/src/main/java/org/bukkit/UnsafeValues.java
index 218adc63a0..4cd43815c3 100644
--- a/paper-api/src/main/java/org/bukkit/UnsafeValues.java
+++ b/paper-api/src/main/java/org/bukkit/UnsafeValues.java
@@ -197,5 +197,22 @@ public interface UnsafeValues {
      * @return true if valid repair, false if not
      */
     public boolean isValidRepairItemStack(@org.jetbrains.annotations.NotNull ItemStack itemToBeRepaired, @org.jetbrains.annotations.NotNull ItemStack repairMaterial);
+
+    /**
+     * Checks if the entity represented by the namespaced key has default attributes.
+     *
+     * @param entityKey the entity's key
+     * @return true if it has default attributes
+     */
+    boolean hasDefaultEntityAttributes(@org.jetbrains.annotations.NotNull NamespacedKey entityKey);
+
+    /**
+     * Gets the default attributes for the entity represented by the namespaced key.
+     *
+     * @param entityKey the entity's key
+     * @return an unmodifiable instance of Attributable for reading default attributes.
+     * @throws IllegalArgumentException if the entity does not exist of have default attributes (use {@link #hasDefaultEntityAttributes(NamespacedKey)} first)
+     */
+    @org.jetbrains.annotations.NotNull org.bukkit.attribute.Attributable getDefaultEntityAttributes(@org.jetbrains.annotations.NotNull NamespacedKey entityKey);
     // Paper end
 }
diff --git a/paper-api/src/main/java/org/bukkit/entity/EntityType.java b/paper-api/src/main/java/org/bukkit/entity/EntityType.java
index fd5d1ad033..6bc514af55 100644
--- a/paper-api/src/main/java/org/bukkit/entity/EntityType.java
+++ b/paper-api/src/main/java/org/bukkit/entity/EntityType.java
@@ -477,6 +477,25 @@ public enum EntityType implements Keyed, Translatable, net.kyori.adventure.trans
         Preconditions.checkArgument(this != UNKNOWN, "UNKNOWN entities do not have translation keys");
         return org.bukkit.Bukkit.getUnsafe().getTranslationKey(this);
     }
+
+    /**
+     * Checks if the entity has default attributes.
+     *
+     * @return true if it has default attributes
+     */
+    public boolean hasDefaultAttributes() {
+        return org.bukkit.Bukkit.getUnsafe().hasDefaultEntityAttributes(this.key);
+    }
+
+    /**
+     * Gets the default attributes for the entity.
+     *
+     * @return an unmodifiable instance of Attributable for reading default attributes.
+     * @throws IllegalArgumentException if the entity does not exist of have default attributes (use {@link #hasDefaultAttributes()} first)
+     */
+    public @NotNull org.bukkit.attribute.Attributable getDefaultAttributes() {
+        return org.bukkit.Bukkit.getUnsafe().getDefaultEntityAttributes(this.key);
+    }
     // Paper end
 
     /**