From f4e92bc2673b29c32ad877de756cbb8ae982acc2 Mon Sep 17 00:00:00 2001
From: HexedHero <6012891+HexedHero@users.noreply.github.com>
Date: Thu, 6 May 2021 14:56:26 +0100
Subject: [PATCH] Add more WanderingTrader API

---
 .../org/bukkit/entity/WanderingTrader.java    | 26 +++++++++++++++++++
 1 file changed, 26 insertions(+)

diff --git a/paper-api/src/main/java/org/bukkit/entity/WanderingTrader.java b/paper-api/src/main/java/org/bukkit/entity/WanderingTrader.java
index 55394ed5c..da76e1ed5 100644
--- a/paper-api/src/main/java/org/bukkit/entity/WanderingTrader.java
+++ b/paper-api/src/main/java/org/bukkit/entity/WanderingTrader.java
@@ -28,4 +28,30 @@ public interface WanderingTrader extends AbstractVillager {
      * {@link WanderingTrader} is forcibly despawned
      */
     public void setDespawnDelay(int despawnDelay);
+    
+    // Paper start - Add more WanderingTrader API
+    /**
+     * Set if the Wandering Trader can and will drink an invisibility potion.
+     * @param bool whether the mob will drink
+     */
+    public void setCanDrinkPotion(boolean bool);
+
+    /**
+     * Get if the Wandering Trader can and will drink an invisibility potion.
+     * @return whether the mob will drink
+     */
+    public boolean canDrinkPotion();
+
+    /**
+     * Set if the Wandering Trader can and will drink milk.
+      * @param bool whether the mob will drink
+     */
+    public void setCanDrinkMilk(boolean bool);
+
+    /**
+     * Get if the Wandering Trader can and will drink milk.
+     * @return whether the mob will drink
+     */
+    public boolean canDrinkMilk();
+    // Paper end
 }