Add dispenser place and lit firework mode

Author: fantahund

src/main/java/de/fanta/fancyfirework/FancyFirework.java

@@ -35,6 +35,8 @@ public final class FancyFirework extends JavaPlugin {
     private WorldGuardHelper worldGuardHelper;
     private long time;
     private boolean redstonemode;
+    private boolean dispenserMode;
+    private int dispenserDelaSeconds;
 
     private boolean disableFireWorkUse = false;
 
@@ -132,7 +134,9 @@ public void reloadFireworkConfig() {
     public void loadConfig() {
         FileConfiguration cfg = getConfig();
         cfg.options().copyDefaults(true);
-        redstonemode = cfg.getBoolean("redstonemode");
+        redstonemode = cfg.getBoolean("redstonemode", false);
+        dispenserMode = cfg.getBoolean("dispenserMode", true);
+        dispenserDelaSeconds = cfg.getInt("dispenserDelaSeconds", 30);
 
         //AFK
         AFKListener.maxIdleTime = Math.max(plugin.getConfig().getInt("maxAFKTime"), 2) * 1000;
@@ -142,6 +146,14 @@ public boolean isRedstonemode() {
         return redstonemode;
     }
 
+    public boolean isDispenserMode() {
+        return dispenserMode;
+    }
+
+    public int getDispenserDelaSeconds() {
+        return dispenserDelaSeconds;
+    }
+
     public boolean isDisableFireWorkUse() {
         return disableFireWorkUse;
     }

src/main/java/de/fanta/fancyfirework/fireworks/BlockFireWork.java

@@ -30,6 +30,7 @@
  * There are a couple of methods that will be called on interactions or updates.
  * <ul>
  *     <li>{@link #onPlace(Block, ArmorStand, Player)} - Called when a player places the firework.</li>
+ *     <li>{@link #onDispenserPlace(Block, ArmorStand)}  Called when a dispenser places the firework.</li>
  *     <li>{@link #onLit(Entity, Player)} - Called when the firework is lit by a player using flint and steel.</li>
  *     <li>{@link #onTick(Task, boolean)} - Called when a task of this firework executes a tick.</li>
  * </ul>
@@ -51,6 +52,14 @@ protected BlockFireWork(NamespacedKey key) {
      */
     public abstract void onPlace(Block block, ArmorStand stand, Player player);
 
+    /**
+     * Called when a dispenser places the firework.
+     *
+     * @param block  The placed block.
+     * @param stand  The spawned ArmorStand of the firework.
+     */
+    public abstract void onDispenserPlace(Block block, ArmorStand stand);
+
     /**
      * Called when the firework is lit by a player using flint and steel.
      *
@@ -228,7 +237,6 @@ public void start() {
          * This removes the entity from the world!
          */
         public void stop() {
-            Bukkit.getLogger().info("Tick Stop");
             Entity entity = getEntity();
 
             if (cancellableTask != null) {

src/main/java/de/fanta/fancyfirework/fireworks/defaults/FireWorkBattery.java

@@ -37,6 +37,11 @@ public void onPlace(Block block, ArmorStand stand, Player player) {
         //Nothing
     }
 
+    @Override
+    public void onDispenserPlace(Block block, ArmorStand stand) {
+        //Nothing
+    }
+
     @Override
     public void onLit(Entity entity, Player player) {
         entity.getWorld().playSound(entity.getLocation(), Sound.ENTITY_CREEPER_PRIMED, SoundCategory.AMBIENT, 1f, 1f);

src/main/java/de/fanta/fancyfirework/listners/FireworkListener.java

@@ -18,8 +18,7 @@
 import org.bukkit.block.Block;
 import org.bukkit.block.BlockFace;
 import org.bukkit.block.DoubleChest;
-import org.bukkit.damage.DamageSource;
-import org.bukkit.damage.DamageType;
+import org.bukkit.block.data.Directional;
 import org.bukkit.enchantments.Enchantment;
 import org.bukkit.entity.ArmorStand;
 import org.bukkit.entity.Entity;
@@ -31,6 +30,7 @@
 import org.bukkit.event.EventPriority;
 import org.bukkit.event.Listener;
 import org.bukkit.event.block.Action;
+import org.bukkit.event.block.BlockDispenseEvent;
 import org.bukkit.event.block.BlockPlaceEvent;
 import org.bukkit.event.block.BlockRedstoneEvent;
 import org.bukkit.event.entity.EntityDamageByEntityEvent;
@@ -49,22 +49,27 @@
 import org.bukkit.event.player.PlayerQuitEvent;
 import org.bukkit.inventory.BlockInventoryHolder;
 import org.bukkit.inventory.EquipmentSlot;
+import org.bukkit.inventory.Inventory;
 import org.bukkit.inventory.InventoryHolder;
 import org.bukkit.inventory.ItemStack;
 import org.bukkit.inventory.MerchantRecipe;
 import org.bukkit.inventory.PlayerInventory;
 import org.bukkit.inventory.meta.Damageable;
 import org.bukkit.inventory.meta.ItemMeta;
+import org.bukkit.util.BlockVector;
 
 import java.util.ArrayList;
 import java.util.Collection;
 import java.util.List;
 import java.util.Random;
+import java.util.concurrent.ConcurrentHashMap;
 
 public class FireworkListener implements Listener {
 
     private final FancyFirework plugin = FancyFirework.getPlugin();
 
+    private final ConcurrentHashMap<BlockVector, Long> dispenserPlaceTimes = new ConcurrentHashMap<>();
+
     @EventHandler
     public void onFireworkExplode(FireworkExplodeEvent e) {
         Firework entity = e.getEntity();
@@ -104,6 +109,63 @@ public void onFireWorkPlace(BlockPlaceEvent event) {
         }
     }
 
+    @EventHandler
+    public void onFireWorkPlaceByDispenser(BlockDispenseEvent event) {
+        if (!plugin.isDispenserMode()) {
+            return;
+        }
+
+        ItemStack stack = event.getItem();
+        AbstractFireWork fireWork = plugin.getRegistry().getByItemStack(stack);
+        if (!(fireWork instanceof BlockFireWork blockFireWork)) {
+            return;
+        }
+
+        if (plugin.isDisableFireWorkUse()) {
+            return;
+        }
+
+        Block block = event.getBlock();
+        if (!(block.getState() instanceof org.bukkit.block.Dispenser dispenser)) {
+            return;
+        }
+        event.setCancelled(true);
+
+        plugin.getScheduler().runLocalDelayed(event.getBlock().getLocation(), () -> {
+            BlockVector blockVector = block.getLocation().toVector().toBlockVector();
+            long lastPlaceTime = dispenserPlaceTimes.getOrDefault(blockVector, 0L);
+
+            if (System.currentTimeMillis() - lastPlaceTime < (plugin.getDispenserDelaSeconds() * 1000L)) {
+                return;
+            }
+            dispenserPlaceTimes.remove(blockVector);
+
+            Inventory dispenserInventory = dispenser.getInventory();
+            for (int i = 0; i < dispenserInventory.getSize(); i++) {
+                ItemStack item = dispenserInventory.getItem(i);
+
+                if (item != null && item.isSimilar(stack)) {
+                    if (item.getAmount() > 1) {
+                        item.setAmount(item.getAmount() - 1);
+                        dispenserInventory.setItem(i, item);
+                    } else {
+                        dispenserInventory.setItem(i, null);
+                    }
+                    break;
+                }
+            }
+
+            //Spawn and lit firework
+            if (block.getBlockData() instanceof Directional directional) {
+                Block placeBlock = block.getRelative(directional.getFacing());
+                ArmorStand armorStand = blockFireWork.spawnAtBlock(placeBlock.getRelative(BlockFace.DOWN));
+                blockFireWork.onDispenserPlace(placeBlock, armorStand);
+                blockFireWork.onLit(armorStand, null);
+                dispenserPlaceTimes.put(blockVector, System.currentTimeMillis());
+            }
+        }, 1L);
+    }
+
     @EventHandler(priority = EventPriority.LOW)
     public void onInteract(PlayerInteractAtEntityEvent event) {
         Entity entity = event.getRightClicked();

src/main/resources/config.yml

@@ -2,6 +2,8 @@ enabled: false
 spawnrate: 100
 votereward: false
 redstonemode: false
+dispenserMode: true
+dispenserDelaSeconds: 30
 litplayer: true
 #after X seconds the player is afk (default: 120 Sec = 2 Min
 maxAFKTime: 120