PaperMC/patches/server/0558-Added-firing-of-PlayerChangeBeaconEffectEvent.patch

From 0000000000000000000000000000000000000000 Mon Sep 17 00:00:00 2001
From: Jake Potrebic <jake.m.potrebic@gmail.com>
Date: Wed, 24 Jun 2020 15:14:51 -0600
Subject: [PATCH] Added firing of PlayerChangeBeaconEffectEvent


diff --git a/src/main/java/net/minecraft/world/inventory/BeaconMenu.java b/src/main/java/net/minecraft/world/inventory/BeaconMenu.java
index 5e6582110c97287f0a2ed6e8d1a8f5e1ee2c81e5..4e40203125b2931d45666fccb022e84622ea61c2 100644
--- a/src/main/java/net/minecraft/world/inventory/BeaconMenu.java
+++ b/src/main/java/net/minecraft/world/inventory/BeaconMenu.java
@@ -158,13 +158,27 @@ public class BeaconMenu extends AbstractContainerMenu {
     public MobEffect getSecondaryEffect() {
         return MobEffect.byId(this.beaconData.get(2));
     }
+    // Paper start
+    private static @Nullable org.bukkit.potion.PotionEffectType convert(Optional<MobEffect> effect) {
+        return effect.flatMap(net.minecraft.core.registries.BuiltInRegistries.MOB_EFFECT::getResourceKey).map(key -> {
+            return org.bukkit.potion.PotionEffectType.getByKey(org.bukkit.craftbukkit.util.CraftNamespacedKey.fromMinecraft(key.location()));
+        }).orElse(null);
+    }
+    // Paper end
 
     public void updateEffects(Optional<MobEffect> primary, Optional<MobEffect> secondary) {
         if (this.paymentSlot.hasItem()) {
-            this.beaconData.set(1, (Integer) primary.map(MobEffect::getId).orElse(-1));
-            this.beaconData.set(2, (Integer) secondary.map(MobEffect::getId).orElse(-1));
+            // Paper start
+            io.papermc.paper.event.player.PlayerChangeBeaconEffectEvent event = new io.papermc.paper.event.player.PlayerChangeBeaconEffectEvent((org.bukkit.entity.Player) this.player.player.getBukkitEntity(), convert(primary), convert(secondary), this.access.getLocation().getBlock());
+            if (event.callEvent()) {
+            this.beaconData.set(1, event.getPrimary() == null ? -1 : event.getPrimary().getId());
+            this.beaconData.set(2, event.getSecondary() == null ? -1 : event.getSecondary().getId());
+            if (event.willConsumeItem()) {
+            // Paper end
             this.paymentSlot.remove(1);
+            }
             this.access.execute(Level::blockEntityChanged);
+            } // Paper end
         }
 
     }