From 0000000000000000000000000000000000000000 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


diff --git a/src/main/java/org/bukkit/entity/WanderingTrader.java b/src/main/java/org/bukkit/entity/WanderingTrader.java
index 55394ed5c68cb0bf4333fc918e3b4c8c4e3db0c6..da76e1ed5406322073dd8c7a89ca55aa68620ac4 100644
--- a/src/main/java/org/bukkit/entity/WanderingTrader.java
+++ b/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
 }