Fix TNTSimulatorListener

Signed-off-by: yoyosource <yoyosource@nidido.de>

BauSystem_Main/src/de/steamwar/bausystem/features/simulator/TNTSimulatorListener.java

@@ -32,12 +32,18 @@ import org.bukkit.entity.Player;
 import org.bukkit.event.EventHandler;
 import org.bukkit.event.Listener;
 import org.bukkit.event.player.PlayerInteractEvent;
+import org.bukkit.event.player.PlayerItemHeldEvent;
 import org.bukkit.event.player.PlayerMoveEvent;
 import org.bukkit.event.player.PlayerQuitEvent;
+import org.bukkit.inventory.Inventory;
+import org.bukkit.inventory.ItemStack;
+import org.bukkit.inventory.PlayerInventory;
 import org.bukkit.util.BoundingBox;
 import org.bukkit.util.RayTraceResult;
 import org.bukkit.util.Vector;
 
+import java.util.function.Function;
+
 import static de.steamwar.bausystem.features.simulator.TNTSimulator.WAND;
 import static de.steamwar.bausystem.features.simulator.TNTSimulator.get;
 
@@ -88,14 +94,22 @@ public class TNTSimulatorListener implements Listener {
 
     @EventHandler
     public void onPlayerMove(PlayerMoveEvent e) {
-        Player player = e.getPlayer();
+        simulatorShowHide(e.getPlayer(), PlayerInventory::getItemInMainHand, e.getTo());
+    }
+
+    @EventHandler
+    public void onPlayerItemHeld(PlayerItemHeldEvent e) {
+        simulatorShowHide(e.getPlayer(), i -> i.getItem(e.getNewSlot()), e.getPlayer().getLocation());
+    }
+
+    private void simulatorShowHide(Player player, Function<PlayerInventory, ItemStack> itemStackFunction, Location location) {
         TNTSimulator simulator = get(player);
-        if (!player.getInventory().getItemInMainHand().isSimilar(WAND)) {
+        if (!itemStackFunction.apply(player.getInventory()).isSimilar(WAND)) {
             simulator.hide();
             return;
         }
 
-        simulator.show(trace(player, e.getTo()));
+        simulator.show(trace(player, location));
     }
 
     @EventHandler