From e289e5508f6251c6e1c236bf739c7b096af8ac78 Mon Sep 17 00:00:00 2001
From: Aikar <aikar@aikar.co>
Date: Tue, 5 Jun 2018 22:47:08 -0400
Subject: [PATCH] WitchReadyPotionEvent

Control what potion the witch readies to use

diff --git a/src/main/java/com/destroystokyo/paper/event/entity/WitchReadyPotionEvent.java b/src/main/java/com/destroystokyo/paper/event/entity/WitchReadyPotionEvent.java
new file mode 100644
index 000000000..5351b523d
--- /dev/null
+++ b/src/main/java/com/destroystokyo/paper/event/entity/WitchReadyPotionEvent.java
@@ -0,0 +1,80 @@
+package com.destroystokyo.paper.event.entity;
+
+import org.bukkit.Material;
+import org.bukkit.entity.Witch;
+import org.bukkit.event.Cancellable;
+import org.bukkit.event.HandlerList;
+import org.bukkit.event.entity.EntityEvent;
+import org.bukkit.inventory.ItemStack;
+import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
+
+public class WitchReadyPotionEvent extends EntityEvent implements Cancellable {
+    private ItemStack potion;
+
+    public WitchReadyPotionEvent(@NotNull Witch witch, @Nullable ItemStack potion) {
+        super(witch);
+        this.potion = potion;
+    }
+
+    /**
+     * Fires thee event, returning the desired potion, or air of cancelled
+     * @param witch the witch whom is readying to use a potion
+     * @param potion the potion to be used
+     * @return The ItemStack to be used
+     */
+    @Nullable
+    public static ItemStack process(@NotNull Witch witch, @Nullable ItemStack potion) {
+        WitchReadyPotionEvent event = new WitchReadyPotionEvent(witch, potion);
+        if (!event.callEvent() || event.getPotion() == null) {
+            return new ItemStack(Material.AIR);
+        }
+        return event.getPotion();
+    }
+
+    @NotNull
+    @Override
+    public Witch getEntity() {
+        return (Witch) super.getEntity();
+    }
+
+    /**
+     * @return the potion the witch is readying to use
+     */
+    @Nullable
+    public ItemStack getPotion() {
+        return potion;
+    }
+
+    /**
+     * Sets the potion the which is going to hold and use
+     * @param potion The potion
+     */
+    public void setPotion(@Nullable ItemStack potion) {
+        this.potion = potion != null ? potion.clone() : null;
+    }
+
+    private static final HandlerList handlers = new HandlerList();
+
+    @NotNull
+    public HandlerList getHandlers() {
+        return handlers;
+    }
+
+    @NotNull
+    public static HandlerList getHandlerList() {
+        return handlers;
+    }
+
+    private boolean cancelled = false;
+
+    @Override
+    public boolean isCancelled() {
+        return cancelled;
+    }
+
+    @Override
+    public void setCancelled(boolean cancel) {
+        cancelled = cancel;
+    }
+}
-- 
2.21.0